Spaces:
Sleeping
Sleeping
George Sergia
commited on
Commit
·
bbb6334
1
Parent(s):
034083e
Fix the return answer
Browse files
agent.py
CHANGED
|
@@ -10,6 +10,7 @@ from excel_file_reader import excel_file_reader
|
|
| 10 |
from image_analyzer import analyze_image
|
| 11 |
from audio_analyzer import analyze_audio
|
| 12 |
from video_analyzer import analyze_video
|
|
|
|
| 13 |
import os
|
| 14 |
|
| 15 |
async def main(query: str, file_url: str | None) -> str:
|
|
@@ -38,7 +39,7 @@ async def main(query: str, file_url: str | None) -> str:
|
|
| 38 |
llm=google_genai_llm,
|
| 39 |
system_prompt="""
|
| 40 |
You are a general AI assistant. I will ask you a question.
|
| 41 |
-
|
| 42 |
YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
|
| 43 |
If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
|
| 44 |
If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
|
|
|
|
| 10 |
from image_analyzer import analyze_image
|
| 11 |
from audio_analyzer import analyze_audio
|
| 12 |
from video_analyzer import analyze_video
|
| 13 |
+
from generate_final_answer import generate_final_answer
|
| 14 |
import os
|
| 15 |
|
| 16 |
async def main(query: str, file_url: str | None) -> str:
|
|
|
|
| 39 |
llm=google_genai_llm,
|
| 40 |
system_prompt="""
|
| 41 |
You are a general AI assistant. I will ask you a question.
|
| 42 |
+
Do not report your thoughts, and finish your answer with the following template: [YOUR FINAL ANSWER].
|
| 43 |
YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
|
| 44 |
If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
|
| 45 |
If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
|