aip-agents-binary 0.6.4__py3-none-any.whl

aip_agents/examples/compare_streaming_client.pyi

@@ -0,0 +1,48 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+from aip_agents.examples.tools.mock_retrieval_tool import MockRetrievalTool as MockRetrievalTool
+from aip_agents.examples.tools.pii_demo_tools import get_customer_info as get_customer_info, get_employee_data as get_employee_data, get_user_profile as get_user_profile
+from aip_agents.examples.tools.random_chart_tool import RandomChartTool as RandomChartTool
+from aip_agents.examples.tools.table_generator_tool import TableGeneratorTool as TableGeneratorTool
+from aip_agents.schema.a2a import A2AStreamEventType as A2AStreamEventType
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+from aip_agents.utils.logger import get_logger as get_logger
+from typing import Any
+
+logger: Incomplete
+SERVER_URL: str
+
+def create_local_agent(enable_token_streaming: bool = False) -> LangGraphAgent:
+    """Create a local agent with the same tools as the server.
+
+    Args:
+        enable_token_streaming: Whether to enable token streaming for content_chunk events.
+    """
+def format_chunk_summary(chunk: dict[str, Any], index: int) -> str:
+    """Format a chunk for display."""
+async def run_direct_streaming(agent: LangGraphAgent, query: str, pii_mapping: dict[str, str] | None = None) -> list[dict[str, Any]]:
+    """Run arun_sse_stream and collect all chunks."""
+async def run_connector_streaming(agent: LangGraphAgent, query: str, pii_mapping: dict[str, str] | None = None) -> list[dict[str, Any]]:
+    """Run astream_to_agent via A2A connector and collect all chunks."""
+def get_chunk_keys(chunks: list[dict[str, Any]]) -> set[str]:
+    """Get all unique keys across all chunks."""
+def get_field_types(chunks: list[dict[str, Any]], field: str) -> set[str]:
+    """Get all types seen for a field across chunks.
+
+    Note: StrEnum values are reported as 'str' since they ARE strings
+    and serialize identically over HTTP/SSE.
+    """
+def group_chunks_by_event_type(chunks: list[dict[str, Any]]) -> dict[str, list[dict[str, Any]]]:
+    """Group chunks by their event_type."""
+def compare_chunk_structure(chunk1: dict[str, Any], chunk2: dict[str, Any]) -> dict[str, Any]:
+    """Compare structure of two chunks and return differences."""
+def compare_event_type_groups(direct_groups: dict[str, list[dict[str, Any]]], connector_groups: dict[str, list[dict[str, Any]]]) -> dict[str, Any]:
+    """Compare chunks grouped by event type - RAW comparison without filtering."""
+def compare_chunks(direct_chunks: list[dict[str, Any]], connector_chunks: list[dict[str, Any]]) -> dict[str, Any]:
+    """Compare the two chunk lists and produce a comprehensive structural summary."""
+def print_event_type_comparison(event_comparison: dict[str, Any]) -> None:
+    """Print per-event-type comparison details with exact key/value examples."""
+def print_structure_comparison(comparison: dict[str, Any]) -> None:
+    """Print detailed structure comparison."""
+async def main() -> None:
+    """Main comparison workflow."""

aip_agents/examples/compare_streaming_server.py

@@ -0,0 +1,142 @@
+"""A2A server with artifact and reference tools for streaming comparison tests.
+
+This server provides an agent with:
+- table_generator tool: Returns artifacts (CSV files)
+- mock_retrieval tool: Returns references
+
+To run this server:
+    cd libs/aip_agents
+    poetry run python -m aip_agents.examples.compare_streaming_server
+
+It will listen on http://localhost:18999 by default.
+
+Authors:
+    AI Agent Platform Team
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.tools.mock_retrieval_tool import MockRetrievalTool
+from aip_agents.examples.tools.pii_demo_tools import (
+    get_customer_info,
+    get_employee_data,
+    get_user_profile,
+)
+from aip_agents.examples.tools.random_chart_tool import RandomChartTool
+from aip_agents.examples.tools.table_generator_tool import TableGeneratorTool
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+SERVER_AGENT_NAME = "StreamingComparisonAgent"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=18999, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the streaming comparison A2A server.
+
+    Args:
+        host: Host to bind the server to.
+        port: Port to bind the server to.
+    """
+    print(f"Starting {SERVER_AGENT_NAME} on http://{host}:{port}")
+
+    agent_card = AgentCard(
+        name=SERVER_AGENT_NAME,
+        description="Agent for comparing direct vs connector streaming with artifacts and references.",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="table_generation",
+                name="Table Generation",
+                description="Generates data tables as CSV artifacts.",
+                examples=["Generate a table with 3 rows"],
+                tags=["table", "artifact"],
+            ),
+            AgentSkill(
+                id="mock_retrieval",
+                name="Mock Retrieval",
+                description="Retrieves mock data with references.",
+                examples=["Search for test data"],
+                tags=["retrieval", "references"],
+            ),
+            AgentSkill(
+                id="customer_info",
+                name="Get Customer Information",
+                description="Retrieves customer information including email and phone number (PII masked).",
+                examples=["Get customer info for C001", "What is the email for customer C002?"],
+                tags=["customer", "pii"],
+            ),
+            AgentSkill(
+                id="employee_data",
+                name="Get Employee Data",
+                description="Retrieves employee data including name, email, and salary (PII masked).",
+                examples=["Get employee data for E001", "What is the salary for employee E002?"],
+                tags=["employee", "pii"],
+            ),
+            AgentSkill(
+                id="user_profile",
+                name="Get User Profile",
+                description="Retrieves user profile information with personal details (PII masked).",
+                examples=["Get user profile for U001", "What is the email for user U002?"],
+                tags=["user", "pii"],
+            ),
+        ],
+        tags=["test", "comparison", "artifacts", "references", "pii"],
+    )
+
+    # Token streaming disabled for now - will compare token streaming later
+    llm = ChatOpenAI(model="gpt-4.1", temperature=0, streaming=False)
+    table_tool = TableGeneratorTool()
+    mock_retrieval_tool = MockRetrievalTool()
+    random_chart_tool = RandomChartTool()
+    tools = [table_tool, mock_retrieval_tool, get_customer_info, get_employee_data, get_user_profile]
+
+    visualization_agent = LangGraphAgent(
+        name="RandomChartAgent",
+        instruction=(
+            "You are a visualization specialist. Whenever asked to produce a chart, visualization, or images "
+            "that summarize insights, call the random_chart_tool to generate a bar chart artifact. "
+            "Describe what the generated image represents."
+        ),
+        model=llm,
+        tools=[random_chart_tool],
+    )
+
+    agent = LangGraphAgent(
+        name=SERVER_AGENT_NAME,
+        instruction=(
+            "You are a helpful assistant for testing streaming comparison. "
+            "When asked for a table, use the table_generator tool. "
+            "When asked to search or retrieve, use the mock_retrieval tool. "
+            "When asked for customer information, use the get_customer_info tool. "
+            "When asked for employee data, use the get_employee_data tool. "
+            "When asked for user profile, use the get_user_profile tool. "
+            "IMPORTANT: When you receive PII placeholders like <PERSON_1>, pass them WITH the angle brackets <> "
+            "to the tools - they are required for the PII system to work correctly. "
+            "Always use the tools when relevant to demonstrate artifacts, references, and PII masking."
+        ),
+        model=llm,
+        tools=tools,
+        enable_a2a_token_streaming=False,
+        agents=[visualization_agent],
+    )
+
+    app = agent.to_a2a(agent_card=agent_card)
+
+    print("A2A application configured. Starting Uvicorn server...")
+    uvicorn.run(app, host=host, port=port)
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/compare_streaming_server.pyi

@@ -0,0 +1,18 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+from aip_agents.examples.tools.mock_retrieval_tool import MockRetrievalTool as MockRetrievalTool
+from aip_agents.examples.tools.pii_demo_tools import get_customer_info as get_customer_info, get_employee_data as get_employee_data, get_user_profile as get_user_profile
+from aip_agents.examples.tools.random_chart_tool import RandomChartTool as RandomChartTool
+from aip_agents.examples.tools.table_generator_tool import TableGeneratorTool as TableGeneratorTool
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+def main(host: str, port: int):
+    """Runs the streaming comparison A2A server.
+
+    Args:
+        host: Host to bind the server to.
+        port: Port to bind the server to.
+    """

aip_agents/examples/hello_world_a2a_google_adk_client.py

@@ -0,0 +1,49 @@
+"""Example of a General Assistant GoogleADKAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import json
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    agent_name = "GoogleAssistantAgent"
+    # Create the agent with simplified instructions tailored to the agent name
+    assistant_agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that can help with various tasks by delegating to specialized agents.",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8002"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    query = "What is the weather in Jakarta?"
+    logger.info(f"Processing Query: {query}")
+
+    response = assistant_agent.send_to_agent(agent_cards[0], query)
+    logger.info(f"Raw response: \n\n{json.dumps(response, indent=4)}Q")
+    logger.info(f"Agent Response: {response['content']}")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_google_adk_client.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_google_adk_client_agent.py

@@ -0,0 +1,48 @@
+"""Example of a General Assistant GoogleADKAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    agent_name = "GoogleAssistantAgent"
+    assistant_agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that can help with various tasks by delegating to specialized agents.",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8002"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    # Register agents
+    assistant_agent.register_a2a_agents(agent_cards)
+
+    query = "What is the weather in Jakarta?"
+    logger.info(f"Processing Query: {query}")
+
+    response = assistant_agent.run(query)
+    logger.info(f"Assistant Agent Response: {response['output']}")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_google_adk_client_agent.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_google_adk_client_streaming.py

@@ -0,0 +1,60 @@
+"""Example of a General Assistant GoogleADKAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A with streaming.
+4. Handling streaming responses and providing relevant advice.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+
+from a2a.types import TaskState
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+async def main():
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""
+    agent_name = "GoogleADKWeatherAgent"
+
+    # Create the agent with simplified instructions and our tool
+    assistant_agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that can help with various tasks by delegating to specialized agents.",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8002"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    query = "What is the weather in Jakarta?"
+    logger.info(f"Processing Query: {query}")
+
+    agent_result = ""
+    async for chunk in assistant_agent.astream_to_agent(agent_card=agent_cards[0], message=query):
+        task_state = chunk.get("task_state")
+        content = chunk.get("content", "")
+        if task_state == TaskState.working and content:
+            logger.info(f"Event Working: {content}")
+        elif task_state == TaskState.completed and content:
+            agent_result += content
+
+    logger.info(f"Agent Response: {agent_result}")
+    return agent_result
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_google_adk_client_streaming.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+async def main():
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""

aip_agents/examples/hello_world_a2a_google_adk_server.py

@@ -0,0 +1,79 @@
+"""Example A2A server for a Google ADK Weather Service.
+
+This server instantiates a Google ADK agent with weather lookup capabilities and serves it
+via the A2A protocol using GoogleADKAgent's to_a2a method.
+
+To run this server:
+    python examples/hello_world_a2a_google_adk_server.py
+
+It will listen on http://localhost:8002 by default.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.tools.adk_weather_tool import weather_tool
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+SERVER_AGENT_NAME = "GoogleADKWeatherAgent"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8002, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the Google ADK Weather A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """
+    logger.info(f"Starting {SERVER_AGENT_NAME} on http://{host}:{port}")
+
+    agent_card = AgentCard(
+        name=SERVER_AGENT_NAME,
+        description="A weather agent based on Google ADK that provides weather information for cities.",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="weather",
+                name="Weather Lookup",
+                description="Provides current weather information for cities.",
+                examples=["What's the weather in Tokyo?", "Get weather for London"],
+                tags=["weather"],
+            )
+        ],
+        tags=["weather"],
+    )
+
+    # Create a Google ADK agent
+    agent = GoogleADKAgent(
+        model="gemini-2.0-flash",
+        name=SERVER_AGENT_NAME,
+        description="A weather agent that provides weather information for cities.",
+        instruction="""You are a weather agent that provides weather information for cities.
+        Always use the weather_tool for looking up weather data. Format your responses clearly and professionally.""",
+        tools=[weather_tool],
+    )
+
+    # Convert the agent to an A2A server directly
+    app = agent.to_a2a(agent_card=agent_card)
+
+    uvicorn.run(app, host=host, port=port)
+    logger.info("A2A application configured. Starting Uvicorn server...")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_google_adk_server.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.tools.adk_weather_tool import weather_tool as weather_tool
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+def main(host: str, port: int):
+    """Runs the Google ADK Weather A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """

aip_agents/examples/hello_world_a2a_langchain_client.py

@@ -0,0 +1,39 @@
+"""Example of a General Assistant LangChainAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    assistant_agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    response = assistant_agent.send_to_agent(agent_cards[0], message="What is the weather in Jakarta?")
+    print(response.get("content", str(response)))
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langchain_client.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_langchain_client_agent.py

@@ -0,0 +1,39 @@
+"""Example of a General Assistant LangChainAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    assistant_agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+    assistant_agent.register_a2a_agents(agent_cards)
+
+    response = assistant_agent.run("What is the weather in Jakarta?")
+    print(response["output"])
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langchain_client_agent.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""

aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.py

@@ -0,0 +1,37 @@
+"""Example of a General Assistant LangChainAgent that can delegate tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Creating a general assistant agent that can help with various queries.
+3. Delegating specific tasks to specialized agents via A2A.
+4. Handling responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+def main():
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""
+    assistant_agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model="openai/gpt-4o",
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    response = assistant_agent.send_to_agent(agent_cards[0], message="What is the weather in Jakarta?")
+    print(response.get("content", str(response)))
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langchain_client_lm_invoker.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+def main() -> None:
+    """Main function demonstrating the General Assistant agent with A2A capabilities."""