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

aip_agents/examples/mcp_configs/configs.py

@@ -0,0 +1,63 @@
+"""Configuration for MCP servers."""
+
+mcp_config_sse = {
+    "playwright_tools": {
+        "url": "http://localhost:8931/sse",
+        "transport": "sse",
+    }
+}
+
+
+mcp_config_stdio = {
+    "weather_tools": {
+        "command": "python",
+        "args": ["aip_agents/examples/mcp_servers/mcp_server_stdio.py"],
+        "transport": "stdio",
+    }
+}
+
+mcp_config_http = {
+    "playwright_tools": {
+        "url": "http://localhost:8931/mcp",
+        "transport": "http",
+    }
+}
+
+
+mcp_multi_server = {
+    "sse_secretmsg": {
+        "url": "http://localhost:8123/sse",
+        "transport": "sse",
+    },
+    "stdio_time": {
+        "command": "python",
+        "args": ["aip_agents/examples/mcp_servers/mcp_time.py"],
+        "transport": "stdio",
+    },
+    "http_playwright_tools": {
+        "url": "http://localhost:8931/mcp",
+        "transport": "http",
+    },
+}
+
+mcp_tool_whitelisting_demo = {
+    # STDIO server - allow only get_current_time (whitelist approach)
+    "stdio_time": {
+        "command": "python",
+        "args": ["aip_agents/examples/mcp_servers/mcp_server_stdio.py"],
+        "transport": "stdio",
+        "allowed_tools": ["get_current_time"],
+    },
+    # SSE server - allow only get_random_quote (whitelist approach)
+    "sse_quotes": {
+        "url": "http://localhost:8123/sse",
+        "transport": "sse",
+        "allowed_tools": ["get_random_quote"],
+    },
+    # HTTP server - allow only get_random_fact (whitelist approach)
+    "http_facts": {
+        "url": "http://localhost:8931/mcp",
+        "transport": "streamable-http",
+        "allowed_tools": ["get_random_fact"],
+    },
+}

aip_agents/examples/mcp_servers/common.py

@@ -0,0 +1,76 @@
+"""Common reusable tool functions for MCP server examples."""
+
+import base64
+import random
+import uuid
+from datetime import datetime
+
+# ===== Random Selection Tools =====
+
+
+def get_random_item_from_list(items: list[str]) -> str:
+    """Return a random item from the given list."""
+    return random.choice(items)
+
+
+def get_random_fact() -> str:
+    """Return a random fun fact."""
+    facts = [
+        "Honey never spoils. Archaeologists have found pots of honey in ancient Egyptian tombs that are over 3,000 years old and still perfectly edible!",
+        "A group of flamingos is called a 'flamboyance'.",
+        "The shortest war in history was between Britain and Zanzibar in 1896. Zanzibar surrendered after 38 minutes.",
+        "There are more stars in the universe than grains of sand on all of Earth's beaches.",
+        "A banana is technically a berry, but a strawberry is not.",
+    ]
+    return get_random_item_from_list(facts)
+
+
+def get_random_quote() -> str:
+    """Return a random inspirational quote."""
+    quotes = [
+        "The only way to do great work is to love what you do. - Steve Jobs",
+        "Innovation distinguishes between a leader and a follower. - Steve Jobs",
+        "Life is what happens when you're busy making other plans. - John Lennon",
+        "The future belongs to those who believe in the beauty of their dreams. - Eleanor Roosevelt",
+        "It is during our darkest moments that we must focus to see the light. - Aristotle",
+    ]
+    return get_random_item_from_list(quotes)
+
+
+def random_name_generator(name_list: list[str]) -> str:
+    """Generate random name from the provided list."""
+    return get_random_item_from_list(name_list)
+
+
+# ===== Time & UUID Tools =====
+
+
+def get_current_time() -> str:
+    """Return the current time with format YYYY-MM-DD HH:MM:SS."""
+    return f"Current time: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+
+
+def generate_uuid() -> str:
+    """Generate a random UUID."""
+    return f"Generated UUID: {uuid.uuid4()}"
+
+
+# ===== Text Utilities =====
+
+
+def convert_to_base64(text: str) -> str:
+    """Convert text to base64 representation."""
+    try:
+        encoded = base64.b64encode(text.encode()).decode()
+        return f"Base64: {encoded}"
+    except Exception as e:
+        return f"Error converting to base64: {str(e)}"
+
+
+def word_count(text: str) -> str:
+    """Count words in the given text."""
+    if not text.strip():
+        return "Word count: 0 (empty text)"
+
+    words = text.split()
+    return f"Word count: {len(words)} words"

aip_agents/examples/mcp_servers/mcp_name.py

@@ -0,0 +1,29 @@
+"""Random Name Generator MCP Tool by SSE."""
+
+from mcp.server.fastmcp import FastMCP
+
+from aip_agents.examples.mcp_servers.common import get_random_item_from_list
+
+name_list = [
+    "Pedri",
+    "Joan Garcia",
+    "De Jong",
+    "Curbasi",
+    "Yamal",
+    "Kounde",
+    "Balde",
+    "Raphina",
+    "Fermin Lopez",
+    "Gavi",
+]
+
+
+def random_name_generator() -> str:
+    """Generate random name."""
+    return get_random_item_from_list(name_list)
+
+
+if __name__ == "__main__":
+    mcp = FastMCP("Random Name Generator", port=8123)
+    mcp.add_tool(random_name_generator, name="random_name_generator")
+    mcp.run(transport="sse")

aip_agents/examples/mcp_servers/mcp_server_http.py

@@ -0,0 +1,19 @@
+"""Weather Forecast & Trivia Tools MCP by Streamable HTTP with multiple tools."""
+
+from mcp.server.fastmcp import FastMCP
+
+from aip_agents.examples.mcp_servers.common import convert_to_base64, get_random_fact
+from aip_agents.examples.tools.weather_forecast_tool import get_weather_forecast
+
+if __name__ == "__main__":
+    mcp = FastMCP("Weather-Trivia HTTP Server", port=8931)
+    mcp.add_tool(get_weather_forecast, name="get_weather_forecast")
+    mcp.add_tool(get_random_fact, name="get_random_fact")
+    mcp.add_tool(convert_to_base64, name="convert_to_base64")
+
+    print("Starting HTTP MCP Server with 3 tools:")
+    print("- get_weather_forecast")
+    print("- get_random_fact")
+    print("- convert_to_base64")
+
+    mcp.run(transport="streamable-http")

aip_agents/examples/mcp_servers/mcp_server_sse.py

@@ -0,0 +1,19 @@
+"""Weather Forecast & Text Tools MCP by SSE with multiple tools."""
+
+from mcp.server.fastmcp import FastMCP
+
+from aip_agents.examples.mcp_servers.common import get_random_quote, word_count
+from aip_agents.examples.tools.weather_forecast_tool import get_weather_forecast
+
+if __name__ == "__main__":
+    mcp = FastMCP("Weather-Text SSE Server", port=8123)
+    mcp.add_tool(get_weather_forecast, name="get_weather_forecast")
+    mcp.add_tool(get_random_quote, name="get_random_quote")
+    mcp.add_tool(word_count, name="word_count")
+
+    print("Starting SSE MCP Server with 3 tools:")
+    print("- get_weather_forecast")
+    print("- get_random_quote")
+    print("- word_count")
+
+    mcp.run(transport="sse")

aip_agents/examples/mcp_servers/mcp_server_stdio.py

@@ -0,0 +1,19 @@
+"""Weather Forecast & Time Tools MCP by STDIO with multiple tools."""
+
+from mcp.server.fastmcp import FastMCP
+
+from aip_agents.examples.mcp_servers.common import generate_uuid, get_current_time
+from aip_agents.examples.tools.weather_forecast_tool import get_weather_forecast
+
+if __name__ == "__main__":
+    mcp = FastMCP("Weather-Time STDIO Server")
+    mcp.add_tool(get_weather_forecast, name="get_weather_forecast")
+    mcp.add_tool(get_current_time, name="get_current_time")
+    mcp.add_tool(generate_uuid, name="generate_uuid")
+
+    print("Starting STDIO MCP Server with 3 tools:")
+    print("- get_weather_forecast")
+    print("- get_current_time")
+    print("- generate_uuid")
+
+    mcp.run(transport="stdio")

aip_agents/examples/mcp_servers/mcp_time.py

@@ -0,0 +1,10 @@
+"""Time MCP Tool STDIO."""
+
+from mcp.server.fastmcp import FastMCP
+
+from aip_agents.examples.mcp_servers.common import get_current_time as time_tool
+
+if __name__ == "__main__":
+    mcp = FastMCP("Time")
+    mcp.add_tool(time_tool, name="time")
+    mcp.run(transport="stdio")

aip_agents/examples/pii_demo_langgraph_client.py

@@ -0,0 +1,69 @@
+"""Streaming PII demo client - inspect streaming chunks in real-time.
+
+To run:
+    1. Start server:
+       export NER_API_URL=http://localhost:8080
+       export NER_API_KEY=your-api-key
+       python examples/pii_demo_langgraph_server.py
+
+    2. Run client:
+       python examples/pii_demo_langgraph_client.py
+
+Authors:
+    Fachriza Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+"""
+
+import asyncio
+from pprint import pprint
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+
+async def main():
+    """Stream queries and inspect chunks with PII handling."""
+    # Setup
+    llm = ChatOpenAI(model="gpt-4.1", streaming=True)
+    client_agent = LangChainAgent(
+        name="PIIDemoClient",
+        instruction="""You are a helpful assistant. When asked for customer, employee, or user info, delegate to the PII demo agent.
+        You will receive output from tool in masked version, just write the response naturally without revealing or changing the masked values.""",
+        model=llm,
+    )
+
+    # Discover agents
+    print("🔍 Discovering agents...\n")
+    config = A2AClientConfig(discovery_urls=["http://localhost:8002"])
+    agent_cards = client_agent.discover_agents(config)
+    client_agent.register_a2a_agents(agent_cards)
+    print(f"✅ Found: {agent_cards[0].name}\n")
+
+    # Example queries
+    queries = [
+        "Can you get me information about customers <PERSON_1> and <PERSON_2>?",
+    ]
+
+    # Stream each query
+    for i, query in enumerate(queries, 1):
+        print(f"\n{'=' * 80}")
+        print(f"Query {i}: {query}")
+        print("=" * 80)
+        print("\n📡 Streaming chunks:\n")
+
+        try:
+            async for chunk in client_agent.astream_to_agent(
+                agent_card=agent_cards[0],
+                message=query,
+                pii_mapping={"<PERSON_1>": "C001", "<PERSON_2>": "C002"},
+            ):
+                pprint(chunk)
+                print("-" * 50)
+
+        except Exception as e:
+            print(f"\n❌ Error: {e}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/pii_demo_langgraph_server.py

@@ -0,0 +1,126 @@
+"""Example A2A server demonstrating PII handling in LangGraph agents.
+
+This server instantiates a LangGraph ReAct agent with tools that return personally
+identifiable information (PII). The agent demonstrates automatic PII anonymization
+and deanonymization during parallel tool execution.
+
+Features demonstrated:
+- Automatic PII detection and anonymization in tool outputs
+- Deanonymization of PII for tool input arguments
+- Collision detection and resolution in parallel tool execution
+- Hybrid tag strategy (sequential tags preserved, UUID tags for collisions)
+
+To run this server:
+    export NER_API_URL=http://localhost:8080
+    export NER_API_KEY=your-api-key
+    python examples/pii_demo_langgraph_server.py
+
+It will listen on http://localhost:8002 by default.
+
+Authors:
+    Fachriza Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+"""
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+from dotenv import load_dotenv
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.examples.tools.pii_demo_tools import (
+    get_customer_info,
+    get_employee_data,
+    get_user_profile,
+)
+from aip_agents.utils.logger import LoggerManager
+
+load_dotenv()
+
+logger = LoggerManager().get_logger(__name__)
+
+
+SERVER_AGENT_NAME = "PIIDemoAgentLangGraph"
+
+
+@click.command()
+@click.option("--host", "host", default="localhost", help="Host to bind the server to.")
+@click.option("--port", "port", default=8002, help="Port to bind the server to.")
+def main(host: str, port: int):
+    """Runs the PII Demo LangGraph A2A server.
+
+    This server demonstrates PII handling capabilities including:
+    - Automatic anonymization of sensitive data in tool outputs
+    - Deanonymization of PII for tool inputs
+    - Collision detection and resolution in parallel tool execution
+
+    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}")
+    logger.info("PII handling is enabled. Sensitive data will be automatically anonymized.")
+
+    agent_card = AgentCard(
+        name=SERVER_AGENT_NAME,
+        description="A LangGraph ReAct agent demonstrating PII handling with automatic anonymization and deanonymization",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="customer_info",
+                name="Get Customer Information",
+                description="Retrieves customer information including email and phone number",
+                examples=["Get customer info for C001", "What is the email for customer C002?"],
+                tags=["customer", "pii"],
+            ),
+            AgentSkill(
+                id="employee_data",
+                name="Get Employee Data",
+                description="Retrieves employee data including name, email, and salary",
+                examples=["Get employee data for E001", "What is the salary for employee E002?"],
+                tags=["employee", "pii"],
+            ),
+            AgentSkill(
+                id="user_profile",
+                name="Get User Profile",
+                description="Retrieves user profile information with personal details",
+                examples=["Get user profile for U001", "What is the email for user U002?"],
+                tags=["user", "pii"],
+            ),
+        ],
+        tags=["pii", "demo", "langgraph"],
+    )
+
+    langgraph_agent = LangChainAgent(
+        name=SERVER_AGENT_NAME,
+        instruction="""You are a helpful assistant that can retrieve customer, employee, and user information.
+
+You have access to three tools:
+1. get_customer_info: Retrieves customer information (email, phone, address)
+2. get_employee_data: Retrieves employee data (email, phone, salary, department)
+3. get_user_profile: Retrieves user profile information (email, phone, date of birth, city)
+
+When users ask for information, use the appropriate tool to retrieve the data.
+The system will automatically handle PII anonymization and deanonymization.
+
+You will receive output from tool in masked version,
+just write the response as it is without revealing or changing the masked values.
+
+Always be helpful and provide the requested information clearly.""",
+        model="openai/gpt-4.1",
+        tools=[get_customer_info, get_employee_data, get_user_profile],
+    )
+
+    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/pii_demo_multi_agent_client.py

@@ -0,0 +1,80 @@
+"""Multi-Agent PII Demo Client - Tests PII handling across agent hierarchy.
+
+This client demonstrates connecting to and using a hierarchical agent system
+that manages PII across multiple specialist agents through A2A protocol.
+
+To run this client:
+1. First start the server:
+   export NER_API_URL=http://localhost:8080
+   export NER_API_KEY=your-api-key
+   python aip_agents/examples/pii_demo_multi_agent_server.py
+
+2. Then run this client:
+   python aip_agents/examples/pii_demo_multi_agent_client.py
+
+Authors:
+    Fachriza Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+"""
+
+import asyncio
+from pprint import pprint
+
+from dotenv import load_dotenv
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+load_dotenv()
+
+
+async def main():
+    """Main function demonstrating the Multi-Agent PII Demo client."""
+    llm = ChatOpenAI(model="gpt-4.1", streaming=True)
+
+    # Create a simple client agent
+    client_agent = LangGraphAgent(
+        name="PIIMultiAgentTestClient",
+        instruction=(
+            "You are a test client that communicates with hierarchical agent systems via A2A. "
+            "You will receive responses with PII tags - present them as-is."
+        ),
+        model=llm,
+        tools=[],
+    )
+
+    # Discover the coordinator agent
+    print("🔍 Discovering PIICoordinator agent...\n")
+    config = A2AClientConfig(discovery_urls=["http://localhost:8003"])
+    agent_cards = client_agent.discover_agents(config)
+    client_agent.register_a2a_agents(agent_cards)
+    print(f"✅ Found: {agent_cards[0].name}\n")
+
+    # Example queries demonstrating multi-agent PII handling
+    queries = [
+        "I need information about customer <PERSON_1> and employee E001",
+        "Find me information about customer <PERSON_1>, and then find information about employee E001, don't run tool in paralel",
+    ]
+
+    # Stream each query
+    for i, query in enumerate(queries, 1):
+        print(f"\n{'=' * 80}")
+        print(f"Query {i}: {query}")
+        print("=" * 80)
+        print("\n📡 Streaming chunks:\n")
+
+        try:
+            async for chunk in client_agent.astream_to_agent(
+                agent_card=agent_cards[0],
+                message=query,
+                pii_mapping={"<EMAIL_ADDRESS_1>": "john.smith@example.com", "<PERSON_1>": "C001"},
+            ):
+                pprint(chunk)
+                print("-" * 50)
+
+        except Exception as e:
+            print(f"\n❌ Error: {e}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())