moved the db.add for agent memory storing to be able to include the sources

app.py

@@ -40,8 +40,6 @@ def initialize_chroma_db() -> Chroma:
 
 if __name__ == "__main__":
     
-    current_id = generate_uuid()
-    
     db = initialize_chroma_db()
     
     def add_text(history, text):
@@ -52,6 +50,17 @@ if __name__ == "__main__":
         response = infer(history[-1][0], history)
         sources = extract_urls(all_sources)
         src_list = '\n'.join(sources)
+        current_id = generate_uuid()
+        db.add(
+            ids=[current_id],
+            documents=[response['output']],
+            metadatas=[
+                {
+                    "query":history[-1][0],
+                    "sources":src_list
+                }
+            ]
+        )
         response_w_sources = response['output']+"\n\n\n Sources: \n\n\n"+src_list
         history[-1][1] = response_w_sources
         return history
@@ -63,19 +72,7 @@ if __name__ == "__main__":
                 "input": question,
                 "chat_history": history
             }
-        )
-        
-        db.add(
-            ids=[current_id],
-            documents=[result['output']],
-            metadatas=[
-                {
-                    "query":query,
-                    "sources":result['sources'].__str__()
-                }
-            ]
-        )
-        
+        )        
         return result
 
     def vote(data: gr.LikeData):

hf_mixtral_agent.py

@@ -71,7 +71,7 @@ agent_executor = AgentExecutor(
     agent=agent, 
     tools=tools, 
     verbose=True,
-    max_iterations=6,       # cap number of iterations
+    max_iterations=10,       # cap number of iterations
     #max_execution_time=60,  # timout at 60 sec
     return_intermediate_steps=True,
     handle_parsing_errors=True,

innovation_pathfinder_ai/utils/utils.py

@@ -32,7 +32,7 @@ def create_wikipedia_urls_from_text(text):
         # Construct the URL and add it to the list
         url = f"https://en.wikipedia.org/wiki/{url_title}"
         urls.append(url)
-        print(urls)
+        #print(urls)
     
     return urls