Update agent.py

agent.py

@@ -1,4 +1,5 @@
-from smolagents import CodeAgent, LiteLLMModel, tool, Tool, load_tool, DuckDuckGoSearchTool, WikipediaSearchTool
+from smolagents import CodeAgent, LiteLLMModel, tool, Tool, load_tool, DuckDuckGoSearchTool
+, WikipediaSearchTool
 import asyncio
 import os
 import re
@@ -168,20 +169,20 @@ class MagAgent:
             max_tokens=8192
         )
 
-        self.imports = [
-            "pandas",
-            "numpy",
-            "os",
-            "requests",
-            "tempfile",
-            "datetime",
-            "json",
-            "time",
-            "re",
-            "openpyxl",
-            "pathlib",
-            "sys",
-        ]
+#        self.imports = [
+#            "pandas",
+#            "numpy",
+#            "os",
+#            "requests",
+#            "tempfile",
+#            "datetime",
+#            "json",
+#            "time",
+#            "re",
+#            "openpyxl",
+#            "pathlib",
+#            "sys",
+#        ]
 
         self.tools = [
             RetryDuckDuckGoSearchTool(),
@@ -200,12 +201,12 @@ class MagAgent:
             Follow these principles:
             - 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 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 the question references an attachment, the file path is provided in the FILE section. Use the appropriate tool based on the file extension to process it.
             - When processing external data (e.g., YouTube transcripts, web searches), expect potential issues like missing punctuation, inconsistent formatting, or conversational text.
             - If the input is ambiguous, prioritize extracting key information relevant to the question.
             - Provide answers that are concise, accurate, and properly punctuated according to standard English grammar.
-            - Use quotation marks for direct quotes (e.g., "Extremely.") and appropriate punctuation for lists, sentences, or clarifications.
+            - Use quotation marks for direct quotes (e.g., "Extremely") and appropriate punctuation for lists, sentences, or clarifications.
             - 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 you cannot retrieve or process data (e.g., due to blocked requests), return a clear error message: "Unable to retrieve data. Please refine the question or check external sources."
             - Reason step-by-step. Think through the solution logically and plan your actions carefully before answering.
@@ -225,7 +226,7 @@ class MagAgent:
             tools=self.tools,
             add_base_tools=True,
             additional_authorized_imports=self.imports,
-            verbosity_level=2,
+            verbosity_level=1,
             max_steps=20
         )
         print("MagAgent initialized.")