v0.8.4

src/chromaIntf.py

@@ -17,7 +17,7 @@ class ChromaIntf():
         model_name = "BAAI/bge-large-en-v1.5"
         encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
 
-        embedding = HuggingFaceBgeEmbeddings(
+        self.embedding = HuggingFaceBgeEmbeddings(
                 model_name=model_name,
                 model_kwargs={'device': 'cpu'},  
                 encode_kwargs=encode_kwargs
@@ -37,8 +37,9 @@ class ChromaIntf():
             ]
 
         self.vectorstore = Chroma.from_documents(documents=docs,
-                                  embedding=embedding,
+                                  embedding=self.embedding,
                                   persist_directory=persist_db_directory)
+        #self.vectorstore._client.
 
         self.metadata_field_info = [
             AttributeInfo(
@@ -121,7 +122,12 @@ class ChromaIntf():
             print("inside expect of addText")
             return self.vectorstore.add_documents(docs,ids=[metadata['ID']])
         
-    def persist(self):
-        self.vectorstore.persist()
+    async def listDocs(self):
+        return self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)
+        
+        
+    async def persist(self):
+        await self.vectorstore.persist()
+        await dbh.backupFolder("db")
 
     

src/main.py

@@ -79,8 +79,7 @@ async def add_text_document(inDoc: DocWithMeta ) -> Any:
 
 @app.get(api_base+"/persist")
 async def persist_db():
-    chromaIntf.persist()
-    return await dbh.backupFolder("db")
+    return await chromaIntf.persist()
 
 @app.get(api_base+"/reset")
 async def reset_db():
@@ -102,6 +101,10 @@ def walk(path: PathRequest):
     response= JSONResponse(content= {"dirs":dirs,"files":fileList})
     return response
 
+@app.get(api_base+"/list")
+async def list_collection():
+    return await chromaIntf.listDocs()
+
 print(__name__)
 
 if __name__ == '__main__' or __name__ == "src.main":