George Sergia commited on
Commit
bbb6334
·
1 Parent(s): 034083e

Fix the return answer

Browse files
Files changed (1) hide show
  1. agent.py +2 -1
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
- Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
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.