vellum-ai 1.0.5__py3-none-any.whl → 1.0.6__py3-none-any.whl

vellum_cli/tests/test_push.py

@@ -13,6 +13,7 @@ from vellum.client.core.api_error import ApiError
 from vellum.client.types.workflow_push_response import WorkflowPushResponse
 from vellum.utils.uuid import is_valid_uuid
 from vellum_cli import main as cli_main
+from vellum_ee.workflows.display.nodes.utils import to_kebab_case
 
 
 def _extract_tar_gz(tar_gz_bytes: bytes) -> dict[str, str]:
@@ -391,7 +392,9 @@ def test_push__deployment(mock_module, vellum_client, base_command):
     assert json.loads(call_args["exec_config"])["workflow_raw_data"]["definition"]["name"] == "ExampleWorkflow"
     assert is_valid_uuid(call_args["workflow_sandbox_id"])
     assert call_args["artifact"].name == expected_artifact_name
-    assert call_args["deployment_config"] == "{}"
+    expected_deployment_name = to_kebab_case(module.split(".")[-1])
+    deployment_config = json.loads(call_args["deployment_config"])
+    assert deployment_config["name"] == expected_deployment_name
 
     extracted_files = _extract_tar_gz(call_args["artifact"].read())
     assert extracted_files["workflow.py"] == workflow_py_file_content
@@ -1039,3 +1042,56 @@ def test_push__deploy_with_release_tags_success(mock_module, vellum_client):
     # AND should show success message
     assert "Successfully pushed" in result.output
     assert "Updated vellum.lock.json file." in result.output
+
+
+def test_push__deploy_stores_deployment_config_in_lock_file(mock_module, vellum_client):
+    # GIVEN a single workflow
+    temp_dir = mock_module.temp_dir
+    module = mock_module.module
+
+    # AND a workflow exists in the module successfully
+    _ensure_workflow_py(temp_dir, module)
+
+    # AND the push API call returns successfully with a deployment
+    workflow_deployment_id = str(uuid4())
+    vellum_client.workflows.push.return_value = WorkflowPushResponse(
+        workflow_sandbox_id=str(uuid4()),
+        workflow_deployment_id=workflow_deployment_id,
+    )
+
+    # WHEN calling `vellum workflows push --deploy` for the first time
+    runner = CliRunner()
+    result = runner.invoke(cli_main, ["workflows", "push", module, "--deploy"])
+
+    # THEN it should succeed
+    assert result.exit_code == 0, result.output
+
+    # AND the deployment config should be stored in the lock file with the deployment ID and module name
+    with open(os.path.join(temp_dir, "vellum.lock.json")) as f:
+        lock_data = json.loads(f.read())
+        assert len(lock_data["workflows"][0]["deployments"]) == 1
+        deployment = lock_data["workflows"][0]["deployments"][0]
+        assert str(deployment["id"]) == workflow_deployment_id
+        assert deployment["name"] == "test-push-deploy-stores-deployment-config-in-lock-file"
+        assert deployment.get("label") is None
+        assert deployment.get("description") is None
+        assert deployment.get("release_tags") is None
+
+    # AND when we do a second push
+    vellum_client.workflows.push.reset_mock()
+    vellum_client.workflows.push.return_value = WorkflowPushResponse(
+        workflow_sandbox_id=str(uuid4()),
+        workflow_deployment_id=workflow_deployment_id,
+    )
+
+    result = runner.invoke(cli_main, ["workflows", "push", module, "--deploy"])
+
+    # THEN it should succeed
+    assert result.exit_code == 0, result.output
+
+    # AND we should have called the push API with the module name as deployment name
+    vellum_client.workflows.push.assert_called_once()
+    call_args = vellum_client.workflows.push.call_args.kwargs
+    deployment_config_str = call_args["deployment_config"]
+    deployment_config = json.loads(deployment_config_str)
+    assert deployment_config["name"] == "test-push-deploy-stores-deployment-config-in-lock-file"

vellum_ee/workflows/display/nodes/vellum/code_execution_node.py

@@ -40,6 +40,8 @@ class BaseCodeExecutionNodeDisplay(BaseNodeDisplay[_CodeExecutionNodeType], Gene
                 node_filepath=node_file_path,
                 script_filepath=filepath,
             )
+            if not file_code:
+                raise Exception(f"Filepath '{filepath}' of node {node.__name__} does not exist")
             code_value = file_code
         else:
             code_value = ""

vellum_ee/workflows/display/nodes/vellum/tests/test_code_execution_node.py

@@ -153,3 +153,19 @@ def test_serialize_node__with_unresolved_secret_references(vellum_client):
     # warnings = list(workflow_display.errors)
     # assert len(warnings) == 1
     # assert "Failed to resolve secret reference 'MY_API_KEY'" in str(warnings[0])
+
+
+def test_serialize_node__with_non_exist_code_input_path():
+    # GIVEN a code node with a non-existent code input path
+    class MyNode(CodeExecutionNode):
+        filepath = "non_existent_file.py"
+
+    # AND a workflow with the code node
+    class Workflow(BaseWorkflow):
+        graph = MyNode
+
+    # WHEN we serialize the workflow
+    workflow_display = get_workflow_display(workflow_class=Workflow)
+    with pytest.raises(Exception) as exc_info:
+        workflow_display.serialize()
+    assert "Filepath 'non_existent_file.py' of node MyNode does not exist" in str(exc_info.value)

vellum_ee/workflows/display/tests/workflow_serialization/test_basic_tool_calling_node_composio_serialization.py

@@ -54,6 +54,9 @@ def test_serialize_workflow():
         "action": "GITHUB_CREATE_AN_ISSUE",
         "description": "Create a new issue in a GitHub repository",
         "display_name": "Create GitHub Issue",
+        "parameters": None,
+        "version": None,
+        "tags": None,
     }
 
     # AND the rest of the node structure should be correct

vellum/client/types/name_enum.py

@@ -1,7 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-NameEnum = typing.Union[
-    typing.Literal["workflow_executions", "prompt_executions", "workflow_runtime_seconds"], typing.Any
-]

vellum/client/types/organization_limit_config.py

@@ -1,25 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .vembda_service_tier_enum import VembdaServiceTierEnum
-from .quota import Quota
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class OrganizationLimitConfig(UniversalBaseModel):
-    vembda_service_tier: typing.Optional[VembdaServiceTierEnum] = None
-    prompt_executions_quota: typing.Optional[Quota] = None
-    workflow_executions_quota: typing.Optional[Quota] = None
-    workflow_runtime_seconds_quota: typing.Optional[Quota] = None
-    max_workflow_runtime_seconds: typing.Optional[int] = 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/client/types/quota.py

@@ -1,22 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.pydantic_utilities import UniversalBaseModel
-from .name_enum import NameEnum
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class Quota(UniversalBaseModel):
-    name: NameEnum
-    value: typing.Optional[int] = None
-    period_seconds: typing.Optional[int] = 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/client/types/vembda_service_tier_enum.py

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

vellum/types/name_enum.py

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

vellum/types/organization_limit_config.py

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

vellum/types/quota.py

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

vellum/types/vembda_service_tier_enum.py

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

vellum/workflows/nodes/displayable/tool_calling_node/composio_service.py

@@ -1,83 +0,0 @@
-from dataclasses import dataclass
-from typing import Any, Dict, List
-
-from composio import Action, Composio
-from composio_client import Composio as ComposioClient
-
-
-@dataclass
-class ConnectionInfo:
-    """Information about a user's authorized connection"""
-
-    connection_id: str
-    integration_name: str
-    created_at: str
-    updated_at: str
-    status: str = "ACTIVE"  # TODO: Use enum if we end up supporting integrations that the user has not yet connected to
-
-
-class ComposioAccountService:
-    """Manages user authorized connections using composio-client"""
-
-    def __init__(self, api_key: str):
-        self.client = ComposioClient(api_key=api_key)
-
-    def get_user_connections(self) -> List[ConnectionInfo]:
-        """Get all authorized connections for the user"""
-        response = self.client.connected_accounts.list()
-
-        return [
-            ConnectionInfo(
-                connection_id=item.id,
-                integration_name=item.toolkit.slug,
-                status=item.status,
-                created_at=item.created_at,
-                updated_at=item.updated_at,
-            )
-            for item in response.items
-        ]
-
-
-class ComposioCoreService:
-    """Handles tool execution using composio-core"""
-
-    def __init__(self, api_key: str):
-        self.client = Composio(api_key=api_key)
-
-    def execute_tool(self, tool_name: str, arguments: Dict[str, Any]) -> Any:
-        """Execute a tool using composio-core
-
-        Args:
-            tool_name: The name of the tool to execute (e.g., "HACKERNEWS_GET_USER")
-            arguments: Dictionary of arguments to pass to the tool
-
-        Returns:
-            The result of the tool execution
-        """
-        # Convert tool name string to Action enum
-        action = getattr(Action, tool_name)
-        return self.client.actions.execute(action, params=arguments)
-
-
-class ComposioService:
-    """Unified interface for Composio operations"""
-
-    def __init__(self, api_key: str):
-        self.accounts = ComposioAccountService(api_key)
-        self.core = ComposioCoreService(api_key)
-
-    def get_user_connections(self) -> List[ConnectionInfo]:
-        """Get user's authorized connections"""
-        return self.accounts.get_user_connections()
-
-    def execute_tool(self, tool_name: str, arguments: Dict[str, Any]) -> Any:
-        """Execute a tool using composio-core
-
-        Args:
-            tool_name: The name of the tool to execute (e.g., "HACKERNEWS_GET_USER")
-            arguments: Dictionary of arguments to pass to the tool
-
-        Returns:
-            The result of the tool execution
-        """
-        return self.core.execute_tool(tool_name, arguments)