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

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

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

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

vellum/client/types/search_filters_request.py

@@ -1,13 +1,12 @@
 # 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 .metadata_filter_rule_request import MetadataFilterRuleRequest
-from .vellum_value_logical_condition_group_request import VellumValueLogicalConditionGroupRequest
+from __future__ import annotations
+
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .metadata_filters_request import MetadataFiltersRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class SearchFiltersRequest(UniversalBaseModel):
@@ -29,3 +28,10 @@ class SearchFiltersRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+from .metadata_filter_rule_request import MetadataFilterRuleRequest  # noqa: E402, F401, I001
+from .vellum_value_logical_condition_group_request import VellumValueLogicalConditionGroupRequest  # noqa: E402, F401, I001
+
+update_forward_refs(SearchFiltersRequest)

vellum/client/types/search_node_result.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_node_result_data import SearchNodeResultData
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .search_node_result_data import SearchNodeResultData
 
 
 class SearchNodeResult(UniversalBaseModel):

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

vellum/client/types/search_request_options_request.py

@@ -1,15 +1,14 @@
 # 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 .metadata_filter_rule_request import MetadataFilterRuleRequest
-from .vellum_value_logical_condition_group_request import VellumValueLogicalConditionGroupRequest
+from __future__ import annotations
+
 import typing
+
 import pydantic
-from .search_weights_request import SearchWeightsRequest
-from .search_result_merging_request import SearchResultMergingRequest
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .search_filters_request import SearchFiltersRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .search_result_merging_request import SearchResultMergingRequest
+from .search_weights_request import SearchWeightsRequest
 
 
 class SearchRequestOptionsRequest(UniversalBaseModel):
@@ -41,3 +40,10 @@ class SearchRequestOptionsRequest(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value_request import ArrayVellumValueRequest  # noqa: E402, F401, I001
+from .metadata_filter_rule_request import MetadataFilterRuleRequest  # noqa: E402, F401, I001
+from .vellum_value_logical_condition_group_request import VellumValueLogicalConditionGroupRequest  # noqa: E402, F401, I001
+
+update_forward_refs(SearchRequestOptionsRequest)

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

vellum/client/types/search_result.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .search_result_document import SearchResultDocument
 from .search_result_meta import SearchResultMeta
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class SearchResult(UniversalBaseModel):

vellum/client/types/search_result_document.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 SearchResultDocument(UniversalBaseModel):

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

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

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

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

vellum/client/types/search_result_request.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .search_result_document_request import SearchResultDocumentRequest
 from .search_result_meta_request import SearchResultMetaRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class SearchResultRequest(UniversalBaseModel):

vellum/client/types/search_results_input.py

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

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

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

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

vellum/client/types/sentence_chunker_config.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 SentenceChunkerConfig(UniversalBaseModel):

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

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

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

vellum/client/types/slim_deployment_read.py

@@ -1,14 +1,15 @@
 # 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 datetime as dt
-import pydantic
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
 from .vellum_variable import VellumVariable
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class SlimDeploymentRead(UniversalBaseModel):
@@ -56,3 +57,8 @@ class SlimDeploymentRead(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(SlimDeploymentRead)

vellum/client/types/slim_document.py

@@ -1,14 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import pydantic
-import typing
 import datetime as dt
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .document_processing_state import DocumentProcessingState
-from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .document_status import DocumentStatus
+from .processing_failure_reason_enum import ProcessingFailureReasonEnum
 from .slim_document_document_to_document_index import SlimDocumentDocumentToDocumentIndex
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class SlimDocument(UniversalBaseModel):

vellum/client/types/slim_document_document_to_document_index.py

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

vellum/client/types/slim_release_review.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
-from .release_review_reviewer import ReleaseReviewReviewer
-from .release_review_state import ReleaseReviewState
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .release_review_reviewer import ReleaseReviewReviewer
+from .release_review_state import ReleaseReviewState
 
 
 class SlimReleaseReview(UniversalBaseModel):

vellum/client/types/slim_workflow_deployment.py

@@ -1,14 +1,15 @@
 # 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 datetime as dt
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 from .entity_status import EntityStatus
 from .environment_enum import EnvironmentEnum
-import datetime as dt
 from .vellum_variable import VellumVariable
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class SlimWorkflowDeployment(UniversalBaseModel):
@@ -65,3 +66,8 @@ class SlimWorkflowDeployment(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(SlimWorkflowDeployment)

vellum/client/types/slim_workflow_execution_read.py

@@ -1,29 +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 .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
-from .array_vellum_value import ArrayVellumValue
-import typing
+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 .execution_vellum_value import ExecutionVellumValue
 from .workflow_error import WorkflowError
 from .workflow_execution_actual import WorkflowExecutionActual
-from .workflow_execution_view_online_eval_metric_result import WorkflowExecutionViewOnlineEvalMetricResult
 from .workflow_execution_usage_result import WorkflowExecutionUsageResult
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .workflow_execution_view_online_eval_metric_result import WorkflowExecutionViewOnlineEvalMetricResult
 
 
 class SlimWorkflowExecutionRead(UniversalBaseModel):
     span_id: str
-    parent_context: typing.Optional[WorkflowDeploymentParentContext] = None
+    parent_context: typing.Optional["WorkflowDeploymentParentContext"] = None
     start: dt.datetime
     end: typing.Optional[dt.datetime] = None
     inputs: typing.List[ExecutionVellumValue]
@@ -41,3 +34,16 @@ class SlimWorkflowExecutionRead(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 .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(SlimWorkflowExecutionRead)

vellum/client/types/span_link.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-from .span_link_type_enum import SpanLinkTypeEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import typing
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .span_link_type_enum import SpanLinkTypeEnum
 
 
 class SpanLink(UniversalBaseModel):
@@ -24,13 +24,13 @@ class SpanLink(UniversalBaseModel):
             extra = pydantic.Extra.allow
 
 
-from .api_request_parent_context import ApiRequestParentContext  # noqa: E402
-from .external_parent_context import ExternalParentContext  # noqa: E402
-from .node_parent_context import NodeParentContext  # noqa: E402
-from .prompt_deployment_parent_context import PromptDeploymentParentContext  # noqa: E402
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402
-from .workflow_parent_context import WorkflowParentContext  # noqa: E402
-from .workflow_sandbox_parent_context import WorkflowSandboxParentContext  # noqa: E402
-from .parent_context import ParentContext  # noqa: E402
+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 .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(SpanLink)

vellum/client/types/streaming_ad_hoc_execute_prompt_event.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .prompt_output import PromptOutput
-from .ad_hoc_streaming_prompt_execution_meta import AdHocStreamingPromptExecutionMeta
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .ad_hoc_streaming_prompt_execution_meta import AdHocStreamingPromptExecutionMeta
+from .prompt_output import PromptOutput
 
 
 class StreamingAdHocExecutePromptEvent(UniversalBaseModel):

vellum/client/types/streaming_execute_prompt_event.py

@@ -1,11 +1,11 @@
 # 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 .prompt_output import PromptOutput
 from .streaming_prompt_execution_meta import StreamingPromptExecutionMeta
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class StreamingExecutePromptEvent(UniversalBaseModel):

vellum/client/types/streaming_prompt_execution_meta.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 StreamingPromptExecutionMeta(UniversalBaseModel):

vellum/client/types/streaming_workflow_node_result_event.py

@@ -1,13 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-import typing
+from __future__ import annotations
+
 import datetime as dt
-from .workflow_node_result_data import WorkflowNodeResultData
-from .node_output_compiled_value import NodeOutputCompiledValue
-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_output_compiled_value import NodeOutputCompiledValue
+from .workflow_node_result_data import WorkflowNodeResultData
 
 
 class StreamingWorkflowNodeResultEvent(UniversalBaseModel):
@@ -33,3 +34,8 @@ class StreamingWorkflowNodeResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(StreamingWorkflowNodeResultEvent)