ibm-watsonx-orchestrate 1.9.0.dev0__py3-none-any.whl → 1.10.0__py3-none-any.whl

ibm_watsonx_orchestrate/__init__.py

@@ -5,8 +5,7 @@
 
 pkg_name = "ibm-watsonx-orchestrate"
 
-__version__ = "1.9.0.dev0"
-
+__version__ = "1.10.0"
 
 
 from ibm_watsonx_orchestrate.utils.logging.logger import setup_logging

ibm_watsonx_orchestrate/agent_builder/agents/types.py

@@ -77,6 +77,8 @@ class BaseAgentSpec(BaseModel):
     description: Annotated[str, Field(json_schema_extra={"min_length_str":1})]
     context_access_enabled: bool = True
     context_variables: Optional[List[str]] = []
+    voice_configuration_id: Optional[str] = None
+    voice_configuration: Optional[str] = None
 
     def dump_spec(self, file: str) -> None:
         dumped = self.model_dump(mode='json', exclude_unset=True, exclude_none=True)

ibm_watsonx_orchestrate/agent_builder/connections/__init__.py

@@ -18,7 +18,7 @@ from .types import (
     OAuth2AuthCodeCredentials,
     OAuth2ClientCredentials,
     # OAuth2ImplicitCredentials,
-    # OAuth2PasswordCredentials,
+    OAuth2PasswordCredentials,
     OAuthOnBehalfOfCredentials,
     KeyValueConnectionCredentials,
     CONNECTION_KIND_SCHEME_MAPPING,

ibm_watsonx_orchestrate/agent_builder/connections/connections.py

@@ -37,7 +37,7 @@ connection_type_security_schema_map = {
     ConnectionType.OAUTH2_CLIENT_CREDS: ConnectionSecurityScheme.OAUTH2,
     ConnectionType.OAUTH_ON_BEHALF_OF_FLOW: ConnectionSecurityScheme.OAUTH2,
     # ConnectionType.OAUTH2_IMPLICIT: ConnectionSecurityScheme.OAUTH2,
-    # ConnectionType.OAUTH2_PASSWORD: ConnectionSecurityScheme.OAUTH2
+    ConnectionType.OAUTH2_PASSWORD: ConnectionSecurityScheme.OAUTH2
 }
 
 def _clean_env_vars(vars: dict[str:str], requirements: List[str], app_id: str) -> dict[str,str]:

ibm_watsonx_orchestrate/agent_builder/connections/types.py

@@ -8,7 +8,7 @@ class ConnectionKind(str, Enum):
     api_key = 'api_key'
     oauth_auth_code_flow = 'oauth_auth_code_flow'
     # oauth_auth_implicit_flow = 'oauth_auth_implicit_flow'
-    # oauth_auth_password_flow = 'oauth_auth_password_flow'
+    oauth_auth_password_flow = 'oauth_auth_password_flow'
     oauth_auth_client_credentials_flow = 'oauth_auth_client_credentials_flow'
     oauth_auth_on_behalf_of_flow = 'oauth_auth_on_behalf_of_flow'
     key_value = 'key_value'
@@ -34,7 +34,7 @@ class ConnectionPreference(str, Enum):
 class ConnectionAuthType(str, Enum):
     OAUTH2_AUTH_CODE = 'oauth2_auth_code'
     # OAUTH2_IMPLICIT = 'oauth2_implicit'
-    # OAUTH2_PASSWORD = 'oauth2_password'
+    OAUTH2_PASSWORD = 'oauth2_password'
     OAUTH2_CLIENT_CREDS = 'oauth2_client_creds'
     OAUTH_ON_BEHALF_OF_FLOW = 'oauth_on_behalf_of_flow'
 
@@ -65,7 +65,7 @@ class ConnectionType(str, Enum):
     API_KEY_AUTH = ConnectionSecurityScheme.API_KEY_AUTH.value
     OAUTH2_AUTH_CODE = ConnectionAuthType.OAUTH2_AUTH_CODE.value
     # OAUTH2_IMPLICIT = ConnectionAuthType.OAUTH2_IMPLICIT.value
-    # OAUTH2_PASSWORD = ConnectionAuthType.OAUTH2_PASSWORD.value
+    OAUTH2_PASSWORD = ConnectionAuthType.OAUTH2_PASSWORD.value
     OAUTH2_CLIENT_CREDS = ConnectionAuthType.OAUTH2_CLIENT_CREDS.value
     OAUTH_ON_BEHALF_OF_FLOW = ConnectionAuthType.OAUTH_ON_BEHALF_OF_FLOW.value
     KEY_VALUE = ConnectionSecurityScheme.KEY_VALUE.value
@@ -90,7 +90,7 @@ OAUTH_CONNECTION_TYPES = {
     ConnectionType.OAUTH2_AUTH_CODE,
     ConnectionType.OAUTH2_CLIENT_CREDS,
     # ConnectionType.OAUTH2_IMPLICIT,
-    # ConnectionType.OAUTH2_PASSWORD,
+    ConnectionType.OAUTH2_PASSWORD,
     ConnectionType.OAUTH_ON_BEHALF_OF_FLOW,
 }
 
@@ -193,11 +193,15 @@ class OAuth2AuthCodeCredentials(BaseModel):
 #     client_id: str
 #     authorization_url: str
 
-# class OAuth2PasswordCredentials(BaseModel):
-#     client_id: str
-#     client_secret: str
-#     token_url: str
-#     authorization_url: str
+class OAuth2PasswordCredentials(BaseModel):
+    username: str
+    password: str
+    client_id: str
+    client_secret: str
+    token_url: str
+    scope: Optional[str] = None
+    grant_type: str = "password"
+    
 
 class OAuth2ClientCredentials(BaseModel):
     client_id: str
@@ -226,7 +230,7 @@ CREDENTIALS_SET = Union[
     APIKeyAuthCredentials,
     OAuth2AuthCodeCredentials,
     # OAuth2ImplicitCredentials,
-    # OAuth2PasswordCredentials,
+    OAuth2PasswordCredentials,
     OAuth2ClientCredentials,
     OAuthOnBehalfOfCredentials,
     KeyValueConnectionCredentials
@@ -240,7 +244,7 @@ CONNECTION_KIND_SCHEME_MAPPING = {
     ConnectionKind.api_key: ConnectionSecurityScheme.API_KEY_AUTH,
     ConnectionKind.oauth_auth_code_flow: ConnectionSecurityScheme.OAUTH2,
     # ConnectionKind.oauth_auth_implicit_flow: ConnectionSecurityScheme.OAUTH2,
-    # ConnectionKind.oauth_auth_password_flow: ConnectionSecurityScheme.OAUTH2,
+    ConnectionKind.oauth_auth_password_flow: ConnectionSecurityScheme.OAUTH2,
     ConnectionKind.oauth_auth_client_credentials_flow: ConnectionSecurityScheme.OAUTH2,
     ConnectionKind.oauth_auth_on_behalf_of_flow: ConnectionSecurityScheme.OAUTH2,
     ConnectionKind.key_value: ConnectionSecurityScheme.KEY_VALUE,
@@ -250,7 +254,7 @@ CONNECTION_KIND_SCHEME_MAPPING = {
 CONNECTION_KIND_OAUTH_TYPE_MAPPING = {
     ConnectionKind.oauth_auth_code_flow: ConnectionAuthType.OAUTH2_AUTH_CODE,
     # ConnectionKind.oauth_auth_implicit_flow: ConnectionAuthType.OAUTH2_IMPLICIT,
-    # ConnectionKind.oauth_auth_password_flow: ConnectionAuthType.OAUTH2_PASSWORD,
+    ConnectionKind.oauth_auth_password_flow: ConnectionAuthType.OAUTH2_PASSWORD,
     ConnectionKind.oauth_auth_client_credentials_flow: ConnectionAuthType.OAUTH2_CLIENT_CREDS,
     ConnectionKind.oauth_auth_on_behalf_of_flow: ConnectionAuthType.OAUTH_ON_BEHALF_OF_FLOW,
 }

ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py

@@ -11,6 +11,7 @@ class SpecVersion(str, Enum):
 class KnowledgeBaseKind(str, Enum):
     KNOWLEDGE_BASE = "knowledge_base"
 class RetrievalConfidenceThreshold(str, Enum):
+    Off = "Off"
     Lowest = "Lowest"
     Low = "Low"
     High = "High"
@@ -24,6 +25,7 @@ class GeneratedResponseLength(str, Enum):
 
 
 class ResponseConfidenceThreshold(str, Enum):
+    Off = "Off"
     Lowest = "Lowest"
     Low = "Low"
     High = "High"
@@ -86,6 +88,8 @@ class GenerationConfiguration(BaseModel):
     {
         "model_id": "meta-llama/llama-3-1-70b-instruct",
         "prompt_instruction": "When the documents are in different languages, you should respond in english.",
+        "max_docs_passed_to_llm": 10,
+        "retrieval_confidence_threshold": "Lowest",
         "generated_response_length": "Moderate",
         "display_text_no_results_found": "no docs found",
         "display_text_connectivity_issue": "conn failed",
@@ -95,6 +99,7 @@ class GenerationConfiguration(BaseModel):
 
     model_id: Optional[str] = None
     prompt_instruction: Optional[str] = None
+    max_docs_passed_to_llm: Optional[int] = None
     generated_response_length: Optional[GeneratedResponseLength] = None
     display_text_no_results_found: Optional[str] = None
     display_text_connectivity_issue: Optional[str] = None
@@ -198,12 +203,46 @@ class CustomSearchConnection(BaseModel):
     filter: Optional[str] = None
     metadata: Optional[dict] = None
 
+class AstraDBConnection(BaseModel):
+    """
+    example:
+    {
+        "api_endpoint": "https://xxx-us-east-2.apps.astra.datastax.com",
+        "key_space": "default_keyspace",
+        "collection": "search_wa_docs",
+        "embedding_model_id": "sentence-transformers/all-minilm-l12-v2",
+        "port": "443",
+        "filter": "productType: \"boots\"",
+        "limit": 5,
+        "field_mapping": {
+                        "title": "title",
+                        "body": "text",
+                        "url": "some-url"
+                    }
+    }
+    """
+    api_endpoint: str
+    port: Optional[str] = None
+    server_cert: Optional[str] = None
+    keyspace: Optional[str]
+    data_type: str
+    collection: Optional[str]
+    table: Optional[str]
+    index_column: Optional[str]
+    embedding_mode: str
+    embedding_model_id: Optional[str]
+    credentials: dict
+    search_mode: str
+    limit: Optional[int] = 5
+    filter: Optional[str] = None
+    field_mapping: Optional[FieldMapping] = None
+
 class IndexConnection(BaseModel):
     connection_id: Optional[str] = None
     milvus: Optional[MilvusConnection] = None
     elastic_search: Optional[ElasticSearchConnection] = None
     custom_search: Optional[CustomSearchConnection] = None
-
+    astradb: Optional[AstraDBConnection] = None
     
 class ConversationalSearchConfig(BaseModel):
     language: Optional[str] = None
@@ -219,6 +258,10 @@ class KnowledgeBaseBuiltInVectorIndexConfig(BaseModel):
     chunk_size: Optional[int] = None
     chunk_overlap: Optional[int] = None
     limit: Optional[int] = None
+
+class FileUpload(BaseModel):
+    path: str
+    url: Optional[str] = None
     
 class KnowledgeBaseSpec(BaseModel):
     """Schema for a complete knowledge-base."""
@@ -237,4 +280,4 @@ class KnowledgeBaseSpec(BaseModel):
     created_on: Optional[datetime] = None 
     updated_at: Optional[datetime] = None
     # For import/update
-    documents: list[str] = None
+    documents: list[str] | list[FileUpload] = None

ibm_watsonx_orchestrate/agent_builder/toolkits/types.py

@@ -1,27 +1,36 @@
-from typing import List, Dict, Optional
+from typing import List, Dict, Optional, Union
 from enum import Enum
-from pydantic import BaseModel, model_validator
+from pydantic import BaseModel 
 
 class ToolkitKind(str, Enum):
     MCP = "mcp"
 
-class Language(str, Enum):
-    NODE = "node"
-    PYTHON ="python"
-
 class ToolkitSource(str, Enum):
     FILES = "files"
     PUBLIC_REGISTRY = "public-registry"
 
+class ToolkitTransportKind(str, Enum):
+    STREAMABLE_HTTP = "streamable_http"
+    SSE = "sse"
 
+class Language(str, Enum):
+    NODE = "node"
+    PYTHON ="python"
 
-class McpModel(BaseModel):
-    source: str
+class LocalMcpModel(BaseModel):
+    source: ToolkitSource
     command: str
     args: List[str]
     tools: List[str]
     connections: Dict[str, str]
 
+class RemoteMcpModel(BaseModel):
+    server_url: str
+    transport: ToolkitTransportKind
+    tools: List[str]
+    connections: Dict[str, str]
+
+McpModel = Union[LocalMcpModel, RemoteMcpModel]
 
 class ToolkitSpec(BaseModel):
     id: str
@@ -33,10 +42,4 @@ class ToolkitSpec(BaseModel):
     created_by: str
     created_by_username: str
     tools: List[str] | None
-    mcp: McpModel
-
-    @model_validator(mode='after')
-    def validate_tools_and_mcp(self) -> 'ToolkitSpec':
-        if self.mcp.source not in {"files", "public-registry"}:
-            raise ValueError("MCP source must be either 'files' or 'public-registry'.")
-        return self
+    mcp: McpModel

ibm_watsonx_orchestrate/agent_builder/tools/python_tool.py

@@ -216,7 +216,7 @@ def _fix_optional(schema):
         }
         schema.properties[k] = JsonSchemaObject(**combined)
         schema.properties[k].anyOf = None
-
+        
     for k in schema.properties.keys():
         if schema.properties[k].type == 'object':
             schema.properties[k] = _fix_optional(schema.properties[k])

ibm_watsonx_orchestrate/agent_builder/tools/types.py

@@ -152,7 +152,7 @@ class ClientSideToolBinding(BaseModel):
 
 class McpToolBinding(BaseModel):
     server_url: Optional[str] = None
-    source: str
+    source: str | None
     connections: Dict[str, str] | None
 
 class FlowToolBinding(BaseModel):

ibm_watsonx_orchestrate/agent_builder/voice_configurations/__init__.py

@@ -0,0 +1 @@
+from .types import VoiceConfiguration

ibm_watsonx_orchestrate/agent_builder/voice_configurations/types.py

@@ -0,0 +1,98 @@
+import json
+from typing import Annotated, Optional, List, Dict
+from pydantic import BaseModel, Field, model_validator
+
+def _validate_exactly_one_of_fields(object: BaseModel, object_name: str, fields: list[str]):
+  present_fields = [getattr(object,field) for field in fields if getattr(object,field) is not None]
+
+  if len(present_fields) != 1:
+    raise ValueError(f"{object_name} requires exactly one of {','.join(fields)}")
+
+
+def _validate_language_uniqueness(config: BaseModel):
+  if hasattr(config,'language') and hasattr(config,'additional_languages'):
+    if config.language and config.additional_languages and config.language in config.additional_languages:
+      raise ValueError(f"Language '{config.language}' cannot be in both the default language and additional_languages")
+
+
+class WatsonSTTConfig(BaseModel):
+  api_url: Annotated[str, Field(min_length=1,max_length=2048)]
+  api_key: Optional[Annotated[str, Field(min_length=1,max_length=2048)]] = None
+  bearer_token: Optional[Annotated[str, Field(min_length=1,max_length=2048)]] = None
+  model: Annotated[str, Field(min_length=1,max_length=256)]
+
+class EmotechSTTConfig(BaseModel):
+  api_key: Annotated[str,Field(min_length=1,max_length=2048)]
+  api_url: Annotated[str,Field(min_length=1,max_length=2048)]
+
+
+class SpeechToTextConfig(BaseModel):
+  provider: Annotated[str, Field(min_length=1,max_length=128)]
+  watson_stt_config: Optional[WatsonSTTConfig] = None
+  emotech_stt_config: Optional[EmotechSTTConfig] = None
+
+  @model_validator(mode='after')
+  def validate_providers(self):
+    _validate_exactly_one_of_fields(self,'SpeechToTextConfig',['watson_stt_config','emotech_stt_config'])
+    return self
+
+class WatsonTTSConfig(BaseModel):
+  api_url: Annotated[str, Field(min_length=1,max_length=2048)]
+  api_key: Optional[Annotated[str, Field(min_length=1,max_length=2048)]] = None
+  bearer_token: Optional[Annotated[str, Field(min_length=1,max_length=2048)]] = None
+  voice: Annotated[str, Field(min_length=1,max_length=128)]
+  rate_percentage: Optional[int] = None
+  pitch_percentage: Optional[int] = None
+  language: Optional[str] = None
+
+class EmotechTTSConfig(BaseModel):
+  api_url: Annotated[str, Field(min_length=1,max_length=2048)]
+  api_key: Annotated[str, Field(min_length=1,max_length=2048)]
+  voice: Optional[Annotated[str, Field(min_length=1,max_length=128)]]
+
+class TextToSpeechConfig(BaseModel):
+  provider: Annotated[str, Field(min_length=1,max_length=128)]
+  watson_tts_config: Optional[WatsonTTSConfig] = None
+  emotech_tts_config: Optional[EmotechTTSConfig] = None
+
+  @model_validator(mode='after')
+  def validate_providers(self):
+    _validate_exactly_one_of_fields(self,'TextToSpeechConfig',['watson_tts_config','emotech_tts_config'])
+    return self
+
+class AdditionalProperties(BaseModel):
+  speech_to_text: Optional[SpeechToTextConfig] = None
+  text_to_speech: Optional[TextToSpeechConfig] = None
+
+class DTMFInput(BaseModel):
+  inter_digit_timeout_ms: Optional[int] = None
+  termination_key: Optional[str] = None
+  maximum_count: Optional[int] = None
+  ignore_speech: Optional[bool] = None
+
+class AttachedAgent(BaseModel):
+  id: str
+  name: Optional[str] = None
+  display_name: Optional[str] = None
+
+class VoiceConfiguration(BaseModel):
+  name: Annotated[str, Field(min_length=1,max_length=128)]
+  speech_to_text: SpeechToTextConfig
+  text_to_speech: TextToSpeechConfig
+  language: Optional[Annotated[str,Field(min_length=2,max_length=16)]] = None
+  additional_languages: Optional[dict[str,AdditionalProperties]] = None
+  dtmf_input: Optional[DTMFInput] = None
+  voice_configuration_id: Optional[str] = None
+  tenant_id: Optional[Annotated[str, Field(min_length=1,max_length=128)]] = None
+  attached_agents: Optional[list[AttachedAgent]] = None
+
+  @model_validator(mode='after')
+  def validate_language(self):
+    _validate_language_uniqueness(self)
+    return self
+
+  def dumps_spec(self) -> str:
+    dumped = self.model_dump(mode='json', exclude_none=True)
+    return json.dumps(dumped, indent=2)
+
+

ibm_watsonx_orchestrate/cli/commands/agents/agents_command.py

@@ -253,3 +253,23 @@ def export_agent(
 ):  
     agents_controller = AgentsController()
     agents_controller.export_agent(name=name, kind=kind, output_path=output_file, agent_only_flag=agent_only_flag)
+
+@agents_app.command(name="deploy", help="Deploy Agent")
+def deploy_agent(
+    name: Annotated[
+        str,
+        typer.Option("--name", "-n", help="Name of the agent you wish to deploy"),
+    ]
+):
+    agents_controller = AgentsController()
+    agents_controller.deploy_agent(name=name)
+
+@agents_app.command(name="undeploy", help="Undeploy Agent")
+def undeploy_agent(
+    name: Annotated[
+        str,
+        typer.Option("--name", "-n", help="Name of the agent you wish to undeploy"),
+    ]
+):
+    agents_controller = AgentsController()
+    agents_controller.undeploy_agent(name=name)

ibm_watsonx_orchestrate/cli/commands/agents/agents_controller.py

@@ -29,13 +29,17 @@ from ibm_watsonx_orchestrate.client.agents.agent_client import AgentClient, Agen
 from ibm_watsonx_orchestrate.client.agents.external_agent_client import ExternalAgentClient
 from ibm_watsonx_orchestrate.client.agents.assistant_agent_client import AssistantAgentClient
 from ibm_watsonx_orchestrate.client.tools.tool_client import ToolClient
+from ibm_watsonx_orchestrate.client.voice_configurations.voice_configurations_client import VoiceConfigurationsClient
 from ibm_watsonx_orchestrate.utils.exceptions import BadRequest
 from ibm_watsonx_orchestrate.client.connections import get_connections_client
 from ibm_watsonx_orchestrate.client.knowledge_bases.knowledge_base_client import KnowledgeBaseClient
 
-from ibm_watsonx_orchestrate.client.utils import instantiate_client
+from ibm_watsonx_orchestrate.client.utils import instantiate_client, is_local_dev
 from ibm_watsonx_orchestrate.utils.utils import check_file_in_zip
 
+from rich.console import Console
+from rich.progress import Progress, SpinnerColumn, TextColumn
+
 logger = logging.getLogger(__name__)
 
 # Helper generic type for any agent
@@ -197,6 +201,7 @@ def get_app_id_from_conn_id(conn_id: str) -> str:
         exit(1)
     return app_id
 
+
 def get_agent_details(name: str, client: AgentClient | ExternalAgentClient | AssistantAgentClient) -> dict:
     agent_specs = client.get_draft_by_name(name)
     if len(agent_specs) > 1:
@@ -219,6 +224,7 @@ class AgentsController:
         self.assistant_client = None
         self.tool_client = None
         self.knowledge_base_client = None
+        self.voice_configuration_client = None
 
     def get_native_client(self):
         if not self.native_client:
@@ -245,6 +251,11 @@ class AgentsController:
             self.knowledge_base_client = instantiate_client(KnowledgeBaseClient)
         return self.knowledge_base_client
     
+    def get_voice_configuration_client(self):
+        if not self.voice_configuration_client:
+            self.voice_configuration_client = instantiate_client(VoiceConfigurationsClient)
+        return self.voice_configuration_client
+    
     @staticmethod
     def import_agent(file: str, app_id: str) -> List[Agent | ExternalAgent | AssistantAgent]:
         agents = parse_file(file)
@@ -520,6 +531,38 @@ class AgentsController:
                 guideline.tool = name
 
         return ref_agent
+    
+    def get_voice_config_name_from_id(self, voice_config_id: str) -> str | None:
+        client = self.get_voice_configuration_client()
+        config = client.get_by_id(voice_config_id)
+        return config.name if config else None
+
+    def get_voice_config_id_from_name(self, voice_config_name: str) -> str | None:
+        client = self.get_voice_configuration_client()
+        configs = client.get_by_name(voice_config_name)
+
+        if len(configs) == 0:
+            logger.error(f"No voice_configs with the name '{voice_config_name}' found. Failed to get config")
+            sys.exit(1)
+        
+        if len(configs) > 1:
+            logger.error(f"Multiple voice_configs with the name '{voice_config_name}' found. Failed to get config")
+            sys.exit(1)
+        
+        return configs[0].voice_configuration_id
+
+
+    def reference_voice_config(self,agent: Agent):
+        deref_agent = deepcopy(agent)
+        deref_agent.voice_configuration = self.get_voice_config_name_from_id(agent.voice_configuration_id)
+        del deref_agent.voice_configuration_id
+        return deref_agent
+
+    def dereference_voice_config(self,agent: Agent):
+        ref_agent = deepcopy(agent)
+        ref_agent.voice_configuration_id = self.get_voice_config_id_from_name(agent.voice_configuration)
+        del ref_agent.voice_configuration
+        return ref_agent
 
     @staticmethod
     def dereference_app_id(agent: ExternalAgent | AssistantAgent) -> ExternalAgent | AssistantAgent:
@@ -540,7 +583,18 @@ class AgentsController:
             agent.config.connection_id = None
 
         return agent
+    
+    def dereference_common_agent_dependencies(self, agent: AnyAgentT) -> AnyAgentT:
+        if agent.voice_configuration:
+            agent = self.dereference_voice_config(agent)
+
+        return agent  
+
+    def reference_common_agent_dependencies(self, agent: AnyAgentT) -> AnyAgentT:
+        if agent.voice_configuration_id:
+            agent = self.reference_voice_config(agent)
 
+        return agent
 
     def dereference_native_agent_dependencies(self, agent: Agent) -> Agent:
         if agent.collaborators and len(agent.collaborators):
@@ -584,6 +638,8 @@ class AgentsController:
     
     # Convert all names used in an agent to the corresponding ids
     def dereference_agent_dependencies(self, agent: AnyAgentT) -> AnyAgentT:
+
+        agent = self.dereference_common_agent_dependencies(agent)
         if isinstance(agent, Agent):
             return self.dereference_native_agent_dependencies(agent)
         if isinstance(agent, ExternalAgent) or isinstance(agent, AssistantAgent):
@@ -591,6 +647,8 @@ class AgentsController:
 
     # Convert all ids used in an agent to the corresponding names
     def reference_agent_dependencies(self, agent: AnyAgentT) -> AnyAgentT:
+
+        agent = self.reference_common_agent_dependencies(agent)
         if isinstance(agent, Agent):
             return self.reference_native_agent_dependencies(agent)
         if isinstance(agent, ExternalAgent) or isinstance(agent, AssistantAgent):
@@ -1111,3 +1169,114 @@ class AgentsController:
             logger.info(f"Successfully wrote agents and tools to '{output_path}'")
             zip_file_out.close()
 
+
+    def deploy_agent(self, name: str):
+        if is_local_dev():
+            logger.error("Agents cannot be deployed in Developer Edition")
+            sys.exit(1)
+        native_client = self.get_native_client()
+        external_client = self.get_external_client()
+        assistant_client = self.get_assistant_client()
+
+        existing_native_agents = native_client.get_draft_by_name(name)
+        existing_external_agents = external_client.get_draft_by_name(name)
+        existing_assistant_agents = assistant_client.get_draft_by_name(name)
+
+        if len(existing_native_agents) == 0 and (len(existing_external_agents) >= 1 or len(existing_assistant_agents) >= 1):
+            logger.error(f"No native agent found with name '{name}'. Only Native Agents can be deployed to a Live Environment")
+            sys.exit(1)
+        if len(existing_native_agents) > 1:
+            logger.error(f"Multiple native agents with the name '{name}' found. Failed to get agent")
+            sys.exit(1)
+        if len(existing_native_agents) == 0:
+            logger.error(f"No native agents with the name '{name}' found. Failed to get agent")
+            sys.exit(1)
+            
+
+        agent_details = existing_native_agents[0]
+        agent_id = agent_details.get("id")
+
+        environments = native_client.get_environments_for_agent(agent_id)
+
+        live_environment = [env for env in environments if env.get("name") == "live"]
+        if live_environment is None:
+            logger.error("No live environment found for this tenant")
+            sys.exit(1)
+
+        live_env_id = live_environment[0].get("id")
+
+        console = Console()
+        with Progress(
+            SpinnerColumn(spinner_name="dots"),
+            TextColumn("[progress.description]{task.description}"),
+            transient=True,
+            console=console,
+                ) as progress:
+                    progress.add_task(description="Deploying agent to Live envrionment", total=None)
+
+                    status = native_client.deploy(agent_id, live_env_id)
+
+        if status:
+            logger.info(f"Successfully deployed agent {name}")
+        else:
+            logger.error(f"Error deploying agent {name}")
+
+    def undeploy_agent(self, name: str):
+        if is_local_dev():
+            logger.error("Agents cannot be undeployed in Developer Edition")
+            sys.exit(1)
+        
+        native_client = self.get_native_client()
+        external_client = self.get_external_client()
+        assistant_client = self.get_assistant_client()
+
+        existing_native_agents = native_client.get_draft_by_name(name)
+        existing_external_agents = external_client.get_draft_by_name(name)
+        existing_assistant_agents = assistant_client.get_draft_by_name(name)
+
+        if len(existing_native_agents) == 0 and (len(existing_external_agents) >= 1 or len(existing_assistant_agents) >= 1):
+            logger.error(f"No native agent found with name '{name}'. Only Native Agents can be undeployed from a Live Environment")
+            sys.exit(1)
+        if len(existing_native_agents) > 1:
+            logger.error(f"Multiple native agents with the name '{name}' found. Failed to get agent")
+            sys.exit(1)
+        if len(existing_native_agents) == 0:
+            logger.error(f"No native agents with the name '{name}' found. Failed to get agent")
+            sys.exit(1)
+
+        agent_details = existing_native_agents[0]
+        agent_id = agent_details.get("id")
+
+        environments = native_client.get_environments_for_agent(agent_id)
+        live_environment = [env for env in environments if env.get("name") == "live"]
+        if live_environment is None:
+            logger.error("No live environment found for this tenant")
+            sys.exit(1)
+        version_id = live_environment[0].get("current_version")
+
+        if version_id is None:
+            agent_name = agent_details.get("name")
+            logger.error(f"Agent {agent_name} does not exist in a Live environment")
+            sys.exit(1)
+
+        draft_environment = [env for env in environments if env.get("name") == "draft"]
+        if draft_environment is None:
+            logger.error("No draft environment found for this tenant")
+            sys.exit(1)
+        draft_env_id = draft_environment[0].get("id")
+
+        console = Console()
+        with Progress(
+            SpinnerColumn(spinner_name="dots"),
+            TextColumn("[progress.description]{task.description}"),
+            transient=True,
+            console=console,
+                ) as progress:
+                    progress.add_task(description="Undeploying agent to Draft envrionment", total=None)
+
+                    status = native_client.undeploy(agent_id, version_id, draft_env_id)
+        if status:
+            logger.info(f"Successfully undeployed agent {name}")
+        else:
+            logger.error(f"Error undeploying agent {name}")
+