v0.8.4

src/chromaIntf.py

@@ -70,11 +70,15 @@ class ChromaIntf():
         )
 
 
-    async def getRelevantDocs(self,query:str,count:int=8):
+    async def getRelevantDocs(self,query:str,kwargs:dict):
         """This should also post the result to firebase"""
         print("retriver state",self.retriever.search_kwargs)
         print("retriver state",self.retriever.search_type)
-        self.retriever.search_kwargs["k"]=count
+        try:
+            for key in kwargs.keys():
+                self.retriever.search_kwargs[key]=kwargs[key]
+        except:
+            print("setting search args failed")
         retVal=await self.retriever.aget_relevant_documents(query)
         value=[]
         excludeMeta=True
@@ -125,10 +129,10 @@ class ChromaIntf():
         with open("./docs/"+metadata['ID'],"w") as fd:
             fd.write(inStr)
         try:
-            return await self.vectorstore.add_documents(docs,ids=[metadata.ID])
+            return await self.vectorstore.aadd_documents(docs,ids=[metadata.ID])
         except:
             print("inside expect of addText")
-            return await self.vectorstore.add_documents(docs,ids=[metadata['ID']])
+            return await self.vectorstore.aadd_documents(docs,ids=[metadata['ID']])
         
     async def listDocs(self):
         collection=self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)

src/main.py

@@ -60,13 +60,14 @@ app.add_middleware(
 api_base="/api/v1"
 
 @app.post(api_base+"/getMatchingDocs")
-async def get_matching_docs(inStr: str ) -> Any:
+async def get_matching_docs(inStr: str, kwargs: Dict [Any, Any] ) -> Any:
     """
     Gets the query embeddings and uses metadata appropriately and gets the matching docs for query
     TODO: Add parameter for type of query and number of docs to return
     TODO: Add parameter to return the source information as well
     """
-    return await chromaIntf.getRelevantDocs(inStr)
+    print("Reaching get matchin docs ")
+    return await chromaIntf.getRelevantDocs(inStr,kwargs)
 
 @app.post(api_base+"/addTextDocument")
 async def add_text_document(inDoc: DocWithMeta ) -> Any: