55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
import asyncio
|
|
import aiohttp
|
|
import json
|
|
|
|
async def clear_documents(workspace):
|
|
url = "http://localhost:3015/documents"
|
|
headers = {
|
|
"X-API-Key": "jleu1212",
|
|
"X-Workspace": workspace,
|
|
"accept": "application/json",
|
|
}
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.delete(url, headers=headers) as resp:
|
|
print(f"Clear status: {resp.status}")
|
|
text = await resp.text()
|
|
print(f"Clear response: {text}")
|
|
return resp.status, text
|
|
|
|
async def search(workspace):
|
|
url = "http://localhost:3015/search"
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
"X-Workspace": workspace,
|
|
"X-API-Key": "jleu1212"
|
|
}
|
|
data = {
|
|
"query": "what is the minimum safe working distance",
|
|
"mode": "mix",
|
|
"return_raw_data": True,
|
|
"use_keywords": False
|
|
}
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.post(url, headers=headers, json=data) as resp:
|
|
print(f"Search status: {resp.status}")
|
|
if resp.status == 200:
|
|
result = await resp.json()
|
|
print(json.dumps(result, indent=2))
|
|
else:
|
|
text = await resp.text()
|
|
print(f"Search error: {text}")
|
|
return resp.status
|
|
|
|
async def main():
|
|
workspace = "test1"
|
|
print("Clearing documents...")
|
|
status, _ = await clear_documents(workspace)
|
|
if status == 200:
|
|
print("Clear succeeded.")
|
|
else:
|
|
print("Clear failed.")
|
|
print("Searching after clear...")
|
|
await search(workspace)
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |