aip-agents-binary 0.5.25b8__py3-none-any.whl → 0.6.0__py3-none-any.whl

aip_agents/examples/hello_world_sentry.pyi

@@ -0,0 +1,21 @@
+from _typeshed import Incomplete
+from aip_agents.examples.hello_world_langgraph import langgraph_example as langgraph_example
+from aip_agents.sentry import setup_telemetry as setup_telemetry
+from aip_agents.utils.logger import get_logger as get_logger
+from fastapi import FastAPI
+
+logger: Incomplete
+BASE_URL: str
+SENTRY_ENVIRONMENT: Incomplete
+USE_OPENTELEMETRY: Incomplete
+
+def fetch_endpoints() -> None:
+    """Fetch all endpoints from the server."""
+def create_app() -> FastAPI:
+    """Create and configure the FastAPI application.
+
+    Returns:
+        FastAPI: The configured FastAPI application.
+    """
+def run_server() -> None:
+    """Run the FastAPI server."""

aip_agents/examples/hello_world_step_limits.pyi

@@ -0,0 +1,17 @@
+from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
+from aip_agents.examples.tools.langchain_arithmetic_tools import add_numbers as add_numbers
+from aip_agents.schema.step_limit import MaxStepsExceededError as MaxStepsExceededError, StepLimitConfig as StepLimitConfig
+from aip_agents.utils.step_limit_manager import StepLimitManager as StepLimitManager
+
+def print_config_validation(config: StepLimitConfig | None, context: str):
+    """Print step limit configuration validation."""
+async def test_case_1_success_no_limits() -> None:
+    """Test Case 1: Success - High limits, no warnings."""
+async def test_case_2_parallel_tool_batch_accounting() -> None:
+    """Test Case 2: Demonstrate that parallel tool batches consume multiple steps."""
+async def test_case_3_step_limit_exceeded() -> None:
+    """Test Case 3: Step Limit Exceeded - Only 2 steps allowed."""
+async def test_case_4_delegation_depth_exceeded() -> None:
+    """Test Case 4: Delegation Depth Exceeded - Zero depth."""
+async def main() -> None:
+    """Run all test cases."""

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

@@ -39,6 +39,15 @@ async def main():
             print(chunk["content"], end="", flush=True)
         if chunk.get("metadata"):
             print(f"\nMetadata: {chunk['metadata']}", end="\n\n", flush=True)
+        tool_info = chunk.get("metadata", {}).get("tool_info") if isinstance(chunk.get("metadata"), dict) else None
+        if isinstance(tool_info, dict):
+            for tool_call in tool_info.get("tool_calls", []):
+                if tool_call.get("name") == "data_visualizer":
+                    data_source = tool_call.get("args", {}).get("data_source")
+                    if not (isinstance(data_source, str) and data_source.startswith("$tool_output.")):
+                        raise RuntimeError(
+                            "Tool output sharing failed: expected data_source to reference $tool_output.<call_id>."
+                        )
     print("\n")
 
 

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.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.
+    """

aip_agents/examples/hitl_demo.pyi

@@ -0,0 +1,67 @@
+import asyncio
+import httpx
+from _typeshed import Incomplete
+from aip_agents.agent import LangGraphReactAgent as LangGraphReactAgent
+from aip_agents.schema.hitl import ApprovalRequest as ApprovalRequest
+from aip_agents.utils.env_loader import load_local_env as load_local_env
+from aip_agents.utils.logger import get_logger as get_logger
+from dataclasses import dataclass
+from langchain_core.tools import tool
+from starlette.applications import Starlette
+
+logger: Incomplete
+
+@tool
+def check_candidate_inbox(candidate_email: str) -> str:
+    """Retrieve the latest email from a candidate (safe tool).
+
+    Args:
+        candidate_email (str): The email address of the candidate.
+
+    Returns:
+        str: The latest email content from the candidate.
+    """
+@tool
+def validate_candidate(candidate_name: str, role: str, score: int) -> str:
+    """Record the candidate decision in the applicant tracking system.
+
+    Args:
+        candidate_name (str): The name of the candidate.
+        role (str): The role the candidate is being evaluated for.
+        score (int): The evaluation score for the candidate.
+
+    Returns:
+        str: The validation result with recommendation and notes.
+    """
+@tool
+def send_candidate_email(candidate_email: str, subject: str, body: str) -> str:
+    """Send an email update to the candidate.
+
+    Args:
+        candidate_email (str): The email address of the candidate.
+        subject (str): The subject line of the email.
+        body (str): The body content of the email.
+
+    Returns:
+        str: Confirmation message that the email was sent.
+    """
+
+CANDIDATE_PROFILES: dict[str, dict[str, str | int | bool]]
+CANDIDATE_SEQUENCE: list[dict[str, str | int | bool]]
+NAME_INDEX: Incomplete
+
+class _ServerContext:
+    def __init__(self, app: Starlette, host: str, port: int) -> None: ...
+    async def __aenter__(self) -> _ServerContext: ...
+    async def __aexit__(self, exc_type, exc, _tb) -> None: ...
+
+@dataclass
+class RunContext:
+    """Context object containing runtime dependencies for the HITL demo server."""
+    http_client: httpx.AsyncClient
+    host: str
+    port: int
+    pending_queue: asyncio.Queue[ApprovalRequest]
+
+async def main() -> None:
+    """Interactive HITL approval demo."""

aip_agents/examples/pii_demo_langgraph_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:
+    """Stream queries and inspect chunks with PII handling."""

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.py

@@ -1,10 +1,10 @@
 """A2A client for the planning LangGraphReactAgent.
 
 Run the planning server first:
-    uv run python -m aip_agents.examples.todolist_planning_a2a_langgraph_server
+    poetry run python -m aip_agents.examples.todolist_planning_a2a_langgraph_server
 
 Then run this client:
-    uv run python -m aip_agents.examples.todolist_planning_a2a_langchain_client
+    poetry run python -m aip_agents.examples.todolist_planning_a2a_langchain_client
 
 You should see streaming output, including when write_todos_tool is called.
 """

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.py

@@ -1,7 +1,7 @@
 """A2A server exposing a LangGraphReactAgent with planning (TodoListMiddleware).
 
 Run:
-    uv run python -m aip_agents.examples.todolist_planning_a2a_langgraph_server \
+    poetry run python -m aip_agents.examples.todolist_planning_a2a_langgraph_server \
         --host localhost --port 8002
 
 Then connect with the matching A2A client to observe write_todos_tool calls.

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]