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

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

34
backend/main.py
View file

@ -108,20 +108,26 @@ class RAGMiddleware(BaseHTTPMiddleware):
for doc in docs: for doc in docs:
context = None context = None
if doc["type"] == "collection":
context = query_collection( try:
collection_names=doc["collection_names"], if doc["type"] == "collection":
query=query, context = query_collection(
k=rag_app.state.TOP_K, collection_names=doc["collection_names"],
embedding_function=rag_app.state.sentence_transformer_ef, query=query,
) k=rag_app.state.TOP_K,
else: embedding_function=rag_app.state.sentence_transformer_ef,
context = query_doc( )
collection_name=doc["collection_name"], else:
query=query, context = query_doc(
k=rag_app.state.TOP_K, collection_name=doc["collection_name"],
embedding_function=rag_app.state.sentence_transformer_ef, query=query,
) k=rag_app.state.TOP_K,
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 = ""