aip-agents-binary 0.5.25b1__py3-none-macosx_13_0_arm64.whl

aip_agents/examples/hello_world_stock_a2a_server.py

@@ -0,0 +1,103 @@
+"""Stock Market A2A Server Example.
+
+This server provides stock market information and can be used as an A2A agent.
+
+To run:
+    python aip_agents/examples/hello_world_stock_a2a_server.py --port 8002
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+import os
+
+import click
+import uvicorn
+from a2a.types import AgentCapabilities, AgentCard, AgentSkill
+from langchain_core.tools import BaseTool
+from langchain_openai import ChatOpenAI
+
+from aip_agents.agent import LangGraphAgent
+from aip_agents.examples.tools.stock_tools import StockNewsInput, StockPriceInput, get_stock_news, get_stock_price
+from aip_agents.utils.logger import get_logger
+
+logger = get_logger(__name__)
+
+SERVER_AGENT_NAME = "StockAgent"
+
+
+stock_tools: list[BaseTool] = [get_stock_price, get_stock_news]
+
+
+@click.command()
+@click.option("--host", "host", default="0.0.0.0", 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 StockAgent 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="Provides stock market information including prices and news.",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True, tools=True),
+        skills=[
+            AgentSkill(
+                id="get_stock_price",
+                name="Get Stock Price",
+                description="Retrieves the current stock price and performance data for a given symbol.",
+                examples=["What's the price of AAPL?", "Stock price for MSFT"],
+                tags=["finance", "stocks", "price"],
+                input_schema=StockPriceInput.model_json_schema(),
+            ),
+            AgentSkill(
+                id="get_stock_news",
+                name="Get Stock News",
+                description="Fetches recent news articles for a specified stock symbol.",
+                examples=[
+                    "Latest news for GOOGL",
+                    "Stock news for TSLA over last 3 days",
+                ],
+                tags=["finance", "stocks", "news"],
+                input_schema=StockNewsInput.model_json_schema(),
+            ),
+        ],
+        tags=["finance", "stocks"],
+    )
+
+    # For A2A, the LLM is used by the *client* agent to decide to call this agent.
+    # This agent itself just executes its tools based on the A2A request.
+    # However, LangGraphAgent requires an LLM, so we provide a basic one.
+    llm = ChatOpenAI(
+        model="gpt-4.1",
+        temperature=0,
+        streaming=True,
+        openai_api_key=os.getenv("OPENAI_API_KEY"),
+    )
+
+    langgraph_agent = LangGraphAgent(
+        name=SERVER_AGENT_NAME,
+        instruction="You are a Stock Agent. You can get stock prices and news. Use your tools to answer user queries.",
+        model=llm,
+        tools=stock_tools,
+        verbose=True,
+    )
+
+    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/hello_world_stock_a2a_server.pyi

@@ -0,0 +1,17 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+from aip_agents.examples.tools.stock_tools import StockNewsInput as StockNewsInput, StockPriceInput as StockPriceInput, get_stock_news as get_stock_news, get_stock_price as get_stock_price
+from aip_agents.utils.logger import get_logger as get_logger
+from langchain_core.tools import BaseTool as BaseTool
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+stock_tools: list[BaseTool]
+
+def main(host: str, port: int):
+    """Runs the StockAgent A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """

aip_agents/examples/hello_world_tool_output_client.py

@@ -0,0 +1,46 @@
+"""Simple A2A client demonstrating tool output management.
+
+This client connects to the data visualization server and demonstrates
+how tools can reference each other's outputs using the tool output
+management system.
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+import asyncio
+
+from dotenv import load_dotenv
+
+from aip_agents.agent import LangGraphReactAgent
+from aip_agents.schema.agent import A2AClientConfig
+
+load_dotenv()
+
+
+async def main():
+    """Demonstrate tool output management with data visualization."""
+    # Create client agent
+    client = LangGraphReactAgent(
+        name="Client",
+        instruction="You request data visualization services.",
+        model="openai/gpt-4o-mini",
+    )
+
+    # Discover server agent
+    agents = client.discover_agents(A2AClientConfig(discovery_urls=["http://localhost:8885"]))
+    server_agent = agents[0]
+
+    async for chunk in client.astream_to_agent(
+        server_agent,
+        "Generate sales data for 1000 months and create a bar chart from it",
+    ):
+        if chunk.get("content"):
+            print(chunk["content"], end="", flush=True)
+        if chunk.get("metadata"):
+            print(f"\nMetadata: {chunk['metadata']}", end="\n\n", flush=True)
+    print("\n")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

aip_agents/examples/hello_world_tool_output_client.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+async def main() -> None:
+    """Demonstrate tool output management with data visualization."""

aip_agents/examples/hello_world_tool_output_server.py

@@ -0,0 +1,114 @@
+"""Example A2A server demonstrating tool output management.
+
+This server provides data generation and visualization capabilities,
+demonstrating how tools can store and reference each other's outputs.
+
+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 import LangGraphReactAgent
+from aip_agents.examples.tools.data_generator_tool import DataGeneratorTool
+from aip_agents.examples.tools.data_visualization_tool import DataVisualizerTool
+from aip_agents.storage.clients.minio_client import MinioConfig, MinioObjectStorage
+from aip_agents.storage.providers.object_storage import ObjectStorageProvider
+from aip_agents.utils.langgraph.tool_output_management import ToolOutputConfig, ToolOutputManager
+from aip_agents.utils.logger import get_logger
+
+load_dotenv()
+logger = get_logger(__name__)
+
+SERVER_AGENT_NAME = "DataVisualizationAgent"
+
+
+@click.command()
+@click.option("--host", default="localhost", help="Host to bind the server to")
+@click.option("--port", default=8885, help="Port to bind the server to")
+def main(host: str, port: int):
+    """Run the Data Visualization 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}")
+
+    # Create agent card
+    agent_card = AgentCard(
+        name=SERVER_AGENT_NAME,
+        description="Agent that generates and visualizes data with tool output references",
+        url=f"http://{host}:{port}",
+        version="1.0.0",
+        defaultInputModes=["text"],
+        defaultOutputModes=["text"],
+        capabilities=AgentCapabilities(streaming=True),
+        skills=[
+            AgentSkill(
+                id="data_generation",
+                name="Data Generation",
+                description="Generate sample datasets (sales, scores, growth)",
+                examples=[
+                    "Generate sales data with 5 points",
+                    "Create test scores for 3 students",
+                ],
+                tags=["data"],
+            ),
+            AgentSkill(
+                id="data_visualization",
+                name="Data Visualization",
+                description="Create charts from data (supports tool output references)",
+                examples=[
+                    "Create a bar chart from the sales data",
+                    "Visualize test scores as a line chart",
+                ],
+                tags=["visualization"],
+            ),
+        ],
+        tags=["data", "visualization"],
+    )
+
+    # Use environment-based MinIO configuration
+    minio_config = MinioConfig.from_env()
+    minio_client = MinioObjectStorage(minio_config)
+    object_storage_provider = ObjectStorageProvider(
+        client=minio_client,
+        prefix="test_prefix",  # Unique test prefix
+    )
+    tool_output_config = ToolOutputConfig(
+        storage_provider=object_storage_provider,
+    )
+
+    # Create agent with tool output management
+    tool_output_manager = ToolOutputManager(tool_output_config)
+    agent = LangGraphReactAgent(
+        name=SERVER_AGENT_NAME,
+        instruction="""You are a data visualization agent that can generate and visualize data.
+
+You have access to these tools:
+1. data_generator: Creates sample datasets (sales, scores, growth data)
+2. data_visualizer: Creates charts from data, supporting tool output references
+
+Always be explicit about what data you're generating and visualizing.""",
+        model="openai/gpt-4.1",
+        tools=[
+            DataGeneratorTool(),
+            DataVisualizerTool(),
+        ],
+        tool_output_manager=tool_output_manager,
+    )
+
+    # Create A2A app
+    app = agent.to_a2a(agent_card=agent_card)
+
+    # Run server
+    uvicorn.run(app, host=host, port=port)
+    logger.info("A2A application configured. Starting Uvicorn server...")
+
+
+if __name__ == "__main__":
+    main()

aip_agents/examples/hello_world_tool_output_server.pyi

@@ -0,0 +1,19 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
+from aip_agents.examples.tools.data_generator_tool import DataGeneratorTool as DataGeneratorTool
+from aip_agents.examples.tools.data_visualization_tool import DataVisualizerTool as DataVisualizerTool
+from aip_agents.storage.clients.minio_client import MinioConfig as MinioConfig, MinioObjectStorage as MinioObjectStorage
+from aip_agents.storage.providers.object_storage import ObjectStorageProvider as ObjectStorageProvider
+from aip_agents.utils.langgraph.tool_output_management import ToolOutputConfig as ToolOutputConfig, ToolOutputManager as ToolOutputManager
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+def main(host: str, port: int):
+    """Run the Data Visualization A2A server.
+
+    Args:
+        host (str): Host to bind the server to.
+        port (int): Port to bind the server to.
+    """