UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 13:40:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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")