Dev Environment

1
#!/usr/bin/env python3
2
## main.py - Main entry point with environment loading
3
"""Main entry point with environment loading."""
4
5
import os
6
from dotenv import load_dotenv
7
8
## Load environment variables from .env file
9
load_dotenv()
10
11
from agents.customer_service import CustomerServiceAgent
12
13
14
def main():
15
    agent = CustomerServiceAgent()
16
17
    # Use environment variables
18
    host = os.getenv("AGENT_HOST", "0.0.0.0")
19
    port = int(os.getenv("AGENT_PORT", "3000"))
20
21
    print(f"Starting agent on {host}:{port}")
22
    agent.run(host=host, port=port)
23
24
25
if __name__ == "__main__":
26
    main()

1
#!/usr/bin/env python3
2
## customer_service.py - Customer service agent
3
"""
4
Customer Service Agent
5
6
A production-ready customer service agent template.
7
"""
8
9
import os
10
from signalwire_agents import AgentBase, SwaigFunctionResult
11
12
13
class CustomerServiceAgent(AgentBase):
14
    """Customer service voice AI agent."""
15
16
    def __init__(self):
17
        super().__init__(
18
            name="customer-service",
19
            route="/",
20
            host="0.0.0.0",
21
            port=int(os.getenv("AGENT_PORT", "3000"))
22
        )
23
24
        self._configure_voice()
25
        self._configure_prompts()
26
        self._configure_functions()
27
28
    def _configure_voice(self):
29
        """Set up voice and language."""
30
        self.add_language("English", "en-US", "rime.spore")
31
32
        self.set_params({
33
            "end_of_speech_timeout": 500,
34
            "attention_timeout": 15000,
35
        })
36
37
        self.add_hints([
38
            "account",
39
            "billing",
40
            "support",
41
            "representative"
42
        ])
43
44
    def _configure_prompts(self):
45
        """Set up AI prompts."""
46
        self.prompt_add_section(
47
            "Role",
48
            "You are a helpful customer service representative for Acme Corp. "
49
            "Help customers with their questions about accounts, billing, and products."
50
        )
51
52
        self.prompt_add_section(
53
            "Guidelines",
54
            body="Follow these guidelines:",
55
            bullets=[
56
                "Be professional and courteous",
57
                "Ask clarifying questions when needed",
58
                "Offer to transfer to a human if you cannot help",
59
                "Keep responses concise"
60
            ]
61
        )
62
63
    def _configure_functions(self):
64
        """Register SWAIG functions."""
65
        self.define_tool(
66
            name="lookup_account",
67
            description="Look up a customer account by phone number or account ID",
68
            parameters={
69
                "type": "object",
70
                "properties": {
71
                    "identifier": {
72
                        "type": "string",
73
                        "description": "Phone number or account ID"
74
                    }
75
                },
76
                "required": ["identifier"]
77
            },
78
            handler=self.lookup_account
79
        )
80
81
        self.define_tool(
82
            name="transfer_to_human",
83
            description="Transfer the call to a human representative",
84
            parameters={"type": "object", "properties": {}},
85
            handler=self.transfer_to_human
86
        )
87
88
    def lookup_account(self, args, raw_data):
89
        """Look up account information."""
90
        identifier = args.get("identifier", "")
91
92
        # In production, query your database here
93
        return SwaigFunctionResult(
94
            f"Found account for {identifier}: Status is Active, Balance is $0.00"
95
        )
96
97
    def transfer_to_human(self, args, raw_data):
98
        """Transfer to human support."""
99
        return SwaigFunctionResult(
100
            "Transferring you to a human representative now."
101
        ).connect("+15551234567", final=True, from_addr="+15559876543")
102
103
104
## Allow running directly for testing
105
if __name__ == "__main__":
106
    agent = CustomerServiceAgent()
107
    agent.run()

1
#!/usr/bin/env python3
2
## test_agents.py - Tests for agents
3
"""Tests for agents."""
4
5
import pytest
6
from agents.customer_service import CustomerServiceAgent
7
8
9
class TestCustomerServiceAgent:
10
    """Test customer service agent."""
11
12
    def setup_method(self):
13
        """Set up test fixtures."""
14
        self.agent = CustomerServiceAgent()
15
16
    def test_agent_name(self):
17
        """Test agent has correct name."""
18
        assert self.agent.name == "customer-service"
19
20
    def test_lookup_account(self):
21
        """Test account lookup function."""
22
        result = self.agent.lookup_account(
23
            {"identifier": "12345"},
24
            {}
25
        )
26
        assert "Found account" in result
27
28
    def test_has_functions(self):
29
        """Test agent has expected functions."""
30
        functions = self.agent._tool_registry.get_function_names()
31
        assert "lookup_account" in functions
32
        assert "transfer_to_human" in functions