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

aip_agents/examples/hello_world_a2a_with_metadata_langchain_server_lm_invoker.py

@@ -0,0 +1,80 @@
+"""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 dotenv import load_dotenv
+
+from aip_agents.agent import LangGraphReactAgent
+from aip_agents.examples.hello_world_a2a_langchain_server import SERVER_AGENT_NAME
+from aip_agents.examples.tools.langchain_currency_exchange_tool import CurrencyExchangeTool
+from aip_agents.utils.logger import get_logger
+
+load_dotenv()
+
+logger = get_logger(__name__)
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8885, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the LangChain Currency Exchange 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="currency_exchange",
+                name="Currency Exchange",
+                description="Provides current currency exchange information for cities.",
+                examples=["What's the currency exchange in Tokyo?", "Get currency exchange for London"],
+                tags=["currency_exchange"],
+            )
+        ],
+        tags=["currency_exchange"],
+    )
+
+    agent = LangGraphReactAgent(
+        name="CurrencyAgent",
+        instruction="You are a currency exchange agent. Use the currency_exchange tool for conversions.",
+        model="openai/gpt-4o-mini",
+        tools=[CurrencyExchangeTool()],
+        tool_configs={"currency_exchange": {"tenant_id": "premium_corp", "auth_key": "premium_key_123"}},
+    )
+
+    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_with_metadata_langchain_server_lm_invoker.pyi

@@ -0,0 +1,15 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
+from aip_agents.examples.hello_world_a2a_langchain_server import SERVER_AGENT_NAME as SERVER_AGENT_NAME
+from aip_agents.examples.tools.langchain_currency_exchange_tool import CurrencyExchangeTool as CurrencyExchangeTool
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+def main(host: str, port: int):
+    """Runs the LangChain Currency Exchange A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """

aip_agents/examples/hello_world_google_adk.py

@@ -0,0 +1,41 @@
+"""Minimal example demonstrating the GoogleADKAgent with tool usage and async operation.
+
+This example shows how to create a simple calculator agent using Google's ADK
+which automatically handles tool calling and conversation flow.
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.tools.adk_arithmetic_tools import sum_numbers
+
+
+async def google_adk_example():
+    """Demonstrates the GoogleADKAgent's arun method."""
+    agent_name = "GoogleADKCalculator"
+
+    # Create the agent with simplified instructions and our tool
+    agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="""You are a calculator assistant. When asked math problems,
+        extract numbers and call sum_numbers tool to add them.
+        For multi-step problems, use multiple tool calls.""",
+        model="gemini-2.0-flash",
+        tools=[sum_numbers],
+        max_iterations=5,  # Allow multiple tool calls if needed
+    )
+
+    # Use the same query as in LangGraph example for consistency
+    query = "What is the sum of 23 and 47? And then add 10 to that, then add 5 more."
+    print(f"--- Agent: {agent_name} ---")
+    print(f"Query: {query}")
+
+    print("\nRunning arun...")
+    response = await agent.arun(query=query)
+    print(f"[arun] Final Response: {response.get('output')}")
+    print("--- End of Google ADK Example ---")
+
+
+if __name__ == "__main__":
+    # GOOGLE_API_KEY should be set in the environment.
+    asyncio.run(google_adk_example())

aip_agents/examples/hello_world_google_adk.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.tools.adk_arithmetic_tools import sum_numbers as sum_numbers
+
+async def google_adk_example() -> None:
+    """Demonstrates the GoogleADKAgent's arun method."""

aip_agents/examples/hello_world_google_adk_mcp_http.py

@@ -0,0 +1,34 @@
+"""Example showing Google ADK agent with MCP tools integration using Streamable HTTP transport.
+
+This example demonstrates how to create a Google ADK agent that can use tools
+from MCP servers via Streamable HTTP transport.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_http
+
+
+async def main():
+    """Demonstrates the GoogleADKAgent with MCP tools via Streamable HTTP transport."""
+    agent = GoogleADKAgent(
+        name="GoogleADKMCPHTTP",
+        instruction="""You are a helpful assistant that can use playwright tools to browse the web.
+        If a user ask something, you try to answer it by browsing the web via playwright tools
+        Do not ask for clarification, just answer to the best of your ability using playwright tools.""",
+        model="gemini-2.0-flash",
+        max_iterations=5,
+    )
+    agent.add_mcp_server(mcp_config_http)
+
+    response = await agent.arun(query="Who is the Winner of La Liga in 2024/2025 Season? Browse it.")
+    print(f"Response: {response.get('output')}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_google_adk_mcp_http.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_http as mcp_config_http
+
+async def main() -> None:
+    """Demonstrates the GoogleADKAgent with MCP tools via Streamable HTTP transport."""

aip_agents/examples/hello_world_google_adk_mcp_http_stream.py

@@ -0,0 +1,40 @@
+"""Example showing Google ADK agent with MCP tools integration.
+
+Transport: Streamable HTTP
+Streaming: Yes
+
+This example demonstrates how to create a Google ADK agent that can use tools
+from MCP servers via Streamable HTTP transport while streaming the response
+in real-time.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_http
+
+
+async def main():
+    """Demonstrates the GoogleADKAgent with MCP tools via Streamable HTTP transport and streaming."""
+    agent = GoogleADKAgent(
+        name="GoogleADKMCPHTTPStream",
+        instruction="""You are a helpful assistant that can use playwright tools to browse the web.
+        If a user ask something, you try to answer it by browsing the web via playwright tools
+        Do not ask for clarification, just answer to the best of your ability using playwright tools.""",
+        model="gemini-2.0-flash",
+        max_iterations=5,
+    )
+    agent.add_mcp_server(mcp_config_http)
+    query = (
+        "Go to huggingface.co, and then browse to the number 1 trending model page to find the abstract of the model"
+    )
+    async for chunk in agent.arun_stream(query=query):
+        print(chunk, end="", flush=True)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_google_adk_mcp_http_stream.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_http as mcp_config_http
+
+async def main() -> None:
+    """Demonstrates the GoogleADKAgent with MCP tools via Streamable HTTP transport and streaming."""

aip_agents/examples/hello_world_google_adk_mcp_sse.py

@@ -0,0 +1,44 @@
+"""Example showing Google ADK agent with MCP tools integration using SSE transport.
+
+This example demonstrates how to create a Google ADK agent that can use tools
+from MCP servers via Server-Sent Events (SSE) transport.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_sse
+
+
+async def main():
+    """Demonstrates the GoogleADKAgent with MCP tools via SSE transport."""
+    agent_name = "GoogleADKMCPSSE"
+
+    agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="""You are a helpful assistant that can use playwright tools to browse the web.
+        If a user ask something, you try to answer it by browsing the web via playwright tools
+        Do not ask for clarification, just answer to the best of your ability using playwright tools.""",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+    agent.add_mcp_server(mcp_config_sse)
+
+    query = "How many trophies did FC Barcelona win in 2024/2025 season?"
+
+    print(f"--- Agent: {agent_name} ---")
+    print(f"Query: {query}")
+
+    print("\nRunning arun with MCP SSE tools...")
+    response = await agent.arun(query=query)
+    print(f"[arun] Final Response: {response.get('output')}")
+    print("--- End of Google ADK MCP SSE Example ---")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_google_adk_mcp_sse.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_sse as mcp_config_sse
+
+async def main() -> None:
+    """Demonstrates the GoogleADKAgent with MCP tools via SSE transport."""

aip_agents/examples/hello_world_google_adk_mcp_sse_stream.py

@@ -0,0 +1,48 @@
+"""Example showing Google ADK agent with MCP tools integration using SSE transport and streaming capabilities.
+
+This example demonstrates how to create a Google ADK agent that can use tools
+from MCP servers via Server-Sent Events (SSE) transport while streaming the response
+in real-time.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_sse
+
+
+async def main():
+    """Demonstrates the GoogleADKAgent with MCP tools via SSE transport and streaming."""
+    agent_name = "GoogleADKMCPSSEStream"
+
+    agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="""You are a helpful assistant that can use playwright tools to browse the web.
+        If a user ask something, you try to answer it by browsing the web via playwright tools
+        Do not ask for clarification, just answer to the best of your ability using playwright tools.""",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+    agent.add_mcp_server(mcp_config_sse)
+
+    query = "How many trophies did FC Barcelona win in 2024/2025 season?"
+
+    print(f"--- Agent: {agent_name} ---")
+    print(f"Query: {query}")
+
+    print("\nRunning arun_stream with MCP SSE tools...")
+    print("Streaming response:")
+
+    async for chunk in agent.arun_stream(query=query):
+        print(chunk, end="", flush=True)
+
+    print("\n--- End of Google ADK MCP SSE Streaming Example ---")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_google_adk_mcp_sse_stream.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_sse as mcp_config_sse
+
+async def main() -> None:
+    """Demonstrates the GoogleADKAgent with MCP tools via SSE transport and streaming."""

aip_agents/examples/hello_world_google_adk_mcp_stdio.py

@@ -0,0 +1,44 @@
+"""Example showing Google ADK agent with MCP tools integration using stdio transport.
+
+This example demonstrates how to create a Google ADK agent that can use tools
+from MCP servers via stdio (standard input/output) transport, which runs the
+MCP server as a subprocess.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_stdio
+
+
+async def main():
+    """Demonstrates the GoogleADKAgent with MCP tools via stdio transport."""
+    agent_name = "GoogleADKMCPStdio"
+
+    agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+    agent.add_mcp_server(mcp_config_stdio)
+
+    query = "What's the weather forecast for monday?"  # Uses MCP weather tool
+
+    print(f"--- Agent: {agent_name} ---")
+    print(f"Query: {query}")
+
+    print("\nRunning arun with MCP stdio tools...")
+    response = await agent.arun(query=query)
+    print(f"[arun] Final Response: {response.get('output')}")
+    print("--- End of Google ADK MCP Stdio Example ---")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_google_adk_mcp_stdio.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_stdio as mcp_config_stdio
+
+async def main() -> None:
+    """Demonstrates the GoogleADKAgent with MCP tools via stdio transport."""

aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.py

@@ -0,0 +1,48 @@
+"""Example showing Google ADK agent with MCP tools integration using stdio transport and streaming capabilities.
+
+This example demonstrates how to create a Google ADK agent that can use tools
+from MCP servers via stdio (standard input/output) transport while streaming the response
+in real-time. The MCP server runs as a subprocess.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_stdio
+
+
+async def main():
+    """Demonstrates the GoogleADKAgent with MCP tools via stdio transport and streaming."""
+    agent_name = "GoogleADKMCPStdioStream"
+
+    agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').
+        Explain your steps clearly for streaming demonstration.""",
+        model="gemini-2.0-flash",
+        tools=[],
+        max_iterations=5,
+    )
+    agent.add_mcp_server(mcp_config_stdio)
+
+    query = "What's the weather forecast for monday?"  # Uses MCP weather tool
+
+    print(f"--- Agent: {agent_name} ---")
+    print(f"Query: {query}")
+
+    print("\nRunning arun_stream with MCP stdio tools...")
+    print("Streaming response:")
+
+    async for chunk in agent.arun_stream(query=query):
+        print(chunk, end="", flush=True)
+
+    print("\n--- End of Google ADK MCP Stdio Streaming Example ---")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_google_adk_mcp_stdio_stream.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_stdio as mcp_config_stdio
+
+async def main() -> None:
+    """Demonstrates the GoogleADKAgent with MCP tools via stdio transport and streaming."""

aip_agents/examples/hello_world_google_adk_stream.py

@@ -0,0 +1,44 @@
+"""Minimal example demonstrating the GoogleADKAgent's streaming capabilities with async execution.
+
+This example shows how to use the arun_stream method to process responses in real-time
+with Google's Agent Development Kit (ADK).
+"""
+
+import asyncio
+
+from aip_agents.agent.google_adk_agent import GoogleADKAgent
+from aip_agents.examples.tools.adk_arithmetic_tools import add_numbers
+
+
+async def google_adk_example_stream():
+    """Demonstrates the GoogleADKAgent's arun_stream method."""
+    agent_name = "GoogleADKStreamingCalculator"
+
+    # Create the agent with simplified instructions for streaming
+    agent = GoogleADKAgent(
+        name=agent_name,
+        instruction="""You are a calculator assistant. When asked math problems,
+        extract numbers and call add_numbers tool to add them.
+        Explain your steps clearly for streaming demonstration.""",
+        model="gemini-2.0-flash",
+        tools=[add_numbers],
+        max_iterations=5,  # Allow multiple tool calls if needed
+    )
+
+    # Use the same query as in LangGraph example for consistency
+    query = "What is the sum of 23 and 47? And then add 10 to that, then add 5 more."
+    print(f"--- Agent: {agent_name} ---")
+    print(f"Query: {query}")
+
+    print("\nRunning arun_stream...")
+    print("Streaming response:")
+
+    # Stream the response chunks
+    async for chunk in agent.arun_stream(query=query):
+        print(chunk, end="", flush=True)
+
+    print("\n--- End of Google ADK Streaming Example ---")
+
+
+if __name__ == "__main__":
+    asyncio.run(google_adk_example_stream())

aip_agents/examples/hello_world_google_adk_stream.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
+from aip_agents.examples.tools.adk_arithmetic_tools import add_numbers as add_numbers
+
+async def google_adk_example_stream() -> None:
+    """Demonstrates the GoogleADKAgent's arun_stream method."""

aip_agents/examples/hello_world_langchain.py

@@ -0,0 +1,28 @@
+"""Minimal LangChain agent example demonstrating asynchronous run."""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers
+
+
+async def langchain_example():
+    """Demonstrates the LangChainAgent's arun method."""
+    langchain_agent = LangChainAgent(
+        name="LangChainArithmeticAgent",
+        instruction="You are a helpful assistant that can add two numbers using the add_numbers tool.",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+        tools=[add_numbers],
+    )
+
+    response = await langchain_agent.arun(
+        query="What is the sum of 23 and 47? And then add 10 to that, then add 5 more."
+    )
+    print(response["output"])
+
+
+if __name__ == "__main__":
+    # OPENAI_API_KEY should be set in the environment.
+    asyncio.run(langchain_example())

aip_agents/examples/hello_world_langchain.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
+
+async def langchain_example() -> None:
+    """Demonstrates the LangChainAgent's arun method."""

aip_agents/examples/hello_world_langchain_lm_invoker.py

@@ -0,0 +1,15 @@
+"""Minimal LangChain agent example demonstrating asynchronous run."""
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.examples.tools.langchain_weather_tool import weather_tool
+
+if __name__ == "__main__":
+    langchain_agent = LangChainAgent(
+        name="LangChainWeatherAgent",
+        instruction="You are a helpful assistant that can get the weather in a given city using the get_weather tool.",
+        model="openai/gpt-4o",
+        tools=[weather_tool],
+    )
+
+    response = langchain_agent.run(query="What is the weather in Tokyo?")
+    print(response["output"])

aip_agents/examples/hello_world_langchain_lm_invoker.pyi

@@ -0,0 +1,2 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool

aip_agents/examples/hello_world_langchain_mcp_http.py

@@ -0,0 +1,34 @@
+"""Example showing LangChain agent with MCP tools integration using Streamable HTTP transport.
+
+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.examples.mcp_configs.configs import mcp_config_http
+
+
+async def main():
+    """Demonstrates the LangChainAgent with MCP tools via Streamable HTTP transport."""
+    langchain_agent = LangChainAgent(
+        name="langchain_mcp_example",
+        instruction=(
+            "You are a helpful assistant that can browse the web. If you are blocked by captcha, try another website."
+        ),
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+    langchain_agent.add_mcp_server(mcp_config_http)
+
+    response = await langchain_agent.arun(query="What's the best restaurant in San Francisco, search it")
+    print(response["output"])
+    print("-" * 50)
+    response = await langchain_agent.arun(query="What's the best restaurant in New York, search it")
+    print(response["output"])
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langchain_mcp_http.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_http as mcp_config_http
+
+async def main() -> None:
+    """Demonstrates the LangChainAgent with MCP tools via Streamable HTTP transport."""