v0.1.7

src/chroma_intf.py

@@ -8,6 +8,9 @@ from langchain.chains.query_constructor.base import AttributeInfo
 from llm.llmFactory import LLMFactory
 from datetime import datetime
 import baseInfra.dropbox_handler as dbh
+from baseInfra.dbInterface import DbInterface
+
+db_interface=DbInterface()
 
 model_name = "BAAI/bge-large-en-v1.5"
 encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
@@ -113,8 +116,20 @@ meta_defaults={
     "source":"conversation",
     "ID":datetime.now().strftime("%Y-%m-%d %H:%M:%S::%f")+"-conversation"
 }
+
 def getRelevantDocs(query:str):
-    return retriever.get_relevant_documents(query)
+    """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+=v+"\n"
+    db_interface.add_to_cache(input=query,value=value)
+    return retVal
+    
 
 def addText(inStr:str,metadata):
     md=meta_defaults
@@ -123,7 +138,6 @@ def addText(inStr:str,metadata):
     md['timestamp']=datetime.now().strftime("%Y-%m-%d %H:%M:%S::%f")
     md['ID']=datetime.now().strftime("%Y-%m-%d %H:%M:%S::%f")+"-conversation"
 
-
     docs = [
         Document(page_content=inStr, metadata=md)]
     try: