hibikigf88 commited on
Commit
b74d35c
·
verified ·
1 Parent(s): 129379e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,7 +21,7 @@ class BasicAgent:
21
 
22
  def __call__(self, question: str) -> str:
23
  model = OpenAIServerModel(
24
- model_id="gpt-4o-mini",
25
  api_base="https://api.openai.com/v1",
26
  api_key=os.getenv("API_KEY"),
27
  )
@@ -30,10 +30,10 @@ class BasicAgent:
30
  tools=[search_tool],
31
  model=model,
32
  add_base_tools=True,
33
- max_steps=1,
34
  additional_authorized_imports=["pandas", "requests"],
35
  )
36
-
37
  response = code_agent.run(
38
  f"""
39
  You are an expert in answering questions.
 
21
 
22
  def __call__(self, question: str) -> str:
23
  model = OpenAIServerModel(
24
+ model_id="gpt-o4-mini",
25
  api_base="https://api.openai.com/v1",
26
  api_key=os.getenv("API_KEY"),
27
  )
 
30
  tools=[search_tool],
31
  model=model,
32
  add_base_tools=True,
33
+ max_steps=2,
34
  additional_authorized_imports=["pandas", "requests"],
35
  )
36
+
37
  response = code_agent.run(
38
  f"""
39
  You are an expert in answering questions.