fix: error handling

2024-03-08 23:21:00 -08:00 · 2024-03-08 23:21:00 -08:00 · 784ee6f521
commit 784ee6f521
parent d936353da0
1 changed files with 20 additions and 14 deletions
--- a/backend/main.py
+++ b/backend/main.py
@ -108,6 +108,8 @@ class RAGMiddleware(BaseHTTPMiddleware):

                for doc in docs:
                    context = None
+
+                    try:
                        if doc["type"] == "collection":
                            context = query_collection(
                                collection_names=doc["collection_names"],
@ -122,6 +124,10 @@ class RAGMiddleware(BaseHTTPMiddleware):
                                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 = ""