aip-agents-binary 0.0.0b2__py3-none-any.whl → 0.5.12__py3-none-any.whl

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_manager import LoggerManager as LoggerManager
-from fastapi import FastAPI as 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_bosa_functions() -> None:
-    """Instrument BOSA 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/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,66 +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.providers.base import BaseStorageProvider as BaseStorageProvider
-from aip_agents.storage.providers.memory import InMemoryStorageProvider as InMemoryStorageProvider
-from aip_agents.storage.providers.object_storage import ObjectStorageProvider as ObjectStorageProvider
-from dataclasses import dataclass
-from enum import StrEnum
-
-OBJECT_STORAGE_PREFIX: str
-
-class StorageType(StrEnum):
-    """Supported storage types."""
-    MEMORY = 'memory'
-    OBJECT_STORAGE = 'object_storage'
-
-@dataclass
-class StorageConfig:
-    """Configuration for storage providers.
-
-    Attributes:
-        storage_type: Type of storage to use
-        object_prefix: Prefix for object storage keys (like a directory path)
-        object_use_json: Use JSON serialization instead of pickle for object storage
-    """
-    storage_type: StorageType = ...
-    object_prefix: str = ...
-    object_use_json: bool = ...
-    @classmethod
-    def from_env(cls) -> StorageConfig:
-        '''Create StorageConfig from environment variables.
-
-        Environment variables:
-        - TOOL_OUTPUT_STORAGE_TYPE: "memory" or "object_storage"
-        - TOOL_OUTPUT_OBJECT_PREFIX: Prefix for object storage keys
-        - TOOL_OUTPUT_USE_JSON: "true" to use JSON serialization
-
-        Returns:
-            StorageConfig instance
-        '''
-
-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: Storage configuration
-            object_storage_client: Object storage client (required for object storage)
-
-        Returns:
-            Configured storage provider
-
-        Raises:
-            ValueError: If configuration is invalid
-        """
-    @staticmethod
-    def create_from_env(object_storage_client: BaseObjectStorageClient | None = None) -> BaseStorageProvider:
-        """Create storage provider from environment variables.
-
-        Args:
-            object_storage_client: Optional object storage client
-
-        Returns:
-            Configured storage provider
-        """

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/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_manager import LoggerManager as LoggerManager
-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,3 +0,0 @@
-from .bosa_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS
-
-__all__ = ['BOSA_AUTOMATED_TOOLS']

aip_agents/tools/base.pyi

@@ -1,44 +0,0 @@
-from langchain_core.tools import BaseTool as LangchainBaseTool
-from typing import Any
-
-class BaseTool(LangchainBaseTool):
-    """Base class for tools inheriting from Langchain's BaseTool.
-
-    Tools should inherit from this class and implement the _run and optionally
-    _arun methods.
-
-    Attributes:
-        name (str): The name of the tool.
-        description (str): A description of what the tool does.
-        args_schema (Optional[Type[Any]]): The schema for the tool's arguments.
-        return_direct (bool): Whether to return the tool's output directly.
-    """
-    name: str
-    description: str
-    args_schema: type[Any] | None
-    return_direct: bool
-    def run(self, tool_input: Any, **_kwargs: Any) -> Any:
-        """Run the tool synchronously. (Not Implemented)
-
-        This method is intentionally not implemented to encourage asynchronous
-        operations via the `arun` method.
-
-        Args:
-            tool_input: The input for the tool.
-            **_kwargs: Unused. Additional keyword arguments.
-
-        Raises:
-            NotImplementedError: Always raised.
-        """
-    async def arun(self, tool_input: Any, **_kwargs: Any) -> Any:
-        """Run the tool asynchronously.
-
-        Args:
-            tool_input: The input for the tool.
-            **_kwargs: Unused. Additional keyword arguments.
-
-        Returns:
-            The output of the tool.
-        """
-    async def __call__(self, tool_input: Any) -> Any:
-        """Allows the tool instance to be called asynchronously."""

aip_agents/tools/base_bosa_tools.pyi

@@ -1,12 +0,0 @@
-from aip_agents.tools.bosa_connector import BOSAConnector as BOSAConnector
-from aip_agents.tools.bosa_tools_interface import BosaToolInterface as BosaToolInterface
-from aip_agents.tools.constants import BOSA_API_BASE_URL as BOSA_API_BASE_URL, BOSA_API_KEY as BOSA_API_KEY
-from langchain_core.callbacks import AsyncCallbackManagerForToolRun as AsyncCallbackManagerForToolRun
-
-class BaseBosaTool(BosaToolInterface):
-    """Base class for tools with BOSA Connector integration."""
-    def __init__(self, **kwargs) -> None:
-        """Initialize the BaseBosaTool.
-
-        This method calls the _init_bosa_connector method to initialize the BosaConnector.
-        """

aip_agents/tools/bosa_connector.pyi

@@ -1,30 +0,0 @@
-from _typeshed import Incomplete
-from bosa_connectors.connector import BosaConnector
-from typing import Any
-
-MAX_ATTEMPTS: int
-logger: Incomplete
-
-class BOSAConnector:
-    """Handler for fetching data using the BOSA Connector."""
-    bosa: BosaConnector
-    def __init__(self, api_base_url: str, api_key: str) -> None:
-        """Initialize the BOSA connector.
-
-        Args:
-            api_base_url: Base URL for the BOSA API
-            api_key: API key for authentication
-        """
-    def execute_endpoint(self, action: str, endpoint: str, input_: dict[str, Any], max_attempts: int = ..., forwards_cursor: bool = True) -> dict[str, Any]:
-        """Execute a BOSA endpoint and fetch data.
-
-        Args:
-            action: The BOSA API action to call
-            endpoint: The BOSA API endpoint to call
-            input_: Parameters matching the API spec
-            max_attempts: Max retry attempts
-            forwards_cursor: Whether to use cursor-based pagination (True) or page-based pagination (False)
-
-        Returns:
-            dict containing response data with '0', 'data', and 'meta' fields
-        """

aip_agents/tools/bosa_tools.pyi

@@ -1,37 +0,0 @@
-from _typeshed import Incomplete
-from aip_agents.tools.constants import BOSA_API_BASE_URL as BOSA_API_BASE_URL, BOSA_API_KEY as BOSA_API_KEY, BOSA_FETCH_MAX_RETRIES as BOSA_FETCH_MAX_RETRIES
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
-from langchain_core.tools import BaseTool as BaseTool
-
-logger: Incomplete
-
-def get_bosa_modules_with_retry() -> list[str]:
-    """Try to get available modules with retries.
-
-    Returns:
-        List of available modules.
-    """
-
-class LazyBosaToolsDict(dict):
-    """Lazy dictionary for BOSA tools."""
-    def __missing__(self, app):
-        """When a key is missing, create the tools and store them in the dictionary.
-
-        Args:
-            app: Name of the BOSA connector.
-
-        Returns:
-            List of tools generated by BOSAConnectorToolGenerator.
-        """
-
-def get_bosa_modules() -> list[str]:
-    """Lazily fetch and cache BOSA modules.
-
-    This is for backwards compatibility with the old BOSA modules.
-
-    Returns:
-        List of BOSA modules.
-    """
-
-BOSA_MODULES: Incomplete
-BOSA_AUTOMATED_TOOLS: dict[str, list[BaseTool]]

aip_agents/tools/bosa_tools_interface.pyi

@@ -1,26 +0,0 @@
-import abc
-from abc import ABC, abstractmethod
-from aip_agents.tools.bosa_connector import BOSAConnector as BOSAConnector
-from pydantic import BaseModel as BaseModel
-
-class BosaToolInterface(ABC, metaclass=abc.ABCMeta):
-    """Interface for BOSA tools.
-
-    Defines the abstract base class for tools that agents can use.
-
-    Attributes:
-        name (str): The name of the tool.
-        description (Optional[str]): The description of the tool.
-        args_schema (Optional[Type[BaseModel]]): The schema for the tool arguments.
-        _bosa_connector (Optional[BOSAConnector]): The BosaConnector instance.
-    """
-    name: str
-    description: str | None
-    args_schema: type[BaseModel] | None
-    @abstractmethod
-    def __init__(self):
-        """Initialize the BosaToolInterface.
-
-        This method should be implemented by subclasses to define the tool's
-        initialization logic. The _init_bosa_connector should be called here.
-        """

aip_agents/tools/constants.pyi

@@ -1,130 +0,0 @@
-from _typeshed import Incomplete
-from enum import Enum
-
-BOSA_API_BASE_URL: Incomplete
-BOSA_API_KEY: Incomplete
-BOSA_FETCH_MAX_RETRIES: int
-
-class Action(Enum):
-    """Actions for BOSA Connector."""
-    GITHUB = 'github'
-    GOOGLE = 'google'
-    GOOGLE_DRIVE = 'google_drive'
-    GOOGLE_MAIL = 'google_mail'
-    TWITTER = 'twitter'
-
-class GitHubEndpoint(Enum):
-    """GitHub endpoints for BOSA Connector."""
-    INTEGRATIONS = 'integrations'
-    USER_HAS_INTEGRATION = 'integration-exists'
-    SUCCESS_AUTHORIZE_CALLBACK = 'success-authorize-callback'
-    CREATE_ISSUE_HANDLER = 'create_issue'
-    GET_ISSUE_HANDLER = 'get_issue'
-    LIST_ISSUES = 'list_issues'
-    LIST_ISSUES_COMMENTS = 'list_issues_comments'
-    SEARCH_ALL_ISSUES = 'search_issues'
-    GET_COMMITS = 'list_commits'
-    SEARCH_COMMITS = 'search_commits'
-    GET_COLLABORATORS = 'list_collaborators'
-    GET_RELEASES = 'list_releases'
-    GET_CONTRIBUTORS = 'list_contributors'
-    GET_LANGUAGES = 'list_languages'
-    SEARCH_CONTRIBUTIONS = 'search_contributions'
-    LIST_PULL_REQUESTS = 'list_pull_requests'
-    SEARCH_PR = 'search_pull_requests'
-    GET_PULL = 'get_pull'
-    GET_ALL_CONTRIBUTOR_COMMIT_ACTIVITIES = 'get_all_contributor_commit_activities'
-    GET_COMMIT_ACTIVITY = 'get_the_last_year_of_commit_activity'
-    GET_WEEKLY_COMMIT_COUNTS = 'get_weekly_commit_count '
-    GET_USER_CONTRIBUTION_STATISTICS = 'get_user_contribution_statistics'
-    LIST_PROJECT_ITEMS = 'list_project_items'
-    LIST_PROJECTS = 'list_projects'
-
-class GoogleDriveEndpoint(Enum):
-    """Google Drive endpoints for BOSA Connector."""
-    INTEGRATIONS = 'integrations'
-    USER_HAS_INTEGRATION = 'integration-exists'
-    SUCCESS_AUTHORIZE_CALLBACK = 'success-authorize-callback'
-    SEARCH_FILES = 'search_files'
-    GET_FILE = 'get_file'
-    CREATE_FILE = 'create_file'
-    CREATE_FOLDER = 'create_folder'
-    UPDATE_FILE = 'update_file'
-    UPDATE_FOLDER = 'update_folder'
-    COPY_FILE = 'copy_file'
-    DELETE_FILE = 'delete_file'
-    SUMMARIZE_FOLDER_FILES_BY_TYPE = 'summarize_folder_files_by_type'
-    SUMMARIZE_TOTAL_FILES_BY_TYPE = 'summarize_total_files_by_type'
-    RECENT_FILES = 'list_recent_files_from_yesterday'
-    CREATE_PERMISSION = 'create_permission'
-    LIST_PERMISSIONS = 'list_permissions'
-    GET_PERMISSION = 'get_permission'
-    UPDATE_PERMISSION = 'update_permission'
-    DELETE_PERMISSION = 'delete_permission'
-    DOWNLOAD_FILE = 'download_file'
-
-class GoogleDocsEndpoint(Enum):
-    """Google Docs endpoints for BOSA Connector."""
-    INTEGRATIONS = 'integrations'
-    USER_HAS_INTEGRATION = 'integration-exists'
-    SUCCESS_AUTHORIZE_CALLBACK = 'success-authorize-callback'
-    GET_DOCUMENT = 'get_document'
-    LIST_DOCUMENTS = 'list_documents'
-    CREATE_DOCUMENT = 'create_document'
-    UPDATE_DOCUMENT = 'update_document'
-    COPY_CONTENT = 'copy_content'
-    UPDATE_DOCUMENT_MARKDOWN = 'update_document_markdown'
-    LIST_COMMENTS = 'list_comments'
-    SUMMARIZE_COMMENTS = 'summarize_comments'
-
-class GoogleEndpoint(Enum):
-    """Google endpoints for BOSA Connector."""
-    INTEGRATIONS = 'integrations'
-    USER_HAS_INTEGRATION = 'integration-exists'
-    SUCCESS_AUTHORIZE_CALLBACK = 'success-authorize-callback'
-    USERINFO = 'userinfo'
-
-class TwitterEndpoint(Enum):
-    """Twitter endpoints for BOSA Connector."""
-    INTEGRATIONS = 'integrations'
-    USER_HAS_INTEGRATION = 'integration-exists'
-    SUCCESS_AUTHORIZE_CALLBACK = 'success-authorize-callback'
-    SEARCH = 'tweet_search'
-    GET_TWEETS = 'get_tweets'
-    GET_THREAD = 'get_thread'
-    GET_USERS = 'get_users'
-
-class GoogleMailEndpoint(Enum):
-    """Google Mail endpoints for BOSA Connector."""
-    INTEGRATIONS = 'integrations'
-    USER_HAS_INTEGRATION = 'integration-exists'
-    SUCCESS_AUTHORIZE_CALLBACK = 'success-authorize-callback'
-    CREATE_DRAFT = 'create_draft'
-    LIST_DRAFTS = 'list_drafts'
-    SEND_DRAFT = 'send_draft'
-    GET_DRAFT = 'get_draft'
-    MODIFY_DRAFT = 'modify_draft'
-    LIST_LABELS = 'list_labels'
-    LABEL_STATS = 'label_stats'
-    GET_LABEL_DETAILS = 'get_label_details'
-    CREATE_LABELS = 'create_labels'
-    MODIFY_LABELS = 'modify_labels'
-    DELETE_LABELS = 'delete_labels'
-    SEND_EMAIL = 'send_email'
-    LIST_EMAILS = 'list_emails'
-    GET_EMAIL_DETAILS = 'get_email_details'
-    MODIFY_EMAIL = 'modify_email'
-    DELETE_EMAIL = 'delete_email'
-    TRASH_EMAIL = 'trash_email'
-    UNTRASH_EMAIL = 'untrash_email'
-    LIST_THREADS = 'list_threads'
-    THREAD_DETAILS = 'thread_details'
-    MODIFY_THREAD = 'modify_thread'
-    GET_AUTO_REPLY = 'get_auto_reply'
-    SET_AUTO_REPLY = 'set_auto_reply'
-    GET_ATTACHMENT = 'get_attachment'
-    USERINFO = 'userinfo'
-
-class ActionEndpointMap:
-    """Maps Action enums to their corresponding Endpoint enums."""
-    MAP: dict[Action, type[Enum]]