aip-agents-binary 0.5.17__py3-none-any.whl → 0.5.19__py3-none-any.whl

aip_agents/examples/pii_demo_langgraph_server.pyi

@@ -0,0 +1,20 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.examples.tools.pii_demo_tools import get_customer_info as get_customer_info, get_employee_data as get_employee_data, get_user_profile as get_user_profile
+from aip_agents.utils.logger import LoggerManager as LoggerManager
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+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.
+    """

aip_agents/examples/pii_demo_multi_agent_client.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+async def main() -> None:
+    """Main function demonstrating the Multi-Agent PII Demo client."""

aip_agents/examples/pii_demo_multi_agent_server.pyi

@@ -0,0 +1,40 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphAgent as LangGraphAgent
+from aip_agents.examples.tools.pii_demo_tools import get_customer_info as get_customer_info, get_employee_data as get_employee_data, get_user_profile as get_user_profile
+from aip_agents.utils.logger import LoggerManager as LoggerManager
+from langchain_openai import ChatOpenAI
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+def create_specialist_agents(llm: ChatOpenAI) -> tuple[LangGraphAgent, LangGraphAgent, LangGraphAgent]:
+    """Create Level 2 specialist agents with PII-returning tools.
+
+    Args:
+        llm: The language model to use for the specialist agents.
+
+    Returns:
+        Tuple of (customer_service_agent, hr_agent, user_support_agent).
+    """
+def create_coordinator_agent(llm: ChatOpenAI, specialist_agents: tuple[LangGraphAgent, LangGraphAgent, LangGraphAgent]) -> LangGraphAgent:
+    """Create Level 1 coordinator agent that orchestrates specialist agents.
+
+    Args:
+        llm: The language model to use for the coordinator agent.
+        specialist_agents: Tuple of specialist agents.
+
+    Returns:
+        The configured coordinator agent.
+    """
+def main(host: str, port: int):
+    """Runs the Multi-Agent PII Demo A2A server.
+
+    This server demonstrates PII handling across a multi-agent hierarchy:
+    - PII anonymization in tool outputs
+    - PII propagation from child agents to parent
+    - Multi-agent delegation with PII mapping merge
+
+    Args:
+        host: Host to bind the server to.
+        port: Port to bind the server to.
+    """

aip_agents/examples/todolist_planning_a2a_langchain_client.pyi

@@ -0,0 +1,5 @@
+from aip_agents.agent import LangChainAgent as LangChainAgent
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig
+
+async def main() -> None:
+    """Connect to the planning A2A server and stream a planning query."""

aip_agents/examples/todolist_planning_a2a_langgraph_server.pyi

@@ -0,0 +1,19 @@
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
+from aip_agents.examples.tools.serper_tool import MockGoogleSerperTool as MockGoogleSerperTool
+from aip_agents.examples.tools.stock_tools import get_stock_price as get_stock_price
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+SERVER_AGENT_NAME: str
+
+def main(host: str, port: int) -> None:
+    """Run an A2A server with a planning-enabled LangGraphReactAgent.
+
+    The agent has TodoListMiddleware attached via planning=True and will
+    expose the write_todos_tool over A2A with token streaming enabled.
+
+    Args:
+        host: The host to bind the server to.
+        port: The port to bind the server to.
+    """

aip_agents/examples/tools/__init__.pyi

@@ -0,0 +1,9 @@
+from aip_agents.examples.tools.adk_weather_tool import get_weather as weather_tool_adk
+from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool_langchain
+from aip_agents.examples.tools.langgraph_streaming_tool import LangGraphStreamingTool as langgraph_streaming_tool
+from aip_agents.examples.tools.mock_retrieval_tool import MockRetrievalTool as mock_retrieval_tool
+from aip_agents.examples.tools.random_chart_tool import RandomChartTool as random_chart_tool
+from aip_agents.examples.tools.serper_tool import MockGoogleSerperTool as google_serper_tool
+from aip_agents.examples.tools.time_tool import TimeTool as time_tool
+
+__all__ = ['weather_tool_langchain', 'weather_tool_adk', 'time_tool', 'google_serper_tool', 'langgraph_streaming_tool', 'mock_retrieval_tool', 'random_chart_tool']

aip_agents/examples/tools/adk_arithmetic_tools.pyi

@@ -0,0 +1,24 @@
+def add_numbers(a: int, b: int) -> str:
+    """Adds two integer numbers together and returns the result as a string.
+
+    For example, to add 5 and 3, you would call this with a=5 and b=3.
+
+    Args:
+        a: The first number.
+        b: The second number.
+
+    Returns:
+        The sum of the two numbers as a string, or an error message if addition fails.
+    """
+def sum_numbers(a: int, b: int) -> int:
+    """Adds two integer numbers together and returns the integer result.
+
+    Used specifically in Google ADK examples where an integer return is expected.
+
+    Args:
+        a: The first number.
+        b: The second number.
+
+    Returns:
+        The sum of the two numbers as an integer.
+    """

aip_agents/examples/tools/adk_weather_tool.pyi

@@ -0,0 +1,18 @@
+from _typeshed import Incomplete
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+WEATHER_DATA: Incomplete
+
+def get_weather(location: str) -> dict:
+    """Gets detailed weather information for a specified location.
+
+    Args:
+        location: The name of the city to get weather for.
+
+    Returns:
+        A dictionary containing:
+            - location: The requested location name
+            - weather: A dictionary with temperature, conditions, and humidity
+    """
+weather_tool = get_weather

aip_agents/examples/tools/data_generator_tool.pyi

@@ -0,0 +1,15 @@
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+from typing import Literal
+
+class DataInput(BaseModel):
+    """Input schema for data generation."""
+    data_type: Literal['sales', 'scores', 'growth']
+    size: int
+
+class DataGeneratorTool(BaseTool):
+    """Tool that generates sample datasets with automatic output storage."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]
+    store_final_output: bool

aip_agents/examples/tools/data_visualization_tool.pyi

@@ -0,0 +1,19 @@
+from aip_agents.a2a.types import MimeType as MimeType
+from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+from typing import Any
+
+PIL_AVAILABLE: bool
+
+class ChartInput(BaseModel):
+    """Input schema for chart generation."""
+    data_source: Any
+    chart_type: str
+    title: str
+
+class DataVisualizerTool(BaseTool):
+    """Tool that creates visualizations with automatic output storage."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]

aip_agents/examples/tools/image_artifact_tool.pyi

@@ -0,0 +1,26 @@
+from aip_agents.a2a.types import MimeType as MimeType
+from aip_agents.utils.artifact_helpers import create_artifact_response as create_artifact_response, create_error_response as create_error_response
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+PIL_AVAILABLE: bool
+
+class ImageArtifactInput(BaseModel):
+    """Input schema for image artifact tool."""
+    width: int
+    height: int
+    color: str
+    text: str
+    image_name: str
+
+class ImageArtifactTool(BaseTool):
+    """Tool that generates simple images with artifact support.
+
+    This tool demonstrates the standardized artifact format for binary data by:
+    1. Generating a simple image using PIL
+    2. Providing a confirmation message for the agent
+    3. Creating an image file artifact for the user
+    """
+    name: str
+    description: str
+    args_schema: type[BaseModel]

aip_agents/examples/tools/langchain_arithmetic_tools.pyi

@@ -0,0 +1,17 @@
+from langchain_core.tools import tool
+
+@tool
+def add_numbers(a: int, b: int) -> str:
+    """Adds two numbers and returns the result as a string.
+
+    This is a self-contained LangChain tool.
+    For example:
+    add_numbers(a=5, b=3)
+
+    Args:
+        a (int): First number to add.
+        b (int): Second number to add.
+
+    Returns:
+        str: Sum of the two numbers as a string.
+    """

aip_agents/examples/tools/langchain_currency_exchange_tool.pyi

@@ -0,0 +1,20 @@
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+class CurrencyConfig(BaseModel):
+    """Tool configuration schema."""
+    tenant_id: str
+    auth_key: str
+
+class CurrencyInput(BaseModel):
+    """Input schema for currency exchange."""
+    amount: float
+    from_currency: str
+    to_currency: str
+
+class CurrencyExchangeTool(BaseTool):
+    """Simple currency exchange tool with SDK auto-injection configuration."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]
+    tool_config_schema: type[BaseModel]

aip_agents/examples/tools/langchain_graph_artifact_tool.pyi

@@ -0,0 +1,25 @@
+from aip_agents.a2a.types import MimeType as MimeType
+from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+PIL_AVAILABLE: bool
+
+class GraphCommandInput(BaseModel):
+    """Input schema for a tiny bar/line chart artifact tool.
+
+    Note: width/height are clamped to a small size to keep artifacts lightweight.
+    """
+    chart_type: str
+    labels: list[str]
+    values: list[float]
+    title: str
+    width: int
+    height: int
+    image_name: str
+
+class GraphArtifactCommandTool(BaseTool):
+    """Generate a very small PNG chart and return it as a Command artifact."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]

aip_agents/examples/tools/langchain_weather_tool.pyi

@@ -0,0 +1,19 @@
+from _typeshed import Incomplete
+from aip_agents.utils.logger import get_logger as get_logger
+from pydantic import BaseModel
+
+logger: Incomplete
+
+class WeatherToolInputSchema(BaseModel):
+    """Schema for weather tool input."""
+    city: str
+
+def weather_tool(city: str) -> str:
+    """Gets the weather for a specified city.
+
+    Args:
+        city: The name of the city to get weather for.
+
+    Returns:
+        A string describing the weather conditions.
+    """

aip_agents/examples/tools/langgraph_streaming_tool.pyi

@@ -0,0 +1,43 @@
+from aip_agents.agent.langgraph_react_agent import LangGraphReactAgent as LangGraphReactAgent
+from aip_agents.examples.tools.langchain_weather_tool import weather_tool as weather_tool
+from aip_agents.examples.tools.time_tool import TimeTool as TimeTool
+from langchain_core.runnables import RunnableConfig
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+from typing import Any
+
+class StreamingToolConfig(BaseModel):
+    """Tool configuration schema for the LangGraph streaming tool."""
+    time_format: str
+
+class LangGraphStreamingToolInput(BaseModel):
+    """Input schema for the LangGraphStreamingTool."""
+    query: str
+
+class LangGraphStreamingTool(BaseTool):
+    """Tool that wraps a LangGraph agent with time and weather forecast capabilities."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]
+    tool_config_schema: type[BaseModel]
+    def __init__(self, model: Any = None, **kwargs) -> None:
+        """Initialize the LangGraphStreamingTool.
+
+        Args:
+            model: The model to use for the agent.
+            **kwargs: Additional keyword arguments.
+        """
+    @property
+    def agent(self):
+        """Access the internal agent."""
+    async def arun_streaming(self, query: str = None, config: RunnableConfig = None, **kwargs):
+        """Execute the LangGraph agent asynchronously with A2A streaming output.
+
+        Args:
+            query: The query to execute.
+            config: Tool configuration containing user context and preferences.
+            **kwargs: Additional keyword arguments.
+
+        Returns:
+            An async generator that yields streaming events.
+        """

aip_agents/examples/tools/mock_retrieval_tool.pyi

@@ -0,0 +1,13 @@
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+class MockRetrievalInput(BaseModel):
+    """Input schema for the MockRetrievalTool."""
+    query: str
+
+class MockRetrievalTool(BaseTool):
+    """Mock tool that returns hardcoded references for testing."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]
+    save_output_history: bool

aip_agents/examples/tools/pii_demo_tools.pyi

@@ -0,0 +1,54 @@
+from _typeshed import Incomplete
+from aip_agents.utils.logger import LoggerManager as LoggerManager
+from pydantic import BaseModel
+
+logger: Incomplete
+
+class CustomerIdSchema(BaseModel):
+    """Schema for customer ID input."""
+    customer_id: str
+
+class EmployeeIdSchema(BaseModel):
+    """Schema for employee ID input."""
+    employee_id: str
+
+class UserIdSchema(BaseModel):
+    """Schema for user ID input."""
+    user_id: str
+
+def get_customer_info(customer_id: str) -> str:
+    """Gets customer information including email and phone number.
+
+    This tool demonstrates PII handling by returning sensitive customer data
+    that will be automatically anonymized by the PII handler.
+
+    Args:
+        customer_id: The ID of the customer to retrieve information for.
+
+    Returns:
+        A string containing customer information with PII.
+    """
+def get_employee_data(employee_id: str) -> str:
+    """Gets employee data including name, email, and salary information.
+
+    This tool demonstrates PII handling with employee-specific data that includes
+    sensitive information like salary and personal email addresses.
+
+    Args:
+        employee_id: The ID of the employee to retrieve data for.
+
+    Returns:
+        A string containing employee information with PII.
+    """
+def get_user_profile(user_id: str) -> str:
+    """Gets user profile information with personal details.
+
+    This tool demonstrates PII handling with user profile data that includes
+    personal information like email, phone, and date of birth.
+
+    Args:
+        user_id: The ID of the user to retrieve profile for.
+
+    Returns:
+        A string containing user profile information with PII.
+    """

aip_agents/examples/tools/random_chart_tool.pyi

@@ -0,0 +1,20 @@
+from aip_agents.a2a.types import MimeType as MimeType
+from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+DEFAULT_RANDOM_CHART_TITLE: str
+PIL_AVAILABLE: bool
+
+class RandomChartInput(BaseModel):
+    """Input schema for random chart generation."""
+    title: str
+    num_bars: int
+    min_value: int
+    max_value: int
+
+class RandomChartTool(BaseTool):
+    """Generate random bar chart images without relying on upstream data."""
+    name: str
+    description: str
+    args_schema: type[BaseModel]

aip_agents/examples/tools/serper_tool.pyi

@@ -0,0 +1,16 @@
+from _typeshed import Incomplete
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+logger: Incomplete
+
+class GoogleSerperInput(BaseModel):
+    """Input schema for the GoogleSerperTool."""
+    query: str
+
+class MockGoogleSerperTool(BaseTool):
+    """Mock Tool to simulate Google Serper API results for testing."""
+    name: str
+    description: str
+    save_output_history: bool
+    args_schema: type[BaseModel]

aip_agents/examples/tools/stock_tools.pyi

@@ -0,0 +1,36 @@
+from _typeshed import Incomplete
+from aip_agents.utils.logger import get_logger as get_logger
+from pydantic import BaseModel
+from typing import Any
+
+logger: Incomplete
+
+class StockPriceInput(BaseModel):
+    """Input for the stock price tool."""
+    symbol: str
+
+def get_stock_price(symbol: str) -> dict[str, Any]:
+    """Get current stock price and performance data for a given symbol.
+
+    Args:
+        symbol (str): The stock symbol (e.g., AAPL, MSFT) to get the price for.
+
+    Returns:
+        dict[str, Any]: Dictionary containing stock price and performance data.
+    """
+
+class StockNewsInput(BaseModel):
+    """Input for the stock news tool."""
+    symbol: str
+    days: int
+
+def get_stock_news(symbol: str, days: int = 7) -> dict[str, Any]:
+    """Get recent news for a stock for a specified number of days.
+
+    Args:
+        symbol (str): The stock symbol (e.g., AAPL, MSFT) to get news for.
+        days (int, optional): Number of days of news to retrieve. Defaults to 7.
+
+    Returns:
+        dict[str, Any]: Dictionary containing stock news data.
+    """

aip_agents/examples/tools/table_generator_tool.pyi

@@ -0,0 +1,22 @@
+from aip_agents.a2a.types import MimeType as MimeType
+from aip_agents.utils.artifact_helpers import create_artifact_response as create_artifact_response
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+class TableGeneratorInput(BaseModel):
+    """Input schema for table generator tool."""
+    rows: int
+    columns: list[str]
+    table_name: str
+
+class TableGeneratorTool(BaseTool):
+    """Tool that generates sample data tables with artifact support.
+
+    This tool demonstrates the standardized artifact format by:
+    1. Generating sample data
+    2. Creating a markdown table for the agent's context
+    3. Creating a CSV file artifact for the user
+    """
+    name: str
+    description: str
+    args_schema: type[BaseModel]

aip_agents/examples/tools/time_tool.pyi

@@ -0,0 +1,15 @@
+from langchain_core.tools import BaseTool
+from pydantic import BaseModel
+
+FORMAT_STRING: str
+
+class TimeToolInput(BaseModel):
+    """Input schema for the TimeTool."""
+    datetime_format: str
+
+class TimeTool(BaseTool):
+    """Tool to get the current time."""
+    name: str
+    description: str
+    save_output_history: bool
+    args_schema: type[BaseModel]

aip_agents/examples/tools/weather_forecast_tool.pyi

@@ -0,0 +1,14 @@
+from _typeshed import Incomplete
+
+DAILY_WEATHER: Incomplete
+DayOfWeek: Incomplete
+
+def get_weather_forecast(day: DayOfWeek) -> str:
+    """Get the weather forecast for a specific day of the week.
+
+    Args:
+        day: The day of the week (monday, tuesday, wednesday, etc.)
+
+    Returns:
+        str: A weather forecast message including conditions and temperature.
+    """

aip_agents/mcp/client/__init__.pyi

@@ -0,0 +1,5 @@
+from aip_agents.mcp.client.base_mcp_client import BaseMCPClient as BaseMCPClient
+from aip_agents.mcp.client.google_adk.client import GoogleADKMCPClient as GoogleADKMCPClient
+from aip_agents.mcp.client.langchain.client import LangchainMCPClient as LangchainMCPClient
+
+__all__ = ['GoogleADKMCPClient', 'LangchainMCPClient', 'BaseMCPClient']