tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Releases 2 Packages Activity Actions

feat: vectordb query error handling

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

17
backend/apps/rag/main.py
View file

@ -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):
collection = CHROMA_CLIENT.get_collection( try:
name=collection_name, collection = CHROMA_CLIENT.get_collection(
) 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")