vellum-ai 0.14.29__py3-none-any.whl → 0.14.30__py3-none-any.whl

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.29",
+            "X-Fern-SDK-Version": "0.14.30",
         }
         headers["X_API_KEY"] = self.api_key
         return headers

vellum/client/resources/sandboxes/client.py

@@ -30,6 +30,7 @@ class SandboxesClient:
         prompt_deployment_name: typing.Optional[str] = OMIT,
         label: typing.Optional[str] = OMIT,
         release_tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        release_description: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DeploymentRead:
         """
@@ -53,6 +54,9 @@ class SandboxesClient:
         release_tags : typing.Optional[typing.Sequence[str]]
             Optionally provide the release tags that you'd like to be associated with the latest release of the created/updated Prompt Deployment.
 
+        release_description : typing.Optional[str]
+            Optionally provide a description that details what's new in this Release.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -82,6 +86,7 @@ class SandboxesClient:
                 "prompt_deployment_name": prompt_deployment_name,
                 "label": label,
                 "release_tags": release_tags,
+                "release_description": release_description,
             },
             request_options=request_options,
             omit=OMIT,
@@ -250,6 +255,7 @@ class AsyncSandboxesClient:
         prompt_deployment_name: typing.Optional[str] = OMIT,
         label: typing.Optional[str] = OMIT,
         release_tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        release_description: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DeploymentRead:
         """
@@ -273,6 +279,9 @@ class AsyncSandboxesClient:
         release_tags : typing.Optional[typing.Sequence[str]]
             Optionally provide the release tags that you'd like to be associated with the latest release of the created/updated Prompt Deployment.
 
+        release_description : typing.Optional[str]
+            Optionally provide a description that details what's new in this Release.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -310,6 +319,7 @@ class AsyncSandboxesClient:
                 "prompt_deployment_name": prompt_deployment_name,
                 "label": label,
                 "release_tags": release_tags,
+                "release_description": release_description,
             },
             request_options=request_options,
             omit=OMIT,

vellum/client/resources/workflow_sandboxes/client.py

@@ -29,6 +29,7 @@ class WorkflowSandboxesClient:
         workflow_deployment_name: typing.Optional[str] = OMIT,
         label: typing.Optional[str] = OMIT,
         release_tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        release_description: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkflowDeploymentRead:
         """
@@ -52,6 +53,9 @@ class WorkflowSandboxesClient:
         release_tags : typing.Optional[typing.Sequence[str]]
             Optionally provide the release tags that you'd like to be associated with the latest release of the created/updated Prompt Deployment.
 
+        release_description : typing.Optional[str]
+            Optionally provide a description that details what's new in this Release.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -81,6 +85,7 @@ class WorkflowSandboxesClient:
                 "workflow_deployment_name": workflow_deployment_name,
                 "label": label,
                 "release_tags": release_tags,
+                "release_description": release_description,
             },
             request_options=request_options,
             omit=OMIT,
@@ -181,6 +186,7 @@ class AsyncWorkflowSandboxesClient:
         workflow_deployment_name: typing.Optional[str] = OMIT,
         label: typing.Optional[str] = OMIT,
         release_tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        release_description: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> WorkflowDeploymentRead:
         """
@@ -204,6 +210,9 @@ class AsyncWorkflowSandboxesClient:
         release_tags : typing.Optional[typing.Sequence[str]]
             Optionally provide the release tags that you'd like to be associated with the latest release of the created/updated Prompt Deployment.
 
+        release_description : typing.Optional[str]
+            Optionally provide a description that details what's new in this Release.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -241,6 +250,7 @@ class AsyncWorkflowSandboxesClient:
                 "workflow_deployment_name": workflow_deployment_name,
                 "label": label,
                 "release_tags": release_tags,
+                "release_description": release_description,
             },
             request_options=request_options,
             omit=OMIT,

vellum/workflows/nodes/bases/tests/test_base_node.py

@@ -1,9 +1,12 @@
+import pytest
 from uuid import UUID
 from typing import Optional
 
 from vellum.client.types.string_vellum_value_request import StringVellumValueRequest
 from vellum.core.pydantic_utilities import UniversalBaseModel
+from vellum.workflows.descriptors.tests.test_utils import FixtureState
 from vellum.workflows.inputs.base import BaseInputs
+from vellum.workflows.nodes import FinalOutputNode
 from vellum.workflows.nodes.bases.base import BaseNode
 from vellum.workflows.outputs.base import BaseOutputs
 from vellum.workflows.state.base import BaseState, StateMeta
@@ -232,3 +235,27 @@ def test_outputs_preserves_non_object_bases():
     # THEN the output values should be correct
     assert outputs.foo == "bar"
     assert outputs.bar == "baz"
+
+
+@pytest.mark.parametrize(
+    "falsy_value,expected_type",
+    [
+        ("", str),
+        (0, int),
+    ],
+)
+def test_resolve_value__for_falsy_values(falsy_value, expected_type):
+    """Test that falsy values in FinalOutputNode are handled correctly."""
+
+    class FalsyOutput(FinalOutputNode[FixtureState, expected_type]):  # type: ignore
+        class Outputs(FinalOutputNode.Outputs):
+            value = falsy_value
+
+    # GIVEN a node with a falsy value
+    falsy_node = FalsyOutput(state=FixtureState())
+
+    # WHEN we run the node
+    falsy_output = falsy_node.run()
+
+    # THEN the output has the correct value
+    assert falsy_output.value == falsy_value

vellum/workflows/outputs/base.py

@@ -145,7 +145,7 @@ class _BaseOutputsMeta(type):
             # we iterate through its inheritance hierarchy to find the first base class that has this attribute
             # and use its mapping.
             instance = vars(cls).get(name, undefined)
-            if not instance:
+            if instance is undefined:
                 for base in cls.__mro__[1:]:
                     if hasattr(base, name):
                         instance = getattr(base, name)

{vellum_ai-0.14.29.dist-info → vellum_ai-0.14.30.dist-info}/METADATA

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

{vellum_ai-0.14.29.dist-info → vellum_ai-0.14.30.dist-info}/RECORD

@@ -7,7 +7,7 @@ vellum_cli/image_push.py,sha256=8DDvRDJEZ-FukUCqGW1827bg1ybF4xBbx9WyqWYQE-g,6816
 vellum_cli/init.py,sha256=WpnMXPItPmh0f0bBGIer3p-e5gu8DUGwSArT_FuoMEw,5093
 vellum_cli/logger.py,sha256=PuRFa0WCh4sAGFS5aqWB0QIYpS6nBWwPJrIXpWxugV4,1022
 vellum_cli/ping.py,sha256=lWyJw6sziXjyTopTYRdFF5hV-sYPVDdX0yVbG5fzcY4,585
-vellum_cli/pull.py,sha256=XrlJqImcqZcr6SRGqJ4x3yyvc_0LHDejBcfeVRpY1mY,9169
+vellum_cli/pull.py,sha256=GTYNJDzEeJ3Z48i6YjL2sF2ERFt4czOR7T6lEoQ_J6Q,9176
 vellum_cli/push.py,sha256=xjTNbLwOVFNU3kpBrm56Bk5QkSRrJ9z86qceghCzfIA,9655
 vellum_cli/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_cli/tests/conftest.py,sha256=AFYZryKA2qnUuCPBxBKmHLFoPiE0WhBFFej9tNwSHdc,1526
@@ -16,7 +16,7 @@ vellum_cli/tests/test_image_push.py,sha256=i3lJuW8nFRwL1M1OF6752IZYvGAFgKmkB2hd_
 vellum_cli/tests/test_init.py,sha256=8UOc_ThfouR4ja5cCl_URuLk7ohr9JXfCnG4yka1OUQ,18754
 vellum_cli/tests/test_main.py,sha256=qDZG-aQauPwBwM6A2DIu1494n47v3pL28XakTbLGZ-k,272
 vellum_cli/tests/test_ping.py,sha256=QtbhYKMYn1DFnDyBij2mkQO32j9KOpZ5Pf0yek7k_Ao,1284
-vellum_cli/tests/test_pull.py,sha256=09BkkBoFvqJXIFRxdCu-_a6CE6FtGzqXkXMPaKlcvwE,30178
+vellum_cli/tests/test_pull.py,sha256=qIoft5Pu49Os6760rgtOG80PNYW0ERdF10N1h0380sA,32089
 vellum_cli/tests/test_push.py,sha256=zDv_Q1hbXtLwmTJDPRAvwDjbuHC09uNRYOy4FQujUow,23476
 vellum_ee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -128,7 +128,7 @@ vellum/client/README.md,sha256=JkCJjmMZl4jrPj46pkmL9dpK4gSzQQmP5I7z4aME4LY,4749
 vellum/client/__init__.py,sha256=Jv9sI5BNFo2OYA9px_aREFSIp655ryC3eaZSRI6yH1k,117826
 vellum/client/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
 vellum/client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-vellum/client/core/client_wrapper.py,sha256=WRK8WU4WFDaADxcUMi93lvUkCQp0McOsjxKX1PJ4Zyc,1869
+vellum/client/core/client_wrapper.py,sha256=n2emxJFfx0tF0nSGCah08qytcpcFnvJBgH2UN0WLCFA,1869
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=X9IbmkZmB2bB_DpmZAO3crWdXagOakAyn6UCOCImCPg,2322
 vellum/client/core/http_client.py,sha256=R0pQpCppnEtxccGvXl4uJ76s7ro_65Fo_erlNNLp_AI,19228
@@ -173,7 +173,7 @@ vellum/client/resources/organizations/client.py,sha256=Uye92moqjAcOCs4astmuFpT92
 vellum/client/resources/prompts/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/client/resources/prompts/client.py,sha256=_rNTUjhl_ZF3vyQa_M1BSTrX4DlFXU_SXkwwCEYKD2s,6598
 vellum/client/resources/sandboxes/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-vellum/client/resources/sandboxes/client.py,sha256=i-6DHap5k6gFcYS-kWI8ayJFVZxb-GENRft6BJwVam4,17158
+vellum/client/resources/sandboxes/client.py,sha256=SG4BV0NG1Ow10mXSu52ybj-c6hR7wcNxfGJK4eiHu_8,17670
 vellum/client/resources/test_suite_runs/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 vellum/client/resources/test_suite_runs/client.py,sha256=tU-N1fEfXQPomt2f058PUNIhnGoMV4vo471PlQanTAs,15128
 vellum/client/resources/test_suites/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -184,7 +184,7 @@ vellum/client/resources/workflow_deployments/types/__init__.py,sha256=W7DKJ1nduw
 vellum/client/resources/workflow_deployments/types/list_workflow_release_tags_request_source.py,sha256=LPETHLX9Ygha_JRT9oWZAZR6clv-W1tTelXzktkTBX8,178
 vellum/client/resources/workflow_deployments/types/workflow_deployments_list_request_status.py,sha256=FXVkVmGM6DZ2RpTGnZXWJYiVlLQ-K5fDtX3WMaBPaWk,182
 vellum/client/resources/workflow_sandboxes/__init__.py,sha256=OR3wE3pTgsZlTS-0ukeMWzSuEZF8PszuQTCHDh6JybI,175
-vellum/client/resources/workflow_sandboxes/client.py,sha256=4FfB7DCAo8DDd3CDlmRiycMjnZhP4oWEbfGuhtzVfwo,12404
+vellum/client/resources/workflow_sandboxes/client.py,sha256=H6GP3yitYKSSb5D0fSF_dKG467d2NYMQf5bnFo0nSZM,12916
 vellum/client/resources/workflow_sandboxes/types/__init__.py,sha256=EaGVRU1w6kJiiHrbZOeEa0c3ggjfgv_jBqsyOkCRWOI,212
 vellum/client/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py,sha256=TEwWit20W3X-zWPPLAhmUG05UudG9gaBSJ4Q4-rNJws,188
 vellum/client/resources/workflows/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
@@ -1498,7 +1498,7 @@ vellum/workflows/nodes/bases/__init__.py,sha256=cniHuz_RXdJ4TQgD8CBzoiKDiPxg62Er
 vellum/workflows/nodes/bases/base.py,sha256=eW-3RSkBgtuGY8x2nmbHYiUg_HXS5U57n3k6Fh-dJ9s,15330
 vellum/workflows/nodes/bases/base_adornment_node.py,sha256=afMwJLHK2Ke7sBpceVLnNdZMlU2O-6UgyG7lBt9SAQ8,3039
 vellum/workflows/nodes/bases/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vellum/workflows/nodes/bases/tests/test_base_node.py,sha256=4SOdZzvugVtN8CIuo5RrapAxSYGXnxUwQ77dXJ64oTU,6295
+vellum/workflows/nodes/bases/tests/test_base_node.py,sha256=4PGjzFh4ZkvffOn-7xhqzsqfdLsYe_2IGqn002TL4LE,7099
 vellum/workflows/nodes/core/__init__.py,sha256=5zDMCmyt1v0HTJzlUBwq3U9L825yZGZhT9JL18-mRR4,455
 vellum/workflows/nodes/core/error_node/__init__.py,sha256=g7RRnlHhqu4qByfLjBwCunmgGA8dI5gNsjS3h6TwlSI,60
 vellum/workflows/nodes/core/error_node/node.py,sha256=MFHU5vITYSK-L9CuMZ49In2ZeNLWnhZD0f8r5dWvb5Y,1270
@@ -1593,7 +1593,7 @@ vellum/workflows/nodes/tests/test_mocks.py,sha256=mfPvrs75PKcsNsbJLQAN6PDFoVqs9T
 vellum/workflows/nodes/tests/test_utils.py,sha256=qNB6ApIsnFtE_9HDaEah9KD-zX8e10FhDixewS1uRcc,5199
 vellum/workflows/nodes/utils.py,sha256=Tc3TAmAytb-gi30BAyzGY7DG0uS1_7KTGYdjrvKUSt0,8362
 vellum/workflows/outputs/__init__.py,sha256=AyZ4pRh_ACQIGvkf0byJO46EDnSix1ZCAXfvh-ms1QE,94
-vellum/workflows/outputs/base.py,sha256=b4Dnha1miKu3uFJYptKKflIHNuajPF2BNKy0BTt8Tjc,8622
+vellum/workflows/outputs/base.py,sha256=jOvSr44GEmtldits-bysJoxme7js6aN_LkdV-PpA7Dk,8631
 vellum/workflows/ports/__init__.py,sha256=bZuMt-R7z5bKwpu4uPW7LlJeePOQWmCcDSXe5frUY5g,101
 vellum/workflows/ports/node_ports.py,sha256=g4A-8iUAvEJSkaWppbvzAR8XU02R9U-qLN4rP2Kq4Aw,2743
 vellum/workflows/ports/port.py,sha256=eI2SOZPZ5rsC3jMsxW6Rbn0NpaYQsrR7AapiIbbiy8Q,3635
@@ -1650,8 +1650,8 @@ vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnad
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=tCxrV3QBHL8wfdEO3bvKteDdw32xBlUl1_WxkAwaONw,8344
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-0.14.29.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-0.14.29.dist-info/METADATA,sha256=KgSQqeoqVLwdeQuikF8QxV4hr9OaVVyym5OlArQRyPM,5484
-vellum_ai-0.14.29.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-0.14.29.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-0.14.29.dist-info/RECORD,,
+vellum_ai-0.14.30.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-0.14.30.dist-info/METADATA,sha256=RwawNldGcpTbIALzGk6SBEVUdlVisV1xiZTdYAdfLRs,5484
+vellum_ai-0.14.30.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-0.14.30.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-0.14.30.dist-info/RECORD,,

vellum_cli/pull.py

@@ -9,7 +9,6 @@ from dotenv import load_dotenv
 from pydash import snake_case
 
 from vellum.client.core.pydantic_utilities import UniversalBaseModel
-from vellum.utils.uuid import is_valid_uuid
 from vellum.workflows.vellum_client import create_vellum_client
 from vellum_cli.config import VellumCliConfig, WorkflowConfig, load_vellum_cli_config
 from vellum_cli.logger import load_cli_logger
@@ -31,6 +30,7 @@ class RunnerConfig(UniversalBaseModel):
 class PullContentsMetadata(UniversalBaseModel):
     label: Optional[str] = None
     runner_config: Optional[RunnerConfig] = None
+    deployment_name: Optional[str] = None
 
 
 def _resolve_workflow_config(
@@ -78,13 +78,8 @@ def _resolve_workflow_config(
             pk=workflow_config.workflow_sandbox_id,
         )
     elif workflow_deployment:
-        module = (
-            f"workflow_{workflow_deployment.split('-')[0]}"
-            if is_valid_uuid(workflow_deployment)
-            else snake_case(workflow_deployment)
-        )
         workflow_config = WorkflowConfig(
-            module=module,
+            module="",
         )
         config.workflows.append(workflow_config)
         return WorkflowConfigResolutionResult(
@@ -169,7 +164,8 @@ def pull_command(
                 workflow_config.container_image_tag = pull_contents_metadata.runner_config.container_image_tag
                 if workflow_config.container_image_name and not workflow_config.container_image_tag:
                     workflow_config.container_image_tag = "latest"
-
+            if not workflow_config.module and workflow_deployment and pull_contents_metadata.deployment_name:
+                workflow_config.module = snake_case(pull_contents_metadata.deployment_name)
             if not workflow_config.module and pull_contents_metadata.label:
                 workflow_config.module = snake_case(pull_contents_metadata.label)
 

vellum_cli/tests/test_pull.py

@@ -338,7 +338,16 @@ def test_pull__workflow_deployment_with_no_config(vellum_client):
     workflow_deployment = "my-deployment"
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter(
+        [
+            _zip_file_map(
+                {
+                    "workflow.py": "print('hello')",
+                    "metadata.json": json.dumps({"deployment_name": workflow_deployment, "label": "Some Label"}),
+                }
+            )
+        ]
+    )
 
     # AND we are currently in a new directory
     current_dir = os.getcwd()
@@ -822,3 +831,47 @@ def test_pull__multiple_instances_of_same_module__keep_when_pulling_another_modu
     with open(lock_json) as f:
         lock_data = json.load(f)
         assert len(lock_data["workflows"]) == 3
+
+
+def test_pull__module_name_from_deployment_name(vellum_client):
+    # GIVEN a workflow deployment
+    workflow_deployment = "test-workflow-deployment-id"
+
+    # AND the workflow pull API call returns a zip file with metadata containing a deployment_name
+    deployment_name = "Test Deployment"
+    vellum_client.workflows.pull.return_value = iter(
+        [
+            _zip_file_map(
+                {
+                    "workflow.py": "print('hello')",
+                    "metadata.json": json.dumps({"deployment_name": deployment_name, "label": "Some Label"}),
+                }
+            )
+        ]
+    )
+
+    # AND we are currently in a new directory
+    current_dir = os.getcwd()
+    temp_dir = tempfile.mkdtemp()
+    os.chdir(temp_dir)
+
+    # WHEN the user runs the pull command with the workflow deployment
+    runner = CliRunner()
+    result = runner.invoke(cli_main, ["workflows", "pull", "--workflow-deployment", workflow_deployment])
+    os.chdir(current_dir)
+
+    # THEN the command returns successfully
+    assert result.exit_code == 0
+
+    # AND the module name is derived from the deployment_name in metadata.json (snake_cased)
+    vellum_lock_json = os.path.join(temp_dir, "vellum.lock.json")
+    assert os.path.exists(vellum_lock_json)
+    with open(vellum_lock_json) as f:
+        lock_data = json.loads(f.read())
+        assert lock_data["workflows"][0]["module"] == "test_deployment"
+
+    # AND the workflow.py file is written to the module directory with the correct name
+    workflow_py = os.path.join(temp_dir, "test_deployment", "workflow.py")
+    assert os.path.exists(workflow_py)
+    with open(workflow_py) as f:
+        assert f.read() == "print('hello')"