aip-agents-binary 0.5.25b1__py3-none-macosx_13_0_arm64.whl

aip_agents/examples/hello_world_a2a_langchain_server.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/hello_world_a2a_langchain_server.py
+
+It will listen on http://localhost:8001 by default.
+
+Authors:
+    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 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 = "Weather Agent LangChain"
+
+
+@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=ChatOpenAI(model="gpt-4.1", temperature=0),
+        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_langchain_server.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.
+    """

aip_agents/examples/hello_world_a2a_langchain_server_lm_invoker.py

@@ -0,0 +1,78 @@
+"""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)
+"""
+
+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_langchain_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.
+    """

aip_agents/examples/hello_world_a2a_langflow_client.py

@@ -0,0 +1,83 @@
+"""Example of a LangflowAgent client that can interact with A2A servers via streaming.
+
+This example demonstrates:
+1. Configuring A2A settings for a client agent.
+2. Discovering and communicating with Langflow agents via A2A.
+3. Using streaming A2A communication for real-time responses.
+4. Handling streaming responses and providing relevant interactions.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+
+from dotenv import load_dotenv
+
+from aip_agents.agent.langflow_agent import LangflowAgent
+from aip_agents.schema.agent import A2AClientConfig
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+load_dotenv()
+
+
+async def main():
+    """Main function demonstrating the Langflow client with streaming A2A capabilities."""
+    # Create the LangflowAgent client
+    client_agent = LangflowAgent(
+        name="LangflowA2AClient",
+        flow_id="dummy-flow-id",  # Not used for client operations
+        description="Client agent for A2A streaming testing",
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8787"],
+    )
+
+    print("🔍 Discovering A2A agents...")
+    agent_cards = client_agent.discover_agents(client_a2a_config)
+
+    if not agent_cards:
+        logger.error("❌ No agents discovered!")
+        print("💡 Make sure the A2A server is running:")
+        print("   python aip_agents/examples/hello_world_a2a_langflow_server.py")
+        return
+
+    print(f"✅ Discovered {len(agent_cards)} agent(s):")
+    for i, agent_card in enumerate(agent_cards, 1):
+        print(f"  {i}. {agent_card.name} - {agent_card.description}")
+
+    # Use streaming A2A communication
+    query = "Tell me a short story about a robot learning to paint"
+    print(f"\n📡 Streaming Query: {query}")
+    print("🎯 Response (streaming):")
+    print("-" * 50)
+
+    try:
+        chunk_count = 0
+
+        async for chunk in client_agent.astream_to_agent(agent_card=agent_cards[0], message=query):
+            chunk_count += 1
+            content = chunk.get("content", "")
+            status = chunk.get("status", "unknown")
+            is_final = chunk.get("final", False)
+
+            if content:
+                if status == "success":
+                    print(content)
+                else:
+                    print(f"\n[{status}] {content}")
+
+            # Log completion but don't break - let stream finish naturally
+            if is_final:
+                print(f"\n\n✅ Stream completed! ({chunk_count} chunks received)")
+
+    except Exception as e:
+        logger.error(f"❌ Streaming error: {e}")
+        print(f"Error during A2A streaming: {e}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langflow_client.pyi

@@ -0,0 +1,9 @@
+from _typeshed import Incomplete
+from aip_agents.agent.langflow_agent import LangflowAgent as LangflowAgent
+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 Langflow client with streaming A2A capabilities."""

aip_agents/examples/hello_world_a2a_langflow_server.py

@@ -0,0 +1,82 @@
+"""Example A2A server for a LangflowAgent.
+
+This server instantiates a LangflowAgent and serves it via the A2A protocol
+using the proper A2A server setup.
+
+To run this server:
+    python examples/hello_world_a2a_langflow_server.py
+
+It will listen on http://localhost:8787 by default.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+from dotenv import load_dotenv
+
+from aip_agents.agent.langflow_agent import LangflowAgent
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+SERVER_AGENT_NAME = "LangflowA2AServer"
+load_dotenv()
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8787, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the Langflow 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 Langflow agent that provides AI assistance through A2A protocol",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        author="AIP Agents SDK",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="general_assistance",
+                name="General Assistance",
+                description="Provides general AI assistance through Langflow",
+                examples=[
+                    "Answer questions about any topic",
+                    "Help with writing and editing",
+                    "Provide explanations and analysis",
+                    "Creative content generation",
+                ],
+                tags=["ai", "assistant", "langflow", "chat"],
+            )
+        ],
+        tags=["langflow", "ai", "assistant"],
+    )
+
+    # Create the Langflow agent
+    langflow_agent = LangflowAgent(
+        name=SERVER_AGENT_NAME,
+        flow_id="6dd45ac0-5c05-44c1-9825-66c6c6e516f7",
+        description="A Langflow agent that provides AI assistance through A2A protocol",
+        base_url="https://langflow.obrol.id",
+    )
+
+    app = langflow_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_langflow_server.pyi

@@ -0,0 +1,14 @@
+from _typeshed import Incomplete
+from aip_agents.agent.langflow_agent import LangflowAgent as LangflowAgent
+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 Langflow 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_artifact_client.py

@@ -0,0 +1,73 @@
+"""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 artifacts in the response.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+import base64
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+async def main():
+    """Main function demonstrating the General Assistant agent with 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:8999"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    # Test query
+    query = "Generate a data table with 5 rows and 3 columns showing Name, Age, and City and Create a simple image."
+    result = assistant_agent.send_to_agent(agent_card=agent_cards[0], message=query)
+    print(f"Result: {result}")
+
+    # Handle artifacts from result
+    artifacts = result.get("artifacts", [])
+    if artifacts:
+        print(f"\n--- {len(artifacts)} Artifacts Received ---")
+        for artifact in artifacts:
+            name = artifact.get("name", "unknown")
+            mime_type = artifact.get("mime_type", "unknown")
+            file_data = artifact.get("file_data")
+
+            print(f"Name: {name}")
+            print(f"Type: {mime_type}")
+
+            if file_data:
+                try:
+                    if mime_type == "text/csv":
+                        decoded_content = base64.b64decode(file_data).decode("utf-8")
+                        print(f"CSV Preview (first 200 chars): {decoded_content[:200]}...")
+                    elif mime_type.startswith("image/"):
+                        file_size = len(base64.b64decode(file_data))
+                        print(f"Image size: {file_size} bytes")
+                    else:
+                        print("File data available (base64 encoded)")
+                except Exception as e:
+                    print(f"Could not decode file data: {e}")
+            print("---")
+        print("--- End Artifacts ---\n")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langgraph_artifact_client.pyi

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

aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.py

@@ -0,0 +1,76 @@
+"""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
+import base64
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+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:8999"],
+    )
+    agent_cards = assistant_agent.discover_agents(client_a2a_config)
+
+    # Test query
+    query = "Generate me just 1 single image and a table data of 10 rows"
+
+    async for chunk in assistant_agent.astream_to_agent(agent_card=agent_cards[0], message=query):
+        content = chunk.get("content", "")
+        if content:
+            print(content)
+
+        # Handle artifacts
+        if chunk.get("status") == "success" and chunk.get("artifacts"):
+            for artifact in chunk.get("artifacts", []):
+                artifact_name = artifact.get("name", "unknown")
+                mime_type = artifact.get("mime_type", "unknown")
+                file_data = artifact.get("file_data")
+
+                print("\n--- Artifact Received ---")
+                print(f"Name: {artifact_name}")
+                print(f"Type: {mime_type}")
+
+                if file_data:
+                    try:
+                        if mime_type == "text/csv":
+                            decoded_content = base64.b64decode(file_data).decode("utf-8")
+                            print(f"CSV Preview (first 200 chars): {decoded_content[:200]}...")
+                        elif mime_type.startswith("image/"):
+                            file_size = len(base64.b64decode(file_data))
+                            print(f"Image size: {file_size} bytes")
+                        else:
+                            print("File data available (base64 encoded)")
+                    except Exception as e:
+                        print(f"Could not decode file data: {e}")
+
+            print("--- End Artifact ---\n")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langgraph_artifact_client_streaming.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+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_artifact_server.py

@@ -0,0 +1,92 @@
+"""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)
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.tools.image_artifact_tool import ImageArtifactTool
+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 = "ArtifactAgent"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8999, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the LangGraph Artifact Generation 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="An agent that can generate data tables and images as downloadable artifacts.",
+        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 and provides them as downloadable CSV files.",
+                examples=["Generate a data table with 5 rows and 3 columns", "Create a table with customer data"],
+                tags=["table", "data", "csv"],
+            ),
+            AgentSkill(
+                id="image_generation",
+                name="Image Generation",
+                description="Creates simple images and provides them as downloadable PNG files.",
+                examples=["Create an image", "Generate a simple graphic"],
+                tags=["image", "graphics", "png"],
+            ),
+        ],
+        tags=["artifacts", "files", "generation"],
+    )
+
+    table_tool = TableGeneratorTool()
+    image_tool = ImageArtifactTool()
+    tools = [table_tool, image_tool]
+
+    langgraph_agent = LangGraphAgent(
+        name=SERVER_AGENT_NAME,
+        instruction=(
+            "You are a helpful assistant that can generate data tables and images. "
+            "When users ask for data tables, use the table_generator tool to create both a preview "
+            "and a downloadable CSV file. When users ask for images, use the image_generator tool "
+            "to create visual content. Always provide clear descriptions of what you're generating."
+            "Format your responses clearly and professionally."
+        ),
+        model="openai/gpt-4.1",
+        tools=tools,
+    )
+
+    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_artifact_server.pyi

@@ -0,0 +1,16 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+from aip_agents.examples.tools.image_artifact_tool import ImageArtifactTool as ImageArtifactTool
+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 LangGraph Artifact Generation A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """