Upload folder using huggingface_hub

db.py

@@ -33,14 +33,24 @@ class SanatanDatabase:
     def search(self, collection_name: str, query: str, n_results=2):
         logger.info("Vector Semantic Search for [%s] in [%s]", query, collection_name)
         collection = self.chroma_client.get_or_create_collection(name=collection_name)
-        response = collection.query(
-            query_embeddings=get_embedding(
-                [query], SanatanConfig().get_embedding_for_collection(collection_name)
-            ),
-            # query_texts=[query],
-            n_results=n_results,
-        )
-        # logger.info("number of matches = %d", len(response["metadatas"]))
+        try:
+            response = collection.query(
+                query_embeddings=get_embedding(
+                    [query], SanatanConfig().get_embedding_for_collection(collection_name)
+                ),
+                # query_texts=[query],
+                n_results=n_results,
+                include=["metadatas","documents","distances"],
+            )
+        except Exception as e:
+            logger.error("Error in search: %s", e)
+            return chromadb.QueryResult(
+                documents=[],
+                ids=[],
+                metadatas=[],
+                distances=[],
+            )
+   
         return response
 
     def search_for_literal(