sergiosampayo commited on
Commit
aa3c857
·
verified ·
1 Parent(s): f7044f8

updated with OpenAI model

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, HfApiModel, LiteLLMModel
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
@@ -11,12 +11,14 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
11
 
12
  # --- Basic Agent Definition ---
13
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
14
- model = HfApiModel("Qwen/Qwen2.5-72B-Instruct")
15
-
16
  class BasicAgent:
17
  def __init__(self):
18
  print("BasicAgent initialized.")
19
- self.agent = CodeAgent(tools=[DuckDuckGoSearchTool(), VisitWebpageTool()], model=model)
 
 
 
 
20
  def __call__(self, question: str) -> str:
21
  print(f"Agent received question (first 50 chars): {question[:50]}...")
22
  answer = self.agent.run(question)
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, OpenAIServerModel, LiteLLMModel
7
 
8
  # (Keep Constants as is)
9
  # --- Constants ---
 
11
 
12
  # --- Basic Agent Definition ---
13
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
 
14
  class BasicAgent:
15
  def __init__(self):
16
  print("BasicAgent initialized.")
17
+ self.agent = CodeAgent(
18
+ tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
19
+ model=OpenAIServerModel(model_id="gpt-4o"),
20
+ add_base_tools=True,
21
+ )
22
  def __call__(self, question: str) -> str:
23
  print(f"Agent received question (first 50 chars): {question[:50]}...")
24
  answer = self.agent.run(question)