abc

app.py

@@ -38,23 +38,43 @@ class SmolGaiaAgent:
             api_key=os.getenv("HF_TOKEN"),
         )
 
-        # 2) CodeAgent with system_prompt in initialization
-        self.agent = CodeAgent(
-            tools=[],
-            add_base_tools=True,
-            model=self.model,
-            max_steps=8,
-            name="gaia_code_agent",
-            description="Agent that uses web search and python to solve GAIA level 1 questions.",
-            system_prompt=GAIA_SYSTEM_PROMPT,
-        )
+        # 2) CodeAgent - try with system_prompt, fallback without if it fails
+        try:
+            self.agent = CodeAgent(
+                tools=[],
+                add_base_tools=True,
+                model=self.model,
+                max_steps=8,
+                system_prompt=GAIA_SYSTEM_PROMPT,
+            )
+            print("Agent initialized with system_prompt parameter")
+        except TypeError as e:
+            # If system_prompt is not supported, initialize without it
+            print(f"system_prompt not supported in this version, initializing without it: {e}")
+            self.agent = CodeAgent(
+                tools=[],
+                add_base_tools=True,
+                model=self.model,
+                max_steps=8,
+            )
+            # We'll prepend the system prompt to each task instead
+            self.use_task_prefix = True
+        else:
+            self.use_task_prefix = False
 
     def __call__(self, question: str) -> str:
         """
         Runs the CodeAgent on one question and returns the final answer string.
         """
         print(f"[SmolGaiaAgent] Question: {question[:80]}...")
-        answer = self.agent.run(question)
+        
+        # If system_prompt wasn't supported during init, prepend it to the task
+        if hasattr(self, 'use_task_prefix') and self.use_task_prefix:
+            task = f"{GAIA_SYSTEM_PROMPT}\n\nTask: {question}"
+        else:
+            task = question
+            
+        answer = self.agent.run(task)
         answer = str(answer).strip()
         print(f"[SmolGaiaAgent] Answer: {answer}")
         return answer
@@ -89,6 +109,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         agent = SmolGaiaAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
+        import traceback
+        traceback.print_exc()
         return f"Error initializing agent: {e}", None
     
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
@@ -131,6 +153,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")
+             import traceback
+             traceback.print_exc()
              results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
 
     if not answers_payload:
@@ -204,7 +228,6 @@ with gr.Blocks() as demo:
         """
     )
 
-    # Store the login button to access its value
     login_button = gr.LoginButton()
 
     run_button = gr.Button("Run Evaluation & Submit All Answers")
@@ -212,8 +235,6 @@ with gr.Blocks() as demo:
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
 
-    # The key is to NOT specify inputs - Gradio will automatically inject the profile
-    # when the function signature includes the profile parameter
     run_button.click(
         fn=run_and_submit_all,
         outputs=[status_output, results_table]