final assignment gaia agent tools submission enabled system prompt litellm llama

agents/gaia.py

@@ -1,7 +1,7 @@
 # --- Gaia Agent Definition ---
 from smolagents import CodeAgent, DuckDuckGoSearchTool
 from tools.gaiatool import fetch_questions
-from smolagents import LiteLLMModel, OpenAIServerModel, HfApiModel
+from smolagents import LiteLLMModel, OpenAIServerModel, HfApiModel, InferenceClientModel
 import os
 
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -19,12 +19,16 @@ class GaiaAgent:
 
         # 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
-        )
+        #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
+        #)
+
+
+        model_id = "meta-llama/Llama-3.3-70B-Instruct"
+        self.model = InferenceClientModel(model_id=model_id)
 
         # Initialize the DuckDuckGo search tool
         self.search_tool = DuckDuckGoSearchTool()
@@ -32,6 +36,8 @@ class GaiaAgent:
         self.agent = CodeAgent(
             model=self.model,
             tools=[self.search_tool],
+            add_base_tools=True,
+            additional_authorized_imports=["pandas"]
         )
 
         self.agent.prompt_templates["system_prompt"] += "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."

app.py

@@ -115,10 +115,7 @@ with gr.Blocks() as demo:
     gr.Markdown(
         """
         **Instructions:**
-
-        1.  (DUPLICATED) Please clone this space, then modify the code to define your agent's logic, the tools, the necessary packages, etc ...
-        2.  Log in to your Hugging Face account using the button below. This uses your HF username for submission.
-        3.  (NOT SUBMITTING FOR NOW: Click 'Run Evaluation & Submit All Answers' to fetch questions, run your agent, submit answers, and see the score.
+        1.  Log in to your Hugging Face account using the button below. This uses your HF username for submission.
 
         ---
         **Disclaimers:**