Update agent.py

agent.py

@@ -1,4 +1,3 @@
-# agent.py
 import os
 import datetime
 import requests
@@ -8,18 +7,26 @@ from typing import List
 
 from langchain.text_splitter import CharacterTextSplitter
 from tools.final_answer import FinalAnswerTool
-from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, load_tool, tool
 
 # === TOOLS ===
 
 @tool
 def web_search(query: str) -> str:
+    """Allows search through DuckDuckGo.
+    Args:
+        query: what you want to search
+    """
     search_tool = DuckDuckGoSearchTool()
     results = search_tool(query)
     return "\n".join(results)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
+    """Fetches the current local time in a specified timezone.
+    Args:
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -29,14 +36,22 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 @tool
 def visit_webpage(url: str) -> str:
+    """Fetches raw HTML content of a web page.
+    Args:
+        url: The url of the webpage.
+    """
     try:
         response = requests.get(url, timeout=5)
-        return response.text[:5000]
+        return response.text[:5000]  # Limit length
     except Exception as e:
         return f"[ERROR fetching {url}]: {str(e)}"
 
 @tool
 def text_splitter(text: str) -> List[str]:
+    """Splits text into chunks using LangChain's CharacterTextSplitter.
+    Args:
+        text: A string of text to split.
+    """
     splitter = CharacterTextSplitter(chunk_size=450, chunk_overlap=10)
     return splitter.split_text(text)
 
@@ -84,10 +99,6 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# === Export for import ===
-def run_agent(question):
-    try:
-        result = agent(question)
-        return [str(result)]
-    except Exception as e:
-        return [f"Error: {e}"]
+# === EXPORT AGENT ===
+def get_agent():
+    return agent