init.

.env.example

@@ -0,0 +1,2 @@
+OPENAI_API_KEY=
+ANTHROPIC_API_KEY=

.gitignore

@@ -0,0 +1 @@
+.env

agent.py

@@ -0,0 +1,51 @@
+import os
+import pathlib
+
+import dotenv
+from smolagents import (
+    CodeAgent,
+    OpenAIServerModel,
+    VisitWebpageTool,
+    WebSearchTool,
+    WikipediaSearchTool,
+)
+
+if pathlib.Path(".env").exists():
+    dotenv.load_dotenv(".env")
+
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", None)
+ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY", None)
+
+if not OPENAI_API_KEY and not ANTHROPIC_API_KEY:
+    raise ValueError(
+        "Please set either OPENAI_API_KEY or ANTHROPIC_API_KEY in your .env file or Huggingface Space settings."
+    )
+
+model = OpenAIServerModel(
+    model_id="gpt-4.1",
+    api_key=OPENAI_API_KEY,
+)
+
+web_search_agent = CodeAgent(
+    model=model,
+    tools=[
+        WebSearchTool(),
+        VisitWebpageTool(),
+        WikipediaSearchTool(),
+    ],
+)
+
+calc_agent = CodeAgent(
+    model=model,
+    additional_authorized_imports=["pandas", "numpy"]
+)
+
+class Agent:
+    def __init__(self):
+        print("Agent initialized.")
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        fixed_answer = "This is a default answer."
+        print(f"Agent returning fixed answer: {fixed_answer}")
+        return fixed_answer

app.py

@@ -5,6 +5,8 @@ import gradio as gr
 import pandas as pd
 import requests
 
+from agent import Agent
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -12,20 +14,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
-    def __init__(self):
-        print("BasicAgent initialized.")
-
-    def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
-
-
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
+    Fetches all questions, runs the Agent on them, submits all answers,
     and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
@@ -44,7 +35,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = Agent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None

prompt.py

@@ -0,0 +1,39 @@
+from smolagents.agents import (
+    FinalAnswerPromptTemplate,
+    PlanningPromptTemplate,
+    PromptTemplates,
+)
+
+web_search_agent_prompt = PromptTemplates(
+    system_prompt="""You are a web search agent. Your task is to find information on the web to answer the user's question.
+You can use web search and visit webpages to gather information. Plan your actions carefully to find the
+most relevant information.
+""",
+        planning=PlanningPromptTemplate(
+            "You are a web search agent. Your task is to find information on the web to answer the user's question. "
+            "You can use web search and visit webpages to gather information. "
+            "Plan your actions carefully to find the most relevant information.",
+            max_steps=5,
+        ),
+        final_answer=FinalAnswerPromptTemplate(
+            "Based on the information you found, provide a final answer to the user's question. "
+            "Make sure your answer is clear and concise. "
+            "If you couldn't find enough information, explain why and what you would do next.",
+            max_steps=3,
+        ),
+    )
+
+calc_agent_prompt = PromptTemplates(
+        planning=PlanningPromptTemplate(
+            "You are a calculator agent. Your task is to perform calculations or data analysis based on "
+            "the user's question. You can use libraries like pandas and numpy to assist you. "
+            "Plan your actions carefully to ensure accurate results.",
+            max_steps=5,
+        ),
+        final_answer=FinalAnswerPromptTemplate(
+            "Based on your calculations or data analysis, provide a final answer to the user's question. "
+            "Make sure your answer is clear and concise. "
+            "If you encountered any issues, explain what they were and how you would resolve them.",
+            max_steps=3,
+        ),
+    )

pyproject.toml

@@ -7,5 +7,6 @@ requires-python = ">=3.11"
 dependencies = [
     "gradio>=5.34.0",
     "requests>=2.32.4",
+    "smolagents>=1.18.0",
     "types-requests>=2.32.4.20250611",
 ]

uv.lock

@@ -7,6 +7,25 @@ resolution-markers = [
     "python_full_version < '3.12'",
 ]
 
+[[package]]
+name = "agents-course-final-assignment"
+version = "0.1.0"
+source = { virtual = "." }
+dependencies = [
+    { name = "gradio" },
+    { name = "requests" },
+    { name = "smolagents" },
+    { name = "types-requests" },
+]
+
+[package.metadata]
+requires-dist = [
+    { name = "gradio", specifier = ">=5.34.0" },
+    { name = "requests", specifier = ">=2.32.4" },
+    { name = "smolagents", specifier = ">=1.18.0" },
+    { name = "types-requests", specifier = ">=2.32.4.20250611" },
+]
+
 [[package]]
 name = "aiofiles"
 version = "24.1.0"
@@ -189,23 +208,6 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/4d/36/2a115987e2d8c300a974597416d9de88f2444426de9571f4b59b2cca3acc/filelock-3.18.0-py3-none-any.whl", hash = "sha256:c401f4f8377c4464e6db25fff06205fd89bdd83b65eb0488ed1b160f780e21de", size = 16215, upload-time = "2025-03-14T07:11:39.145Z" },
 ]
 
-[[package]]
-name = "final-assignment-template"
-version = "0.1.0"
-source = { virtual = "." }
-dependencies = [
-    { name = "gradio" },
-    { name = "requests" },
-    { name = "types-requests" },
-]
-
-[package.metadata]
-requires-dist = [
-    { name = "gradio", specifier = ">=5.34.0" },
-    { name = "requests", specifier = ">=2.32.4" },
-    { name = "types-requests", specifier = ">=2.32.4.20250611" },
-]
-
 [[package]]
 name = "fsspec"
 version = "2025.5.1"
@@ -772,6 +774,15 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/ec/57/56b9bcc3c9c6a792fcbaf139543cee77261f3651ca9da0c93f5c1221264b/python_dateutil-2.9.0.post0-py2.py3-none-any.whl", hash = "sha256:a8b2bc7bffae282281c8140a97d3aa9c14da0b136dfe83f850eea9a5f7470427", size = 229892, upload-time = "2024-03-01T18:36:18.57Z" },
 ]
 
+[[package]]
+name = "python-dotenv"
+version = "1.1.0"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/88/2c/7bb1416c5620485aa793f2de31d3df393d3686aa8a8506d11e10e13c5baf/python_dotenv-1.1.0.tar.gz", hash = "sha256:41f90bc6f5f177fb41f53e87666db362025010eb28f60a01c9143bfa33a2b2d5", size = 39920, upload-time = "2025-03-25T10:14:56.835Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/1e/18/98a99ad95133c6a6e2005fe89faedf294a748bd5dc803008059409ac9b1e/python_dotenv-1.1.0-py3-none-any.whl", hash = "sha256:d7c01d9e2293916c18baf562d95698754b0dbbb5e74d457c45d4f6561fb9d55d", size = 20256, upload-time = "2025-03-25T10:14:55.034Z" },
+]
+
 [[package]]
 name = "python-multipart"
 version = "0.0.20"
@@ -917,6 +928,23 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/b7/ce/149a00dd41f10bc29e5921b496af8b574d8413afcd5e30dfa0ed46c2cc5e/six-1.17.0-py2.py3-none-any.whl", hash = "sha256:4721f391ed90541fddacab5acf947aa0d3dc7d27b2e1e8eda2be8970586c3274", size = 11050, upload-time = "2024-12-04T17:35:26.475Z" },
 ]
 
+[[package]]
+name = "smolagents"
+version = "1.18.0"
+source = { registry = "https://pypi.org/simple" }
+dependencies = [
+    { name = "huggingface-hub" },
+    { name = "jinja2" },
+    { name = "pillow" },
+    { name = "python-dotenv" },
+    { name = "requests" },
+    { name = "rich" },
+]
+sdist = { url = "https://files.pythonhosted.org/packages/00/31/f4aa75afd9a98724932fef9b6f1356e9305a22359a0a32eff48e0ee6cb42/smolagents-1.18.0.tar.gz", hash = "sha256:2a44bd9ac85a2c47102bf30c1bcf32e5d399e1f9d66bb9518cc5fcb30879660a", size = 181958, upload-time = "2025-06-10T14:14:34.633Z" }
+wheels = [
+    { url = "https://files.pythonhosted.org/packages/d3/28/493fcf6fb5f7c4ae21b9541fd588fbc69a22c22d3d9477771554a0779006/smolagents-1.18.0-py3-none-any.whl", hash = "sha256:c8f6d0c37808090213f45ec742be7f818e7f4772516f6e776bf991936ecabaf8", size = 136036, upload-time = "2025-06-10T14:14:32.949Z" },
+]
+
 [[package]]
 name = "sniffio"
 version = "1.3.1"