Update tools.py

tools.py

@@ -1,30 +1,35 @@
 # Tool Section
 
-from langchain_community.tools import DuckDuckGoSearchRun
 from langchain.tools import Tool
 from huggingface_hub import list_models
-import random
-
-# Internet Search Tool
-search_tool = DuckDuckGoSearchRun()
-
-def get_weather_info(location: str) -> str:
-    """Fetches dummy weather information for a given location."""
-    # Dummy weather data
-    weather_conditions = [
-        {"condition": "Rainy", "temp_c": 15},
-        {"condition": "Clear", "temp_c": 25},
-        {"condition": "Windy", "temp_c": 20}
-    ]
-    # Randomly select a weather condition
-    data = random.choice(weather_conditions)
-    return f"Weather in {location}: {data['condition']}, {data['temp_c']}°C"
+import requests
+import os
+from tavily import TavilyClient
+
+def internet_search_tool(query: str) -> str:
+    """Fetches information, news, etc from internet."""
+    client = TavilyClient(os.getenv("TAVILY_API_KEY"))
+    response = client.search(query=query)
+    return response
+
+search_tool = Tool(
+    name="internet_search_tool",
+    func=internet_search_tool,
+    description="Fetches information, news, etc from internet."
+)
+
+def get_weather_info(city: str) -> str:
+    """Fetches the current weather information of specific city."""
+    weather_api = "http://api.weatherapi.com/v1/current.json?q={city}".format(city=city)
+    weather_api_response = requests.get(weather_api, headers={"key":os.getenv("WEATHER_API_KEY")})
+    if weather_api_response.status_code == 200:
+        return weather_api_response.text
+    return "Problem at moment with API requests, please try after some time"
 
-# Initialize the tool
 weather_info_tool = Tool(
     name="get_weather_info",
     func=get_weather_info,
-    description="Fetches dummy weather information for a given location."
+    description="Fetches the current weather information of specific city."
 )
 
 def get_hub_stats(author: str) -> str: