vellum-ai 0.14.63__py3-none-any.whl → 1.13.5__py3-none-any.whl

vellum_cli/tests/test_pull.py

@@ -1,34 +1,17 @@
 import pytest
-import io
 import json
 import os
-import tempfile
+import pathlib
 from unittest import mock
 from uuid import uuid4
-import zipfile
 
 from click.testing import CliRunner
 
 from vellum.client.core.api_error import ApiError
+from vellum.workflows.utils.zip import zip_file_map
 from vellum_cli import main as cli_main
 
 
-def _zip_file_map(file_map: dict[str, str]) -> bytes:
-    # Create an in-memory bytes buffer to store the zip
-    zip_buffer = io.BytesIO()
-
-    # Create zip file and add files from file_map
-    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
-        for filename, content in file_map.items():
-            zip_file.writestr(filename, content)
-
-    # Get the bytes from the buffer
-    zip_bytes = zip_buffer.getvalue()
-    zip_buffer.close()
-
-    return zip_bytes
-
-
 @pytest.mark.parametrize(
     "base_command",
     [
@@ -44,7 +27,7 @@ def test_pull(vellum_client, mock_module, base_command):
     workflow_sandbox_id = mock_module.workflow_sandbox_id
 
     # 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')"})])
 
     # WHEN the user runs the pull command
     runner = CliRunner()
@@ -89,7 +72,7 @@ def test_pull__second_module(vellum_client, mock_module):
     set_pyproject_toml = mock_module.set_pyproject_toml
 
     # 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')"})])
 
     # AND the module we're about to pull is configured second
     set_pyproject_toml(
@@ -134,7 +117,7 @@ def test_pull__with_target_dir(vellum_client, mock_module, base_command):
     os.makedirs(target_dir, exist_ok=True)
 
     # 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')"})])
 
     # WHEN the user runs the pull command with target-dir
     runner = CliRunner()
@@ -195,7 +178,7 @@ def test_pull__with_nested_target_dir(vellum_client, mock_module, base_command):
     nested_target_dir = os.path.join(temp_dir, "dir-1", "dir-2")
 
     # 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')"})])
 
     # WHEN the user runs the pull command with nested target-dir
     runner = CliRunner()
@@ -244,42 +227,35 @@ def test_pull__with_nested_target_dir(vellum_client, mock_module, base_command):
         }
 
 
-def test_pull__sandbox_id_with_no_config(vellum_client):
+def test_pull__sandbox_id_with_no_config(vellum_client, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path):
     # GIVEN a workflow sandbox id
     workflow_sandbox_id = "87654321-0000-0000-0000-000000000000"
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [
-            _zip_file_map(
-                {"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})}
-            )
-        ]
+        [zip_file_map({"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})})]
     )
 
     # AND we are currently in a new directory
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    monkeypatch.chdir(tmp_path)
 
     # WHEN the user runs the pull command with the workflow sandbox id and no module
     runner = CliRunner()
     result = runner.invoke(cli_main, ["workflows", "pull", "--workflow-sandbox-id", workflow_sandbox_id])
-    os.chdir(current_dir)
 
     # THEN the command returns successfully
     assert result.exit_code == 0
 
     # AND the pull api is called with the workflow sandbox id
     vellum_client.workflows.pull.assert_called_once()
-    workflow_py = os.path.join(temp_dir, "super_cool_workflow", "workflow.py")
-    assert os.path.exists(workflow_py)
+    workflow_py = tmp_path / "super_cool_workflow" / "workflow.py"
+    assert workflow_py.exists()
     with open(workflow_py) as f:
         assert f.read() == "print('hello')"
 
     # AND the vellum.lock.json file is created
-    vellum_lock_json = os.path.join(temp_dir, "vellum.lock.json")
-    assert os.path.exists(vellum_lock_json)
+    vellum_lock_json = tmp_path / "vellum.lock.json"
+    assert vellum_lock_json.exists()
     with open(vellum_lock_json) as f:
         lock_data = json.loads(f.read())
         assert lock_data == {
@@ -309,11 +285,7 @@ def test_pull__sandbox_id_with_other_workflow_configured(vellum_client, mock_mod
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [
-            _zip_file_map(
-                {"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})}
-            )
-        ]
+        [zip_file_map({"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})})]
     )
 
     # WHEN the user runs the pull command with the new workflow sandbox id
@@ -335,7 +307,9 @@ def test_pull__sandbox_id_with_other_workflow_configured(vellum_client, mock_mod
         assert f.read() == "print('hello')"
 
 
-def test_pull__workflow_deployment_with_no_config(vellum_client):
+def test_pull__workflow_deployment_with_no_config(
+    vellum_client, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path
+):
     # GIVEN a workflow deployment
     workflow_deployment = "my-deployment"
     deployment_id = str(uuid4())
@@ -343,7 +317,7 @@ def test_pull__workflow_deployment_with_no_config(vellum_client):
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -355,28 +329,25 @@ def test_pull__workflow_deployment_with_no_config(vellum_client):
     )
 
     # AND we are currently in a new directory
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    monkeypatch.chdir(tmp_path)
 
     # 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 pull api is called with the workflow deployment
     vellum_client.workflows.pull.assert_called_once()
-    workflow_py = os.path.join(temp_dir, "my_deployment", "workflow.py")
-    assert os.path.exists(workflow_py)
+    workflow_py = tmp_path / "my_deployment" / "workflow.py"
+    assert workflow_py.exists()
     with open(workflow_py) as f:
         assert f.read() == "print('hello')"
 
     # AND the vellum.lock.json file is created
-    vellum_lock_json = os.path.join(temp_dir, "vellum.lock.json")
-    assert os.path.exists(vellum_lock_json)
+    vellum_lock_json = tmp_path / "vellum.lock.json"
+    assert vellum_lock_json.exists()
     with open(vellum_lock_json) as f:
         lock_data = json.loads(f.read())
         assert lock_data == {
@@ -393,6 +364,7 @@ def test_pull__workflow_deployment_with_no_config(vellum_client):
                             "name": "my-deployment",
                             "description": None,
                             "release_tags": None,
+                            "release_description": None,
                         }
                     ],
                     "container_image_tag": None,
@@ -405,7 +377,9 @@ def test_pull__workflow_deployment_with_no_config(vellum_client):
         }
 
 
-def test_pull__both_workflow_sandbox_id_and_deployment(vellum_client):
+def test_pull__both_workflow_sandbox_id_and_deployment(
+    vellum_client, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path
+):
     # GIVEN a workflow sandbox id
     workflow_sandbox_id = "87654321-0000-0000-0000-000000000000"
 
@@ -413,12 +387,10 @@ def test_pull__both_workflow_sandbox_id_and_deployment(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')"})])
 
     # AND we are currently in a new directory
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    monkeypatch.chdir(tmp_path)
 
     # WHEN the user runs the pull command with the workflow deployment
     runner = CliRunner()
@@ -433,7 +405,6 @@ def test_pull__both_workflow_sandbox_id_and_deployment(vellum_client):
             workflow_deployment,
         ],
     )
-    os.chdir(current_dir)
 
     # THEN the command returns successfully
     assert result.exit_code == 1
@@ -446,7 +417,7 @@ def test_pull__remove_missing_files(vellum_client, mock_module):
     module = mock_module.module
 
     # 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')"})])
 
     # AND there is already a different file in the module directory
     other_file_path = os.path.join(temp_dir, *module.split("."), "other_file.py")
@@ -477,7 +448,7 @@ def test_pull__remove_missing_files__ignore_pattern(vellum_client, mock_module):
     set_pyproject_toml = mock_module.set_pyproject_toml
 
     # 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')"})])
 
     # AND there is already a different file in the module directory
     other_file_path = os.path.join(temp_dir, *module.split("."), "other_file.py")
@@ -529,7 +500,7 @@ def test_pull__include_json(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
+        [zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
     )
 
     # WHEN the user runs the pull command
@@ -551,7 +522,7 @@ def test_pull__exclude_code(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
+        [zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
     )
 
     # WHEN the user runs the pull command
@@ -605,7 +576,7 @@ def test_pull__sandbox_id_with_other_workflow_deployment_in_lock(vellum_client,
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -641,6 +612,7 @@ def test_pull__sandbox_id_with_other_workflow_deployment_in_lock(vellum_client,
                         "name": None,
                         "description": None,
                         "release_tags": None,
+                        "release_description": None,
                     },
                 ],
                 "container_image_name": None,
@@ -669,7 +641,7 @@ def test_pull__handle_error_log(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file with an error log
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "error.log": "test error"})]
+        [zip_file_map({"workflow.py": "print('hello')", "error.log": "test error"})]
     )
 
     # WHEN the user runs the pull command with the new workflow sandbox id
@@ -691,7 +663,7 @@ def test_pull__strict__with_error(vellum_client, mock_module):
     workflow_sandbox_id = mock_module.workflow_sandbox_id
 
     # 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')"})])
 
     # WHEN the user runs the pull command with the new workflow sandbox id
     runner = CliRunner()
@@ -713,7 +685,7 @@ def test_pull__include_sandbox(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "sandbox.py": "print('hello')"})]
+        [zip_file_map({"workflow.py": "print('hello')", "sandbox.py": "print('hello')"})]
     )
 
     # WHEN the user runs the pull command
@@ -742,10 +714,10 @@ def test_pull__same_pull_twice__one_entry_in_lockfile(vellum_client, mock_module
     workflow_sandbox_id = mock_module.workflow_sandbox_id
 
     # AND the workflow pull API call returns a zip file both times
-    zip_contents = _zip_file_map({"workflow.py": "print('hello')"})
+    zip_contents = zip_file_map({"workflow.py": "print('hello')"})
     responses = iter([zip_contents, zip_contents])
 
-    def workflows_pull_side_effect(*args, **kwargs):
+    def workflows_pull_side_effect(*_args, **_kwargs):
         return iter([next(responses)])
 
     vellum_client.workflows.pull.side_effect = workflows_pull_side_effect
@@ -778,7 +750,7 @@ def test_pull__module_not_in_config(vellum_client, mock_module):
     set_pyproject_toml({"workflows": []})
 
     # 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')"})])
 
     # WHEN the user runs the pull command again with the workflow sandbox id and module
     runner = CliRunner()
@@ -831,7 +803,7 @@ def test_pull__multiple_instances_of_same_module__keep_when_pulling_another_modu
         json.dump(lock_data, f)
 
     # 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')"})])
 
     # WHEN the user runs the pull command on the new module
     runner = CliRunner()
@@ -846,7 +818,7 @@ def test_pull__multiple_instances_of_same_module__keep_when_pulling_another_modu
         assert len(lock_data["workflows"]) == 3
 
 
-def test_pull__module_name_from_deployment_name(vellum_client):
+def test_pull__module_name_from_deployment_name(vellum_client, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path):
     # GIVEN a workflow deployment
     workflow_deployment = "test-workflow-deployment-id"
 
@@ -854,7 +826,7 @@ def test_pull__module_name_from_deployment_name(vellum_client):
     deployment_name = "Test Deployment"
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps({"deployment_name": deployment_name, "label": "Some Label"}),
@@ -863,33 +835,33 @@ def test_pull__module_name_from_deployment_name(vellum_client):
         ]
     )
 
-    # AND we are currently in a new directory
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    # AND we are currently in a new directory with a .env file
+    monkeypatch.chdir(tmp_path)
+    with open(tmp_path / ".env", "w") as f:
+        f.write("VELLUM_API_KEY=test-api-key")
 
     # 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)
+    vellum_lock_json = tmp_path / "vellum.lock.json"
+    assert vellum_lock_json.exists()
     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)
+    workflow_py = tmp_path / "test_deployment" / "workflow.py"
+    assert workflow_py.exists()
     with open(workflow_py) as f:
         assert f.read() == "print('hello')"
 
 
+@pytest.mark.usefixtures("mock_module")
 def test_pull__invalid_zip_file(vellum_client):
     workflow_deployment = "test-workflow-deployment-id"
 
@@ -903,11 +875,12 @@ def test_pull__invalid_zip_file(vellum_client):
     # THEN the command returns an error
     assert result.exit_code == 1
     assert (
-        str(result.exception)
-        == "The API we tried to pull from returned an invalid zip file. Please make sure your `VELLUM_API_URL` environment variable is set correctly."  # noqa: E501
+        "Invalid response format" in result.output
+        or "Please verify your `VELLUM_API_URL` environment variable is set correctly" in result.output
     )
 
 
+@pytest.mark.usefixtures("mock_module")
 def test_pull__json_decode_error(vellum_client):
     workflow_deployment = "test-workflow-deployment-id"
 
@@ -927,11 +900,12 @@ def test_pull__json_decode_error(vellum_client):
     # THEN the command returns an error
     assert result.exit_code == 1
     assert (
-        str(result.exception)
-        == "The API we tried to pull is invalid. Please make sure your `VELLUM_API_URL` environment variable is set correctly."  # noqa: E501
+        "API request failed" in result.output
+        or "Please verify your `VELLUM_API_URL` environment variable is set correctly" in result.output
     )
 
 
+@pytest.mark.usefixtures("mock_module")
 def test_pull__unauthorized_error_path(vellum_client):
     workflow_deployment = "test-workflow-deployment-id"
 
@@ -948,9 +922,10 @@ def test_pull__unauthorized_error_path(vellum_client):
 
     # THEN the command returns an error
     assert result.exit_code == 1
-    assert str(result.exception) == "Please make sure your `VELLUM_API_KEY` environment variable is set correctly."
+    assert "Please make sure your `VELLUM_API_KEY` environment variable is set correctly" in result.output
 
 
+@pytest.mark.usefixtures("mock_module")
 def test_pull__unexpected_error_path(vellum_client):
     workflow_deployment = "test-workflow-deployment-id"
 
@@ -968,9 +943,8 @@ def test_pull__unexpected_error_path(vellum_client):
     # THEN the command returns an error
     assert result.exit_code == 1
     assert (
-        str(result.exception)
-        == """The Pull API failed with an unexpected error. \
-Please try again later and contact support if the problem persists."""
+        "Server error occurred" in result.output
+        or "Please try again in a few moments. If the problem persists, contact Vellum support" in result.output
     )
 
 
@@ -978,10 +952,12 @@ Please try again later and contact support if the problem persists."""
     "workflow_deployment",
     [
         "test-workflow-deployment-id",
-        str(uuid4()),
+        "377a59fb-1dd0-4760-9568-ae4bd6188da0",
     ],
 )
-def test_pull__workflow_deployment_adds_deployment_to_config(vellum_client, workflow_deployment):
+def test_pull__workflow_deployment_adds_deployment_to_config(
+    vellum_client, workflow_deployment, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path
+):
     # GIVEN a workflow deployment ID
     deployment_id = str(uuid4())  # config will always use the deployment_id return from the API
     deployment_name = "Test Deployment"
@@ -990,7 +966,7 @@ def test_pull__workflow_deployment_adds_deployment_to_config(vellum_client, work
     # AND the workflow pull API call returns a zip file with metadata
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1006,9 +982,8 @@ def test_pull__workflow_deployment_adds_deployment_to_config(vellum_client, work
     )
 
     # AND we are currently in a new directory
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setenv("VELLUM_API_KEY", "abcdef123456")
 
     # WHEN the user runs the pull command with the workflow deployment
     runner = CliRunner()
@@ -1018,8 +993,8 @@ def test_pull__workflow_deployment_adds_deployment_to_config(vellum_client, work
     assert result.exit_code == 0
 
     # AND the deployment is saved in the config
-    vellum_lock_json = os.path.join(temp_dir, "vellum.lock.json")
-    assert os.path.exists(vellum_lock_json)
+    vellum_lock_json = tmp_path / "vellum.lock.json"
+    assert vellum_lock_json.exists()
     with open(vellum_lock_json) as f:
         lock_data = json.loads(f.read())
         assert len(lock_data["workflows"]) == 1
@@ -1029,21 +1004,18 @@ def test_pull__workflow_deployment_adds_deployment_to_config(vellum_client, work
         assert deployment["name"] == deployment_name
         assert deployment["label"] == deployment_label
 
-    os.chdir(current_dir)
-
 
-def test_pull__workflow_deployment_name_is_uuid(vellum_client):
+def test_pull__workflow_deployment_name_is_uuid(vellum_client, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path):
     # GIVEN a workflow deployment name that is a valid UUID
     deployment_id = str(uuid4())
     deployment_name = str(uuid4())
 
     # AND an existing configuration with this deployment
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setenv("VELLUM_API_KEY", "abcdef123456")
 
     # Create initial config with a deployment
-    vellum_lock_json = os.path.join(temp_dir, "vellum.lock.json")
+    vellum_lock_json = tmp_path / "vellum.lock.json"
     with open(vellum_lock_json, "w") as f:
         json.dump(
             {
@@ -1077,7 +1049,7 @@ def test_pull__workflow_deployment_name_is_uuid(vellum_client):
     updated_label = "Updated Label"
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1109,11 +1081,11 @@ def test_pull__workflow_deployment_name_is_uuid(vellum_client):
         assert deployment["name"] == deployment_name
         assert deployment["label"] == updated_label
 
-    os.chdir(current_dir)
-
 
 @pytest.mark.parametrize("get_identifier", [(lambda d: d), (lambda d: "Test Name")])
-def test_pull__workflow_deployment_updates_existing_deployment(vellum_client, get_identifier):
+def test_pull__workflow_deployment_updates_existing_deployment(
+    vellum_client, get_identifier, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path
+):
     """
     This test is to ensure that the deployment info is updated in the config
     when the user runs the pull command with the workflow deployment
@@ -1126,12 +1098,11 @@ def test_pull__workflow_deployment_updates_existing_deployment(vellum_client, ge
     deployment_name = "Test Name"
 
     # AND an existing configuration with this deployment
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setenv("VELLUM_API_KEY", "abcdef123456")
 
     # Create initial config with a deployment
-    vellum_lock_json = os.path.join(temp_dir, "vellum.lock.json")
+    vellum_lock_json = tmp_path / "vellum.lock.json"
     with open(vellum_lock_json, "w") as f:
         json.dump(
             {
@@ -1165,7 +1136,7 @@ def test_pull__workflow_deployment_updates_existing_deployment(vellum_client, ge
     updated_label = "Updated Label"
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1198,10 +1169,10 @@ def test_pull__workflow_deployment_updates_existing_deployment(vellum_client, ge
         assert deployment["name"] == deployment_name
         assert deployment["label"] == updated_label
 
-    os.chdir(current_dir)
 
-
-def test_pull__workflow_deployment_with_name_and_id(vellum_client):
+def test_pull__workflow_deployment_with_name_and_id(
+    vellum_client, monkeypatch: pytest.MonkeyPatch, tmp_path: pathlib.Path
+):
     """
     This test is to ensure that pulling with id and name will not add a new deployment to the config
     """
@@ -1213,7 +1184,7 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
     # AND the workflow pull API call returns a zip file with metadata
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1230,9 +1201,8 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
     )
 
     # AND we are currently in a new directory
-    current_dir = os.getcwd()
-    temp_dir = tempfile.mkdtemp()
-    os.chdir(temp_dir)
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setenv("VELLUM_API_KEY", "abcdef123456")
 
     # WHEN the user runs the pull command with the workflow deployment
     runner = CliRunner()
@@ -1242,8 +1212,8 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
     assert result.exit_code == 0
 
     # AND the deployment is saved in the config
-    vellum_lock_json = os.path.join(temp_dir, "vellum.lock.json")
-    assert os.path.exists(vellum_lock_json)
+    vellum_lock_json = tmp_path / "vellum.lock.json"
+    assert vellum_lock_json.exists()
     with open(vellum_lock_json) as f:
         lock_data = json.loads(f.read())
         assert len(lock_data["workflows"]) == 1
@@ -1254,12 +1224,10 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
         assert deployment["label"] == deployment_label
         assert lock_data["workflows"][0]["workflow_sandbox_id"] == workflow_sandbox_id
 
-    os.chdir(current_dir)
-
     # AND pull with name will not add a new deployment to the config
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1274,7 +1242,6 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
         ]
     )
 
-    os.chdir(temp_dir)
     result = runner.invoke(cli_main, ["workflows", "pull", "--workflow-deployment", deployment_name])
     assert result.exit_code == 0
     with open(vellum_lock_json) as f:
@@ -1284,8 +1251,6 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
         assert lock_data["workflows"][0]["deployments"][0]["name"] == deployment_name
         assert lock_data["workflows"][0]["deployments"][0]["label"] == deployment_label
 
-    os.chdir(current_dir)
-
 
 def test_pull__workspace_option__uses_different_api_key(mock_module, vellum_client_class):
     # GIVEN a module and workflow_sandbox_id
@@ -1318,7 +1283,7 @@ MY_OTHER_VELLUM_API_KEY=aaabbbcccddd
 
     # AND the workflow pull API call returns a zip file
     vellum_client_class.return_value.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')"})]
+        [zip_file_map({"workflow.py": "print('hello')"})]
     )
 
     # WHEN calling `vellum pull` with --workspace
@@ -1343,6 +1308,7 @@ MY_OTHER_VELLUM_API_KEY=aaabbbcccddd
     vellum_client_class.assert_called_once_with(
         api_key="aaabbbcccddd",
         environment=mock.ANY,
+        api_version=None,
     )
 
     # AND the vellum lock file should have been updated with the correct workspace
@@ -1356,3 +1322,33 @@ MY_OTHER_VELLUM_API_KEY=aaabbbcccddd
     assert os.path.exists(workflow_py)
     with open(workflow_py) as f:
         assert f.read() == "print('hello')"
+
+
+@pytest.mark.parametrize(
+    "base_command",
+    [
+        ["pull"],
+        ["workflows", "pull"],
+    ],
+    ids=["pull", "workflows_pull"],
+)
+def test_pull__release_tag(vellum_client, mock_module, base_command):
+    """Tests that the --release-tag option is passed to the API."""
+
+    # GIVEN a module on the user's filesystem
+    module = mock_module.module
+
+    # AND the workflow pull API call returns a zip file
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
+
+    # WHEN the user runs the pull command with --release-tag
+    runner = CliRunner()
+    result = runner.invoke(cli_main, base_command + [module, "--release-tag", "my-release-tag"])
+
+    # THEN the command returns successfully
+    assert result.exit_code == 0
+
+    # AND the pull api is called with release_tag="my-release-tag"
+    vellum_client.workflows.pull.assert_called_once()
+    call_args = vellum_client.workflows.pull.call_args.kwargs
+    assert call_args["release_tag"] == "my-release-tag"