aip-agents-binary 0.5.25b1__py3-none-any.whl → 0.5.25b8__py3-none-any.whl

aip_agents/examples/pii_demo_langgraph_server.pyi

@@ -1,20 +0,0 @@
-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

@@ -1,5 +0,0 @@
-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

@@ -1,40 +0,0 @@
-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

@@ -1,5 +0,0 @@
-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

@@ -1,19 +0,0 @@
-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

@@ -1,9 +0,0 @@
-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

@@ -1,24 +0,0 @@
-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

@@ -1,18 +0,0 @@
-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

@@ -1,15 +0,0 @@
-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

@@ -1,19 +0,0 @@
-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

@@ -1,26 +0,0 @@
-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

@@ -1,17 +0,0 @@
-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

@@ -1,20 +0,0 @@
-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

@@ -1,25 +0,0 @@
-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

@@ -1,19 +0,0 @@
-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

@@ -1,43 +0,0 @@
-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

@@ -1,13 +0,0 @@
-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

@@ -1,54 +0,0 @@
-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

@@ -1,20 +0,0 @@
-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

@@ -1,16 +0,0 @@
-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

@@ -1,36 +0,0 @@
-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

@@ -1,22 +0,0 @@
-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

@@ -1,15 +0,0 @@
-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

@@ -1,14 +0,0 @@
-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/guardrails/__init__.pyi

@@ -1,6 +0,0 @@
-from aip_agents.guardrails.exceptions import GuardrailViolationError as GuardrailViolationError
-from aip_agents.guardrails.manager import GuardrailManager as GuardrailManager
-from aip_agents.guardrails.middleware import GuardrailMiddleware as GuardrailMiddleware
-from aip_agents.guardrails.schemas import BaseGuardrailEngineConfig as BaseGuardrailEngineConfig, GuardrailInput as GuardrailInput, GuardrailMode as GuardrailMode, GuardrailResult as GuardrailResult
-
-__all__ = ['GuardrailViolationError', 'GuardrailManager', 'GuardrailMiddleware', 'BaseGuardrailEngineConfig', 'GuardrailMode', 'GuardrailInput', 'GuardrailResult']

aip_agents/guardrails/engines/__init__.pyi

@@ -1,4 +0,0 @@
-from aip_agents.guardrails.engines.nemo import NemoGuardrailEngine as NemoGuardrailEngine
-from aip_agents.guardrails.engines.phrase_matcher import PhraseMatcherEngine as PhraseMatcherEngine
-
-__all__ = ['NemoGuardrailEngine', 'PhraseMatcherEngine']

aip_agents/guardrails/engines/base.pyi

@@ -1,61 +0,0 @@
-from _typeshed import Incomplete
-from abc import ABC, abstractmethod
-from aip_agents.guardrails.schemas import BaseGuardrailEngineConfig as BaseGuardrailEngineConfig, GuardrailResult as GuardrailResult
-from typing import Protocol
-
-class GuardrailEngine(Protocol):
-    """Protocol defining the interface for guardrail engines.
-
-    All guardrail engines must implement this protocol to be compatible
-    with GuardrailManager. Engines check content for safety violations.
-
-    Attributes:
-        config: Configuration for this engine's behavior
-    """
-    config: BaseGuardrailEngineConfig
-    @abstractmethod
-    async def check_input(self, content: str) -> GuardrailResult:
-        """Check user input content for safety violations.
-
-        Args:
-            content: The user input content to check
-
-        Returns:
-            GuardrailResult indicating if content is safe
-        """
-    @abstractmethod
-    async def check_output(self, content: str) -> GuardrailResult:
-        """Check AI output content for safety violations.
-
-        Args:
-            content: The AI output content to check
-
-        Returns:
-            GuardrailResult indicating if content is safe
-        """
-    @abstractmethod
-    def model_dump(self) -> dict:
-        """Serialize engine configuration into a JSON-compatible dictionary."""
-
-class BaseGuardrailEngine(ABC):
-    """Abstract base class for guardrail engines.
-
-    Provides common functionality and ensures proper configuration handling.
-    Concrete engines should inherit from this class.
-    """
-    config: Incomplete
-    def __init__(self, config: BaseGuardrailEngineConfig | None = None) -> None:
-        """Initialize the engine with configuration.
-
-        Args:
-            config: Engine configuration. Uses defaults if None provided.
-        """
-    @abstractmethod
-    async def check_input(self, content: str) -> GuardrailResult:
-        """Check user input content for safety violations."""
-    @abstractmethod
-    async def check_output(self, content: str) -> GuardrailResult:
-        """Check AI output content for safety violations."""
-    @abstractmethod
-    def model_dump(self) -> dict:
-        """Serialize engine configuration into a JSON-compatible dictionary."""