ibm-watsonx-orchestrate 1.11.0b1__py3-none-any.whl → 1.12.0b1__py3-none-any.whl

ibm_watsonx_orchestrate/docker/compose-lite.yml

@@ -49,8 +49,28 @@ services:
       MAX_POOL: 60
       DEPLOYMENT_MODE: laptop
       SUFFIXLIST: ${CM_SUFFIXLIST:-[]}
+      BACKEND_BASE_URL: http://localhost:3001
+      CALLBACK_URL: http://localhost:3001/auth/oauth2/callback
     ports:
       - 3001:3001
+  
+  wxo-server-connections-ui:
+    image: ${CONNECTIONS_UI_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-connections-ui:${CONNECTIONS_UI_TAG:-latest}
+    platform: linux/amd64
+    profiles: [connections-ui]
+    environment:
+      PORT: 3412
+      WXO_CONNECTIONS_URL: http://wxo-server-connection-manager:3001
+      DEPLOYMENT_MODE: laptop 
+      WXO_URL: http://localhost:3412
+      ENV: production
+      WXO_SERVER_URL: http://localhost:4321
+      TENANT_ID: ${REACT_APP_TENANT_ID}
+      WXO_CHAT_URL: http://localhost:3000
+    ports:
+     - "3412:3412"
+    networks:
+      - default
 
   ai-gateway:
     image: ${AI_GATEWAY_REGISTRY:-us.icr.io/watson-orchestrate-private}/ai-gateway:${AI_GATEWAY_TAG:-latest}
@@ -73,6 +93,24 @@ services:
     environment:
       JWT_SECRET: ${JWT_SECRET}
       DEPLOYMENT_PLATFORM: laptop-lite
+      CALLBACK_HOST_URL: ${CALLBACK_HOST_URL:-}
+      REDIS_URL: redis://wxo-server-redis:6379/0
+    entrypoint: >
+      sh -c '
+        if [ ! -z "$$CALLBACK_HOST_URL" ]; then
+          GATEWAY_IP=$$(echo "$$CALLBACK_HOST_URL" | sed "s|http://||" | cut -d: -f1)
+          export CALL_BACK_URL="http://$$GATEWAY_IP:8989"
+          echo "Auto-configured CALL_BACK_URL: $$CALL_BACK_URL"
+        else
+          export CALL_BACK_URL="http://wxo-agent-gateway:8989"
+          echo "Using fallback CALL_BACK_URL: $$CALL_BACK_URL"
+        fi
+        if [ $$# -eq 0 ]; then
+          exec ./conditional_script.sh
+        else
+          exec "$$@"
+        fi
+      '
 
   ui:
     image: ${UI_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-chat:${UITAG:-latest}
@@ -105,6 +143,7 @@ services:
       STANDALONE: "true"
       STREAM_TIMEOUT: ${STREAM_TIMEOUT:-120000}
       WXO_VOICE_URL: http://localhost:9876/v1
+      DPS_CACHE_URL: http://localhost:18083
     command: "./StartServer.sh"
     ports:
       - "3000:4002"
@@ -127,14 +166,15 @@ services:
       AGENTS_OPS_RUNTIME_ENDPOINT: https://frontend-server:443
       DPI_WO_WDU_SERVER_ENDPOINT: https://wxo-doc-processing-service:8080
       DPI_RAG_SERVER_ENDPOINT: https://wxo-doc-processing-llm-service:8083
-      IBM_DPS_CACHE_SERVICE: https://wxo-doc-processing-cache:8080
+      IBM_DPS_CACHE_SERVICE: http://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: "false"
+      ENABLE_EDIT_PROMPTS: "true"
       LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
+      ENABLE_EMBED_SCRIPT: "true"
     command: 'npm start'
     ports:
       - "4025:4025"
@@ -352,6 +392,10 @@ services:
       ES_HOST: http://elasticsearch:9200
       ORIGIN_HEADER: "internal"
       LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
+      WATSONX_SERVER: WXO
+      EVALUATION_PLATFORM: lite-local
+      AGENTOPS_BACKEND_URL: https://frontend-server:443
+      OBSERVABILITY_POSTGRES_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@wxo-server-db:5432/wxo_observability
 
   wxo-server-worker:
     image: ${WORKER_REGISTRY:-us.icr.io/watson-orchestrate-private}/wxo-server-conversation_controller:${WORKER_TAG:-latest}
@@ -452,6 +496,10 @@ services:
       CPD_VERIFY: ${CPD_VERIFY}
       CALLBACK_HOST_URL: ${CALLBACK_HOST_URL:-http://wxo-server:4321}
       LANGFLOW_ENABLED: ${LANGFLOW_ENABLED:-false}
+      WATSONX_SERVER: WXO
+      EVALUATION_PLATFORM: lite-local
+      AGENTOPS_BACKEND_URL: https://frontend-server:443
+      OBSERVABILITY_POSTGRES_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@wxo-server-db:5432/wxo_observability
   
   tools-runtime-manager:
     image: ${TRM_REGISTRY:-us.icr.io/watson-orchestrate-private}/tools-runtime-manager:${TRM_TAG:-latest}
@@ -812,6 +860,7 @@ services:
       AUTHORIZATION_URL: ${AUTHORIZATION_URL}
       WO_AUTH_TYPE: ${WO_AUTH_TYPE}
       FLOW_CALLBACK_REQUEST_RETRIES: "1"
+      DOCPROC_CACHE_SERVER_URL: http://wxo-doc-processing-cache:8080
     healthcheck:
       test: curl -k http://localhost:9044/readiness --fail
       interval: 30s
@@ -857,13 +906,13 @@ services:
       SERVICE_NAME: wxo-doc-processing-service
       LOG_LEVEL: info
       VERIFY_CLIENT_CERT: 0
-      SERVICE_URL: ${SERVICE_URL:-https://wxo-doc-processing-cache:8080}
+      SERVICE_URL: ${SERVICE_URL:-http://wxo-doc-processing-cache:8080}
       RATE_LIMITING_ENABLED: "false"
       SHOULD_CACHE_IMAGES: "false"
       HANDWRITING_ENABLED: "true"
       WDU_RUNTIME_URL: wdu-runtime:8080
       WDU_TIMEOUT_SECS: "180"
-      IBM_DPS_CACHE_SERVICE: https://wxo-doc-processing-cache:8080
+      IBM_DPS_CACHE_SERVICE: http://wxo-doc-processing-cache:8080
     profiles:
       - docproc
     depends_on:
@@ -887,18 +936,18 @@ 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_API_KEY: ${WXAI_API_KEY:-"gateway"}
+      WXAI_IMAGE_DESCRIPTION_MODEL_ID: ${WXAI_IMAGE_DESCRIPTION_MODEL_ID:-mistralai/mistral-small-3-1-24b-instruct-2503}
+      WXAI_IMAGE_DESCRIPTION_SPACE_ID: ${WATSONX_SPACE_ID:-}
+      WXAI_KVP_MODEL_ID: ${WXAI_KVP_MODEL_ID:-mistralai/mistral-small-3-1-24b-instruct-2503}
+      WXAI_KVP_SPACE_ID: ${WATSONX_SPACE_ID:-}
+      WXAI_SEMANTIC_KVP_MODEL_ID: ${WXAI_SEMANTIC_KVP_MODEL_ID:-mistralai/mistral-small-3-1-24b-instruct-2503}
+      WXAI_SEMANTIC_KVP_SPACE_ID: ${WATSONX_SPACE_ID:-}
+      WXAI_URL: ${AI_GATEWAY_BASE_URL:-}
       WXAI_USERNAME: ${WXAI_USERNAME:-}
       WXAI_INSTANCE_ID: ${WXAI_INSTANCE_ID:-}
-      WXAI_VERSION: ${WXAI_VERSION:-5.1}
+      WXAI_VERSION: ${WXAI_VERSION:-2023-05-29}
+      WXAI_PROVIDER: ${WXAI_PROVIDER:-gateway}
       
     profiles:
       - docproc
@@ -958,6 +1007,8 @@ services:
       VERIFY_CLIENT_CERT: 1
       # S3_USE_SSL: false
       # S3_VERIFY_CERT: false
+      ISTIO_ENABLED: true
+      IS_ADK_ENV: true
     profiles:
       - docproc
     depends_on:
@@ -970,7 +1021,7 @@ services:
     # volumes:
       # - ./docker/wo_doc_processing_rag_base/etc/certs:/etc/certs
     healthcheck:
-      test: ["CMD-SHELL", "curl -fk https://localhost:8080/ping"]
+      test: ["CMD-SHELL", "curl -fk http://localhost:8080/ping"]
       interval: 30s
       timeout: 30s
       retries: 10
@@ -1029,7 +1080,7 @@ services:
       LOG_LEVEL: info
       WATSONX_API_ENDPOINT: ${WATSONX_URL:-https://us-south.ml.cloud.ibm.com}
       WXO_SERVER_BASE_URL : http://wxo-server:4321
-      IBM_DPS_CACHE_SERVICE: 'wxo-doc-processing-cache:8080'
+      IBM_DPS_CACHE_SERVICE: 'http://wxo-doc-processing-cache:8080'
     profiles:
       - docproc
     depends_on:
@@ -1140,7 +1191,7 @@ services:
     environment:
       LANGFLOW_PORT: 7861
       LANGFLOW_DATABASE_URL: postgresql://${POSTGRES_USER:-postgres}:${POSTGRES_PASSWORD:-postgres}@wxo-server-db:5432/langflow
-      LANGFLOW_CONFIG_DIR: app/langflow
+      LANGFLOW_CONFIG_DIR: /app/app/langflow
     volumes:
       - langflow-data:/app/langflow
 

ibm_watsonx_orchestrate/docker/default.env

@@ -58,41 +58,44 @@ REGISTRY_URL=
 
 
 
-SERVER_TAG=05-09-2025-Hb2zH5Ggf
+SERVER_TAG=17-10-2025-8a9aff2
 SERVER_REGISTRY=
 
-WORKER_TAG=05-09-2025-Hb2zH5Ggf
+WORKER_TAG=17-10-2025-8a9aff2
 WORKER_REGISTRY=
 
 AI_GATEWAY_TAG=20-08-2025-9ed6d40
 AI_GATEWAY_REGISTRY=
 
-AGENT_GATEWAY_TAG=29-07-2025
+AGENT_GATEWAY_TAG=16-09-2025-115ef29
 AGENT_GATEWAY_REGISTRY=
 
 DB_REGISTRY=
 # If you build multiarch set all three of these to the same, we have a pr against main
 # to not have this separation, but we can merge it later
-DBTAG=04-09-2025-48c39
-AMDDBTAG=04-09-2025-48c39
-ARM64DBTAG=04-09-2025-48c39
+DBTAG=17-09-2025-8a9aff2
+AMDDBTAG=17-09-2025-8a9aff2
+ARM64DBTAG=17-09-2025-8a9aff2
 
 UI_REGISTRY=
-UITAG=29-08-2025
+UITAG=19-09-2025-188b5fe
 
 CM_REGISTRY=
-CM_TAG=24-07-2025
+CM_TAG=16-09-2025-e33b344
 
-TRM_TAG=19-08-2025-fe105eb0b950ff304f712a1a5b9fa3cba92d09da
+CONNECTIONS_UI_REGISTRY=
+CONNECTIONS_UI_TAG=15-09-2025-98aa9da
+
+TRM_TAG=15-09-2025-228f397
 TRM_REGISTRY=
 
-TR_TAG=25-08-2025-58ae475
+TR_TAG=15-09-2025-228f397
 TR_REGISTRY=
 
-BUILDER_TAG=27-08-2025-7432aca
+BUILDER_TAG=19-09-2025-4ab5c88
 BUILDER_REGISTRY=
 
-FLOW_RUNTIME_TAG=26-08-2025-3b7f19e
+FLOW_RUNTIME_TAG=15-09-2025-c208dc5
 FLOW_RUMTIME_REGISTRY=
 
 
@@ -108,20 +111,20 @@ SOCKET_HANDLER_REGISTRY=
 CPE_TAG=29-08-2025-e612bea
 CPE_REGISTRY=
 
-VOICE_CONTROLLER_TAG=12-08-2025
+VOICE_CONTROLLER_TAG=12-09-2025-0e04772
 VOICE_CONTROLLER_REGISTRY=
 
 LANGFLOW_TAG=
 LANGFLOW_IMAGE=
 
 # IBM Document Processing
-WDU_TAG=2.6.1
+WDU_TAG=2.7.0
 WDU_REGISTRY=
 
-DOCPROC_DPS_TAG=20250826-201302-283-24eb106
-DOCPROC_LLMSERVICE_TAG=20250826-main-133-42a5320
-DOCPROC_CACHE_TAG=20250814-master-82-cf33f87
-DOCPROC_DPI_TAG=20250828-154151-285-e4dd1cff
+DOCPROC_DPS_TAG=20250910-165658-290-c566031
+DOCPROC_LLMSERVICE_TAG=20250915-main-139-7a36ad3
+DOCPROC_CACHE_TAG=20250916-master-86-454157f
+DOCPROC_DPI_TAG=20250910-214413-288-a348dfd9
 DOCPROC_REGISTRY=
 
 # END -- IMAGE REGISTRIES AND TAGS

ibm_watsonx_orchestrate/flow_builder/flows/decorators.py

@@ -8,6 +8,8 @@ import logging
 import inspect
 from typing import Callable, Optional, Sequence
 from pydantic import BaseModel
+
+from ...agent_builder.models.types import ListVirtualModel
 from ..types import extract_node_spec, UserNodeSpec, FlowSpec
 
 from .flow import FlowFactory, Flow
@@ -34,7 +36,9 @@ def flow(*args,
          input_schema: type[BaseModel] | None = None,
          output_schema: type[BaseModel] | None = None,
          initiators: Sequence[str] = (),
-         schedulable: bool = False):
+         schedulable: bool = False,
+         llm_model: str|ListVirtualModel|None=None,
+         agent_conversation_memory_turns_limit: int|None=None):
     """Decorator to mark a function as a flow model builder."""
 
     def decorator(func: Callable):
@@ -60,7 +64,9 @@ def flow(*args,
                              input_schema = input_schema,
                              output_schema = output_schema,
                              initiators = initiators,
-                             schedulable = schedulable)
+                             schedulable = schedulable,
+                             llm_model = llm_model,
+                             agent_conversation_memory_turns_limit = agent_conversation_memory_turns_limit)
 
         # logger.info("Creating flow model: %s", a_model.spec.name)
 

ibm_watsonx_orchestrate/flow_builder/flows/flow.py

@@ -21,6 +21,7 @@ from typing_extensions import Self
 from pydantic import BaseModel, Field, SerializeAsAny, create_model, TypeAdapter
 import yaml
 from ibm_watsonx_orchestrate.agent_builder.tools.python_tool import PythonTool
+from ibm_watsonx_orchestrate.agent_builder.models.types import ListVirtualModel
 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
@@ -84,6 +85,21 @@ class Flow(Node):
         # get Tool Client
         self._tool_client = instantiate_client(ToolClient)
 
+        # set llm_model to use for the flow if any
+        llm_model = kwargs.get("llm_model")
+        if llm_model:
+            if isinstance(llm_model, ListVirtualModel):
+                self.metadata["llm_model"] = llm_model.name
+            elif isinstance(llm_model, str):
+                self.metadata["llm_model"] = llm_model
+            else:
+                raise AssertionError(f"flow llm_model should be either a str or ListVirtualModel")
+        
+        # set agent_conversation_memory_turns_limit for the flow if any
+        agent_conversation_memory_turns_limit = kwargs.get("agent_conversation_memory_turns_limit")
+        if agent_conversation_memory_turns_limit:
+            self.metadata["agent_conversation_memory_turns_limit"] = agent_conversation_memory_turns_limit
+
     def _find_topmost_flow(self) -> Self:
         if self.parent:
             return self.parent._find_topmost_flow()
@@ -519,7 +535,9 @@ class Flow(Node):
             fields: type[BaseModel]| None = None, 
             description: str | None = None,
             input_map: DataMap = None,
-            enable_hw: bool = False) -> tuple[DocExtNode, type[BaseModel]]:
+            enable_hw: bool = False,
+            min_confidence: float = 0, # Setting a small value because htil is not supported for pro code. 
+            review_fields: List[str] = []) -> tuple[DocExtNode, type[BaseModel]]:
         
         if name is None :
             raise ValueError("name must be provided.")
@@ -544,7 +562,9 @@ class Flow(Node):
             output_schema_object = output_schema_obj,
             config=doc_ext_config,
             version=version,
-            enable_hw=enable_hw
+            enable_hw=enable_hw,
+            min_confidence=min_confidence,
+            review_fields=review_fields
         )
         node = DocExtNode(spec=task_spec)
 
@@ -609,7 +629,8 @@ class Flow(Node):
             input_map: DataMap = None,
             document_structure: bool = False,
             kvp_schemas: list[DocProcKVPSchema] = None,
-            enable_hw: bool = False) -> DocProcNode:
+            enable_hw: bool = False,
+            kvp_model_name: str | None = None) -> DocProcNode:
 
         if name is None :
             raise ValueError("name must be provided.")
@@ -635,7 +656,8 @@ class Flow(Node):
             document_structure=document_structure,
             plain_text_reading_order=plain_text_reading_order,
             enable_hw=enable_hw,
-            kvp_schemas=kvp_schemas
+            kvp_schemas=kvp_schemas,
+            kvp_model_name=kvp_model_name
         )
 
         node = DocProcNode(spec=task_spec)
@@ -1201,7 +1223,7 @@ class CompiledFlow(BaseModel):
         dumped = self.flow.to_json()
         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:
@@ -1223,7 +1245,9 @@ class FlowFactory(BaseModel):
                     initiators: Sequence[str]|None=None,
                     input_schema: type[BaseModel]|None=None,
                     output_schema: type[BaseModel]|None=None,
-                    schedulable: bool=False) -> Flow:
+                    schedulable: bool=False,
+                    llm_model: str|ListVirtualModel|None=None,
+                    agent_conversation_memory_turns_limit: int|None = None) -> Flow:
         if isinstance(name, Callable):
             flow_spec = getattr(name, "__flow_spec__", None)
             if not flow_spec:
@@ -1248,7 +1272,7 @@ class FlowFactory(BaseModel):
             schedulable=schedulable,
         )
 
-        return Flow(spec = flow_spec)
+        return Flow(spec = flow_spec, llm_model=llm_model, agent_conversation_memory_turns_limit=agent_conversation_memory_turns_limit)
 
 
 class FlowControl(Node):

ibm_watsonx_orchestrate/flow_builder/node.py

@@ -23,7 +23,7 @@ class Node(BaseModel):
                                       exclude_unset=True, exclude_none=True, by_alias=True)
         with open(file, 'w', encoding="utf-8") 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/flow_builder/types.py

@@ -271,6 +271,8 @@ class DocClassifierSpec(DocProcCommonNodeSpec):
 class DocExtSpec(DocProcCommonNodeSpec):
     version : str = Field(description="A version of the spec")
     config : DocExtConfig
+    min_confidence: float = Field(description="The minimal confidence acceptable for an extracted field value", default=0.0,le=1.0, ge=0.0 ,title="Minimum Confidence")
+    review_fields: List[str] = Field(description="The fields that require user to review", default=[])
 
     def __init__(self, **data):
         super().__init__(**data)
@@ -281,6 +283,8 @@ class DocExtSpec(DocProcCommonNodeSpec):
         model_spec["version"] = self.version
         model_spec["config"] = self.config.model_dump()
         model_spec["task"] = DocProcTask.custom_field_extraction
+        model_spec["min_confidence"] = self.min_confidence
+        model_spec["review_fields"] = self.review_fields
         return model_spec
     
 class DocProcField(BaseModel):
@@ -337,6 +341,11 @@ class DocProcSpec(DocProcCommonNodeSpec):
         title='KVP schemas',
         description="Optional list of key-value pair schemas to use for extraction.",
         default=None)
+    kvp_model_name: str | None = Field(
+        title='KVP Model Name',
+        description="The LLM model to be used for key-value pair extraction",
+        default=None
+    )
     plain_text_reading_order : PlainTextReadingOrder = Field(default=PlainTextReadingOrder.block_structure)
     document_structure: bool = Field(default=False,description="Requests the entire document structure computed by WDU to be returned")
     
@@ -352,6 +361,8 @@ class DocProcSpec(DocProcCommonNodeSpec):
             model_spec["plain_text_reading_order"] = self.plain_text_reading_order
         if self.kvp_schemas is not None:
             model_spec["kvp_schemas"] = self.kvp_schemas
+        if self.kvp_model_name is not None:
+            model_spec["kvp_model_name"] = self.kvp_model_name
         return model_spec
 
 class StartNodeSpec(NodeSpec):
@@ -905,8 +916,7 @@ class UserFlowSpec(FlowSpec):
 class ForeachPolicy(Enum):
  
     SEQUENTIAL = 1
-    # support only SEQUENTIAL for now
-    # PARALLEL = 2
+    PARALLEL = 2
 
 class ForeachSpec(FlowSpec):
  
@@ -923,7 +933,7 @@ class ForeachSpec(FlowSpec):
         if isinstance(self.item_schema, JsonSchemaObject):
             my_dict["item_schema"] = _to_json_from_json_schema(self.item_schema)
         else:
-            my_dict["item_schema"] = self.item_schema.model_dump(exclude_defaults=True, exclude_none=True, exclude_unset=True)
+            my_dict["item_schema"] = self.item_schema.model_dump(exclude_defaults=True, exclude_none=True, exclude_unset=True, by_alias=True)
 
         my_dict["foreach_policy"] = self.foreach_policy.name
         return my_dict
@@ -1312,6 +1322,11 @@ class DocProcInput(DocumentProcessingCommonInput):
         title='KVP schemas',
         description="Optional list of key-value pair schemas to use for extraction.",
         default=None)
+    kvp_model_name: str | None = Field(
+        title='KVP Model Name',
+        description="The LLM model to be used for key-value pair extraction",
+        default=None
+    )
 
 class TextExtractionResponse(BaseModel):
     '''