Update app.py

app.py

@@ -14,9 +14,8 @@ import requests
 VERBOSE_SHELL = True
 ENDPOINT_URL = "https://api.hyperbolic.xyz/v1"
 OAI_API_KEY = os.environ['HYPERBOLIC_XYZ_API_KEY']
-WEATHER_API_KEY = os.environ["WEATHER_API_KEY"]
+PERPLEXITY_API_KEY = os.environ["PERPLEXITY_API_KEY"]
 MODEL_NAME = "meta-llama/Llama-3.3-70B-Instruct"
-#MODEL_NAME = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
 def lgs(log_string):
     if VERBOSE_SHELL:
@@ -287,72 +286,104 @@ class NIHRefSNPTool(ToolBase):
 
 nih_ref_snp_tool=NIHRefSNPTool()
 
-def get_weather_data(location):
+def query_perplexity(query: str, api_key: str) -> Dict:
     """
-    Fetch current weather data for a given location using WeatherAPI.com.
-
+    Query the Perplexity API for research information using default settings.
+    
     Args:
-        location (str): The location for which to retrieve weather (e.g., "London", "90210", or "48.8567,2.3510").
-
+        query: The research question or topic to query
+        api_key: Your Perplexity API key
+    
     Returns:
-        dict: A dictionary containing the current weather data or an error message.
+        Dictionary containing the response from Perplexity API
     """
-    base_url = "https://api.weatherapi.com/v1/current.json"
-    params = {
-        "key": WEATHER_API_KEY,
-        "q": location,
-        "aqi": "no"  # Set to "yes" to include air quality data if desired.
+    if not api_key:
+        return {
+            "status": "error",
+            "message": "API key is required. Please provide your Perplexity API key."
+        }
+    
+    url = "https://api.perplexity.ai/chat/completions"
+    
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/json"
     }
-    full_url = base_url + "?" + "&".join([f"{k}={urllib.parse.quote(str(v))}" for k, v in params.items()])
-    try:
-        response = requests.get(full_url)
-    except:
-        lgs("FAILED PARAMS: " + str(params))
-        lgs("FAILED RESPONSE: " + str(response.text))
-    lgs("RAW RESPONSE: " + str(response))
-    if response.status_code != 200:
-        return {"error": f"Failed to retrieve weather data for {location}. Status code: {response.status_code}"}
-    data = response.json()
-    formatted_data = {
-        "location": data.get("location", {}),
-        "current": {
-            "last_updated": data.get("current", {}).get("last_updated"),
-            "temp_c": data.get("current", {}).get("temp_c"),
-            "temp_f": data.get("current", {}).get("temp_f"),
-            "precip_mm": data.get("current", {}).get("precip_mm"),
-            "precip_in": data.get("current", {}).get("precip_in"),
-            "humidity": data.get("current", {}).get("humidity"),
-            "wind_kph": data.get("current", {}).get("wind_kph"),
-            "wind_mph": data.get("current", {}).get("wind_mph"),
-            "condition": data.get("current", {}).get("condition", {})
+    
+    # Using Perplexity's recommended defaults
+    messages = [
+        {
+            "role": "system",
+            "content": "You are a helpful AI research assistant. Provide accurate, detailed information with relevant citations."
+        },
+        {
+            "role": "user",
+            "content": query
         }
+    ]
+    
+    payload = {
+        "model": "sonar",
+        "messages": messages,
+        "temperature": 0.2,
+        "top_p": 0.9,
     }
-    return formatted_data
+    
+    try:
+        response = requests.post(url, json=payload, headers=headers)
+        response.raise_for_status()  # Raise exception for 4XX/5XX responses
+        data = response.json()
+        
+        # Format the response for easier consumption
+        result = {
+            "status": "success",
+            "content": data["choices"][0]["message"]["content"] if data.get("choices") else None,
+            "citations": data.get("citations", []),
+        }
+            
+        return result
+    
+    except requests.exceptions.RequestException as e:
+        return {
+            "status": "error",
+            "message": f"API request failed: {str(e)}"
+        }
+    except Exception as e:
+        return {
+            "status": "error",
+            "message": f"An unexpected error occurred: {str(e)}"
+        }
 
-class WeatherAPITool(ToolBase):
+class PerplexityQueryTool(ToolBase):
     def __init__(self):
         super().__init__(
-            programmatic_name="get_weather_data",
-            natural_name="Weather Report Fetcher",
-            active_voice_description_of_capability="You can fetch real-time weather data for any location worldwide.",
-            passive_voice_description_of_function="a service that retrieves current weather details including temperature, precipitation, humidity, and wind data.",
-            prescriptive_conditional="When provided with a location (city, ZIP, or lat,long) call the get_weather_data function to retrieve its weather information.",
+            programmatic_name="query_perplexity",
+            natural_name="Perplexity Research Assistant",
+            active_voice_description_of_capability="You can search for up-to-date information using the Perplexity API.",
+            passive_voice_description_of_function="a service that retrieves information from the web using the Perplexity AI model with proper citations.",
+            prescriptive_conditional="When you need to find current information or answer research questions, you should use the query_perplexity function.",
             input_params={
-                "location": {
+                "query": {
                     "type": "string",
-                    "description": "The location to retrieve weather data for (e.g., 'London', '90210', or '48.8567,2.3510')."
+                    "description": "The research question or topic to query."
+                },
+                "api_key": {
+                    "type": "string",
+                    "description": "Your Perplexity API key."
                 },
             },
-            required_params=["location"],
+            required_params=["query", "api_key"],
         )
 
     def actual_function(self, **kwargs):
-        return get_weather_data(kwargs["location"])
+        """
+        Query the Perplexity API for research information.
+        """
+        return query_perplexity(**kwargs)
 
-# Instance of the weather tool.
-weather_tool = WeatherAPITool()
+perplexity_tool = PerplexityQueryTool()
 
-tool_objects_list = [arxiv_tool, nih_ref_snp_tool,weather_tool]
+tool_objects_list = [arxiv_tool, nih_ref_snp_tool, perplexity_tool]
 system_prompt = build_sys_prompt(tool_objects_list)
 functions_dict = {t.json_name: t.actual_function for t in tool_objects_list}