UnForkRAG/quick_qdrant_test.py at main · iamthebrainman/UnForkRAG · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#!/usr/bin/env python3
"""
Quick test script for UnForkRAG Qdrant API

A simple test to verify basic functionality without extensive logging.
"""

import requests
import json
import time

def quick_test():
    base_url = "http://localhost:8000/qdrant"
    collection_name = f"quick-test-{int(time.time())}"

    print("🧪 Quick Qdrant API Test")
    print(f"Collection: {collection_name}")

    # 1. Health check
    print("1. Health check...", end=" ")
    r = requests.get(f"{base_url}/")
    print("✅" if r.status_code == 200 else f"❌ {r.status_code}")

    # 2. Create collection
    print("2. Create collection...", end=" ")
    r = requests.put(f"{base_url}/collections/{collection_name}", json={})
    print("✅" if r.status_code == 200 else f"❌ {r.status_code}")

    # 3. Add points
    print("3. Add points...", end=" ")
    points = {
        "points": [
            {"id": "1", "payload": {"text": "Hello world", "type": "greeting"}},
            {"id": "2", "payload": {"text": "Python programming", "type": "code"}},
            {"id": "3", "payload": {"text": "Vector search", "type": "tech"}}
        ]
    }
    r = requests.put(f"{base_url}/collections/{collection_name}/points", json=points)
    print("✅" if r.status_code == 200 else f"❌ {r.status_code}")

    # 4. Search
    print("4. Search points...", end=" ")
    search = {"query_text": "programming", "limit": 2}
    r = requests.post(f"{base_url}/collections/{collection_name}/points/search", json=search)
    if r.status_code == 200:
        results = r.json().get("result", [])
        print(f"✅ Found {len(results)} results")
    else:
        print(f"❌ {r.status_code}")

    # 5. Query points
    print("5. Query points...", end=" ")
    query = {"query_text": "hello", "limit": 1}
    r = requests.post(f"{base_url}/collections/{collection_name}/points/query", json=query)
    if r.status_code == 200:
        points = r.json().get("result", {}).get("points", [])
        print(f"✅ Found {len(points)} points")
    else:
        print(f"❌ {r.status_code}")

    # 6. Delete points
    print("6. Delete points...", end=" ")
    delete = {"points": ["1", "2"]}
    r = requests.post(f"{base_url}/collections/{collection_name}/points/delete", json=delete)
    if r.status_code == 200:
        deleted = r.json().get("result", {}).get("deleted_count", 0)
        print(f"✅ Deleted {deleted} points")
    else:
        print(f"❌ {r.status_code}")

    # 7. Delete collection
    print("7. Delete collection...", end=" ")
    r = requests.delete(f"{base_url}/collections/{collection_name}")
    print("✅" if r.status_code == 200 else f"❌ {r.status_code}")

    print("\n🎉 Quick test complete!")

if __name__ == "__main__":
    try:
        quick_test()
    except requests.exceptions.RequestException as e:
        print(f"❌ Connection error: {e}")
        print("Make sure the server is running: python unfork_server.py --port 8000")