Update app.py

app.py

@@ -7,6 +7,7 @@ from langchain.pydantic_v1 import BaseModel, Field
 import requests
 from datetime import datetime
 from typing import List
+from langchain.tools import Tool
 from langchain.prompts import ChatPromptTemplate
 from langchain.output_parsers import PydanticOutputParser
 from langchain.memory import ConversationBufferMemory
@@ -29,10 +30,6 @@ llm = ChatGoogleGenerativeAI(
 
 
 
-class WeatherInput(BaseModel):
-    city: str = Field(default=None, description="The city to get the weather for.")
-
-
 def get_location_from_ip():
     ip = requests.get('https://api.ipify.org').text
     response = requests.get(f"https://ipapi.co/{ip}/json/").json()
@@ -43,11 +40,14 @@ def get_location_from_ip():
     }
 
 
+class WeatherInput(BaseModel):
+    city: str = Field(default=None, description="The city to get the weather for.")
+
+
 @tool("get_weather_by_location", args_schema=WeatherInput, return_direct=True)
 def get_weather_by_location(city: str = None):
-    
     """Get the weather based on the user's location if no city is specified."""
-    
+
     if not city:
         location = get_location_from_ip()
         city = location['city']
@@ -73,6 +73,7 @@ def get_weather_by_location(city: str = None):
 
 
 def format_weather_response(weather_data, city):
+    """Format the weather data into a readable string."""
     intervals = weather_data['data']['timelines'][0]['intervals']
     response = f"Weather forecast for {city}:\n\n"
 
@@ -89,6 +90,14 @@ def format_weather_response(weather_data, city):
 
     return response
 
+
+get_weather_tool = Tool(
+    name="get_weather_by_location",
+    func=get_weather_by_location,
+    description="Get the current weather for a specific location. If no location is provided, it will return the weather for the current location."
+)
+
+
 class DailyWeather(BaseModel):
     date: str
     temperature: float