vellum-ai 0.14.5__tar.gz → 0.14.6__tar.gz

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/PKG-INFO

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

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/ee/vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py

@@ -7,6 +7,7 @@ from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.inputs import BaseInputs
 from vellum.workflows.nodes.bases import BaseNode
 from vellum.workflows.outputs import BaseOutputs
+from vellum.workflows.references import LazyReference
 from vellum_ee.workflows.display.nodes.base_node_vellum_display import BaseNodeVellumDisplay
 from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input_value_pointer_rules
@@ -43,6 +44,10 @@ class MyNodeADisplay(BaseNodeVellumDisplay[MyNodeA]):
 class MyNodeB(BaseNode):
     example = MyNodeA.Outputs.output
     fallback_example = MyNodeA.Outputs.output.coalesce(Inputs.example_workflow_input).coalesce("fallback")
+    constant_coalesce = Inputs.example_workflow_input.coalesce("default_value")
+    lazy_coalesce: BaseDescriptor[str] = LazyReference(
+        lambda: MyNodeA.Outputs.output.coalesce(Inputs.example_workflow_input)
+    )
 
 
 @pytest.mark.parametrize(
@@ -76,6 +81,32 @@ class MyNodeB(BaseNode):
                 ConstantValuePointer(type="CONSTANT_VALUE", data=StringVellumValue(value="fallback")),
             ],
         ),
+        (
+            MyNodeB.constant_coalesce,
+            [
+                InputVariablePointer(
+                    type="INPUT_VARIABLE",
+                    data=InputVariableData(input_variable_id="a154c29d-fac0-4cd0-ba88-bc52034f5470"),
+                ),
+                ConstantValuePointer(type="CONSTANT_VALUE", data=StringVellumValue(value="default_value")),
+            ],
+        ),
+        (
+            MyNodeB.lazy_coalesce,
+            [
+                NodeOutputPointer(
+                    type="NODE_OUTPUT",
+                    data=NodeOutputData(
+                        node_id="b48fa5e0-d7d3-4fe3-ae48-615415011cc5",
+                        output_id="4b16a629-11a1-4b3f-a965-a57b872d13b8",
+                    ),
+                ),
+                InputVariablePointer(
+                    type="INPUT_VARIABLE",
+                    data=InputVariableData(input_variable_id="a154c29d-fac0-4cd0-ba88-bc52034f5470"),
+                ),
+            ],
+        ),
     ],
 )
 def test_create_node_input_value_pointer_rules(

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/ee/vellum_ee/workflows/display/nodes/vellum/utils.py

@@ -5,8 +5,10 @@ from vellum.workflows.descriptors.base import BaseDescriptor
 from vellum.workflows.expressions.coalesce_expression import CoalesceExpression
 from vellum.workflows.nodes.displayable.bases.utils import primitive_to_vellum_value
 from vellum.workflows.references import NodeReference
+from vellum.workflows.references.lazy import LazyReference
 from vellum.workflows.utils.uuids import uuid4_from_hash
 from vellum_ee.workflows.display.types import WorkflowDisplayContext
+from vellum_ee.workflows.display.utils.expressions import get_child_descriptor
 from vellum_ee.workflows.display.utils.vellum import create_node_input_value_pointer_rule
 from vellum_ee.workflows.display.vellum import (
     ConstantValuePointer,
@@ -56,6 +58,12 @@ def create_node_input_value_pointer_rules(
                 raise ValueError(f"Expected NodeReference {value.name} to have an instance")
             value = cast(BaseDescriptor, value.instance)
 
+        if isinstance(value, LazyReference):
+            child_descriptor = get_child_descriptor(value, display_context)
+            return create_node_input_value_pointer_rules(
+                child_descriptor, display_context, [], pointer_type=pointer_type
+            )
+
         if isinstance(value, CoalesceExpression):
             # Recursively handle the left-hand side
             lhs_rules = create_node_input_value_pointer_rules(value.lhs, display_context, [], pointer_type=pointer_type)

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vellum-ai"
-version = "0.14.5"
+version = "0.14.6"
 description = ""
 readme = "README.md"
 authors = []

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/src/vellum/__init__.py

@@ -259,6 +259,7 @@ from .types import (
     PaginatedTestSuiteRunExecutionList,
     PaginatedTestSuiteTestCaseList,
     PaginatedWorkflowReleaseTagReadList,
+    PaginatedWorkflowSandboxExampleList,
     PdfSearchResultMetaSource,
     PdfSearchResultMetaSourceRequest,
     PlainTextPromptBlock,
@@ -495,6 +496,7 @@ from .types import (
     WorkflowResultEventOutputDataNumber,
     WorkflowResultEventOutputDataSearchResults,
     WorkflowResultEventOutputDataString,
+    WorkflowSandboxExample,
     WorkflowStreamEvent,
     WorkspaceRead,
     WorkspaceSecretRead,
@@ -506,6 +508,7 @@ from .resources import (
     FolderEntitiesListRequestEntityStatus,
     ListDeploymentReleaseTagsRequestSource,
     ListWorkflowReleaseTagsRequestSource,
+    ListWorkflowSandboxExamplesRequestTag,
     WorkflowDeploymentsListRequestStatus,
     WorkflowsPullRequestFormat,
     ad_hoc,
@@ -713,6 +716,7 @@ __all__ = [
     "JsonVellumValueRequest",
     "ListDeploymentReleaseTagsRequestSource",
     "ListWorkflowReleaseTagsRequestSource",
+    "ListWorkflowSandboxExamplesRequestTag",
     "LogicalOperator",
     "LogprobsEnum",
     "MapNodeResult",
@@ -796,6 +800,7 @@ __all__ = [
     "PaginatedTestSuiteRunExecutionList",
     "PaginatedTestSuiteTestCaseList",
     "PaginatedWorkflowReleaseTagReadList",
+    "PaginatedWorkflowSandboxExampleList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlock",
@@ -1035,6 +1040,7 @@ __all__ = [
     "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
+    "WorkflowSandboxExample",
     "WorkflowStreamEvent",
     "WorkflowsPullRequestFormat",
     "WorkspaceRead",

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/src/vellum/client/__init__.py

@@ -360,7 +360,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -508,7 +508,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -667,7 +667,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
@@ -794,7 +794,7 @@ class Vellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         event_types : typing.Optional[typing.Sequence[WorkflowExecutionEventType]]
             Optionally specify which events you want to receive. Defaults to only WORKFLOW events. Note that the schema of non-WORKFLOW events is unstable and should be used with caution.
@@ -1717,7 +1717,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -1873,7 +1873,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Prompt Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique within a given Prompt Deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         expand_meta : typing.Optional[PromptDeploymentExpandMetaRequest]
             An optionally specified configuration used to opt in to including additional metadata about this prompt execution in the API response. Corresponding values will be returned under the `meta` key of the API response.
@@ -2040,7 +2040,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         metadata : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
             Arbitrary JSON metadata associated with this request. Can be used to capture additional monitoring data such as user id, session id, etc. for future analysis.
@@ -2175,7 +2175,7 @@ class AsyncVellum:
             Optionally specify a release tag if you want to pin to a specific release of the Workflow Deployment
 
         external_id : typing.Optional[str]
-            Optionally include a unique identifier for tracking purposes. Must be unique for a given workflow deployment.
+            Optionally include a unique identifier for tracking purposes. Must be unique within a given Workspace.
 
         event_types : typing.Optional[typing.Sequence[WorkflowExecutionEventType]]
             Optionally specify which events you want to receive. Defaults to only WORKFLOW events. Note that the schema of non-WORKFLOW events is unstable and should be used with caution.

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/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.5",
+            "X-Fern-SDK-Version": "0.14.6",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/src/vellum/client/resources/__init__.py

@@ -23,6 +23,7 @@ from .deployments import DeploymentsListRequestStatus, ListDeploymentReleaseTags
 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__ = [
@@ -31,6 +32,7 @@ __all__ = [
     "FolderEntitiesListRequestEntityStatus",
     "ListDeploymentReleaseTagsRequestSource",
     "ListWorkflowReleaseTagsRequestSource",
+    "ListWorkflowSandboxExamplesRequestTag",
     "WorkflowDeploymentsListRequestStatus",
     "WorkflowsPullRequestFormat",
     "ad_hoc",

vellum_ai-0.14.6/src/vellum/client/resources/workflow_sandboxes/__init__.py

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

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/src/vellum/client/resources/workflow_sandboxes/client.py

@@ -8,6 +8,8 @@ from ...core.jsonable_encoder import jsonable_encoder
 from ...core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
 from ...core.api_error import ApiError
+from .types.list_workflow_sandbox_examples_request_tag import ListWorkflowSandboxExamplesRequestTag
+from ...types.paginated_workflow_sandbox_example_list import PaginatedWorkflowSandboxExampleList
 from ...core.client_wrapper import AsyncClientWrapper
 
 # this is used as the default value for optional parameters
@@ -97,6 +99,74 @@ class WorkflowSandboxesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def list_workflow_sandbox_examples(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        tag: typing.Optional[ListWorkflowSandboxExamplesRequestTag] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedWorkflowSandboxExampleList:
+        """
+        List Workflow Sandbox examples that were previously cloned into the User's Workspace
+
+        Parameters
+        ----------
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        tag : typing.Optional[ListWorkflowSandboxExamplesRequestTag]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedWorkflowSandboxExampleList
+
+
+        Examples
+        --------
+        from vellum import Vellum
+
+        client = Vellum(
+            api_key="YOUR_API_KEY",
+        )
+        client.workflow_sandboxes.list_workflow_sandbox_examples()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "v1/workflow-sandboxes/examples",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "tag": tag,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedWorkflowSandboxExampleList,
+                    parse_obj_as(
+                        type_=PaginatedWorkflowSandboxExampleList,  # 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 AsyncWorkflowSandboxesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -188,3 +258,79 @@ class AsyncWorkflowSandboxesClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def list_workflow_sandbox_examples(
+        self,
+        *,
+        limit: typing.Optional[int] = None,
+        offset: typing.Optional[int] = None,
+        ordering: typing.Optional[str] = None,
+        tag: typing.Optional[ListWorkflowSandboxExamplesRequestTag] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> PaginatedWorkflowSandboxExampleList:
+        """
+        List Workflow Sandbox examples that were previously cloned into the User's Workspace
+
+        Parameters
+        ----------
+        limit : typing.Optional[int]
+            Number of results to return per page.
+
+        offset : typing.Optional[int]
+            The initial index from which to return the results.
+
+        ordering : typing.Optional[str]
+            Which field to use when ordering the results.
+
+        tag : typing.Optional[ListWorkflowSandboxExamplesRequestTag]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        PaginatedWorkflowSandboxExampleList
+
+
+        Examples
+        --------
+        import asyncio
+
+        from vellum import AsyncVellum
+
+        client = AsyncVellum(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.workflow_sandboxes.list_workflow_sandbox_examples()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "v1/workflow-sandboxes/examples",
+            base_url=self._client_wrapper.get_environment().default,
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
+                "ordering": ordering,
+                "tag": tag,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    PaginatedWorkflowSandboxExampleList,
+                    parse_obj_as(
+                        type_=PaginatedWorkflowSandboxExampleList,  # 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_ai-0.14.6/src/vellum/client/resources/workflow_sandboxes/types/__init__.py

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

vellum_ai-0.14.6/src/vellum/client/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ListWorkflowSandboxExamplesRequestTag = typing.Union[typing.Literal["ONBOARDING", "TEMPLATES"], typing.Any]

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/src/vellum/client/types/__init__.py

@@ -267,6 +267,7 @@ from .paginated_slim_workflow_deployment_list import PaginatedSlimWorkflowDeploy
 from .paginated_test_suite_run_execution_list import PaginatedTestSuiteRunExecutionList
 from .paginated_test_suite_test_case_list import PaginatedTestSuiteTestCaseList
 from .paginated_workflow_release_tag_read_list import PaginatedWorkflowReleaseTagReadList
+from .paginated_workflow_sandbox_example_list import PaginatedWorkflowSandboxExampleList
 from .pdf_search_result_meta_source import PdfSearchResultMetaSource
 from .pdf_search_result_meta_source_request import PdfSearchResultMetaSourceRequest
 from .plain_text_prompt_block import PlainTextPromptBlock
@@ -519,6 +520,7 @@ from .workflow_result_event_output_data_json import WorkflowResultEventOutputDat
 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
+from .workflow_sandbox_example import WorkflowSandboxExample
 from .workflow_stream_event import WorkflowStreamEvent
 from .workspace_read import WorkspaceRead
 from .workspace_secret_read import WorkspaceSecretRead
@@ -779,6 +781,7 @@ __all__ = [
     "PaginatedTestSuiteRunExecutionList",
     "PaginatedTestSuiteTestCaseList",
     "PaginatedWorkflowReleaseTagReadList",
+    "PaginatedWorkflowSandboxExampleList",
     "PdfSearchResultMetaSource",
     "PdfSearchResultMetaSourceRequest",
     "PlainTextPromptBlock",
@@ -1015,6 +1018,7 @@ __all__ = [
     "WorkflowResultEventOutputDataNumber",
     "WorkflowResultEventOutputDataSearchResults",
     "WorkflowResultEventOutputDataString",
+    "WorkflowSandboxExample",
     "WorkflowStreamEvent",
     "WorkspaceRead",
     "WorkspaceSecretRead",

vellum_ai-0.14.6/src/vellum/client/types/paginated_workflow_sandbox_example_list.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .workflow_sandbox_example import WorkflowSandboxExample
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class PaginatedWorkflowSandboxExampleList(UniversalBaseModel):
+    count: typing.Optional[int] = None
+    next: typing.Optional[str] = None
+    previous: typing.Optional[str] = None
+    results: typing.Optional[typing.List[WorkflowSandboxExample]] = 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

vellum_ai-0.14.6/src/vellum/client/types/workflow_sandbox_example.py

@@ -0,0 +1,22 @@
+# 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
+
+
+class WorkflowSandboxExample(UniversalBaseModel):
+    id: str
+    label: str
+    description: typing.Optional[str] = None
+    icon_name: typing.Optional[str] = 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

vellum_ai-0.14.6/src/vellum/resources/workflow_sandboxes/types/__init__.py

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

vellum_ai-0.14.6/src/vellum/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py

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

vellum_ai-0.14.6/src/vellum/types/paginated_workflow_sandbox_example_list.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.paginated_workflow_sandbox_example_list import *

vellum_ai-0.14.6/src/vellum/types/workflow_sandbox_example.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.workflow_sandbox_example import *

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/src/vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py

@@ -127,10 +127,20 @@ class SubworkflowDeploymentNode(BaseNode[StateType], Generic[StateType]):
             "execution_context": {"parent_context": parent_context},
             **request_options.get("additional_body_parameters", {}),
         }
+
+        try:
+            deployment_id = str(self.deployment) if isinstance(self.deployment, UUID) else None
+            deployment_name = self.deployment if isinstance(self.deployment, str) else None
+        except AttributeError:
+            raise NodeException(
+                code=WorkflowErrorCode.NODE_EXECUTION,
+                message="Expected subworkflow deployment attribute to be either a UUID or STR, got None instead",
+            )
+
         subworkflow_stream = self._context.vellum_client.execute_workflow_stream(
             inputs=self._compile_subworkflow_inputs(),
-            workflow_deployment_id=str(self.deployment) if isinstance(self.deployment, UUID) else None,
-            workflow_deployment_name=self.deployment if isinstance(self.deployment, str) else None,
+            workflow_deployment_id=deployment_id,
+            workflow_deployment_name=deployment_name,
             release_tag=self.release_tag,
             external_id=self.external_id,
             event_types=["WORKFLOW"],

{vellum_ai-0.14.5 → vellum_ai-0.14.6}/src/vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/test_node.py

@@ -11,6 +11,8 @@ from vellum.client.types.workflow_request_chat_history_input_request import Work
 from vellum.client.types.workflow_request_json_input_request import WorkflowRequestJsonInputRequest
 from vellum.client.types.workflow_result_event import WorkflowResultEvent
 from vellum.client.types.workflow_stream_event import WorkflowStreamEvent
+from vellum.workflows.errors import WorkflowErrorCode
+from vellum.workflows.exceptions import NodeException
 from vellum.workflows.nodes.displayable.subworkflow_deployment_node.node import SubworkflowDeploymentNode
 
 
@@ -129,3 +131,24 @@ def test_run_workflow__any_array(vellum_client):
     assert call_kwargs["inputs"] == [
         WorkflowRequestJsonInputRequest(name="fruits", value=["apple", "banana", "cherry"]),
     ]
+
+
+def test_run_workflow__no_deployment():
+    """Confirm that we raise error when running a subworkflow deployment node with no deployment attribute set"""
+
+    # GIVEN a Subworkflow Deployment Node
+    class ExampleSubworkflowDeploymentNode(SubworkflowDeploymentNode):
+        subworkflow_inputs = {
+            "fruits": ["apple", "banana", "cherry"],
+        }
+
+    # WHEN/THEN running the node should raise a NodeException
+    node = ExampleSubworkflowDeploymentNode()
+    with pytest.raises(NodeException) as exc_info:
+        list(node.run())
+
+    # AND the error message should be correct
+    assert exc_info.value.code == WorkflowErrorCode.NODE_EXECUTION
+    assert "Expected subworkflow deployment attribute to be either a UUID or STR, got None instead" in str(
+        exc_info.value
+    )