vellum-ai 0.14.25__tar.gz → 0.14.26__tar.gz

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vellum-ai
-Version: 0.14.25
+Version: 0.14.26
 Summary: 
 License: MIT
 Requires-Python: >=3.9,<4.0

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/ee/vellum_ee/workflows/display/base.py

@@ -1,6 +1,6 @@
 from dataclasses import dataclass, field
 from uuid import UUID
-from typing import TypeVar
+from typing import Optional, TypeVar
 
 from pydantic import Field
 
@@ -50,17 +50,19 @@ WorkflowInputsDisplayOverridesType = TypeVar("WorkflowInputsDisplayOverridesType
 
 
 @dataclass
-class StateValueDisplayOverrides:
+class StateValueDisplay:
     id: UUID
+    name: Optional[str] = None
+    color: Optional[str] = None
 
 
 @dataclass
-class StateValueDisplay(StateValueDisplayOverrides):
-    pass
-
+class StateValueDisplayOverrides(StateValueDisplay):
+    """
+    DEPRECATED: Use StateValueDisplay instead. Will be removed in 0.15.0
+    """
 
-StateValueDisplayType = TypeVar("StateValueDisplayType", bound=StateValueDisplay)
-StateValueDisplayOverridesType = TypeVar("StateValueDisplayOverridesType", bound=StateValueDisplayOverrides)
+    pass
 
 
 @dataclass
@@ -85,6 +87,10 @@ class EntrypointDisplay:
 
 @dataclass
 class EntrypointDisplayOverrides(EntrypointDisplay):
+    """
+    DEPRECATED: Use EntrypointDisplay instead. Will be removed in 0.15.0
+    """
+
     pass
 
 

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/ee/vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/conftest.py

@@ -1,19 +1,20 @@
 import pytest
 from uuid import uuid4
-from typing import Any, Dict, Tuple, Type
+from typing import Any, Dict, Type
 
-from vellum.workflows.nodes.bases.base import BaseNode
-from vellum.workflows.references.output import OutputReference
-from vellum.workflows.references.state_value import StateValueReference
 from vellum.workflows.references.workflow_input import WorkflowInputReference
 from vellum.workflows.types.core import JsonObject
 from vellum.workflows.types.generics import NodeType
-from vellum_ee.workflows.display.base import StateValueDisplayType, WorkflowInputsDisplayType
+from vellum_ee.workflows.display.base import WorkflowInputsDisplayType
 from vellum_ee.workflows.display.editor.types import NodeDisplayData
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
-from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay
-from vellum_ee.workflows.display.types import WorkflowDisplayContext
+from vellum_ee.workflows.display.types import (
+    NodeDisplays,
+    NodeOutputDisplays,
+    StateValueDisplays,
+    WorkflowDisplayContext,
+)
 from vellum_ee.workflows.display.vellum import WorkflowMetaVellumDisplay
 from vellum_ee.workflows.display.workflows.vellum_workflow_display import VellumWorkflowDisplay
 
@@ -24,9 +25,9 @@ def serialize_node():
         node_class: Type[NodeType],
         base_class: type[BaseNodeDisplay[Any]] = BaseNodeDisplay,
         global_workflow_input_displays: Dict[WorkflowInputReference, WorkflowInputsDisplayType] = {},
-        global_state_value_displays: Dict[StateValueReference, StateValueDisplayType] = {},
-        global_node_displays: Dict[Type[BaseNode], BaseNodeDisplay] = {},
-        global_node_output_displays: Dict[OutputReference, Tuple[Type[BaseNode], NodeOutputDisplay]] = {},
+        global_state_value_displays: StateValueDisplays = {},
+        global_node_displays: NodeDisplays = {},
+        global_node_output_displays: NodeOutputDisplays = {},
     ) -> JsonObject:
         node_display_class = get_node_display_class(base_class, node_class)
         node_display = node_display_class()

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_default_state_serialization.py

@@ -50,7 +50,7 @@ def test_serialize_workflow():
                 "key": "example",
                 "type": "NUMBER",
                 "default": {"type": "NUMBER", "value": 5.0},
-                "required": True,
+                "required": False,
                 "extensions": {"color": None},
             },
         ],

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/ee/vellum_ee/workflows/display/types.py

@@ -9,7 +9,7 @@ from vellum.workflows.references import OutputReference, StateValueReference, Wo
 from vellum_ee.workflows.display.base import (
     EdgeDisplay,
     EntrypointDisplay,
-    StateValueDisplayType,
+    StateValueDisplay,
     WorkflowInputsDisplayType,
     WorkflowMetaDisplay,
     WorkflowOutputDisplay,
@@ -22,6 +22,7 @@ if TYPE_CHECKING:
 
 WorkflowDisplayType = TypeVar("WorkflowDisplayType", bound="BaseWorkflowDisplay")
 
+StateValueDisplays = Dict[StateValueReference, StateValueDisplay]
 NodeDisplays = Dict[Type[BaseNode], BaseNodeDisplay]
 NodeOutputDisplays = Dict[OutputReference, Tuple[Type[BaseNode], NodeOutputDisplay]]
 EntrypointDisplays = Dict[Type[BaseNode], EntrypointDisplay]
@@ -31,20 +32,15 @@ PortDisplays = Dict[Port, PortDisplay]
 
 
 @dataclass
-class WorkflowDisplayContext(
-    Generic[
-        WorkflowInputsDisplayType,
-        StateValueDisplayType,
-    ]
-):
+class WorkflowDisplayContext(Generic[WorkflowInputsDisplayType,]):
     workflow_display_class: Type["BaseWorkflowDisplay"]
     workflow_display: WorkflowMetaDisplay
     workflow_input_displays: Dict[WorkflowInputReference, WorkflowInputsDisplayType] = field(default_factory=dict)
     global_workflow_input_displays: Dict[WorkflowInputReference, WorkflowInputsDisplayType] = field(
         default_factory=dict
     )
-    state_value_displays: Dict[StateValueReference, StateValueDisplayType] = field(default_factory=dict)
-    global_state_value_displays: Dict[StateValueReference, StateValueDisplayType] = field(default_factory=dict)
+    state_value_displays: StateValueDisplays = field(default_factory=dict)
+    global_state_value_displays: StateValueDisplays = field(default_factory=dict)
     node_displays: NodeDisplays = field(default_factory=dict)
     global_node_displays: NodeDisplays = field(default_factory=dict)
     global_node_output_displays: NodeOutputDisplays = field(default_factory=dict)

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/ee/vellum_ee/workflows/display/vellum.py

@@ -6,7 +6,6 @@ from vellum.core import UniversalBaseModel
 from vellum_ee.workflows.display.base import (
     EdgeDisplayOverrides,
     EntrypointDisplayOverrides,
-    StateValueDisplay,
     StateValueDisplayOverrides,
     WorkflowInputsDisplay,
     WorkflowInputsDisplayOverrides,
@@ -57,14 +56,20 @@ class WorkflowInputsVellumDisplay(WorkflowInputsVellumDisplayOverrides):
 
 
 @dataclass
-class StateValueVellumDisplayOverrides(StateValueDisplay, StateValueDisplayOverrides):
-    name: Optional[str] = None
+class StateValueVellumDisplayOverrides(StateValueDisplayOverrides):
+    """
+    DEPRECATED: Use StateValueDisplay instead. Will be removed in 0.15.0
+    """
+
     required: Optional[bool] = None
-    color: Optional[str] = None
 
 
 @dataclass
 class StateValueVellumDisplay(StateValueVellumDisplayOverrides):
+    """
+    DEPRECATED: Use StateValueDisplay instead. Will be removed in 0.15.0
+    """
+
     pass
 
 

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/ee/vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -13,15 +13,14 @@ from vellum.workflows.events.workflow import NodeEventDisplayContext, WorkflowEv
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.nodes.utils import get_unadorned_node, get_wrapped_node
 from vellum.workflows.ports import Port
-from vellum.workflows.references import OutputReference, StateValueReference, WorkflowInputReference
+from vellum.workflows.references import OutputReference, WorkflowInputReference
 from vellum.workflows.types.core import JsonObject
 from vellum.workflows.types.generics import WorkflowType
 from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.base import (
     EdgeDisplay,
     EntrypointDisplay,
-    StateValueDisplayOverridesType,
-    StateValueDisplayType,
+    StateValueDisplay,
     WorkflowInputsDisplayOverridesType,
     WorkflowInputsDisplayType,
     WorkflowMetaDisplay,
@@ -39,6 +38,7 @@ from vellum_ee.workflows.display.types import (
     NodeDisplays,
     NodeOutputDisplays,
     PortDisplays,
+    StateValueDisplays,
     WorkflowDisplayContext,
     WorkflowOutputDisplays,
 )
@@ -52,8 +52,6 @@ class BaseWorkflowDisplay(
         WorkflowType,
         WorkflowInputsDisplayType,
         WorkflowInputsDisplayOverridesType,
-        StateValueDisplayType,
-        StateValueDisplayOverridesType,
     ]
 ):
     # Used to specify the display data for a workflow.
@@ -63,7 +61,7 @@ class BaseWorkflowDisplay(
     inputs_display: Dict[WorkflowInputReference, WorkflowInputsDisplayOverridesType] = {}
 
     # Used to explicitly specify display data for a workflow's state values.
-    state_value_displays: Dict[StateValueReference, StateValueDisplayOverridesType] = {}
+    state_value_displays: StateValueDisplays = {}
 
     # Used to explicitly specify display data for a workflow's entrypoints.
     entrypoint_displays: EntrypointDisplays = {}
@@ -88,12 +86,7 @@ class BaseWorkflowDisplay(
         self,
         workflow: Type[WorkflowType],
         *,
-        parent_display_context: Optional[
-            WorkflowDisplayContext[
-                WorkflowInputsDisplayType,
-                StateValueDisplayType,
-            ]
-        ] = None,
+        parent_display_context: Optional[WorkflowDisplayContext[WorkflowInputsDisplayType,]] = None,
         dry_run: bool = False,
     ):
         self._workflow = workflow
@@ -183,10 +176,7 @@ class BaseWorkflowDisplay(
     @cached_property
     def display_context(
         self,
-    ) -> WorkflowDisplayContext[
-        WorkflowInputsDisplayType,
-        StateValueDisplayType,
-    ]:
+    ) -> WorkflowDisplayContext[WorkflowInputsDisplayType]:
         workflow_meta_display = self._generate_workflow_meta_display()
 
         global_node_output_displays: NodeOutputDisplays = (
@@ -232,7 +222,7 @@ class BaseWorkflowDisplay(
             workflow_input_displays[workflow_input] = input_display
             global_workflow_input_displays[workflow_input] = input_display
 
-        state_value_displays: Dict[StateValueReference, StateValueDisplayType] = {}
+        state_value_displays: StateValueDisplays = {}
         global_state_value_displays = (
             copy(self._parent_display_context.global_state_value_displays) if self._parent_display_context else {}
         )
@@ -327,11 +317,20 @@ class BaseWorkflowDisplay(
     ) -> WorkflowInputsDisplayType:
         pass
 
-    @abstractmethod
     def _generate_state_value_display(
-        self, state_value: StateValueReference, overrides: Optional[StateValueDisplayOverridesType] = None
-    ) -> StateValueDisplayType:
-        pass
+        self, state_value: BaseDescriptor, overrides: Optional[StateValueDisplay] = None
+    ) -> StateValueDisplay:
+        state_value_id: UUID
+        name = None
+        color = None
+        if overrides:
+            state_value_id = overrides.id
+            name = overrides.name
+            color = overrides.color
+        else:
+            state_value_id = uuid4_from_hash(f"{self.workflow_id}|state_values|id|{state_value.name}")
+
+        return StateValueDisplay(id=state_value_id, name=name, color=color)
 
     def _generate_entrypoint_display(
         self,

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/ee/vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -2,7 +2,6 @@ import logging
 from uuid import UUID
 from typing import Optional, cast
 
-from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.nodes.displayable.bases.utils import primitive_to_vellum_value
 from vellum.workflows.nodes.displayable.final_output_node import FinalOutputNode
 from vellum.workflows.nodes.utils import get_unadorned_node, get_unadorned_port
@@ -16,12 +15,7 @@ from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input
 from vellum_ee.workflows.display.utils.vellum import infer_vellum_variable_type
-from vellum_ee.workflows.display.vellum import (
-    StateValueVellumDisplay,
-    StateValueVellumDisplayOverrides,
-    WorkflowInputsVellumDisplay,
-    WorkflowInputsVellumDisplayOverrides,
-)
+from vellum_ee.workflows.display.vellum import WorkflowInputsVellumDisplay, WorkflowInputsVellumDisplayOverrides
 from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
 
 logger = logging.getLogger(__name__)
@@ -32,8 +26,6 @@ class VellumWorkflowDisplay(
         WorkflowType,
         WorkflowInputsVellumDisplay,
         WorkflowInputsVellumDisplayOverrides,
-        StateValueVellumDisplay,
-        StateValueVellumDisplayOverrides,
     ]
 ):
     node_display_base_class = BaseNodeDisplay
@@ -60,20 +52,17 @@ class VellumWorkflowDisplay(
             )
 
         state_variables: JsonArray = []
-        for state_value, state_value_display in self.display_context.state_value_displays.items():
-            default = primitive_to_vellum_value(state_value.instance) if state_value.instance else None
-            required = (
-                state_value_display.required
-                if state_value_display.required is not None
-                else type(None) not in state_value.types
+        for state_value_reference, state_value_display in self.display_context.state_value_displays.items():
+            default = (
+                primitive_to_vellum_value(state_value_reference.instance) if state_value_reference.instance else None
             )
             state_variables.append(
                 {
                     "id": str(state_value_display.id),
-                    "key": state_value_display.name or state_value.name,
-                    "type": infer_vellum_variable_type(state_value),
+                    "key": state_value_display.name or state_value_reference.name,
+                    "type": infer_vellum_variable_type(state_value_reference),
                     "default": default.dict() if default else None,
-                    "required": required,
+                    "required": state_value_reference.instance is None,
                     "extensions": {"color": state_value_display.color},
                 }
             )
@@ -297,20 +286,3 @@ class VellumWorkflowDisplay(
             workflow_input_id = uuid4_from_hash(f"{self.workflow_id}|inputs|id|{workflow_input.name}")
 
         return WorkflowInputsVellumDisplay(id=workflow_input_id, name=name, required=required, color=color)
-
-    def _generate_state_value_display(
-        self, state_value: BaseDescriptor, overrides: Optional[StateValueVellumDisplayOverrides] = None
-    ) -> StateValueVellumDisplay:
-        state_value_id: UUID
-        name = None
-        required = None
-        color = None
-        if overrides:
-            state_value_id = overrides.id
-            name = overrides.name
-            required = overrides.required
-            color = overrides.color
-        else:
-            state_value_id = uuid4_from_hash(f"{self.workflow_id}|state_values|id|{state_value.name}")
-
-        return StateValueVellumDisplay(id=state_value_id, name=name, required=required, color=color)

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.14.25"
+version = "0.14.26"
 description = ""
 readme = "README.md"
 authors = []
@@ -69,6 +69,7 @@ ipython = "8.12.3"
 isort = "5.13.2"
 pre-commit = "3.5.0"
 pytest-mock = "3.10.0"
+pytest-timeout = "2.3.1"
 requests-mock = "1.11.0"
 ruff = "^0.5.6"
 tomli_w = "1.1.0"

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/__init__.py

@@ -322,7 +322,7 @@ from .types import (
     RejectedExecuteWorkflowWorkflowResultEvent,
     RejectedPromptExecutionMeta,
     RejectedWorkflowNodeResultEvent,
-    Release,
+    ReleaseTagRelease,
     ReleaseTagSource,
     ReplaceTestSuiteTestCaseRequest,
     RichTextChildBlock,
@@ -580,7 +580,6 @@ from .resources import (
     ListWorkflowReleaseTagsRequestSource,
     ListWorkflowSandboxExamplesRequestTag,
     WorkflowDeploymentsListRequestStatus,
-    WorkflowsPullRequestFormat,
     ad_hoc,
     container_images,
     deployments,
@@ -933,7 +932,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
-    "Release",
+    "ReleaseTagRelease",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",
@@ -1182,7 +1181,6 @@ __all__ = [
     "WorkflowSandboxExample",
     "WorkflowSandboxParentContext",
     "WorkflowStreamEvent",
-    "WorkflowsPullRequestFormat",
     "WorkspaceRead",
     "WorkspaceSecretRead",
     "__version__",

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/client/core/client_wrapper.py

@@ -18,7 +18,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "vellum-ai",
-            "X-Fern-SDK-Version": "0.14.25",
+            "X-Fern-SDK-Version": "0.14.26",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/client/resources/__init__.py

@@ -24,7 +24,6 @@ from .document_indexes import DocumentIndexesListRequestStatus
 from .folder_entities import FolderEntitiesListRequestEntityStatus
 from .workflow_deployments import ListWorkflowReleaseTagsRequestSource, WorkflowDeploymentsListRequestStatus
 from .workflow_sandboxes import ListWorkflowSandboxExamplesRequestTag
-from .workflows import WorkflowsPullRequestFormat
 
 __all__ = [
     "DeploymentsListRequestStatus",
@@ -34,7 +33,6 @@ __all__ = [
     "ListWorkflowReleaseTagsRequestSource",
     "ListWorkflowSandboxExamplesRequestTag",
     "WorkflowDeploymentsListRequestStatus",
-    "WorkflowsPullRequestFormat",
     "ad_hoc",
     "container_images",
     "deployments",

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/client/resources/workflows/client.py

@@ -2,7 +2,6 @@
 
 import typing
 from ...core.client_wrapper import SyncClientWrapper
-from .types.workflows_pull_request_format import WorkflowsPullRequestFormat
 from ...core.request_options import RequestOptions
 from ...core.jsonable_encoder import jsonable_encoder
 from ...errors.bad_request_error import BadRequestError
@@ -28,7 +27,6 @@ class WorkflowsClient:
         id: str,
         *,
         exclude_code: typing.Optional[bool] = None,
-        format: typing.Optional[WorkflowsPullRequestFormat] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
@@ -42,8 +40,6 @@ class WorkflowsClient:
 
         exclude_code : typing.Optional[bool]
 
-        format : typing.Optional[WorkflowsPullRequestFormat]
-
         include_json : typing.Optional[bool]
 
         include_sandbox : typing.Optional[bool]
@@ -64,7 +60,6 @@ class WorkflowsClient:
             method="GET",
             params={
                 "exclude_code": exclude_code,
-                "format": format,
                 "include_json": include_json,
                 "include_sandbox": include_sandbox,
                 "strict": strict,
@@ -180,7 +175,6 @@ class AsyncWorkflowsClient:
         id: str,
         *,
         exclude_code: typing.Optional[bool] = None,
-        format: typing.Optional[WorkflowsPullRequestFormat] = None,
         include_json: typing.Optional[bool] = None,
         include_sandbox: typing.Optional[bool] = None,
         strict: typing.Optional[bool] = None,
@@ -194,8 +188,6 @@ class AsyncWorkflowsClient:
 
         exclude_code : typing.Optional[bool]
 
-        format : typing.Optional[WorkflowsPullRequestFormat]
-
         include_json : typing.Optional[bool]
 
         include_sandbox : typing.Optional[bool]
@@ -216,7 +208,6 @@ class AsyncWorkflowsClient:
             method="GET",
             params={
                 "exclude_code": exclude_code,
-                "format": format,
                 "include_json": include_json,
                 "include_sandbox": include_sandbox,
                 "strict": strict,

vellum_ai-0.14.26/src/vellum/client/resources/workspaces/__init__.py

@@ -0,0 +1,2 @@
+# This file was auto-generated by Fern from our API Definition.
+

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/client/types/__init__.py

@@ -330,7 +330,7 @@ from .rejected_execute_prompt_response import RejectedExecutePromptResponse
 from .rejected_execute_workflow_workflow_result_event import RejectedExecuteWorkflowWorkflowResultEvent
 from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
 from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
-from .release import Release
+from .release_tag_release import ReleaseTagRelease
 from .release_tag_source import ReleaseTagSource
 from .replace_test_suite_test_case_request import ReplaceTestSuiteTestCaseRequest
 from .rich_text_child_block import RichTextChildBlock
@@ -914,7 +914,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
-    "Release",
+    "ReleaseTagRelease",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/client/types/deployment_release_tag_read.py

@@ -4,6 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
 import pydantic
 from .release_tag_source import ReleaseTagSource
 from .deployment_release_tag_deployment_history_item import DeploymentReleaseTagDeploymentHistoryItem
+from .release_tag_release import ReleaseTagRelease
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
@@ -24,7 +25,12 @@ class DeploymentReleaseTagRead(UniversalBaseModel):
 
     history_item: DeploymentReleaseTagDeploymentHistoryItem = pydantic.Field()
     """
-    The Deployment History Item that this Release Tag is associated with
+    Deprecated. Reference the `release` field instead.
+    """
+
+    release: ReleaseTagRelease = pydantic.Field()
+    """
+    The Release that this Release Tag points to.
     """
 
     if IS_PYDANTIC_V2:

vellum_ai-0.14.25/src/vellum/client/types/release.py → vellum_ai-0.14.26/src/vellum/client/types/release_tag_release.py

@@ -7,7 +7,7 @@ import typing
 import pydantic
 
 
-class Release(UniversalBaseModel):
+class ReleaseTagRelease(UniversalBaseModel):
     id: str
     timestamp: dt.datetime
 

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/client/types/workflow_release_tag_read.py

@@ -4,7 +4,7 @@ from ..core.pydantic_utilities import UniversalBaseModel
 import pydantic
 from .release_tag_source import ReleaseTagSource
 from .workflow_release_tag_workflow_deployment_history_item import WorkflowReleaseTagWorkflowDeploymentHistoryItem
-from .release import Release
+from .release_tag_release import ReleaseTagRelease
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
 
@@ -28,7 +28,7 @@ class WorkflowReleaseTagRead(UniversalBaseModel):
     Deprecated. Reference the `release` field instead.
     """
 
-    release: Release = pydantic.Field()
+    release: ReleaseTagRelease = pydantic.Field()
     """
     The Release that this Release Tag points to.
     """

vellum_ai-0.14.26/src/vellum/client/types/workflow_release_tag_workflow_deployment_history_item.py

@@ -0,0 +1,21 @@
+# 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
+
+
+class WorkflowReleaseTagWorkflowDeploymentHistoryItem(UniversalBaseModel):
+    id: str
+    timestamp: dt.datetime
+
+    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_ai-0.14.26/src/vellum/types/release_tag_release.py

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

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/workflows/events/types.py

@@ -28,13 +28,16 @@ def serialize_type_encoder(obj: type) -> Dict[str, Any]:
     }
 
 
-def serialize_type_encoder_with_id(obj: type) -> Dict[str, Any]:
-    if not hasattr(obj, "__id__"):
-        raise AttributeError(f"The object of type '{type(obj).__name__}' must have an '__id__' attribute.")
-    return {
-        "id": getattr(obj, "__id__"),
-        **serialize_type_encoder(obj),
-    }
+def serialize_type_encoder_with_id(obj: Union[type, "CodeResourceDefinition"]) -> Dict[str, Any]:
+    if hasattr(obj, "__id__") and isinstance(obj, type):
+        return {
+            "id": getattr(obj, "__id__"),
+            **serialize_type_encoder(obj),
+        }
+    elif isinstance(obj, CodeResourceDefinition):
+        return obj.model_dump(mode="json")
+
+    raise AttributeError(f"The object of type '{type(obj).__name__}' must have an '__id__' attribute.")
 
 
 def default_serializer(obj: Any) -> Any:

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/workflows/nodes/displayable/bases/inline_prompt_node/node.py

@@ -87,13 +87,11 @@ class BaseInlinePromptNode(BasePromptNode[StateType], Generic[StateType]):
 
     def _get_prompt_event_stream(self) -> Iterator[AdHocExecutePromptEvent]:
         input_variables, input_values = self._compile_prompt_inputs()
-        current_context = get_execution_context()
-        parent_context = current_context.parent_context
-        trace_id = current_context.trace_id
+        execution_context = get_execution_context()
         request_options = self.request_options or RequestOptions()
 
         request_options["additional_body_parameters"] = {
-            "execution_context": {"parent_context": parent_context, "trace_id": trace_id},
+            "execution_context": execution_context.model_dump(mode="json"),
             **request_options.get("additional_body_parameters", {}),
         }
         normalized_functions = (

{vellum_ai-0.14.25 → vellum_ai-0.14.26}/src/vellum/workflows/nodes/displayable/bases/prompt_deployment_node.py

@@ -55,12 +55,10 @@ class BasePromptDeploymentNode(BasePromptNode, Generic[StateType]):
         merge_behavior = MergeBehavior.AWAIT_ANY
 
     def _get_prompt_event_stream(self) -> Iterator[ExecutePromptEvent]:
-        current_context = get_execution_context()
-        trace_id = current_context.trace_id
-        parent_context = current_context.parent_context.model_dump() if current_context.parent_context else None
+        execution_context = get_execution_context()
         request_options = self.request_options or RequestOptions()
         request_options["additional_body_parameters"] = {
-            "execution_context": {"parent_context": parent_context, "trace_id": trace_id},
+            "execution_context": execution_context.model_dump(mode="json"),
             **request_options.get("additional_body_parameters", {}),
         }
         return self._context.vellum_client.execute_prompt_stream(