vellum-ai 0.14.63__py3-none-any.whl → 1.11.2__py3-none-any.whl

vellum/client/types/workflow_execution_node_result_event.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .workflow_node_result_event import WorkflowNodeResultEvent
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .workflow_node_result_event import WorkflowNodeResultEvent
 
 
 class WorkflowExecutionNodeResultEvent(UniversalBaseModel):
@@ -27,3 +28,8 @@ class WorkflowExecutionNodeResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionNodeResultEvent)

vellum/client/types/workflow_execution_paused_body.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_code_resource_definition import VellumCodeResourceDefinition
 import typing
-from .external_input_descriptor import ExternalInputDescriptor
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .external_input_descriptor import ExternalInputDescriptor
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
 
 
 class WorkflowExecutionPausedBody(UniversalBaseModel):

vellum/client/types/workflow_execution_paused_event.py

@@ -1,25 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .api_request_parent_context import ApiRequestParentContext
-from .node_parent_context import NodeParentContext
-from .prompt_deployment_parent_context import PromptDeploymentParentContext
-from .span_link import SpanLink
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
-from .workflow_parent_context import WorkflowParentContext
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
-import typing
-from .parent_context import ParentContext
-from .workflow_execution_paused_body import WorkflowExecutionPausedBody
+from __future__ import annotations
+
 import datetime as dt
-from .api_version_enum import ApiVersionEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .api_version_enum import ApiVersionEnum
+from .workflow_execution_paused_body import WorkflowExecutionPausedBody
 
 
 class WorkflowExecutionPausedEvent(UniversalBaseModel):
-    parent: typing.Optional[ParentContext] = None
-    links: typing.Optional[typing.List[SpanLink]] = None
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
     name: typing.Literal["workflow.execution.paused"] = "workflow.execution.paused"
     body: WorkflowExecutionPausedBody
     id: str
@@ -36,3 +30,18 @@ class WorkflowExecutionPausedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionPausedEvent)

vellum/client/types/workflow_execution_rejected_body.py

@@ -1,16 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_code_resource_definition import VellumCodeResourceDefinition
-from .vellum_sdk_error import VellumSdkError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
+from .vellum_sdk_error import VellumSdkError
 
 
 class WorkflowExecutionRejectedBody(UniversalBaseModel):
     workflow_definition: VellumCodeResourceDefinition
     error: VellumSdkError
+    stacktrace: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/workflow_execution_rejected_event.py

@@ -1,25 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .api_request_parent_context import ApiRequestParentContext
-from .node_parent_context import NodeParentContext
-from .prompt_deployment_parent_context import PromptDeploymentParentContext
-from .span_link import SpanLink
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
-from .workflow_parent_context import WorkflowParentContext
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
-import typing
-from .parent_context import ParentContext
-from .workflow_execution_rejected_body import WorkflowExecutionRejectedBody
+from __future__ import annotations
+
 import datetime as dt
-from .api_version_enum import ApiVersionEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .api_version_enum import ApiVersionEnum
+from .workflow_execution_rejected_body import WorkflowExecutionRejectedBody
 
 
 class WorkflowExecutionRejectedEvent(UniversalBaseModel):
-    parent: typing.Optional[ParentContext] = None
-    links: typing.Optional[typing.List[SpanLink]] = None
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
     name: typing.Literal["workflow.execution.rejected"] = "workflow.execution.rejected"
     body: WorkflowExecutionRejectedBody
     id: str
@@ -36,3 +30,18 @@ class WorkflowExecutionRejectedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionRejectedEvent)

vellum/client/types/workflow_execution_resumed_body.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_code_resource_definition import VellumCodeResourceDefinition
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
 
 
 class WorkflowExecutionResumedBody(UniversalBaseModel):

vellum/client/types/workflow_execution_resumed_event.py

@@ -1,25 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .api_request_parent_context import ApiRequestParentContext
-from .node_parent_context import NodeParentContext
-from .prompt_deployment_parent_context import PromptDeploymentParentContext
-from .span_link import SpanLink
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
-from .workflow_parent_context import WorkflowParentContext
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
-import typing
-from .parent_context import ParentContext
-from .workflow_execution_resumed_body import WorkflowExecutionResumedBody
+from __future__ import annotations
+
 import datetime as dt
-from .api_version_enum import ApiVersionEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .api_version_enum import ApiVersionEnum
+from .workflow_execution_resumed_body import WorkflowExecutionResumedBody
 
 
 class WorkflowExecutionResumedEvent(UniversalBaseModel):
-    parent: typing.Optional[ParentContext] = None
-    links: typing.Optional[typing.List[SpanLink]] = None
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
     name: typing.Literal["workflow.execution.resumed"] = "workflow.execution.resumed"
     body: WorkflowExecutionResumedBody
     id: str
@@ -36,3 +30,18 @@ class WorkflowExecutionResumedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionResumedEvent)

vellum/client/types/workflow_execution_snapshotted_body.py

@@ -1,14 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_code_resource_definition import VellumCodeResourceDefinition
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
 
 
 class WorkflowExecutionSnapshottedBody(UniversalBaseModel):
     workflow_definition: VellumCodeResourceDefinition
+    edited_by: typing.Optional[VellumCodeResourceDefinition] = None
     state: typing.Dict[str, typing.Optional[typing.Any]]
 
     if IS_PYDANTIC_V2:

vellum/client/types/workflow_execution_snapshotted_event.py

@@ -1,25 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .api_request_parent_context import ApiRequestParentContext
-from .node_parent_context import NodeParentContext
-from .prompt_deployment_parent_context import PromptDeploymentParentContext
-from .span_link import SpanLink
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
-from .workflow_parent_context import WorkflowParentContext
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
-import typing
-from .parent_context import ParentContext
-from .workflow_execution_snapshotted_body import WorkflowExecutionSnapshottedBody
+from __future__ import annotations
+
 import datetime as dt
-from .api_version_enum import ApiVersionEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .api_version_enum import ApiVersionEnum
+from .workflow_execution_snapshotted_body import WorkflowExecutionSnapshottedBody
 
 
 class WorkflowExecutionSnapshottedEvent(UniversalBaseModel):
-    parent: typing.Optional[ParentContext] = None
-    links: typing.Optional[typing.List[SpanLink]] = None
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
     name: typing.Literal["workflow.execution.snapshotted"] = "workflow.execution.snapshotted"
     body: WorkflowExecutionSnapshottedBody
     id: str
@@ -36,3 +30,18 @@ class WorkflowExecutionSnapshottedEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionSnapshottedEvent)

vellum/client/types/workflow_execution_span.py

@@ -1,25 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .api_request_parent_context import ApiRequestParentContext
-from .node_parent_context import NodeParentContext
-from .prompt_deployment_parent_context import PromptDeploymentParentContext
-from .span_link import SpanLink
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
-from .workflow_parent_context import WorkflowParentContext
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
+from __future__ import annotations
+
+import datetime as dt
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .vellum_workflow_execution_event import VellumWorkflowExecutionEvent
 from .workflow_execution_span_attributes import WorkflowExecutionSpanAttributes
-import datetime as dt
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .workflow_execution_usage_calculation_fulfilled_body import WorkflowExecutionUsageCalculationFulfilledBody
 
 
 class WorkflowExecutionSpan(UniversalBaseModel):
     name: typing.Literal["workflow.execution"] = "workflow.execution"
     events: typing.List[VellumWorkflowExecutionEvent]
     attributes: WorkflowExecutionSpanAttributes
+    usage_result: typing.Optional[WorkflowExecutionUsageCalculationFulfilledBody] = None
     span_id: str
     start_ts: dt.datetime
     end_ts: dt.datetime
@@ -33,3 +30,17 @@ class WorkflowExecutionSpan(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionSpan)

vellum/client/types/workflow_execution_span_attributes.py

@@ -1,13 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class WorkflowExecutionSpanAttributes(UniversalBaseModel):
     label: str
+    workflow_id: str
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/workflow_execution_streaming_body.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .vellum_code_resource_definition import VellumCodeResourceDefinition
-from .base_output import BaseOutput
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .base_output import BaseOutput
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
 
 
 class WorkflowExecutionStreamingBody(UniversalBaseModel):

vellum/client/types/workflow_execution_streaming_event.py

@@ -1,25 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .api_request_parent_context import ApiRequestParentContext
-from .node_parent_context import NodeParentContext
-from .prompt_deployment_parent_context import PromptDeploymentParentContext
-from .span_link import SpanLink
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext
-from .workflow_parent_context import WorkflowParentContext
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext
-import typing
-from .parent_context import ParentContext
-from .workflow_execution_streaming_body import WorkflowExecutionStreamingBody
+from __future__ import annotations
+
 import datetime as dt
-from .api_version_enum import ApiVersionEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .api_version_enum import ApiVersionEnum
+from .workflow_execution_streaming_body import WorkflowExecutionStreamingBody
 
 
 class WorkflowExecutionStreamingEvent(UniversalBaseModel):
-    parent: typing.Optional[ParentContext] = None
-    links: typing.Optional[typing.List[SpanLink]] = None
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
     name: typing.Literal["workflow.execution.streaming"] = "workflow.execution.streaming"
     body: WorkflowExecutionStreamingBody
     id: str
@@ -36,3 +30,18 @@ class WorkflowExecutionStreamingEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .api_request_parent_context import ApiRequestParentContext  # noqa: E402, F401, I001
+from .external_parent_context import ExternalParentContext  # noqa: E402, F401, I001
+from .integration_trigger_context import IntegrationTriggerContext  # noqa: E402, F401, I001
+from .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402, F401, I001
+from .scheduled_trigger_context import ScheduledTriggerContext  # noqa: E402, F401, I001
+from .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402, F401, I001
+from .workflow_parent_context import WorkflowParentContext  # noqa: E402, F401, I001
+from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402, F401, I001
+from .parent_context import ParentContext  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionStreamingEvent)

vellum/client/types/workflow_execution_usage_calculation_error.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .workflow_execution_usage_calculation_error_code_enum import WorkflowExecutionUsageCalculationErrorCodeEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .workflow_execution_usage_calculation_error_code_enum import WorkflowExecutionUsageCalculationErrorCodeEnum
 
 
 class WorkflowExecutionUsageCalculationError(UniversalBaseModel):

vellum/client/types/workflow_execution_usage_calculation_fulfilled_body.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .ml_model_usage_wrapper import MlModelUsageWrapper
+from .price import Price
+
+
+class WorkflowExecutionUsageCalculationFulfilledBody(UniversalBaseModel):
+    usage: typing.List[MlModelUsageWrapper]
+    cost: typing.List[Price]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

vellum/client/types/workflow_execution_usage_result.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .ml_model_usage_wrapper import MlModelUsageWrapper
 from .price import Price
 from .workflow_execution_usage_calculation_error import WorkflowExecutionUsageCalculationError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
 
 
 class WorkflowExecutionUsageResult(UniversalBaseModel):

vellum/client/types/workflow_execution_view_online_eval_metric_result.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .execution_vellum_value import ExecutionVellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .execution_vellum_value import ExecutionVellumValue
 
 
 class WorkflowExecutionViewOnlineEvalMetricResult(UniversalBaseModel):
@@ -21,3 +22,8 @@ class WorkflowExecutionViewOnlineEvalMetricResult(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionViewOnlineEvalMetricResult)

vellum/client/types/workflow_execution_workflow_result_event.py

@@ -1,11 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
+from __future__ import annotations
+
 import typing
-from .workflow_result_event import WorkflowResultEvent
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .workflow_result_event import WorkflowResultEvent
 
 
 class WorkflowExecutionWorkflowResultEvent(UniversalBaseModel):
@@ -27,3 +28,8 @@ class WorkflowExecutionWorkflowResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowExecutionWorkflowResultEvent)

vellum/client/types/workflow_expand_meta_request.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class WorkflowExpandMetaRequest(UniversalBaseModel):

vellum/client/types/workflow_initialization_error.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class WorkflowInitializationError(UniversalBaseModel):

vellum/client/types/workflow_input.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .array_input import ArrayInput
+from .audio_input import AudioInput
+from .chat_history_input import ChatHistoryInput
+from .document_input import DocumentInput
+from .error_input import ErrorInput
+from .function_call_input import FunctionCallInput
+from .image_input import ImageInput
+from .json_input import JsonInput
+from .number_input import NumberInput
+from .search_results_input import SearchResultsInput
+from .string_input import StringInput
+from .video_input import VideoInput
+
+WorkflowInput = typing.Union[
+    StringInput,
+    JsonInput,
+    ChatHistoryInput,
+    NumberInput,
+    SearchResultsInput,
+    ErrorInput,
+    ArrayInput,
+    FunctionCallInput,
+    AudioInput,
+    VideoInput,
+    ImageInput,
+    DocumentInput,
+]

vellum/client/types/workflow_node_result_data.py

@@ -1,17 +1,18 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .prompt_node_result import PromptNodeResult
-from .search_node_result import SearchNodeResult
-from .templating_node_result import TemplatingNodeResult
+
+from .api_node_result import ApiNodeResult
 from .code_execution_node_result import CodeExecutionNodeResult
 from .conditional_node_result import ConditionalNodeResult
-from .api_node_result import ApiNodeResult
-from .terminal_node_result import TerminalNodeResult
+from .map_node_result import MapNodeResult
 from .merge_node_result import MergeNodeResult
-from .subworkflow_node_result import SubworkflowNodeResult
 from .metric_node_result import MetricNodeResult
-from .map_node_result import MapNodeResult
+from .prompt_node_result import PromptNodeResult
+from .search_node_result import SearchNodeResult
+from .subworkflow_node_result import SubworkflowNodeResult
+from .templating_node_result import TemplatingNodeResult
+from .terminal_node_result import TerminalNodeResult
 
 WorkflowNodeResultData = typing.Union[
     PromptNodeResult,