vellum-ai 1.1.2__py3-none-any.whl → 1.1.4__py3-none-any.whl

vellum/client/types/metric_definition_history_item.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
-import pydantic
+from __future__ import annotations
+
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .vellum_variable import VellumVariable
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class MetricDefinitionHistoryItem(UniversalBaseModel):
@@ -32,3 +33,8 @@ class MetricDefinitionHistoryItem(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(MetricDefinitionHistoryItem)

vellum/client/types/metric_definition_input.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .string_input import StringInput
-from .json_input import JsonInput
+
 from .chat_history_input import ChatHistoryInput
+from .json_input import JsonInput
 from .number_input import NumberInput
+from .string_input import StringInput
 
 MetricDefinitionInput = typing.Union[StringInput, JsonInput, ChatHistoryInput, NumberInput]

vellum/client/types/metric_node_result.py

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

vellum/client/types/ml_model_read.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import datetime as dt
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class MlModelRead(UniversalBaseModel):
     """
@@ -17,6 +18,8 @@ class MlModelRead(UniversalBaseModel):
     The unique name of the ML Model.
     """
 
+    introduced_on: dt.datetime
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

vellum/client/types/ml_model_usage.py

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

vellum/client/types/ml_model_usage_wrapper.py

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

vellum/client/types/named_scenario_input_chat_history_variable_value_request.py

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

vellum/client/types/named_scenario_input_json_variable_value_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
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedScenarioInputJsonVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_scenario_input_request.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
-from .named_scenario_input_json_variable_value_request import NamedScenarioInputJsonVariableValueRequest
+
 from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
+from .named_scenario_input_json_variable_value_request import NamedScenarioInputJsonVariableValueRequest
+from .named_scenario_input_string_variable_value_request import NamedScenarioInputStringVariableValueRequest
 
 NamedScenarioInputRequest = typing.Union[
     NamedScenarioInputStringVariableValueRequest,

vellum/client/types/named_scenario_input_string_variable_value_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
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedScenarioInputStringVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_array_variable_value.py

@@ -1,11 +1,11 @@
 # 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 .vellum_value import VellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class NamedTestCaseArrayVariableValue(UniversalBaseModel):
@@ -14,7 +14,7 @@ class NamedTestCaseArrayVariableValue(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValue]] = None
+    value: typing.Optional[typing.List["VellumValue"]] = None
     name: str
 
     if IS_PYDANTIC_V2:
@@ -25,3 +25,9 @@ class NamedTestCaseArrayVariableValue(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+from .vellum_value import VellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(NamedTestCaseArrayVariableValue)

vellum/client/types/named_test_case_array_variable_value_request.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value_request import ArrayVellumValueRequest
+from __future__ import annotations
+
 import typing
-from .vellum_value_request import VellumValueRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class NamedTestCaseArrayVariableValueRequest(UniversalBaseModel):
@@ -14,7 +14,7 @@ class NamedTestCaseArrayVariableValueRequest(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValueRequest]] = None
+    value: typing.Optional[typing.List["VellumValueRequest"]] = None
     name: str
 
     if IS_PYDANTIC_V2:
@@ -25,3 +25,9 @@ class NamedTestCaseArrayVariableValueRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+from .vellum_value_request import VellumValueRequest  # noqa: E402, F401, I001
+
+update_forward_refs(NamedTestCaseArrayVariableValueRequest)

vellum/client/types/named_test_case_chat_history_variable_value.py

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

vellum/client/types/named_test_case_chat_history_variable_value_request.py

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

vellum/client/types/named_test_case_error_variable_value.py

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

vellum/client/types/named_test_case_error_variable_value_request.py

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

vellum/client/types/named_test_case_function_call_variable_value.py

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

vellum/client/types/named_test_case_function_call_variable_value_request.py

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

vellum/client/types/named_test_case_json_variable_value.py

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

vellum/client/types/named_test_case_json_variable_value_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
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseJsonVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_number_variable_value.py

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

vellum/client/types/named_test_case_number_variable_value_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
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseNumberVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_search_results_variable_value.py

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

vellum/client/types/named_test_case_search_results_variable_value_request.py

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

vellum/client/types/named_test_case_string_variable_value.py

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

vellum/client/types/named_test_case_string_variable_value_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
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class NamedTestCaseStringVariableValueRequest(UniversalBaseModel):

vellum/client/types/named_test_case_variable_value.py

@@ -1,14 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .named_test_case_string_variable_value import NamedTestCaseStringVariableValue
-from .named_test_case_number_variable_value import NamedTestCaseNumberVariableValue
-from .named_test_case_json_variable_value import NamedTestCaseJsonVariableValue
+
+from .named_test_case_array_variable_value import NamedTestCaseArrayVariableValue
 from .named_test_case_chat_history_variable_value import NamedTestCaseChatHistoryVariableValue
-from .named_test_case_search_results_variable_value import NamedTestCaseSearchResultsVariableValue
 from .named_test_case_error_variable_value import NamedTestCaseErrorVariableValue
 from .named_test_case_function_call_variable_value import NamedTestCaseFunctionCallVariableValue
-from .named_test_case_array_variable_value import NamedTestCaseArrayVariableValue
+from .named_test_case_json_variable_value import NamedTestCaseJsonVariableValue
+from .named_test_case_number_variable_value import NamedTestCaseNumberVariableValue
+from .named_test_case_search_results_variable_value import NamedTestCaseSearchResultsVariableValue
+from .named_test_case_string_variable_value import NamedTestCaseStringVariableValue
 
 NamedTestCaseVariableValue = typing.Union[
     NamedTestCaseStringVariableValue,

vellum/client/types/named_test_case_variable_value_request.py

@@ -1,14 +1,15 @@
 # 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_chat_history_variable_value_request import NamedTestCaseChatHistoryVariableValueRequest
-from .named_test_case_search_results_variable_value_request import NamedTestCaseSearchResultsVariableValueRequest
 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_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
 
 NamedTestCaseVariableValueRequest = typing.Union[
     NamedTestCaseStringVariableValueRequest,

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):

vellum/client/types/node_execution_fulfilled_event.py

@@ -1,26 +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 .external_parent_context import ExternalParentContext
-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
@@ -37,3 +30,16 @@ 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 .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # 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,26 +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 .external_parent_context import ExternalParentContext
-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
@@ -37,3 +30,16 @@ 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 .node_parent_context import NodeParentContext  # noqa: E402, F401, I001
+from .prompt_deployment_parent_context import PromptDeploymentParentContext  # 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_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):