ibm-watsonx-orchestrate 1.10.0b1__py3-none-any.whl → 1.10.1__py3-none-any.whl

ibm_watsonx_orchestrate/__init__.py

@@ -5,8 +5,7 @@
 
 pkg_name = "ibm-watsonx-orchestrate"
 
-__version__ = "1.10.0b1"
-
+__version__ = "1.10.1"
 
 
 

ibm_watsonx_orchestrate/agent_builder/connections/connections.py

@@ -1,5 +1,6 @@
 import os
 import logging
+from copy import deepcopy
 from typing import List
 from ibm_watsonx_orchestrate.agent_builder.connections.types import (
     BasicAuthCredentials,
@@ -67,7 +68,8 @@ def _clean_env_vars(vars: dict[str:str], requirements: List[str], app_id: str) -
     return required_env_vars
 
 def _build_credentials_model(credentials_type: type[CREDENTIALS], vars: dict[str,str], base_prefix: str) -> type[CREDENTIALS]:
-    requirements = connection_type_requirements_mapping[credentials_type]
+    requirements_lut = deepcopy(connection_type_requirements_mapping)
+    requirements = requirements_lut[credentials_type]
 
     if requirements:
         requirements.append("url")
@@ -101,7 +103,8 @@ def _get_credentials_model(connection_type: ConnectionSecurityScheme, app_id: st
     
     credentials_type = CONNECTION_TYPE_CREDENTIAL_MAPPING[connection_type]
 
-    requirements = connection_type_requirements_mapping.get(credentials_type)
+    requirements_lut = deepcopy(connection_type_requirements_mapping)
+    requirements = requirements_lut.get(credentials_type)
     if requirements:
         variables = _clean_env_vars(vars=variables, requirements=requirements, app_id=app_id)
 

ibm_watsonx_orchestrate/agent_builder/connections/types.py

@@ -165,6 +165,53 @@ class ConnectionConfiguration(BaseModel):
             raise ValueError("Connection of type 'key_value' cannot be configured at the 'member' level. Key value connections must be of type 'team'")
         return self
 
+class ConnectionCredentialsEntryLocation(str, Enum):
+    BODY = 'body'
+    HEADER = 'header',
+    QUERY = 'query'
+
+    def __str__(self):
+        return self.value
+
+class ConnectionCredentialsEntry(BaseModel):
+    key: str
+    value: str
+    location: ConnectionCredentialsEntryLocation
+
+    def __str__(self):
+        return f"<ConnectionCredentialsEntry: {self.location}:{self.key}={self.value}>"
+
+class BaseOAuthCredentials(BaseModel):
+    custom_token_query: Optional[dict] = None
+    custom_token_header: Optional[dict] = None
+    custom_token_body: Optional[dict] = None
+    custom_auth_query: Optional[dict] = None
+
+class ConnectionCredentialsCustomFields(BaseOAuthCredentials):
+    def add_field(self, entry: ConnectionCredentialsEntry, is_token:bool=True) -> None:
+        match entry.location:
+            case ConnectionCredentialsEntryLocation.HEADER:
+                if not is_token:
+                    return
+                attribute = "custom_token_header"
+            case ConnectionCredentialsEntryLocation.BODY:
+                if not is_token:
+                    return
+                attribute = "custom_token_body"
+            case ConnectionCredentialsEntryLocation.QUERY:
+                if is_token:
+                     attribute = "custom_token_query"
+                else:
+                    attribute = "custom_auth_query"
+            case _:
+                return
+        
+        fields = getattr(self, attribute)
+        if not fields:
+            setattr(self, attribute, {})
+            fields = getattr(self, attribute)
+        fields[entry.key] = entry.value
+
 class BasicAuthCredentials(BaseModel):
     username: str
     password: str
@@ -182,7 +229,7 @@ class OAuth2TokenCredentials(BaseModel):
     access_token: str
     url: Optional[str] = None
 
-class OAuth2AuthCodeCredentials(BaseModel):
+class OAuth2AuthCodeCredentials(BaseOAuthCredentials):
     client_id: str
     client_secret: str
     token_url: str
@@ -193,7 +240,7 @@ class OAuth2AuthCodeCredentials(BaseModel):
 #     client_id: str
 #     authorization_url: str
 
-class OAuth2PasswordCredentials(BaseModel):
+class OAuth2PasswordCredentials(BaseOAuthCredentials):
     username: str
     password: str
     client_id: str
@@ -203,7 +250,7 @@ class OAuth2PasswordCredentials(BaseModel):
     grant_type: str = "password"
     
 
-class OAuth2ClientCredentials(BaseModel):
+class OAuth2ClientCredentials(BaseOAuthCredentials):
     client_id: str
     client_secret: str
     token_url: str
@@ -211,7 +258,7 @@ class OAuth2ClientCredentials(BaseModel):
     send_via: ConnectionSendVia = ConnectionSendVia.HEADER
     grant_type: str = "client_credentials"
 
-class OAuthOnBehalfOfCredentials(BaseModel):
+class OAuthOnBehalfOfCredentials(BaseOAuthCredentials):
     client_id: str
     access_token_url: str
     grant_type: str
@@ -267,7 +314,7 @@ CONNECTION_TYPE_CREDENTIAL_MAPPING = {
     ConnectionSecurityScheme.KEY_VALUE: KeyValueConnectionCredentials,
 }
 
-class IdentityProviderCredentials(BaseModel):
+class IdentityProviderCredentials(BaseOAuthCredentials):
     idp_url: str = Field(validation_alias=AliasChoices('idp_url', 'url'), serialization_alias='idp_url')
     client_id: str
     client_secret: str
@@ -276,4 +323,4 @@ class IdentityProviderCredentials(BaseModel):
 
 class ExpectedCredentials(BaseModel):
     app_id: str
-    type: ConnectionType | List[ConnectionType]
+    type: ConnectionType | List[ConnectionType]

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
+from ibm_watsonx_orchestrate.agent_builder.connections.types import ConnectionEnvironment, ConnectionPreference, ConnectionKind, ConnectionCredentialsEntry
 from ibm_watsonx_orchestrate.cli.commands.connections.connections_controller import (
     add_connection,
     remove_connection,
@@ -8,7 +8,9 @@ from ibm_watsonx_orchestrate.cli.commands.connections.connections_controller imp
     import_connection,
     configure_connection,
     set_credentials_connection,
-    set_identity_provider_connection
+    set_identity_provider_connection,
+    token_entry_connection_credentials_parse,
+    auth_entry_connection_credentials_parse
 )
 
 connections_app = typer.Typer(no_args_is_help=True)
@@ -244,6 +246,22 @@ def set_credentials_connection_command(
             help="For key_value, a key value pair in the form '<key>=<value>'. Multiple values can be passed using `-e key1=value1 -e key2=value2`"
         )
     ] = None,
+    token_entries: Annotated[
+        List[ConnectionCredentialsEntry],
+        typer.Option(
+            '--token-entries', "-t",
+            parser=token_entry_connection_credentials_parse,
+            help="Custom field options for oauth types token request, a key value location option in the form 'location:<key>=<value>' or '<key>=<value>' with location defaulting to 'header'. Multiple values can be passed using `-t key1=value1 -t location:key2=value2`"
+        )
+    ] = None,
+    auth_entries: Annotated[
+        List[ConnectionCredentialsEntry],
+        typer.Option(
+            '--auth-entries',
+            parser=auth_entry_connection_credentials_parse,
+            help="Custom field options for oauth_auth_code_flow auth server request, a key value location option in the form 'location:<key>=<value>' or '<key>=<value>' with location defaulting to 'query'. Note only 'query' is a valid location. Multiple values can be passed using `--auth-entries key1=value1 --auth-entries location:key2=value2`"
+        )
+    ] = None,
 ):
     set_credentials_connection(
         app_id=app_id,
@@ -259,7 +277,9 @@ def set_credentials_connection_command(
         auth_url=auth_url,
         grant_type=grant_type,
         scope=scope,
-        entries=entries
+        entries=entries,
+        token_entries=token_entries,
+        auth_entries=auth_entries
     )
 
 @connections_app.command(name="set-identity-provider")
@@ -311,6 +331,14 @@ def set_identity_provider_connection_command(
             help='The grant-type of the the identity provider'
         )
     ],
+    token_entries: Annotated[
+        List[ConnectionCredentialsEntry],
+        typer.Option(
+            '--token-entries', "-t",
+            parser=token_entry_connection_credentials_parse,
+            help="Custom field options for oauth types token request, a key value location option in the form 'location:<key>=<value>' or '<key>=<value>' with location defaulting to 'header'. Multiple values can be passed using `-t key1=value1 -t location:key2=value2`"
+        )
+    ] = None,
 ):
     set_identity_provider_connection(
         app_id=app_id,
@@ -319,5 +347,6 @@ def set_identity_provider_connection_command(
         client_id=client_id,
         client_secret=client_secret,
         scope=scope,
-        grant_type=grant_type
+        grant_type=grant_type,
+        token_entries=token_entries
     )

ibm_watsonx_orchestrate/cli/commands/connections/connections_controller.py

@@ -27,7 +27,10 @@ from ibm_watsonx_orchestrate.agent_builder.connections.types import (
     KeyValueConnectionCredentials,
     CREDENTIALS,
     IdentityProviderCredentials,
-    OAUTH_CONNECTION_TYPES
+    OAUTH_CONNECTION_TYPES,
+    ConnectionCredentialsEntryLocation,
+    ConnectionCredentialsEntry,
+    ConnectionCredentialsCustomFields
 
 )
 
@@ -167,6 +170,13 @@ def _validate_connection_params(type: ConnectionType, **args) -> None:
             f"Missing flags --grant-type is required for type {type}"
         )
     
+    if type != ConnectionType.OAUTH2_AUTH_CODE and (
+        args.get('auth_entries')
+    ):
+        raise typer.BadParameter(
+            f"The flag --auth-entries is only supported by type {type}"
+        )
+    
 
 def _parse_entry(entry: str) -> dict[str,str]:
     split_entry = entry.split('=', 1)
@@ -176,6 +186,19 @@ def _parse_entry(entry: str) -> dict[str,str]:
         exit(1)
     return {split_entry[0]: split_entry[1]}
 
+def _get_oauth_custom_fields(token_entries: List[ConnectionCredentialsEntry] | None, auth_entries: List[ConnectionCredentialsEntry] | None) -> dict:
+    custom_fields = ConnectionCredentialsCustomFields()
+
+    if token_entries:
+        for entry in token_entries:
+            custom_fields.add_field(entry, is_token=True)
+    
+    if auth_entries:
+        for entry in auth_entries:
+            custom_fields.add_field(entry, is_token=False)
+    
+    return custom_fields.model_dump(exclude_none=True)
+
 def _get_credentials(type: ConnectionType, **kwargs):
     match type:
         case ConnectionType.BASIC_AUTH:
@@ -192,18 +215,21 @@ def _get_credentials(type: ConnectionType, **kwargs):
                 api_key=kwargs.get("api_key")
             )
         case ConnectionType.OAUTH2_AUTH_CODE:
+            custom_fields = _get_oauth_custom_fields(kwargs.get("token_entries"), kwargs.get("auth_entries"))
             return OAuth2AuthCodeCredentials(
                 authorization_url=kwargs.get("auth_url"),
                 client_id=kwargs.get("client_id"),
                 client_secret=kwargs.get("client_secret"),
                 token_url=kwargs.get("token_url"),
-                scope=kwargs.get("scope")
+                scope=kwargs.get("scope"),
+                **custom_fields
             )
         case ConnectionType.OAUTH2_CLIENT_CREDS:
             # using filtered args as default values will not be set if 'None' is passed, causing validation errors
             keys = ["client_id","client_secret","token_url","grant_type","send_via", "scope"]
             filtered_args = { key_name: kwargs[key_name] for key_name in keys if kwargs.get(key_name) }
-            return OAuth2ClientCredentials(**filtered_args)
+            custom_fields = _get_oauth_custom_fields(kwargs.get("token_entries"), kwargs.get("auth_entries"))
+            return OAuth2ClientCredentials(**filtered_args, **custom_fields)
         # case ConnectionType.OAUTH2_IMPLICIT:
         #     return OAuth2ImplicitCredentials(
         #         authorization_url=kwargs.get("auth_url"),
@@ -212,13 +238,16 @@ def _get_credentials(type: ConnectionType, **kwargs):
         case ConnectionType.OAUTH2_PASSWORD:
             keys = ["username", "password", "client_id","client_secret","token_url","grant_type", "scope"]
             filtered_args = { key_name: kwargs[key_name] for key_name in keys if kwargs.get(key_name) }
-            return OAuth2PasswordCredentials(**filtered_args)
+            custom_fields = _get_oauth_custom_fields(kwargs.get("token_entries"), kwargs.get("auth_entries"))
+            return OAuth2PasswordCredentials(**filtered_args, **custom_fields)
         
         case ConnectionType.OAUTH_ON_BEHALF_OF_FLOW:
+            custom_fields = _get_oauth_custom_fields(kwargs.get("token_entries"), kwargs.get("auth_entries"))
             return OAuthOnBehalfOfCredentials(
                 client_id=kwargs.get("client_id"),
                 access_token_url=kwargs.get("token_url"),
-                grant_type=kwargs.get("grant_type")
+                grant_type=kwargs.get("grant_type"),
+                **custom_fields
             )
         case ConnectionType.KEY_VALUE:
             env = {}
@@ -231,6 +260,23 @@ def _get_credentials(type: ConnectionType, **kwargs):
         case _:
             raise ValueError(f"Invalid type '{type}' selected")
 
+def _connection_credentials_parse_entry(text: str, default_location: ConnectionCredentialsEntryLocation) -> ConnectionCredentialsEntry:
+    location_kv_pair = text.split(":", 1)
+    key_value = location_kv_pair[-1]
+    location = location_kv_pair[0] if len(location_kv_pair)>1 else default_location
+
+    valid_locations = [item.value for item in ConnectionCredentialsEntryLocation]
+    if location not in valid_locations:
+        raise typer.BadParameter(f"The provided location '{location}' is not in the allowed values {valid_locations}.")
+
+    key_value_pair = key_value.split('=', 1)
+    if len(key_value_pair) != 2:
+        message = f"The entry '{text}' is not in the expected form '<location>:<key>=<value>' or '<key>=<value>'"
+        raise typer.BadParameter(message)
+    key, value = key_value_pair[0], key_value_pair[1]
+    
+    return ConnectionCredentialsEntry(key=key, value=value, location=location)
+    
 
 def add_configuration(config: ConnectionConfiguration) -> None:
     client = get_connections_client()
@@ -524,5 +570,15 @@ def set_identity_provider_connection(
         logger.error(f"Cannot set Identity Provider when 'sso' is false in configuration. Please enable sso for connection '{app_id}' in environment '{environment}' and try again.")
         sys.exit(1)
 
-    idp = IdentityProviderCredentials.model_validate(kwargs)
+    custom_fields = _get_oauth_custom_fields(token_entries=kwargs.get("token_entries"), auth_entries=None)
+    idp = IdentityProviderCredentials(**kwargs, **custom_fields)
     add_identity_provider(app_id=app_id, environment=environment, idp=idp)
+
+def token_entry_connection_credentials_parse(text: str) -> ConnectionCredentialsEntry:
+    return _connection_credentials_parse_entry(text=text, default_location=ConnectionCredentialsEntryLocation.HEADER)
+    
+def auth_entry_connection_credentials_parse(text: str) -> ConnectionCredentialsEntry:
+    entry = _connection_credentials_parse_entry(text=text, default_location=ConnectionCredentialsEntryLocation.QUERY)
+    if entry.location != ConnectionCredentialsEntryLocation.QUERY:
+        raise typer.BadParameter(f"Only location '{ConnectionCredentialsEntryLocation.QUERY}' is supported for --auth-entry")
+    return entry

ibm_watsonx_orchestrate/cli/commands/server/server_command.py

@@ -191,9 +191,6 @@ def get_default_registry_env_vars_by_dev_edition_source(default_env: dict, user_
             parsed = urlparse(wo_url)
             hostname = parsed.hostname
             
-            if not hostname or not hostname.startswith("api."):
-                raise ValueError(f"Invalid WO_INSTANCE URL: '{wo_url}'. It should starts with 'api.'")
-            
             registry_url = f"registry.{hostname[4:]}/cp/wxo-lite"
         else:
             raise ValueError(f"Unknown value for developer edition source: {source}. Must be one of ['internal', 'myibm', 'orchestrate'].")

ibm_watsonx_orchestrate/cli/commands/server/types.py

@@ -1,5 +1,6 @@
 import logging
 import sys
+import uuid
 from enum import Enum
 from pydantic import BaseModel, model_validator, ConfigDict
 
@@ -43,9 +44,6 @@ class WatsonXAIEnvConfig(BaseModel):
         if not config.get("WATSONX_SPACE_ID") and not config.get("WATSONX_APIKEY"):
              raise ValueError("Missing configuration requirements 'WATSONX_SPACE_ID' and 'WATSONX_APIKEY'")
         
-        if config.get("WATSONX_SPACE_ID") and not config.get("WATSONX_APIKEY"):
-            logger.error("Cannot use env var 'WATSONX_SPACE_ID' without setting the corresponding 'WATSONX_APIKEY'")
-            sys.exit(1)
         
         if not config.get("WATSONX_SPACE_ID") and config.get("WATSONX_APIKEY"):
             logger.error("Cannot use env var 'WATSONX_APIKEY' without setting the corresponding 'WATSONX_SPACE_ID'")
@@ -54,6 +52,12 @@ class WatsonXAIEnvConfig(BaseModel):
         config["USE_SAAS_ML_TOOLS_RUNTIME"] = False
         return config
 
+def is_valid_uuid(value) -> bool:
+    try:
+        uuid.UUID(str(value))
+        return True
+    except (ValueError, TypeError, AttributeError):
+        return False
 
 class ModelGatewayEnvConfig(BaseModel):
     WO_API_KEY: str | None = None
@@ -84,7 +88,10 @@ class ModelGatewayEnvConfig(BaseModel):
         if not config.get("AUTHORIZATION_URL"):
             inferred_auth_url = AUTH_TYPE_DEFAULT_URL_MAPPING.get(auth_type)
             if not inferred_auth_url:
-                logger.error(f"No 'AUTHORIZATION_URL' found. Auth type '{auth_type}' does not support defaulting. Please set the 'AUTHORIZATION_URL' explictly")
+                if auth_type == WoAuthType.CPD:
+                    inferred_auth_url = config.get("WO_INSTANCE") + '/icp4d-api/v1/authorize'   
+                else: 
+                    logger.error(f"No 'AUTHORIZATION_URL' found. Auth type '{auth_type}' does not support defaulting. Please set the 'AUTHORIZATION_URL' explictly")
                 sys.exit(1)
             config["AUTHORIZATION_URL"] = inferred_auth_url
         
@@ -101,6 +108,7 @@ class ModelGatewayEnvConfig(BaseModel):
                 sys.exit(1)
         
         config["USE_SAAS_ML_TOOLS_RUNTIME"] = True
-        # Fake (but valid) UUIDv4 for knowledgebase check
-        config["WATSONX_SPACE_ID"] = "aaaaaaaa-aaaa-4aaa-aaaa-aaaaaaaaaaaa"
+        if not is_valid_uuid(config.get("WATSONX_SPACE_ID")):
+            # Fake (but valid) UUIDv4 for knowledgebase check
+            config["WATSONX_SPACE_ID"] = "aaaaaaaa-aaaa-4aaa-aaaa-aaaaaaaaaaaa"
         return config

ibm_watsonx_orchestrate/docker/compose-lite.yml

@@ -62,6 +62,7 @@ services:
       WATSONX_API_KEY: ${WATSONX_APIKEY}
       WATSONX_URL: ${WATSONX_URL}
       WATSONX_SPACE_ID: ${WATSONX_SPACE_ID}
+      NODE_TLS_REJECT_UNAUTHORIZED: ${AI_GATEWAY_TLS_REJECT_UNAUTHORIZED}
 
   wxo-agent-gateway:
     image: ${AGENT_GATEWAY_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-agent-gateway:${AGENT_GATEWAY_TAG:-latest}
@@ -327,6 +328,7 @@ services:
       ENABLE_WEBHOOKS: false
       DISABLE_JSON_LOG_CELERY: true
       WXO_DEPLOYMENT_PLATFORM: saas
+      CPD_VERIFY: ${CPD_VERIFY}
       CONNECTION_MANAGER_URL: http://wxo-server-connection-manager:3001
       CHANNEL_SESSION_REDIS_URL: redis://wxo-server-redis:6379/5
       WXO_MILVUS_URI: http://wxo-milvus-standalone:19530
@@ -446,6 +448,7 @@ services:
       IBM_TELEMETRY_TRACER_ENDPOINT: http://jaeger:4318/v1/traces
       USE_IBM_TELEMETRY: ${USE_IBM_TELEMETRY:-false}
       WXO_DEPLOYMENT_PLATFORM: saas
+      CPD_VERIFY: ${CPD_VERIFY}
       CALLBACK_HOST_URL: ${CALLBACK_HOST_URL:-http://wxo-server:4321}
       LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
   
@@ -653,6 +656,7 @@ services:
       - WATSONX_URL=${WATSONX_URL}
       - PROXY_SERVER_URL=http://jaeger-proxy:9201
       - WXO_DEPLOYMENT_PLATFORM=saas
+      - CPD_VERIFY=${CPD_VERIFY}      
       - TENANT_API_KEY=${AGENTOPS_API_KEY}
       - TENANT_CONFIG_URL=http://wxo-server:4321
       - TENANT_DEFAULT_USERNAME=${ES_USERNAME}

ibm_watsonx_orchestrate/docker/default.env

@@ -58,10 +58,10 @@ REGISTRY_URL=
 
 
 
-SERVER_TAG=20-08-2025-cf6e342
+SERVER_TAG=29-08-2025-59ef405
 SERVER_REGISTRY=
 
-WORKER_TAG=20-08-2025-cf6e342
+WORKER_TAG=29-08-2025-cf32b2d
 WORKER_REGISTRY=
 
 AI_GATEWAY_TAG=20-08-2025-9ed6d40
@@ -78,7 +78,7 @@ AMDDBTAG=29-07-2025-9f3661b
 ARM64DBTAG=29-07-2025-9f3661b
 
 UI_REGISTRY=
-UITAG=31-07-2025
+UITAG=29-08-2025
 
 CM_REGISTRY=
 CM_TAG=24-07-2025
@@ -86,13 +86,13 @@ CM_TAG=24-07-2025
 TRM_TAG=19-08-2025-fe105eb0b950ff304f712a1a5b9fa3cba92d09da
 TRM_REGISTRY=
 
-TR_TAG=19-08-2025-fe105eb0b950ff304f712a1a5b9fa3cba92d09da
+TR_TAG=25-08-2025-58ae475
 TR_REGISTRY=
 
-BUILDER_TAG=19-08-2025-1a79d34
+BUILDER_TAG=27-08-2025-7432aca
 BUILDER_REGISTRY=
 
-FLOW_RUNTIME_TAG=18-08-2025-v2
+FLOW_RUNTIME_TAG=26-08-2025-3b7f19e
 FLOW_RUMTIME_REGISTRY=
 
 
@@ -126,6 +126,8 @@ DOCPROC_REGISTRY=
 
 # END -- IMAGE REGISTRIES AND TAGS
 
+CPD_VERIFY=true
+AI_GATEWAY_TLS_REJECT_UNAUTHORIZED=1
 TAVILY_API_KEY=dummy_tavily_api_key
 PREFERRED_MODELS=meta-llama/llama-3-2-90b-vision-instruct,meta-llama/llama-3-405b-instruct
 INCOMPATIBLE_MODELS=flan,embedding,cross-encoder,tinytimemixers

ibm_watsonx_orchestrate/flow_builder/data_map.py

@@ -12,7 +12,10 @@ class DataMap(BaseModel):
     def to_json(self) -> dict[str, Any]:
         model_spec = {}
         if self.maps and len(self.maps) > 0:
-            model_spec["maps"] = [assignment.model_dump() for assignment in self.maps]
+            model_spec["maps"] = []
+            for assignment in self.maps:
+                model_spec["maps"].append(assignment.model_dump())
+        return model_spec
 
     def add(self, line: Assignment) -> Self:
         self.maps.append(line)

ibm_watsonx_orchestrate/flow_builder/flows/flow.py

@@ -25,7 +25,7 @@ from ibm_watsonx_orchestrate.client.tools.tool_client import ToolClient
 from ibm_watsonx_orchestrate.client.tools.tempus_client import TempusClient
 from ibm_watsonx_orchestrate.client.utils import instantiate_client
 from ..types import (
-    DocProcKVPSchema, EndNodeSpec, Expression, ForeachPolicy, ForeachSpec, LoopSpec, BranchNodeSpec, MatchPolicy, PlainTextReadingOrder, PromptLLMParameters, PromptNodeSpec, TimerNodeSpec,
+    DocProcKVPSchema, Assignment, Conditions, EndNodeSpec, Expression, ForeachPolicy, ForeachSpec, LoopSpec, BranchNodeSpec, MatchPolicy, NodeIdCondition, PlainTextReadingOrder, PromptExample, PromptLLMParameters, PromptNodeSpec, TimerNodeSpec,
     StartNodeSpec, ToolSpec, JsonSchemaObject, ToolRequestBody, ToolResponseBody, UserFieldKind, UserFieldOption, UserFlowSpec, UserNodeSpec, WaitPolicy,
     DocProcSpec, TextExtractionResponse, DocProcInput, DecisionsNodeSpec, DecisionsRule, DocExtSpec, File, DocumentClassificationResponse, DocClassifierSpec, DocumentProcessingCommonInput
 )
@@ -64,7 +64,7 @@ class FlowEdge(BaseModel):
 
 class Flow(Node):
     '''Flow represents a flow that will be run by wxO Flow engine.'''
-    output_map: DataMap | None = None
+    output_map: dict[str, DataMap] | None = None
     nodes: dict[str, SerializeAsAny[Node]] = {}
     edges: List[FlowEdge] = []
     schemas: dict[str, JsonSchemaObject] = {}
@@ -401,6 +401,7 @@ class Flow(Node):
             display_name: str|None=None,
             system_prompt: str | list[str] | None = None,
             user_prompt: str | list[str] | None = None,
+            prompt_examples: list[PromptExample] | None = None,
             llm: str | None = None,
             llm_parameters: PromptLLMParameters | None = None,
             description: str | None = None,
@@ -422,6 +423,7 @@ class Flow(Node):
             description=description,
             system_prompt=system_prompt,
             user_prompt=user_prompt,
+            prompt_examples=prompt_examples,
             llm=llm,
             llm_parameters=llm_parameters,
             input_schema=_get_tool_request_body(input_schema_obj),
@@ -721,7 +723,7 @@ class Flow(Node):
         '''Create a single node flow with an automatic START and END node.'''
         return self.sequence(START, node, END)
 
-    def branch(self, evaluator: Union[Callable, Expression]) -> "Branch":
+    def branch(self, evaluator: Union[Callable, Expression, Conditions]) -> 'Branch':
         '''Create a BRANCH node'''
         e = evaluator
         if isinstance(evaluator, Callable):
@@ -735,11 +737,19 @@ class Flow(Node):
             # e = new_script_spec
         elif isinstance(evaluator, str):
             e = Expression(expression=evaluator)
+        elif isinstance(evaluator, list):
+            e = Conditions(conditions=evaluator)
 
         spec = BranchNodeSpec(name = "branch_" + str(self._next_sequence_id()), evaluator=e)
         branch_node = Branch(spec = spec, containing_flow=self)
         return cast(Branch, self._add_node(branch_node))
     
+    def conditions(self) -> 'Branch':
+        '''Create a Branch node with empty Conditions evaluator (if-else)'''
+        spec = BranchNodeSpec(name = "branch_" + str(self._next_sequence_id()), evaluator=Conditions(conditions=[]))
+        branch_conditions_node = Branch(spec = spec, containing_flow=self)
+        return cast(Branch, self._add_node(branch_conditions_node))
+    
     def wait_for(self, *args) -> "Wait":
         '''Wait for all incoming nodes to complete.'''
         raise ValueError("Not implemented yet.")
@@ -754,6 +764,77 @@ class Flow(Node):
             
         # return cast(Wait, self.node(wait_node))
             
+    def map_flow_output_with_variable(self, target_output_variable: str, variable: str, default_value: str = None) -> Self:
+        if self.output_map and "spec" in self.output_map:
+            maps = self.output_map["spec"].maps or []
+        else:
+            maps = []
+        
+        curr_map_metadata = {
+            "assignmentType": "variable"
+        }
+
+        target_variable = "flow.output." + target_output_variable
+        value_expression = "flow." + variable
+
+        if default_value:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, default_value=default_value, metadata=curr_map_metadata))
+        else:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, metadata=curr_map_metadata))
+
+        flow_output_map_spec = DataMap(maps=maps)
+
+        if self.output_map and "spec" in self.output_map:
+            self.output_map["spec"] = flow_output_map_spec
+        else:
+            self.output_map = {"spec": flow_output_map_spec}
+        return self
+    
+    def map_output(self, output_variable: str, expression: str, default_value: str = None) -> Self:
+        if self.output_map and "spec" in self.output_map:
+            maps = self.output_map["spec"].maps or []
+        else:
+            maps = []
+        
+        curr_map_metadata = {
+            "assignmentType": "pyExpression"
+        }
+
+        target_variable = "flow.output." + output_variable
+        value_expression = expression
+
+        if default_value:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, default_value=default_value, metadata=curr_map_metadata))
+        else:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, metadata=curr_map_metadata))
+
+        flow_output_map_spec = DataMap(maps=maps)
+
+        if self.output_map and "spec" in self.output_map:
+            self.output_map["spec"] = flow_output_map_spec
+        else:
+            self.output_map = {"spec": flow_output_map_spec}
+        return self
+    
+    def map_flow_output_with_none(self, target_output_variable: str) -> Self:
+        if self.output_map and "spec" in self.output_map:
+            maps = self.output_map["spec"].maps or []
+        else:
+            maps = []
+        
+
+        target_variable = "flow.output." + target_output_variable
+
+        maps.append(Assignment(target_variable=target_variable, value_expression=None))
+
+        flow_output_map_spec = DataMap(maps=maps)
+
+        if self.output_map and "spec" in self.output_map:
+            self.output_map["spec"] = flow_output_map_spec
+        else:
+            self.output_map = {"spec": flow_output_map_spec}
+        return self
+
 
     def foreach(self, item_schema: type[BaseModel],
                 input_schema: type[BaseModel] |None=None,
@@ -814,8 +895,6 @@ class Flow(Node):
                  input_schema: type[BaseModel] |None=None,
                  output_schema: type[BaseModel] |None=None) -> "UserFlow": # return a UserFlow object
 
-        raise ValueError("userflow is NOT supported yet and it's interface will change.")
-
         output_schema_obj = _get_json_schema_obj("output", output_schema)
         input_schema_obj = _get_json_schema_obj("input", input_schema)
 
@@ -917,6 +996,11 @@ class Flow(Node):
         for key, value in self.metadata.items():
             metadata_dict[key] = value
         flow_dict["metadata"] = metadata_dict
+
+        if self.output_map and "spec" in self.output_map:
+            flow_dict["output_map"] = {
+                "spec": self.output_map["spec"].to_json()
+            }
         return flow_dict
 
     def _get_node_id(self, node: Union[str, Node]) -> str:
@@ -1226,6 +1310,27 @@ class Branch(FlowControl):
             raise ValueError("Cannot have custom label __default__. Use default() instead.")
 
         return self._add_case(label, node)
+    
+    def condition(self, to_node: Node, expression: str="", default: bool=False) -> Self:
+        '''
+        Add a condition to this branch node. 
+
+        Parameters:
+        expression (str): The expression of this condition.
+        to_node (Node): The node to go to when expression is evaluated to true.
+        default (bool): The condition is the default (else) case.
+        '''
+
+        node_id = self.containing_flow._get_node_id(to_node)
+        if default:
+            condition = NodeIdCondition(node_id=node_id, default=default)
+        else:
+            condition = NodeIdCondition(expression=expression, node_id=node_id, default=default)
+
+        self.spec.evaluator.conditions.append(condition)
+        self.containing_flow.edge(self, to_node)
+
+        return self
 
     def default(self, node: Node) -> Self:
         '''
@@ -1445,13 +1550,14 @@ class UserFlow(Flow):
               kind: UserFieldKind = UserFieldKind.Text,
               display_name: str | None = None,
               description: str | None = None,
+              direction: str | None = None,
               default: Any | None = None,
               text: str = None, # The text used to ask question to the user, e.g. 'what is your name?'
               option: UserFieldOption | None = None,
               is_list: bool = False,
               min: Any | None = None,
               max: Any | None = None,
-              input_map: DataMap = None,
+              input_map: DataMap | None= None,
               custom: dict[str, Any] = {}) -> UserNode:
         '''create a node in the flow'''
         # create a json schema object based on the single field
@@ -1483,6 +1589,8 @@ class UserFlow(Flow):
                    description = description,
                    default = default,
                    text = text,
+                   direction = direction,
+                   input_map = input_map,
                    option = option,
                    is_list = is_list,
                    min = min,

ibm_watsonx_orchestrate/flow_builder/node.py

@@ -6,14 +6,14 @@ import yaml
 from pydantic import BaseModel, Field, SerializeAsAny, create_model
 from enum import Enum
 
-from .types import DocExtConfigField, EndNodeSpec, NodeSpec, AgentNodeSpec, PromptNodeSpec, TimerNodeSpec, StartNodeSpec, ToolNodeSpec, UserFieldKind, UserFieldOption, UserNodeSpec, DocProcSpec, \
+from .types import Assignment, DocExtConfigField, EndNodeSpec, NodeSpec, AgentNodeSpec, PromptNodeSpec, TimerNodeSpec, StartNodeSpec, ToolNodeSpec, UserFieldKind, UserFieldOption, UserNodeSpec, DocProcSpec, \
                     DocExtSpec, DocExtConfig, DocClassifierSpec, DecisionsNodeSpec, DocClassifierConfig
 
 from .data_map import DataMap
 
 class Node(BaseModel):
     spec: SerializeAsAny[NodeSpec]
-    input_map: DataMap | None = None
+    input_map: dict[str, DataMap] | None = None
 
     def __call__(self, **kwargs):
         pass
@@ -40,10 +40,77 @@ class Node(BaseModel):
     def to_json(self) -> dict[str, Any]:
         model_spec = {}
         model_spec["spec"] = self.spec.to_json()
-        if self.input_map is not None:
-            model_spec['input_map'] = self.input_map.to_json()
+        if self.input_map is not None and "spec" in self.input_map:
+            model_spec['input_map'] = {
+                "spec": self.input_map["spec"].to_json()
+            }
 
         return model_spec
+    
+    def map_node_input_with_variable(self, target_input_variable: str, variable: str, default_value: str = None) -> None:
+        if self.input_map and "spec" in self.input_map:
+            maps = self.input_map["spec"].maps or []
+        else:
+            maps = []
+        
+        curr_map_metadata = {
+            "assignmentType": "variable"
+        }
+
+        target_variable = "self.input." + target_input_variable
+        value_expression = "flow." + variable
+
+        if default_value:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, default_value=default_value, metadata=curr_map_metadata))
+        else:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, metadata=curr_map_metadata))
+
+        node_input_map_spec = DataMap(maps=maps)
+        if self.input_map and "spec" in self.input_map:
+            self.input_map["spec"] = node_input_map_spec
+        else:
+            self.input_map = {"spec": node_input_map_spec}
+
+    def map_input(self, input_variable: str, expression: str, default_value: str = None) -> None:
+        if self.input_map and "spec" in self.input_map:
+            maps = self.input_map["spec"].maps or []
+        else:
+            maps = []
+        
+        curr_map_metadata = {
+            "assignmentType": "pyExpression"
+        }
+
+        target_variable = "self.input." + input_variable
+        value_expression = expression
+
+        if default_value:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, default_value=default_value, metadata=curr_map_metadata))
+        else:
+            maps.append(Assignment(target_variable=target_variable, value_expression=value_expression, metadata=curr_map_metadata))
+
+        node_input_map_spec = DataMap(maps=maps)
+        if self.input_map and "spec" in self.input_map:
+            self.input_map["spec"] = node_input_map_spec
+        else:
+            self.input_map = {"spec": node_input_map_spec}
+
+    def map_node_input_with_none(self, target_input_variable: str) -> None:
+        if self.input_map and "spec" in self.input_map:
+            maps = self.input_map["spec"].maps or []
+        else:
+            maps = []
+        
+
+        target_variable = "self.input." + target_input_variable
+
+        maps.append(Assignment(target_variable=target_variable, value_expression=None))
+
+        node_input_map_spec = DataMap(maps=maps)
+        if self.input_map and "spec" in self.input_map:
+            self.input_map["spec"] = node_input_map_spec
+        else:
+            self.input_map = {"spec": node_input_map_spec}
 
 class StartNode(Node):
     def __repr__(self):
@@ -83,6 +150,8 @@ class UserNode(Node):
               option: UserFieldOption | None = None,
               min: Any | None = None,
               max: Any | None = None,
+              direction: str | None = None,
+              input_map: DataMap | None = None,
               is_list: bool = False,
               custom: dict[str, Any] | None = None,
               widget: str | None = None):
@@ -97,7 +166,9 @@ class UserNode(Node):
                               max=max,
                               is_list=is_list,
                               custom=custom,
-                              widget=widget)
+                              widget=widget,
+                              direction=direction,
+                              input_map=input_map)
 
 class AgentNode(Node):
     def __repr__(self):

ibm_watsonx_orchestrate/flow_builder/types.py

@@ -28,6 +28,7 @@ from ibm_watsonx_orchestrate.agent_builder.tools.types import (
 )
 from .utils import get_valid_name
 
+
 logger = logging.getLogger(__name__)
 
 class JsonSchemaObjectRef(JsonSchemaObject):
@@ -360,7 +361,6 @@ class EndNodeSpec(NodeSpec):
     def __init__(self, **data):
         super().__init__(**data)
         self.kind = "end"
-
 class ToolNodeSpec(NodeSpec):
     tool: Union[str, ToolSpec] = Field(default = None, description="the tool to use")
 
@@ -421,9 +421,10 @@ class UserFieldKind(str, Enum):
     DateTime: str = "datetime"
     Time: str = "time"
     Number: str = "number"
-    Document: str = "document"
+    File: str = "file"
     Boolean: str = "boolean"
     Object: str = "object"
+    Choice: str = "any"
 
     def convert_python_type_to_kind(python_type: type) -> "UserFieldKind":
         if inspect.isclass(python_type):
@@ -463,8 +464,8 @@ class UserFieldKind(str, Enum):
             model_spec["format"] = "number"
         elif kind == UserFieldKind.Boolean:
             model_spec["type"] = "boolean"
-        elif kind == UserFieldKind.Document:
-            model_spec["format"] = "uri"
+        elif kind == UserFieldKind.File:
+            model_spec["format"] = "wxo-file"
         elif kind == UserFieldKind.Object:
             raise ValueError("Object user fields are not supported.")
         
@@ -483,6 +484,8 @@ class UserField(BaseModel):
     text: str | None = Field(default=None, description="A descriptive text that can be used to ask user about this field.")
     display_name: str | None = None
     description: str | None = None
+    direction: str | None = None
+    input_map: Any | None = None,
     default: Any | None = None
     option: UserFieldOption | None = None
     min: Any | None = None,
@@ -497,6 +500,15 @@ class UserField(BaseModel):
             model_spec["name"] = self.name
         if self.kind:
             model_spec["kind"] = self.kind.value
+        if self.direction:
+            model_spec["direction"] = self.direction  
+        if self.input_map:
+            # workaround for circular dependency related to Assigments in the Datamap module
+            from .data_map import DataMap
+            if self.input_map and not isinstance(self.input_map, DataMap):
+                raise TypeError("input_map must be an instance of DataMap")
+            #model_spec["input_map"] = self.input_map.to_json() 
+            model_spec["input_map"] = {"spec": self.input_map.to_json()}        
         if self.text:
             model_spec["text"] = self.text
         if self.display_name:
@@ -556,7 +568,15 @@ class UserNodeSpec(NodeSpec):
               max: Any | None = None,
               is_list: bool = False,
               custom: dict[str, Any] | None = None,
-              widget: str | None = None):
+              widget: str | None = None,
+              input_map: Any | None = None,
+              direction: str | None = None):
+        
+        # workaround for circular dependency related to Assigments in the Datamap module
+        from .data_map import DataMap
+        if input_map and not isinstance(input_map, DataMap):
+            raise TypeError("input_map must be an instance of DataMap")
+        
         userfield = UserField(name=name, 
                               kind=kind, 
                               text=text,
@@ -568,7 +588,9 @@ class UserNodeSpec(NodeSpec):
                               max=max,
                               is_list=is_list,
                               custom=custom,
-                              widget=widget)
+                              widget=widget,
+                              direction=direction,
+                              input_map=input_map)
         
         # find the index of the field
         i = 0
@@ -660,11 +682,28 @@ class PromptLLMParameters(BaseModel):
         if self.stop_sequences:
             model_spec["stop_sequences"] = self.stop_sequences
         return model_spec
+    
+class PromptExample(BaseModel):
+    input: Optional[str] = None
+    expected_output: Optional[str] = None
+    enabled: bool
+
+    def to_json(self) -> dict[str, Any]:
+        model_spec = {}
+        if self.input:
+            model_spec["input"] = self.input
+        if self.expected_output:
+            model_spec["expected_output"] = self.expected_output
+        if self.enabled:
+            model_spec["enabled"] = self.enabled
+        return model_spec
+
 
 
 class PromptNodeSpec(NodeSpec):
     system_prompt: str | list[str]
     user_prompt: str | list[str]
+    prompt_examples: Optional[list[PromptExample]]
     llm: Optional[str] 
     llm_parameters: Optional[PromptLLMParameters] 
 
@@ -682,7 +721,10 @@ class PromptNodeSpec(NodeSpec):
             model_spec["llm"] = self.llm
         if self.llm_parameters:
             model_spec["llm_parameters"] = self.llm_parameters.to_json()
-
+        if self.prompt_examples:
+            model_spec["prompt_examples"] = []
+            for example in self.prompt_examples:
+                model_spec["prompt_examples"].append(example.to_json())
         return model_spec
     
 class TimerNodeSpec(NodeSpec):
@@ -707,6 +749,47 @@ class Expression(BaseModel):
         model_spec["expression"] = self.expression;
         return model_spec
     
+class NodeIdCondition(BaseModel):
+    '''One Condition contains an expression, a node_id that branch should go to when expression is true, and a default indicator. '''
+    expression: Optional[str] = Field(description="A python expression to be run by the flow engine", default=None)
+    node_id: str = Field(description="ID of the node in the flow that branch node should go to")
+    default: bool = Field(description="Boolean indicating if the condition is default case")
+
+    def to_json(self) -> dict[str, Any]:
+        model_spec = {}
+        if self.expression:
+            model_spec["expression"] = self.expression
+        model_spec["node_id"] = self.node_id
+        model_spec["default"] = self.default
+        return model_spec
+
+
+class EdgeIdCondition(BaseModel):
+    '''One Condition contains an expression, an edge_id that branch should go to when expression is true, and a default indicator. '''
+    expression: Optional[str] = Field(description="A python expression to be run by the flow engine")
+    edge_id: str = Field(description="ID of the edge in the flow that branch node should go to")
+    default: bool = Field(description="Boolean indicating if the condition is default case")
+
+    def to_json(self) -> dict[str, Any]:
+        model_spec = {}
+        if self.expression:
+            model_spec["expression"] = self.expression
+        model_spec["edge_id"] = self.edge_id
+        model_spec["default"] = self.default
+        return model_spec
+
+class Conditions(BaseModel):
+    '''One Conditions is an array represents the if-else conditions of a complex branch'''
+    conditions: list = List[Union[NodeIdCondition, EdgeIdCondition]]
+
+    def to_json(self) -> dict[str, Any]:
+        model_spec = {}
+        condition_list = []
+        for condition in self.conditions:
+            condition_list.append(NodeIdCondition.model_validate(condition).to_json())
+        model_spec["conditions"] = condition_list
+        return model_spec
+    
 class MatchPolicy(Enum):
  
     FIRST_MATCH = 1
@@ -724,7 +807,7 @@ class BranchNodeSpec(FlowControlNodeSpec):
     cases (dict[str | bool, str]): A dictionary of labels to node names. The keys can be strings or booleans.
     match_policy (MatchPolicy): The policy to use when evaluating the expression.
     '''
-    evaluator: Expression
+    evaluator: Expression | Conditions
     cases: dict[str | bool, str] = Field(default = {},
                                          description="A dictionary of labels to node names.")
     match_policy: MatchPolicy = Field(default = MatchPolicy.FIRST_MATCH)
@@ -808,7 +891,7 @@ class UserFlowSpec(FlowSpec):
 
     def __init__(self, **kwargs):
         super().__init__(**kwargs)
-        self.kind = "userflow"
+        self.kind = "user_flow"
 
     def to_json(self) -> dict[str, Any]:
         model_spec = super().to_json()

{ibm_watsonx_orchestrate-1.10.0b1.dist-info → ibm_watsonx_orchestrate-1.10.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ibm-watsonx-orchestrate
-Version: 1.10.0b1
+Version: 1.10.1
 Summary: IBM watsonx.orchestrate SDK
 Author-email: IBM <support@ibm.com>
 License: MIT License

{ibm_watsonx_orchestrate-1.10.0b1.dist-info → ibm_watsonx_orchestrate-1.10.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ibm_watsonx_orchestrate/__init__.py,sha256=G8rgpNWoKMo_SVWHA47iL6iuMrXwKIw9aaWYcFRb-v0,429
+ibm_watsonx_orchestrate/__init__.py,sha256=PRy6NfSG4mSt4yawB2bLg9gwssVrWTF8ssncV9sp0vU,426
 ibm_watsonx_orchestrate/agent_builder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/agent_builder/agents/__init__.py,sha256=lmZwaiWXD4Ea19nrMwZXaqCxFMG29xNS8vUoZtK3yI4,392
 ibm_watsonx_orchestrate/agent_builder/agents/agent.py,sha256=W0uya81fQPrYZFaO_tlsxBL56Bfpw0xrqdxQJhAZ6XI,983
@@ -9,8 +9,8 @@ ibm_watsonx_orchestrate/agent_builder/agents/webchat_customizations/__init__.py,
 ibm_watsonx_orchestrate/agent_builder/agents/webchat_customizations/prompts.py,sha256=jNVF_jgz1Dmt7-RxAceAS0XWXk_fx9h3sS_fGrvZT28,941
 ibm_watsonx_orchestrate/agent_builder/agents/webchat_customizations/welcome_content.py,sha256=U76wZrblSXx4qv7phcPYs3l8SiFzwZ5cJ74u8Y2iYhU,608
 ibm_watsonx_orchestrate/agent_builder/connections/__init__.py,sha256=B9FwmBbdJxFj3KmJ87Fc78TbzxOr1MIVhaHPJePOGSQ,716
-ibm_watsonx_orchestrate/agent_builder/connections/connections.py,sha256=Vo1ucn4c_Db68x7GGuq-uyM89dInnkRztmsF0uo-yBk,5421
-ibm_watsonx_orchestrate/agent_builder/connections/types.py,sha256=OGuFEQhc1dj8GVjHuT_V3coCC_YtBGqQ68qGy96m6EQ,9587
+ibm_watsonx_orchestrate/agent_builder/connections/connections.py,sha256=ozRTZ1Y10YbipE6oegI8QIP31fWqUaBUFoHAo-WTY3g,5547
+ibm_watsonx_orchestrate/agent_builder/connections/types.py,sha256=cfM2L4VscC-0muDwqq5EhCKKQxwWyHmA-rQBE9LE7r0,11244
 ibm_watsonx_orchestrate/agent_builder/knowledge_bases/knowledge_base.py,sha256=_KuGF0RZpKpwdt31rzjlTjrhGRFz2RtLzleNkhMNX4k,1831
 ibm_watsonx_orchestrate/agent_builder/knowledge_bases/knowledge_base_requests.py,sha256=3xTfFMZR17EN8eYRhsVyBfOEzlTqyi0eYaMXyv0_ZtQ,862
 ibm_watsonx_orchestrate/agent_builder/knowledge_bases/types.py,sha256=2MP3_c6V3OdbwrdDDffGtrBoAhoa3q-YoUXj_RsMG_M,8340
@@ -42,8 +42,8 @@ ibm_watsonx_orchestrate/cli/commands/channels/types.py,sha256=hMFvWPr7tAmDrhBqtz
 ibm_watsonx_orchestrate/cli/commands/channels/webchat/channels_webchat_command.py,sha256=vPCr6z1n1yzGDjTlM4f3_9MiuVRzNmXbvUifm6XyQi8,1103
 ibm_watsonx_orchestrate/cli/commands/channels/webchat/channels_webchat_controller.py,sha256=CGfmKsCBX4E3HMZ8C0IXD-DHQNwe96V1Y_BxUZM2us0,8557
 ibm_watsonx_orchestrate/cli/commands/chat/chat_command.py,sha256=Q9vg2Z5Fsunu6GQFY_TIsNRhUCa0SSGSPnK4jxSGK34,1581
-ibm_watsonx_orchestrate/cli/commands/connections/connections_command.py,sha256=bYn9Z4DDDdoaAYkUddgrQas8H7h7h7hbO4fle5Ovdmo,10688
-ibm_watsonx_orchestrate/cli/commands/connections/connections_controller.py,sha256=XpcJBAzJHczZcJttH7kzfIgZUidj_aW86mmQ5qk_T8c,23367
+ibm_watsonx_orchestrate/cli/commands/connections/connections_command.py,sha256=lwPhDPemDvFU-j7QUj8E07hsnzLBUJDNcyhQDzxs3S8,12442
+ibm_watsonx_orchestrate/cli/commands/connections/connections_controller.py,sha256=_i1gaR2Nuy9gPLJZTJWoVnRvg9vKcwyg2UoiK5Azsw4,26390
 ibm_watsonx_orchestrate/cli/commands/copilot/copilot_command.py,sha256=IxasApIyQYWRMKPXKa38ZPVkUvOc4chggSmSGjgQGXc,2345
 ibm_watsonx_orchestrate/cli/commands/copilot/copilot_controller.py,sha256=SC2Tjq6r-tHIiyPBMajsxdMIY3BQpRWpkYGZS2XbJyU,18981
 ibm_watsonx_orchestrate/cli/commands/copilot/copilot_server_controller.py,sha256=AcBE97qNYsRN0ftY_E-AAjKFyVea4fHtU5eB2HsB42I,5619
@@ -58,8 +58,8 @@ ibm_watsonx_orchestrate/cli/commands/login/login_command.py,sha256=xArMiojoozg7E
 ibm_watsonx_orchestrate/cli/commands/models/model_provider_mapper.py,sha256=mbvBR5o9M7W6OpTZyd6TtSEOIXq07dPYz4hv5zDrsd0,8129
 ibm_watsonx_orchestrate/cli/commands/models/models_command.py,sha256=PW-PIM5Nq0qdCopWjANGBWEuEoA3NLTFThYrN8ggGCI,6425
 ibm_watsonx_orchestrate/cli/commands/models/models_controller.py,sha256=eZSYQUg9TL_-8lgcPVpKIx7MtOE7K_NCvZW9Y9YsFA0,18466
-ibm_watsonx_orchestrate/cli/commands/server/server_command.py,sha256=RBGH3ssDDhrcvmGpvpVtvqpWIhJm_odH4kqf4UQMRrI,46669
-ibm_watsonx_orchestrate/cli/commands/server/types.py,sha256=UCrgGErbSVBnOzesfjrIii4tTCuVfWemYz5AKGZX0oA,4213
+ibm_watsonx_orchestrate/cli/commands/server/server_command.py,sha256=cMy_GGYXbug5A6IGoddUzBUCmunySCI8BC-ebRL6p4A,46489
+ibm_watsonx_orchestrate/cli/commands/server/types.py,sha256=lOrPLzadXJ3pVnIVsQRb0keXckFBGMDD3Holjfge9GI,4412
 ibm_watsonx_orchestrate/cli/commands/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/cli/commands/settings/settings_command.py,sha256=CzXRkd-97jXyS6LtaaNtMah-aZu0919dYl-mDwzGThc,344
 ibm_watsonx_orchestrate/cli/commands/settings/observability/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -101,23 +101,23 @@ ibm_watsonx_orchestrate/client/toolkit/toolkit_client.py,sha256=TLFNS39EeBD_t4Y-
 ibm_watsonx_orchestrate/client/tools/tempus_client.py,sha256=24fKDZUOVHBW-Vj4mubnpnUmab5LgGn8u5hOVyJaozI,1804
 ibm_watsonx_orchestrate/client/tools/tool_client.py,sha256=d3i3alVwa0TCN72w9sWOrM20GCbNmnpTnqEOJVbBIFM,1994
 ibm_watsonx_orchestrate/client/voice_configurations/voice_configurations_client.py,sha256=M5xIPLiVNpP-zxQw8CTNT9AiBjeXXmJiNaE142e2A3E,2682
-ibm_watsonx_orchestrate/docker/compose-lite.yml,sha256=wylLRS6iGEfMitfoXuT2tVuSiWej0HLAE5k-6Onl-kI,45671
-ibm_watsonx_orchestrate/docker/default.env,sha256=ol3f-UGtJ0i2sTiwmEQPNSNzOe_VBNFc6zsgR0dC3XY,6268
+ibm_watsonx_orchestrate/docker/compose-lite.yml,sha256=iFk3JcGUs4aanAzTZkgG1EcsunWp_awkuIphpjLmeX8,45848
+ibm_watsonx_orchestrate/docker/default.env,sha256=jz6MNLTvVIVTUOY8AYLOLy589nMs_v0nzQM5rKYACwg,6293
 ibm_watsonx_orchestrate/docker/proxy-config-single.yaml,sha256=WEbK4ENFuTCYhzRu_QblWp1_GMARgZnx5vReQafkIG8,308
 ibm_watsonx_orchestrate/docker/start-up.sh,sha256=LTtwHp0AidVgjohis2LXGvZnkFQStOiUAxgGABOyeUI,1811
 ibm_watsonx_orchestrate/docker/sdk/ibm_watsonx_orchestrate-0.6.0-py3-none-any.whl,sha256=Hi3-owh5OM0Jz2ihX9nLoojnr7Ky1TV-GelyqLcewLE,2047417
 ibm_watsonx_orchestrate/docker/sdk/ibm_watsonx_orchestrate-0.6.0.tar.gz,sha256=e5T-q7XPAtiCyQljwZp6kk3Q_4Tg6y5sijHTkscmqqQ,2025466
 ibm_watsonx_orchestrate/docker/tempus/common-config.yaml,sha256=Zo3F36F5DV4VO_vUg1RG-r4WhcukVh79J2fXhGl6j0A,22
 ibm_watsonx_orchestrate/flow_builder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ibm_watsonx_orchestrate/flow_builder/data_map.py,sha256=1brmWWFERDsNG2XGais-5-r58LKUUwBtqwdaLQIFRhE,503
-ibm_watsonx_orchestrate/flow_builder/node.py,sha256=2b4BL64tt1isvA5O_uFvXZG5qqvDYrAVSYyeW6wyk0U,6978
-ibm_watsonx_orchestrate/flow_builder/types.py,sha256=ohtfMFneFtK7KOEEEdYpD5hmiRwfyTnkHLkOr4zNhGs,47611
+ibm_watsonx_orchestrate/flow_builder/data_map.py,sha256=LinePFgb5mBnrvNmPkFe3rq5oYJZSjcgmaEGpE6dVwc,586
+ibm_watsonx_orchestrate/flow_builder/node.py,sha256=U7-cYYhe7Gj_j6U0dw1ufaPROvXmFo30ZI8s7eTXZlk,9940
+ibm_watsonx_orchestrate/flow_builder/types.py,sha256=FJw9pvXag9paNtwH1p9pnva0bvLMJ-ISFemyDJe7WXY,51392
 ibm_watsonx_orchestrate/flow_builder/utils.py,sha256=8q1jr5i_TzoJpoQxmLiO0g5Uv03BLbTUaRfw8_0VWIY,11931
 ibm_watsonx_orchestrate/flow_builder/flows/__init__.py,sha256=iRYV0_eXgBBGhuNnvg-mUyPUyCIw5BiallPOp27bzYM,1083
 ibm_watsonx_orchestrate/flow_builder/flows/constants.py,sha256=-TGneZyjA4YiAtJJK7OmmjDHYQC4mw2e98MPAZqiB50,324
 ibm_watsonx_orchestrate/flow_builder/flows/decorators.py,sha256=lr4qSWq5PWqlGFf4fzUQZCVQDHBYflrYwZ24S89Aq80,2794
 ibm_watsonx_orchestrate/flow_builder/flows/events.py,sha256=VyaBm0sADwr15LWfKbcBQS1M80NKqzYDj3UlW3OpOf4,2984
-ibm_watsonx_orchestrate/flow_builder/flows/flow.py,sha256=P-MazfKiQr8BLiFAqOxqKlFHM2KyY7Dgp8O6xdCxIjk,60388
+ibm_watsonx_orchestrate/flow_builder/flows/flow.py,sha256=UqFgGxYM9EIXk6pZjMMI5s6r1C11Gb9kRN5NVbJChoc,64824
 ibm_watsonx_orchestrate/run/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/run/connections.py,sha256=9twXkNeUx83fP_qYUbJRtumE-wzPPNN2v-IY_8hGndM,1820
 ibm_watsonx_orchestrate/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -126,8 +126,8 @@ ibm_watsonx_orchestrate/utils/utils.py,sha256=U7z_2iASoFiZ2zM0a_2Mc2Y-P5oOT7hOwu
 ibm_watsonx_orchestrate/utils/logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ibm_watsonx_orchestrate/utils/logging/logger.py,sha256=FzeGnidXAjC7yHrvIaj4KZPeaBBSCniZFlwgr5yV3oA,1037
 ibm_watsonx_orchestrate/utils/logging/logging.yaml,sha256=9_TKfuFr1barnOKP0fZT5D6MhddiwsXVTFjtRbcOO5w,314
-ibm_watsonx_orchestrate-1.10.0b1.dist-info/METADATA,sha256=Md6fKs7y3oZCmVgOR6jVw5eSF31wwRMtXSn1w0SFvN8,1363
-ibm_watsonx_orchestrate-1.10.0b1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ibm_watsonx_orchestrate-1.10.0b1.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
-ibm_watsonx_orchestrate-1.10.0b1.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
-ibm_watsonx_orchestrate-1.10.0b1.dist-info/RECORD,,
+ibm_watsonx_orchestrate-1.10.1.dist-info/METADATA,sha256=a2z_4K9IvBH1Elg31ecfkyAuPEIS4j5mHYn29boz2eg,1361
+ibm_watsonx_orchestrate-1.10.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ibm_watsonx_orchestrate-1.10.1.dist-info/entry_points.txt,sha256=SfIT02-Jen5e99OcLhzbcM9Bdyf8SGVOCtnSplgZdQI,69
+ibm_watsonx_orchestrate-1.10.1.dist-info/licenses/LICENSE,sha256=Shgxx7hTdCOkiVRmfGgp_1ISISrwQD7m2f0y8Hsapl4,1083
+ibm_watsonx_orchestrate-1.10.1.dist-info/RECORD,,