Development Environment Setup

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