fix: error handling

784ee6f5 · Timothy J. Baek · d936353d · 784ee6f5
Commit 784ee6f5 authored Mar 08, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 20 additions and 14 deletions

backend/main.py backend/main.py +20 -14

No files found.
--- a/backend/main.py
+++ b/backend/main.py
@@ -108,20 +108,26 @@ class RAGMiddleware(BaseHTTPMiddleware):
                for doc in docs:
                    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)
                context_string = ""