abliznyuk commited on
Commit
f381c1d
·
1 Parent(s): e715173

improvements

Browse files
Files changed (2) hide show
  1. agent.py +4 -4
  2. app.py +1 -1
agent.py CHANGED
@@ -75,14 +75,14 @@ class GAIAAgent:
75
  visual_qa,
76
  transcribe_audio,
77
  ],
78
- model=OpenAIServerModel(model_id='gpt-4o-mini', max_tokens=4096, temperature=0),
79
  add_base_tools=False,
80
- max_steps=10,
81
  )
82
  self.prompt = get_prompt()
83
 
84
- def __call__(self, question: str, source: str = None) -> str:
85
- args = {"question": question, "task_file_source": source} if source else {"question": question, }
86
  return self.agent.run(self.prompt, additional_args=args)
87
 
88
 
 
75
  visual_qa,
76
  transcribe_audio,
77
  ],
78
+ model=OpenAIServerModel(model_id='gpt-4.1-mini', max_tokens=4096, temperature=0),
79
  add_base_tools=False,
80
+ max_steps=15,
81
  )
82
  self.prompt = get_prompt()
83
 
84
+ def __call__(self, question: str) -> str:
85
+ args = {"question": question}
86
  return self.agent.run(self.prompt, additional_args=args)
87
 
88
 
app.py CHANGED
@@ -75,7 +75,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
75
  continue
76
  try:
77
  if data_url:
78
- submitted_answer = agent(question_text, data_url)
79
  else:
80
  submitted_answer = agent(question_text)
81
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
 
75
  continue
76
  try:
77
  if data_url:
78
+ submitted_answer = agent(question_text + f" Data file: {data_url}")
79
  else:
80
  submitted_answer = agent(question_text)
81
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})