aip-agents-binary 0.0.0b2__py3-none-any.whl → 0.5.11__py3-none-any.whl

aip_agents/examples/hello_world_langchain_mcp_stdio.py

@@ -0,0 +1,30 @@
+"""Example showing LangChain agent with MCP tools integration using stdio 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_stdio
+
+
+async def main():
+    """Demonstrates the LangChainAgent with MCP tools via stdio transport."""
+    langchain_agent = LangChainAgent(
+        name="langchain_mcp_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+    langchain_agent.add_mcp_server(mcp_config_stdio)
+
+    response = await langchain_agent.arun(query="What's the weather forecast for monday?")
+    print(response["output"])
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langchain_mcp_stdio_stream.py

@@ -0,0 +1,41 @@
+"""Example showing LangChain agent with MCP tools integration and streaming capabilities using stdio 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_stdio
+
+
+async def main():
+    """Demonstrates the LangChainAgent with MCP tools via stdio transport and streaming capabilities."""
+    langchain_agent = LangChainAgent(
+        name="langchain_mcp_stream_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+    langchain_agent.add_mcp_server(mcp_config_stdio)
+
+    full_response = ""
+    async for chunk in langchain_agent.arun_stream(
+        query="What's the weather forecast for monday?",
+    ):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)
+            full_response += chunk
+        elif isinstance(chunk, dict) and "messages" in chunk:
+            print("\n(Stream finished with final state object)")
+        elif isinstance(chunk, dict):
+            pass
+
+    print(full_response)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langchain_stream.py

@@ -0,0 +1,36 @@
+"""Minimal LangChain agent example demonstrating streaming capabilities.
+
+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.tools.langchain_arithmetic_tools import add_numbers
+
+
+async def langchain_stream_example():
+    """Demonstrates the LangChainAgent's arun_stream method with async execution."""
+    agent = LangChainAgent(
+        name="LangChainStreamingCalculator",
+        instruction="You are a helpful calculator assistant that can add numbers. "
+        "When asked to add numbers, use the add_numbers tool. "
+        "Explain your steps clearly for streaming demonstration.",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+        tools=[add_numbers],
+    )
+
+    # Stream the response chunks
+    async for chunk in agent.arun_stream(
+        query="What is the sum of 23 and 47? And then add 10 to that, then add 5 more."
+    ):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)
+
+
+if __name__ == "__main__":
+    asyncio.run(langchain_stream_example())

aip_agents/examples/hello_world_langchain_stream_lm_invoker.py

@@ -0,0 +1,39 @@
+"""Minimal LangChain agent example demonstrating streaming capabilities.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+
+import dotenv
+
+from aip_agents.agent import LangChainAgent
+from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers
+
+dotenv.load_dotenv(override=True)
+
+
+async def langchain_stream_example():
+    """Demonstrates the LangChainAgent's arun_stream method with async execution."""
+    agent = LangChainAgent(
+        name="LangChainStreamingCalculator",
+        instruction="You are a helpful calculator assistant that can add numbers. "
+        "When asked to add numbers, use the add_numbers tool. "
+        "Explain your steps clearly for streaming demonstration.",
+        model="openai/gpt-4o",
+        tools=[add_numbers],
+    )
+
+    # Stream the response chunks
+    async for chunk in agent.arun_stream(
+        query="What is the sum of 23 and 47? And then add 10 to that, then add 5 more."
+    ):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)  # Print text chunks as they come in
+        elif isinstance(chunk, dict):
+            print(f"Dict chunk: {chunk}")
+
+
+if __name__ == "__main__":
+    asyncio.run(langchain_stream_example())

aip_agents/examples/hello_world_langflow_agent.py

@@ -0,0 +1,163 @@
+"""Example demonstrating basic Langflow agent usage.
+
+This example shows how to create and use a LangflowAgent for both
+synchronous and asynchronous execution modes.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+import os
+import uuid
+
+from dotenv import load_dotenv
+
+from aip_agents.agent.langflow_agent import LangflowAgent
+from aip_agents.clients.langflow import LangflowApiClient
+
+load_dotenv()
+
+
+async def fetch_flow_id() -> tuple[str, str]:
+    """Fetch available flows and return the first flow ID and name."""
+    print("Fetching available flows...")
+    temp_client = LangflowApiClient(base_url="https://langflow.obrol.id", api_key=os.getenv("LANGFLOW_API_KEY"))
+
+    try:
+        flows = await temp_client.get_all_flows(remove_example_flows=True)
+        print(flows)
+
+        if not flows:
+            print("❌ No flows found!")
+            return "", ""
+
+        print(f"Found {len(flows)} flows:")
+        for i, flow in enumerate(flows[:5]):  # Show first 5 flows
+            flow_name = flow.get("name", "Unnamed")
+            flow_id = flow.get("id", "No ID")
+            print(f"  {i + 1}. {flow_name} (ID: {flow_id})")
+
+        selected_flow = flows[2]
+        flow_id = selected_flow.get("id")
+        flow_name = selected_flow.get("name", "Unknown Flow")
+
+        if not flow_id:
+            print("❌ Selected flow has no ID!")
+            return "", ""
+
+        print(f"✅ Selected flow: {flow_name} (ID: {flow_id})")
+        print("-" * 50)
+        return flow_id, flow_name
+
+    except Exception as e:
+        print(f"❌ Error fetching flows: {e}")
+        print("Falling back to hardcoded flow ID...")
+        return "6dd45ac0-5c05-44c1-9825-66c6c6e516f7", "Fallback Flow"
+
+
+async def create_agent(flow_id: str, flow_name: str) -> LangflowAgent:
+    """Create and configure the Langflow agent.
+
+    Args:
+        flow_id (str): The flow ID to use for the agent.
+        flow_name (str): The name of the flow.
+
+    Returns:
+        LangflowAgent: The configured Langflow agent.
+    """
+    agent = LangflowAgent(
+        name="HelloWorldLangflow",
+        flow_id=flow_id,
+        description=f"A Langflow agent using flow: {flow_name}",
+        base_url="https://langflow.obrol.id",
+    )
+
+    print(f"Created agent: {agent.name}")
+    print(f"Flow ID: {flow_id}")
+    print(f"API Base URL: {agent.api_client.base_url}")
+
+    # Test health check
+    print("Testing health check...")
+    is_healthy = await agent.health_check()
+    print(f"API Health: {'✓ Healthy' if is_healthy else '✗ Unhealthy'}")
+    print("-" * 50)
+
+    return agent
+
+
+async def demonstrate_regular_execution(agent: LangflowAgent) -> None:
+    """Demonstrate regular execution.
+
+    Args:
+        agent (LangflowAgent): The Langflow agent to demonstrate with.
+    """
+    print("Example 1: Regular execution")
+    result1 = await agent.arun("Hello, how are you?")
+    print(f"Message 1: {result1}")
+    print("-" * 50)
+
+
+async def demonstrate_streaming(agent: LangflowAgent) -> None:
+    """Demonstrate streaming execution.
+
+    Args:
+        agent (LangflowAgent): The Langflow agent to demonstrate with.
+    """
+    print("Example 2: Regular streaming")
+    try:
+        print("Streaming response:")
+        async for chunk in agent.arun_stream("Hello, how are you?"):
+            print(chunk)
+        print()  # New line after streaming
+    except Exception as e:
+        print(f"Streaming Error: {e}")
+    print("-" * 50)
+
+
+async def demonstrate_session_management(agent: LangflowAgent) -> None:
+    """Demonstrate session management.
+
+    Args:
+        agent (LangflowAgent): The Langflow agent to demonstrate with.
+    """
+    print("Example 3: Session management")
+    session_id = f"conversation-{uuid.uuid4()}"
+    try:
+        # First message with thread_id
+        result1 = await agent.arun("My name is Alice", configurable={"thread_id": session_id})
+        print(f"Message 1: {result1}")
+
+        # Second message in same thread - should remember context
+        result2 = await agent.arun("What's my name?", configurable={"thread_id": session_id})
+        print(f"Message 2: {result2}")
+
+        # Third message in different thread - should not remember
+        result3 = await agent.arun("What's my name?", configurable={"thread_id": f"new-conversation-{uuid.uuid4()}"})
+        print(f"Message 3 (new thread): {result3}")
+
+    except Exception as e:
+        print(f"Session Error: {e}")
+    print("-" * 50)
+
+
+async def main():
+    """Demonstrate basic Langflow agent usage."""
+    # Fetch flow ID and name
+    flow_id, flow_name = await fetch_flow_id()
+    if not flow_id:
+        return
+
+    # Create agent
+    agent = await create_agent(flow_id, flow_name)
+
+    # Run demonstrations
+    await demonstrate_regular_execution(agent)
+    await demonstrate_streaming(agent)
+    await demonstrate_session_management(agent)
+
+    print("Demo completed!")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langflow_agent.pyi

@@ -0,0 +1,35 @@
+from aip_agents.agent.langflow_agent import LangflowAgent as LangflowAgent
+from aip_agents.clients.langflow import LangflowApiClient as LangflowApiClient
+
+async def fetch_flow_id() -> tuple[str, str]:
+    """Fetch available flows and return the first flow ID and name."""
+async def create_agent(flow_id: str, flow_name: str) -> LangflowAgent:
+    """Create and configure the Langflow agent.
+
+    Args:
+        flow_id (str): The flow ID to use for the agent.
+        flow_name (str): The name of the flow.
+
+    Returns:
+        LangflowAgent: The configured Langflow agent.
+    """
+async def demonstrate_regular_execution(agent: LangflowAgent) -> None:
+    """Demonstrate regular execution.
+
+    Args:
+        agent (LangflowAgent): The Langflow agent to demonstrate with.
+    """
+async def demonstrate_streaming(agent: LangflowAgent) -> None:
+    """Demonstrate streaming execution.
+
+    Args:
+        agent (LangflowAgent): The Langflow agent to demonstrate with.
+    """
+async def demonstrate_session_management(agent: LangflowAgent) -> None:
+    """Demonstrate session management.
+
+    Args:
+        agent (LangflowAgent): The Langflow agent to demonstrate with.
+    """
+async def main() -> None:
+    """Demonstrate basic Langflow agent usage."""

aip_agents/examples/hello_world_langgraph.py

@@ -0,0 +1,39 @@
+"""Minimal LangGraph agent example demonstrating asynchronous run."""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers
+
+
+async def langgraph_example():
+    """Demonstrates the LangGraphAgent's arun method."""
+    model = ChatOpenAI(model="gpt-4.1", temperature=0)
+    tools = [add_numbers]
+    agent_name = "LangGraphArithmeticAgent"
+
+    langgraph_agent = LangGraphAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that can add two numbers using the add_numbers tool.",
+        model=model,
+        tools=tools,
+    )
+
+    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 langgraph_agent.arun(
+        query=query,
+        configurable={"configurable": {"thread_id": "lgraph_arith_example_arun"}},
+    )
+    print(f"[arun] Final Response: {response['output']}")
+    print("--- End of LangGraph Example ---")
+
+
+if __name__ == "__main__":
+    # OPENAI_API_KEY should be set in the environment.
+    asyncio.run(langgraph_example())

aip_agents/examples/hello_world_langgraph_bosa_twitter.py

@@ -0,0 +1,41 @@
+"""Minimal LangGraph agent with bosa support example demonstrating asynchronous run.
+
+Authors:
+    Saul Sayers (saul.sayers@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.tools import BOSA_AUTOMATED_TOOLS
+
+
+async def langgraph_bosa_example():
+    """Demonstrates the LangGraphAgent's arun method."""
+    model = ChatOpenAI(model="gpt-4.1", temperature=0)
+    agent_name = "BOSAConnectorTwitterAgent"
+
+    langgraph_agent = LangGraphAgent(
+        name=agent_name,
+        instruction="You are a helpful assistant that use BOSA connector to connect with Twitter API.",
+        model=model,
+        tools=BOSA_AUTOMATED_TOOLS["twitter"],
+    )
+
+    query = "Get 3 tweets about the latest Air India Incident"
+    print(f"--- Agent: {agent_name} ---")
+    print(f"Query: {query}")
+
+    print("\nRunning arun...")
+    response = await langgraph_agent.arun(
+        query=query,
+        configurable={"configurable": {"thread_id": "lgraph_arith_example_arun"}},
+    )
+    print(f"[arun] Final Response: {response['output']}")
+    print("--- End of LangGraph Example ---")
+
+
+if __name__ == "__main__":
+    asyncio.run(langgraph_bosa_example())

aip_agents/examples/hello_world_langgraph_mcp_http.py

@@ -0,0 +1,31 @@
+"""Example showing LangGraph agent with MCP tools integration.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_http
+
+
+async def main():
+    """Demonstrates the LangGraphAgent with MCP tools via Streamable HTTP transport."""
+    langgraph_agent = LangGraphAgent(
+        name="langgraph_mcp_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+    langgraph_agent.add_mcp_server(mcp_config_http)
+
+    response = await langgraph_agent.arun(query="What's the weather forecast for monday?")
+    print(f"Response: {response['output']}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langgraph_mcp_http_stream.py

@@ -0,0 +1,34 @@
+"""Example showing LangGraph agent with MCP tools integration and streaming capabilities.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_http
+
+
+async def main():
+    """Demonstrates the LangGraphAgent with MCP tools via Streamable HTTP transport and streaming capabilities."""
+    langgraph_agent = LangGraphAgent(
+        name="langgraph_mcp_stream_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+    )
+    langgraph_agent.add_mcp_server(mcp_config_http)
+
+    async for chunk in langgraph_agent.arun_stream(
+        query="What's the weather forecast for monday?",
+    ):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langgraph_mcp_sse.py

@@ -0,0 +1,35 @@
+"""Example showing LangGraph agent with MCP tools integration.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_sse
+
+
+async def main():
+    """Demonstrates the LangGraphAgent with MCP tools via SSE transport."""
+    langgraph_agent = LangGraphAgent(
+        name="langgraph_mcp_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+        tools=[],
+    )
+    langgraph_agent.add_mcp_server(mcp_config_sse)
+
+    query = "What's the weather forecast for monday?"  # Uses MCP weather tool
+
+    print(f"\nQuery: {query}")
+    response = await langgraph_agent.arun(query=query)
+    print(f"Response: {response['output']}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langgraph_mcp_sse_stream.py

@@ -0,0 +1,50 @@
+"""Example showing LangGraph agent with MCP tools integration and streaming capabilities.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_sse
+
+
+async def main():
+    """Demonstrates the LangGraphAgent with MCP tools via SSE transport and streaming capabilities."""
+    langgraph_agent = LangGraphAgent(
+        name="langgraph_mcp_stream_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+        tools=[],
+    )
+    langgraph_agent.add_mcp_server(mcp_config_sse)
+
+    query = "What's the weather forecast for monday?"  # Uses MCP weather tool
+
+    stream_thread_id = "langgraph_mcp_stream_example"
+
+    print(f"\nQuery: {query}")
+    print("Streaming response:")
+
+    full_response = ""
+    async for chunk in langgraph_agent.arun_stream(
+        query=query, configurable={"configurable": {"thread_id": stream_thread_id}}
+    ):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)
+            full_response += chunk
+        elif isinstance(chunk, dict) and "messages" in chunk:
+            print("\n(Stream finished with final state object)")
+        elif isinstance(chunk, dict):
+            pass
+
+    print(f"\nFull response collected: {full_response}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langgraph_mcp_stdio.py

@@ -0,0 +1,35 @@
+"""Example showing LangGraph agent with MCP tools integration using stdio transport.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_stdio
+
+
+async def main():
+    """Demonstrates the LangGraphAgent with MCP tools via stdio transport."""
+    langgraph_agent = LangGraphAgent(
+        name="langgraph_mcp_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+        tools=[],
+    )
+    langgraph_agent.add_mcp_server(mcp_config_stdio)
+
+    query = "What's the weather forecast for monday?"  # Uses MCP weather tool
+
+    print(f"\nQuery: {query}")
+    response = await langgraph_agent.arun(query=query)
+    print(f"Response: {response['output']}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_langgraph_mcp_stdio_stream.py

@@ -0,0 +1,50 @@
+"""Example showing LangGraph agent with MCP tools integration and streaming capabilities using stdio transport.
+
+Authors:
+    Fachriza Dian Adhiatma (fachriza.d.adhiatma@gdplabs.id)
+    Putu Ravindra Wiguna (putu.r.wiguna@gdplabs.id)
+"""
+
+import asyncio
+
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.mcp_configs.configs import mcp_config_stdio
+
+
+async def main():
+    """Demonstrates the LangGraphAgent with MCP tools via stdio transport and streaming."""
+    langgraph_agent = LangGraphAgent(
+        name="langgraph_mcp_stream_example",
+        instruction="""You are a helpful assistant that can provide weather forecasts.
+        For weather, specify the day in lowercase (e.g., 'monday').""",
+        model=ChatOpenAI(model="gpt-4.1", temperature=0),
+        tools=[],
+    )
+    langgraph_agent.add_mcp_server(mcp_config_stdio)
+
+    query = "What's the weather forecast for monday?"  # Uses MCP weather tool
+
+    stream_thread_id = "langgraph_mcp_stream_example"
+
+    print(f"\nQuery: {query}")
+    print("Streaming response:")
+
+    full_response = ""
+    async for chunk in langgraph_agent.arun_stream(
+        query=query, configurable={"configurable": {"thread_id": stream_thread_id}}
+    ):
+        if isinstance(chunk, str):
+            print(chunk, end="", flush=True)
+            full_response += chunk
+        elif isinstance(chunk, dict) and "messages" in chunk:
+            print("\n(Stream finished with final state object)")
+        elif isinstance(chunk, dict):
+            pass
+
+    print(f"\nFull response collected: {full_response}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())