OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-29 17:52:48 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

fix: error handling

Browse Source
This commit is contained in:
Timothy J. Baek 2024-03-08 23:21:00 -08:00
parent d936353da0
commit 784ee6f521
1 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/main.py
View File

@ -108,6 +108,8 @@ class RAGMiddleware(BaseHTTPMiddleware):
for doc in docs: for doc in docs:
context = None context = None
try:
if doc["type"] == "collection": if doc["type"] == "collection":
context = query_collection( context = query_collection(
collection_names=doc["collection_names"], collection_names=doc["collection_names"],
@ -122,6 +124,10 @@ class RAGMiddleware(BaseHTTPMiddleware):
k=rag_app.state.TOP_K, k=rag_app.state.TOP_K,
embedding_function=rag_app.state.sentence_transformer_ef, embedding_function=rag_app.state.sentence_transformer_ef,
) )
except Exception as e:
print(e)
context = None
relevant_contexts.append(context) relevant_contexts.append(context)
context_string = "" context_string = ""