Merge pull request #27 from almutareb/retrieve_conv_mem

.gitignore

@@ -171,3 +171,4 @@ obj
 *.zip
 *.faiss
 *.pkl
+*.db

app.py

@@ -61,7 +61,7 @@ if __name__ == "__main__":
             metadatas=[
                 {
                     "human_message":history[-1][0],
-                    "sources": src_list
+                    "sources": 'Internal Knowledge Base From: \n\n' + src_list
                 }
             ]
         )
@@ -70,6 +70,7 @@ if __name__ == "__main__":
         else:
             response_w_sources = response['output']+"\n\n\n Sources: \n\n\n"+src_list
         history[-1][1] = response_w_sources
+        all_sources.clear()
         return history
 
     def infer(question, history):

hf_mixtral_agent.py

@@ -14,7 +14,10 @@ from innovation_pathfinder_ai.structured_tools.structured_tools import (
 from langchain.prompts import PromptTemplate
 from innovation_pathfinder_ai.templates.react_json_with_memory import template_system
 from innovation_pathfinder_ai.utils import logger
+from langchain.globals import set_llm_cache
+from langchain.cache import SQLiteCache
 
+set_llm_cache(SQLiteCache(database_path=".cache.db"))
 logger = logger.get_console_logger("hf_mixtral_agent")
 
 config = load_dotenv(".env")