Thanh Vinh Vo commited on
Commit
7c71cec
·
1 Parent(s): 125e003
Files changed (1) hide show
  1. app.py +13 -4
app.py CHANGED
@@ -284,7 +284,7 @@ def get_wikipedia_section_tables(
284
  @tool
285
  def get_file(question_id: str, file_name: str) -> str:
286
  """
287
- A tool that downloads data file that was mentioned in a question and store it as local file.
288
  Args:
289
  question_id: Question ID.
290
  file_name: File name.
@@ -421,9 +421,18 @@ class BasicAgent:
421
  def __call__(self, question: str, question_id: str, file_name: str) -> str:
422
  print(f"Agent received question: {question}")
423
  file = f"Provided data file: {file_name}" if file_name else ""
 
 
 
 
 
424
  prompt = f"""
425
- Answer the following question (question_id is {question_id}):):
426
- "{question}" {file}
 
 
 
 
427
  Follow below rules when possible:
428
  - Please take the question literally! Do not add any additional information or assumptions.
429
  - Please answer as concisely as possible.
@@ -559,7 +568,7 @@ def run_and_submit_all(question_id: str, profile: gr.OAuthProfile | None):
559
  f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
560
  f"Message: {result_data.get('message', 'No message received.')}"
561
  )
562
- print("Submission successful.")
563
  results_df = pd.DataFrame(results_log)
564
  return final_status, results_df
565
  except requests.exceptions.HTTPError as e:
 
284
  @tool
285
  def get_file(question_id: str, file_name: str) -> str:
286
  """
287
+ A tool that downloads file that was mentioned in a question and store it as local file.
288
  Args:
289
  question_id: Question ID.
290
  file_name: File name.
 
421
  def __call__(self, question: str, question_id: str, file_name: str) -> str:
422
  print(f"Agent received question: {question}")
423
  file = f"Provided data file: {file_name}" if file_name else ""
424
+ metadata = {}
425
+ metadata["question_id"] = question_id
426
+ if file_name:
427
+ metadata["file_name"] = file_name
428
+
429
  prompt = f"""
430
+ Answer the following question:
431
+ "{question}".
432
+ Question metadata in JSON format:
433
+ ```
434
+ {json.dumps(metadata)}
435
+ ```
436
  Follow below rules when possible:
437
  - Please take the question literally! Do not add any additional information or assumptions.
438
  - Please answer as concisely as possible.
 
568
  f"({result_data.get('correct_count', '?')}/{result_data.get('total_attempted', '?')} correct)\n"
569
  f"Message: {result_data.get('message', 'No message received.')}"
570
  )
571
+ print(f"Submission successful. Final status: {final_status}")
572
  results_df = pd.DataFrame(results_log)
573
  return final_status, results_df
574
  except requests.exceptions.HTTPError as e: