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

vellum/client/types/named_test_case_variable_value_request.py

@@ -1,14 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .named_test_case_string_variable_value_request import NamedTestCaseStringVariableValueRequest
-from .named_test_case_number_variable_value_request import NamedTestCaseNumberVariableValueRequest
-from .named_test_case_json_variable_value_request import NamedTestCaseJsonVariableValueRequest
+
+from .named_test_case_array_variable_value_request import NamedTestCaseArrayVariableValueRequest
+from .named_test_case_audio_variable_value_request import NamedTestCaseAudioVariableValueRequest
 from .named_test_case_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
-from .named_test_case_search_results_variable_value_request import NamedTestCaseSearchResultsVariableValueRequest
+from .named_test_case_document_variable_value_request import NamedTestCaseDocumentVariableValueRequest
 from .named_test_case_error_variable_value_request import NamedTestCaseErrorVariableValueRequest
 from .named_test_case_function_call_variable_value_request import NamedTestCaseFunctionCallVariableValueRequest
-from .named_test_case_array_variable_value_request import NamedTestCaseArrayVariableValueRequest
+from .named_test_case_image_variable_value_request import NamedTestCaseImageVariableValueRequest
+from .named_test_case_json_variable_value_request import NamedTestCaseJsonVariableValueRequest
+from .named_test_case_number_variable_value_request import NamedTestCaseNumberVariableValueRequest
+from .named_test_case_search_results_variable_value_request import NamedTestCaseSearchResultsVariableValueRequest
+from .named_test_case_string_variable_value_request import NamedTestCaseStringVariableValueRequest
+from .named_test_case_video_variable_value_request import NamedTestCaseVideoVariableValueRequest
 
 NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseStringVariableValueRequest,
@@ -19,4 +24,8 @@ NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseErrorVariableValueRequest,
     NamedTestCaseFunctionCallVariableValueRequest,
     NamedTestCaseArrayVariableValueRequest,
+    NamedTestCaseAudioVariableValueRequest,
+    NamedTestCaseVideoVariableValueRequest,
+    NamedTestCaseImageVariableValueRequest,
+    NamedTestCaseDocumentVariableValueRequest,
 ]

vellum/client/types/named_test_case_video_variable_value.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 .vellum_video import VellumVideo
+
+
+class NamedTestCaseVideoVariableValue(UniversalBaseModel):
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: typing.Optional[VellumVideo] = None
+    name: str
+
+    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/named_test_case_video_variable_value_request.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 .vellum_video_request import VellumVideoRequest
+
+
+class NamedTestCaseVideoVariableValueRequest(UniversalBaseModel):
+    type: typing.Literal["VIDEO"] = "VIDEO"
+    value: typing.Optional[VellumVideoRequest] = None
+    name: str
+
+    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/node_execution_fulfilled_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 .invoked_port import InvokedPort
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .invoked_port import InvokedPort
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
 
 
 class NodeExecutionFulfilledBody(UniversalBaseModel):
@@ -13,6 +13,7 @@ class NodeExecutionFulfilledBody(UniversalBaseModel):
     invoked_ports: typing.Optional[typing.List[InvokedPort]] = None
     outputs: typing.Dict[str, typing.Optional[typing.Any]]
     mocked: typing.Optional[bool] = None
+    redacted: typing.Optional[bool] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

vellum/client/types/node_execution_fulfilled_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 .node_execution_fulfilled_body import NodeExecutionFulfilledBody
+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 .node_execution_fulfilled_body import NodeExecutionFulfilledBody
 
 
 class NodeExecutionFulfilledEvent(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["node.execution.fulfilled"] = "node.execution.fulfilled"
     body: NodeExecutionFulfilledBody
     id: str
@@ -36,3 +30,18 @@ class NodeExecutionFulfilledEvent(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(NodeExecutionFulfilledEvent)

vellum/client/types/node_execution_initiated_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
 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 NodeExecutionInitiatedBody(UniversalBaseModel):

vellum/client/types/node_execution_initiated_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 .node_execution_initiated_body import NodeExecutionInitiatedBody
+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 .node_execution_initiated_body import NodeExecutionInitiatedBody
 
 
 class NodeExecutionInitiatedEvent(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["node.execution.initiated"] = "node.execution.initiated"
     body: NodeExecutionInitiatedBody
     id: str
@@ -36,3 +30,18 @@ class NodeExecutionInitiatedEvent(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(NodeExecutionInitiatedEvent)

vellum/client/types/node_execution_log_body.py

@@ -0,0 +1,24 @@
+# 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 .severity_enum import SeverityEnum
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
+
+
+class NodeExecutionLogBody(UniversalBaseModel):
+    node_definition: VellumCodeResourceDefinition
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    severity: SeverityEnum
+    message: str
+
+    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/node_execution_log_event.py

@@ -0,0 +1,47 @@
+# This file was auto-generated by Fern from our API Definition.
+
+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 .api_version_enum import ApiVersionEnum
+from .node_execution_log_body import NodeExecutionLogBody
+
+
+class NodeExecutionLogEvent(UniversalBaseModel):
+    parent: typing.Optional["ParentContext"] = None
+    links: typing.Optional[typing.List["SpanLink"]] = None
+    name: typing.Literal["node.execution.log"] = "node.execution.log"
+    body: NodeExecutionLogBody
+    id: str
+    timestamp: dt.datetime
+    api_version: typing.Optional[ApiVersionEnum] = None
+    trace_id: str
+    span_id: str
+
+    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
+
+
+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(NodeExecutionLogEvent)

vellum/client/types/node_execution_paused_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 NodeExecutionPausedBody(UniversalBaseModel):

vellum/client/types/node_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 .node_execution_paused_body import NodeExecutionPausedBody
+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 .node_execution_paused_body import NodeExecutionPausedBody
 
 
 class NodeExecutionPausedEvent(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["node.execution.paused"] = "node.execution.paused"
     body: NodeExecutionPausedBody
     id: str
@@ -36,3 +30,18 @@ class NodeExecutionPausedEvent(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(NodeExecutionPausedEvent)

vellum/client/types/node_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 NodeExecutionRejectedBody(UniversalBaseModel):
     node_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/node_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 .node_execution_rejected_body import NodeExecutionRejectedBody
+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 .node_execution_rejected_body import NodeExecutionRejectedBody
 
 
 class NodeExecutionRejectedEvent(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["node.execution.rejected"] = "node.execution.rejected"
     body: NodeExecutionRejectedBody
     id: str
@@ -36,3 +30,18 @@ class NodeExecutionRejectedEvent(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(NodeExecutionRejectedEvent)

vellum/client/types/node_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 NodeExecutionResumedBody(UniversalBaseModel):

vellum/client/types/node_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 .node_execution_resumed_body import NodeExecutionResumedBody
+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 .node_execution_resumed_body import NodeExecutionResumedBody
 
 
 class NodeExecutionResumedEvent(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["node.execution.resumed"] = "node.execution.resumed"
     body: NodeExecutionResumedBody
     id: str
@@ -36,3 +30,18 @@ class NodeExecutionResumedEvent(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(NodeExecutionResumedEvent)

vellum/client/types/node_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
-import typing
-from .vellum_node_execution_event import VellumNodeExecutionEvent
-from .node_execution_span_attributes import NodeExecutionSpanAttributes
+from __future__ import annotations
+
 import datetime as dt
-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 .node_execution_span_attributes import NodeExecutionSpanAttributes
+from .vellum_node_execution_event import VellumNodeExecutionEvent
+from .workflow_execution_usage_calculation_fulfilled_body import WorkflowExecutionUsageCalculationFulfilledBody
 
 
 class NodeExecutionSpan(UniversalBaseModel):
     name: typing.Literal["node.execution"] = "node.execution"
     events: typing.List[VellumNodeExecutionEvent]
     attributes: NodeExecutionSpanAttributes
+    usage_result: typing.Optional[WorkflowExecutionUsageCalculationFulfilledBody] = None
     span_id: str
     start_ts: dt.datetime
     end_ts: dt.datetime
@@ -33,3 +30,17 @@ class NodeExecutionSpan(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(NodeExecutionSpan)

vellum/client/types/node_execution_span_attributes.py

@@ -1,12 +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 NodeExecutionSpanAttributes(UniversalBaseModel):
+    label: str
+    filepath: typing.Optional[str] = None
     node_id: str
 
     if IS_PYDANTIC_V2:

vellum/client/types/node_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 NodeExecutionStreamingBody(UniversalBaseModel):

vellum/client/types/node_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 .node_execution_streaming_body import NodeExecutionStreamingBody
+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 .node_execution_streaming_body import NodeExecutionStreamingBody
 
 
 class NodeExecutionStreamingEvent(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["node.execution.streaming"] = "node.execution.streaming"
     body: NodeExecutionStreamingBody
     id: str
@@ -36,3 +30,18 @@ class NodeExecutionStreamingEvent(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(NodeExecutionStreamingEvent)