vellum-ai 0.10.8__py3-none-any.whl → 0.11.0__py3-none-any.whl

vellum_ee/workflows/display/types.py

@@ -1,6 +1,10 @@
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Dict, Generic, Tuple, Type, TypeVar
 
+from vellum.workflows.descriptors.base import BaseDescriptor
+from vellum.workflows.nodes import BaseNode
+from vellum.workflows.ports import Port
+from vellum.workflows.references import OutputReference, WorkflowInputReference
 from vellum_ee.workflows.display.base import (
     EdgeDisplayType,
     EntrypointDisplayType,
@@ -9,10 +13,6 @@ from vellum_ee.workflows.display.base import (
     WorkflowOutputDisplayType,
 )
 from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplay
-from vellum.workflows.descriptors.base import BaseDescriptor
-from vellum.workflows.nodes import BaseNode
-from vellum.workflows.ports import Port
-from vellum.workflows.references import OutputReference, WorkflowInputReference
 
 if TYPE_CHECKING:
     from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay

vellum_ee/workflows/display/utils/vellum.py

@@ -4,9 +4,7 @@ import typing
 from typing import Any, List, Union, cast
 
 from vellum import ChatMessage, SearchResult, SearchResultRequest, VellumVariableType
-from vellum.client.types.array_vellum_value import ArrayVellumValue
 from vellum.workflows.descriptors.base import BaseDescriptor
-from vellum.workflows.nodes.core.map_node.node import MapNodeItemType
 from vellum.workflows.references import OutputReference, WorkflowInputReference
 from vellum.workflows.references.execution_count import ExecutionCountReference
 from vellum.workflows.references.node import NodeReference
@@ -48,7 +46,7 @@ def infer_vellum_variable_type(value: Any) -> VellumVariableType:
                     f"Expected NodeReference {descriptor.name} to have an instance pointing to a descriptor"
                 )
             descriptor = descriptor.instance
-        
+
         inferred_type = primitive_type_to_vellum_variable_type(descriptor)
     else:
         vellum_variable_value = primitive_to_vellum_value(value)
@@ -68,9 +66,7 @@ def create_node_input_value_pointer_rule(
         )
     if isinstance(value, WorkflowInputReference):
         workflow_input_display = display_context.workflow_input_displays[value]
-        return InputVariablePointer(
-            data=InputVariableData(input_variable_id=str(workflow_input_display.id))
-        )
+        return InputVariablePointer(data=InputVariableData(input_variable_id=str(workflow_input_display.id)))
     if isinstance(value, VellumSecretReference):
         # TODO: Pass through the name instead of retrieving the ID
         # https://app.shortcut.com/vellum/story/5072

vellum_ee/workflows/display/workflows/get_vellum_workflow_display_class.py

@@ -24,4 +24,7 @@ def get_workflow_display(
         except IndexError:
             return base_display_class(workflow_class)
 
-    return workflow_display_class(workflow_class, parent_display_context=parent_display_context) # type: ignore[return-value]
+    return workflow_display_class(  # type: ignore[return-value]
+        workflow_class,
+        parent_display_context=parent_display_context,
+    )

vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -57,7 +57,11 @@ 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
+            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
 
             input_variables.append(
@@ -67,7 +71,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},
                 }
             )
 

vellum/workflows/events/utils.py

@@ -1,5 +0,0 @@
-from vellum.workflows.events.workflow import WorkflowEvent
-
-
-def is_terminal_event(event: WorkflowEvent) -> bool:
-    return event.name in {"workflow.execution.fulfilled", "workflow.execution.rejected", "workflow.execution.paused"}

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"