aip-agents-binary 0.5.20__py3-none-manylinux_2_31_x86_64.whl

aip_agents/examples/hello_world_a2a_langgraph_client.py

@@ -0,0 +1,54 @@
+"""Example of a General Assistant LangGraphAgent 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)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import json
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+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."""
+    # Init Assistant Agent
+    llm = ChatOpenAI(model="gpt-4.1")
+
+    # Create and return the LangGraphAgent
+    assistant_agent = LangGraphAgent(
+        name="AssistantAgent",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=llm,
+        tools=[],
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    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)}")
+    logger.info(f"Agent Response: {response['content']}")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langgraph_client.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+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_langgraph_client_agent.py

@@ -0,0 +1,54 @@
+"""Example of a General Assistant LangGraphAgent 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)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+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."""
+    # Create the LLM
+    llm = ChatOpenAI(model="gpt-4.1")
+
+    # Create and return the LangGraphAgent
+    assistant_agent = LangGraphAgent(
+        name="AssistantAgent",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=llm,
+        tools=[],
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    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_langgraph_client_agent.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+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_langgraph_client_agent_lm_invoker.py

@@ -0,0 +1,32 @@
+"""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)
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+if __name__ == "__main__":
+    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-4.1",
+    )
+
+    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"])

aip_agents/examples/hello_world_a2a_langgraph_client_agent_lm_invoker.pyi

@@ -0,0 +1,2 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig

aip_agents/examples/hello_world_a2a_langgraph_client_streaming.py

@@ -0,0 +1,50 @@
+"""Example of a General Assistant LangGraphAgent 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)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+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."""
+    llm = ChatOpenAI(model="gpt-4.1", streaming=True)
+    assistant_agent = LangGraphAgent(
+        name="AssistantAgent",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model=llm,
+        tools=[],
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    query = "What is the weather in Jakarta?"
+    logger.info(f"Processing Query: {query}")
+
+    async for chunk in assistant_agent.astream_to_agent(agent_card=agent_cards[0], message=query):
+        print(chunk.get("content", ""))
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langgraph_client_streaming.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+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() -> None:
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""

aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.py

@@ -0,0 +1,44 @@
+"""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 with streaming.
+4. Handling streaming responses and providing relevant advice.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+from pprint import pprint
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+async def main():
+    """Main function demonstrating the General Assistant agent with streaming A2A capabilities."""
+    agent = LangChainAgent(
+        name="AssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents.""",
+        model="openai/gpt-4.1",
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8001"],
+    )
+    agent_cards = agent.discover_agents(client_a2a_config)
+
+    async for chunk in agent.astream_to_agent(
+        agent_card=agent_cards[0], message="What is the weather in Jakarta and London?"
+    ):
+        pprint(chunk)
+        print("-" * 50)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langgraph_client_streaming_lm_invoker.pyi

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

aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.py

@@ -0,0 +1,92 @@
+"""Example of a General Assistant LangChainAgent that delegates time and weather tasks to specialized agents.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent to connect to time/weather services.
+2. Creating a general assistant agent that can help with time and weather queries.
+3. Delegating specific time and weather tasks to specialized agents via A2A with streaming.
+4. Handling streaming responses from time and weather tasks.
+5. Real-time display of time and weather information and agent thinking process.
+
+The client requests the time/weather agent to get current time and weather forecast,
+demonstrating streaming capabilities of the LangGraph agent wrapper.
+
+To run this example:
+1. First start the time/weather agent server:
+   python examples/hello_world_a2a_langgraph_server_tool_streaming.py --port 8003
+2. Then run this client:
+   python examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.py
+
+Environment Variables Required:
+    OPENAI_API_KEY: OpenAI API key for the LLM
+
+Authors:
+    Fachriza Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+"""
+
+import asyncio
+from pprint import pprint
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+async def main():
+    """Main function demonstrating the General Assistant agent with streaming A2A time/weather capabilities."""
+    print("🚀 Starting Time & Weather Agent Client Demo")
+    print("=" * 60)
+
+    # Create a general assistant agent
+    agent = LangChainAgent(
+        name="TimeWeatherAssistantAgentLangChain",
+        instruction="""You are a helpful assistant that can help with various tasks
+        by delegating to specialized agents. You have access to time and weather
+        agents that can provide current time information and weather data
+        for specific cities.
+
+        When users ask for time or weather-related tasks, delegate them to the
+        time/weather agent. Provide clear summaries of the results and help
+        interpret the information that's provided.""",
+        model="openai/gpt-4o",
+    )
+
+    # Configure A2A client to discover time/weather agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8003"],  # Time/weather agent server
+    )
+
+    print("🔍 Discovering available time & weather agents...")
+    try:
+        agent_cards = agent.discover_agents(client_a2a_config)
+
+        if not agent_cards:
+            print("❌ No time & weather agents found!")
+            print("   Make sure the agent server is running on http://localhost:8003")
+            return
+
+        print(f"✅ Found {len(agent_cards)} time & weather agent(s)")
+
+        for i, card in enumerate(agent_cards):
+            print(f"   {i + 1}. {card.name}: {card.description}")
+            if card.skills:
+                print(f"      Skills: {', '.join([skill.name for skill in card.skills])}")
+
+        task = "what time is it and what is the weather in jakarta?"
+
+        print(f"📝 Task: {task}")
+
+        async for chunk in agent.astream_to_agent(agent_card=agent_cards[0], message=task):
+            print("-" * 40)
+            pprint(chunk)
+
+        print("\n" + "=" * 60)
+        print("\n🎉 Time & Weather agent task finished!")
+
+    except Exception as e:
+        print(f"❌ Error during time & weather query: {str(e)}")
+        print("   Make sure:")
+        print("   1. The time & weather server is running (python hello_world_a2a_langgraph_server_agent_wrapper.py)")
+        print("   2. OPENAI_API_KEY is set in your environment")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langgraph_client_streaming_tool_streaming.pyi

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

aip_agents/examples/hello_world_a2a_langgraph_server.py

@@ -0,0 +1,84 @@
+"""Example A2A server for a LangGraphAgent Weather Service.
+
+This server instantiates a LangGraphAgent with weather lookup capabilities and serves it
+via the A2A protocol using the proper A2A server setup with LangGraphA2AExecutor.
+
+To run this server:
+    python examples/hello_world_a2a_langgraph_server.py
+
+It will listen on http://localhost:8001 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 langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.tools import weather_tool_langchain as weather_tool
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+SERVER_AGENT_NAME = "WeatherAgent"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8001, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the LangGraph 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 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"],
+    )
+
+    llm = ChatOpenAI(model="gpt-4.1", temperature=0, streaming=True)
+    tools = [weather_tool]
+
+    langgraph_agent = LangGraphAgent(
+        name=SERVER_AGENT_NAME,
+        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."
+        ),
+        model=llm,
+        tools=tools,
+        enable_a2a_token_streaming=True,
+    )
+
+    app = langgraph_agent.to_a2a(agent_card=agent_card)
+
+    logger.info("A2A application configured. Starting Uvicorn server...")
+    uvicorn.run(app, host=host, port=port)
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_a2a_langgraph_server.pyi

@@ -0,0 +1,14 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+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 LangGraph 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_langgraph_server_lm_invoker.py

@@ -0,0 +1,79 @@
+"""Example A2A server for a LangChainAgent Weather Service.
+
+This server instantiates a LangChainAgent with weather lookup capabilities and serves it
+via the A2A protocol using the to_a2a convenience method.
+
+To run this server:
+    python examples/a2a/langchain_server_example.py
+
+It will listen on http://localhost:8001 by default.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.examples.tools.langchain_weather_tool import weather_tool
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+SERVER_AGENT_NAME = "WeatherAgentLangChain"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8001, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the LangChain 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 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"],
+    )
+
+    langchain_agent = LangChainAgent(
+        name=SERVER_AGENT_NAME,
+        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.""",
+        model="openai/gpt-4o",
+        tools=[weather_tool],
+    )
+
+    app = langchain_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_langgraph_server_lm_invoker.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.examples.tools.langchain_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 LangChain Weather A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """