Fixed runtime errors.

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from agent.basic_agent import BasicAgent
+from src.agent.basic_agent import BasicAgent
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"

src/agent/basic_agent.py

@@ -30,7 +30,7 @@ class BasicAgent:
         self.graph = builder.compile()
         print("BasicAgent initialized.")
     
-    def __call__(self, question: str) -> MessagesState:
+    def __call__(self, question: str) -> str:
         print(f"Agent received question...")
         message = HumanMessage(content=question)
         answer = self.graph.invoke({"messages": [message]})

src/agent/tools/__init__.py

@@ -1,8 +1,10 @@
 from .math_tools import math_tools
 from .web_tools import web_tools
+from .coding_tools import code_agent
 
 # Consolidate all tools into a single exportable array
-tools = web_tools + math_tools
+coding_tools = [code_agent]
+tools = web_tools + math_tools + coding_tools
 
 # For backward compatibility, also export individual tool categories
-__all__ = ['tools', 'math_tools', 'web_tools'] 
+__all__ = ['tools', 'math_tools', 'web_tools', 'coding_tools'] 

src/agent/tools/coding_tools.py

@@ -1,9 +1,9 @@
 from langchain_core.tools import tool
-from smolagents import CodeAgent, HfApiModel
+from smolagents import CodeAgent, InferenceClientModel
 
 
 imports = ['datetime', 'pandas', 'numpy', 'python-chess', 'set', 'math']
-agent = CodeAgent(tools=[], model=HfApiModel(), additional_authorized_imports=imports, max_steps=10,)
+agent = CodeAgent(tools=[], model=InferenceClientModel(), additional_authorized_imports=imports, max_steps=10,)
 
 @tool
 def code_agent(question: str) -> str: