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

aip_agents/examples/hello_world_a2a_langchain_client_streaming.py

@@ -0,0 +1,41 @@
+"""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)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+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=ChatOpenAI(model="gpt-4.1", streaming=True, temperature=0),
+    )
+
+    # 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?"):
+        print(chunk.get("content", ""))
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langchain_client_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 capabilities."""

aip_agents/examples/hello_world_a2a_langchain_reference_client_streaming.py

@@ -0,0 +1,60 @@
+"""Example A2A streaming client for the Google Search Tool and Time Tool Server.
+
+This client demonstrates streaming interaction with the reference server
+that provides time and mock Google search functionality via A2A protocol.
+
+To use this client:
+1. First start the reference server: python hello_world_a2a_langchain_reference_server.py
+2. Then run this client: python hello_world_a2a_langchain_reference_client_streaming.py
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+import pprint
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+async def main():
+    """Main function demonstrating streaming A2A client for the reference server."""
+    agent = LangChainAgent(
+        name="ReferenceClientAgent",
+        instruction="""You are a helpful assistant that can delegate tasks to specialized agents.
+        You can ask the mock retrieval server for mock data or mock Google search results.""",
+        model=ChatOpenAI(model="gpt-4.1", streaming=True, temperature=0),
+    )
+
+    # Discover agents
+    client_a2a_config = A2AClientConfig(
+        discovery_urls=["http://localhost:8455"],
+    )
+    agent_cards = agent.discover_agents(client_a2a_config)
+
+    if not agent_cards:
+        print("No agents discovered. Make sure the reference server is running on http://localhost:8455")
+        return
+
+    print(f"Discovered agent: {agent_cards[0].name}")
+    print(f"Description: {agent_cards[0].description}")
+    print(f"Skills: {[skill.name for skill in agent_cards[0].skills]}")
+
+    # Test queries to demonstrate the server's capabilities
+    test_queries = "Search for information about NeoAI and retrieve some mock data about artificial intelligence"
+
+    print(f"\n{'=' * 60}")
+    print(f"Query: {test_queries}")
+    print(f"{'=' * 60}")
+
+    async for chunk in agent.astream_to_agent(agent_card=agent_cards[0], message=test_queries):
+        print("-" * 50)
+        pprint.pprint(chunk)
+        print("-" * 50)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_a2a_langchain_reference_client_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 streaming A2A client for the reference server."""

aip_agents/examples/hello_world_a2a_langchain_reference_server.py

@@ -0,0 +1,105 @@
+"""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 import google_serper_tool, mock_retrieval_tool, time_tool
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+
+SERVER_AGENT_NAME = "Mock Retrieval Tool and Google Search Tool Server"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8455, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the LangChain Mock Retrieval and Google Search 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 server that provides mock retrieval data and mock google search information",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="mock_retrieval",
+                name="Mock Data Retrieval",
+                description="Retrieves mock data and returns references for testing.",
+                examples=["Retrieve mock data", "Get test information"],
+                tags=["retrieval", "mock"],
+            ),
+            AgentSkill(
+                id="google_serper",
+                name="Mock Google Search",
+                description="Provides mock google search information.",
+                examples=["What is NeoAI?", "Get NeoAI"],
+                tags=["google serper"],
+            ),
+        ],
+        tags=["retrieval", "mock", "google serper"],
+    )
+
+    time_delegate_agent = LangChainAgent(
+        name="TimeLookupAgent",
+        instruction="""You are a specialized time lookup assistant.
+        Always answer user queries by calling the `time_tool` to obtain the most up-to-date information.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0, streaming=True),
+        tools=[time_tool()],
+    )
+
+    google_delegate_agent = LangChainAgent(
+        name="GoogleSerperAgent",
+        instruction="""You are a specialized research assistant that must use the `google_serper_tool`
+        for every query. Provide concise summaries using the tool output.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0, streaming=True),
+        tools=[google_serper_tool()],
+    )
+
+    langchain_agent = LangChainAgent(
+        name=SERVER_AGENT_NAME,
+        instruction="""You are a coordinator agent that delegates research tasks to sub-agents but uses
+        mock_retrieval_tool directly. For research questions, delegate to `GoogleSerperAgent`.
+        For retrieval queries, use the mock_retrieval_tool directly. Synthesize their responses
+        and present a clear combined answer.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0, streaming=True),
+        tools=[mock_retrieval_tool()],  # Direct tool usage
+        agents=[time_delegate_agent, google_delegate_agent],  # Delegation for others
+    )
+
+    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_reference_server.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.examples.tools import google_serper_tool as google_serper_tool, mock_retrieval_tool as mock_retrieval_tool, time_tool as time_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 Mock Retrieval and Google Search 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.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.
+    """