aip-agents-binary 0.4.8__cp312-cp312-win_amd64.whl → 0.5.4__cp312-cp312-win_amd64.whl

aip_agents/storage/config.pyi

@@ -1,42 +1,8 @@
-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
+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
 
-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
-        '''
+__all__ = ['OBJECT_STORAGE_PREFIX', 'StorageConfig', 'StorageType', 'StorageProviderFactory']
 
 class StorageProviderFactory:
     """Factory for creating storage providers based on configuration."""
@@ -45,22 +11,19 @@ class StorageProviderFactory:
         """Create storage provider based on configuration.
 
         Args:
-            config: Storage configuration
-            object_storage_client: Object storage client (required for object storage)
+            config (StorageConfig): Storage configuration object.
+            object_storage_client (BaseObjectStorageClient | None, optional): Optional object storage client for object storage type.
 
         Returns:
-            Configured storage provider
-
-        Raises:
-            ValueError: If configuration is invalid
+            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: Optional object storage client
+            object_storage_client (BaseObjectStorageClient | None, optional): Optional object storage client for object storage type.
 
         Returns:
-            Configured storage provider
+            BaseStorageProvider: The created storage provider instance.
         """

aip_agents/storage/providers/base.pyi

@@ -1,7 +1,6 @@
-import abc
 from _typeshed import Incomplete
 from abc import ABC, abstractmethod
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
+from aip_agents.utils.logger import get_logger as get_logger
 from typing import Any
 
 logger: Incomplete
@@ -9,7 +8,7 @@ logger: Incomplete
 class StorageError(Exception):
     """Base exception for storage operations."""
 
-class BaseStorageProvider(ABC, metaclass=abc.ABCMeta):
+class BaseStorageProvider(ABC):
     """Base interface for storage providers.
 
     This abstract class defines the contract that all storage providers

aip_agents/storage/providers/object_storage.pyi

@@ -1,7 +1,7 @@
 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 aip_agents.utils.logger import get_logger as get_logger
 from typing import Any
 
 logger: Incomplete

aip_agents/tools/__init__.pyi

@@ -1,3 +1,3 @@
-from .bosa_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS
+from aip_agents.tools.bosa_tools import BOSA_AUTOMATED_TOOLS as BOSA_AUTOMATED_TOOLS
 
 __all__ = ['BOSA_AUTOMATED_TOOLS']

aip_agents/tools/bosa_tools.pyi

@@ -1,6 +1,6 @@
 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 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, ToolType as ToolType
+from aip_agents.utils.logger import get_logger as get_logger
 from langchain_core.tools import BaseTool as BaseTool
 
 logger: Incomplete

aip_agents/tools/constants.pyi

@@ -1,129 +1,134 @@
 from _typeshed import Incomplete
-from enum import Enum
+from enum import Enum, StrEnum
 
 BOSA_API_BASE_URL: Incomplete
 BOSA_API_KEY: Incomplete
 BOSA_FETCH_MAX_RETRIES: int
 
+class ToolType(StrEnum):
+    """Tool types for BOSA Connector."""
+    GLLM: str
+    LANGCHAIN: str
+
 class Action(Enum):
     """Actions for BOSA Connector."""
-    GITHUB = 'github'
-    GOOGLE = 'google'
-    GOOGLE_DRIVE = 'google_drive'
-    GOOGLE_MAIL = 'google_mail'
-    TWITTER = 'twitter'
+    GITHUB: str
+    GOOGLE: str
+    GOOGLE_DRIVE: str
+    GOOGLE_MAIL: str
+    TWITTER: str
 
 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'
+    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 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'
+    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 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'
+    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 BOSA Connector."""
-    INTEGRATIONS = 'integrations'
-    USER_HAS_INTEGRATION = 'integration-exists'
-    SUCCESS_AUTHORIZE_CALLBACK = 'success-authorize-callback'
-    USERINFO = 'userinfo'
+    INTEGRATIONS: str
+    USER_HAS_INTEGRATION: str
+    SUCCESS_AUTHORIZE_CALLBACK: str
+    USERINFO: str
 
 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'
+    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 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'
+    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."""

aip_agents/tools/memory_search/__init__.pyi

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

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

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

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

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

@@ -1,4 +1,4 @@
-from aip_agents.types.a2a_events import A2AEvent as A2AEvent, A2AStreamEventType as A2AStreamEventType, ToolCallInfo as ToolCallInfo, ToolResultInfo as ToolResultInfo
+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
 
@@ -15,6 +15,22 @@ class AgentProtocol(Protocol):
     name: str
     description: str
     def run(self, query: str, **kwargs: Any) -> dict[str, Any]:
-        """Synchronous execution method."""
+        """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."""
+        """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,74 +1,3 @@
-from enum import Enum
-from typing import Any
-from typing_extensions import TypedDict
+from aip_agents.schema.a2a import A2AEvent as A2AEvent, A2AStreamEventType as A2AStreamEventType, ToolCallInfo as ToolCallInfo, ToolResultInfo as ToolResultInfo
 
-class A2AStreamEventType(Enum):
-    """Semantic event types for A2A agent-executor communication.
-
-    These event types preserve the semantic meaning of LangChain message types
-    and eliminate the need for string-based content classification downstream.
-    """
-    STATUS_UPDATE = 'status_update'
-    CONTENT_CHUNK = 'content_chunk'
-    FINAL_RESPONSE = 'final_response'
-    TOOL_CALL = 'tool_call'
-    TOOL_RESULT = 'tool_result'
-    ERROR = 'error'
-
-class A2AEvent(TypedDict):
-    """Structured event data for A2A agent-executor communication.
-
-    This TypedDict replaces the generic dictionary format previously used,
-    providing type safety and semantic clarity for agent streaming events.
-
-    Attributes:
-        event_type: The semantic type of this event (tool call, content, etc.).
-        content: The main text content of the event.
-        metadata: Additional metadata associated with the event.
-        tool_info: Tool-specific information (only present for tool-related events).
-        is_final: Whether this event represents the final output.
-        artifacts: List of artifacts generated by tools or agents.
-        references: Reference data from tools that support reference collection.
-        step_usage: Usage metadata associated with the event.
-        total_usage: Accumulated usage metadata associated with the event.
-    """
-    event_type: A2AStreamEventType
-    content: str
-    metadata: dict[str, Any]
-    tool_info: dict[str, Any] | None
-    is_final: bool
-    artifacts: list[dict[str, Any]] | None
-    references: list[Any] | None
-    step_usage: dict[str, Any] | None
-    total_usage: dict[str, Any] | None
-    thinking_and_activity_info: dict[str, Any] | None
-
-class ToolCallInfo(TypedDict):
-    '''Structured information for tool invocation events.
-
-    This provides a standardized format for tool call information,
-    eliminating the need for JSON encoding/decoding.
-
-    Attributes:
-        tool_calls: List of tool calls with names and arguments.
-        status: Current status of the tool invocation ("running", "finished").
-    '''
-    tool_calls: list[dict[str, Any]]
-    status: str
-
-class ToolResultInfo(TypedDict):
-    """Structured information for tool completion events.
-
-    This provides a standardized format for tool execution results,
-    eliminating the need for JSON encoding/decoding.
-
-    Attributes:
-        name: Name of the executed tool.
-        args: Arguments passed to the tool.
-        output: Output returned by the tool.
-        execution_time: Time taken for tool execution in seconds.
-    """
-    name: str
-    args: dict[str, Any]
-    output: str
-    execution_time: float | None
+__all__ = ['A2AStreamEventType', 'A2AEvent', 'ToolCallInfo', 'ToolResultInfo']

aip_agents/utils/__init__.pyi

@@ -1,5 +1,10 @@
 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.logger_manager import LoggerManager as LoggerManager
+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__ = ['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']
+__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', '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']

aip_agents/utils/a2a_connector.pyi

@@ -1,9 +1,14 @@
 from _typeshed import Incomplete
-from a2a.types import AgentCard as AgentCard
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
-from aip_agents.utils.metadata_helper import MetadataFieldKeys as MetadataFieldKeys, MetadataTimeTracker as MetadataTimeTracker, Status as Status, create_metadata as create_metadata, create_status_update_metadata as create_status_update_metadata, create_tool_processing_metadata as create_tool_processing_metadata
+from a2a.types import AgentCard
+from aip_agents.schema.a2a import A2AStreamEventType as A2AStreamEventType
+from aip_agents.utils.event_handler_registry import DEFAULT_EVENT_HANDLER_REGISTRY as DEFAULT_EVENT_HANDLER_REGISTRY, EventHandlerRegistry as EventHandlerRegistry
+from aip_agents.utils.final_response_builder import FinalResponseMetadataOptions as FinalResponseMetadataOptions, assemble_final_response as assemble_final_response
+from aip_agents.utils.logger import get_logger as get_logger
+from aip_agents.utils.metadata_helper import Kind as Kind, MetadataFieldKeys as MetadataFieldKeys, MetadataTimeTracker as MetadataTimeTracker, Status as Status, create_metadata as create_metadata, create_status_update_metadata as create_status_update_metadata, create_tool_processing_metadata as create_tool_processing_metadata
+from aip_agents.utils.sse_chunk_transformer import SSEChunkTransformer as SSEChunkTransformer
+from collections.abc import AsyncGenerator
 from pydantic import BaseModel
-from typing import Any, AsyncGenerator
+from typing import Any
 
 logger: Incomplete
 
@@ -25,6 +30,10 @@ class ArtifactInfo(BaseModel):
     description: str | None
     parts: int | None
 
+class StreamingConfig(BaseModel):
+    """Configuration for A2A streaming operations."""
+    http_kwargs: dict[str, Any] | None
+
 class A2AConnector:
     """Handles A2A protocol communication between agents.
 
@@ -33,6 +42,7 @@ class A2AConnector:
     responses with immediate artifact event handling.
     """
     FLOAT_EPSILON: float
+    event_registry: EventHandlerRegistry
     @staticmethod
     def send_to_agent(agent_card: AgentCard, message: str | dict[str, Any], **kwargs: Any) -> dict[str, Any]:
         """Synchronously sends a message to another agent using the A2A protocol.