text_client_cli.py

import argparse
import asyncio

import grpc
import httpx

from a2a.client import A2ACardResolver, create_text_client


async def main() -> None:
    """Run the simple A2A terminal client using TextClient."""
    parser = argparse.ArgumentParser(description='A2A Simple Text Client')
    parser.add_argument(
        '--url', default='http://127.0.0.1:41241', help='Agent base URL'
    )
    args = parser.parse_args()

    print(f'Connecting to {args.url}')

    async with httpx.AsyncClient() as httpx_client:
        resolver = A2ACardResolver(httpx_client, args.url)
        card = await resolver.get_agent_card()
        print(f'\n✓ Agent Card Found: {card.name}')

    text_client = await create_text_client(card)

    print('\nConnected! Send a message or type /quit to exit.')

    while True:
        try:
            loop = asyncio.get_running_loop()
            user_input = await loop.run_in_executor(None, input, 'You: ')
        except KeyboardInterrupt:
            break

        if user_input.lower() in ('/quit', '/exit'):
            break
        if not user_input.strip():
            continue

        try:
            response = await text_client.send_text_message(user_input)
            print(f'Agent: {response}')
        except (httpx.RequestError, grpc.RpcError) as e:
            print(f'Error communicating with agent: {e}')

    await text_client.close()


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