vellum-ai 0.14.37__tar.gz → 0.14.39__tar.gz

{vellum_ai-0.14.37 → vellum_ai-0.14.39}/PKG-INFO

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

{vellum_ai-0.14.37 → vellum_ai-0.14.39}/ee/vellum_cli/push.py

@@ -36,9 +36,6 @@ def push_command(
     logger = load_cli_logger()
     config = load_vellum_cli_config()
 
-    if not config.workflows:
-        raise ValueError("No Workflows found in project to push.")
-
     workflow_configs = (
         [
             w
@@ -61,6 +58,8 @@ def push_command(
             )
             config.workflows.append(new_config)
             workflow_configs = [new_config]
+        elif not module:
+            raise ValueError("No Workflows found in project to push.")
         else:
             raise ValueError(f"No workflow config for '{module}' found in project to push.")
 

{vellum_ai-0.14.37 → vellum_ai-0.14.39}/ee/vellum_cli/tests/test_push.py

@@ -133,6 +133,58 @@ def test_push__happy_path(mock_module, vellum_client, base_command):
     assert extracted_files["workflow.py"] == workflow_py_file_content
 
 
+def test_push__no_config__module_found(mock_module, vellum_client):
+    # GIVEN no config file set
+    temp_dir = mock_module.temp_dir
+    module = mock_module.module
+    mock_module.set_pyproject_toml({"workflows": []})
+
+    # AND a workflow exists in the module successfully
+    workflow_py_file_content = _ensure_workflow_py(temp_dir, module)
+
+    # AND the push API call returns successfully
+    new_workflow_sandbox_id = str(uuid4())
+    vellum_client.workflows.push.return_value = WorkflowPushResponse(
+        workflow_sandbox_id=new_workflow_sandbox_id,
+    )
+
+    # WHEN calling `vellum push`
+    runner = CliRunner()
+    result = runner.invoke(cli_main, ["push", module])
+
+    # THEN it should successfully push the workflow
+    assert result.exit_code == 0, result.stdout
+
+    # Get the last part of the module path and format it
+    expected_artifact_name = f"{mock_module.module.replace('.', '__')}.tar.gz"
+
+    # AND we should have called the push API with the correct args
+    vellum_client.workflows.push.assert_called_once()
+    call_args = vellum_client.workflows.push.call_args.kwargs
+    assert json.loads(call_args["exec_config"])["workflow_raw_data"]["definition"]["name"] == "ExampleWorkflow"
+    assert call_args["workflow_sandbox_id"] is None
+    assert "deplyment_config" not in call_args
+
+    # AND we should have pushed the correct artifact
+    assert call_args["artifact"].name == expected_artifact_name
+    extracted_files = _extract_tar_gz(call_args["artifact"].read())
+    assert extracted_files["workflow.py"] == workflow_py_file_content
+
+    # AND there should be a new entry in the lock file
+    with open(os.path.join(temp_dir, "vellum.lock.json")) as f:
+        lock_file_content = json.load(f)
+        assert lock_file_content["workflows"][0] == {
+            "container_image_name": None,
+            "container_image_tag": None,
+            "deployments": [],
+            "ignore": None,
+            "module": module,
+            "target_directory": None,
+            "workflow_sandbox_id": new_workflow_sandbox_id,
+            "workspace": "default",
+        }
+
+
 @pytest.mark.parametrize(
     "base_command",
     [

{vellum_ai-0.14.37 → vellum_ai-0.14.39}/ee/vellum_ee/workflows/display/vellum.py

@@ -19,11 +19,6 @@ from vellum_ee.workflows.display.editor.types import NodeDisplayPosition  # noqa
 from vellum_ee.workflows.display.utils.vellum import NodeInputValuePointerRule
 
 
-class CodeResourceDefinition(UniversalBaseModel):
-    name: str
-    module: List[str]
-
-
 @dataclass
 class WorkflowMetaVellumDisplayOverrides(WorkflowMetaDisplayOverrides):
     """

{vellum_ai-0.14.37 → vellum_ai-0.14.39}/pyproject.toml

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

{vellum_ai-0.14.37 → vellum_ai-0.14.39}/src/vellum/__init__.py

@@ -443,6 +443,9 @@ from .types import (
     TestSuiteRunMetricNumberOutput,
     TestSuiteRunMetricOutput,
     TestSuiteRunMetricStringOutput,
+    TestSuiteRunProgress,
+    TestSuiteRunPromptSandboxExecConfigDataRequest,
+    TestSuiteRunPromptSandboxExecConfigRequest,
     TestSuiteRunPromptSandboxHistoryItemExecConfig,
     TestSuiteRunPromptSandboxHistoryItemExecConfigData,
     TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest,
@@ -454,6 +457,8 @@ from .types import (
     TestSuiteRunWorkflowReleaseTagExecConfigData,
     TestSuiteRunWorkflowReleaseTagExecConfigDataRequest,
     TestSuiteRunWorkflowReleaseTagExecConfigRequest,
+    TestSuiteRunWorkflowSandboxExecConfigDataRequest,
+    TestSuiteRunWorkflowSandboxExecConfigRequest,
     TestSuiteRunWorkflowSandboxHistoryItemExecConfig,
     TestSuiteRunWorkflowSandboxHistoryItemExecConfigData,
     TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest,
@@ -1068,6 +1073,9 @@ __all__ = [
     "TestSuiteRunMetricNumberOutput",
     "TestSuiteRunMetricOutput",
     "TestSuiteRunMetricStringOutput",
+    "TestSuiteRunProgress",
+    "TestSuiteRunPromptSandboxExecConfigDataRequest",
+    "TestSuiteRunPromptSandboxExecConfigRequest",
     "TestSuiteRunPromptSandboxHistoryItemExecConfig",
     "TestSuiteRunPromptSandboxHistoryItemExecConfigData",
     "TestSuiteRunPromptSandboxHistoryItemExecConfigDataRequest",
@@ -1079,6 +1087,8 @@ __all__ = [
     "TestSuiteRunWorkflowReleaseTagExecConfigData",
     "TestSuiteRunWorkflowReleaseTagExecConfigDataRequest",
     "TestSuiteRunWorkflowReleaseTagExecConfigRequest",
+    "TestSuiteRunWorkflowSandboxExecConfigDataRequest",
+    "TestSuiteRunWorkflowSandboxExecConfigRequest",
     "TestSuiteRunWorkflowSandboxHistoryItemExecConfig",
     "TestSuiteRunWorkflowSandboxHistoryItemExecConfigData",
     "TestSuiteRunWorkflowSandboxHistoryItemExecConfigDataRequest",

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