Rename search_web to weather_tool and make it weather-specific

app.py

@@ -48,17 +48,27 @@ app = FastAPI()
 
 # Define tools
 @tool
-def search_web(query: str) -> str:
-    """Search the web for information on a given query."""
-    logger.info(f"Searching web for query: {query}")
+def weather_tool(query: str) -> str:
+    """Get weather information for a specific location or query."""
+    logger.info(f"Getting weather information for: {query}")
     try:
         search = DuckDuckGoSearchAPIWrapper()
-        results = search.run(query)
-        logger.info(f"Search results: {results}")
-        return results
+        # Focus the search on weather information
+        weather_query = f"current weather {query} site:weather.com OR site:accuweather.com"
+        results = search.run(weather_query)
+        
+        if not results:
+            # If no specific weather results, try a more general weather search
+            results = search.run(f"weather {query}")
+            
+        if not results:
+            return f"I couldn't find weather information for '{query}'. Please specify a location or try rephrasing your question."
+            
+        logger.info(f"Weather results: {results}")
+        return f"Here's the weather information for {query}:\n\n{results}"
     except Exception as e:
-        logger.error(f"Error in web search: {str(e)}")
-        return f"Error searching the web: {str(e)}"
+        logger.error(f"Error getting weather information: {str(e)}")
+        return f"Error getting weather information: {str(e)}"
 
 
 @tool
@@ -96,7 +106,7 @@ def generate_image_prompt(description: str) -> str:
 
 # Create tools list
 logger.info("Creating tools list...")
-tools = [search_web, calculate, generate_image_prompt]
+tools = [weather_tool, calculate, generate_image_prompt]
 
 # Set up the language model
 logger.info("Initializing ChatOpenAI model...")
@@ -145,7 +155,7 @@ def create_agent_node():
     # System prompt for the agent
     system_prompt = """You are a helpful AI assistant with access to the following tools:
 
-1. search_web: Search the web for information
+1. weather_tool: Get weather information
 2. calculate: Calculate the result of mathematical expressions
 3. generate_image_prompt: Generate prompts for image generation