abliznyuk commited on
Commit
5e11dec
·
1 Parent(s): f381c1d
Files changed (2) hide show
  1. agent.py +1 -0
  2. app.py +1 -1
agent.py CHANGED
@@ -78,6 +78,7 @@ class GAIAAgent:
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
 
 
78
  model=OpenAIServerModel(model_id='gpt-4.1-mini', max_tokens=4096, temperature=0),
79
  add_base_tools=False,
80
  max_steps=15,
81
+ additional_authorized_imports=["pandas"],
82
  )
83
  self.prompt = get_prompt()
84
 
app.py CHANGED
@@ -68,7 +68,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
68
  task_id = item.get("task_id")
69
  question_text = item.get("question")
70
  data_url = None
71
- if "file_name" in item:
72
  data_url = f"{api_url}/files/{item['task_id']}"
73
  if not task_id or question_text is None:
74
  print(f"Skipping item with missing task_id or question: {item}")
 
68
  task_id = item.get("task_id")
69
  question_text = item.get("question")
70
  data_url = None
71
+ if item["file_name"]:
72
  data_url = f"{api_url}/files/{item['task_id']}"
73
  if not task_id or question_text is None:
74
  print(f"Skipping item with missing task_id or question: {item}")