fix: error handling

mmartel86 · Mar 9, 2024 · 784ee6f · 784ee6f
1 parent d936353
commit 784ee6f
Showing 1 changed file with 20 additions and 14 deletions.
diff --git a/backend/main.py b/backend/main.py
@@ -108,20 +108,26 @@ async def dispatch(self, request: Request, call_next):
 
                 for doc in docs:
                     context = None
-                    if doc["type"] == "collection":
-                        context = query_collection(
-                            collection_names=doc["collection_names"],
-                            query=query,
-                            k=rag_app.state.TOP_K,
-                            embedding_function=rag_app.state.sentence_transformer_ef,
-                        )
-                    else:
-                        context = query_doc(
-                            collection_name=doc["collection_name"],
-                            query=query,
-                            k=rag_app.state.TOP_K,
-                            embedding_function=rag_app.state.sentence_transformer_ef,
-                        )
+
+                    try:
+                        if doc["type"] == "collection":
+                            context = query_collection(
+                                collection_names=doc["collection_names"],
+                                query=query,
+                                k=rag_app.state.TOP_K,
+                                embedding_function=rag_app.state.sentence_transformer_ef,
+                            )
+                        else:
+                            context = query_doc(
+                                collection_name=doc["collection_name"],
+                                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)
 
                 context_string = ""