fix: error handling

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 = ""