Fix errors + set recursion limit to be 10

app.py

@@ -23,6 +23,7 @@ from langgraph.prebuilt import ToolNode, tools_condition
 
 from langchain_openai import ChatOpenAI
 from langchain_core.messages import SystemMessage, HumanMessage, AnyMessage
+from langchain_core.runnables.config import RunnableConfig
 from langchain_core.tools import tool
 from langchain_tavily import TavilySearch
 
@@ -75,7 +76,7 @@ class BasicAgent:
         
         **Answer Format Rules**  
         - If the answer is a number, output digits only (no commas, no units, no strings like “one”, “twenty three”).  
-        - If it’s a word or phrase, don't use articles, neither abbreviations (e.g. for cities).  
+        - If it’s a word or phrase, don't use articles, neither abbreviations (e.g. for cities - New York, not NY).  
         - If it’s a comma separated list, output a comma-separated list following the above rules for each element.  
         - **Always** output exactly one line as an answer and nothing else.
         
@@ -108,7 +109,8 @@ class BasicAgent:
             question=question, 
             messages=[sys_msg, HumanMessage(content=question)]
         )
-        result = self.compiled_graph.invoke(state)
+        config = RunnableConfig(recursion_limit=10)
+        result = self.compiled_graph.invoke(state, config)
         final_answer = result["messages"][-1].content
         print(f"\nFinal Answer: {final_answer}")
         return final_answer, result["messages"]
@@ -129,7 +131,7 @@ class BasicAgent:
     def search_tool(question: str, max_length: int = 100000) -> str:
         print(f"\nCalling search tool with: {question}, max_lentgh: {max_length}")
         search_ = TavilySearch(
-            max_results=5,
+            max_results=3,
             topic="general",
         )
         info = search_.invoke({"query": question})
@@ -300,9 +302,10 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             file_url = f"{api_url}/files/{task_id}"
             file_name = f"task_file_{task_id}"
             with open(file_name, "wb") as file:
-                response = get(file_url)
+                response = requests.get(file_url, timeout=15)
                 file.write(response.content)
-        except:
+        except Exception as e:
+            print(f"Expection occurred while trying to download {file_name} from {file_url}:", e)
             print("Didn't manage to download a file, probably it's not expected for this task")
         
         try: