vellum-ai 0.14.25__py3-none-any.whl → 0.14.27__py3-none-any.whl

vellum/__init__.py

@@ -300,6 +300,7 @@ from .types import (
     PromptDeploymentExpandMetaRequest,
     PromptDeploymentInputRequest,
     PromptDeploymentParentContext,
+    PromptExecConfig,
     PromptExecutionMeta,
     PromptNodeExecutionMeta,
     PromptNodeResult,
@@ -322,7 +323,7 @@ from .types import (
     RejectedExecuteWorkflowWorkflowResultEvent,
     RejectedPromptExecutionMeta,
     RejectedWorkflowNodeResultEvent,
-    Release,
+    ReleaseTagRelease,
     ReleaseTagSource,
     ReplaceTestSuiteTestCaseRequest,
     RichTextChildBlock,
@@ -580,7 +581,6 @@ from .resources import (
     ListWorkflowReleaseTagsRequestSource,
     ListWorkflowSandboxExamplesRequestTag,
     WorkflowDeploymentsListRequestStatus,
-    WorkflowsPullRequestFormat,
     ad_hoc,
     container_images,
     deployments,
@@ -590,6 +590,7 @@ from .resources import (
     metric_definitions,
     ml_models,
     organizations,
+    prompts,
     sandboxes,
     test_suite_runs,
     test_suites,
@@ -911,6 +912,7 @@ __all__ = [
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
     "PromptDeploymentParentContext",
+    "PromptExecConfig",
     "PromptExecutionMeta",
     "PromptNodeExecutionMeta",
     "PromptNodeResult",
@@ -933,7 +935,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
-    "Release",
+    "ReleaseTagRelease",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",
@@ -1182,7 +1184,6 @@ __all__ = [
     "WorkflowSandboxExample",
     "WorkflowSandboxParentContext",
     "WorkflowStreamEvent",
-    "WorkflowsPullRequestFormat",
     "WorkspaceRead",
     "WorkspaceSecretRead",
     "__version__",
@@ -1195,6 +1196,7 @@ __all__ = [
     "metric_definitions",
     "ml_models",
     "organizations",
+    "prompts",
     "sandboxes",
     "test_suite_runs",
     "test_suites",

vellum/client/__init__.py

@@ -13,6 +13,7 @@ from .resources.folder_entities.client import FolderEntitiesClient
 from .resources.metric_definitions.client import MetricDefinitionsClient
 from .resources.ml_models.client import MlModelsClient
 from .resources.organizations.client import OrganizationsClient
+from .resources.prompts.client import PromptsClient
 from .resources.sandboxes.client import SandboxesClient
 from .resources.test_suite_runs.client import TestSuiteRunsClient
 from .resources.test_suites.client import TestSuitesClient
@@ -69,6 +70,7 @@ from .resources.folder_entities.client import AsyncFolderEntitiesClient
 from .resources.metric_definitions.client import AsyncMetricDefinitionsClient
 from .resources.ml_models.client import AsyncMlModelsClient
 from .resources.organizations.client import AsyncOrganizationsClient
+from .resources.prompts.client import AsyncPromptsClient
 from .resources.sandboxes.client import AsyncSandboxesClient
 from .resources.test_suite_runs.client import AsyncTestSuiteRunsClient
 from .resources.test_suites.client import AsyncTestSuitesClient
@@ -145,6 +147,7 @@ class Vellum:
         self.metric_definitions = MetricDefinitionsClient(client_wrapper=self._client_wrapper)
         self.ml_models = MlModelsClient(client_wrapper=self._client_wrapper)
         self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
+        self.prompts = PromptsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = SandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suite_runs = TestSuiteRunsClient(client_wrapper=self._client_wrapper)
         self.test_suites = TestSuitesClient(client_wrapper=self._client_wrapper)
@@ -1486,6 +1489,7 @@ class AsyncVellum:
         self.metric_definitions = AsyncMetricDefinitionsClient(client_wrapper=self._client_wrapper)
         self.ml_models = AsyncMlModelsClient(client_wrapper=self._client_wrapper)
         self.organizations = AsyncOrganizationsClient(client_wrapper=self._client_wrapper)
+        self.prompts = AsyncPromptsClient(client_wrapper=self._client_wrapper)
         self.sandboxes = AsyncSandboxesClient(client_wrapper=self._client_wrapper)
         self.test_suite_runs = AsyncTestSuiteRunsClient(client_wrapper=self._client_wrapper)
         self.test_suites = AsyncTestSuitesClient(client_wrapper=self._client_wrapper)

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.27",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/client/core/jsonable_encoder.py

@@ -45,7 +45,7 @@ def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any]
             encoder = getattr(obj.__config__, "json_encoders", {})  # type: ignore # Pydantic v1
         if custom_encoder:
             encoder.update(custom_encoder)
-        obj_dict = obj.dict(by_alias=True)
+        obj_dict = obj.model_dump(mode="json")
         if "__root__" in obj_dict:
             obj_dict = obj_dict["__root__"]
         if "root" in obj_dict:

vellum/client/resources/__init__.py

@@ -10,6 +10,7 @@ from . import (
     metric_definitions,
     ml_models,
     organizations,
+    prompts,
     sandboxes,
     test_suite_runs,
     test_suites,
@@ -24,7 +25,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 +34,6 @@ __all__ = [
     "ListWorkflowReleaseTagsRequestSource",
     "ListWorkflowSandboxExamplesRequestTag",
     "WorkflowDeploymentsListRequestStatus",
-    "WorkflowsPullRequestFormat",
     "ad_hoc",
     "container_images",
     "deployments",
@@ -44,6 +43,7 @@ __all__ = [
     "metric_definitions",
     "ml_models",
     "organizations",
+    "prompts",
     "sandboxes",
     "test_suite_runs",
     "test_suites",

vellum/client/resources/prompts/__init__.py

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

vellum/client/resources/prompts/client.py

@@ -0,0 +1,197 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ...core.client_wrapper import SyncClientWrapper
+import typing
+from ...core.request_options import RequestOptions
+from ...types.prompt_exec_config import PromptExecConfig
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.pydantic_utilities import parse_obj_as
+from ...errors.bad_request_error import BadRequestError
+from ...errors.not_found_error import NotFoundError
+from json.decoder import JSONDecodeError
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper
+
+
+class PromptsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def pull(
+        self,
+        id: str,
+        *,
+        prompt_variant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PromptExecConfig:
+        """
+        Used to pull the definition of a Prompt from Vellum.
+
+        Parameters
+        ----------
+        id : str
+            The ID of the Prompt to pull from. Prompt Sandbox IDs are currently supported.
+
+        prompt_variant_id : typing.Optional[str]
+            The ID of the Prompt Variant within a Prompt Sandbox to pull. Must be included if providing the ID of a Prompt Sandbox.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PromptExecConfig
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.prompts.pull(
+            id="id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"v1/prompts/{jsonable_encoder(id)}/pull",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "prompt_variant_id": prompt_variant_id,
+            },
+            headers={
+                "Accept": "application/json",
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PromptExecConfig,
+                    parse_obj_as(
+                        type_=PromptExecConfig,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncPromptsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def pull(
+        self,
+        id: str,
+        *,
+        prompt_variant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PromptExecConfig:
+        """
+        Used to pull the definition of a Prompt from Vellum.
+
+        Parameters
+        ----------
+        id : str
+            The ID of the Prompt to pull from. Prompt Sandbox IDs are currently supported.
+
+        prompt_variant_id : typing.Optional[str]
+            The ID of the Prompt Variant within a Prompt Sandbox to pull. Must be included if providing the ID of a Prompt Sandbox.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PromptExecConfig
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.prompts.pull(
+                id="id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"v1/prompts/{jsonable_encoder(id)}/pull",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "prompt_variant_id": prompt_variant_id,
+            },
+            headers={
+                "Accept": "application/json",
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PromptExecConfig,
+                    parse_obj_as(
+                        type_=PromptExecConfig,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    )
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

vellum/client/resources/workflows/__init__.py

@@ -1,5 +1,2 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import WorkflowsPullRequestFormat
-
-__all__ = ["WorkflowsPullRequestFormat"]

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/client/types/__init__.py

@@ -308,6 +308,7 @@ from .prompt_block_state import PromptBlockState
 from .prompt_deployment_expand_meta_request import PromptDeploymentExpandMetaRequest
 from .prompt_deployment_input_request import PromptDeploymentInputRequest
 from .prompt_deployment_parent_context import PromptDeploymentParentContext
+from .prompt_exec_config import PromptExecConfig
 from .prompt_execution_meta import PromptExecutionMeta
 from .prompt_node_execution_meta import PromptNodeExecutionMeta
 from .prompt_node_result import PromptNodeResult
@@ -330,7 +331,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
@@ -892,6 +893,7 @@ __all__ = [
     "PromptDeploymentExpandMetaRequest",
     "PromptDeploymentInputRequest",
     "PromptDeploymentParentContext",
+    "PromptExecConfig",
     "PromptExecutionMeta",
     "PromptNodeExecutionMeta",
     "PromptNodeResult",
@@ -914,7 +916,7 @@ __all__ = [
     "RejectedExecuteWorkflowWorkflowResultEvent",
     "RejectedPromptExecutionMeta",
     "RejectedWorkflowNodeResultEvent",
-    "Release",
+    "ReleaseTagRelease",
     "ReleaseTagSource",
     "ReplaceTestSuiteTestCaseRequest",
     "RichTextChildBlock",

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/client/types/prompt_exec_config.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+from ..core.pydantic_utilities import UniversalBaseModel
+from .array_vellum_value import ArrayVellumValue
+from .chat_message_prompt_block import ChatMessagePromptBlock
+import typing
+from .vellum_variable import VellumVariable
+from .prompt_parameters import PromptParameters
+from .prompt_settings import PromptSettings
+from .prompt_block import PromptBlock
+from .function_definition import FunctionDefinition
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class PromptExecConfig(UniversalBaseModel):
+    ml_model: str
+    input_variables: typing.List[VellumVariable]
+    parameters: PromptParameters
+    settings: typing.Optional[PromptSettings] = None
+    blocks: typing.List[PromptBlock]
+    functions: typing.Optional[typing.List[FunctionDefinition]] = None
+
+    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
+
+
+update_forward_refs(ArrayVellumValue, PromptExecConfig=PromptExecConfig)
+update_forward_refs(ChatMessagePromptBlock, PromptExecConfig=PromptExecConfig)

vellum/client/types/{release.py → release_tag_release.py}

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

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/client/types/workflow_release_tag_workflow_deployment_history_item.py

@@ -1,22 +1,15 @@
 # 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
 
 
 class WorkflowReleaseTagWorkflowDeploymentHistoryItem(UniversalBaseModel):
-    id: str = pydantic.Field()
-    """
-    The ID of the Workflow Deployment History Item
-    """
-
-    timestamp: dt.datetime = pydantic.Field()
-    """
-    The timestamp representing when this History Item was created
-    """
+    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

vellum/{types/release.py → resources/prompts/__init__.py}

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

vellum/resources/{workflows/types/__init__.py → prompts/client.py}

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

vellum/{resources/workflows/types/workflows_pull_request_format.py → types/prompt_exec_config.py}

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

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/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/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/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(