Commit 14226937 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

feat: vectordb query error handling

parent ad3d69be
...@@ -85,12 +85,19 @@ async def get_status(): ...@@ -85,12 +85,19 @@ async def get_status():
@app.get("/query/{collection_name}") @app.get("/query/{collection_name}")
def query_collection(collection_name: str, query: str, k: Optional[int] = 4): def query_collection(collection_name: str, query: str, k: Optional[int] = 4):
try:
collection = CHROMA_CLIENT.get_collection( collection = CHROMA_CLIENT.get_collection(
name=collection_name, name=collection_name,
) )
result = collection.query(query_texts=[query], n_results=k) 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") @app.post("/web")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment