final assignment gaia agent tools submission enabled system prompt litellm login models

README.md

@@ -1,5 +1,5 @@
 ---
-title: Template Final Assignment
+title: HF Smolagents Course: Gaia Agent Evaluation 
 emoji: 🕵🏻‍♂️
 colorFrom: indigo
 colorTo: indigo

agents/gaia.py

@@ -1,8 +1,10 @@
 # --- Gaia Agent Definition ---
 from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool
 from tools.gaiatool import fetch_questions
-from smolagents import LiteLLMModel, OpenAIServerModel, HfApiModel, InferenceClientModel, TransformersModel
+from smolagents import LiteLLMModel, InferenceClientModel
 import os
+from smolagents import HfApiModel
+#from smolagents import OpenAIServerModel #uncoment in case of using OpenAI API, or Hugging Face API
 
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 # (Keep Constants as is)
@@ -17,20 +19,26 @@ class GaiaAgent:
 
         self.questions_url = f"{api_url}/questions"
 
-        # Initialize the model
-        #self.model = LiteLLMModel(model_id="gemini/gemini-1.5-flash-latest", api_key=os.getenv("GEMINI_API_KEY"))
-        """self.model = OpenAIServerModel(
-            model_id="gpt-4o-mini",
-            api_base="https://api.openai.com/v1",
-            api_key=os.environ["OPENAI_API_KEY"],
-            temperature=0.0
-        )"""
+        if not os.getenv("SPACE_ID"):
+            # Initialize local model
+            model_id = "ollama/qwen2.5:14b"
+            self.model = LiteLLMModel(
+                model_id=model_id, 
+                api_base="http://localhost:11434"
+                #api_key=os.getenv("GEMINI_API_KEY")
+            )
+        else: # Run with Hugging Face Inference API
+            #model_id = "meta-llama/Llama-3.3-70B-Instruct"
+            #self.model = InferenceClientModel(model_id=model_id)
 
-        #model_id = "meta-llama/Llama-3.3-70B-Instruct"
-        #self.model = InferenceClientModel(model_id=model_id)
+            self.model = HfApiModel(temperature=0.0)
 
-        model_id = "meta-llama/Llama-4-Maverick-17B-128E-Instruct"
-        self.model = TransformersModel(model_id=model_id)
+            """self.model = OpenAIServerModel(
+                model_id="gpt-4o-mini",
+                api_base="https://api.openai.com/v1",
+                api_key=os.environ["OPENAI_API_KEY"],
+                temperature=0.0
+            )"""
 
         search= DuckDuckGoSearchTool()
         web_browser = VisitWebpageTool()

app.py

@@ -15,12 +15,16 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
 
-    if profile:
-        username= f"{profile.username}"
-        print(f"User logged in: {username}")
+    if space_id:
+        if profile:
+            username= f"{profile.username}"
+            print(f"User logged in: {username}")
+        else:
+            print("User not logged in.")
+            return "Please Login to Hugging Face with the button.", None
     else:
-        print("User not logged in.")
-        return "Please Login to Hugging Face with the button.", None
+        username = "avfranco"
+        print(f"Running locally with user {username}.")
 
     submit_url = f"{DEFAULT_API_URL}/submit"
 

requirements.txt

@@ -4,4 +4,3 @@ pandas==2.2.3
 requests==2.32.3
 smolagents
 smolagents[litellm]
-transformers

tools/gaiatool.py

@@ -1,7 +1,5 @@
-from smolagents import tool
 import requests
 
-
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"