Customer Service Chatbot

# A2A Python Client Example
# Source: https://github.com/a2aproject/a2a-samples/tree/main/samples/python

import asyncio
import httpx
from uuid import uuid4
from a2a.client import A2ACardResolver, A2AClient
from a2a.types import (
    AgentCard,
    Message,
    MessageSendParams,
    Part,
    Role,
    SendStreamingMessageRequest,
    TextPart,
)

class AgentClient:
    def __init__(self, agent_url: str):
        self.agent_url = agent_url
    
    async def get_agent_card(self) -> AgentCard:
        """Get the agent card information."""
        async with httpx.AsyncClient() as httpx_client:
            card_resolver = A2ACardResolver(httpx_client, self.agent_url)
            return await card_resolver.get_agent_card()
    
    async def send_message(self, message_text: str):
        """Send a message to the agent and get streaming response."""
        async with httpx.AsyncClient() as httpx_client:
            # Get agent card first
            agent_card = await self.get_agent_card()
            client = A2AClient(httpx_client, agent_card=agent_card)
            
            # Create message
            message = MessageSendParams(
                message=Message(
                    role=Role.user,
                    parts=[Part(TextPart(text=message_text))],
                    messageId=uuid4().hex,
                    taskId=uuid4().hex,
                )
            )
            
            # Send streaming message
            streaming_request = SendStreamingMessageRequest(
                id=str(uuid4().hex),
                params=message
            )
            
            async for chunk in client.send_message_streaming(streaming_request):
                if hasattr(chunk.root, 'result') and hasattr(chunk.root.result, 'status'):
                    message = chunk.root.result.status.message
                    if message and message.parts:
                        yield message.parts[0].root.text

async def main():
    """Example usage of the agent client."""
    client = AgentClient("https://chatbot.wingtip.com/a2a")
    
    # Get agent information
    agent_card = await client.get_agent_card()
    print(f"Agent: {agent_card.name}")
    print(f"Description: {agent_card.description}")
    
    # Send a message and stream the response
    async for response_chunk in client.send_message("Your input text here"):
        print(response_chunk, end='', flush=True)

if __name__ == '__main__':
    asyncio.run(main())