refactor BasicAgent initialization to include AgentName and update requirements.txt to include huggingface-hub

app.py

@@ -1,15 +1,14 @@
 import os
 import gradio as gr
 import requests
-import inspect
 import pandas as pd
 from smolagents import (
     CodeAgent,
     DuckDuckGoSearchTool,
-    InferenceClientModel,
     VisitWebpageTool,
     PythonInterpreterTool,
     WikipediaSearchTool,
+    LiteLLMModel
 )
 
 
@@ -17,10 +16,18 @@ from smolagents import (
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+
+model = LiteLLMModel(
+    model_id="gemini/gemini-2.5-flash",
+    temperature=0.2
+)
+
 class BasicAgent:
-    def __init__(self):
+    def __init__(self, name: str = "GGSAgent"):
+        self.name = name
         self.code_agent = CodeAgent(
             tools=[
                 DuckDuckGoSearchTool(),
@@ -28,8 +35,10 @@ class BasicAgent:
                 PythonInterpreterTool(),
                 WikipediaSearchTool(),
             ],
-            model=InferenceClientModel("Qwen/Qwen2.5-Coder-32b-instruct"),
-            max_steps=20,
+            model=model,
+            max_steps=15,
+            verbosity_level=1,
+            additional_authorized_imports=["json", "math"]
         )
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
@@ -64,7 +73,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent_name = os.getenv("AGENT_NAME", "GGSAgent")
+        agent = BasicAgent(name=agent_name)
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -115,8 +125,13 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
-    # 4. Prepare Submission 
-    submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
+    # 4. Prepare Submission
+    submission_data = {
+        "username": username.strip(),
+        "agent_code": agent_code,
+        "agent_name": getattr(agent, "name", "BasicAgent"),
+        "answers": answers_payload,
+    }
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."
     print(status_update)
 

requirements.txt

@@ -4,3 +4,4 @@ pandas==1.5.3
 requests>=2.28.0
 smolagents==1.22.0
 ddgs==9.6.1
+huggingface-hub>=0.15.1