ibm-watsonx-orchestrate 1.10.0b0__py3-none-any.whl → 1.10.0b1__py3-none-any.whl

ibm_watsonx_orchestrate/cli/commands/server/server_command.py

@@ -332,7 +332,6 @@ def write_merged_env_file(merged_env: dict, target_path: str = None) -> Path:
             file.write(f"{key}={val}\n")
     return Path(file.name)
 
-
 def get_dbtag_from_architecture(merged_env_dict: dict) -> str:
     """Detects system architecture and returns the corresponding DBTAG."""
     arch = platform.machine()
@@ -375,7 +374,8 @@ def run_compose_lite(
         experimental_with_langfuse=False, 
         experimental_with_ibm_telemetry=False, 
         with_doc_processing=False,
-        with_voice=False
+        with_voice=False,
+        experimental_with_langflow=False,
     ) -> None:
     compose_path = get_compose_file()
 
@@ -404,7 +404,11 @@ def run_compose_lite(
     logger.info("Database container started successfully. Now starting other services...")
 
 
-    # Step 2: Start all remaining services (except DB)
+    # Step 2: Create Langflow DB (if enabled)
+    if experimental_with_langflow:
+        create_langflow_db()
+
+    # Step 3: Start all remaining services (except DB)
     profiles = []
     if experimental_with_langfuse:
         profiles.append("langfuse")
@@ -414,6 +418,8 @@ def run_compose_lite(
         profiles.append("docproc")
     if with_voice:
         profiles.append("voice")
+    if experimental_with_langflow:
+        profiles.append("langflow")
 
     command = compose_command[:]
     for profile in profiles:
@@ -665,7 +671,6 @@ def run_compose_lite_down(final_env_file: Path, is_reset: bool = False) -> None:
         )
         sys.exit(1)
 
-
 def run_compose_lite_logs(final_env_file: Path, is_reset: bool = False) -> None:
     compose_path = get_compose_file()
     compose_command = ensure_docker_compose_installed()
@@ -866,6 +871,12 @@ def server_start(
         '--with-voice', '-v',
         help='Enable voice controller to interact with the chat via voice channels'
     ),
+    experimental_with_langflow: bool = typer.Option(
+        False,
+        '--experimental-with-langflow',
+        help='(Experimental) Enable Langflow UI, available at http://localhost:7861',
+        hidden=True
+    ),
 ):
     confirm_accepts_license_agreement(accept_terms_and_conditions)
 
@@ -907,6 +918,9 @@ def server_start(
     if experimental_with_ibm_telemetry:
         merged_env_dict['USE_IBM_TELEMETRY'] = 'true'
 
+    if experimental_with_langflow:
+        merged_env_dict['LANGFLOW_ENABLED'] = 'true'
+    
 
     try:
         dev_edition_source = get_dev_edition_source(merged_env_dict)
@@ -921,7 +935,8 @@ def server_start(
                      experimental_with_langfuse=experimental_with_langfuse,
                      experimental_with_ibm_telemetry=experimental_with_ibm_telemetry,
                      with_doc_processing=with_doc_processing,
-                     with_voice=with_voice)
+                     with_voice=with_voice,
+                     experimental_with_langflow=experimental_with_langflow)
     
     run_db_migration()
 
@@ -951,6 +966,8 @@ def server_start(
         logger.info(f"You can access the observability platform Langfuse at http://localhost:3010, username: orchestrate@ibm.com, password: orchestrate")
     if with_doc_processing:
         logger.info(f"Document processing in Flows (Public Preview) has been enabled.")
+    if experimental_with_langflow:
+        logger.info("Langflow has been enabled, the Langflow UI is available at http://localhost:7861")
 
 @server_app.command(name="stop")
 def server_stop(
@@ -1031,15 +1048,11 @@ def run_db_migration() -> None:
     merged_env_dict['ROUTING_LLM_API_KEY'] = ''
     merged_env_dict['ASSISTANT_LLM_API_KEY'] = ''
     final_env_file = write_merged_env_file(merged_env_dict)
+    
 
-    command = compose_command + [
-        "-f", str(compose_path),
-        "--env-file", str(final_env_file),
-        "exec",
-        "wxo-server-db",
-        "bash",
-        "-c",
-        '''
+    pg_user = merged_env_dict.get("POSTGRES_USER","postgres")
+
+    migration_command = f'''
         APPLIED_MIGRATIONS_FILE="/var/lib/postgresql/applied_migrations/applied_migrations.txt"
         touch "$APPLIED_MIGRATIONS_FILE"
 
@@ -1050,7 +1063,7 @@ def run_db_migration() -> None:
                 echo "Skipping already applied migration: $filename"
             else
                 echo "Applying migration: $filename"
-                if psql -U postgres -d postgres -q -f "$file" > /dev/null 2>&1; then
+                if psql -U {pg_user} -d postgres -q -f "$file" > /dev/null 2>&1; then
                     echo "$filename" >> "$APPLIED_MIGRATIONS_FILE"
                 else
                     echo "Error applying $filename. Stopping migrations."
@@ -1059,6 +1072,15 @@ def run_db_migration() -> None:
             fi
         done
         '''
+
+    command = compose_command + [
+        "-f", str(compose_path),
+        "--env-file", str(final_env_file),
+        "exec",
+        "wxo-server-db",
+        "bash",
+        "-c",
+        migration_command
     ]
 
     logger.info("Running Database Migration...")
@@ -1073,6 +1095,65 @@ def run_db_migration() -> None:
         )
         sys.exit(1)
 
+def create_langflow_db() -> None:
+    compose_path = get_compose_file()
+    compose_command = ensure_docker_compose_installed()
+    default_env_path = get_default_env_file()
+    merged_env_dict = merge_env(default_env_path, user_env_path=None)
+    merged_env_dict['WATSONX_SPACE_ID']='X'
+    merged_env_dict['WATSONX_APIKEY']='X'
+    merged_env_dict['WXAI_API_KEY'] = ''
+    merged_env_dict['ASSISTANT_EMBEDDINGS_API_KEY'] = ''
+    merged_env_dict['ASSISTANT_LLM_SPACE_ID'] = ''
+    merged_env_dict['ROUTING_LLM_SPACE_ID'] = ''
+    merged_env_dict['USE_SAAS_ML_TOOLS_RUNTIME'] = ''
+    merged_env_dict['BAM_API_KEY'] = ''
+    merged_env_dict['ASSISTANT_EMBEDDINGS_SPACE_ID'] = ''
+    merged_env_dict['ROUTING_LLM_API_KEY'] = ''
+    merged_env_dict['ASSISTANT_LLM_API_KEY'] = ''
+    final_env_file = write_merged_env_file(merged_env_dict)
+
+    pg_timeout = merged_env_dict.get('POSTGRES_READY_TIMEOUT','10')
+
+    pg_user = merged_env_dict.get("POSTGRES_USER","postgres")
+
+    creation_command = f"""
+    echo 'Waiting for pg to initialize...'
+
+    timeout={pg_timeout}
+    while [[ -z `pg_isready | grep 'accepting connections'` ]] && (( timeout > 0 )); do
+      ((timeout-=1)) && sleep 1;
+    done
+
+    if psql -U {pg_user} -lqt | cut -d \\| -f 1 | grep -qw langflow; then
+        echo 'Existing Langflow DB found'
+    else
+        echo 'Creating Langflow DB'
+        createdb -U "{pg_user}" -O "{pg_user}" langflow;
+        psql -U {pg_user} -q -d postgres -c "GRANT CONNECT ON DATABASE langflow TO {pg_user}";
+    fi
+    """
+    command = compose_command + [
+        "-f", str(compose_path),
+        "--env-file", str(final_env_file),
+        "exec",
+        "wxo-server-db",
+        "bash",
+        "-c",
+        creation_command
+    ]
+
+    logger.info("Preparing Langflow resources...")
+    result = subprocess.run(command, capture_output=False)
+
+    if result.returncode == 0:
+        logger.info("Langflow resources sucessfully created")
+    else:
+        error_message = result.stderr.decode('utf-8') if result.stderr else "Error occurred."
+        logger.error(
+            f"Failed to create Langflow resources\n{error_message}"
+        )
+        sys.exit(1)
 
 def bump_file_iteration(filename: str) -> str:
     regex = re.compile(f"^(?P<name>[^\\(\\s\\.\\)]+)(\\((?P<num>\\d+)\\))?(?P<type>\\.(?:{'|'.join(_EXPORT_FILE_TYPES)}))?$")

ibm_watsonx_orchestrate/cli/commands/toolkit/toolkit_command.py

@@ -1,7 +1,7 @@
 import typer
 from typing import List
 from typing_extensions import Annotated, Optional
-from ibm_watsonx_orchestrate.agent_builder.toolkits.types import ToolkitKind, Language
+from ibm_watsonx_orchestrate.agent_builder.toolkits.types import ToolkitKind, Language, ToolkitTransportKind
 from ibm_watsonx_orchestrate.cli.commands.toolkit.toolkit_controller import ToolkitController
 import logging
 import sys
@@ -45,6 +45,14 @@ def import_toolkit(
                 "The first argument will be used as the executable, the rest as its arguments."
         ),
     ] = None,
+    url: Annotated[
+        Optional[str],
+        typer.Option("--url", "-u", help="The URL of the remote MCP server", hidden=True),
+    ] = None,
+    transport: Annotated[
+        ToolkitTransportKind,
+        typer.Option("--transport", help="The communication protocol to use for the remote MCP server. Only \"sse\" or \"streamable_http\" supported", hidden=True),
+    ] = None,
     tools: Annotated[
         Optional[str],
         typer.Option("--tools", "-t", help="Comma-separated list of tools to import. Or you can use \"*\" to use all tools"),
@@ -64,18 +72,41 @@ def import_toolkit(
     else:
         tool_list = None
 
-    if not package and not package_root and not command:
-        logger.error("You must provide either '--package', '--package-root' or '--command'.")
-        sys.exit(1)
+    if not url and not transport:
+        if not package and not package_root and not command:
+            logger.error("You must provide either '--package', '--package-root' or '--command'.")
+            sys.exit(1)
 
-    if package_root and not command:
-        logger.error("Error: '--command' flag must be provided when '--package-root' is specified.")
-        sys.exit(1)
-    
-    if package_root and package:
-        logger.error("Please choose either '--package-root' or '--package' but not both.")
+        if package_root and not command:
+            logger.error("Error: '--command' flag must be provided when '--package-root' is specified.")
+            sys.exit(1)
+        
+        if package_root and package:
+            logger.error("Please choose either '--package-root' or '--package' but not both.")
+            sys.exit(1)
+
+    if (url and not transport) or (transport and not url):
+        logger.error("Both '--url' and '--transport' must be provided together for remote MCP.")
         sys.exit(1)
 
+    if url and transport:
+        forbidden_local_opts = []
+        if package:
+            forbidden_local_opts.append("--package")
+        if package_root:
+            forbidden_local_opts.append("--package-root")
+        if language:
+            forbidden_local_opts.append("--language")
+        if command:
+            forbidden_local_opts.append("--command")
+        
+        if forbidden_local_opts:
+            logger.error(
+                f"When using '--url' and '--transport' for a remote MCP, you cannot specify: "
+                f"{', '.join(forbidden_local_opts)}"
+            )
+            sys.exit(1)
+
     if package and not package_root:
         if not command:
             if language == Language.NODE:
@@ -97,6 +128,8 @@ def import_toolkit(
     package_root=package_root,
     language=language,
     command=command,
+    url=url,
+    transport=transport
 )
     toolkit_controller.import_toolkit(tools=tool_list, app_id=app_id)
 

ibm_watsonx_orchestrate/cli/commands/toolkit/toolkit_controller.py

@@ -10,7 +10,7 @@ import requests
 from ibm_watsonx_orchestrate.client.toolkit.toolkit_client import ToolKitClient
 from ibm_watsonx_orchestrate.client.tools.tool_client import ToolClient
 from ibm_watsonx_orchestrate.agent_builder.toolkits.base_toolkit import BaseToolkit, ToolkitSpec
-from ibm_watsonx_orchestrate.agent_builder.toolkits.types import ToolkitKind, Language, ToolkitSource
+from ibm_watsonx_orchestrate.agent_builder.toolkits.types import ToolkitKind, Language, ToolkitSource, ToolkitTransportKind
 from ibm_watsonx_orchestrate.client.utils import instantiate_client
 from ibm_watsonx_orchestrate.utils.utils import sanatize_app_id
 from ibm_watsonx_orchestrate.client.connections import get_connections_client
@@ -26,15 +26,16 @@ import json
 
 logger = logging.getLogger(__name__)
 
-def get_connection_id(app_id: str) -> str:
+def get_connection_id(app_id: str, is_local_mcp: bool) -> str:
     connections_client = get_connections_client()
     existing_draft_configuration = connections_client.get_config(app_id=app_id, env='draft')
     existing_live_configuration = connections_client.get_config(app_id=app_id, env='live')
 
     for config in [existing_draft_configuration, existing_live_configuration]:
-        if config and config.security_scheme != 'key_value_creds':
-            logger.error("Only key_value credentials are currently supported")
-            exit(1)
+        if is_local_mcp is True:
+            if config and config.security_scheme != 'key_value_creds':
+                logger.error("Only key_value credentials are currently supported for local MCP")
+                exit(1)
     connection_id = None
     if app_id is not None:
         connection = connections_client.get(app_id=app_id)
@@ -59,6 +60,8 @@ class ToolkitController:
         package_root: str = None,
         language: Language = None,
         command: str = None,
+        url: str = None,
+        transport: ToolkitTransportKind = None
     ):
         self.kind = kind
         self.name = name
@@ -68,6 +71,8 @@ class ToolkitController:
         self.language = language
         self.command = command
         self.client = None
+        self.url = url
+        self.transport = transport
 
         self.source: ToolkitSource = (
             ToolkitSource.FILES if package_root else ToolkitSource.PUBLIC_REGISTRY
@@ -95,15 +100,23 @@ class ToolkitController:
             logger.error(f"Existing toolkit found with name '{self.name}'. Failed to create toolkit.")
             sys.exit(1)
 
-        try:
-            command_parts = json.loads(self.command)
-            if not isinstance(command_parts, list):
-                raise ValueError("JSON command must be a list of strings")
-        except (json.JSONDecodeError, ValueError):
-            command_parts = self.command.split()
+        if not self.command:
+            command_parts = []
+        else:
+            try:
+                command_parts = json.loads(self.command)
+                if not isinstance(command_parts, list):
+                    raise ValueError("JSON command must be a list of strings")
+            except (json.JSONDecodeError, ValueError):
+                command_parts = self.command.split()
+
+        if command_parts:
+            command = command_parts[0]
+            args = command_parts[1:]
+        else:
+            command = None
+            args = []
 
-        command = command_parts[0]
-        args = command_parts[1:]
 
         if self.package_root:
             is_folder = os.path.isdir(self.package_root)
@@ -150,18 +163,31 @@ class ToolkitController:
                     console.print(f"  • {tool}")
 
             # Create toolkit metadata
-            payload = {
-                "name": self.name,
-                "description": self.description,
-                "mcp": {
-                    "source": self.source.value,
-                    "command": command,
-                    "args": args,
-                    "tools": tools,
-                    "connections": remapped_connections,
+            payload = {}
+
+            if self.transport is not None and self.url is not None:
+                payload = {
+                    "name": self.name,
+                    "description": self.description,
+                    "mcp": {
+                        "server_url": self.url,
+                        "transport": self.transport.value,
+                        "tools": tools,
+                        "connections": remapped_connections,
+                    }
+                }
+            else:
+                payload = {
+                    "name": self.name,
+                    "description": self.description,
+                    "mcp": {
+                        "source": self.source.value,
+                        "command": command,
+                        "args": args,
+                        "tools": tools,
+                        "connections": remapped_connections,
+                    }
                 }
-            }
-
 
             with Progress(
                 SpinnerColumn(spinner_name="dots"),
@@ -215,7 +241,8 @@ class ToolkitController:
                 raise typer.BadParameter(f"The provided --app-id '{app_id}' is not valid. --app-id cannot be empty or whitespace")
 
             runtime_id = sanatize_app_id(runtime_id)
-            app_id_dict[runtime_id] = get_connection_id(local_id)
+            is_local_mcp = self.package is not None or self.package_root is not None
+            app_id_dict[runtime_id] = get_connection_id(local_id, is_local_mcp)
 
         return app_id_dict
 

ibm_watsonx_orchestrate/client/connections/connections_client.py

@@ -19,7 +19,7 @@ class FetchConfigAuthTypes(str, Enum):
     API_KEY_AUTH = ConnectionType.API_KEY_AUTH.value
     OAUTH2_AUTH_CODE = ConnectionType.OAUTH2_AUTH_CODE.value
     OAUTH2_IMPLICIT = 'oauth2_implicit'
-    OAUTH2_PASSWORD = 'oauth2_password'
+    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
@@ -50,7 +50,7 @@ class GetConfigResponse(BaseModel):
 class GetConnectionResponse(BaseModel):
     connection_id: str = None
     app_id: str = None
-    tenant_id: str = None
+    tenant_id: Optional[str] = None
 
 
 
@@ -141,7 +141,8 @@ class ConnectionsClient(BaseAPIClient):
             else:
                 return self._get(f"/connections/applications/runtime_credentials?app_id={app_id}&env={env}")
         except ClientAPIException as e:
-            if e.response.status_code == 404:
+            # Returns 400 when app creds exist but runtime cred don't yet exist
+            if e.response.status_code == 404 or e.response.status_code == 400:
                 return None
             raise e
 

ibm_watsonx_orchestrate/client/knowledge_bases/knowledge_base_client.py

@@ -15,10 +15,10 @@ class KnowledgeBaseClient(BaseAPIClient):
         self.base_endpoint = "/orchestrate/knowledge-bases" if is_local_dev(self.base_url) else "/knowledge-bases"
 
     def create(self, payload: dict) -> dict:
-        return self._post_form_data(f"{self.base_endpoint}/documents", data={ "knowledge_base" : json.dumps(payload) })
+        return self._post_form_data(f"{self.base_endpoint}/documents", data=payload)
     
     def create_built_in(self, payload: dict, files: list) -> dict:
-        return self._post_form_data(f"{self.base_endpoint}/documents", data={ "knowledge_base" : json.dumps(payload) }, files=files)
+        return self._post_form_data(f"{self.base_endpoint}/documents", data=payload, files=files)
 
     def get(self) -> dict:
         return self._get(self.base_endpoint)
@@ -38,10 +38,10 @@ class KnowledgeBaseClient(BaseAPIClient):
         return self._get(f"{self.base_endpoint}/{knowledge_base_id}/status")
 
     def update(self, knowledge_base_id: str, payload: dict) -> dict:
-        return self._patch_form_data(f"{self.base_endpoint}/{knowledge_base_id}/documents", data={ "knowledge_base" : json.dumps(payload) })
+        return self._patch_form_data(f"{self.base_endpoint}/{knowledge_base_id}/documents", data=payload)
     
     def update_with_documents(self, knowledge_base_id: str, payload: dict, files: list) -> dict:
-        return self._patch_form_data(f"{self.base_endpoint}/{knowledge_base_id}/documents", data={ "knowledge_base" : json.dumps(payload) }, files=files)
+        return self._patch_form_data(f"{self.base_endpoint}/{knowledge_base_id}/documents", data=payload, files=files)
 
     def delete(self, knowledge_base_id: str,) -> dict:
         return self._delete(f"{self.base_endpoint}/{knowledge_base_id}")

ibm_watsonx_orchestrate/docker/compose-lite.yml

@@ -32,6 +32,7 @@ services:
       - wxo-applied-migrations:/var/lib/postgresql/applied_migrations
     command: -c shared_preload_libraries=pgsodium
 
+
   wxo-server-connection-manager:
     image: ${CM_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-connections:${CM_TAG:-latest}
     platform: linux/amd64
@@ -128,8 +129,11 @@ services:
       IBM_DPS_CACHE_SERVICE: https://wxo-doc-processing-cache:8080
       DOCPROC_BASE_URL: http://wxo-doc-processing-infra-standalone:9080
       BUILDER_ASYNC_CALLBACK_ENDPOINT: http://wxo-builder:4025
+      DOCPROC_ENABLED: ${DOCPROC_ENABLED:-false}
       IS_OBSERVABILITY_FEATURE_ENABLED: "true"
       ALLOW_INSECURE_TLS: "true"
+      ENABLE_EDIT_PROMPTS: "true"
+      LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
     command: 'npm start'
     ports:
       - "4025:4025"
@@ -261,6 +265,7 @@ services:
       - default
       - jaeger-network
     environment:
+      DEFAULT_AGENT_ENABLED: "true"
       RUNTIME_MANAGER_API_KEY: ${RUNTIME_MANAGER_API_KEY:-testapikey} 
       PIP_NO_CACHE_DIR:
       JWT_SECRET: ${JWT_SECRET}
@@ -344,6 +349,7 @@ services:
       AGENTOPS_API_KEY: ${AGENTOPS_API_KEY}
       ES_HOST: http://elasticsearch:9200
       ORIGIN_HEADER: "internal"
+      LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
 
   wxo-server-worker:
     image: ${WORKER_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-server-conversation_controller:${WORKER_TAG:-latest}
@@ -441,6 +447,7 @@ services:
       USE_IBM_TELEMETRY: ${USE_IBM_TELEMETRY:-false}
       WXO_DEPLOYMENT_PLATFORM: saas
       CALLBACK_HOST_URL: ${CALLBACK_HOST_URL:-http://wxo-server:4321}
+      LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
   
   tools-runtime-manager:
     image: ${TRM_REGISTRY:-us.icr.io/watson-orchestrate-private}/tools-runtime-manager:${TRM_TAG:-latest}
@@ -482,7 +489,8 @@ services:
       STORAGE_S3_REGION: us-east-1
       AWS_ACCESS_KEY_ID: ${MINIO_ROOT_USER:-minioadmin}
       AWS_SECRET_ACCESS_KEY: ${MINIO_ROOT_PASSWORD:-watsonxorchestrate}
-      TEMPUS_HOST_NAME: http://wxo-tempus-runtime:9044
+      TEMPUS_HOST_NAME: http://wxo-tempus-runtime:9044      
+      LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
     extra_hosts:
       - "host.docker.internal:host-gateway"
 
@@ -514,6 +522,7 @@ services:
       STORAGE_S3_REGION: us-east-1
       AWS_ACCESS_KEY_ID: ${MINIO_ROOT_USER:-minioadmin}
       AWS_SECRET_ACCESS_KEY: ${MINIO_ROOT_PASSWORD:-watsonxorchestrate}
+      LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
 
   ########################
   # IBM AGENT-OPS
@@ -792,6 +801,7 @@ services:
       LOG_LEVEL: info
       DOCPROC_ENABLED: ${DOCPROC_ENABLED:-false}
       DOCPROC_BASE_URL: http://wxo-doc-processing-infra-standalone:9080
+      DOCPROC_OUTPUT_FILE_SERVER_URL: ${DOCPROC_OUTPUT_FILE_SERVER_URL:-http://wxo-server:4321}
       WO_API_KEY: ${WO_API_KEY}
       WO_INSTANCE: ${WO_INSTANCE}
       AUTHORIZATION_URL: ${AUTHORIZATION_URL}
@@ -872,6 +882,19 @@ services:
       PREFLIGHT_MAX_SIZE: 10Mb
       PREFLIGHT_MAX_PAGES: 600
       RUNTIME_LIBRARY: "watson_doc_understanding"
+      WXAI_API_KEY: ${WXAI_API_KEY:-}
+      WXAI_IMAGE_DESCRIPTION_MODEL_ID: ${WXAI_IMAGE_DESCRIPTION_MODEL_ID:-mistralai/pixtral-12b}
+      WXAI_IMAGE_DESCRIPTION_PROJECT_ID: ${WXAI_IMAGE_DESCRIPTION_PROJECT_ID:-}
+      WXAI_KVP_MODEL_ID: ${WXAI_KVP_MODEL_ID:-}
+      WXAI_KVP_PROJECT_ID: ${WXAI_KVP_PROJECT_ID:-}
+      WXAI_SEMANTIC_KVP_PROJECT_ID: ${WXAI_SEMANTIC_KVP_PROJECT_ID:-}
+      WXAI_SEMANTIC_KVP_MODEL_ID: ${WXAI_SEMANTIC_KVP_MODEL_ID:-mistralai/pixtral-12b}
+      WXAI_SEMANTIC_KVP_SPACE_ID: ${WXAI_SEMANTIC_KVP_SPACE_ID:-}
+      WXAI_URL: ${WXAI_URL:-}
+      WXAI_USERNAME: ${WXAI_USERNAME:-}
+      WXAI_INSTANCE_ID: ${WXAI_INSTANCE_ID:-}
+      WXAI_VERSION: ${WXAI_VERSION:-5.1}
+      
     profiles:
       - docproc
     ports:
@@ -1000,9 +1023,7 @@ services:
       SERVICE_NAME: wxo-doc-processing-llm-service
       LOG_LEVEL: info
       WATSONX_API_ENDPOINT: ${WATSONX_URL:-https://us-south.ml.cloud.ibm.com}
-      WATSONX_API_KEY: ${WATSONX_APIKEY}
-      WATSONX_PROJECT_ID: ${WATSONX_PROJECT_ID}
-      WATSONX_SPACE_ID: ${WATSONX_SPACE_ID}
+      WXO_SERVER_BASE_URL : http://wxo-server:4321
       IBM_DPS_CACHE_SERVICE: 'wxo-doc-processing-cache:8080'
     profiles:
       - docproc
@@ -1096,15 +1117,28 @@ services:
       "
 
   wxo-server-voice:
-      image: ${VOICE_CONTROLLER_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-server-voice:${VOICE_CONTROLLER_TAG:-latest}
-      profiles: [voice]
-      restart: unless-stopped
-      ports: 
-        - 9876:8765
-      environment:
-        WXO_BASE_DOMAIN_URL: "http://wxo-server:4321"
-        AI_GATEWAY_URL: "http://ai-gateway:8787"
-        WXO_ORIGIN_HEADER: "internal"
+    image: ${VOICE_CONTROLLER_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-server-voice:${VOICE_CONTROLLER_TAG:-latest}
+    profiles: [voice]
+    restart: unless-stopped
+    ports: 
+      - 9876:8765
+    environment:
+      WXO_BASE_DOMAIN_URL: "http://wxo-server:4321"
+      AI_GATEWAY_URL: "http://ai-gateway:8787"
+      WXO_ORIGIN_HEADER: "internal"
+
+  langflow:
+    image: ${OPENSOURCE_REGISTRY_PROXY:-docker.io}/langflowai/${LANGFLOW_IMAGE:-langflow}:${LANGFLOW_TAG:-latest}
+    profiles: [langflow]
+    ports:
+      - 7861:7861
+    environment:
+      LANGFLOW_PORT: 7861
+      LANGFLOW_DATABASE_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@wxo-server-db:5432/langflow
+      LANGFLOW_CONFIG_DIR: app/langflow
+    volumes:
+      - langflow-data:/app/langflow
+
 
 volumes:
   tools:
@@ -1138,6 +1172,7 @@ volumes:
   wdu-models:
     driver: local
   es_data:
+  langflow-data:
 
 networks:
   default: