Update weather_tool to use OpenWeatherMap API

app.py

@@ -19,6 +19,7 @@ import logging
 import sys
 from langchain_community.utilities import DuckDuckGoSearchAPIWrapper
 from openai import OpenAI
+import requests
 
 # LangChain
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage, ToolMessage
@@ -48,24 +49,44 @@ app = FastAPI()
 
 # Define tools
 @tool
-def weather_tool(query: str) -> str:
-    """Get weather information for a specific location or query."""
-    logger.info(f"Getting weather information for: {query}")
+def weather_tool(location: str) -> str:
+    """Get current weather information for a specific location using OpenWeatherMap API."""
+    logger.info(f"Getting weather information for: {location}")
     try:
-        search = DuckDuckGoSearchAPIWrapper()
-        # 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}")
+        # Get API key from environment
+        api_key = os.getenv("OPENWEATHER_API_KEY")
+        if not api_key:
+            return "Error: OPENWEATHER_API_KEY not found in environment variables"
             
-        if not results:
-            return f"I couldn't find weather information for '{query}'. Please specify a location or try rephrasing your question."
+        # Make API request
+        base_url = "http://api.openweathermap.org/data/2.5/weather"
+        params = {
+            "q": location,
+            "appid": api_key,
+            "units": "metric"  # Use metric units
+        }
+        
+        response = requests.get(base_url, params=params)
+        data = response.json()
+        
+        if response.status_code != 200:
+            return f"Error getting weather: {data.get('message', 'Unknown error')}"
             
-        logger.info(f"Weather results: {results}")
-        return f"Here's the weather information for {query}:\n\n{results}"
+        # Extract relevant weather information
+        weather = data["weather"][0]["description"]
+        temp = data["main"]["temp"]
+        humidity = data["main"]["humidity"]
+        wind_speed = data["wind"]["speed"]
+        
+        result = f"""Current weather in {location}:
+- Temperature: {temp}°C
+- Conditions: {weather}
+- Humidity: {humidity}%
+- Wind Speed: {wind_speed} m/s"""
+        
+        logger.info(f"Weather results: {result}")
+        return result
+        
     except Exception as e:
         logger.error(f"Error getting weather information: {str(e)}")
         return f"Error getting weather information: {str(e)}"

requirements.txt

@@ -7,3 +7,4 @@ langchain-core>=0.1.0
 langchain-community>=0.0.10
 langchain-openai>=0.0.5
 websockets>=12.0
+requests>=2.31.0