abliznyuk commited on
Commit
c432527
·
1 Parent(s): a73f583

add files to agent queries

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -67,11 +67,17 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
67
  for item in questions_data:
68
  task_id = item.get("task_id")
69
  question_text = item.get("question")
 
 
 
70
  if not task_id or question_text is None:
71
  print(f"Skipping item with missing task_id or question: {item}")
72
  continue
73
  try:
74
- submitted_answer = agent(question_text)
 
 
 
75
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
76
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
77
  except Exception as e:
 
67
  for item in questions_data:
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}")
75
  continue
76
  try:
77
+ if data_url:
78
+ submitted_answer = agent(question_text + f"Task file source URL: {data_url}.")
79
+ else:
80
+ submitted_answer = agent(question_text)
81
  answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
82
  results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
83
  except Exception as e: