aip-agents-binary 0.4.8__cp311-cp311-win_amd64.whl → 0.5.4__cp311-cp311-win_amd64.whl

aip_agents/__init__.pyi

@@ -0,0 +1,19 @@
+__all__ = ['a2a', 'agent', 'clients', 'constants', 'credentials', 'executor', 'memory', 'mcp', 'schema', 'sentry', 'storage', 'tools', 'types', 'utils', '__version__']
+
+__version__: str
+
+# Names in __all__ with no definition:
+#   a2a
+#   agent
+#   clients
+#   constants
+#   credentials
+#   executor
+#   mcp
+#   memory
+#   schema
+#   sentry
+#   storage
+#   tools
+#   types
+#   utils

aip_agents/a2a/__init__.pyi

@@ -1,3 +1,3 @@
-from .types import ArtifactType as ArtifactType, MimeType as MimeType, get_extension_from_mime_type as get_extension_from_mime_type, get_mime_type_from_filename as get_mime_type_from_filename
+from aip_agents.a2a.types import ArtifactType as ArtifactType, MimeType as MimeType, get_extension_from_mime_type as get_extension_from_mime_type, get_mime_type_from_filename as get_mime_type_from_filename
 
 __all__ = ['ArtifactType', 'MimeType', 'get_mime_type_from_filename', 'get_extension_from_mime_type']

aip_agents/a2a/server/base_executor.pyi

@@ -1,13 +1,13 @@
-import abc
 from _typeshed import Incomplete
-from a2a.server.agent_execution import AgentExecutor as A2ASDKExecutor, RequestContext as RequestContext
-from a2a.server.events.event_queue import EventQueue as EventQueue
+from a2a.server.agent_execution import AgentExecutor as A2ASDKExecutor, RequestContext
+from a2a.server.events.event_queue import EventQueue
 from abc import ABC, abstractmethod
 from aip_agents.types import A2AEvent as A2AEvent, A2AStreamEventType as A2AStreamEventType
 from aip_agents.utils import serialize_references_for_metadata as serialize_references_for_metadata
 from aip_agents.utils.artifact_helpers import ArtifactHandler as ArtifactHandler
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
+from aip_agents.utils.logger import get_logger as get_logger
 from aip_agents.utils.metadata_helper import MetadataFieldKeys as MetadataFieldKeys
+from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from typing import Any
 
@@ -21,12 +21,12 @@ class StatusUpdateParams:
     task_id: str | None = ...
     context_id: str | None = ...
 
-class BaseA2AExecutor(A2ASDKExecutor, ABC, metaclass=abc.ABCMeta):
+class BaseA2AExecutor(A2ASDKExecutor, ABC):
     """Abstract base class for GLLM Agent framework's A2A server-side executors.
 
     This class extends the A2A SDK's `AgentExecutor`. It serves as a common
     foundation for specific executors tailored to different agent types within the
-    `gllm-agents` framework, such as `LangGraphA2AExecutor` or
+    `aip-agents` framework, such as `LangGraphA2AExecutor` or
     `GoogleADKA2AExecutor`.
 
     Subclasses are required to implement the `execute` method to handle A2A

aip_agents/a2a/server/google_adk_executor.pyi

@@ -1,10 +1,10 @@
 from _typeshed import Incomplete
-from a2a.server.agent_execution import RequestContext as RequestContext
-from a2a.server.events.event_queue import EventQueue as EventQueue
-from a2a.server.tasks import TaskUpdater as TaskUpdater
+from a2a.server.agent_execution import RequestContext
+from a2a.server.events.event_queue import EventQueue
 from aip_agents.a2a.server.base_executor import BaseA2AExecutor as BaseA2AExecutor, StatusUpdateParams as StatusUpdateParams
-from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
+from aip_agents.agent.google_adk_constants import DEFAULT_AUTH_URL as DEFAULT_AUTH_URL
+from aip_agents.agent.interfaces import GoogleADKAgentProtocol as GoogleADKAgentProtocol
+from aip_agents.utils.logger import get_logger as get_logger
 
 logger: Incomplete
 
@@ -12,7 +12,7 @@ class GoogleADKExecutor(BaseA2AExecutor):
     '''A2A Executor for serving a `GoogleADKAgent`.
 
     This executor bridges the A2A server protocol with a `aip_agents.agent.GoogleADKAgent`.
-    It handles incoming requests by invoking the agent\'s `arun_a2a_stream` method
+    It handles incoming requests by invoking the agent\'s `arun_a2a_stream` method,
     which is specifically designed to yield ADK events in an A2A-compatible dictionary
     format. This executor\'s `_process_stream` method is tailored to handle this stream,
     including ADK-specific statuses like "auth_required", before delegating common
@@ -22,17 +22,17 @@ class GoogleADKExecutor(BaseA2AExecutor):
     initial request checks, and cancellation.
 
     Attributes:
-        agent (GoogleADKAgent): The instance of `GoogleADKAgent` to be executed.
+        agent (GoogleADKAgentProtocol): The instance of `GoogleADKAgent`-compatible class to execute.
     '''
-    agent: GoogleADKAgent
-    def __init__(self, agent: GoogleADKAgent) -> None:
+    agent: GoogleADKAgentProtocol
+    def __init__(self, agent: GoogleADKAgentProtocol) -> None:
         """Initializes the GoogleADKExecutor.
 
         Args:
-            agent (GoogleADKAgent): A fully initialized instance of `GoogleADKAgent`.
+            agent: Component implementing `GoogleADKAgentProtocol`.
 
         Raises:
-            TypeError: If `agent` is not an instance of `GoogleADKAgent`.
+            TypeError: If the provided agent does not satisfy `GoogleADKAgentProtocol`.
         """
     async def execute(self, context: RequestContext, event_queue: EventQueue) -> None:
         """Processes an incoming agent request using the `GoogleADKAgent`.

aip_agents/a2a/server/langflow_executor.pyi

@@ -0,0 +1,43 @@
+from _typeshed import Incomplete
+from a2a.server.agent_execution import RequestContext
+from a2a.server.events.event_queue import EventQueue
+from abc import ABC
+from aip_agents.a2a.server.base_executor import BaseA2AExecutor as BaseA2AExecutor, StatusUpdateParams as StatusUpdateParams
+from aip_agents.agent.interfaces import LangflowAgentProtocol as LangflowAgentProtocol
+from aip_agents.utils.logger import get_logger as get_logger
+
+logger: Incomplete
+
+class LangflowA2AExecutor(BaseA2AExecutor, ABC):
+    """A2A executor for Langflow agents.
+
+    This class extends BaseA2AExecutor to provide A2A execution capabilities
+    for Langflow agents. It follows the same patterns as LangGraphA2AExecutor
+    but handles Langflow-specific streaming and execution logic.
+
+    Attributes:
+        agent: The LangflowAgent-compatible instance to be executed.
+    """
+    agent: LangflowAgentProtocol
+    def __init__(self, langflow_agent_instance: LangflowAgentProtocol) -> None:
+        """Initialize the LangflowA2AExecutor.
+
+        Args:
+            langflow_agent_instance: Component implementing `LangflowAgentProtocol`.
+
+        Raises:
+            TypeError: If the agent does not satisfy `LangflowAgentProtocol`.
+        """
+    async def execute(self, context: RequestContext, event_queue: EventQueue) -> None:
+        """Process an incoming agent request using a Langflow agent.
+
+        This method handles the execution lifecycle for Langflow agents:
+        1. Performs initial validation and setup
+        2. Creates agent processing coroutine
+        3. Manages the execution lifecycle through BaseA2AExecutor
+
+        Args:
+            context: The A2A request context containing message details,
+                task ID, and context ID.
+            event_queue: The queue for sending A2A events back to the server.
+        """

aip_agents/a2a/server/langgraph_executor.pyi

@@ -1,11 +1,11 @@
 from _typeshed import Incomplete
-from a2a.server.agent_execution import RequestContext as RequestContext
-from a2a.server.events.event_queue import EventQueue as EventQueue
-from a2a.server.tasks import TaskUpdater as TaskUpdater
+from a2a.server.agent_execution import RequestContext
+from a2a.server.events.event_queue import EventQueue
 from abc import ABC
 from aip_agents.a2a.server.base_executor import BaseA2AExecutor as BaseA2AExecutor, StatusUpdateParams as StatusUpdateParams
-from aip_agents.agent import BaseLangGraphAgent as BaseLangGraphAgent
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
+from aip_agents.agent.interfaces import LangGraphAgentProtocol as LangGraphAgentProtocol
+from aip_agents.schema.step_limit import StepLimitConfig as StepLimitConfig
+from aip_agents.utils.logger import get_logger as get_logger
 
 logger: Incomplete
 
@@ -18,18 +18,17 @@ class LangGraphA2AExecutor(BaseA2AExecutor, ABC):
     agent execution, while leaving agent-specific initialization to subclasses.
 
     Attributes:
-        agent (BaseLangGraphAgent): The LangChain-based agent instance to be executed.
+        agent (LangGraphAgentProtocol): The LangChain-based agent instance to be executed.
     """
-    agent: BaseLangGraphAgent
-    def __init__(self, langgraph_agent_instance: BaseLangGraphAgent) -> None:
+    agent: LangGraphAgentProtocol
+    def __init__(self, langgraph_agent_instance: LangGraphAgentProtocol) -> None:
         """Initializes the LangGraphA2AExecutor.
 
         Args:
-            langgraph_agent_instance (BaseLangGraphAgent): A fully initialized instance
-                of `BaseLangGraphAgent`.
+            langgraph_agent_instance: Component implementing `LangGraphAgentProtocol`.
 
         Raises:
-            TypeError: If `langgraph_agent_instance` is not an instance of `BaseLangGraphAgent`.
+            TypeError: If the provided agent does not satisfy `LangGraphAgentProtocol`.
         """
     async def execute(self, context: RequestContext, event_queue: EventQueue) -> None:
         """Processes an incoming agent request using a LangChain-based agent.

aip_agents/a2a/types.pyi

@@ -9,58 +9,58 @@ class MimeType(StrEnum):
     This class provides constants for commonly used MIME types in artifact generation,
     ensuring consistency and reducing typos across the codebase.
     """
-    TEXT_PLAIN = 'text/plain'
-    TEXT_HTML = 'text/html'
-    TEXT_CSS = 'text/css'
-    TEXT_JAVASCRIPT = 'text/javascript'
-    TEXT_CSV = 'text/csv'
-    TEXT_XML = 'text/xml'
-    TEXT_MARKDOWN = 'text/markdown'
-    APPLICATION_JSON = 'application/json'
-    APPLICATION_XML = 'application/xml'
-    APPLICATION_PDF = 'application/pdf'
-    APPLICATION_ZIP = 'application/zip'
-    APPLICATION_GZIP = 'application/gzip'
-    APPLICATION_TAR = 'application/x-tar'
-    APPLICATION_OCTET_STREAM = 'application/octet-stream'
-    APPLICATION_MSWORD = 'application/msword'
-    APPLICATION_EXCEL = 'application/vnd.ms-excel'
-    APPLICATION_POWERPOINT = 'application/vnd.ms-powerpoint'
-    APPLICATION_DOCX = 'application/vnd.openxmlformats-officedocument.wordprocessingml.document'
-    APPLICATION_XLSX = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
-    APPLICATION_PPTX = 'application/vnd.openxmlformats-officedocument.presentationml.presentation'
-    IMAGE_JPEG = 'image/jpeg'
-    IMAGE_PNG = 'image/png'
-    IMAGE_GIF = 'image/gif'
-    IMAGE_WEBP = 'image/webp'
-    IMAGE_SVG = 'image/svg+xml'
-    IMAGE_BMP = 'image/bmp'
-    IMAGE_TIFF = 'image/tiff'
-    IMAGE_ICO = 'image/x-icon'
-    AUDIO_MP3 = 'audio/mpeg'
-    AUDIO_WAV = 'audio/wav'
-    AUDIO_OGG = 'audio/ogg'
-    AUDIO_AAC = 'audio/aac'
-    AUDIO_FLAC = 'audio/flac'
-    VIDEO_MP4 = 'video/mp4'
-    VIDEO_AVI = 'video/x-msvideo'
-    VIDEO_MOV = 'video/quicktime'
-    VIDEO_WMV = 'video/x-ms-wmv'
-    VIDEO_WEBM = 'video/webm'
-    VIDEO_MKV = 'video/x-matroska'
-    FONT_TTF = 'font/ttf'
-    FONT_OTF = 'font/otf'
-    FONT_WOFF = 'font/woff'
-    FONT_WOFF2 = 'font/woff2'
+    TEXT_PLAIN: str
+    TEXT_HTML: str
+    TEXT_CSS: str
+    TEXT_JAVASCRIPT: str
+    TEXT_CSV: str
+    TEXT_XML: str
+    TEXT_MARKDOWN: str
+    APPLICATION_JSON: str
+    APPLICATION_XML: str
+    APPLICATION_PDF: str
+    APPLICATION_ZIP: str
+    APPLICATION_GZIP: str
+    APPLICATION_TAR: str
+    APPLICATION_OCTET_STREAM: str
+    APPLICATION_MSWORD: str
+    APPLICATION_EXCEL: str
+    APPLICATION_POWERPOINT: str
+    APPLICATION_DOCX: str
+    APPLICATION_XLSX: str
+    APPLICATION_PPTX: str
+    IMAGE_JPEG: str
+    IMAGE_PNG: str
+    IMAGE_GIF: str
+    IMAGE_WEBP: str
+    IMAGE_SVG: str
+    IMAGE_BMP: str
+    IMAGE_TIFF: str
+    IMAGE_ICO: str
+    AUDIO_MP3: str
+    AUDIO_WAV: str
+    AUDIO_OGG: str
+    AUDIO_AAC: str
+    AUDIO_FLAC: str
+    VIDEO_MP4: str
+    VIDEO_AVI: str
+    VIDEO_MOV: str
+    VIDEO_WMV: str
+    VIDEO_WEBM: str
+    VIDEO_MKV: str
+    FONT_TTF: str
+    FONT_OTF: str
+    FONT_WOFF: str
+    FONT_WOFF2: str
 
 class ArtifactType(StrEnum):
     """Common artifact types for A2A artifacts.
 
     This class provides constants for artifact types used in the A2A protocol.
     """
-    FILE = 'file'
-    DATA = 'data'
-    TEXT = 'text'
+    FILE: str
+    DATA: str
+    TEXT: str
 
 EXTENSION_TO_MIME_TYPE: Incomplete
 MIME_TYPE_TO_EXTENSION: Incomplete
@@ -122,4 +122,11 @@ class Artifact(BaseModel):
     metadata: dict[str, Any] | None
     @classmethod
     def validate_base64(cls, v: str) -> str:
-        """Validate that 'data' is a valid base64 string."""
+        """Validate that 'data' is a valid base64 string.
+
+        Args:
+            v (str): The base64 string to validate.
+
+        Returns:
+            str: The validated base64 string if valid.
+        """

aip_agents/agent/__init__.pyi

@@ -2,6 +2,8 @@ from aip_agents.agent.base_agent import BaseAgent as BaseAgent
 from aip_agents.agent.base_langgraph_agent import BaseLangGraphAgent as BaseLangGraphAgent
 from aip_agents.agent.google_adk_agent import GoogleADKAgent as GoogleADKAgent
 from aip_agents.agent.interface import AgentInterface as AgentInterface
+from aip_agents.agent.langflow_agent import LangflowAgent as LangflowAgent
+from aip_agents.agent.langgraph_memory_enhancer_agent import LangGraphMemoryEnhancerAgent as LangGraphMemoryEnhancerAgent
 from aip_agents.agent.langgraph_react_agent import LangChainAgent as LangChainAgent, LangGraphAgent as LangGraphAgent, LangGraphReactAgent as LangGraphReactAgent
 
-__all__ = ['AgentInterface', 'BaseAgent', 'BaseLangGraphAgent', 'LangGraphReactAgent', 'GoogleADKAgent', 'LangGraphAgent', 'LangChainAgent']
+__all__ = ['AgentInterface', 'BaseAgent', 'BaseLangGraphAgent', 'LangGraphReactAgent', 'GoogleADKAgent', 'LangGraphAgent', 'LangChainAgent', 'LangflowAgent', 'LangGraphMemoryEnhancerAgent']

aip_agents/agent/base_agent.pyi

@@ -1,16 +1,16 @@
-import abc
 from _typeshed import Incomplete
 from a2a.types import AgentCard
-from aip_agents.a2a.server.google_adk_executor import GoogleADKExecutor as GoogleADKExecutor
-from aip_agents.a2a.server.langgraph_executor import LangGraphA2AExecutor as LangGraphA2AExecutor
 from aip_agents.agent.interface import AgentInterface as AgentInterface
-from aip_agents.agent.types import A2AClientConfig as A2AClientConfig, AgentConfig as AgentConfig, BaseAgentConfig as BaseAgentConfig, CredentialType as CredentialType
 from aip_agents.credentials.manager import CredentialsManager as CredentialsManager
+from aip_agents.mcp.client.base_mcp_client import BaseMCPClient as BaseMCPClient
+from aip_agents.schema.agent import A2AClientConfig as A2AClientConfig, AgentConfig as AgentConfig, BaseAgentConfig as BaseAgentConfig, CredentialType as CredentialType
+from aip_agents.schema.model_id import ModelId as ModelId, ModelProvider as ModelProvider
 from aip_agents.utils.a2a_connector import A2AConnector as A2AConnector
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
+from aip_agents.utils.logger import get_logger as get_logger
 from aip_agents.utils.name_preprocessor.name_preprocessor import NamePreprocessor as NamePreprocessor
+from collections.abc import AsyncGenerator
 from starlette.applications import Starlette
-from typing import Any, AsyncGenerator
+from typing import Any
 
 logger: Incomplete
 AGENT_EXECUTOR_MAPPING: Incomplete
@@ -19,7 +19,7 @@ LM_EXCLUDE_FIELDS: Incomplete
 CUSTOM_PROVIDERS: Incomplete
 OUTPUT_ANALYTICS_KEY: str
 
-class BaseAgent(AgentInterface, metaclass=abc.ABCMeta):
+class BaseAgent(AgentInterface):
     """Base class for agents, providing common A2A client method implementations.
 
     Concrete agent implementations (e.g., LangGraphAgent, GoogleADKAgent)
@@ -34,8 +34,8 @@ class BaseAgent(AgentInterface, metaclass=abc.ABCMeta):
     """
     model: Incomplete
     tools: Incomplete
-    mcp_config: Incomplete
     tool_configs: Incomplete
+    mcp_client: BaseMCPClient | None
     name_preprocessor: Incomplete
     def __init__(self, name: str, instruction: str, description: str | None = None, model: Any | None = None, tools: list[Any] | None = None, config: BaseAgentConfig | dict[str, Any] | None = None, tool_configs: dict[str, Any] | None = None, **kwargs: Any) -> None:
         """Initializes the BaseAgent.
@@ -69,6 +69,22 @@ class BaseAgent(AgentInterface, metaclass=abc.ABCMeta):
         Returns:
             str: The provider of the model.
         """
+    @property
+    def mcp_config(self) -> dict[str, dict[str, Any]]:
+        """Read-only view of MCP configuration.
+
+        Returns a copy to prevent direct mutation; use add_mcp_server() for changes.
+        """
+    @mcp_config.setter
+    def mcp_config(self, value: dict[str, dict[str, Any]]) -> None:
+        """Set MCP configuration and maintain synchronization.
+
+        Automatically resets initialization flag and recreates client to ensure consistency.
+        Prefer using add_mcp_server() for proper validation.
+
+        Args:
+            value (dict[str, dict[str, Any]]): The MCP configuration to set.
+        """
     def to_a2a(self, agent_card: AgentCard, **kwargs: Any) -> Starlette:
         """Converts the agent to an A2A-compatible ASGI application.
 

aip_agents/agent/base_langgraph_agent.pyi

@@ -1,29 +1,35 @@
-import abc
 from _typeshed import Incomplete
 from a2a.types import AgentCard as AgentCard
 from abc import abstractmethod
 from aip_agents.agent.base_agent import BaseAgent as BaseAgent
-from aip_agents.agent.types import StreamMode as StreamMode
-from aip_agents.mcp.client.langchain import LangchainMCPClient as LangchainMCPClient
+from aip_agents.agent.system_instruction_context import get_current_date_context as get_current_date_context
+from aip_agents.constants import TEXT_PREVIEW_LENGTH as TEXT_PREVIEW_LENGTH
+from aip_agents.mcp.client import LangchainMCPClient as LangchainMCPClient
 from aip_agents.memory import BaseMemory as BaseMemory, MemoryFactory as MemoryFactory, MemoryMethod as MemoryMethod
 from aip_agents.memory.constants import MemoryDefaults as MemoryDefaults
+from aip_agents.schema.agent import StreamMode as StreamMode
+from aip_agents.schema.hitl import HitlMetadata as HitlMetadata
 from aip_agents.tools.tool_config_injector import CONFIG_SCHEMA_ATTR as CONFIG_SCHEMA_ATTR, TOOL_CONFIG_SCHEMA_ATTR as TOOL_CONFIG_SCHEMA_ATTR, inject_config_methods_into_tool as inject_config_methods_into_tool
 from aip_agents.types import A2AEvent as A2AEvent, A2AStreamEventType as A2AStreamEventType
 from aip_agents.utils import validate_references as validate_references
 from aip_agents.utils.langgraph.tool_managers.a2a_tool_manager import A2AToolManager as A2AToolManager
 from aip_agents.utils.langgraph.tool_managers.delegation_tool_manager import DelegationToolManager as DelegationToolManager
-from aip_agents.utils.logger_manager import LoggerManager as LoggerManager
-from aip_agents.utils.metadata_helper import DefaultStepMessages as DefaultStepMessages, FINAL_AGENT_THINKING_FINISHED_CONTENT as FINAL_AGENT_THINKING_FINISHED_CONTENT, Status as Status
+from aip_agents.utils.logger import get_logger as get_logger
+from aip_agents.utils.metadata.activity_metadata_helper import create_tool_activity_info as create_tool_activity_info
+from aip_agents.utils.metadata_helper import DefaultStepMessages as DefaultStepMessages, Kind as Kind, MetadataFieldKeys as MetadataFieldKeys, Status as Status, end_step_counter_scope as end_step_counter_scope, get_next_step_number as get_next_step_number, start_step_counter_scope as start_step_counter_scope
+from aip_agents.utils.pii import deanonymize_final_response_content as deanonymize_final_response_content
+from aip_agents.utils.sse_chunk_transformer import SSEChunkTransformer as SSEChunkTransformer
 from aip_agents.utils.token_usage_helper import STEP_USAGE_KEY as STEP_USAGE_KEY, TOTAL_USAGE_KEY as TOTAL_USAGE_KEY, USAGE_METADATA_KEY as USAGE_METADATA_KEY
+from collections.abc import AsyncGenerator, Sequence
 from dataclasses import dataclass
 from gllm_core.event import EventEmitter
-from langchain_core.messages import BaseMessage as BaseMessage
+from gllm_core.schema import Chunk as Chunk
 from langchain_core.tools import BaseTool
 from langgraph.graph import StateGraph
 from langgraph.graph.message import add_messages as add_messages
-from langgraph.graph.state import CompiledStateGraph as CompiledStateGraph
-from langgraph.types import Checkpointer as Checkpointer, StreamWriter as StreamWriter
-from typing import Any, AsyncGenerator, Sequence
+from langgraph.graph.state import CompiledStateGraph
+from langgraph.types import Checkpointer
+from typing import Any
 
 logger: Incomplete
 
@@ -31,12 +37,12 @@ logger: Incomplete
 class _StreamingContext:
     """Context object for managing streaming state and configuration."""
     original_query: str
-    augmented_query: str
     graph_input: dict[str, Any]
     config: dict[str, Any]
     memory_user_id: str | None
     current_thread_id: str
     token: Any
+    enable_token_streaming: bool
     final_event_yielded: bool = ...
     pending_artifacts: list | None = ...
     seen_artifact_hashes: set | None = ...
@@ -44,10 +50,11 @@ class _StreamingContext:
     final_state: dict[str, Any] | None = ...
     last_final_content: str | None = ...
     saved_memory: bool = ...
+    is_token_streaming: bool = ...
     def __post_init__(self) -> None:
         """Initialize mutable defaults."""
 
-class BaseLangGraphAgent(BaseAgent, metaclass=abc.ABCMeta):
+class BaseLangGraphAgent(BaseAgent):
     """Base class for LangGraph-based agents with unified tool approach.
 
     Provides core LangGraph functionality including:
@@ -73,6 +80,7 @@ class BaseLangGraphAgent(BaseAgent, metaclass=abc.ABCMeta):
     """
     state_schema: Incomplete
     thread_id_key: Incomplete
+    enable_a2a_token_streaming: Incomplete
     event_emitter: Incomplete
     checkpointer: Incomplete
     memory: BaseMemory | None
@@ -81,8 +89,8 @@ class BaseLangGraphAgent(BaseAgent, metaclass=abc.ABCMeta):
     regular_tools: list[BaseTool]
     mcp_tools: list[BaseTool]
     resolved_tools: list[BaseTool]
-    def __init__(self, name: str, instruction: str, description: str | None = None, model: Any | None = None, tools: Sequence[BaseTool] | None = None, state_schema: type | None = None, thread_id_key: str = 'thread_id', event_emitter: EventEmitter | None = None, checkpointer: Checkpointer | None = None, **kwargs: Any) -> None:
-        """Initialize the BaseLangGraphAgent.
+    def __init__(self, name: str, instruction: str, description: str | None = None, model: Any | None = None, tools: Sequence[BaseTool] | None = None, state_schema: type | None = None, thread_id_key: str = 'thread_id', event_emitter: EventEmitter | None = None, checkpointer: Checkpointer | None = None, enable_a2a_token_streaming: bool = False, **kwargs: Any) -> None:
+        '''Initialize the BaseLangGraphAgent.
 
         Args:
             name: The name of the agent.
@@ -94,8 +102,22 @@ class BaseLangGraphAgent(BaseAgent, metaclass=abc.ABCMeta):
             thread_id_key: Key for thread ID in configuration.
             event_emitter: Optional event emitter for streaming updates.
             checkpointer: Optional checkpointer for conversation persistence.
-            **kwargs: Additional keyword arguments passed to BaseAgent (including tool_configs).
-        """
+            enable_a2a_token_streaming: Enable token-level streaming for A2A responses.
+                - False (default): Stream message-level events only
+                - True: Stream individual tokens plus message-level events
+            **kwargs: Additional keyword arguments passed to BaseAgent (including tool_configs and memory settings).
+                Memory settings include:
+                - memory_backend: Memory backend (e.g., "mem0")
+                - agent_id: Agent identifier for memory scoping
+                - memory_namespace: Memory namespace
+                - save_interaction_to_memory: Whether to save interactions (default True)
+        '''
+    def set_operation_mode(self, mode: str) -> None:
+        '''Set the operation mode for dependency tracking.
+
+        Args:
+            mode: Operation mode - "parallel" (default) or "sequential"
+        '''
     @abstractmethod
     def define_graph(self, graph_builder: StateGraph) -> CompiledStateGraph:
         """Define the specific graph structure for this agent type.
@@ -143,9 +165,7 @@ class BaseLangGraphAgent(BaseAgent, metaclass=abc.ABCMeta):
             Dictionary containing the agent's response.
         """
     async def arun(self, query: str, **kwargs: Any) -> dict[str, Any]:
-        """Asynchronously run the LangGraph agent.
-
-        If MCP configuration exists, connects to the MCP server and registers tools before running.
+        """Asynchronously run the LangGraph agent with lazy MCP initialization.
 
         Args:
             query: The input query for the agent.
@@ -169,16 +189,15 @@ class BaseLangGraphAgent(BaseAgent, metaclass=abc.ABCMeta):
         Yields:
             Chunks of output (strings or dicts) from the streaming response.
         """
-    def add_mcp_server(self, mcp_config: dict[str, dict[str, Any]]) -> None:
-        """Add MCP server configuration.
+    async def cleanup(self) -> None:
+        """Cleanup MCP resources including persistent sessions.
 
-        Args:
-            mcp_config (dict[str, dict[str, Any]]): Dictionary containing MCP server configurations.
+        This method performs best-effort cleanup of MCP client resources.
+        Errors during cleanup are logged but do not raise exceptions to ensure
+        the cleanup process completes gracefully.
         """
     async def arun_a2a_stream(self, query: str, **kwargs: Any) -> AsyncGenerator[dict[str, Any], None]:
-        '''Asynchronously streams the agent\'s response in a generic format for A2A.
-
-        If MCP configuration exists, connects to the MCP server and registers tools before streaming.
+        '''Asynchronously streams the agent\'s response in A2A format.
 
         Args:
             query: The input query for the agent.
@@ -188,3 +207,26 @@ class BaseLangGraphAgent(BaseAgent, metaclass=abc.ABCMeta):
             Dictionaries with "status" and "content" keys.
             Possible statuses: "working", "completed", "failed", "canceled".
         '''
+    async def arun_sse_stream(self, query: str, task_id: str | None = None, context_id: str | None = None, **kwargs: Any) -> AsyncGenerator[dict[str, Any], None]:
+        '''Stream agent response as SSE-compatible chunks.
+
+        This method wraps arun_a2a_stream and transforms output to the normalized
+        dict format matching A2AConnector.astream_to_agent output, enabling direct
+        streaming without A2A server overhead.
+
+        Args:
+            query: The input query for the agent.
+            task_id: Optional task identifier for the stream.
+            context_id: Optional context identifier for the stream.
+            **kwargs: Additional arguments passed to arun_a2a_stream.
+
+        Yields:
+            SSEChunk dicts with normalized structure:
+            - status: "success" | "error"
+            - task_state: "working" | "completed" | "failed" | "canceled"
+            - content: Text content or None
+            - event_type: Always string (never enum)
+            - final: True for terminal events
+            - metadata: Normalized metadata dict
+            - artifacts: Only present when non-empty
+        '''