Collection Agent using LiveKit

1
import logging
2
from dotenv import load_dotenv
3
from livekit.agents import JobContext, WorkerOptions, cli
4
from livekit.agents.voice import Agent, AgentSession
5
from livekit.plugins import sarvam
6
7
# Load environment variables
8
load_dotenv()
9
10
# Set up logging
11
logger = logging.getLogger("collection-agent")
12
logger.setLevel(logging.INFO)
13
14
15
class CollectionAgent(Agent):
16
    def __init__(self) -> None:
17
        super().__init__(
18
            # Collection agent personality and instructions
19
            instructions="""
20
                You are a professional and empathetic collection agent working for ABC Bank.
21
                
22
                Customer Account Details:
23
                - Bank Name: ABC Bank
24
                - EMI Amount: ₹5,000
25
                - Due Date: 15th January 2025
26
                - Loan Type: Personal Loan
27
                - Account Status: Payment Overdue
28
                
29
                Your responsibilities:
30
                - Remind customers about their pending EMI payment of ₹5,000 which was due on 15th January
31
                - Provide information about payment due dates, amounts, and available payment methods
32
                - Help customers understand their payment options and any applicable late fees
33
                - Guide customers through the payment process if they want to pay immediately
34
                - Address customer concerns about their account with empathy
35
                - Offer payment plans or extensions when appropriate (mention that you can connect them with a specialist)
36
                
37
                Payment Methods to mention:
38
                - UPI payment to ABC Bank
39
                - Net Banking
40
                - ABC Bank mobile app
41
                - Visit nearest ABC Bank branch
42
                
43
                Communication guidelines:
44
                - Always maintain a professional yet friendly tone
45
                - Be empathetic to customer's financial situations
46
                - Never be aggressive, threatening, or use inappropriate language
47
                - If a customer is upset, remain calm and understanding
48
                - Speak clearly and concisely
49
                - Confirm important details like EMI amount (₹5,000) and due date (15th January)
50
                - If customer requests to speak to a human, acknowledge and offer to transfer
51
                
52
                Start by greeting the customer, introducing yourself as calling from ABC Bank, 
53
                and politely remind them about their pending EMI of ₹5,000.
54
            """,
55
            
56
            # Saaras v3 STT - Converts speech to text
57
            stt=sarvam.STT(
58
                language="unknown",  # Auto-detect language
59
                model="saaras:v3",
60
                mode="transcribe"
61
            ),
62
            
63
            # Sarvam LLM - The "brain" that processes and generates responses
64
            llm=sarvam.LLM(model="sarvam-105b"),
65
            
66
            # Bulbul TTS - Converts text to speech
67
            tts=sarvam.TTS(
68
                target_language_code="en-IN",
69
                model="bulbul:v3",
70
                speaker="aditya"  # Professional male voice
71
            ),
72
        )
73
    
74
    async def on_enter(self):
75
        """Called when user joins - agent starts the conversation"""
76
        self.session.generate_reply()
77
78
79
async def entrypoint(ctx: JobContext):
80
    """Main entry point - LiveKit calls this when a user connects"""
81
    logger.info(f"User connected to room: {ctx.room.name}")
82
    
83
    # Create and start the agent session
84
    session = AgentSession()
85
    await session.start(
86
        agent=CollectionAgent(),
87
        room=ctx.room
88
    )
89
90
91
if __name__ == "__main__":
92
    # Run the agent
93
    cli.run_app(WorkerOptions(entrypoint_fnc=entrypoint))

1
stt=sarvam.STT(
2
    language="hi-IN",  # Hindi
3
    model="saaras:v3",
4
    mode="transcribe"
5
),
6
tts=sarvam.TTS(
7
    target_language_code="hi-IN",
8
    model="bulbul:v3",
9
    speaker="anand"  # Professional Hindi male voice
10
)

1
stt=sarvam.STT(language="ta-IN", model="saaras:v3", mode="transcribe"),
2
tts=sarvam.TTS(
3
    target_language_code="ta-IN",
4
    model="bulbul:v3",
5
    speaker="priya"
6
)

1
stt=sarvam.STT(language="unknown", model="saaras:v3", mode="transcribe"),  # Auto-detects language
2
tts=sarvam.TTS(target_language_code="en-IN", model="bulbul:v3", speaker="aditya")