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

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
@@ -102,6 +103,7 @@ services:
       DISMISS_NOTIFICATION_TIMEOUT: 10000
       STANDALONE: "true"
       STREAM_TIMEOUT: ${STREAM_TIMEOUT:-120000}
+      WXO_VOICE_URL: http://localhost:9876/v1
     command: "./StartServer.sh"
     ports:
       - "3000:4002"
@@ -127,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"
@@ -260,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}
@@ -342,6 +348,8 @@ services:
       AGENTOPS_API_KEY_AUTH_ENABLED: ${AGENTOPS_API_KEY_AUTH_ENABLED:-false}
       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}
@@ -439,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}
@@ -480,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"
 
@@ -512,12 +522,13 @@ 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
   ########################
   elasticsearch:
-    image: docker.elastic.co/elasticsearch/elasticsearch:8.17.5
+    image: docker.elastic.co/elasticsearch/elasticsearch:8.19.0
     profiles: [ibm-telemetry]
     environment:
       - discovery.type=single-node
@@ -790,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}
@@ -870,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:
@@ -998,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
@@ -1093,6 +1116,30 @@ services:
       exit 0;
       "
 
+  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"
+
+  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:
     driver: local
@@ -1125,6 +1172,7 @@ volumes:
   wdu-models:
     driver: local
   es_data:
+  langflow-data:
 
 networks:
   default:

ibm_watsonx_orchestrate/docker/default.env

@@ -57,13 +57,14 @@ EVENT_BROKER_TTL="3600"
 REGISTRY_URL=
 
 
-SERVER_TAG=01-08-2025
+
+SERVER_TAG=29-08-2025-59ef405
 SERVER_REGISTRY=
 
-WORKER_TAG=01-08-2025-v2
+WORKER_TAG=29-08-2025-cf32b2d
 WORKER_REGISTRY=
 
-AI_GATEWAY_TAG=01-08-2025-v1
+AI_GATEWAY_TAG=20-08-2025-9ed6d40
 AI_GATEWAY_REGISTRY=
 
 AGENT_GATEWAY_TAG=29-07-2025
@@ -77,21 +78,21 @@ 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
 
-TRM_TAG=23-07-2025-3c60549f0bac275de3e5736265a3fd49cdd3a203
+TRM_TAG=19-08-2025-fe105eb0b950ff304f712a1a5b9fa3cba92d09da
 TRM_REGISTRY=
 
-TR_TAG=23-07-2025-3c60549f0bac275de3e5736265a3fd49cdd3a203
+TR_TAG=25-08-2025-58ae475
 TR_REGISTRY=
 
-BUILDER_TAG=31-07-2025-d7145cb
+BUILDER_TAG=27-08-2025-7432aca
 BUILDER_REGISTRY=
 
-FLOW_RUNTIME_TAG=01-08-2025
+FLOW_RUNTIME_TAG=18-08-2025-v3
 FLOW_RUMTIME_REGISTRY=
 
 
@@ -104,17 +105,23 @@ JAEGER_PROXY_REGISTRY=
 SOCKET_HANDLER_TAG=29-05-2025
 SOCKET_HANDLER_REGISTRY=
 
-CPE_TAG=06-08-2025-b0a20ad
+CPE_TAG=18-08-2025-ae1308e
 CPE_REGISTRY=
 
+VOICE_CONTROLLER_TAG=12-08-2025
+VOICE_CONTROLLER_REGISTRY=
+
+LANGFLOW_TAG=
+LANGFLOW_IMAGE=
+
 # IBM Document Processing
-WDU_TAG=2.5.0
+WDU_TAG=2.6.1
 WDU_REGISTRY=
 
-DOCPROC_DPS_TAG=20250721-164412-250-503756a
-DOCPROC_LLMSERVICE_TAG=20250725-100249-111-51d3e51
-DOCPROC_CACHE_TAG=20250723-100852-70-9edc1ab
-DOCPROC_DPI_TAG=20250731-155328-257-06879e86
+DOCPROC_DPS_TAG=20250815-010747-277-173db2a
+DOCPROC_LLMSERVICE_TAG=20250820-153924-128-55cf4d5
+DOCPROC_CACHE_TAG=20250814-master-82-cf33f87
+DOCPROC_DPI_TAG=20250815-004755-273-e65f26b4
 DOCPROC_REGISTRY=
 
 # END -- IMAGE REGISTRIES AND TAGS
@@ -149,6 +156,7 @@ TOOLS_RUNTIME_MANAGER_BASE_URL="http://tools-runtime-manager:8080"
 CONNECTION_SERVICE_BASE_URL="http://wxo-server-connection-manager:3001"
 AI_GATEWAY_BASE_URL="http://ai-gateway:8787/v1"
 AI_GATEWAY_ENABLED=True
+DEFAULT_AGENT_ENABLED=True
 AGENT_GATEWAY_URI="http://wxo-agent-gateway:8989"
 DEFAULT_TENANT_ID=10000000-0000-0000-0000-000000000000
 ES_USERNAME=elastic

ibm_watsonx_orchestrate/flow_builder/flows/__init__.py

@@ -19,6 +19,8 @@ __all__ = [
     "DocProcInput",
 
     "DocProcNode",
+    "DocExtNode",
+    "DocClassifierNode"
     "UserNode",
     "AgentNode",
     "StartNode",

ibm_watsonx_orchestrate/flow_builder/flows/flow.py

@@ -25,13 +25,13 @@ 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 (
-    EndNodeSpec, Expression, ForeachPolicy, ForeachSpec, LoopSpec, BranchNodeSpec, MatchPolicy, PromptLLMParameters, PromptNodeSpec, 
+    DocProcKVPSchema, EndNodeSpec, Expression, ForeachPolicy, ForeachSpec, LoopSpec, BranchNodeSpec, MatchPolicy, PlainTextReadingOrder, PromptLLMParameters, PromptNodeSpec, TimerNodeSpec,
     StartNodeSpec, ToolSpec, JsonSchemaObject, ToolRequestBody, ToolResponseBody, UserFieldKind, UserFieldOption, UserFlowSpec, UserNodeSpec, WaitPolicy,
-    DocProcSpec, TextExtractionResponse, DocProcInput, DecisionsNodeSpec, DecisionsRule, DocExtSpec, File
+    DocProcSpec, TextExtractionResponse, DocProcInput, DecisionsNodeSpec, DecisionsRule, DocExtSpec, File, DocumentClassificationResponse, DocClassifierSpec, DocumentProcessingCommonInput
 )
 from .constants import CURRENT_USER, START, END, ANY_USER
 from ..node import (
-    EndNode, Node, PromptNode, StartNode, UserNode, AgentNode, DataMap, ToolNode, DocProcNode, DecisionsNode, DocExtNode
+    EndNode, Node, PromptNode, StartNode, UserNode, AgentNode, DataMap, ToolNode, DocProcNode, DecisionsNode, DocExtNode, DocClassifierNode
 )
 from ..types import (
     AgentNodeSpec, extract_node_spec, FlowContext, FlowEventType, FlowEvent, FlowSpec,
@@ -115,6 +115,10 @@ class Flow(Node):
                 # we need a deep compare if the incoming schema and existing_schema is the same
                 # pydantic suppport nested comparison by default
 
+                if isinstance(schema, dict):
+                    # recast schema to support direct access
+                    schema = JsonSchemaObject.model_validate(schema)
+
                 schema.title = title
 
                 if schema == existing_schema:
@@ -199,30 +203,31 @@ class Flow(Node):
         self._refactor_spec_to_schemaref(node.spec)
                 
     def _refactor_spec_to_schemaref(self, spec: NodeSpec):
-        if spec.input_schema and (spec.input_schema.type == "object" or spec.input_schema.type == "array") :
+        if spec.input_schema and not isinstance(spec.input_schema, SchemaRef) and (spec.input_schema.type == "object" or spec.input_schema.type == "array") :
             if isinstance(spec.input_schema, ToolRequestBody):
                 spec.input_schema = self._add_schema_ref(JsonSchemaObject(type = spec.input_schema.type,
                                                                                 properties= spec.input_schema.properties,
                                                                                 required= spec.input_schema.required), 
                                                                 f"{spec.name}_input")
-        if spec.output_schema_object is not None and spec.output_schema_object.type == "object":
-            spec.output_schema = self._add_schema_ref(spec.output_schema_object, spec.output_schema_object.title)
-            spec.output_schema_object = None
-        elif spec.output_schema is not None:
-            if isinstance(spec.output_schema, ToolResponseBody):
-                if spec.output_schema.type == "object":
-                    json_obj = JsonSchemaObject(type = spec.output_schema.type,
-                            description=spec.output_schema.description,
-                            properties= spec.output_schema.properties,
-                            items = spec.output_schema.items,
-                            uniqueItems=spec.output_schema.uniqueItems,
-                            anyOf=spec.output_schema.anyOf,
-                            required= spec.output_schema.required)
-                    spec.output_schema = self._add_schema_ref(json_obj, f"{spec.name}_output")
-                elif spec.output_schema.type == "array":
-                    if hasattr(spec.output_schema, "items") and hasattr(spec.output_schema.items, "type") and spec.output_schema.items.type == "object":
-                        schema_ref = self._add_schema_ref(spec.output_schema.items)
-                        spec.output_schema.items = JsonSchemaObjectRef(ref=f"{schema_ref.ref}")
+        if not isinstance(spec.output_schema, SchemaRef):
+            if spec.output_schema_object is not None and spec.output_schema_object.type == "object":
+                spec.output_schema = self._add_schema_ref(spec.output_schema_object, spec.output_schema_object.title)
+                spec.output_schema_object = None
+            elif spec.output_schema is not None:
+                if isinstance(spec.output_schema, ToolResponseBody):
+                    if spec.output_schema.type == "object":
+                        json_obj = JsonSchemaObject(type = spec.output_schema.type,
+                                description=spec.output_schema.description,
+                                properties= spec.output_schema.properties,
+                                items = spec.output_schema.items,
+                                uniqueItems=spec.output_schema.uniqueItems,
+                                anyOf=spec.output_schema.anyOf,
+                                required= spec.output_schema.required)
+                        spec.output_schema = self._add_schema_ref(json_obj, f"{spec.name}_output")
+                    elif spec.output_schema.type == "array":
+                        if hasattr(spec.output_schema, "items") and hasattr(spec.output_schema.items, "type") and spec.output_schema.items.type == "object":
+                            schema_ref = self._add_schema_ref(spec.output_schema.items)
+                            spec.output_schema.items = JsonSchemaObjectRef(ref=f"{schema_ref.ref}")
 
     # def refactor_datamap_spec_to_schemaref(self, spec: FnDataMapSpec):
     #    '''TODO'''
@@ -433,23 +438,95 @@ class Flow(Node):
         node = self._add_node(node)
         return cast(PromptNode, node)
     
+    def docclassfier(self, 
+            name: str, 
+            llm : str = "watsonx/meta-llama/llama-3-2-90b-vision-instruct",
+            version: str = "TIP",
+            display_name: str| None = None,
+            classes: type[BaseModel]| None = None, 
+            description: str | None = None,
+            min_confidence: float = 0.0,
+            input_map: DataMap = None) -> DocClassifierNode:
+        
+        if name is None :
+            raise ValueError("name must be provided.")
+        
+        doc_classifier_config = DocClassifierNode.generate_config(llm=llm, min_confidence=min_confidence,input_classes=classes)
+
+        input_schema_obj = _get_json_schema_obj(parameter_name = "input", type_def = DocumentProcessingCommonInput)
+        output_schema_obj = _get_json_schema_obj(parameter_name = "output", type_def = DocumentClassificationResponse)
+        
+        if "$defs" in output_schema_obj.model_extra:
+            output_schema_obj.model_extra.pop("$defs")
+        # Create the docclassifier spec
+        task_spec = DocClassifierSpec(
+            name=name,
+            display_name=display_name if display_name is not None else name,
+            description=description,
+            input_schema=_get_tool_request_body(input_schema_obj),
+            output_schema=_get_tool_response_body(output_schema_obj),
+            output_schema_object = output_schema_obj,
+            config=doc_classifier_config,
+            version=version
+        )
+        node = DocClassifierNode(spec=task_spec)
+
+        # setup input map
+        if input_map:
+            node.input_map = self._get_data_map(input_map)
+        
+        # add the node to the list of node
+        
+        node = self._add_node(node)
+        return cast(DocClassifierNode, node)
+        
+        
+    def timer(self,
+          name: str,
+          delay: int,
+          display_name: str | None = None,
+          description: str | None = None,
+          input_map: DataMap = None) -> Node:
+
+        if name is None:
+            raise ValueError("name must be provided.")
+        if delay < 0:
+            raise ValueError("delay must be non-negative.")
+
+        timer_spec = TimerNodeSpec(
+            name=name,
+            display_name=display_name if display_name is not None else name,
+            description=description,
+            delay=delay
+        )
+
+        node = Node(spec=timer_spec)
+
+        if input_map:
+            node.input_map = self._get_data_map(input_map)
+
+        node = self._add_node(node)
+        return node
+
+    
     def docext(self, 
             name: str, 
-            llm : str = "meta-llama/llama-3-2-11b-vision-instruct",
+            llm : str = "watsonx/meta-llama/llama-3-2-90b-vision-instruct",
             version: str = "TIP",
             display_name: str| None = None,
-            input_entities: type[BaseModel]| None = None, 
+            fields: type[BaseModel]| None = None, 
             description: str | None = None,
-            input_map: DataMap = None) -> tuple[DocExtNode, type[BaseModel]]:
+            input_map: DataMap = None,
+            enable_hw: bool = False) -> tuple[DocExtNode, type[BaseModel]]:
         
         if name is None :
             raise ValueError("name must be provided.")
 
-        doc_ext_config = DocExtNode.generate_config(llm=llm, input_entites=input_entities)
+        doc_ext_config = DocExtNode.generate_config(llm=llm, fields=fields)
 
-        DocExtFieldValue = DocExtNode.generate_docext_field_value_model(input_entities=input_entities)
+        DocExtFieldValue = DocExtNode.generate_docext_field_value_model(fields=fields)
         
-        input_schema_obj = _get_json_schema_obj(parameter_name = "input", type_def = File)
+        input_schema_obj = _get_json_schema_obj(parameter_name = "input", type_def = DocumentProcessingCommonInput)
         output_schema_obj = _get_json_schema_obj("output", DocExtFieldValue)
 
         if "$defs" in output_schema_obj.model_extra:
@@ -464,7 +541,8 @@ class Flow(Node):
             output_schema=_get_tool_response_body(output_schema_obj),
             output_schema_object = output_schema_obj,
             config=doc_ext_config,
-            version=version
+            version=version,
+            enable_hw=enable_hw
         )
         node = DocExtNode(spec=task_spec)
 
@@ -523,9 +601,12 @@ class Flow(Node):
     def docproc(self, 
             name: str, 
             task: str,
+            plain_text_reading_order : PlainTextReadingOrder = PlainTextReadingOrder.block_structure,
             display_name: str|None=None,
             description: str | None = None,
-            input_map: DataMap = None) -> DocProcNode:
+            input_map: DataMap = None,
+            kvp_schemas: list[DocProcKVPSchema] = None,
+            enable_hw: bool = False) -> DocProcNode:
 
         if name is None :
             raise ValueError("name must be provided.")
@@ -547,7 +628,10 @@ class Flow(Node):
             input_schema=_get_tool_request_body(input_schema_obj),
             output_schema=_get_tool_response_body(output_schema_obj),
             output_schema_object = output_schema_obj,
-            task=task
+            task=task,
+            plain_text_reading_order=plain_text_reading_order,
+            enable_hw=enable_hw,
+            kvp_schemas=kvp_schemas
         )
 
         node = DocProcNode(spec=task_spec)

ibm_watsonx_orchestrate/flow_builder/node.py

@@ -6,8 +6,8 @@ import yaml
 from pydantic import BaseModel, Field, SerializeAsAny, create_model
 from enum import Enum
 
-from .types import EndNodeSpec, NodeSpec, AgentNodeSpec, PromptNodeSpec, StartNodeSpec, ToolNodeSpec, UserFieldKind, UserFieldOption, UserNodeSpec, DocProcSpec, \
-                    DocExtSpec, DocExtConfig, LanguageCode, DecisionsNodeSpec
+from .types import DocExtConfigField, EndNodeSpec, NodeSpec, AgentNodeSpec, PromptNodeSpec, TimerNodeSpec, StartNodeSpec, ToolNodeSpec, UserFieldKind, UserFieldOption, UserNodeSpec, DocProcSpec, \
+                    DocExtSpec, DocExtConfig, DocClassifierSpec, DecisionsNodeSpec, DocClassifierConfig
 
 from .data_map import DataMap
 
@@ -120,6 +120,24 @@ class DocProcNode(Node):
     def get_spec(self) -> DocProcSpec:
         return cast(DocProcSpec, self.spec)
     
+class DocClassifierNode(Node):
+    def __repr__(self):
+        return f"DocClassifierNode(name='{self.spec.name}', description='{self.spec.description}')"
+
+    def get_spec(self) -> DocClassifierSpec:
+        return cast(DocClassifierSpec, self.spec)
+
+    @staticmethod
+    def generate_config(llm: str, input_classes: type[BaseModel], min_confidence: float) -> DocClassifierConfig:
+        return DocClassifierConfig(llm=llm, classes=input_classes.__dict__.values(), min_confidence=min_confidence)
+    
+class TimerNode(Node):
+    def __repr__(self):
+        return f"TimerNode(name='{self.spec.name}', description='{self.spec.description}')"
+
+    def get_spec(self) -> TimerNodeSpec:
+        return cast(TimerNodeSpec, self.spec)
+    
 class DocExtNode(Node):
     def __repr__(self):
         return f"DocExtNode(name='{self.spec.name}', description='{self.spec.description}')"
@@ -128,23 +146,29 @@ class DocExtNode(Node):
         return cast(DocExtSpec, self.spec)
     
     @staticmethod
-    def generate_config(llm: str, input_entites: type[BaseModel]) -> DocExtConfig:
-        entities = input_entites.__dict__.values()
-        return DocExtConfig(llm=llm, entities=entities)
+    def generate_config(llm: str, fields: type[BaseModel]) -> DocExtConfig:
+        return DocExtConfig(llm=llm, fields=fields.__dict__.values())
     
     @staticmethod
-    def generate_docext_field_value_model(input_entities: type[BaseModel]) -> type[BaseModel]:
+    def generate_docext_field_value_model(fields: type[BaseModel]) -> type[BaseModel]:
         create_field_value_description = lambda field_name: "Extracted value for " + field_name
+        field_definitions = {}
+
+        for name, value in fields.model_dump().items():
+            field_type = str  
+            field_kwargs = {
+                "title": value['name'],
+                "description": create_field_value_description(value['name']),
+                "type": value["type"] if value["type"] != "date" else "string"
+            }
+
+            # Add json_schema_extra if type is 'date'
+            if value["type"] == "date":
+                field_kwargs["json_schema_extra"] = {"format": "date"}
+
+            field_definitions[name] = (field_type, Field(**field_kwargs))
 
-        DocExtFieldValue = create_model(
-                "DocExtFieldValue",
-                    **{
-                        name: (str, Field(
-                            title=value['name'], 
-                            description=create_field_value_description(value['name']),
-                            )
-                        )
-                        for name, value in input_entities.model_dump().items()})
+        DocExtFieldValue = create_model("DocExtFieldValue", **field_definitions)
         return DocExtFieldValue
     
 class DecisionsNode(Node):