marin-iuga commited on
Commit
86d3e53
·
verified ·
1 Parent(s): f25aa32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py CHANGED
@@ -19,6 +19,24 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
19
  class BasicAgent:
20
  def __init__(self):
21
  print("BasicAgent initialized.")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  def __call__(self, question: str) -> str:
23
  print(f"Agent received question (first 50 chars): {question[:50]}...")
24
  fixed_answer = "This is a default answer."
 
19
  class BasicAgent:
20
  def __init__(self):
21
  print("BasicAgent initialized.")
22
+
23
+ model_api_key = os.getenv("OPENAI_API_KEY")
24
+
25
+ model = OpenAIServerModel(
26
+ model_id="gpt-4o-mini-2024-07-18",
27
+ api_key = model_api_key
28
+ )
29
+
30
+ self.tools = [
31
+ DuckDuckGoSearchTool()
32
+ ]
33
+
34
+ self.agent = CodeAgent(
35
+ model = model,
36
+ tools = self.tools,
37
+ add_base_tools=True # Add basic tools like math
38
+ )
39
+
40
  def __call__(self, question: str) -> str:
41
  print(f"Agent received question (first 50 chars): {question[:50]}...")
42
  fixed_answer = "This is a default answer."