v0.1.4

src/chromaIntf.py

@@ -11,12 +11,12 @@ import baseInfra.dropbox_handler as dbh
 from baseInfra.dbInterface import DbInterface
 from uuid import UUID
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-import logging
+import logging, asyncio
 
 logger=logging.getLogger("root")
 
 class ChromaIntf():
-    async def __init__(self):
+    def __init__(self):
         self.db_interface=DbInterface()
 
         model_name = "BAAI/bge-large-en-v1.5"
@@ -31,9 +31,10 @@ class ChromaIntf():
         self.persist_db_directory = 'db'
         self.persist_docs_directory = "persistence-docs"
         self.logger_file = "persistence.log"
+        loop=asyncio.get_event_loop()
         try:
-            await dbh.restoreFolder(self.persist_db_directory)
-            await dbh.restoreFolder(self.persist_docs_directory)
+            loop.run_until_complete(dbh.restoreFolder(self.persist_db_directory))
+            loop.run_until_complete(dbh.restoreFolder(self.persist_docs_directory))
         except:
             print("Probably folder doesn't exist as it is brand new setup")
         docs = [