changed the memory key "query" to "human_message" as it could confuse the agent

updated the sources output to indicate when knowledge comes from the kb
moved the sources cache location to config.ini

app.py

@@ -56,12 +56,15 @@ if __name__ == "__main__":
             documents=[response['output']],
             metadatas=[
                 {
-                    "query":history[-1][0],
-                    "sources":src_list
+                    "human_message":history[-1][0],
+                    "sources": src_list
                 }
             ]
         )
-        response_w_sources = response['output']+"\n\n\n Sources: \n\n\n"+src_list
+        if not sources:
+            response_w_sources = response['output']+"\n\n\n Sources:  \n\n\n Internal knowledge base"
+        else:
+            response_w_sources = response['output']+"\n\n\n Sources: \n\n\n"+src_list
         history[-1][1] = response_w_sources
         return history
 

innovation_pathfinder_ai/config.ini

@@ -1,3 +1,4 @@
 [main]
-VECTOR_DATABASE_LOCATION = innovation_pathfinder_ai/chroma_db/
-CONVERSATION_COLLECTION_NAME = ConversationMemory
+VECTOR_DATABASE_LOCATION = innovation_pathfinder_ai/knowledge_base/
+CONVERSATION_COLLECTION_NAME = ConversationMemory
+SOURCES_CACHE = innovation_pathfinder_ai/database/sources_cache.sqlite3

innovation_pathfinder_ai/database/db_handler.py

@@ -1,8 +1,13 @@
 from sqlmodel import SQLModel, create_engine, Session, select
 from innovation_pathfinder_ai.database.schema import Sources
 from innovation_pathfinder_ai.utils.logger import get_console_logger
+from configparser import ConfigParser
+# from innovation_pathfinder_ai.utils import create_wikipedia_urls_from_text
 
-sqlite_file_name = "innovation_pathfinder_ai/database/database.sqlite3"
+config = ConfigParser()
+config.read('innovation_pathfinder_ai/config.ini')
+
+sqlite_file_name = config.get('main', 'SOURCES_CACHE')
 sqlite_url = f"sqlite:///{sqlite_file_name}"
 engine = create_engine(sqlite_url, echo=False)