Update agent.py

agent.py

@@ -64,7 +64,7 @@ def describe_image(request:str, file_path: str) -> str:
     Extract and return the requested information from an image.
     
     Args:
-        request: The information to retreive from the image.
+        request: The information to retreive from the image. The request must be simple, short and precise.
         file_path (str): Path to the audio file to be transcribed. The file should
                         be in a format compatible with the SpeechRecognition library.
     
@@ -107,6 +107,15 @@ def describe_image(request:str, file_path: str) -> str:
     )
     
     return response.output_text
+
+
+@tool
+def get_transcript(file_id:str) -> str:
+    from youtube_transcript_api import YouTubeTranscriptApi
+    ytt_api = YouTubeTranscriptApi()
+    transcript = ytt_api.fetch(video_id)
+    return transcript
+
     
 class TestAgent:
     def __init__(self):
@@ -149,9 +158,23 @@ class TestAgent:
         )
         # V3. add Guidance
         prompt_for_guidance = "\n10. Provide the answer axactly as it is asked, be concise and precise\n\nNow Begin!"
-        self.agent.prompt_templates['system_prompt'] = self.agent.prompt_templates['system_prompt'] + prompt_for_guidance
+        #self.agent.prompt_templates['system_prompt'] = self.agent.prompt_templates['system_prompt'] + prompt_for_guidance
+
+        # V4. use prompt from the paper as guidance
+        prompt = """\n\n
+                Remember: You are a general AI assistant. I will ask you a question. Report your thoughts, and
+                finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
+                YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated
+                list of numbers and/or strings.
+                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.
+                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.
+                If you are asked for a comma separated list, apply the above rules depending of whether the element
+                to be put in the list is a number or a string.
+                """
+        self.agent.prompt_templates['system_prompt'] = self.agent.prompt_templates['system_prompt'] + prompt
 
-        # V4. use prompt from the paper ?
 
     def __call__(self, question: str) -> str: