fix: added agent initialisation in the app to be able to load everything.

app.py

@@ -5,7 +5,7 @@ import requests
 import inspect
 import pandas as pd
 import logging
-
+from universal_agent import agent
 # Setup logger
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)

universal_agent.py

@@ -22,7 +22,7 @@ from smolagents import (
     WikipediaSearchTool,
 )
 
-_ = load_dotenv(dotenv_path=find_dotenv(raise_error_if_not_found=True), override=True)
+_ = load_dotenv(dotenv_path=find_dotenv(raise_error_if_not_found=False), override=True)
 nest_asyncio.apply()
 with open("all_questions.pkl", "rb") as f:
     all_questions = pickle.load(f)
@@ -280,29 +280,31 @@ agent = create_react_agent(
 # agent_w_recursion_limit = agent.with_config(recursion_limit=recursion_limit)
 # all_questions[0]
 
-results = []
-for q in all_questions:
-    try:
-        answer = await agent.ainvoke(
-        # answer = agent_w_recursion_limit.invoke(
-            input={
-                "messages": f"""Complete the following task: {q["question"]}. Relevant file: {
-                    q["file_name"]
-                    if q["file_name"]
-                    else "There's no relevant file to use."
-                }"""
-            }
-        )
-        results.append(answer)
-    except GraphRecursionError:
-        print("❌ Agent stopped due to max iterations.")
-        results.append(q["task_id"])
+async def run_agent():
+    results = []
+    for q in all_questions:
+        try:
+            answer = await agent.ainvoke(
+            # answer = agent_w_recursion_limit.invoke(
+                input={
+                    "messages": f"""Complete the following task: {q["question"]}. Relevant file: {
+                        q["file_name"]
+                        if q["file_name"]
+                        else "There's no relevant file to use."
+                    }"""
+                }
+            )
+            results.append(answer)
+        except GraphRecursionError:
+            print("❌ Agent stopped due to max iterations.")
+            results.append(q["task_id"])
+    return results
 
 # with open("results_gpt_mini.pkl", "wb") as f:
 #     pickle.dump(obj=results, file=f, protocol=pickle.HIGHEST_PROTOCOL)
-
-answers = [{"task_id":j['task_id'],
-            "submitted_answer": results[i]["structured_response"].answer
-            if isinstance(results[i], dict) else "No answer"} 
- for i,j in enumerate(all_questions)]
+# results = asyncio.run(run_agent())
+# answers = [{"task_id":j['task_id'],
+#             "submitted_answer": results[i]["structured_response"].answer
+#             if isinstance(results[i], dict) else "No answer"} 
+#  for i,j in enumerate(all_questions)]