ibm-watsonx-orchestrate 1.12.0b0__tar.gz → 1.13.0b0__tar.gz

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ibm-watsonx-orchestrate
-Version: 1.12.0b0
+Version: 1.13.0b0
 Summary: IBM watsonx.orchestrate SDK
 Author-email: IBM <support@ibm.com>
 License: MIT License
@@ -11,7 +11,7 @@ Requires-Dist: click<8.2.0,>=8.0.0
 Requires-Dist: docstring-parser<1.0,>=0.16
 Requires-Dist: httpx<1.0.0,>=0.28.1
 Requires-Dist: ibm-cloud-sdk-core>=3.24.2
-Requires-Dist: ibm-watsonx-orchestrate-evaluation-framework==1.1.2
+Requires-Dist: ibm-watsonx-orchestrate-evaluation-framework==1.1.3
 Requires-Dist: jsonref==1.1.0
 Requires-Dist: langchain-core<=0.3.63
 Requires-Dist: langsmith<=0.3.45

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/pyproject.toml

@@ -35,7 +35,7 @@ dependencies = [
     "urllib3>=2.2.3",
     "pytz>=2025.2",
     "redis>=6.0.0",
-    "ibm_watsonx_orchestrate_evaluation_framework==1.1.2"
+    "ibm_watsonx_orchestrate_evaluation_framework==1.1.3"
 ]
 
 [project.optional-dependencies]

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/__init__.py

@@ -5,7 +5,8 @@
 
 pkg_name = "ibm-watsonx-orchestrate"
 
-__version__ = "1.12.0b0"
+__version__ = "1.13.0b0"
+
 
 
 from ibm_watsonx_orchestrate.utils.logging.logger import setup_logging

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/agents/types.py

@@ -88,7 +88,7 @@ class BaseAgentSpec(BaseModel):
         dumped = self.model_dump(mode='json', exclude_unset=True, exclude_none=True)
         with open(file, 'w') as f:
             if file.endswith('.yaml') or file.endswith('.yml'):
-                yaml.dump(dumped, f, sort_keys=False)
+                yaml.dump(dumped, f, sort_keys=False, allow_unicode=True)
             elif file.endswith('.json'):
                 json.dump(dumped, f, indent=2)
             else:
@@ -219,8 +219,8 @@ def validate_agent_fields(values: dict) -> dict:
 # ===============================
 
 class ExternalAgentConfig(BaseModel):
-    hidden: bool = False
-    enable_cot: bool = False
+    hidden: Optional[bool] = False
+    enable_cot: Optional[bool] = False
 
 class ExternalAgentSpec(BaseAgentSpec):
     model_config = ConfigDict(arbitrary_types_allowed=True)
@@ -243,8 +243,8 @@ class ExternalAgentSpec(BaseAgentSpec):
         # The get api responds with a flat object with no config
         if values.get("config") is None:
             values["config"] = {}
-            values["config"]["enable_cot"] = values.get("enable_cot", None)
-            values["config"]["hidden"] = values.get("hidden", None)
+            values["config"]["enable_cot"] = values.get("enable_cot", False)
+            values["config"]["hidden"] = values.get("hidden", False)
         return validate_external_agent_fields(values)
 
     @model_validator(mode="after")

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/connections/types.py

@@ -27,6 +27,9 @@ class ConnectionEnvironment(str, Enum):
 
     def __str__(self):
         return self.value
+    
+    def __repr__(self):
+        return repr(self.value)
 
 class ConnectionPreference(str, Enum):
     MEMBER = 'member'
@@ -188,9 +191,9 @@ class ConnectionCredentialsEntryLocation(str, Enum):
         return self.value
 
 class ConnectionCredentialsEntry(BaseModel):
-    key: str
-    value: str
-    location: ConnectionCredentialsEntryLocation
+    key: str = Field(description="The key of the custom credential entry.")
+    value: str = Field(description="The value of the custom credential entry.")
+    location: ConnectionCredentialsEntryLocation = Field(description="How the custom credential should be sent to the server")
 
     def __str__(self):
         return f"<ConnectionCredentialsEntry: {self.location}:{self.key}={self.value}>"
@@ -338,3 +341,31 @@ class IdentityProviderCredentials(BaseOAuthCredentials):
 class ExpectedCredentials(BaseModel):
     app_id: str
     type: ConnectionType | List[ConnectionType]
+
+class FetchConfigAuthTypes(str, Enum):
+    BASIC_AUTH = ConnectionType.BASIC_AUTH.value
+    BEARER_TOKEN = ConnectionType.BEARER_TOKEN.value
+    API_KEY_AUTH = ConnectionType.API_KEY_AUTH.value
+    OAUTH2_AUTH_CODE = ConnectionType.OAUTH2_AUTH_CODE.value
+    OAUTH2_IMPLICIT = 'oauth2_implicit'
+    OAUTH2_PASSWORD = ConnectionType.OAUTH2_PASSWORD.value
+    OAUTH2_CLIENT_CREDS = ConnectionType.OAUTH2_CLIENT_CREDS.value
+    OAUTH_ON_BEHALF_OF_FLOW = ConnectionType.OAUTH_ON_BEHALF_OF_FLOW.value
+    KEY_VALUE = ConnectionType.KEY_VALUE.value
+
+class ConnectionsListEntry(BaseModel):
+    app_id: str = Field(description="A unique identifier for the connection.")
+    auth_type: Optional[FetchConfigAuthTypes] = Field(default=None, description="The kind of auth used by the connections")
+    type: Optional[ConnectionPreference] = Field(default=None, description="The type of the connections. If set to 'team' the credentails will be shared by all users. If set to 'member' each user will have to provide their own credentials")
+    credentials_set: bool = Field(default=False, description="Are the credentials set for the current user. If using OAuth connection types this value will be False unless there isn a stored token from runtime usage")
+
+    def get_row_details(self):
+        auth_type = self.auth_type if self.auth_type else "n/a"
+        type = self.type if self.type else "n/a"
+        credentials_set = "✅" if self.credentials_set else "❌"
+        return [self.app_id, auth_type, type, credentials_set]
+
+class ConnectionsListResponse(BaseModel):
+    non_configured: Optional[List[dict] | str] = None
+    draft: Optional[List[dict] | str] = None
+    live: Optional[List[dict] | str] = None

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py

@@ -3,7 +3,7 @@ from datetime import datetime
 from uuid import UUID
 from enum import Enum
 
-from pydantic import BaseModel, model_validator
+from pydantic import BaseModel, model_validator, Field
 
 
 class SpecVersion(str, Enum):
@@ -295,4 +295,13 @@ class KnowledgeBaseSpec(BaseModel):
     created_on: Optional[datetime] = None 
     updated_at: Optional[datetime] = None
     # For import/update
-    documents: list[str] | list[FileUpload] = None
+    documents: list[str] | list[FileUpload] = None
+
+class KnowledgeBaseListEntry(BaseModel):
+    name: str = Field(description="Name of the knowledge base")
+    description: Optional[str] = Field(description="A description of the content contained in a knowledge base")
+    app_id: Optional[str] = Field(description="The app id for a connection that connects the knowledge base to an external knowledge store")
+    id: Optional[str] = Field(description="Unique identifier of the knowledge base")
+
+    def get_row_details(self):
+        return [self.name, self.description, self.app_id, self.id]

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/models/types.py

@@ -23,6 +23,7 @@ class  ModelProvider(str, Enum):
     STABILITY_AI = 'stability-ai'
     TOGETHER_AI = 'together-ai'
     WATSONX = 'watsonx'
+    X_AI='x-ai'
 
     def __str__(self):
         return self.value
@@ -225,5 +226,21 @@ class ListVirtualModel(BaseModel):
     provider_config: Optional[ProviderConfig] = None
     tags: Optional[List[str]] = None
     model_type: Optional[str] = None
+    connection_id: Optional[str] = None
 
-ANTHROPIC_DEFAULT_MAX_TOKENS = 4096
+class ModelListEntry(BaseModel):
+    name: Optional[str] = Field(default=None, description="Name of the model")
+    description: Optional[str] = Field(default=None, description="A description of the model")
+    is_custom: bool = Field(default=False, description="Is the model a third party model imported into Orchestrate")
+    recommended: bool = Field(default=False, description="Is the model a reccomended model in watsonx. Non-custom models only")
+
+    def get_row_details(self):
+        description = self.description or 'No description provided.'
+        if self.is_custom:
+            return [f"✨️ {self.name}", description]
+        else:
+            name = self.name or "N/A"
+            marker = "★ " if self.recommended else ""
+            return [f"[yellow]{marker}[/yellow]watsonx/{name}", description]
+
+ANTHROPIC_DEFAULT_MAX_TOKENS = 4096

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/toolkits/base_toolkit.py

@@ -18,7 +18,7 @@ class BaseToolkit:
         dumped = self.__toolkit_spec__.model_dump(mode='json', exclude_unset=True, exclude_none=True, by_alias=True)
         with open(file, 'w') as f:
             if file.endswith('.yaml') or file.endswith('.yml'):
-                yaml.dump(dumped, f)
+                yaml.dump(dumped, f, allow_unicode=True)
             elif file.endswith('.json'):
                 json.dump(dumped, f, indent=2)
             else:

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/toolkits/types.py

@@ -1,6 +1,6 @@
 from typing import List, Dict, Optional, Union
 from enum import Enum
-from pydantic import BaseModel 
+from pydantic import BaseModel, Field
 
 class ToolkitKind(str, Enum):
     MCP = "mcp"
@@ -42,4 +42,16 @@ class ToolkitSpec(BaseModel):
     created_by: str
     created_by_username: str
     tools: List[str] | None
-    mcp: McpModel
+    mcp: McpModel
+
+class ToolkitListEntry(BaseModel):
+    name: str = Field(description="The name of the Toolkit")
+    description: Optional[str] = Field(description="The description of the Toolkit")
+    type: str = Field(default="MCP", description="The type of Toolkit.")
+    tools: Optional[List[str]] = Field(description = "A list of tool names for every tool in the Toolkit")
+    app_ids: Optional[List[str]] = Field(description = "A list of connection app_ids showing every connection bound to the Toolkit")
+
+    def get_row_details(self):
+        tools = ", ".join(self.tools) if self.tools else ""
+        app_ids = ", ".join(self.app_ids) if self.app_ids else ""
+        return [self.name, self.description, self.type, tools, app_ids]

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/tools/__init__.py

@@ -2,4 +2,4 @@ from .base_tool import BaseTool
 from .python_tool import tool, PythonTool, get_all_python_tools
 from .openapi_tool import create_openapi_json_tool, create_openapi_json_tool_from_uri, create_openapi_json_tools_from_uri, OpenAPITool, HTTPException
 from .langflow_tool import LangflowTool
-from .types import ToolPermission, JsonSchemaObject, ToolRequestBody, ToolResponseBody, OpenApiSecurityScheme, OpenApiToolBinding, PythonToolBinding, WxFlowsToolBinding, SkillToolBinding, ClientSideToolBinding, ToolBinding, ToolSpec
+from .types import ToolPermission, JsonSchemaObject, ToolRequestBody, ToolResponseBody, OpenApiSecurityScheme, OpenApiToolBinding, PythonToolBinding, WxFlowsToolBinding, SkillToolBinding, ClientSideToolBinding, ToolBinding, ToolSpec, ToolListEntry

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/tools/base_tool.py

@@ -19,7 +19,7 @@ class BaseTool:
         dumped = self.__tool_spec__.model_dump(mode='json', exclude_unset=True, exclude_none=True, by_alias=True)
         with open(file, 'w') as f:
             if file.endswith('.yaml') or file.endswith('.yml'):
-                yaml.dump(dumped, f)
+                yaml.dump(dumped, f, allow_unicode=True)
             elif file.endswith('.json'):
                 json.dump(dumped, f, indent=2)
             else:

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/tools/langflow_tool.py

@@ -2,8 +2,10 @@ import json
 import re
 
 from pydantic import BaseModel
+import rich
 
 from ibm_watsonx_orchestrate.agent_builder.connections.types import ConnectionSecurityScheme
+from ibm_watsonx_orchestrate.langflow.langflow_utils import parse_langflow_model
 from .base_tool import BaseTool
 from .types import LangflowToolBinding, ToolBinding, ToolPermission, ToolRequestBody, ToolResponseBody, ToolSpec
 from ibm_watsonx_orchestrate.utils.exceptions import BadRequest
@@ -14,6 +16,9 @@ LANGFLOW_CHAT_OUTPUT_LABEL = "ChatOutput"
 VALID_NAME_PATTERN = re.compile("^[a-zA-Z](\\w|_)+$")
 
 class LangflowTool(BaseTool):
+  # additional python module requirements for langflow based tools
+  requirements: list[str] = []
+
   def __init__(self,spec: ToolSpec):
     BaseTool.__init__(self,spec=spec)
 
@@ -103,6 +108,59 @@ def create_langflow_tool(
   if not validate_langflow_version(langflow_version):
     raise ValueError(f"Langflow version is below minimum requirements, found '{langflow_version}', miniumum required version '{'.'.join(map(str,MIN_LANGFLOW_VERSION))}'")
   
+  # find all the component in Langflow and display its credential
+  langflow_spec = parse_langflow_model(tool_definition)
+  if langflow_spec:
+    rich.print(f"[bold white]Langflow version used: {langflow_version}[/bold white]")
+    rich.print("Please ensure this flow is compatible with the Langflow version bundled in ADK.")
+    rich.print("\nLangflow components:")
+
+    table = rich.table.Table(show_header=True, header_style="bold white", show_lines=True)
+    column_args = {
+      "ID": {},
+      "Name": {},
+      "Credentials": {},
+      "Requirements": {}
+    }
+    for column in column_args:
+        table.add_column(column,**column_args[column])
+
+    requirements = set()
+    api_key_not_set = False
+    for component in langflow_spec.components:
+      if component.credentials and len(component.credentials) > 0:
+        # create a command separated list with newline
+        component_creds = None
+        for k, v in component.credentials.items():
+          if v is None or v == "":
+            v = 'NOT SET'
+            api_key_not_set = True
+          if component_creds is None:
+            component_creds = f"{k} {v}"
+          else:
+            component_creds += "\n" + f"{k} {v}"
+      else:
+        component_creds = "N/A"
+
+      if component.requirements and len(component.requirements) > 0:
+        # create a command separated list with newline
+        component_req = "\n".join([f"{k}" for k in component.requirements])
+        for r in component.requirements:
+          requirements.add(r)
+      else:
+        component_req = "N/A"
+      table.add_row(component.id,component.name,component_creds,component_req)
+    rich.print(table)
+
+    rich.print("[bold yellow]Tip:[/bold yellow] Langflow tool might require additional python modules.  Identified requirements will be added.")
+    rich.print("[bold yellow]Tip:[/bold yellow] Avoid hardcoding sensitive values. Use Orchestrate connections to manage secrets securely.")
+    if api_key_not_set:
+      rich.print("[bold yellow]Warning:[/bold yellow] Some required api key(s) were not set in the flow. Please adjust the flow to include them.")
+    rich.print("Ensure each credential follows the <app-id>_<variable> naming convention within the Langflow model.")
+
+    for connection in connections:
+        rich.print(f"* Connection: {connection} → Suggested naming: {connection}_<variable>")
+
   spec = ToolSpec(
     name=name,
     description=description,
@@ -121,4 +179,6 @@ def create_langflow_tool(
     )
   )
 
-  return LangflowTool(spec)
+  tool = LangflowTool(spec=spec)
+  tool.requirements = requirements
+  return tool

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/tools/openapi_tool.py

@@ -80,6 +80,7 @@ def create_openapi_json_tool(
         http_success_response_code: int = 200,
         http_response_content_type='application/json',
         name: str = None,
+        display_name: str = None,
         description: str = None,
         permission: ToolPermission = None,
         input_schema: ToolRequestBody = None,
@@ -130,8 +131,10 @@ def create_openapi_json_tool(
         raise BadRequest(
             f"No description provided for tool. {http_method}: {http_path} did not specify a description field, and no description was provided")
 
+    spec_display_name = display_name if display_name else route_spec.get('summary')
     spec = ToolSpec(
         name=spec_name,
+        display_name=spec_display_name,
         description=spec_description,
         permission=spec_permission
     )
@@ -348,6 +351,7 @@ async def create_openapi_json_tool_from_uri(
         http_response_content_type='application/json',
         permission: ToolPermission = ToolPermission.READ_ONLY,
         name: str = None,
+        display_name: str = None,
         description: str = None,
         input_schema: ToolRequestBody = None,
         output_schema: ToolResponseBody = None,
@@ -362,6 +366,7 @@ async def create_openapi_json_tool_from_uri(
     :param http_success_response_code: Which http status code should be considered a successful call (defaults to 200)
     :param http_response_content_type: Which http response type should be considered successful (default to application/json)
     :param name: The name of the resulting tool (used to invoke the tool by the agent)
+    :param display_name: The name of the resulting tool to be displayed
     :param description: The description of the resulting tool (used as the semantic layer to help the agent with tool selection)
     :param permission: Which orchestrate permission level does a user need to have to invoke this tool
     :param input_schema: The JSONSchema of the inputs to the http request
@@ -379,6 +384,7 @@ async def create_openapi_json_tool_from_uri(
         http_response_content_type=http_response_content_type,
         permission=permission,
         name=name,
+        display_name=display_name,
         description=description,
         input_schema=input_schema,
         output_schema=output_schema,

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/tools/types.py

@@ -1,10 +1,12 @@
 from enum import Enum
 from typing import List, Any, Dict, Literal, Optional, Union
+import logging
 
-from pydantic import BaseModel, model_validator, ConfigDict, Field, AliasChoices, ValidationError
+from pydantic import BaseModel, model_validator, ConfigDict, Field, AliasChoices, ValidationError, ValidationInfo
 from ibm_watsonx_orchestrate.utils.exceptions import BadRequest
 from ibm_watsonx_orchestrate.agent_builder.connections import KeyValueConnectionCredentials
 
+logger = logging.getLogger(__name__)
 
 class ToolPermission(str, Enum):
     READ_ONLY = 'read_only'
@@ -117,9 +119,14 @@ class OpenApiToolBinding(BaseModel):
     acknowledgement: Optional[AcknowledgementBinding] = None
 
     @model_validator(mode='after')
-    def validate_openapi_tool_binding(self):
+    def validate_openapi_tool_binding(self, info: ValidationInfo):
+        context = getattr(info, "context", None)
+
         if len(self.servers) != 1:
-            raise BadRequest("OpenAPI definition must include exactly one server")
+            if isinstance(context, str) and context == "list":
+                logger.warning("OpenAPI definition must include exactly one server")
+            else:
+                raise BadRequest("OpenAPI definition must include exactly one server")
         return self
 
 
@@ -259,3 +266,14 @@ class ToolSpec(BaseModel):
             return False
 
         return True
+
+class ToolListEntry(BaseModel):
+    name: str = Field(description="The name of the tool")
+    description: Optional[str] = Field(description="A description of the purpose of the tool")
+    type: Optional[str] = Field(description="The type of the tool"),
+    toolkit: Optional[str] = Field(description="The name of the Toolkit the tool belongs. Empty if the tool is not from a Toolkit"),
+    app_ids: Optional[List[str]] = Field(description="A list of app_ids that show what connections are bound to a tool")
+
+    def get_row_details(self):
+        app_ids = ", ".join(self.app_ids) if self.app_ids else ""
+        return [self.name, self.description, self.type, self.toolkit, app_ids]

ibm_watsonx_orchestrate-1.13.0b0/src/ibm_watsonx_orchestrate/agent_builder/voice_configurations/__init__.py

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

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/agent_builder/voice_configurations/types.py

@@ -95,4 +95,15 @@ class VoiceConfiguration(BaseModel):
     dumped = self.model_dump(mode='json', exclude_none=True)
     return json.dumps(dumped, indent=2)
 
+class VoiceConfigurationListEntry(BaseModel):
+    name: str = Field(description="Name of the voice configuration.")
+    id: str = Field(default=None, description="A unique identifier for the voice configuration.")
+    speech_to_text_provider: Optional[str] = Field("The speech to text service provider.")
+    text_to_speech_provider: Optional[str] = Field("The text to speech service provider.")
+    attached_agents: Optional[List[str]] = Field("A list of agent names that use the voice configuration.")
+
+    def get_row_details(self):
+        attached_agents = ", ".join(self.attached_agents) if self.attached_agents else ""
+        return [self.name, self.id, self.speech_to_text_provider, self.text_to_speech_provider, attached_agents]
+
 

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/cli/commands/agents/agents_controller.py

@@ -18,6 +18,7 @@ from ibm_watsonx_orchestrate.agent_builder.tools.types import ToolSpec
 from ibm_watsonx_orchestrate.cli.commands.tools.tools_controller import import_python_tool, ToolsController
 from ibm_watsonx_orchestrate.cli.commands.knowledge_bases.knowledge_bases_controller import import_python_knowledge_base
 from ibm_watsonx_orchestrate.cli.commands.models.models_controller import import_python_model
+from ibm_watsonx_orchestrate.cli.common import ListFormats, rich_table_to_markdown
 
 from ibm_watsonx_orchestrate.agent_builder.agents import (
     Agent,
@@ -41,23 +42,11 @@ from ibm_watsonx_orchestrate.utils.utils import check_file_in_zip
 from rich.console import Console
 from rich.progress import Progress, SpinnerColumn, TextColumn
 
-from enum import Enum
-
 logger = logging.getLogger(__name__)
 
 # Helper generic type for any agent
 AnyAgentT = TypeVar("AnyAgentT", bound=Agent | ExternalAgent | AssistantAgent)
 
-class AgentListFormats(str, Enum):
-    Table = "table"
-    JSON = "json"
-
-    def __str__(self):
-        return self.value 
-
-    def __repr__(self):
-        return repr(self.value)
-
 
 def import_python_agent(file: str) -> List[Agent | ExternalAgent | AssistantAgent]:
     # Import tools
@@ -858,7 +847,7 @@ class AgentsController:
 
     def _get_all_unique_agent_resources(self, agents: List[Agent], target_attr: str) -> List[str]:
         """
-            Given a list if agents get all the unique values of a certain field
+            Given a list of agents get all the unique values of a certain field
             Example: agent1.tools = [1 ,2 ,3] and agent2.tools = [2, 4, 5] then return [1, 2, 3, 4, 5]
             Example: agent1.id = "123" and agent2.id = "456" then return ["123", "456"]
 
@@ -959,18 +948,14 @@ class AgentsController:
             if tool_names:
                 agent.tools = tool_names
         return new_agents
-    
-    # TODO: Once bulk knowledge base is added create a generaic fucntion as opposed to 3 seperate ones
+
     def _bulk_resolve_agent_knowledge_bases(self, agents: List[Agent]) -> List[Agent]:
         new_agents = agents.copy()
         all_kb_ids = self._get_all_unique_agent_resources(new_agents, "knowledge_base")
+        if not all_kb_ids:
+            return new_agents
 
-        all_kbs = []
-        for id in all_kb_ids:
-            try:
-                all_kbs.append(self.get_knowledge_base_client().get_by_id(id))
-            except:
-                continue
+        all_kbs = self._batch_request_resource(self.get_knowledge_base_client().get_by_ids, all_kb_ids)
 
         kb_lut = self._construct_lut_agent_resource(all_kbs, "id", "name")
         
@@ -1016,24 +1001,7 @@ class AgentsController:
                 agent.app_id = app_id
         return new_agents
 
-    # TODO: Make a shared util
-    def _rich_table_to_markdown(self, table: rich.table.Table) -> str:
-        headers = [column.header for column in table.columns]
-        cols = [[cell for cell in col.cells] for col in table.columns]
-        rows = list(map(list, zip(*cols)))
-
-        # Header row
-        md = "| " + " | ".join(headers) + " |\n"
-        # Separator row
-        md += "| " + " | ".join(["---"] * len(headers)) + " |\n"
-        # # Data rows
-        for row in rows:
-            md += "| " + " | ".join(row) + " |\n"
-        return md
-
-
-
-    def list_agents(self, kind: AgentKind=None, verbose: bool=False, format: AgentListFormats | None = None) -> dict[str, dict] | None:
+    def list_agents(self, kind: AgentKind=None, verbose: bool=False, format: ListFormats | None = None) -> dict[str, dict] | dict[str, str] | None:
         """
         List agents in the active wxo environment
 
@@ -1068,7 +1036,7 @@ class AgentsController:
                 resolved_native_agents = self._bulk_resolve_agent_knowledge_bases(resolved_native_agents)
                 resolved_native_agents = self._bulk_resolve_agent_collaborators(resolved_native_agents)
 
-                if format and format == AgentListFormats.JSON:
+                if format and format == ListFormats.JSON:
                     agents_list = []
                     for agent in resolved_native_agents:
                         agents_list.append(json.loads(agent.dumps_spec()))
@@ -1096,8 +1064,9 @@ class AgentsController:
                         native_table.add_column(column, **column_args[column])
 
                     for agent in resolved_native_agents:
+                        agent_name = self._format_agent_display_name(agent)
                         native_table.add_row(
-                            agent.name,
+                            agent_name,
                             agent.description,
                             agent.llm,
                             agent.style,
@@ -1106,8 +1075,8 @@ class AgentsController:
                             ", ".join(agent.knowledge_base),
                             agent.id,
                         )
-                    if format == AgentListFormats.Table:
-                        output_dictionary["native"] = self._rich_table_to_markdown(native_table)
+                    if format == ListFormats.Table:
+                        output_dictionary["native"] = rich_table_to_markdown(native_table)
                     else:
                         rich.print(native_table)
       
@@ -1123,7 +1092,7 @@ class AgentsController:
             else:
                 resolved_external_agents = self._bulk_resolve_agent_app_ids(external_agents)
                 
-                if format and format == AgentListFormats.JSON:
+                if format and format == ListFormats.JSON:
                     external_agents_list = []
                     for agent in resolved_external_agents:
                         external_agents_list.append(json.loads(agent.dumps_spec()))
@@ -1157,8 +1126,9 @@ class AgentsController:
                         app_id = connections_client.get_draft_by_id(agent.connection_id)
                         resolved_native_agents = self._bulk_resolve_agent_app_ids(external_agents)
 
+                        agent_name = self._format_agent_display_name(agent)
                         external_table.add_row(
-                            agent.name,
+                            agent_name,
                             agent.title,
                             agent.description,
                             ", ".join(agent.tags or []),
@@ -1169,8 +1139,8 @@ class AgentsController:
                             app_id,
                             agent.id
                         )
-                    if format == AgentListFormats.Table:
-                        output_dictionary["external"] = self._rich_table_to_markdown(external_table)
+                    if format == ListFormats.Table:
+                        output_dictionary["external"] = rich_table_to_markdown(external_table)
                     else:
                         rich.print(external_table)
         
@@ -1186,7 +1156,7 @@ class AgentsController:
             else:
                 resolved_external_agents = self._bulk_resolve_agent_app_ids(assistant_agents)
                 
-                if format and format == AgentListFormats.JSON:
+                if format and format == ListFormats.JSON:
                     assistant_agents_list = []
                     for agent in resolved_external_agents:
                         assistant_agents_list.append(json.loads(agent.dumps_spec()))
@@ -1215,8 +1185,9 @@ class AgentsController:
                         assistants_table.add_column(column, **column_args[column])
                     
                     for agent in assistant_agents:
+                        agent_name = self._format_agent_display_name(agent)
                         assistants_table.add_row(
-                            agent.name,
+                            agent_name,
                             agent.title,
                             agent.description,
                             ", ".join(agent.tags or []),
@@ -1227,8 +1198,8 @@ class AgentsController:
                             agent.config.environment_id,
                             agent.id
                         )
-                    if format == AgentListFormats.Table:
-                        output_dictionary["assistant"] = self._rich_table_to_markdown(assistants_table)
+                    if format == ListFormats.Table:
+                        output_dictionary["assistant"] = rich_table_to_markdown(assistants_table)
                     else:
                         rich.print(assistants_table)
 
@@ -1330,7 +1301,7 @@ class AgentsController:
         if agent_only_flag:
             logger.info(f"Exported agent definition for '{name}' to '{output_path}'")
             with open(output_path, 'w') as outfile:
-                yaml.dump(agent_spec_file_content, outfile, sort_keys=False, default_flow_style=False)
+                yaml.dump(agent_spec_file_content, outfile, sort_keys=False, default_flow_style=False, allow_unicode=True)
             return
         
         close_file_flag = False
@@ -1340,7 +1311,7 @@ class AgentsController:
 
         logger.info(f"Exporting agent definition for '{name}'")
         
-        agent_spec_yaml = yaml.dump(agent_spec_file_content, sort_keys=False, default_flow_style=False)
+        agent_spec_yaml = yaml.dump(agent_spec_file_content, sort_keys=False, default_flow_style=False, allow_unicode=True)
         agent_spec_yaml_bytes = agent_spec_yaml.encode("utf-8")
         agent_spec_yaml_file = io.BytesIO(agent_spec_yaml_bytes)
 
@@ -1523,4 +1494,9 @@ class AgentsController:
             logger.info(f"Successfully undeployed agent {name}")
         else:
             logger.error(f"Error undeploying agent {name}")
+    
+    @staticmethod
+    def _format_agent_display_name(agent: AnyAgentT) -> str:
+        return f"{agent.name} ({agent.display_name})" if agent.display_name and agent.name != agent.display_name else agent.name
+
 

{ibm_watsonx_orchestrate-1.12.0b0 → ibm_watsonx_orchestrate-1.13.0b0}/src/ibm_watsonx_orchestrate/cli/commands/connections/connections_command.py

@@ -1,6 +1,6 @@
 import typer
 from typing_extensions import Annotated, List
-from ibm_watsonx_orchestrate.agent_builder.connections.types import ConnectionEnvironment, ConnectionPreference, ConnectionKind, ConnectionCredentialsEntry
+from ibm_watsonx_orchestrate.agent_builder.connections.types import ConnectionEnvironment, ConnectionPreference, ConnectionKind, ConnectionCredentialsEntry, ConnectionSendVia
 from ibm_watsonx_orchestrate.cli.commands.connections.connections_controller import (
     add_connection,
     remove_connection,
@@ -222,7 +222,7 @@ def set_credentials_connection_command(
         )
     ] = None,
     send_via: Annotated[
-        str,
+        ConnectionSendVia,
         typer.Option(
             '--send-via',
             help='For oauth_auth_client_credentials_flow, how the token will be sent to the server. Defaults to using `header`'