Launch A2A Agent

1
import uvicorn
2
import os
3
from a2a.server.apps import A2AStarletteApplication
4
from a2a.server.request_handlers import DefaultRequestHandler
5
from a2a.server.tasks import InMemoryTaskStore
6
from a2a.types import (
7
    AgentCapabilities,
8
    AgentCard,
9
    AgentProvider,
10
    AgentSkill,
11
)
12
13
from uagents_core.agentverse.sdk.a2a import agentverse_sdk  # Added(AVSDK): Agentverse SDK
14
15
from agent_executor import HelloWorldAgentExecutor  # type: ignore[import-untyped]
16
17
AGENT_URI = os.environ["AGENT_URI"]
18
AGENT_PUBLIC_URL = os.environ["AGENT_PUBLIC_URL"]
19
20
agentverse_sdk.init(AGENT_URI)  # Added(AVSDK): register this agent with Agentverse
21
22
if __name__ == "__main__":
23
    skill = AgentSkill(
24
        id="hello_world",
25
        name="Returns hello world",
26
        description="just returns hello world",
27
        tags=["hello world"],
28
        examples=["hi", "hello world"],
29
    )
30
31
    extended_skill = AgentSkill(
32
        id="super_hello_world",
33
        name="Returns a SUPER Hello World",
34
        description="A more enthusiastic greeting, only for authenticated users.",
35
        tags=["hello world", "super", "extended"],
36
        examples=["super hi", "give me a super hello"],
37
    )
38
39
    # This will be the public-facing agent card
40
    public_agent_card = AgentCard(
41
        name="Hello World Agent",
42
        description="Just a hello world agent",
43
        url=AGENT_PUBLIC_URL,  # Added(AVSDK): Agentverse public URL
44
        version="1.0.0",
45
        default_input_modes=["text"],
46
        default_output_modes=["text"],
47
        capabilities=AgentCapabilities(streaming=True),
48
        skills=[skill],  # Only the basic skill for the public card
49
        supports_authenticated_extended_card=True,
50
        provider=AgentProvider(organization="Agentverse", url="https://agentverse.ai/"),
51
        documentation_url="https://agentverse.ai/agents/details/agent1qf9v9nqc0j2yllctgk3eelpxj823dwhl4mefl6n7y3pftcdjpcvx5a8c08h/profile",
52
        icon_url="https://res.cloudinary.com/fetch-ai/image/upload/v1736444685/flockx-community-app/Community%20AI%20Assets/Avatar/fetch_ai_avatar_nnhewq.png",
53
    )
54
55
    # This will be the authenticated extended agent card
56
    # It includes the additional 'extended_skill'
57
    specific_extended_agent_card = public_agent_card.model_copy(
58
        update={
59
            "name": "Hello World Agent - Extended Edition",  # Different name for clarity
60
            "description": "The full-featured hello world agent for authenticated users.",
61
            "version": "1.0.1",  # Could even be a different version
62
            # Capabilities and other fields like url, default_input_modes, default_output_modes,
63
            # supports_authenticated_extended_card are inherited from public_agent_card unless specified here.
64
            "skills": [
65
                skill,
66
                extended_skill,
67
            ],  # Both skills for the extended card
68
        }
69
    )
70
71
    request_handler = DefaultRequestHandler(
72
        agent_executor=HelloWorldAgentExecutor(),
73
        task_store=InMemoryTaskStore(),
74
    )
75
76
    server = A2AStarletteApplication(
77
        agent_card=public_agent_card,
78
        http_handler=request_handler,
79
        extended_agent_card=specific_extended_agent_card,
80
    )
81
82
    app = server.build()
83
84
    uvicorn.run(app, host="0.0.0.0", port=9999)