openAI

app.py

@@ -4,7 +4,14 @@ import requests
 import inspect
 import pandas as pd
 
-from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, InferenceClientModel, HfApiModel
+from smolagents import (
+    ToolCallingAgent, 
+    CodeAgent,
+    DuckDuckGoSearchTool, 
+    InferenceClientModel, 
+    HfApiModel, 
+    OpenAIServerModel
+)
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -14,16 +21,21 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        self.model = HfApiModel()
+        self.model = OpenAIServerModel(
+            model_id='o4-mini',
+            api_base="https://api.openai.com/v1",
+            api_key=os.environ["OPENAI_API_KEY"],
+        )
         self.agent = ToolCallingAgent(
             tools=[DuckDuckGoSearchTool()], 
-            model=self.model
+            model=self.model,
+            add_base_tools=True
         )
-        self.agent.prompt_templates['system_prompt'] = """
+        #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.
+        #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.
 
-        """
+        #"""
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
@@ -85,7 +97,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-    for item in questions_data:
+    for i, item in enumerate(questions_data):
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None:
@@ -98,6 +110,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")
              results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+        
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")