feat: vectordb query error handling

This commit is contained in:
Timothy J. Baek 2024-01-07 01:59:00 -08:00
parent ad3d69be30
commit 142269374f
1 changed files with 12 additions and 5 deletions

View File

@ -85,12 +85,19 @@ async def get_status():
@app.get("/query/{collection_name}")
def query_collection(collection_name: str, query: str, k: Optional[int] = 4):
collection = CHROMA_CLIENT.get_collection(
name=collection_name,
)
result = collection.query(query_texts=[query], n_results=k)
try:
collection = CHROMA_CLIENT.get_collection(
name=collection_name,
)
result = collection.query(query_texts=[query], n_results=k)
return result
return result
except Exception as e:
print(e)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.DEFAULT(e),
)
@app.post("/web")