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

aip_agents/schema/step_limit.pyi

@@ -1,63 +0,0 @@
-from _typeshed import Incomplete
-from dataclasses import dataclass, field
-from enum import StrEnum
-
-class StepLimitErrorType(StrEnum):
-    """Types of step-related limit violations."""
-    STEP_LIMIT_EXCEEDED: str
-    DELEGATION_DEPTH_EXCEEDED: str
-
-@dataclass
-class StepLimitConfig:
-    """Configuration for agent step and delegation limits.
-
-    Attributes:
-        max_steps: Maximum number of execution steps allowed per invocation.
-                   Includes agent node (LLM call) and every tool call (parallel batches count per call).
-        max_delegation_depth: Maximum depth of delegation chain allowed.
-                              Depth 0 means no delegation allowed.
-    """
-    max_steps: int = field(default_factory=Incomplete)
-    max_delegation_depth: int = field(default_factory=Incomplete)
-    def __post_init__(self) -> None:
-        """Validate configuration values and normalize range."""
-
-@dataclass
-class StepLimitErrorResponse:
-    """Structured error response for step limit violations.
-
-    Attributes:
-        error_type: The type of limit that was exceeded.
-        agent_name: Name of the agent that hit the limit.
-        current_value: Current step count or delegation depth.
-        configured_limit: The configured limit that was exceeded.
-        message: Human-readable error message.
-        delegation_chain: Full chain for delegation errors.
-        partial_result: Any output generated before hitting the limit.
-    """
-    error_type: StepLimitErrorType
-    agent_name: str
-    current_value: int
-    configured_limit: int
-    message: str
-    delegation_chain: list[str] | None = ...
-    partial_result: str | None = ...
-
-class StepLimitError(Exception):
-    """Base exception for step and delegation limit violations.
-
-    Attributes:
-        error_response: Structured error response with details.
-    """
-    error_response: Incomplete
-    def __init__(self, error_response: StepLimitErrorResponse) -> None:
-        """Initialize with error response.
-
-        Args:
-            error_response: Structured error details.
-        """
-
-class MaxStepsExceededError(StepLimitError):
-    """Raised when agent exceeds configured max_steps limit."""
-class MaxDelegationDepthExceededError(StepLimitError):
-    """Raised when delegation would exceed max_delegation_depth limit."""

aip_agents/schema/storage.pyi

@@ -1,21 +0,0 @@
-from dataclasses import dataclass
-from enum import StrEnum
-
-__all__ = ['OBJECT_STORAGE_PREFIX', 'StorageType', 'StorageConfig']
-
-OBJECT_STORAGE_PREFIX: str
-
-class StorageType(StrEnum):
-    """Supported storage types."""
-    MEMORY: str
-    OBJECT_STORAGE: str
-
-@dataclass
-class StorageConfig:
-    """Configuration for storage providers."""
-    storage_type: StorageType = ...
-    object_prefix: str = ...
-    object_use_json: bool = ...
-    @classmethod
-    def from_env(cls) -> StorageConfig:
-        """Create StorageConfig from environment variables."""

aip_agents/sentry/__init__.pyi

@@ -1,3 +0,0 @@
-from aip_agents.sentry.sentry import setup_telemetry as setup_telemetry
-
-__all__ = ['setup_telemetry']

aip_agents/sentry/sentry.pyi

@@ -1,48 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.agent import BaseAgent as BaseAgent, GoogleADKAgent as GoogleADKAgent, LangChainAgent as LangChainAgent, LangGraphAgent as LangGraphAgent
-from aip_agents.utils.logger import get_logger as get_logger
-from fastapi import FastAPI
-
-logger: Incomplete
-SENTRY_DSN: Incomplete
-SENTRY_ENVIRONMENT: Incomplete
-SENTRY_PROJECT: Incomplete
-VERSION_NUMBER: Incomplete
-BUILD_NUMBER: Incomplete
-USE_OPENTELEMETRY: Incomplete
-CLASSES_TO_INSTRUMENT: Incomplete
-
-def get_all_methods(cls) -> list:
-    """Get all methods from a class.
-
-    Args:
-        cls: The class to get methods from.
-
-    Returns:
-        list: A list of methods.
-    """
-def instrument_gl_functions() -> None:
-    """Instrument GL functions."""
-def traces_sampler(*args) -> float:
-    """Determine appropriate sampling rate for Sentry transactions.
-
-    Args:
-        *args: Additional positional arguments
-
-    Returns:
-        float: Sampling rate between 0 and 1
-    """
-def setup_sentry_with_open_telemetry(app: FastAPI) -> None:
-    """Configure telemetry with both Sentry and OpenTelemetry.
-
-    Args:
-        app: FastAPI application instance
-    """
-def setup_sentry_only() -> None:
-    """Configure telemetry with Sentry only (no OpenTelemetry)."""
-def setup_telemetry(app: FastAPI) -> None:
-    """Configure and initialize telemetry based on configuration.
-
-    Args:
-        app: FastAPI application instance
-    """

aip_agents/storage/__init__.pyi

@@ -1,8 +0,0 @@
-from aip_agents.storage.base import BaseObjectStorageClient as BaseObjectStorageClient
-from aip_agents.storage.clients.minio_client import MinioConfig as MinioConfig, MinioObjectStorage as MinioObjectStorage
-from aip_agents.storage.config import StorageConfig as StorageConfig, StorageProviderFactory as StorageProviderFactory, StorageType as StorageType
-from aip_agents.storage.providers.base import BaseStorageProvider as BaseStorageProvider, StorageError as StorageError
-from aip_agents.storage.providers.memory import InMemoryStorageProvider as InMemoryStorageProvider
-from aip_agents.storage.providers.object_storage import ObjectStorageProvider as ObjectStorageProvider
-
-__all__ = ['BaseObjectStorageClient', 'MinioConfig', 'MinioObjectStorage', 'BaseStorageProvider', 'StorageError', 'InMemoryStorageProvider', 'ObjectStorageProvider', 'StorageConfig', 'StorageType', 'StorageProviderFactory']

aip_agents/storage/base.pyi

@@ -1,58 +0,0 @@
-from abc import ABC, abstractmethod
-from typing import BinaryIO
-
-class BaseObjectStorageClient(ABC):
-    """Abstract base class for object storage clients."""
-    @abstractmethod
-    def object_exists(self, object_key: str) -> bool:
-        """Check if an object exists in the storage.
-
-        Args:
-            object_key: The key of the object to check.
-
-        Returns:
-            True if the object exists, False otherwise.
-        """
-    @abstractmethod
-    def upload(self, object_key: str, file_stream: bytes | BinaryIO, filename: str | None = None, content_type: str | None = None, metadata: dict[str, str] | None = None) -> str:
-        """Upload data to object storage.
-
-        Args:
-            object_key: The key of the object in the storage
-            file_stream: The binary data to upload
-            filename: The name of the file
-            content_type: The content type of the file
-            metadata: Additional metadata to store with the object
-
-        Returns:
-            The key of the uploaded object
-        """
-    @abstractmethod
-    def get(self, object_key: str) -> bytes:
-        """Get data from object storage.
-
-        Args:
-            object_key: The key of the object in the storage
-
-        Returns:
-            The binary data of the object
-        """
-    @abstractmethod
-    def delete(self, object_key: str) -> None:
-        """Delete data from object storage.
-
-        Args:
-            object_key: The key of the object in the storage
-        """
-    @abstractmethod
-    def generate_presigned_url(self, object_key: str, expires: int = 24, response_headers: dict[str, str] | None = None) -> str:
-        """Generate a presigned URL for accessing the object.
-
-        Args:
-            object_key: The key of the object in the storage
-            expires: The number of hours the URL is valid for
-            response_headers: Additional headers to include in the response
-
-        Returns:
-            The presigned URL
-        """

aip_agents/storage/clients/__init__.pyi

@@ -1,3 +0,0 @@
-from aip_agents.storage.clients.minio_client import MinioConfig as MinioConfig, MinioObjectStorage as MinioObjectStorage
-
-__all__ = ['MinioConfig', 'MinioObjectStorage']

aip_agents/storage/clients/minio_client.pyi

@@ -1,137 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.storage.base import BaseObjectStorageClient as BaseObjectStorageClient
-from dataclasses import dataclass
-from typing import BinaryIO
-
-S3_ERR_NO_SUCH_KEY: str
-S3_ERR_NO_SUCH_BUCKET: str
-S3_ERR_ACCESS_DENIED: str
-
-@dataclass
-class MinioConfig:
-    """Configuration for MinIO object storage client.
-
-    Attributes:
-        endpoint: MinIO server endpoint URL
-        access_key: Access key for authentication
-        secret_key: Secret key for authentication
-        bucket: Bucket name to use for storage
-        secure: Whether to use HTTPS (defaults to True)
-    """
-    endpoint: str
-    access_key: str
-    secret_key: str
-    bucket: str
-    secure: bool = ...
-    @classmethod
-    def from_env(cls) -> MinioConfig:
-        """Create MinioConfig from environment variables.
-
-        Expected environment variables:
-        - OBJECT_STORAGE_URL
-        - OBJECT_STORAGE_USER
-        - OBJECT_STORAGE_PASSWORD
-        - OBJECT_STORAGE_BUCKET
-        - OBJECT_STORAGE_SECURE (optional, defaults to True)
-
-        Returns:
-            MinioConfig instance
-
-        Raises:
-            ValueError: If required environment variables are not set
-        """
-
-class MinioObjectStorage(BaseObjectStorageClient):
-    """Implementation of ObjectStorageInterface using Minio."""
-    config: Incomplete
-    client: Incomplete
-    bucket: Incomplete
-    def __init__(self, config: MinioConfig | None = None, ensure_bucket: bool = True) -> None:
-        """Initialize MinioObjectStorage with configuration.
-
-        Args:
-            config: MinioConfig instance. If None, will attempt to load from environment variables.
-            ensure_bucket: Whether to ensure bucket exists during initialization (optional). Defaults to True.
-
-        Raises:
-            ValueError: If configuration is invalid or incomplete
-        """
-    def upload(self, object_key: str, file_stream: bytes | BinaryIO, filename: str | None = None, content_type: str | None = None, metadata: dict[str, str] | None = None) -> str:
-        """Upload data to Minio object storage.
-
-        Args:
-            object_key: The key to store the object under
-            file_stream: The data to upload (bytes or file-like object)
-            filename: The filename of the data (optional)
-            content_type: The content type of the data (optional)
-            metadata: Additional metadata to store with the object (optional)
-
-        Returns:
-            The object key of the uploaded data
-
-        Raises:
-            ValueError: If the file stream is empty or invalid
-            ConnectionError: If there's an issue connecting to Minio
-            Exception: For other unexpected errors during upload
-        """
-    def get(self, object_key: str) -> bytes:
-        """Get data from Minio object storage.
-
-        Args:
-            object_key: The key of the object to retrieve
-
-        Returns:
-            The object data as bytes
-
-        Raises:
-            KeyError: If the object is not found
-            ConnectionError: If there's a network or connection issue
-        """
-    def delete(self, object_key: str) -> None:
-        """Delete data from Minio object storage.
-
-        Args:
-            object_key: The key of the object to delete
-
-        Raises:
-            ConnectionError: If there's a network or connection issue
-        """
-    def object_exists(self, object_key: str) -> bool:
-        """Check if an object exists in the MinIO bucket.
-
-        Args:
-            object_key: The key of the object to check.
-
-        Returns:
-            True if the object exists, False otherwise.
-
-        Raises:
-            ConnectionError: If there's a network or connection issue (excluding not found errors)
-        """
-    def list_objects(self, prefix: str = '') -> list[str]:
-        """List objects in the bucket with optional prefix filter.
-
-        Args:
-            prefix: Optional prefix to filter objects
-
-        Returns:
-            List of object keys
-
-        Raises:
-            ConnectionError: If there's an issue listing objects
-        """
-    def generate_presigned_url(self, object_key: str, expires: int = 24, response_headers: dict[str, str] | None = None) -> str:
-        """Generate a presigned URL for accessing the object.
-
-        Args:
-            object_key: The key of the object
-            expires: Expiration time in hours (defaults to 24)
-            response_headers: Additional response headers (optional)
-
-        Returns:
-            A presigned URL
-
-        Raises:
-            ValueError: If expiration time is not positive
-            ConnectionError: If there's an issue generating the URL
-        """

aip_agents/storage/config.pyi

@@ -1,29 +0,0 @@
-from aip_agents.schema.storage import OBJECT_STORAGE_PREFIX as OBJECT_STORAGE_PREFIX, StorageConfig as StorageConfig, StorageType as StorageType
-from aip_agents.storage.base import BaseObjectStorageClient
-from aip_agents.storage.providers.base import BaseStorageProvider
-
-__all__ = ['OBJECT_STORAGE_PREFIX', 'StorageConfig', 'StorageType', 'StorageProviderFactory']
-
-class StorageProviderFactory:
-    """Factory for creating storage providers based on configuration."""
-    @staticmethod
-    def create(config: StorageConfig, object_storage_client: BaseObjectStorageClient | None = None) -> BaseStorageProvider:
-        """Create storage provider based on configuration.
-
-        Args:
-            config (StorageConfig): Storage configuration object.
-            object_storage_client (BaseObjectStorageClient | None, optional): Optional object storage client for object storage type.
-
-        Returns:
-            BaseStorageProvider: The created storage provider instance.
-        """
-    @staticmethod
-    def create_from_env(object_storage_client: BaseObjectStorageClient | None = None) -> BaseStorageProvider:
-        """Create storage provider from environment variables.
-
-        Args:
-            object_storage_client (BaseObjectStorageClient | None, optional): Optional object storage client for object storage type.
-
-        Returns:
-            BaseStorageProvider: The created storage provider instance.
-        """

aip_agents/storage/providers/__init__.pyi

@@ -1,5 +0,0 @@
-from aip_agents.storage.providers.base import BaseStorageProvider as BaseStorageProvider, StorageError as StorageError
-from aip_agents.storage.providers.memory import InMemoryStorageProvider as InMemoryStorageProvider
-from aip_agents.storage.providers.object_storage import ObjectStorageProvider as ObjectStorageProvider
-
-__all__ = ['BaseStorageProvider', 'StorageError', 'InMemoryStorageProvider', 'ObjectStorageProvider']

aip_agents/storage/providers/base.pyi

@@ -1,88 +0,0 @@
-from _typeshed import Incomplete
-from abc import ABC, abstractmethod
-from aip_agents.utils.logger import get_logger as get_logger
-from typing import Any
-
-logger: Incomplete
-
-class StorageError(Exception):
-    """Base exception for storage operations."""
-
-class BaseStorageProvider(ABC):
-    """Base interface for storage providers.
-
-    This abstract class defines the contract that all storage providers
-    must implement to store and retrieve tool outputs.
-    """
-    @abstractmethod
-    def store(self, key: str, data: Any) -> None:
-        """Store data with the given key.
-
-        Args:
-            key: Unique identifier for the data
-            data: Data to store (must be serializable)
-
-        Raises:
-            StorageError: If storage operation fails
-        """
-    @abstractmethod
-    def retrieve(self, key: str) -> Any:
-        """Retrieve data by key.
-
-        Args:
-            key: Unique identifier for the data
-
-        Returns:
-            The stored data
-
-        Raises:
-            KeyError: If key not found
-            StorageError: If retrieval operation fails
-        """
-    @abstractmethod
-    def exists(self, key: str) -> bool:
-        """Check if key exists.
-
-        Args:
-            key: Unique identifier to check
-
-        Returns:
-            True if key exists, False otherwise
-        """
-    @abstractmethod
-    def delete(self, key: str) -> None:
-        """Delete data by key.
-
-        Args:
-            key: Unique identifier for the data
-
-        Note:
-            Should not raise error if key doesn't exist
-        """
-    @abstractmethod
-    def list_keys(self, prefix: str = '') -> list[str]:
-        """List all keys with optional prefix filter.
-
-        Args:
-            prefix: Optional prefix to filter keys
-
-        Returns:
-            List of matching keys
-        """
-    @abstractmethod
-    def clear(self) -> None:
-        """Clear all stored data.
-
-        Warning:
-            This operation is irreversible
-        """
-    def get_presigned_url(self, key: str, expires_hours: int = 24) -> str | None:
-        """Generate presigned URL for direct access (optional).
-
-        Args:
-            key: Storage key
-            expires_hours: URL expiration in hours
-
-        Returns:
-            Presigned URL if supported, None otherwise
-        """

aip_agents/storage/providers/memory.pyi

@@ -1,79 +0,0 @@
-from aip_agents.storage.providers.base import BaseStorageProvider as BaseStorageProvider, StorageError as StorageError
-from typing import Any
-
-class InMemoryStorageProvider(BaseStorageProvider):
-    """In-memory storage provider for fast access to small data.
-
-    This provider stores all data in memory, providing the fastest
-    access times but limited by available RAM.
-
-    Best for:
-    - Small to medium datasets
-    - Temporary storage
-    - Development and testing
-    - High-frequency access patterns
-    """
-    def __init__(self) -> None:
-        """Initialize in-memory storage."""
-    def store(self, key: str, data: Any) -> None:
-        """Store data in memory.
-
-        Args:
-            key: Unique identifier for the data
-            data: Data to store
-
-        Raises:
-            StorageError: If storage operation fails
-        """
-    def retrieve(self, key: str) -> Any:
-        """Retrieve data from memory.
-
-        Args:
-            key: Unique identifier for the data
-
-        Returns:
-            The stored data
-
-        Raises:
-            KeyError: If key not found
-        """
-    def exists(self, key: str) -> bool:
-        """Check if key exists in memory.
-
-        Args:
-            key: Unique identifier to check
-
-        Returns:
-            True if key exists, False otherwise
-        """
-    def delete(self, key: str) -> None:
-        """Delete data from memory.
-
-        Args:
-            key: Unique identifier for the data
-        """
-    def list_keys(self, prefix: str = '') -> list[str]:
-        """List all keys with optional prefix.
-
-        Args:
-            prefix: Optional prefix to filter keys
-
-        Returns:
-            List of matching keys
-        """
-    def clear(self) -> None:
-        """Clear all data from memory."""
-    @property
-    def size_bytes(self) -> int:
-        """Get approximate memory usage in bytes.
-
-        Returns:
-            Approximate memory usage in bytes
-        """
-    @property
-    def count(self) -> int:
-        """Get number of stored items.
-
-        Returns:
-            Number of items in storage
-        """

aip_agents/storage/providers/object_storage.pyi

@@ -1,98 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.storage.base import BaseObjectStorageClient as BaseObjectStorageClient
-from aip_agents.storage.providers.base import BaseStorageProvider as BaseStorageProvider, StorageError as StorageError
-from aip_agents.utils.logger import get_logger as get_logger
-from typing import Any
-
-logger: Incomplete
-
-class ObjectStorageProvider(BaseStorageProvider):
-    """Object storage provider for S3-compatible storage.
-
-    Works with any S3-compatible storage including AWS S3, MinIO,
-    Google Cloud Storage (with S3 compatibility), etc.
-
-    Best for:
-    - Very large datasets
-    - Distributed systems
-    - Cloud deployments
-    - Long-term storage
-    - Multi-region access
-    """
-    client: Incomplete
-    prefix: Incomplete
-    use_json: Incomplete
-    def __init__(self, client: BaseObjectStorageClient, prefix: str = '', use_json: bool = False) -> None:
-        """Initialize object storage provider.
-
-        Args:
-            client: Object storage client instance
-            prefix: Prefix for all keys (like a directory)
-            use_json: Use JSON format (True) or pickle (False)
-        """
-    def store(self, key: str, data: Any) -> None:
-        """Store data in object storage.
-
-        Args:
-            key: Unique identifier for the data
-            data: Data to store
-
-        Raises:
-            StorageError: If storage operation fails
-        """
-    def retrieve(self, key: str) -> Any:
-        """Retrieve data from object storage.
-
-        Args:
-            key: Unique identifier for the data
-
-        Returns:
-            The stored data
-
-        Raises:
-            KeyError: If key not found
-            StorageError: If retrieval operation fails
-        """
-    def exists(self, key: str) -> bool:
-        """Check if object exists.
-
-        Args:
-            key: Unique identifier to check
-
-        Returns:
-            True if key exists, False otherwise
-        """
-    def delete(self, key: str) -> None:
-        """Delete object.
-
-        Args:
-            key: Unique identifier for the data
-        """
-    def list_keys(self, prefix: str = '') -> list[str]:
-        """List all keys with optional prefix.
-
-        Args:
-            prefix: Optional prefix to filter keys
-
-        Raises:
-            NotImplementedError: list_keys is not implemented
-        """
-    def clear(self) -> None:
-        """Clear all objects with the configured prefix.
-
-        Warning:
-            This is a dangerous operation!
-        """
-    def get_presigned_url(self, key: str, expires_hours: int = 24) -> str | None:
-        """Generate presigned URL for direct access.
-
-        Args:
-            key: Storage key
-            expires_hours: URL expiration in hours
-
-        Returns:
-            Presigned URL for direct access
-
-        Raises:
-            StorageError: If URL generation fails
-        """

aip_agents/tools/__init__.pyi

@@ -1,9 +0,0 @@
-from aip_agents.tools.browser_use import BrowserUseTool as BrowserUseTool
-from aip_agents.tools.code_sandbox import E2BCodeSandboxTool as E2BCodeSandboxTool
-from aip_agents.tools.document_loader import DocxReaderTool as DocxReaderTool, ExcelReaderTool as ExcelReaderTool, PDFReaderTool as PDFReaderTool
-from aip_agents.tools.gl_connector import GLConnectorTool as GLConnectorTool
-from aip_agents.tools.gl_connector_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS, GL_CONNECTORS_AUTOMATED_TOOLS as GL_CONNECTORS_AUTOMATED_TOOLS
-from aip_agents.tools.time_tool import TimeTool as TimeTool
-from aip_agents.tools.web_search import GoogleSerperTool as GoogleSerperTool
-
-__all__ = ['BOSA_AUTOMATED_TOOLS', 'GL_CONNECTORS_AUTOMATED_TOOLS', 'GLConnectorTool', 'GoogleSerperTool', 'TimeTool', 'BrowserUseTool', 'E2BCodeSandboxTool', 'DocxReaderTool', 'ExcelReaderTool', 'PDFReaderTool']

aip_agents/tools/browser_use/__init__.pyi

@@ -1,14 +0,0 @@
-from enum import StrEnum
-
-__all__ = [<ImportableName.BROWSER_USE_TOOL: 'BrowserUseTool'>, <ImportableName.BROWSER_USE_TOOL_INPUT: 'BrowserUseToolInput'>, <ImportableName.BROWSER_USE_TOOL_CONFIG: 'BrowserUseToolConfig'>]
-
-class ImportableName(StrEnum):
-    """Names of the importable attributes."""
-    BROWSER_USE_TOOL: str
-    BROWSER_USE_TOOL_INPUT: str
-    BROWSER_USE_TOOL_CONFIG: str
-
-# Names in __all__ with no definition:
-#   BrowserUseTool
-#   BrowserUseToolConfig
-#   BrowserUseToolInput