workspace working
This commit is contained in:
48
verify_indexing.py
Normal file
48
verify_indexing.py
Normal file
@@ -0,0 +1,48 @@
|
||||
import requests
|
||||
import json
|
||||
|
||||
def get_token():
|
||||
resp = requests.get("http://localhost:3015/auth-status")
|
||||
return resp.json()["access_token"]
|
||||
|
||||
def list_documents():
|
||||
token = get_token()
|
||||
url = "http://localhost:3015/documents"
|
||||
headers = {"Authorization": f"Bearer {token}"}
|
||||
params = {"workspace": "test_workspace"}
|
||||
resp = requests.get(url, headers=headers, params=params)
|
||||
print(f"Status: {resp.status_code}")
|
||||
if resp.status_code == 200:
|
||||
data = resp.json()
|
||||
print(f"Documents: {json.dumps(data, indent=2)}")
|
||||
return data
|
||||
else:
|
||||
print(f"Error: {resp.text}")
|
||||
return None
|
||||
|
||||
def search(query):
|
||||
token = get_token()
|
||||
url = "http://localhost:3015/search"
|
||||
headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
|
||||
data = {
|
||||
"workspace": "test_workspace",
|
||||
"query": query,
|
||||
"top_k": 5
|
||||
}
|
||||
resp = requests.post(url, headers=headers, json=data)
|
||||
print(f"Search status: {resp.status_code}")
|
||||
if resp.status_code == 200:
|
||||
results = resp.json()
|
||||
print(f"Search results: {json.dumps(results, indent=2)}")
|
||||
return results
|
||||
else:
|
||||
print(f"Search error: {resp.text}")
|
||||
return None
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("Listing documents...")
|
||||
docs = list_documents()
|
||||
print("\nSearching for 'bee'...")
|
||||
search("bee")
|
||||
print("\nSearching for 'table'...")
|
||||
search("table")
|
||||
Reference in New Issue
Block a user