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

vellum/client/types/workflow_output_image.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 .vellum_image import VellumImage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class WorkflowOutputImage(UniversalBaseModel):

vellum/client/types/workflow_output_json.py

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

vellum/client/types/workflow_output_number.py

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

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

vellum/client/types/workflow_output_string.py

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

vellum/client/types/workflow_parent_context.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
+
 import typing
-from .vellum_code_resource_definition import VellumCodeResourceDefinition
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .vellum_code_resource_definition import VellumCodeResourceDefinition
 
 
 class WorkflowParentContext(UniversalBaseModel):
@@ -26,13 +26,13 @@ class WorkflowParentContext(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 .span_link import SpanLink  # noqa: E402
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # 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 .span_link import SpanLink  # noqa: E402, F401, I001
+from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # 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(WorkflowParentContext)

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

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

vellum/client/types/workflow_release_tag_read.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 .release_tag_release import ReleaseTagRelease
 from .release_tag_source import ReleaseTagSource
 from .workflow_release_tag_workflow_deployment_history_item import WorkflowReleaseTagWorkflowDeploymentHistoryItem
-from .release_tag_release import ReleaseTagRelease
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import typing
 
 
 class WorkflowReleaseTagRead(UniversalBaseModel):

vellum/client/types/workflow_release_tag_workflow_deployment_history_item.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 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
 
 
 class WorkflowReleaseTagWorkflowDeploymentHistoryItem(UniversalBaseModel):

vellum/client/types/workflow_request_chat_history_input_request.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 .chat_message_request import ChatMessageRequest
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 
 
 class WorkflowRequestChatHistoryInputRequest(UniversalBaseModel):

vellum/client/types/workflow_request_input_request.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .workflow_request_string_input_request import WorkflowRequestStringInputRequest
-from .workflow_request_json_input_request import WorkflowRequestJsonInputRequest
+
 from .workflow_request_chat_history_input_request import WorkflowRequestChatHistoryInputRequest
+from .workflow_request_json_input_request import WorkflowRequestJsonInputRequest
 from .workflow_request_number_input_request import WorkflowRequestNumberInputRequest
+from .workflow_request_string_input_request import WorkflowRequestStringInputRequest
 
 WorkflowRequestInputRequest = typing.Union[
     WorkflowRequestStringInputRequest,

vellum/client/types/workflow_request_json_input_request.py

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

vellum/client/types/workflow_request_number_input_request.py

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

vellum/client/types/workflow_request_string_input_request.py

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

vellum/client/types/workflow_result_event.py

@@ -1,16 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .array_vellum_value import ArrayVellumValue
-from .workflow_node_result_event_state import WorkflowNodeResultEventState
+from __future__ import annotations
+
 import datetime as dt
 import typing
-from .workflow_result_event_output_data import WorkflowResultEventOutputData
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .execution_vellum_value import ExecutionVellumValue
 from .workflow_event_error import WorkflowEventError
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 from .workflow_output import WorkflowOutput
-from .execution_vellum_value import ExecutionVellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .workflow_result_event_output_data import WorkflowResultEventOutputData
 
 
 class WorkflowResultEvent(UniversalBaseModel):
@@ -30,3 +31,8 @@ class WorkflowResultEvent(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .array_vellum_value import ArrayVellumValue  # noqa: E402, F401, I001
+
+update_forward_refs(WorkflowResultEvent)

vellum/client/types/workflow_result_event_output_data.py

@@ -1,14 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
-from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
-from .workflow_result_event_output_data_json import WorkflowResultEventOutputDataJson
-from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
-from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
+
 from .workflow_result_event_output_data_array import WorkflowResultEventOutputDataArray
-from .workflow_result_event_output_data_function_call import WorkflowResultEventOutputDataFunctionCall
+from .workflow_result_event_output_data_chat_history import WorkflowResultEventOutputDataChatHistory
 from .workflow_result_event_output_data_error import WorkflowResultEventOutputDataError
+from .workflow_result_event_output_data_function_call import WorkflowResultEventOutputDataFunctionCall
+from .workflow_result_event_output_data_json import WorkflowResultEventOutputDataJson
+from .workflow_result_event_output_data_number import WorkflowResultEventOutputDataNumber
+from .workflow_result_event_output_data_search_results import WorkflowResultEventOutputDataSearchResults
+from .workflow_result_event_output_data_string import WorkflowResultEventOutputDataString
 
 WorkflowResultEventOutputData = typing.Union[
     WorkflowResultEventOutputDataString,

vellum/client/types/workflow_result_event_output_data_array.py

@@ -1,12 +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_state import WorkflowNodeResultEventState
+
 import pydantic
-from .vellum_value import VellumValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataArray(UniversalBaseModel):
@@ -24,7 +24,7 @@ class WorkflowResultEventOutputDataArray(UniversalBaseModel):
     """
 
     type: typing.Literal["ARRAY"] = "ARRAY"
-    value: typing.Optional[typing.List[VellumValue]] = None
+    value: typing.Optional[typing.List["VellumValue"]] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -34,3 +34,9 @@ class WorkflowResultEventOutputDataArray(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(WorkflowResultEventOutputDataArray)

vellum/client/types/workflow_result_event_output_data_chat_history.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 .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .chat_message import ChatMessage
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataChatHistory(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_error.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 .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .vellum_error import VellumError
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataError(UniversalBaseModel):

vellum/client/types/workflow_result_event_output_data_function_call.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 .workflow_node_result_event_state import WorkflowNodeResultEventState
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .function_call import FunctionCall
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .workflow_node_result_event_state import WorkflowNodeResultEventState
 
 
 class WorkflowResultEventOutputDataFunctionCall(UniversalBaseModel):

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

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

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

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

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

vellum/client/types/workflow_sandbox_parent_context.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class WorkflowSandboxParentContext(UniversalBaseModel):
@@ -27,13 +27,13 @@ class WorkflowSandboxParentContext(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 .span_link import SpanLink  # noqa: E402
-from .workflow_deployment_parent_context import WorkflowDeploymentParentContext  # noqa: E402
-from .workflow_parent_context import WorkflowParentContext  # 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 .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 .parent_context import ParentContext  # noqa: E402, F401, I001
 
 update_forward_refs(WorkflowSandboxParentContext)

vellum/client/types/workflow_stream_event.py

@@ -1,7 +1,8 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
+
 from .workflow_execution_node_result_event import WorkflowExecutionNodeResultEvent
+from .workflow_execution_workflow_result_event import WorkflowExecutionWorkflowResultEvent
 
 WorkflowStreamEvent = typing.Union[WorkflowExecutionWorkflowResultEvent, WorkflowExecutionNodeResultEvent]

vellum/client/types/workspace_read.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 datetime as dt
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
 
 class WorkspaceRead(UniversalBaseModel):
     id: str

vellum/client/types/workspace_secret_read.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import datetime as dt
-from .secret_type_enum import SecretTypeEnum
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .secret_type_enum import SecretTypeEnum
 
 
 class WorkspaceSecretRead(UniversalBaseModel):

vellum/core/force_multipart.py

@@ -0,0 +1,3 @@
+# WARNING: This file will be removed in a future release. Please import from "vellum.client" instead.
+
+from vellum.client.core.force_multipart import *

vellum/workflows/events/node.py

@@ -1,6 +1,6 @@
 from typing import TYPE_CHECKING, Any, Dict, Generic, List, Literal, Optional, Set, Type, Union
 
-from pydantic import SerializerFunctionWrapHandler, field_serializer, model_serializer
+from pydantic import field_serializer, model_serializer
 
 from vellum.client.core.pydantic_utilities import UniversalBaseModel
 from vellum.workflows.errors import WorkflowError
@@ -24,15 +24,6 @@ class _BaseNodeExecutionBody(UniversalBaseModel):
     def serialize_node_definition(self, node_definition: Type, _info: Any) -> Dict[str, Any]:
         return serialize_type_encoder_with_id(node_definition)
 
-    # Couldn't get this to work with model_config.exclude_none or model_config.exclude_defaults
-    # so we're excluding null invoked_ports manually here for now
-    @model_serializer(mode="wrap", when_used="json")
-    def serialize_model(self, handler: SerializerFunctionWrapHandler) -> Any:
-        serialized = super().serialize_model(handler)  # type: ignore[call-arg, arg-type]
-        if "invoked_ports" in serialized and serialized["invoked_ports"] is None:
-            del serialized["invoked_ports"]
-        return serialized
-
 
 class _BaseNodeEvent(BaseEvent):
     body: _BaseNodeExecutionBody
@@ -90,6 +81,18 @@ class NodeExecutionStreamingEvent(_BaseNodeEvent):
     def invoked_ports(self) -> InvokedPorts:
         return self.body.invoked_ports
 
+    @model_serializer(mode="plain", when_used="json")
+    def serialize_model(self) -> Any:
+        serialized = super().serialize_model()
+        if (
+            "body" in serialized
+            and isinstance(serialized["body"], dict)
+            and "invoked_ports" in serialized["body"]
+            and serialized["body"]["invoked_ports"] is None
+        ):
+            del serialized["body"]["invoked_ports"]
+        return serialized
+
 
 class NodeExecutionFulfilledBody(_BaseNodeExecutionBody, Generic[OutputsType]):
     outputs: OutputsType
@@ -123,6 +126,18 @@ class NodeExecutionFulfilledEvent(_BaseNodeEvent, Generic[OutputsType]):
     def mocked(self) -> Optional[bool]:
         return self.body.mocked
 
+    @model_serializer(mode="plain", when_used="json")
+    def serialize_model(self) -> Any:
+        serialized = super().serialize_model()
+        if (
+            "body" in serialized
+            and isinstance(serialized["body"], dict)
+            and "invoked_ports" in serialized["body"]
+            and serialized["body"]["invoked_ports"] is None
+        ):
+            del serialized["body"]["invoked_ports"]
+        return serialized
+
 
 class NodeExecutionRejectedBody(_BaseNodeExecutionBody):
     error: WorkflowError