Recussion_limit

app.py

@@ -36,7 +36,7 @@ class BasicAgent:
         
         # IMPORTANT: invoke() returns a **new** state instance (or an AddableValuesDict),
         # not the object we pass in.  Use the returned value to fetch final_answer.
-        out_state = self.graph.invoke(init_state)
+        out_state = self.graph.invoke(init_state, {"recursion_limit": 5})
         print("out_state: ", out_state,'\n\n\n\n')
         
         # Extract the final answer from the last message

tools.py

@@ -209,13 +209,17 @@ def wikipedia_search_tool(wiki_query: str) -> str:
     """
     print("reached wikipedia search tool")
     query = wiki_query
-    docs = WikipediaLoader(query=query, load_max_docs=5).load()
-    result = ""
-    counter = 1
-    for doc in docs:
-        result += f"\n\nDocument{counter}: {doc.metadata['title']}\n. {doc.page_content}"
-        counter += 1
-    return result
+    try:
+        docs = WikipediaLoader(query=query, load_max_docs=5).load()
+        result = ""
+        counter = 1
+        for doc in docs:
+            result += f"\n\nDocument{counter}: {doc.metadata['title']}\n. {doc.page_content}"
+            counter += 1
+        return result
+    except Exception as e:
+        print(f"Error during Wikipedia search: {e}")
+        return "Wikipedia search failed, try another tool or query"
     
 @tool
 def arxiv_search_tool(arxiv_query: str) -> str:
@@ -225,13 +229,17 @@ def arxiv_search_tool(arxiv_query: str) -> str:
     Returns: text summary of first matching page or an error message>"
     """
     print("reached arxiv_search_tool")
-    docs = ArxivLoader(query=arxiv_query, load_max_docs=5).load()
-    result = ""
-    counter = 1
-    for doc in docs:
-        result += f"\n\nDocument{counter}: {doc.metadata['title']}\n. {doc.page_content}"
-        counter += 1
-    return result
+    try:
+        docs = ArxivLoader(query=arxiv_query, load_max_docs=5).load()
+        result = ""
+        counter = 1
+        for doc in docs:
+            result += f"\n\nDocument{counter}: {doc.metadata['title']}\n. {doc.page_content}"
+            counter += 1
+        return result
+    except Exception as e:
+        print(f"Error during Arxiv search: {e}")
+        return "Arxiv search failed, try another tool or query"
 
 
 from langchain_openai import ChatOpenAI