vellum-ai 0.10.9__py3-none-any.whl → 0.11.1__py3-none-any.whl

vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -14,7 +14,6 @@ from vellum.workflows.types.core import JsonArray, JsonObject
 from vellum.workflows.types.generics import WorkflowType
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.nodes.types import PortDisplay
-from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input
 from vellum_ee.workflows.display.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.utils.vellum import infer_vellum_variable_type, primitive_to_vellum_value
@@ -57,8 +56,12 @@ class VellumWorkflowDisplay(
     def serialize(self, raise_errors: bool = True) -> JsonObject:
         input_variables: JsonArray = []
         for workflow_input, workflow_input_display in self.display_context.workflow_input_displays.items():
-            default = primitive_to_vellum_value(raise_if_descriptor(workflow_input.instance)) if workflow_input.instance else None
-            required = type(None) not in workflow_input.types
+            default = primitive_to_vellum_value(workflow_input.instance) if workflow_input.instance else None
+            required = (
+                workflow_input_display.required
+                if workflow_input_display.required is not None
+                else type(None) not in workflow_input.types
+            )
 
             input_variables.append(
                 {
@@ -67,7 +70,7 @@ class VellumWorkflowDisplay(
                     "type": infer_vellum_variable_type(workflow_input),
                     "default": default.dict() if default else None,
                     "required": required,
-                    "extensions": { "color" : workflow_input_display.color },
+                    "extensions": {"color": workflow_input_display.color},
                 }
             )
 
@@ -266,12 +269,16 @@ class VellumWorkflowDisplay(
         self, workflow_input: WorkflowInputReference, overrides: Optional[WorkflowInputsVellumDisplayOverrides] = None
     ) -> WorkflowInputsVellumDisplay:
         workflow_input_id: UUID
+        required = None
+        color = None
         if overrides:
             workflow_input_id = overrides.id
+            required = overrides.required
+            color = overrides.color
         else:
             workflow_input_id = uuid4_from_hash(f"{self.workflow_id}|inputs|id|{workflow_input.name}")
 
-        return WorkflowInputsVellumDisplay(id=workflow_input_id)
+        return WorkflowInputsVellumDisplay(id=workflow_input_id, required=required, color=color)
 
     def _generate_entrypoint_display(
         self,

vellum/workflows/runner/types.py

@@ -1,16 +0,0 @@
-"""Only intenral types and enums for WorkflowRunner should be defined in this module."""
-
-from dataclasses import dataclass
-from typing import TYPE_CHECKING, Generic
-
-from vellum.workflows.types.generics import StateType
-
-if TYPE_CHECKING:
-    from vellum.workflows.events import NodeEvent
-    from vellum.workflows.nodes.bases import BaseNode
-
-
-@dataclass(frozen=True)
-class WorkItemEvent(Generic[StateType]):
-    node: "BaseNode[StateType]"
-    event: "NodeEvent"