Update agent.py

agent.py

@@ -17,32 +17,32 @@ import shutil
 import traceback
 
 
-@tool
-def GoogleSearchTool(query: str) -> str:
-    """Tool for performing Google searches using Custom Search JSON API
-    Args:
-        query (str): Search query string
-    Returns:
-        str: Formatted search results
-    """
-    api_key = os.environ.get("GOOGLE_API_KEY")
-    cse_id = os.environ.get("GOOGLE_CSE_ID")
-    if not api_key or not cse_id:
-        raise ValueError("GOOGLE_API_KEY and GOOGLE_CSE_ID must be set in environment variables.")
-    url = "https://www.googleapis.com/customsearch/v1"
-    params = {
-        "key": api_key,
-        "cx": cse_id,
-        "q": query,
-        "num": 5  # Number of results to return
-    }
-    try:
-        response = requests.get(url, params=params)
-        response.raise_for_status()
-        results = response.json().get("items", [])
-        return "\n".join([f"{item['title']}: {item['link']}" for item in results]) or "No results found."
-    except Exception as e:
-        return f"Error performing Google search: {str(e)}"
+#@tool
+#def GoogleSearchTool(query: str) -> str:
+#    """Tool for performing Google searches using Custom Search JSON API
+#    Args:
+#        query (str): Search query string
+#    Returns:
+#        str: Formatted search results
+#    """
+#    api_key = os.environ.get("GOOGLE_API_KEY")
+#    cse_id = os.environ.get("GOOGLE_CSE_ID")
+#    if not api_key or not cse_id:
+#        raise ValueError("GOOGLE_API_KEY and GOOGLE_CSE_ID must be set in environment variables.")
+#    url = "https://www.googleapis.com/customsearch/v1"
+#    params = {
+#        "key": api_key,
+#        "cx": cse_id,
+#        "q": query,
+#        "num": 5  # Number of results to return
+#    }
+#    try:
+#        response = requests.get(url, params=params)
+#        response.raise_for_status()
+#        results = response.json().get("items", [])
+#        return "\n".join([f"{item['title']}: {item['link']}" for item in results]) or "No results found."
+#    except Exception as e:
+#        return f"Error performing Google search: {str(e)}"
 
 
 from langchain_community.document_loaders import ArxivLoader
@@ -297,13 +297,13 @@ class MagAgent:
                 search_arxiv,
 #                PNG2FENTool,
 #                ChessEngineTool(),
-                GoogleSearchTool,
+#                GoogleSearchTool,
 #                ImageAnalysisTool,
             ],
             verbosity_level=2,
 #            prompt_templates=prompt_templates,
             add_base_tools=False,
-            max_steps=15
+            max_steps=20
         )
         print("MagAgent initialized.")
 
@@ -330,7 +330,7 @@ class MagAgent:
                 "   - Provide answers that are concise, accurate, and properly punctuated according to standard English grammar.\n"
                 "   - Use quotation marks for direct quotes (e.g., \"Extreamly.\") and appropriate punctuation for lists, sentences, or clarifications.\n"
                 "   - If the question asks for a specific quote or response (e.g., what a character says), format the answer clearly, e.g., 'Character says, \"Exact quote.\"'\n\n"
-                "   -If asked about name of place or city, use full complete name without abbreviations (e.g. use Saint Petersburg instead of St.Petersburg). \n"
+                "   - If asked about name of place or city, use full complete name without abbreviations (e.g. use Saint Petersburg instead of St.Petersburg). \n"
                 "4. **Error Handling**:\n"
                 "   - 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.\"\n\n"
                 f"Answer the following question: {question} \n"