rasa-pro 3.13.6__tar.gz → 3.14.0.dev1__tar.gz

{rasa_pro-3.13.6 → rasa_pro-3.14.0.dev1}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.3
 Name: rasa-pro
-Version: 3.13.6
+Version: 3.14.0.dev1
 Summary: State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development.
 Keywords: nlp,machine-learning,machine-learning-library,bot,bots,botkit,rasa conversational-agents,conversational-ai,chatbot,chatbot-framework,bot-framework
 Author: Rasa Technologies GmbH
 Author-email: hi@rasa.com
 Maintainer: Tom Bocklisch
 Maintainer-email: tom@rasa.com
-Requires-Python: >=3.9.2,<3.12
+Requires-Python: >=3.10,<3.12
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3
@@ -23,7 +23,7 @@ Provides-Extra: spacy
 Provides-Extra: transformers
 Requires-Dist: CacheControl (>=0.14.2,<0.15.0)
 Requires-Dist: PyJWT[crypto] (>=2.8.0,<3.0.0)
-Requires-Dist: SQLAlchemy (>=2.0.41,<2.1.0)
+Requires-Dist: SQLAlchemy (>=2.0.42,<2.1.0)
 Requires-Dist: absl-py (>=2.0,<2.1)
 Requires-Dist: aio-pika (>=8.2.3,<9.4.4)
 Requires-Dist: aiogram (>=3.15,<3.16)
@@ -38,14 +38,14 @@ Requires-Dist: colorama (>=0.4.6,<0.5.0) ; sys_platform == "win32"
 Requires-Dist: colorclass (>=2.2,<2.3)
 Requires-Dist: coloredlogs (>=15,<16)
 Requires-Dist: colorhash (>=2.0,<2.1.0)
-Requires-Dist: confluent-kafka (>=2.10.0,<3.0.0)
+Requires-Dist: confluent-kafka (>=2.11.0,<3.0.0)
 Requires-Dist: cryptography (>=44.0.1)
 Requires-Dist: cvg-python-sdk (>=0.5.1,<0.6.0)
-Requires-Dist: dask (>=2024.7.0,<2024.8.0)
+Requires-Dist: dask (>=2024.8.0,<2024.9.0)
 Requires-Dist: demoji (>=1.1.0,<2.0.0)
 Requires-Dist: diskcache (>=5.6.3,<5.7.0)
 Requires-Dist: dnspython (==2.6.1)
-Requires-Dist: faiss-cpu (>=1.7.4,<2.0.0)
+Requires-Dist: faiss-cpu (>=1.11.0,<1.12.0)
 Requires-Dist: fbmessenger (>=6.0.0,<6.1.0)
 Requires-Dist: github3.py (>=3.2.0,<3.3.0) ; extra == "gh-release-notes"
 Requires-Dist: gitpython (>=3.1.41,<3.2.0) ; extra == "full"
@@ -65,8 +65,9 @@ Requires-Dist: langchain (>=0.2.17,<0.3.0)
 Requires-Dist: langchain-community (>=0.2.19,<0.3.0)
 Requires-Dist: langcodes (>=3.5.0,<4.0.0)
 Requires-Dist: litellm (>=1.69.0,<1.70.0)
-Requires-Dist: matplotlib (>=3.7,<3.8)
+Requires-Dist: matplotlib (>=3.9.4,<3.10.0)
 Requires-Dist: mattermostwrapper (>=2.2,<2.3)
+Requires-Dist: mcp (>=1.12.0,<1.13.0)
 Requires-Dist: networkx (>=3.1,<3.2)
 Requires-Dist: numpy (>=1.26.4,<1.27.0)
 Requires-Dist: onnxruntime (==1.19.2) ; extra == "pii"
@@ -93,13 +94,13 @@ Requires-Dist: python-dateutil (>=2.8.2,<2.9.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: python-engineio (>=4.12.2,<4.13.0)
 Requires-Dist: python-keycloak (>=3.12.0,<4.0.0)
-Requires-Dist: python-socketio (>=5.8,<6)
+Requires-Dist: python-socketio (>=5.13,<6)
 Requires-Dist: pytz (>=2022.7.1,<2023.0)
-Requires-Dist: pyyaml (>=6.0)
+Requires-Dist: pyyaml (>=6.0.2,<6.1.0)
 Requires-Dist: qdrant-client (>=1.9.1,<1.10.0)
 Requires-Dist: questionary (>=1.10.0,<2.1.0)
 Requires-Dist: randomname (>=0.2.1,<0.3.0)
-Requires-Dist: rasa-sdk (==3.13.0)
+Requires-Dist: rasa-sdk (==3.14.0.dev2)
 Requires-Dist: redis (>=4.6.0,<6.0)
 Requires-Dist: regex (>=2024.7.24,<2024.8.0)
 Requires-Dist: requests (>=2.32.3,<2.33.0)
@@ -111,16 +112,16 @@ Requires-Dist: sanic (>=22.12,<22.13)
 Requires-Dist: sanic-cors (>=2.2.0,<2.3.0)
 Requires-Dist: sanic-jwt (>=1.8.0,<2.0.0)
 Requires-Dist: sanic-routing (>=22.8.0,<23.0.0)
-Requires-Dist: scikit-learn (>=1.5.1,<1.6.0)
+Requires-Dist: scikit-learn (>=1.6.1,<1.7.0)
 Requires-Dist: scipy (>=1.13.1,<1.14.0)
 Requires-Dist: sentencepiece[sentencepiece] (>=0.1.99,<0.2.0) ; extra == "transformers" or extra == "full"
 Requires-Dist: sentry-sdk (>=2.8.0,<3)
 Requires-Dist: setuptools (>=78.1.1,<78.2.0)
-Requires-Dist: sklearn-crfsuite (>=0.3.6,<0.4.0)
-Requires-Dist: skops (>=0.10.0,<0.11.0)
+Requires-Dist: sklearn-crfsuite (>=0.5.0,<0.6.0)
+Requires-Dist: skops (>=0.11.0,<0.12.0)
 Requires-Dist: slack-sdk (>=3.27.1,<3.28.0)
 Requires-Dist: spacy (>=3.5.4,<4.0.0) ; extra == "spacy" or extra == "full"
-Requires-Dist: structlog (>=23.1.0,<23.2.0)
+Requires-Dist: structlog (>=25.4.0,<25.5.0)
 Requires-Dist: structlog-sentry (>=2.0.3,<2.1.0)
 Requires-Dist: tarsafe (>=0.0.5,<0.0.6)
 Requires-Dist: tenacity (>=8.4.1,<8.5.0)
@@ -135,7 +136,7 @@ Requires-Dist: tensorflow-metal (==1.1.0) ; (sys_platform == "darwin" and platfo
 Requires-Dist: tensorflow-text (==2.14.0) ; sys_platform != "win32" and platform_machine != "arm64" and platform_machine != "aarch64"
 Requires-Dist: tensorflow_hub (>=0.13.0,<0.14.0)
 Requires-Dist: terminaltables (>=3.1.10,<3.2.0)
-Requires-Dist: tiktoken (>=0.7.0,<0.8.0)
+Requires-Dist: tiktoken (>=0.9.0,<0.10.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)
 Requires-Dist: transformers (>=4.38.2,<4.39.0) ; extra == "transformers" or extra == "full"
 Requires-Dist: twilio (>=8.4,<8.5)

{rasa_pro-3.13.6 → rasa_pro-3.14.0.dev1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "rasa-pro"
-version = "3.13.6"
+version = "3.14.0.dev1"
 description = "State-of-the-art open-core Conversational AI framework for Enterprises that natively leverages generative AI for effortless assistant development."
 authors = [ "Rasa Technologies GmbH <hi@rasa.com>",]
 maintainers = [ "Tom Bocklisch <tom@rasa.com>",]
@@ -123,12 +123,12 @@ target-version = "py39"
 exclude = [ "eggs", ".git", ".pytest_cache", "build", "dist",]
 
 [tool.poetry.dependencies]
-python = ">=3.9.2,<3.12"
+python = ">=3.10,<3.12"
 boto3 = "~1.35.15"
 google-cloud-storage = "~2.14.0"
 azure-storage-blob = ">=12.16.0, <12.17.0"
 requests = "~2.32.3"
-matplotlib = ">=3.7,<3.8"
+matplotlib = "~3.9.4"
 attrs = ">=23.1,<23.2"
 jsonpickle = ">=3.3.0,<3.4"
 redis = ">=4.6.0, <6.0"
@@ -140,7 +140,7 @@ fbmessenger = "~6.0.0"
 pykwalify = ">=1.8,<1.9"
 coloredlogs = ">=15,<16"
 "ruamel.yaml" = ">=0.17.21,<0.17.22"
-pyyaml = ">=6.0"
+pyyaml = "~=6.0.2"
 twilio = ">=8.4,<8.5"
 webexteamssdk = ">=1.6.1,<1.7.0"
 mattermostwrapper = "~2.2"
@@ -159,11 +159,11 @@ websockets = ">=10.4,<11.0"
 aiohttp = ">=3.10,<3.11"
 questionary = ">=1.10.0,<2.1.0"
 prompt-toolkit = "^3.0.28,<3.0.29"
-python-socketio = ">=5.8,<6"
+python-socketio = ">=5.13,<6"
 python-engineio = "~4.12.2" # skip 5.0.0 upgrade
 pydot = "~1.4"
-SQLAlchemy = ">=2.0.41,<2.1.0"
-sklearn-crfsuite = "~0.3.6"
+SQLAlchemy = ">=2.0.42,<2.1.0"
+sklearn-crfsuite = "~0.5.0"
 psycopg2-binary = ">=2.9.10,<2.10.0"
 python-dateutil = "~2.8.2"
 protobuf = "~4.25.8"
@@ -182,10 +182,10 @@ CacheControl = "~0.14.2"
 randomname = ">=0.2.1,<0.3.0"
 pluggy = "^1.2.0"
 slack-sdk = "~3.27.1"
-confluent-kafka = ">=2.10.0,<3.0.0"
+confluent-kafka = ">=2.11.0,<3.0.0"
 portalocker = "^2.7.0"
 pypred = "^0.4.0"
-structlog = ">=23.1.0,<23.2.0"
+structlog = ">=25.4.0,<25.5.0"
 structlog-sentry = ">=2.0.3,<2.1.0"
 langchain = "~0.2.17"
 langchain-community = "~0.2.19"
@@ -208,8 +208,8 @@ hvac = "^1.2.1"
 rich = "^13.4.2"
 cvg-python-sdk = "^0.5.1"
 python-keycloak = "^3.12.0"
-tiktoken = "~0.7.0"
-faiss-cpu = "^1.7.4"
+tiktoken = "~0.9.0"
+faiss-cpu = "~1.11.0"
 pymilvus = ">=2.4.1,<2.4.2" # pinned because 2.4.2 introduces milvus-lite dependency that fails on windows
 qdrant-client = "~1.9.1"
 psutil = "^5.9.5"
@@ -218,15 +218,16 @@ types-protobuf = "4.25.0.20240417"
 litellm = "~1.69.0"
 openpyxl = "^3.1.5"
 diskcache = "~5.6.3"
-skops = "~0.10.0"
+skops = "~0.11.0"
 safetensors = "~0.4.5"
-dask = "~2024.7.0" # later versions drop python 3.9 support
+dask = "~2024.8.0" # later versions drop python 3.9 support
 numpy = "~1.26.4"
 scipy = "~1.13.1"
-scikit-learn = "~1.5.1"
+scikit-learn = "~1.6.1"
 demoji = "^1.1.0"
 azure-identity = "~1.19.0"
 langcodes = "^3.5.0"
+mcp = "~1.12.0"
 
 [[tool.poetry.dependencies.gliner]]
 version = "~0.2.20"
@@ -301,7 +302,7 @@ timeout_func_only = true
 asyncio_mode = "auto"
 
 [tool.poetry.dependencies.rasa-sdk]
-version = "3.13.0"
+version = "3.14.0.dev2"
 allow-prereleases = true
 
 [tool.poetry.dependencies.keras]
@@ -372,17 +373,6 @@ convention = "google"
 "tests/shared/core/test_domain.py" = [ "E501",]
 
 
-
-
-
-
-
-
-
-
-
-
-
 [tool.poetry.group.dev.dependencies]
 ruff = ">=0.3.5,<0.4.0"
 docker = "~7.1.0"

rasa_pro-3.14.0.dev1/rasa/agents/agent_factory.py

@@ -0,0 +1,122 @@
+from typing import Any, ClassVar, Dict, List, Type
+
+import structlog
+
+from rasa.agents.core.agent_protocol import AgentProtocol
+from rasa.agents.core.types import ProtocolType
+from rasa.agents.protocol import A2AAgent, MCPOpenAgent, MCPTaskAgent
+from rasa.core.available_agents import AgentConfig
+from rasa.shared.utils.common import class_from_module_path
+
+structlogger = structlog.get_logger()
+
+
+class AgentFactory:
+    """Factory for creating agent instances based on the protocol type."""
+
+    _protocols: ClassVar[Dict[ProtocolType, Type[AgentProtocol]]] = {
+        ProtocolType.A2A: A2AAgent,
+        ProtocolType.MCP_OPEN: MCPOpenAgent,
+        ProtocolType.MCP_TASK: MCPTaskAgent,
+    }
+
+    @classmethod
+    def create_client(
+        cls, protocol_type: ProtocolType, config: AgentConfig
+    ) -> AgentProtocol:
+        """Create an agent instance based on the protocol type.
+
+        Args:
+            protocol_type: The protocol type of the agent.
+            config: The configuration for the agent as an AgentConfig object.
+
+        Returns:
+            An instance of the agent.
+        """
+        # If the agent is a custom agent, we need to create it from the module
+        if config.configuration and config.configuration.module:
+            agent_class: Type[AgentProtocol] = class_from_module_path(
+                config.configuration.module
+            )
+            if not cls._is_valid_custom_agent(agent_class, protocol_type):
+                raise ValueError(
+                    f"Agent class {agent_class} does not subclass the "
+                    f"{cls._get_agent_class_from_protocol(protocol_type).__name__} "
+                    f"agent class."
+                )
+            structlogger.debug(
+                "agent_factory.create_client.custom_agent",
+                event_info=(
+                    f"Initializing `{agent_class.__name__}` for agent "
+                    f"`{config.agent.name}` with protocol `{protocol_type.value}`"
+                ),
+                agent_name=config.agent.name,
+                protocol_type=protocol_type.value,
+                agent_class=agent_class.__name__,
+            )
+            return agent_class.from_config(config)
+
+        # If the agent is a built-in agent, we need to create it from the protocol class
+        protocol_class = cls._get_agent_class_from_protocol(protocol_type)
+        if protocol_class is None:
+            raise ValueError(
+                f"Unsupported protocol: {protocol_type}. "
+                f"Supported protocols: {cls.get_supported_protocols()}"
+            )
+
+        structlogger.debug(
+            "agent_factory.create_client",
+            event_info=(
+                f"Initializing `{protocol_class.__name__}` for agent "
+                f"`{config.agent.name}` with protocol `{protocol_type.value}`"
+            ),
+            agent_name=config.agent.name,
+            protocol_type=protocol_type.value,
+            agent_class=protocol_class.__name__,
+        )
+        return protocol_class.from_config(config)
+
+    @classmethod
+    def register_protocol(
+        cls, protocol_type: ProtocolType, protocol_class: Type[AgentProtocol]
+    ) -> None:
+        """Register new protocol implementation.
+
+        Args:
+            protocol_type: The protocol type of the agent.
+            protocol_class: The class that implements the protocol.
+        """
+        if cls.is_protocol_supported(protocol_type):
+            raise ValueError(f"Protocol {protocol_type} already registered.")
+        cls._protocols[protocol_type] = protocol_class
+
+    @classmethod
+    def get_supported_protocols(cls) -> List[ProtocolType]:
+        """Get all supported protocol types."""
+        return list(cls._protocols.keys())
+
+    @classmethod
+    def _get_agent_class_from_protocol(
+        cls, protocol_type: ProtocolType
+    ) -> Type[AgentProtocol]:
+        """Get the class that implements the protocol."""
+        if not cls.is_protocol_supported(protocol_type):
+            raise ValueError(
+                f"Unsupported protocol: {protocol_type}. "
+                f"Supported protocols: {cls.get_supported_protocols()}"
+            )
+        return cls._protocols[protocol_type]
+
+    @classmethod
+    def is_protocol_supported(cls, protocol_type: ProtocolType) -> bool:
+        """Check if the protocol is supported."""
+        return protocol_type in cls._protocols
+
+    @classmethod
+    def _is_valid_custom_agent(
+        cls, agent_class: Any, protocol_type: ProtocolType
+    ) -> bool:
+        """Check if the agent class is valid."""
+        return issubclass(
+            agent_class, cls._get_agent_class_from_protocol(protocol_type)
+        )

rasa_pro-3.14.0.dev1/rasa/agents/agent_manager.py

@@ -0,0 +1,162 @@
+from typing import ClassVar, Dict
+
+import structlog
+
+from rasa.agents.agent_factory import AgentFactory
+from rasa.agents.core.agent_protocol import AgentProtocol
+from rasa.agents.core.types import AgentIdentifier, ProtocolType
+from rasa.agents.schemas import AgentInput, AgentOutput
+from rasa.core.available_agents import AgentConfig
+from rasa.shared.agents.utils import make_agent_identifier
+from rasa.utils.singleton import Singleton
+
+structlogger = structlog.get_logger()
+
+
+class AgentManager(metaclass=Singleton):
+    """High-level agent management with protocol abstraction"""
+
+    agents: ClassVar[Dict[AgentIdentifier, AgentProtocol]] = {}
+
+    def _add_agent(
+        self, agent_identifier: AgentIdentifier, agent: AgentProtocol
+    ) -> None:
+        """Add an agent to the manager.
+
+        Args:
+            agent_identifier: The identifier of the agent.
+            agent: The agent to add.
+        """
+        if agent_identifier in self.agents:
+            raise ValueError(f"Agent {agent_identifier} already exists")
+        self.agents[agent_identifier] = agent
+
+    def _remove_agent(self, agent_identifier: AgentIdentifier) -> None:
+        """Remove an agent from the manager.
+
+        Args:
+            agent_identifier: The identifier of the agent.
+
+        Raises:
+            ValueError: If the agent is not connected.
+        """
+        if agent_identifier not in self.agents:
+            raise ValueError(f"Agent {agent_identifier} not connected")
+        del self.agents[agent_identifier]
+
+    def get_agent(self, agent_name: str, protocol_type: ProtocolType) -> AgentProtocol:
+        """Retrieve connected agent instance.
+
+        Args:
+            agent_name: The name of the agent.
+            protocol_type: The protocol type of the agent.
+
+        Returns:
+            The agent.
+
+        Raises:
+            ValueError: If the agent is not connected.
+        """
+        agent_identifier = make_agent_identifier(agent_name, protocol_type)
+        if agent_identifier not in self.agents:
+            raise ValueError(f"Agent {agent_identifier} not connected")
+        return self.agents[agent_identifier]
+
+    async def connect_agent(
+        self, agent_name: str, protocol_type: ProtocolType, config: AgentConfig
+    ) -> None:
+        """Connect to agent using specified protocol, load the default resources
+        and persist the agent to the manager in a ready-to-use state so that
+        it can be used immediately to send messages to the agent.
+
+        Args:
+            agent_name: The name of the agent.
+            protocol_type: The protocol type of the agent.
+            config: The configuration for the agent as an AgentConfig object.
+
+        Raises:
+            ConnectionError: If the agent connection fails.
+        """
+        try:
+            # Create the agent client
+            client = AgentFactory.create_client(protocol_type, config)
+
+            # Connect the agent client
+            await client.connect()
+
+            # Add the agent to the manager
+            agent_identifier = make_agent_identifier(agent_name, protocol_type)
+            self._add_agent(agent_identifier, client)
+            structlogger.info(
+                "agent_manager.connect_agent.success",
+                agent_id=str(agent_identifier),
+                agent_name=agent_name,
+                event_info=f"Connected to agent - `{agent_identifier}` successfully",
+            )
+        except Exception as e:
+            event_info = f"Failed to connect agent {agent_identifier}"
+            structlogger.error(
+                "agent_manager.connect_agent.failed_to_connect",
+                agent_id=str(agent_identifier),
+                event_info=event_info,
+            )
+            raise ConnectionError(e)
+
+    async def run_agent(
+        self, agent_name: str, protocol_type: ProtocolType, context: AgentInput
+    ) -> AgentOutput:
+        """Run an agent, send the input to the agent and return the agent response.
+
+        Args:
+            agent_name: The name of the agent.
+            protocol_type: The protocol type of the agent.
+            context: The input to the agent as an AgentInput object.
+
+        Returns:
+            The response from the agent.
+        """
+        agent = self.get_agent(agent_name, protocol_type)
+
+        # Process input before sending
+        processed_input = await agent.process_input(context)
+
+        # Send message to agent
+        output = await agent.run(processed_input)
+
+        # Process output before returning
+        processed_output = await agent.process_output(output)
+
+        return processed_output
+
+    async def disconnect_agent(
+        self, agent_name: str, protocol_type: ProtocolType
+    ) -> None:
+        """Disconnect agent and remove from agent manager.
+
+        Args:
+            agent_name: The name of the agent.
+            protocol_type: The protocol type of the agent.
+
+        Raises:
+            ValueError: If the agent is not connected.
+            ConnectionError: If the agent disconnection fails.
+        """
+        agent_identifier = make_agent_identifier(agent_name, protocol_type)
+        if agent_identifier not in self.agents:
+            raise ValueError(f"Agent {agent_identifier} not connected")
+        try:
+            await self.get_agent(agent_name, protocol_type).disconnect()
+            self._remove_agent(agent_identifier)
+            structlogger.info(
+                "agent_manager.disconnect_agent.success",
+                agent_id=str(agent_identifier),
+                event_info=f"Disconnected from agent - {agent_identifier} successfully",
+            )
+        except Exception as e:
+            event_info = f"Failed to disconnect agent {agent_identifier}"
+            structlogger.error(
+                "agent_manager.disconnect_agent.failed_to_disconnect",
+                agent_id=str(agent_identifier),
+                event_info=event_info,
+            )
+            raise ConnectionError(e)

rasa_pro-3.14.0.dev1/rasa/agents/constants.py

@@ -0,0 +1,31 @@
+# MCP Tool related constants (primary)
+TOOL_TYPE_KEY = "type"
+TOOL_NAME_KEY = "name"
+TOOL_DESCRIPTION_KEY = "description"
+TOOL_PARAMETERS_KEY = "parameters"
+TOOL_STRICT_KEY = "strict"
+TOOL_TYPE_FUNCTION_KEY = "function"
+
+# MCP Tool related constants (secondary)
+TOOL_ADDITIONAL_PROPERTIES_KEY = "additionalProperties"
+TOOL_PROPERTIES_KEY = "properties"
+TOOL_REQUIRED_KEY = "required"
+TOOL_PROPERTY_TYPE_KEY = "type"
+
+# MCP Message related constants
+KEY_ROLE = "role"
+KEY_CONTENT = "content"
+
+# MCP Tool call related constants
+KEY_TOOL_CALL_ID = "tool_call_id"
+KEY_FUNCTION = "function"
+KEY_NAME = "name"
+KEY_ARGUMENTS = "arguments"
+KEY_ID = "id"
+KEY_TYPE = "type"
+KEY_TOOL_CALLS = "tool_calls"
+
+# Agent output metadata related constants
+AGENT_METADATA_AGENT_RESPONSE_KEY = "agent_response"
+AGENT_METADATA_TOOL_RESULTS_KEY = "tool_results"
+AGENT_METADATA_EXIT_IF_KEY = "exit_if"

rasa_pro-3.14.0.dev1/rasa/agents/core/agent_protocol.py

@@ -0,0 +1,108 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Protocol, runtime_checkable
+
+if TYPE_CHECKING:
+    from rasa.agents.core.types import ProtocolType
+    from rasa.agents.schemas.agent_input import AgentInput
+    from rasa.agents.schemas.agent_output import AgentOutput
+    from rasa.core.available_agents import AgentConfig
+
+
+@runtime_checkable
+class AgentProtocol(Protocol):
+    """
+    Python protocol for interfacing with agent clients that implement the
+    agent communication protocols like MCP, A2A, ACP, etc.
+    """
+
+    @classmethod
+    def from_config(cls, config: AgentConfig) -> AgentProtocol:
+        """
+        Initialize the Agent with the given configuration.
+
+        This class method should be implemented to parse the given
+        configuration and create an instance of an agent.
+
+        Args:
+            config: The configuration for the agent as an AgentConfig object.
+
+        Returns:
+            An instance of the agent.
+        """
+        ...
+
+    @property
+    def protocol_type(self) -> "ProtocolType":
+        """
+        Returns the protocol type of the agent.
+
+        This property should be implemented to return the protocol type of the agent.
+        This is used to determine the type of agent to create in the AgentFactory.
+        """
+        ...
+
+    async def connect(self) -> None:
+        """
+        Establish connection to agent/server.
+
+        This method should be implemented to establish a connection to the agent/server
+        and load any necessary resources.
+        """
+        ...
+
+    async def disconnect(self) -> None:
+        """
+        Close connection to agent/server.
+
+        This method should be implemented to close the connection to the agent/server
+        and release any necessary resources.
+        """
+        ...
+
+    async def process_input(self, input: "AgentInput") -> "AgentInput":
+        """
+        Pre-process the input before sending it to the agent.
+
+        This method should be implemented to pre-process the input before sending it
+        to the agent. This can be used to add any necessary metadata or context to the
+        input, or to filter the input.
+
+        Args:
+            input: The input to the agent as an AgentInput object.
+
+        Returns:
+            The processed input to the agent as an AgentInput object.
+        """
+        ...
+
+    async def run(self, input: "AgentInput") -> "AgentOutput":
+        """
+        Send a message to Agent/server and return response.
+
+        This method should be implemented to send a message to the agent/server and
+        return the response in an AgentOutput object.
+
+        Args:
+            input: The input to the agent as an AgentInput object.
+
+        Returns:
+            The output from the agent as an AgentOutput object.
+        """
+        ...
+
+    async def process_output(self, output: "AgentOutput") -> "AgentOutput":
+        """
+        Post-process the output before returning it to Rasa.
+
+        This method should be implemented to post-process the output before returning
+        it to Rasa. This can be used to add any necessary metadata or context to the
+        output, or to filter the output.
+
+        Args:
+            output: The output from the agent as an AgentOutput object.
+
+        Returns:
+            The processed output from the agent as an AgentOutput object.
+        """
+        ...