use openai

app.py

@@ -6,18 +6,19 @@ from typing import Any
 import gradio as gr
 import pandas as pd
 import requests
-from smolagents import CodeAgent, InferenceClientModel, tool
+from smolagents import CodeAgent, OpenAIServerModel, tool
 from smolagents.default_tools import DuckDuckGoSearchTool, VisitWebpageTool
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-DEFAULT_HF_MODEL = os.getenv("HF_MODEL", "Qwen/Qwen2.5-72B-Instruct")
+DEFAULT_OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-4o-mini")
 
 
 @dataclass
 class AgentConfig:
     api_base_url: str = DEFAULT_API_URL
-    hf_model: str = DEFAULT_HF_MODEL
-    hf_token_env: str = "HF_TOKEN"
+    openai_model: str = DEFAULT_OPENAI_MODEL
+    openai_api_key_env: str = "OPENAI_API_KEY"
+    openai_api_base: str = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
     max_steps: int = 8
     web_timeout_sec: int = 15
     max_file_chars: int = 12000
@@ -49,14 +50,15 @@ def submit_answers(api_base_url: str, payload: dict[str, Any]) -> dict[str, Any]
 class GAIASmolAgent:
     def __init__(self, config: AgentConfig):
         self.config = config
-        token = os.getenv(config.hf_token_env)
-        if not token:
-            raise ValueError(f"Missing required secret: {config.hf_token_env}")
-
-        self.model = InferenceClientModel(
-            model_id=config.hf_model,
-            token=token,
-            temperature=0,
+        api_key = os.getenv(config.openai_api_key_env)
+        if not api_key:
+            raise ValueError(f"Missing required secret: {config.openai_api_key_env}")
+
+        self.model = OpenAIServerModel(
+            model_id=config.openai_model,
+            api_base=config.openai_api_base,
+            api_key=api_key,
+            temperature=0.0,
             max_tokens=1200,
         )
         self.http = requests.Session()
@@ -300,9 +302,10 @@ with gr.Blocks() as demo:
         2. Submit generated answers to leaderboard scoring.
 
         Required Space secrets:
-        - `HF_TOKEN`
+        - `OPENAI_API_KEY`
         Optional:
-        - `HF_MODEL` (default: `Qwen/Qwen2.5-72B-Instruct`)
+        - `OPENAI_MODEL` (default: `gpt-4o-mini`)
+        - `OPENAI_API_BASE` (default: `https://api.openai.com/v1`)
         """
     )
 

requirements.txt

@@ -1,5 +1,5 @@
 gradio
 requests
 pandas
-huggingface_hub
 smolagents[toolkit]
+openai