v0.8.4

src/chromaIntf.py

@@ -9,6 +9,8 @@ from llm.llmFactory import LLMFactory
 from datetime import datetime
 import baseInfra.dropbox_handler as dbh
 from baseInfra.dbInterface import DbInterface
+from uuid import UUID
+
 
 class ChromaIntf():
     def __init__(self):
@@ -96,7 +98,7 @@ class ChromaIntf():
         return retVal
     
 
-    def addText(self,inStr:str,metadata):
+    async def addText(self,inStr:str,metadata):
         metadata=metadata.dict()
         if "timestamp" not in metadata.keys():
             metadata['timestamp']=datetime.now().isoformat()
@@ -117,17 +119,25 @@ class ChromaIntf():
         docs = [
             Document(page_content=inStr, metadata=metadata)]
         try:
-            return self.vectorstore.add_documents(docs,ids=[metadata.ID])
+            return await self.vectorstore.add_documents(docs,ids=[metadata.ID])
         except:
             print("inside expect of addText")
-            return self.vectorstore.add_documents(docs,ids=[metadata['ID']])
+            return await self.vectorstore.add_documents(docs,ids=[metadata['ID']])
         
     async def listDocs(self):
-        return self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)
+        collectionInfo=self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)[0]
+        return self.vectorstore._client._get(collection_id=self._uuid(collectionInfo.id))
         
         
     async def persist(self):
         await self.vectorstore.persist()
-        await dbh.backupFolder("db")
+        return await dbh.backupFolder("db")
+    
+    def _uuid(uuid_str: str) -> UUID:
+        try:
+            return UUID(uuid_str)
+        except ValueError:
+            print("Error generating uuid")
+            raise ValueError(f"Could not parse {uuid_str} as a UUID")
 
     

src/main.py

@@ -36,8 +36,8 @@ async def catch_exceptions_middleware(
     request: Request, call_next: Callable[[Request], Any]
 ) -> Response:
     try:
-        print("In exception cater middleware")
-        print(request.headers)
+        #print("In exception cater middleware")
+        #print(request.headers)
         #print(await request.body())
         return await call_next(request)
     except Exception as e:
@@ -66,7 +66,7 @@ async def get_matching_docs(inStr: str ) -> Any:
     TODO: Add parameter for type of query and number of docs to return
     TODO: Add parameter to return the source information as well
     """
-    return chromaIntf.getRelevantDocs(inStr)
+    return await chromaIntf.getRelevantDocs(inStr)
 
 @app.post(api_base+"/addTextDocument")
 async def add_text_document(inDoc: DocWithMeta ) -> Any:
@@ -75,7 +75,7 @@ async def add_text_document(inDoc: DocWithMeta ) -> Any:
     """
     print("Received request for")
     print(inDoc) 
-    return chromaIntf.addText(inDoc.text,inDoc.metadata)
+    return await chromaIntf.addText(inDoc.text,inDoc.metadata)
 
 @app.get(api_base+"/persist")
 async def persist_db():