v0.1.71

src/chroma_intf.py

@@ -23,7 +23,7 @@ embedding = HuggingFaceBgeEmbeddings(
 
 persist_directory = 'db'
 try:
-    dbh.restoreFolder("./db")
+    dbh.restoreFolder("db")
 except:
     print("Probably folder doesn't exist as it is brand new setup")
 docs = [
@@ -121,13 +121,22 @@ def getRelevantDocs(query:str):
     """This should also post the result to firebase"""
     retVal=retriever.get_relevant_documents(query)
     value=[]
-    for item in retVal:
-        v="Info:"+item['page_content']+" "
-        for key in item.metadata.keys():
-            if key != "ID":
-                v+=key+":"+item.metadata[key]+" "
-        value.append(v)
-    db_interface.add_to_cache(input=query,value=value)
+    try:
+        for item in retVal:
+            v="Info:"+item['page_content']+" "
+            for key in item.metadata.keys():
+                if key != "ID":
+                    v+=key+":"+item.metadata[key]+" "
+            value.append(v)
+        db_interface.add_to_cache(input=query,value=value)
+    except:
+        for item in retVal:
+            v="Info:"+item.page_content+" "
+            for key in item.metadata.keys():
+                if key != "ID":
+                    v+=key+":"+item.metadata[key]+" "
+            value.append(v)
+        db_interface.add_to_cache(input=query,value=value)
     return retVal
     
 

src/main.py

@@ -58,11 +58,11 @@ async def add_text_document(inStr: str,metadata: dict ) -> Any:
 @app.get(api_base+"/persist")
 async def persist_db():
     persist()
-    return await dbh.backupFolder("./db")
+    return await dbh.backupFolder("db")
 
 @app.get(api_base+"/reset")
 async def reset_db():
-    return await dbh.restoreFolder("./db")
+    return await dbh.restoreFolder("db")
 
 print(__name__)