rasa-pro 3.13.7__py3-none-any.whl → 3.14.0.dev2__py3-none-any.whl

rasa/agents/templates/mcp_task_agent_prompt_template.jinja2

@@ -0,0 +1,13 @@
+You are a helpful assistant that should assist the user in the best possible way.
+
+### Task
+{{ description }}
+
+### Instructions
+* Always make sure to output responses to the user in a clear, helpful format.
+* Always avoid asking multiple questions at once. Ask questions sequentially one at a time and wait for the user's response before proceeding to next question.
+* Always avoid making assumptions about what values to pass into tools. Ask for clarification if a user's request is ambiguous.
+* Strictly avoid making up information or ability to take some action which is not available in `tool` provided.
+
+### Conversation history
+{{ conversation_history }}

rasa/agents/utils.py

@@ -0,0 +1,72 @@
+from typing import List, Optional
+
+from rasa.agents.agent_manager import AgentManager
+from rasa.agents.exceptions import AgentNotFoundException
+from rasa.core.available_agents import (
+    AgentConfig,
+    AgentMCPServerConfig,
+    AvailableAgents,
+)
+from rasa.core.available_endpoints import AvailableEndpoints
+from rasa.shared.agents.utils import get_protocol_type
+from rasa.shared.core.domain import Domain
+from rasa.shared.core.flows import FlowsList
+from rasa.shared.core.flows.steps import CallFlowStep
+
+
+def resolve_agent_config(
+    agent_config: AgentConfig,
+    available_endpoints: AvailableEndpoints,
+) -> Optional[AgentConfig]:
+    if agent_config is None:
+        return None
+
+    connections = agent_config.connections
+    mcp_connections: List[AgentMCPServerConfig] = (
+        connections.mcp_servers
+        if connections and connections.mcp_servers is not None
+        else []
+    )
+
+    for mcp_server in mcp_connections:
+        for mcp_server_endpoint in available_endpoints.mcp_servers or []:
+            if mcp_server_endpoint.name == mcp_server.name:
+                mcp_server.url = mcp_server_endpoint.url
+                mcp_server.type = mcp_server_endpoint.type
+
+    return agent_config
+
+
+async def initialize_agents(
+    flows: FlowsList,
+    domain: Domain,
+    sub_agents: AvailableAgents,
+) -> None:
+    """Iterate over flows and create/connect the referenced agents."""
+    agent_manager: AgentManager = AgentManager()
+    endpoints = AvailableEndpoints.get_instance()
+
+    for flow in flows.underlying_flows:
+        for step in flow.steps:
+            if isinstance(step, CallFlowStep):
+                if flows.flow_by_id(step.call) is not None:
+                    continue
+
+                if step.is_calling_mcp_tool():
+                    # The call step is calling an MCP tool, so we don't need to
+                    # initialize any agent.
+                    continue
+
+                if not step.is_calling_agent():
+                    raise AgentNotFoundException(step.call)
+
+                agent_name = step.call
+                agent_config = sub_agents.get_agent_config(agent_name)
+                resolved_agent_config = resolve_agent_config(agent_config, endpoints)
+                protocol_type = get_protocol_type(step, agent_config)
+
+                await agent_manager.connect_agent(
+                    agent_name,
+                    protocol_type,
+                    resolved_agent_config,
+                )

rasa/api.py

@@ -20,6 +20,7 @@ if TYPE_CHECKING:
 def run(
     model: Text,
     endpoints: Text,
+    sub_agents: Text,
     connector: Optional[Text] = None,
     credentials: Optional[Text] = None,
     **kwargs: Dict[Text, Any],
@@ -29,6 +30,7 @@ def run(
     Args:
         model: Path to model archive.
         endpoints: Path to endpoints file.
+        sub_agents: Path to sub-agents directory.
         connector: Connector which should be use (overwrites `credentials`
         field).
         credentials: Path to channel credentials file.
@@ -38,11 +40,13 @@ def run(
     """
     import rasa.core.run
     import rasa.shared.utils.common
+    from rasa.core.available_agents import AvailableAgents
     from rasa.core.available_endpoints import AvailableEndpoints
     from rasa.shared.constants import DOCS_BASE_URL
     from rasa.shared.utils.cli import print_warning
 
     _endpoints = AvailableEndpoints.get_instance(endpoints)
+    _sub_agents = AvailableAgents.get_instance(sub_agents)
 
     if not connector and not credentials:
         connector = "rest"
@@ -61,6 +65,7 @@ def run(
         channel=connector,
         credentials=credentials,
         endpoints=_endpoints,
+        sub_agents=_sub_agents,
         **kwargs,
     )
 

rasa/cli/arguments/default_arguments.py

@@ -3,6 +3,7 @@ import logging
 from enum import Enum
 from typing import List, Optional, Text, Union
 
+from rasa.core.constants import DEFAULT_SUB_AGENTS
 from rasa.core.persistor import RemoteStorageType, StorageType, parse_remote_storage
 from rasa.shared.constants import (
     DEFAULT_CONFIG_PATH,
@@ -217,3 +218,14 @@ def add_skip_validation_flag(
         action="append",
         help="Skip YAML validation for selected parts of the training data.",
     )
+
+
+def add_sub_agents_param(
+    parser: Union[argparse.ArgumentParser, argparse._ActionsContainer],
+) -> None:
+    parser.add_argument(
+        "--sub-agents",
+        type=str,
+        default=DEFAULT_SUB_AGENTS,
+        help="Directory that specifies sub-agents to use (default: %(default)s).",
+    )

rasa/cli/arguments/run.py

@@ -7,6 +7,7 @@ from rasa.cli.arguments.default_arguments import (
     add_model_param,
     add_remote_storage_param,
     add_skip_validation_flag,
+    add_sub_agents_param,
 )
 from rasa.core import constants
 from rasa.env import (
@@ -24,6 +25,7 @@ def set_run_arguments(parser: argparse.ArgumentParser) -> None:
     add_server_arguments(parser)
     add_inspect_argument(parser)
     add_skip_validation_flag(parser)
+    add_sub_agents_param(parser)
 
 
 def set_run_action_arguments(parser: argparse.ArgumentParser) -> None:

rasa/cli/dialogue_understanding_test.py

@@ -14,6 +14,7 @@ from rasa.cli.arguments.default_arguments import (
     add_endpoint_param,
     add_model_param,
     add_remote_storage_param,
+    add_sub_agents_param,
 )
 from rasa.core.agent import Agent
 from rasa.core.available_endpoints import AvailableEndpoints
@@ -174,6 +175,8 @@ def add_du_test_arguments(parser: argparse.ArgumentParser) -> None:
         ),
     )
 
+    add_sub_agents_param(du_arguments)
+
 
 def ensure_calm_only_bot(agent: Agent) -> None:
     if agent.domain is None or agent.processor is None:
@@ -218,6 +221,7 @@ def execute_dialogue_understanding_tests(args: argparse.Namespace) -> None:
             model_path=args.model,
             model_server=endpoints.model,
             remote_storage=args.remote_storage,
+            sub_agents_path=args.sub_agents,
         )
     except AgentNotReady as error:
         structlogger.error(

rasa/cli/e2e_test.py

@@ -17,6 +17,7 @@ from rasa.cli.arguments.default_arguments import (
     add_endpoint_param,
     add_model_param,
     add_remote_storage_param,
+    add_sub_agents_param,
 )
 from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.exceptions import AgentNotReady
@@ -147,6 +148,8 @@ def add_e2e_test_arguments(parser: argparse.ArgumentParser) -> None:
         help="Directory where to save coverage report to.",
     )
 
+    add_sub_agents_param(parser)
+
 
 def execute_e2e_tests(args: argparse.Namespace) -> None:
     """Run the end-to-end tests.
@@ -192,6 +195,7 @@ def execute_e2e_tests(args: argparse.Namespace) -> None:
             model_server=endpoints.model,
             endpoints=endpoints,
             test_case_path=Path(test_case_path),
+            sub_agents_path=args.sub_agents,
         )
     except AgentNotReady as error:
         structlogger.error(

rasa/cli/inspect.py

@@ -8,6 +8,7 @@ from sanic import Sanic
 from rasa import telemetry
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments import shell as arguments
+from rasa.cli.arguments.default_arguments import add_sub_agents_param
 from rasa.core import constants
 from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.engine.storage.local_model_storage import LocalModelStorage
@@ -44,6 +45,8 @@ def add_subparser(
         "--voice", help="Enable voice", action="store_true", default=False
     )
 
+    add_sub_agents_param(inspect_parser)
+
     # it'd be confusing to expose those arguments to the user,
     # so we remove them
     remove_argument_from_parser(inspect_parser, "--credentials")

rasa/cli/llm_fine_tuning.py

@@ -14,6 +14,7 @@ from rasa.cli.arguments.default_arguments import (
     add_endpoint_param,
     add_model_param,
     add_remote_storage_param,
+    add_sub_agents_param,
 )
 from rasa.cli.e2e_test import (
     RASA_PRO_BETA_FINE_TUNING_RECIPE_ENV_VAR_NAME,
@@ -137,6 +138,8 @@ def add_data_preparation_arguments(parser: argparse.ArgumentParser) -> None:
         default=DEFAULT_INPUT_E2E_TEST_PATH,
     )
 
+    add_sub_agents_param(parser)
+
     add_remote_storage_param(parser)
 
     rephrasing_arguments = parser.add_argument_group("Rephrasing Module")
@@ -376,6 +379,7 @@ def write_params(
             "rephrase_config": rephrase_config,
             "model": args.model,
             "endpoints": args.endpoints,
+            "sub-agents": args.sub_agents,
             "remote-storage": args.remote_storage,
             "train_frac": args.train_frac,
             "output_format": args.output_format,
@@ -420,6 +424,7 @@ def set_up_e2e_test_runner(args: argparse.Namespace) -> E2ETestRunner:
             model_path=args.model,
             model_server=endpoints.model,
             endpoints=endpoints,
+            sub_agents_path=args.sub_agents,
         )
     except AgentNotReady as error:
         structlogger.error(

rasa/cli/run.py

@@ -109,6 +109,7 @@ def run(args: argparse.Namespace) -> None:
     # configured
 
     import rasa.model
+    from rasa.core.available_agents import AvailableAgents
     from rasa.core.available_endpoints import AvailableEndpoints
 
     # start server if remote storage is configured
@@ -123,6 +124,9 @@ def run(args: argparse.Namespace) -> None:
         rasa_run(**vars(args))
         return
 
+    # load sub-agents
+    AvailableAgents.get_instance(args.sub_agents)
+
     # start server if local model found
     args.model = _validate_model_path(args.model, "model", DEFAULT_MODELS_PATH)
     local_model_set = True

rasa/cli/shell.py

@@ -6,6 +6,7 @@ from typing import List
 from rasa import telemetry
 from rasa.cli import SubParsersAction
 from rasa.cli.arguments import shell as arguments
+from rasa.cli.arguments.default_arguments import add_sub_agents_param
 from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.engine.storage.local_model_storage import LocalModelStorage
 from rasa.exceptions import ModelNotFound
@@ -45,6 +46,8 @@ def add_subparser(
         help="Set the conversation ID.",
     )
 
+    add_sub_agents_param(shell_parser)
+
     run_subparsers = shell_parser.add_subparsers()
 
     shell_nlu_subparser = run_subparsers.add_parser(

rasa/cli/train.py

@@ -15,7 +15,7 @@ from rasa.core.nlg.contextual_response_rephraser import ContextualResponseRephra
 from rasa.core.nlg.generator import NaturalLanguageGenerator
 from rasa.core.train import do_compare_training
 from rasa.engine.validation import validate_api_type_config_key_usage
-from rasa.exceptions import ValidationError
+from rasa.exceptions import DetailedRasaException
 from rasa.shared.constants import (
     CONFIG_MANDATORY_KEYS,
     CONFIG_MANDATORY_KEYS_CORE,
@@ -82,7 +82,7 @@ def _check_nlg_endpoint_validity(endpoint: Union[Path, str]) -> None:
                 ContextualResponseRephraser.__name__,
             )
         NaturalLanguageGenerator.create(endpoints.nlg)
-    except ValidationError as e:
+    except DetailedRasaException as e:
         structlogger.error(
             e.code,
             event_info=e.info,

rasa/constants.py

@@ -22,8 +22,14 @@ MINIMUM_COMPATIBLE_VERSION = "3.11.0rc1"
 
 GLOBAL_USER_CONFIG_PATH = os.path.expanduser("~/.config/rasa/global.yml")
 
+# Logging level for external libraries (default: ERROR to reduce noise)
 DEFAULT_LOG_LEVEL_LIBRARIES = "ERROR"
 ENV_LOG_LEVEL_LIBRARIES = "LOG_LEVEL_LIBRARIES"
+
+# MCP (Model Context Protocol) logging configuration
+ENV_LOG_LEVEL_MCP = "LOG_LEVEL_MCP"
+ENV_MCP_LOGGING_ENABLED = "MCP_LOGGING_ENABLED"
+
 ENV_LOG_LEVEL_MATPLOTLIB = "LOG_LEVEL_MATPLOTLIB"
 ENV_LOG_LEVEL_RABBITMQ = "LOG_LEVEL_RABBITMQ"
 ENV_LOG_LEVEL_KAFKA = "LOG_LEVEL_KAFKA"

rasa/core/actions/action.py

@@ -23,11 +23,9 @@ from rasa.core.constants import (
     KEY_IS_COEXISTENCE_ASSISTANT,
     UTTER_SOURCE_METADATA_KEY,
 )
-from rasa.core.nlg.translate import get_translated_buttons, get_translated_text
 from rasa.core.policies.policy import PolicyPrediction
 from rasa.core.utils import add_bot_utterance_metadata
 from rasa.e2e_test.constants import KEY_STUB_CUSTOM_ACTIONS
-from rasa.engine.language import Language
 from rasa.nlu.constants import (
     RESPONSE_SELECTOR_DEFAULT_INTENT,
     RESPONSE_SELECTOR_PREDICTION_KEY,
@@ -49,6 +47,7 @@ from rasa.shared.constants import (
     UTTER_PREFIX,
 )
 from rasa.shared.core.constants import (
+    ACTION_AGENT_REQUEST_USER_INPUT_NAME,
     ACTION_BACK_NAME,
     ACTION_DEACTIVATE_LOOP_NAME,
     ACTION_DEFAULT_ASK_AFFIRMATION_NAME,
@@ -58,6 +57,8 @@ from rasa.shared.core.constants import (
     ACTION_LISTEN_NAME,
     ACTION_METADATA_EXECUTION_ERROR_MESSAGE,
     ACTION_METADATA_EXECUTION_SUCCESS,
+    ACTION_METADATA_MESSAGE_KEY,
+    ACTION_METADATA_TEXT_KEY,
     ACTION_RESET_ROUTING,
     ACTION_RESTART_NAME,
     ACTION_REVERT_FALLBACK_EVENTS_NAME,
@@ -84,7 +85,6 @@ from rasa.shared.core.events import (
     UserUttered,
 )
 from rasa.shared.core.flows import FlowsList
-from rasa.shared.core.flows.constants import KEY_TRANSLATION
 from rasa.shared.core.slot_mappings import (
     SlotFillingManager,
     extract_slot_value,
@@ -116,6 +116,12 @@ def default_actions(action_endpoint: Optional[EndpointConfig] = None) -> List["A
     from rasa.core.actions.two_stage_fallback import TwoStageFallbackAction
     from rasa.dialogue_understanding.patterns.cancel import ActionCancelFlow
     from rasa.dialogue_understanding.patterns.clarify import ActionClarifyFlows
+    from rasa.dialogue_understanding.patterns.continue_interrupted import (
+        ActionAskInterruptedFlowToContinue,
+        ActionCancelInterruptedFlow,
+        ActionContinueInterruptedFlow,
+        ActionSetInterruptedFlows,
+    )
     from rasa.dialogue_understanding.patterns.correction import ActionCorrectFlowSlot
 
     return [
@@ -130,6 +136,7 @@ def default_actions(action_endpoint: Optional[EndpointConfig] = None) -> List["A
         TwoStageFallbackAction(action_endpoint),
         ActionUnlikelyIntent(),
         ActionSendText(),
+        ActionAgentRequestUserInfo(),
         ActionBack(),
         ActionExtractSlots(action_endpoint),
         ActionCancelFlow(),
@@ -142,6 +149,10 @@ def default_actions(action_endpoint: Optional[EndpointConfig] = None) -> List["A
         ActionResetRouting(),
         ActionHangup(),
         ActionRepeatBotMessages(),
+        ActionContinueInterruptedFlow(),
+        ActionSetInterruptedFlows(),
+        ActionCancelInterruptedFlow(),
+        ActionAskInterruptedFlowToContinue(),
     ]
 
 
@@ -251,36 +262,25 @@ def action_for_name_or_text(
     return RemoteAction(action_name_or_text, action_endpoint)
 
 
-def create_bot_utterance(
-    message: Dict[Text, Any], language: Optional[Language] = None
-) -> BotUttered:
-    """Create BotUttered event from message with translation support."""
-    message_copy = copy.deepcopy(message)
-
-    text = get_translated_text(
-        text=message_copy.pop(TEXT, None),
-        translation=message_copy.pop(KEY_TRANSLATION, {}),
-        language=language,
-    )
-
-    buttons = get_translated_buttons(
-        buttons=message_copy.pop(BUTTONS, None), language=language
+def create_bot_utterance(message: Dict[Text, Any]) -> BotUttered:
+    """Create BotUttered event from message."""
+    bot_message = BotUttered(
+        text=message.pop(TEXT, None),
+        data={
+            ELEMENTS: message.pop(ELEMENTS, None),
+            QUICK_REPLIES: message.pop(QUICK_REPLIES, None),
+            BUTTONS: message.pop(BUTTONS, None),
+            # for legacy / compatibility reasons we need to set the image
+            # to be the attachment if there is no other attachment (the
+            # `.get` is intentional - no `pop` as we still need the image`
+            # property to set it in the following line)
+            ATTACHMENT: message.pop(ATTACHMENT, None) or message.get(IMAGE, None),
+            IMAGE: message.pop(IMAGE, None),
+            CUSTOM: message.pop(CUSTOM, None),
+        },
+        metadata=message,
     )
-
-    data = {
-        ELEMENTS: message_copy.pop(ELEMENTS, None),
-        QUICK_REPLIES: message_copy.pop(QUICK_REPLIES, None),
-        BUTTONS: buttons,
-        # for legacy / compatibility reasons we need to set the image
-        # to be the attachment if there is no other attachment (the
-        # `.get` is intentional - no `pop` as we still need the image`
-        # property to set it in the following line)
-        ATTACHMENT: message_copy.pop(ATTACHMENT, None) or message_copy.get(IMAGE, None),
-        IMAGE: message_copy.pop(IMAGE, None),
-        CUSTOM: message_copy.pop(CUSTOM, None),
-    }
-
-    return BotUttered(text=text, data=data, metadata=message_copy)
+    return bot_message
 
 
 class Action:
@@ -393,7 +393,7 @@ class ActionBotResponse(Action):
         message = add_bot_utterance_metadata(
             message, self.utter_action, nlg, domain, tracker
         )
-        return [create_bot_utterance(message, tracker.current_language)]
+        return [create_bot_utterance(message)]
 
     def name(self) -> Text:
         """Returns action name."""
@@ -427,7 +427,7 @@ class ActionEndToEndResponse(Action):
     ) -> List[Event]:
         """Runs action (see parent class for full docstring)."""
         message = {"text": self.action_text}
-        return [create_bot_utterance(message, tracker.current_language)]
+        return [create_bot_utterance(message)]
 
     def event_for_successful_execution(
         self,
@@ -893,7 +893,10 @@ class RemoteAction(Action):
             generated_response = response.pop("response", None)
             if generated_response is not None:
                 draft = await nlg.generate(
-                    generated_response, tracker, output_channel.name(), **response
+                    generated_response,
+                    tracker,
+                    output_channel.name(),
+                    **response,
                 )
                 if not draft:
                     continue
@@ -1068,6 +1071,7 @@ def _revert_rephrasing_events() -> List[Event]:
     ]
 
 
+# TODO: this should be removed, e.g. it uses a hardcoded message and no translation
 class ActionDefaultAskAffirmation(Action):
     """Default implementation which asks the user to affirm his intent.
 
@@ -1119,7 +1123,7 @@ class ActionDefaultAskAffirmation(Action):
             "utter_action": self.name(),
         }
 
-        return [create_bot_utterance(message, tracker.current_language)]
+        return [create_bot_utterance(message)]
 
 
 class ActionDefaultAskRephrase(ActionBotResponse):
@@ -1149,16 +1153,42 @@ class ActionSendText(Action):
         metadata: Optional[Dict[Text, Any]] = None,
     ) -> List[Event]:
         """Runs action. Please see parent class for the full docstring."""
-        fallback = {"text": ""}
+        fallback = {ACTION_METADATA_TEXT_KEY: ""}
         metadata_copy = copy.deepcopy(metadata) if metadata else {}
-        message = metadata_copy.get("message", fallback)
+        message = metadata_copy.get(ACTION_METADATA_MESSAGE_KEY, fallback)
 
         should_send_text = metadata_copy.get("should_send_text", True)
         if should_send_text:
-            return [create_bot_utterance(message, tracker.current_language)]
+            return [create_bot_utterance(message)]
         return []
 
 
+class ActionAgentRequestUserInfo(Action):
+    """Sends a text message to the output channel that requests some user information.
+
+    The difference to `ActionSendText` is that this action will pause the main advancing
+    flow loop until the user responds to the request
+    (`should_predict_another_action` will return False for this action).
+    """
+
+    def name(self) -> Text:
+        return ACTION_AGENT_REQUEST_USER_INPUT_NAME
+
+    async def run(
+        self,
+        output_channel: "OutputChannel",
+        nlg: "NaturalLanguageGenerator",
+        tracker: "DialogueStateTracker",
+        domain: "Domain",
+        metadata: Optional[Dict[Text, Any]] = None,
+    ) -> List[Event]:
+        """Runs action. Please see parent class for the full docstring."""
+        fallback = {ACTION_METADATA_TEXT_KEY: ""}
+        metadata_copy = copy.deepcopy(metadata) if metadata else {}
+        message = metadata_copy.get(ACTION_METADATA_MESSAGE_KEY, fallback)
+        return [create_bot_utterance(message)]
+
+
 class ActionExtractSlots(Action):
     """Default action that runs after each user turn.
 

rasa/core/actions/action_run_slot_rejections.py

@@ -216,6 +216,6 @@ class ActionRunSlotRejections(Action):
             message = add_bot_utterance_metadata(
                 message, utterance, nlg, domain, tracker
             )
-            events.append(create_bot_utterance(message, tracker.current_language))
+            events.append(create_bot_utterance(message))
 
         return events

rasa/core/agent.py

@@ -12,7 +12,9 @@ import aiohttp
 from aiohttp import ClientError
 
 import rasa.shared.utils.io
+from rasa.agents.utils import initialize_agents
 from rasa.core import jobs
+from rasa.core.available_agents import AvailableAgents
 from rasa.core.available_endpoints import AvailableEndpoints
 from rasa.core.channels.channel import OutputChannel, UserMessage
 from rasa.core.constants import DEFAULT_REQUEST_TIMEOUT
@@ -202,6 +204,7 @@ async def load_agent(
     model_server: Optional[EndpointConfig] = None,
     remote_storage: Optional[StorageType] = None,
     endpoints: Optional[AvailableEndpoints] = None,
+    sub_agents: Optional[AvailableAgents] = None,
     loop: Optional[AbstractEventLoop] = None,
 ) -> Agent:
     """Loads agent from server, remote storage or disk.
@@ -211,6 +214,7 @@ async def load_agent(
         model_server: Configuration for a potential server which serves the model.
         remote_storage: Remote storage to use for loading the model.
         endpoints: Endpoint configuration.
+        sub_agents: Sub-agents configuration.
         loop: Optional async loop to pass to broker creation.
 
     Returns:
@@ -281,6 +285,16 @@ async def load_agent(
                 "No valid configuration given to load agent. "
                 "Agent loaded with no model!"
             )
+
+        if agent.processor:
+            flows = await agent.processor.get_flows()
+            domain = agent.processor.domain
+            # CLI commands that accept --sub-agents
+            # all route through this entry point.
+            # With all required data available,
+            # this is the best spot to initialize the subagents.
+            await initialize_agents(flows, domain, sub_agents)
+
         return agent
 
     except Exception as e:
@@ -322,6 +336,7 @@ class Agent:
         remote_storage: Optional[StorageType] = None,
         http_interpreter: Optional[RasaNLUHttpInterpreter] = None,
         endpoints: Optional[AvailableEndpoints] = None,
+        sub_agents: Optional[AvailableAgents] = None,
         privacy_manager: Optional[BackgroundPrivacyManager] = None,
     ):
         """Initializes an `Agent`."""
@@ -334,6 +349,7 @@ class Agent:
         self.action_endpoint = action_endpoint
         self.http_interpreter = http_interpreter
         self.endpoints = endpoints
+        self.sub_agents = sub_agents
 
         self._set_fingerprint(fingerprint)
         self.model_server = model_server