Omnitopia commited on
Commit
48ef67d
·
verified ·
1 Parent(s): f6eaa5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -20,11 +20,11 @@ class BasicAgent:
20
  if not api_key:
21
  raise ValueError("OPENAI_API_KEY not set in environment variables!")
22
  model = LiteLLMModel(
23
- model_id="gpt-4o",
24
  api_key=api_key
25
  )
26
 
27
- self.agent = CodeAgent(model=model, tools=my_tool_list, max_steps=6)
28
 
29
  def __call__(self, question: str) -> str:
30
  print(f"Agent received question: {question[:80]}...")
@@ -87,7 +87,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
87
  results_log = []
88
  answers_payload = []
89
  print(f"Running agent on {len(questions_data)} questions...")
90
- for item in questions_data[:3]:
91
  task_id = item.get("task_id")
92
  question_text = item.get("question")
93
  if not task_id or question_text is None:
 
20
  if not api_key:
21
  raise ValueError("OPENAI_API_KEY not set in environment variables!")
22
  model = LiteLLMModel(
23
+ model_id="gpt-4.1",
24
  api_key=api_key
25
  )
26
 
27
+ self.agent = CodeAgent(model=model, tools=my_tool_list, max_steps=3)
28
 
29
  def __call__(self, question: str) -> str:
30
  print(f"Agent received question: {question[:80]}...")
 
87
  results_log = []
88
  answers_payload = []
89
  print(f"Running agent on {len(questions_data)} questions...")
90
+ for item in questions_data:
91
  task_id = item.get("task_id")
92
  question_text = item.get("question")
93
  if not task_id or question_text is None: