Thanh Vinh Vo
commited on
Commit
·
7c71cec
1
Parent(s):
125e003
update
Browse files
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
|
| 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
|
| 426 |
-
"{question}"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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:
|