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

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_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_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_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_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_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_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_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_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_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_interactive.py

@@ -0,0 +1,130 @@
+"""Interactive example showing LangChain agent with MCP tools integration using Streamable HTTP transport.
+
+This script provides an interactive CLI interface with chat history functionality,
+similar to hello_world_model_switch_cli.py, but with MCP tools via HTTP transport.
+
+Authors:
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+import os
+
+from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.utils.env_loader import load_local_env
+
+load_local_env()
+
+
+def print_help():
+    """Prints available commands and their descriptions."""
+    help_text = (
+        "\nAvailable commands:\n"
+        "  /help    Show this help message\n"
+        "  /exit    Quit the chat\n"
+        "Type anything else to chat with the assistant.\n"
+    )
+    print(help_text)
+
+
+def build_mcp_config_from_env() -> dict:
+    """Construct MCP configuration from environment variables.
+
+    Returns:
+        dict: MCP configuration dictionary.
+
+    Raises:
+        RuntimeError: If required environment variables are missing.
+    """
+    url = "https://huggingface.co/mcp"
+    token = os.getenv("TEST_HF_TOKEN")
+
+    missing = [
+        name
+        for name, value in {
+            "TEST_HF_TOKEN": token,
+        }.items()
+        if not value
+    ]
+
+    if missing:
+        missing_vars = ", ".join(missing)
+        raise RuntimeError(
+            "Missing required environment variables for MCP configuration: "
+            f"{missing_vars}. Please set them before running the example."
+        )
+
+    return {
+        "mcp_server": {
+            "url": url,
+            "transport": "http",
+            "headers": {
+                "Authorization": f"Bearer {token}",
+            },
+        }
+    }
+
+
+async def main():
+    """Runs the interactive LangChainAgent with MCP tools via Streamable HTTP transport."""
+    print("Welcome to the Interactive LangChain MCP HTTP Example!")
+    print("🎯 Demonstrating MCP tools integration with HTTP transport:")
+    print("  • Web browsing capabilities via Playwright tools")
+    print()
+    print("Type your message, or type /help to show available commands, or /exit to quit.")
+
+    # Initialize the agent with MCP tools
+    langchain_agent = LangChainAgent(
+        name="interactive_langchain_mcp_example",
+        instruction=("You are a helpful assistant."),
+        # Using a basic model for the example - you can adjust this as needed
+        model="openai/gpt-4.1",  # Changed to a string identifier for consistency
+    )
+    mcp_config = build_mcp_config_from_env()
+    langchain_agent.add_mcp_server(mcp_config)
+
+    # Conversation history as LangChain message objects
+    messages = [SystemMessage(content="You are a helpful assistant.")]
+
+    while True:
+        user_input = input("You: ").strip()
+        if user_input.lower() == "/help":
+            print_help()
+            continue
+        if user_input.lower() == "/exit":
+            print("Goodbye!")
+            break
+        elif user_input == "":
+            continue
+
+        # Add user message to history
+        messages.append(HumanMessage(content=user_input))
+
+        # Run the agent with history
+        try:
+            # Pass the conversation history to the agent
+            response = await langchain_agent.arun(query=user_input, messages=messages[:-1])
+
+            # Extract AI message (should be last in response state)
+            output = response.get("output")
+            ai_message = None
+
+            # Try to find the returned messages in the response (standard pattern)
+            if "messages" in response and response["messages"]:
+                # Use the last message as the AI reply
+                ai_message = response["messages"][-1]
+            elif output:
+                ai_message = AIMessage(content=output)
+            else:
+                ai_message = AIMessage(content=str(response))
+
+            messages.append(ai_message)
+            print(f"Agent: {ai_message.content}")
+        except Exception as e:
+            print(f"Error: {e}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langchain_mcp_http_stream.py

@@ -0,0 +1,42 @@
+"""Example showing LangChain agent with MCP tools integration.
+
+Transport: Streamable HTTP
+Streaming: Yes
+
+This example demonstrates how to create a LangChain agent that can use tools
+from MCP servers via Streamable HTTP transport while streaming the response
+in real-time.
+
+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 and streaming capabilities."""
+    langchain_agent = LangChainAgent(
+        name="langchain_mcp_stream_example",
+        instruction="""You are a helpful assistant that can browse the web.""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+    langchain_agent.add_mcp_server(mcp_config_http)
+
+    async for chunk in langchain_agent.arun_stream(query="What's the latest news about FC Barcelona?"):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)
+
+    # run again to make sure it use persistent session
+    async for chunk in langchain_agent.arun_stream(query="What's the latest news about UEFA Champions League?"):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())