Omnitopia commited on
Commit
e324889
·
verified ·
1 Parent(s): 1344117

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,7 +38,7 @@ class BasicAgent:
38
  if not api_key:
39
  raise ValueError("OPENAI_API_KEY not set in environment variables!")
40
  model = LiteLLMModel(
41
- model_id="gpt-4.1-mini",
42
  api_key=api_key
43
  )
44
 
@@ -46,7 +46,7 @@ class BasicAgent:
46
  self.agent = CodeAgent(
47
  model=model,
48
  tools=my_tool_list,
49
- max_steps=3,
50
  )
51
 
52
  def __call__(self, question: str, files=None, idx=None, total=None) -> str:
 
38
  if not api_key:
39
  raise ValueError("OPENAI_API_KEY not set in environment variables!")
40
  model = LiteLLMModel(
41
+ model_id="gpt-4.1",
42
  api_key=api_key
43
  )
44
 
 
46
  self.agent = CodeAgent(
47
  model=model,
48
  tools=my_tool_list,
49
+ max_steps=4,
50
  )
51
 
52
  def __call__(self, question: str, files=None, idx=None, total=None) -> str: