aip-agents-binary 0.5.23__py3-none-any.whl → 0.5.24__py3-none-any.whl

aip_agents/tools/constants.pyi

@@ -1,138 +0,0 @@
-from _typeshed import Incomplete
-from enum import Enum, StrEnum
-
-GL_CONNECTORS_BASE_URL: Incomplete
-GL_CONNECTORS_API_KEY: Incomplete
-GL_CONNECTORS_FETCH_MAX_RETRIES: Incomplete
-BOSA_API_BASE_URL = GL_CONNECTORS_BASE_URL
-BOSA_API_KEY = GL_CONNECTORS_API_KEY
-BOSA_FETCH_MAX_RETRIES = GL_CONNECTORS_FETCH_MAX_RETRIES
-
-class ToolType(StrEnum):
-    """Tool types for GL Connectors."""
-    GLLM: str
-    LANGCHAIN: str
-
-class Action(Enum):
-    """Actions for GL Connectors."""
-    GITHUB: str
-    GOOGLE: str
-    GOOGLE_DRIVE: str
-    GOOGLE_MAIL: str
-    TWITTER: str
-
-class GitHubEndpoint(Enum):
-    """GitHub endpoints for GL Connectors."""
-    INTEGRATIONS: str
-    USER_HAS_INTEGRATION: str
-    SUCCESS_AUTHORIZE_CALLBACK: str
-    CREATE_ISSUE_HANDLER: str
-    GET_ISSUE_HANDLER: str
-    LIST_ISSUES: str
-    LIST_ISSUES_COMMENTS: str
-    SEARCH_ALL_ISSUES: str
-    GET_COMMITS: str
-    SEARCH_COMMITS: str
-    GET_COLLABORATORS: str
-    GET_RELEASES: str
-    GET_CONTRIBUTORS: str
-    GET_LANGUAGES: str
-    SEARCH_CONTRIBUTIONS: str
-    LIST_PULL_REQUESTS: str
-    SEARCH_PR: str
-    GET_PULL: str
-    GET_ALL_CONTRIBUTOR_COMMIT_ACTIVITIES: str
-    GET_COMMIT_ACTIVITY: str
-    GET_WEEKLY_COMMIT_COUNTS: str
-    GET_USER_CONTRIBUTION_STATISTICS: str
-    LIST_PROJECT_ITEMS: str
-    LIST_PROJECTS: str
-
-class GoogleDriveEndpoint(Enum):
-    """Google Drive endpoints for GL Connectors."""
-    INTEGRATIONS: str
-    USER_HAS_INTEGRATION: str
-    SUCCESS_AUTHORIZE_CALLBACK: str
-    SEARCH_FILES: str
-    GET_FILE: str
-    CREATE_FILE: str
-    CREATE_FOLDER: str
-    UPDATE_FILE: str
-    UPDATE_FOLDER: str
-    COPY_FILE: str
-    DELETE_FILE: str
-    SUMMARIZE_FOLDER_FILES_BY_TYPE: str
-    SUMMARIZE_TOTAL_FILES_BY_TYPE: str
-    RECENT_FILES: str
-    CREATE_PERMISSION: str
-    LIST_PERMISSIONS: str
-    GET_PERMISSION: str
-    UPDATE_PERMISSION: str
-    DELETE_PERMISSION: str
-    DOWNLOAD_FILE: str
-
-class GoogleDocsEndpoint(Enum):
-    """Google Docs endpoints for GL Connectors."""
-    INTEGRATIONS: str
-    USER_HAS_INTEGRATION: str
-    SUCCESS_AUTHORIZE_CALLBACK: str
-    GET_DOCUMENT: str
-    LIST_DOCUMENTS: str
-    CREATE_DOCUMENT: str
-    UPDATE_DOCUMENT: str
-    COPY_CONTENT: str
-    UPDATE_DOCUMENT_MARKDOWN: str
-    LIST_COMMENTS: str
-    SUMMARIZE_COMMENTS: str
-
-class GoogleEndpoint(Enum):
-    """Google endpoints for GL Connectors."""
-    INTEGRATIONS: str
-    USER_HAS_INTEGRATION: str
-    SUCCESS_AUTHORIZE_CALLBACK: str
-    USERINFO: str
-
-class TwitterEndpoint(Enum):
-    """Twitter endpoints for GL Connectors."""
-    INTEGRATIONS: str
-    USER_HAS_INTEGRATION: str
-    SUCCESS_AUTHORIZE_CALLBACK: str
-    SEARCH: str
-    GET_TWEETS: str
-    GET_THREAD: str
-    GET_USERS: str
-
-class GoogleMailEndpoint(Enum):
-    """Google Mail endpoints for GL Connectors."""
-    INTEGRATIONS: str
-    USER_HAS_INTEGRATION: str
-    SUCCESS_AUTHORIZE_CALLBACK: str
-    CREATE_DRAFT: str
-    LIST_DRAFTS: str
-    SEND_DRAFT: str
-    GET_DRAFT: str
-    MODIFY_DRAFT: str
-    LIST_LABELS: str
-    LABEL_STATS: str
-    GET_LABEL_DETAILS: str
-    CREATE_LABELS: str
-    MODIFY_LABELS: str
-    DELETE_LABELS: str
-    SEND_EMAIL: str
-    LIST_EMAILS: str
-    GET_EMAIL_DETAILS: str
-    MODIFY_EMAIL: str
-    DELETE_EMAIL: str
-    TRASH_EMAIL: str
-    UNTRASH_EMAIL: str
-    LIST_THREADS: str
-    THREAD_DETAILS: str
-    MODIFY_THREAD: str
-    GET_AUTO_REPLY: str
-    SET_AUTO_REPLY: str
-    GET_ATTACHMENT: str
-    USERINFO: str
-
-class ActionEndpointMap:
-    """Maps Action enums to their corresponding Endpoint enums."""
-    MAP: dict[Action, type[Enum]]

aip_agents/tools/document_loader/__init__.pyi

@@ -1,7 +0,0 @@
-from aip_agents.tools.document_loader.base_reader import BaseDocumentReaderTool as BaseDocumentReaderTool, DocumentReaderInput as DocumentReaderInput
-from aip_agents.tools.document_loader.docx_reader_tool import DocxReaderTool as DocxReaderTool
-from aip_agents.tools.document_loader.excel_reader_tool import ExcelReaderTool as ExcelReaderTool
-from aip_agents.tools.document_loader.pdf_reader_tool import PDFReaderTool as PDFReaderTool
-from aip_agents.tools.document_loader.pdf_splitter import PDFSplitter as PDFSplitter
-
-__all__ = ['BaseDocumentReaderTool', 'DocumentReaderInput', 'PDFReaderTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFSplitter']

aip_agents/tools/document_loader/base_reader.pyi

@@ -1,75 +0,0 @@
-from _typeshed import Incomplete
-from abc import ABC
-from aip_agents.utils.logger import get_logger as get_logger
-from gllm_docproc.loader.pipeline_loader import PipelineLoader as PipelineLoaderType
-from langchain_core.tools import BaseTool
-from pydantic import BaseModel
-from typing import Any
-
-logger: Incomplete
-DOCPROC_MISSING_MESSAGE: str
-PipelineLoaderType = Any
-
-class _MissingDocprocLoader:
-    """Fallback loader that errors when document processing is attempted."""
-    loaders: list[object]
-    def __init__(self) -> None: ...
-    def add_loader(self, loader: object) -> None: ...
-    def load(self, *_args: object, **_kwargs: object) -> list[dict[str, str]]: ...
-    def clear_cache(self) -> None: ...
-
-DOCPROC_AVAILABLE: Incomplete
-
-class BaseDocumentConfig(BaseModel):
-    """Base tool configuration schema for document processing with batching functionality.
-
-    This configuration enables page-by-page batching to optimize memory usage when
-    processing large document files. When batching is enabled, documents are processed
-    sequentially by pages rather than loading the entire document into memory at once.
-
-    Attributes:
-        batching (bool): Enable page-by-page batching to reduce memory usage.
-            When True, documents are processed page by page sequentially.
-            When False, maintains current behavior of loading entire document.
-            Defaults to False for backward compatibility.
-        batch_size (int): Number of pages to process in each batch.
-            Must be between 1 and 100 pages inclusive.
-            Larger batch sizes may use more memory but could be more efficient.
-            Smaller batch sizes use less memory but may have more overhead.
-            Defaults to 10 for balanced memory usage and efficiency.
-
-    Examples:
-        >>> # Default configuration (no batching)
-        >>> config = BaseDocumentConfig()
-        >>> print(config.batching)  # False
-        >>> print(config.batch_size)  # 10
-
-        >>> # Enable batching with single page processing
-        >>> config = BaseDocumentConfig(batching=True, batch_size=1)
-
-        >>> # Enable batching with multi-page batches
-        >>> config = BaseDocumentConfig(batching=True, batch_size=3)
-    """
-    batching: bool
-    batch_size: int
-
-class DocumentReaderInput(BaseModel):
-    """Input schema for the DocumentReader tool."""
-    file_path: str
-
-class BaseDocumentReaderTool(BaseTool, ABC):
-    """Base tool to read and extract text from document files."""
-    name: str
-    description: str
-    args_schema: type[BaseModel]
-    tool_config_schema: type[BaseModel]
-    loader: PipelineLoaderType
-    def __init__(self) -> None:
-        """Initialize the base document reader tool."""
-    def cleanup_memory(self) -> None:
-        """Explicitly clean up memory and force garbage collection.
-
-        This method can be called after processing to minimize memory usage.
-        While it won't reset memory to exactly 0, it will free up as much
-        memory as possible by clearing internal caches and forcing garbage collection.
-        """

aip_agents/tools/document_loader/docx_reader_tool.pyi

@@ -1,10 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.tools.document_loader.base_reader import BaseDocumentReaderTool as BaseDocumentReaderTool, DOCPROC_MISSING_MESSAGE as DOCPROC_MISSING_MESSAGE
-from aip_agents.utils.logger import get_logger as get_logger
-
-logger: Incomplete
-
-class DocxReaderTool(BaseDocumentReaderTool):
-    """Tool to read and extract text from Word documents."""
-    name: str
-    description: str

aip_agents/tools/document_loader/excel_reader_tool.pyi

@@ -1,26 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.tools.document_loader.base_reader import BaseDocumentReaderTool as BaseDocumentReaderTool, DOCPROC_MISSING_MESSAGE as DOCPROC_MISSING_MESSAGE
-from aip_agents.utils.logger import get_logger as get_logger
-
-logger: Incomplete
-
-class ExcelReaderTool(BaseDocumentReaderTool):
-    '''Tool to read and extract content from Excel files.
-
-    This tool reads Excel files (.xlsx, .xlsm) and extracts their content using
-    the gllm_docproc loader pipeline. The content is formatted as Markdown tables
-    for easy readability.
-
-    Features:
-    - Supports .xlsx and .xlsm formats
-    - Extracts all sheets or specific sheets
-    - Formats output as Markdown tables
-    - Configurable row limits and file size limits
-
-    Examples:
-        >>> tool = ExcelReaderTool()
-        >>> result = tool._run("/tmp/data.xlsx")
-        >>> print(result)
-    '''
-    name: str
-    description: str

aip_agents/tools/document_loader/pdf_reader_tool.pyi

@@ -1,11 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.tools.document_loader.base_reader import BaseDocumentReaderTool as BaseDocumentReaderTool, DOCPROC_MISSING_MESSAGE as DOCPROC_MISSING_MESSAGE
-from aip_agents.tools.document_loader.pdf_splitter import PDFSplitter as PDFSplitter
-from aip_agents.utils.logger import get_logger as get_logger
-
-logger: Incomplete
-
-class PDFReaderTool(BaseDocumentReaderTool):
-    """Tool to read and extract text from PDF files."""
-    name: str
-    description: str

aip_agents/tools/document_loader/pdf_splitter.pyi

@@ -1,18 +0,0 @@
-class PDFSplitter:
-    """Utility class for splitting PDF files into page-based temporary files."""
-    @staticmethod
-    def split_by_pages(file_path: str, batch_size: int) -> list[str]:
-        """Split PDF into temporary files containing batch_size pages each.
-
-        Args:
-            file_path: Path to the PDF file to split
-            batch_size: Number of pages to include in each batch (1-10)
-
-        Returns:
-            List of temporary file paths containing the split PDF batches
-
-        Raises:
-            FileNotFoundError: If the input PDF file doesn't exist
-            ValueError: If batch_size is invalid or PDF is corrupted
-            Exception: For other PDF processing errors
-        """

aip_agents/tools/gl_connector/__init__.pyi

@@ -1,3 +0,0 @@
-from aip_agents.tools.gl_connector.tool import GLConnectorTool as GLConnectorTool
-
-__all__ = ['GLConnectorTool']

aip_agents/tools/gl_connector/tool.pyi

@@ -1,74 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.tools.constants import ToolType as ToolType
-from langchain_core.runnables import RunnableConfig
-from langchain_core.tools import BaseTool
-from typing import Any
-
-class _InjectedTool(BaseTool):
-    """Wrap a BaseTool to inject token and optional identifier into inputs."""
-    model_config: Incomplete
-    def __init__(self, base_tool: BaseTool, token: str, identifier: str | None) -> None:
-        """Initialize the injected tool wrapper.
-
-        Args:
-            base_tool: The base tool to wrap.
-            token: Authentication token to inject into tool inputs.
-            identifier: Optional identifier to inject into tool inputs.
-
-        Returns:
-            None
-        """
-    def invoke(self, input: Any, config: RunnableConfig | None = None, **kwargs: Any) -> Any:
-        """Invoke the tool with token and optional identifier injected.
-
-        Args:
-            input: Tool input to process.
-            config: Optional runnable configuration.
-            **kwargs: Additional keyword arguments.
-
-        Returns:
-            The result of invoking the tool with injected parameters.
-        """
-    async def ainvoke(self, input: Any, config: RunnableConfig | None = None, **kwargs: Any) -> Any:
-        """Invoke the tool asynchronously with token and optional identifier injected.
-
-        Args:
-            input: Tool input to process.
-            config: Optional runnable configuration.
-            **kwargs: Additional keyword arguments.
-
-        Returns:
-            The result of invoking the tool with injected parameters.
-        """
-    def run(self, tool_input: Any, **kwargs: Any) -> Any:
-        """Run the tool with token and optional identifier injected.
-
-        Args:
-            tool_input: Tool input to process.
-            **kwargs: Additional keyword arguments.
-
-        Returns:
-            The result of running the tool with injected parameters.
-        """
-    async def arun(self, tool_input: Any, **kwargs: Any) -> Any:
-        """Run the tool asynchronously with token and optional identifier injected.
-
-        Args:
-            tool_input: Tool input to process.
-            **kwargs: Additional keyword arguments.
-
-        Returns:
-            The result of running the tool with injected parameters.
-        """
-
-def GLConnectorTool(tool_name: str, *, api_key: str | None = None, identifier: str | None = None) -> BaseTool:
-    """Create a single tool from GL Connectors by exact tool name.
-
-    Args:
-        tool_name: Exact tool name (not module name).
-        api_key: Optional override for GL Connectors API key.
-        identifier: Optional override for GL Connectors identifier.
-
-    Returns:
-        A single LangChain BaseTool with token injection.
-    """

aip_agents/tools/gl_connector_tools.pyi

@@ -1,39 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.tools.constants import GL_CONNECTORS_API_KEY as GL_CONNECTORS_API_KEY, GL_CONNECTORS_BASE_URL as GL_CONNECTORS_BASE_URL, GL_CONNECTORS_FETCH_MAX_RETRIES as GL_CONNECTORS_FETCH_MAX_RETRIES, ToolType as ToolType
-from aip_agents.utils.logger import get_logger as get_logger
-from langchain_core.tools import BaseTool as BaseTool
-
-logger: Incomplete
-
-def get_gl_connector_modules_with_retry() -> list[str]:
-    """Try to get available modules with retries.
-
-    Returns:
-        List of available modules.
-    """
-
-class LazyGLConnectorToolsDict(dict):
-    """Lazy dictionary for GL Connectors."""
-    def __missing__(self, app):
-        """When a key is missing, create the tools and store them in the dictionary.
-
-        Args:
-            app: Name of the GL Connectors.
-
-        Returns:
-            List of tools generated by the GL Connectors tool generator.
-        """
-
-def get_gl_connector_modules() -> list[str]:
-    """Lazily fetch and cache GL Connectors modules.
-
-    Returns:
-        List of GL Connectors modules.
-    """
-def get_bosa_modules() -> list[str]:
-    """Backward-compatible alias for get_gl_connector_modules."""
-
-GL_CONNECTORS_MODULES: Incomplete
-BOSA_MODULES = GL_CONNECTORS_MODULES
-GL_CONNECTORS_AUTOMATED_TOOLS: dict[str, list[BaseTool]]
-BOSA_AUTOMATED_TOOLS = GL_CONNECTORS_AUTOMATED_TOOLS

aip_agents/tools/memory_search/__init__.pyi

@@ -1,5 +0,0 @@
-from aip_agents.tools.memory_search.base import LongTermMemorySearchTool as LongTermMemorySearchTool
-from aip_agents.tools.memory_search.mem0 import MEMORY_SEARCH_TOOL_NAME as MEMORY_SEARCH_TOOL_NAME, Mem0SearchInput as Mem0SearchInput, Mem0SearchTool as Mem0SearchTool
-from aip_agents.tools.memory_search.schema import LongTermMemorySearchInput as LongTermMemorySearchInput, MemoryConfig as MemoryConfig
-
-__all__ = ['MemoryConfig', 'LongTermMemorySearchInput', 'LongTermMemorySearchTool', 'Mem0SearchInput', 'Mem0SearchTool', 'MEMORY_SEARCH_TOOL_NAME']

aip_agents/tools/memory_search/base.pyi

@@ -1,69 +0,0 @@
-from _typeshed import Incomplete
-from abc import ABC
-from aip_agents.memory.constants import MemoryDefaults as MemoryDefaults
-from aip_agents.tools.memory_search.schema import LongTermMemorySearchInput as LongTermMemorySearchInput, MemoryConfig as MemoryConfig
-from aip_agents.utils.logger import get_logger as get_logger
-from collections.abc import Callable
-from langchain_core.tools import BaseTool
-from pydantic import BaseModel as BaseModel
-from typing import Any, ClassVar, Protocol
-
-logger: Incomplete
-
-class LongTermMemoryBackend(Protocol):
-    """Protocol for memory adapters that support retrieval and formatting."""
-    def retrieve(self, *, query: str | None, user_id: str, limit: int | None = None, filters: dict[str, Any] | None = None, page: int | None = None) -> list[dict[str, Any]]:
-        """Retrieve memories matching the given query and filters.
-
-        Args:
-            query: Optional search query string.
-            user_id: User identifier for scoped retrieval.
-            limit: Maximum number of results to return.
-            filters: Optional filters to apply to the retrieval.
-            page: Page number for pagination.
-
-        Returns:
-            List of memory hit dictionaries.
-        """
-    def format_hits(self, hits: list[dict[str, Any]], max_items: int = ..., with_tag: bool = True) -> str:
-        """Format memory hits into a readable string.
-
-        Args:
-            hits: List of memory hit dictionaries to format.
-            max_items: Maximum number of hits to include in output.
-            with_tag: Whether to wrap output with memory tags.
-
-        Returns:
-            Formatted string representation of memory hits.
-        """
-
-class LongTermMemorySearchTool(BaseTool, ABC):
-    """Abstract base class for provider-specific long-term memory search tools."""
-    name: str
-    description: str
-    args_schema: type[LongTermMemorySearchInput]
-    tool_config_schema: type[BaseModel]
-    memory: LongTermMemoryBackend
-    default_user_id: str | None
-    user_id_provider: Callable[[], str | None] | None
-    MINIMUM_MEMORY_RETRIEVAL: ClassVar[int]
-    LOG_PREFIX: ClassVar[str]
-    def __init__(self, memory: LongTermMemoryBackend, *, default_user_id: str | None = None, user_id_provider: Callable[[], str | None] | None = None, **kwargs: Any) -> None:
-        """Initialize the long-term memory search tool.
-
-        Args:
-            memory: Memory adapter instance with retrieve() and format_hits() methods.
-            default_user_id: Default user ID to use if not provided in metadata.
-            user_id_provider: Callable that returns a user ID.
-            **kwargs: Additional keyword arguments passed to the parent class.
-        """
-    def format_hits(self, hits: list[dict[str, Any]], with_tag: bool = False) -> str:
-        """Format hits into a string with optional tags.
-
-        Args:
-            hits: List of memory hit dictionaries to format.
-            with_tag: Whether to wrap the output with memory tags.
-
-        Returns:
-            Formatted string representation of the memory hits.
-        """

aip_agents/tools/memory_search/mem0.pyi

@@ -1,19 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.memory.constants import MemoryDefaults as MemoryDefaults
-from aip_agents.tools.memory_search.base import LongTermMemorySearchTool as LongTermMemorySearchTool
-from aip_agents.tools.memory_search.schema import LongTermMemorySearchInput as LongTermMemorySearchInput
-from aip_agents.utils.datetime import is_valid_date_string as is_valid_date_string, next_day_iso as next_day_iso
-from aip_agents.utils.logger import get_logger as get_logger
-from typing import ClassVar
-
-logger: Incomplete
-MEMORY_SEARCH_TOOL_NAME: str
-
-class Mem0SearchTool(LongTermMemorySearchTool):
-    """Mem0-specific implementation of the long-term memory search tool."""
-    name: str
-    description: str
-    args_schema: type[LongTermMemorySearchInput]
-    LOG_PREFIX: ClassVar[str]
-    METADATA_FILTER_BLOCKLIST: ClassVar[set[str]]
-Mem0SearchInput = LongTermMemorySearchInput

aip_agents/tools/memory_search/schema.pyi

@@ -1,15 +0,0 @@
-from pydantic import BaseModel
-from typing import Any
-
-class MemoryConfig(BaseModel):
-    """Tool configuration schema for memory operations."""
-    user_id: str
-
-class LongTermMemorySearchInput(BaseModel):
-    """Input schema for unified long-term memory retrieval."""
-    query: str | None
-    start_date: str | None
-    end_date: str | None
-    limit: int | None
-    categories: list[str] | None
-    metadata: dict[str, Any] | None

aip_agents/tools/memory_search_tool.pyi

@@ -1,3 +0,0 @@
-from aip_agents.tools.memory_search import LongTermMemorySearchInput as LongTermMemorySearchInput, LongTermMemorySearchTool as LongTermMemorySearchTool, MEMORY_SEARCH_TOOL_NAME as MEMORY_SEARCH_TOOL_NAME, Mem0SearchInput as Mem0SearchInput, Mem0SearchTool as Mem0SearchTool, MemoryConfig as MemoryConfig
-
-__all__ = ['MemoryConfig', 'LongTermMemorySearchInput', 'LongTermMemorySearchTool', 'Mem0SearchInput', 'Mem0SearchTool', 'MEMORY_SEARCH_TOOL_NAME']

aip_agents/tools/time_tool.pyi

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

aip_agents/tools/tool_config_injector.pyi

@@ -1,26 +0,0 @@
-from langchain_core.tools import BaseTool
-from pydantic import BaseModel
-from typing import TypeVar
-
-__all__ = ['inject_config_methods_into_tool', 'CONFIG_SCHEMA_ATTR', 'CONFIG_ATTR', 'TOOL_CONFIG_SCHEMA_ATTR', 'TOOL_CONFIGS_KEY']
-
-ConfigSchema = TypeVar('ConfigSchema', bound=BaseModel)
-CONFIG_SCHEMA_ATTR: str
-CONFIG_ATTR: str
-TOOL_CONFIG_SCHEMA_ATTR: str
-TOOL_CONFIGS_KEY: str
-
-def inject_config_methods_into_tool(tool: BaseTool, config_schema: type[ConfigSchema]) -> None:
-    """Inject configuration methods into a tool.
-
-    This function is used by the SDK to automatically inject configuration capabilities
-    into tools that have a tool_config_schema attribute. It can also be used directly
-    by developers who want to add configuration support to their tools.
-
-    Args:
-        tool: The tool to inject configuration capabilities into.
-        config_schema: Pydantic model class for configuration validation.
-
-    Raises:
-        ValueError: If tool is not a BaseTool instance or config_schema is not a Pydantic BaseModel subclass.
-    """

aip_agents/tools/web_search/__init__.pyi

@@ -1,3 +0,0 @@
-from aip_agents.tools.web_search.serper_tool import GoogleSerperTool as GoogleSerperTool
-
-__all__ = ['GoogleSerperTool']

aip_agents/tools/web_search/serper_tool.pyi

@@ -1,19 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.utils.logger import get_logger as get_logger
-from langchain_community.utilities.google_serper import GoogleSerperAPIWrapper
-from langchain_core.tools import BaseTool
-from pydantic import BaseModel
-
-logger: Incomplete
-
-class GoogleSerperInput(BaseModel):
-    """Input schema for the GoogleSerperTool."""
-    query: str
-
-class GoogleSerperTool(BaseTool):
-    """Tool to search Google Serper API."""
-    name: str
-    description: str
-    save_output_history: bool
-    args_schema: type[BaseModel]
-    api_wrapper: GoogleSerperAPIWrapper

aip_agents/types/__init__.pyi

@@ -1,36 +0,0 @@
-from aip_agents.schema.a2a import A2AEvent as A2AEvent, A2AStreamEventType as A2AStreamEventType, ToolCallInfo as ToolCallInfo, ToolResultInfo as ToolResultInfo
-from pydantic import BaseModel
-from typing import Any, Protocol
-
-__all__ = ['A2AEvent', 'A2AStreamEventType', 'ToolCallInfo', 'ToolResultInfo', 'AgentProtocol', 'ChatMessage']
-
-class ChatMessage(BaseModel):
-    """Represents a single message in a chat conversation."""
-    role: str
-    content: str
-
-class AgentProtocol(Protocol):
-    """Defines the expected interface for an agent."""
-    id: str
-    name: str
-    description: str
-    def run(self, query: str, **kwargs: Any) -> dict[str, Any]:
-        """Synchronous execution method.
-
-        Args:
-            query (str): The query to execute.
-            **kwargs (Any): Additional keyword arguments.
-
-        Returns:
-            dict[str, Any]: The execution result.
-        """
-    async def arun(self, query: str, **kwargs: Any) -> dict[str, Any]:
-        """Asynchronous execution method.
-
-        Args:
-            query (str): The query to execute.
-            **kwargs (Any): Additional keyword arguments.
-
-        Returns:
-            dict[str, Any]: The execution result.
-        """

aip_agents/types/a2a_events.pyi

@@ -1,3 +0,0 @@
-from aip_agents.schema.a2a import A2AEvent as A2AEvent, A2AStreamEventType as A2AStreamEventType, ToolCallInfo as ToolCallInfo, ToolResultInfo as ToolResultInfo
-
-__all__ = ['A2AStreamEventType', 'A2AEvent', 'ToolCallInfo', 'ToolResultInfo']

aip_agents/utils/__init__.pyi

@@ -1,11 +0,0 @@
-from aip_agents.utils.artifact_helpers import create_artifact_command as create_artifact_command, create_artifact_response as create_artifact_response, create_error_response as create_error_response, create_multiple_artifacts_command as create_multiple_artifacts_command, create_text_artifact_response as create_text_artifact_response
-from aip_agents.utils.file_prompt_utils import augment_query_with_file_paths as augment_query_with_file_paths
-from aip_agents.utils.final_response_builder import assemble_final_response as assemble_final_response
-from aip_agents.utils.logger import LoggerManager as LoggerManager, get_logger as get_logger
-from aip_agents.utils.pii.pii_handler import ToolPIIHandler as ToolPIIHandler
-from aip_agents.utils.pii.pii_helper import add_pii_mappings as add_pii_mappings, extract_pii_mapping_from_agent_response as extract_pii_mapping_from_agent_response
-from aip_agents.utils.reference_helper import add_references_chunks as add_references_chunks, serialize_references_for_metadata as serialize_references_for_metadata, validate_references as validate_references
-from aip_agents.utils.sse_chunk_transformer import SSEChunkTransformer as SSEChunkTransformer
-from aip_agents.utils.step_limit_manager import StepExecutionContext as StepExecutionContext, StepLimitManager as StepLimitManager, _DELEGATION_CHAIN_CVAR as _DELEGATION_CHAIN_CVAR, _DELEGATION_DEPTH_CVAR as _DELEGATION_DEPTH_CVAR, _REMAINING_STEP_BUDGET_CVAR as _REMAINING_STEP_BUDGET_CVAR, _STEP_LIMIT_CONFIG_CVAR as _STEP_LIMIT_CONFIG_CVAR
-
-__all__ = ['get_logger', 'LoggerManager', 'create_artifact_response', 'create_error_response', 'create_artifact_command', 'create_multiple_artifacts_command', 'create_text_artifact_response', 'validate_references', 'serialize_references_for_metadata', 'add_references_chunks', 'assemble_final_response', 'augment_query_with_file_paths', 'SSEChunkTransformer', 'StepExecutionContext', 'StepLimitManager', '_DELEGATION_CHAIN_CVAR', '_DELEGATION_DEPTH_CVAR', '_REMAINING_STEP_BUDGET_CVAR', '_STEP_LIMIT_CONFIG_CVAR', 'ToolPIIHandler', 'add_pii_mappings', 'extract_pii_mapping_from_agent_response']