mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-10-31 04:45:23 +08:00
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
# Auto generated by LangBot Plugin SDK.
|
|
# Please refer to https://langbot.app/docs/en/plugin/dev/tutor.html for more details.
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from langbot_plugin.api.definition.components.tool.tool import Tool
|
|
import httpx
|
|
|
|
# Constants
|
|
NWS_API_BASE = "https://api.weather.gov"
|
|
USER_AGENT = "weather-app/1.0"
|
|
|
|
|
|
async def make_nws_request(url: str) -> dict[str, Any] | None:
|
|
"""Make a request to the NWS API with proper error handling."""
|
|
headers = {
|
|
"User-Agent": USER_AGENT,
|
|
"Accept": "application/geo+json"
|
|
}
|
|
async with httpx.AsyncClient() as client:
|
|
try:
|
|
response = await client.get(url, headers=headers, timeout=30.0)
|
|
response.raise_for_status()
|
|
return response.json()
|
|
except Exception:
|
|
return None
|
|
|
|
def format_alert(feature: dict) -> str:
|
|
"""Format an alert feature into a readable string."""
|
|
props = feature["properties"]
|
|
return f"""
|
|
Event: {props.get('event', 'Unknown')}
|
|
Area: {props.get('areaDesc', 'Unknown')}
|
|
Severity: {props.get('severity', 'Unknown')}
|
|
Description: {props.get('description', 'No description available')}
|
|
Instructions: {props.get('instruction', 'No specific instructions provided')}
|
|
"""
|
|
|
|
class GetWeatherAlerts(Tool):
|
|
|
|
async def call(self, params: dict[str, Any]) -> dict[str, Any]:
|
|
"""Fill your tool code here"""
|
|
state = params.get("state", "CA")
|
|
|
|
url = f"{NWS_API_BASE}/alerts/active/area/{state}"
|
|
data = await make_nws_request(url)
|
|
|
|
if not data or "features" not in data:
|
|
return "Unable to fetch alerts or no alerts found."
|
|
|
|
if not data["features"]:
|
|
return "No active alerts for this state."
|
|
|
|
alerts = [format_alert(feature) for feature in data["features"]]
|
|
return "\n---\n".join(alerts)
|