SkillBase

1
import os
2
from signalwire.core.skill_base import SkillBase
3
from signalwire import FunctionResult
4
5
class WeatherSkill(SkillBase):
6
    SKILL_NAME = "weather"
7
    SKILL_DESCRIPTION = "Provides weather information"
8
    SKILL_VERSION = "1.0.0"
9
    REQUIRED_PACKAGES = ["requests"]
10
    REQUIRED_ENV_VARS = ["WEATHER_API_KEY"]
11
12
    def setup(self) -> bool:
13
        if not self.validate_packages():
14
            return False
15
        if not self.validate_env_vars():
16
            return False
17
        self.api_key = os.getenv("WEATHER_API_KEY")
18
        return True
19
20
    def register_tools(self):
21
        self.define_tool(
22
            name="get_weather",
23
            description="Get current weather for a location",
24
            handler=self._get_weather,
25
            parameters={
26
                "type": "object",
27
                "properties": {
28
                    "location": {
29
                        "type": "string",
30
                        "description": "City name or zip code"
31
                    }
32
                },
33
                "required": ["location"]
34
            }
35
        )
36
37
    def _get_weather(self, args, raw_data):
38
        import requests
39
        location = args.get("location")
40
        resp = requests.get(
41
            f"https://api.weather.com/v1/current?q={location}&key={self.api_key}"
42
        )
43
        data = resp.json()
44
        return FunctionResult(
45
            f"Weather in {location}: {data['condition']}, {data['temp']}F"
46
        )
47
48
    def get_hints(self):
49
        return ["weather", "temperature", "forecast"]
50
51
    def _get_prompt_sections(self):
52
        return [{
53
            "title": "Weather Information",
54
            "body": "You can check weather for any location."
55
        }]
56
57
    @classmethod
58
    def get_parameter_schema(cls):
59
        schema = super().get_parameter_schema()
60
        schema.update({
61
            "units": {
62
                "type": "string",
63
                "description": "Temperature units",
64
                "default": "fahrenheit",
65
                "enum": ["fahrenheit", "celsius"]
66
            }
67
        })
68
        return schema

1
from signalwire import AgentBase
2
3
agent = AgentBase(name="weather-agent")
4
agent.set_prompt_text("You are a helpful assistant.")
5
agent.add_skill("weather")
6
7
# Or with custom configuration
8
agent.add_skill("weather", {"units": "celsius"})
9
10
if __name__ == "__main__":
11
    agent.run()