Update agent.py

agent.py

@@ -167,6 +167,7 @@ from tenacity import retry, stop_after_attempt, wait_exponential, retry_if_excep
 
 #class RetryDuckDuckGoSearchTool(DuckDuckGoSearchTool):
 #    @retry(
+
 #        stop=stop_after_attempt(3),
 #        wait=wait_exponential(multiplier=1, min=4, max=10),
 #        retry=retry_if_exception_type(Exception)
@@ -183,7 +184,7 @@ class MagAgent:
 
         print("Initializing MagAgent with search tools...")
         model = LiteLLMModel(
-            model_id="gemini/gemini-2.0-flash",
+            model_id="gemini/gemini-1.5-flash",
             api_key=os.environ.get("GEMINI_KEY"),
             max_tokens=8192
         )
@@ -223,12 +224,14 @@ class MagAgent:
             - Reason step-by-step. Think through the solution logically and plan your actions carefully before answering.
             - Validate information. Always verify facts when possible instead of guessing.
             - When processing external data (e.g., YouTube transcripts, web searches), expect potential issues like missing punctuation, inconsistent formatting, or conversational text.
+            - When asked to transcript YouTube video, try searching it in www.youtubetotranscript.com.
             - If the input is ambiguous, prioritize extracting key information relevant to the question.
             - Use code if needed. For calculations, parsing, or transformations, generate Python code and execute it. Be cautious, as some questions contain time-consuming tasks, so analyze the question and choose the most efficient solution.
             - Be precise and concise. The final answer must strictly match the required format with no extra commentary.
             - Use tools intelligently. If a question involves external information, structured data, images, or audio, call the appropriate tool to retrieve or process it.
             - If the question includes direct speech or quoted text (e.g., "Isn't that hot?"), treat it as a precise query and preserve the quoted structure in your response, including quotation marks for direct quotes (e.g., final_answer('"Extremely."')).
             - If asked about the name of a place or city, use the full complete name without abbreviations (e.g., use Saint Petersburg instead of St.Petersburg).
+            - If asked to look up page numbers, make sure you don't mix them with problem or excercise numbers.
             - If you cannot retrieve or process data (e.g., due to blocked requests), retry after 15 seconds delay, try another tool (try wikipedia_search, then web_search, then search_arxiv). Otherwise, return a clear error message: "Unable to retrieve data. Search has failed."
             - Use `final_answer` to give the final answer.