vellum-ai 0.14.32__tar.gz → 0.14.33__tar.gz

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/PKG-INFO

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

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/ee/vellum_cli/ping.py

@@ -14,7 +14,12 @@ def ping_command():
     try:
         workspace = client.workspaces.workspace_identity()
         organization = client.organizations.organization_identity()
-    except ApiError:
+    except ApiError as e:
+        # If user did not provide an API key, we will get a 403 error
+        if e.status_code == 401 or e.status_code == 403:
+            raise Exception(
+                "Please make sure your `VELLUM_API_KEY` environment variable is set correctly."  # noqa: E501
+            ) from e
         raise Exception(
             "The API we tried to ping returned an invalid response. Please make sure your `VELLUM_API_URL` environment variable is set correctly."  # noqa: E501
         )

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/ee/vellum_cli/tests/test_ping.py

@@ -2,6 +2,7 @@ from datetime import datetime
 
 from click.testing import CliRunner
 
+from vellum.client.core.api_error import ApiError
 from vellum.client.types.organization_read import OrganizationRead
 from vellum.client.types.workspace_read import WorkspaceRead
 from vellum_cli import main as cli_main
@@ -51,9 +52,6 @@ def test_ping__error_path(vellum_client):
     # GIVEN a cli
     runner = CliRunner()
 
-    # Mock the workspace_identity method to raise ApiError directly
-    from vellum.client.core.api_error import ApiError
-
     vellum_client.workspaces.workspace_identity.side_effect = ApiError(status_code=400, body="Invalid JSON")
 
     # WHEN the user runs the ping command
@@ -65,3 +63,20 @@ def test_ping__error_path(vellum_client):
         str(result.exception)
         == "The API we tried to ping returned an invalid response. Please make sure your `VELLUM_API_URL` environment variable is set correctly."  # noqa: E501
     )
+
+
+def test_ping__unauthorized_error_path(vellum_client):
+    # GIVEN a cli
+    runner = CliRunner()
+
+    # GIVEN an unauthorized error with the error message from the API
+    vellum_client.workspaces.workspace_identity.side_effect = ApiError(
+        status_code=403, body={"detail": "Authentication credentials were not provided."}
+    )
+
+    # WHEN the user runs the ping command
+    result = runner.invoke(cli_main, ["ping"])
+
+    # 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."

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/ee/vellum_ee/workflows/display/nodes/vellum/guardrail_node.py

@@ -1,5 +1,5 @@
 from uuid import UUID
-from typing import ClassVar, Dict, Generic, Optional, TypeVar
+from typing import Generic, Optional, TypeVar
 
 from vellum.workflows.nodes import GuardrailNode
 from vellum.workflows.types.core import JsonObject
@@ -12,8 +12,6 @@ _GuardrailNodeType = TypeVar("_GuardrailNodeType", bound=GuardrailNode)
 
 
 class BaseGuardrailNodeDisplay(BaseNodeVellumDisplay[_GuardrailNodeType], Generic[_GuardrailNodeType]):
-    metric_input_ids_by_name: ClassVar[Dict[str, UUID]] = {}
-
     def serialize(
         self, display_context: WorkflowDisplayContext, error_output_id: Optional[UUID] = None, **kwargs
     ) -> JsonObject:
@@ -27,7 +25,8 @@ class BaseGuardrailNodeDisplay(BaseNodeVellumDisplay[_GuardrailNodeType], Generi
                 input_name=variable_name,
                 value=variable_value,
                 display_context=display_context,
-                input_id=self.metric_input_ids_by_name.get(variable_name),
+                input_id=self.node_input_ids_by_name.get(f"{GuardrailNode.metric_inputs.name}.{variable_name}")
+                or self.node_input_ids_by_name.get(variable_name),
             )
             for variable_name, variable_value in metric_inputs.items()
         ]

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py

@@ -7,7 +7,7 @@ from vellum import WorkspaceSecretRead
 from vellum_ee.workflows.display.workflows import VellumWorkflowDisplay
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
-from tests.workflows.basic_vellum_api_node.workflow import SimpleAPIWorkflow
+from tests.workflows.basic_api_node.workflow import SimpleAPIWorkflow
 
 
 def test_serialize_workflow(vellum_client):
@@ -99,7 +99,7 @@ def test_serialize_workflow(vellum_client):
                     "id": "c3a17ceb-e201-4025-b18a-9162aac7705e",
                     "key": "body",
                     "value": {
-                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "JSON", "value": None}}],
+                        "rules": [{"type": "CONSTANT_VALUE", "data": {"type": "JSON", "value": {"key": "value"}}}],
                         "combinator": "OR",
                     },
                 },
@@ -193,7 +193,7 @@ def test_serialize_workflow(vellum_client):
             },
             "definition": {
                 "name": "SimpleAPINode",
-                "module": ["tests", "workflows", "basic_vellum_api_node", "workflow"],
+                "module": ["tests", "workflows", "basic_api_node", "workflow"],
             },
         },
         api_node,
@@ -373,7 +373,7 @@ def test_serialize_workflow(vellum_client):
         "module": [
             "tests",
             "workflows",
-            "basic_vellum_api_node",
+            "basic_api_node",
             "workflow",
         ],
     }

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/ee/vellum_ee/workflows/display/tests/workflow_serialization/test_basic_guardrail_node_serialization.py

@@ -73,7 +73,7 @@ def test_serialize_workflow():
         "type": "METRIC",
         "inputs": [
             {
-                "id": "3ec00ee3-c068-4d41-9488-87b7778a649e",
+                "id": "e5e9757e-27db-4200-a650-2f7bb2cd61f5",
                 "key": "expected",
                 "value": {
                     "rules": [
@@ -86,7 +86,7 @@ def test_serialize_workflow():
                 },
             },
             {
-                "id": "a43ce7ba-1685-4977-a34d-65580c42853f",
+                "id": "f5cd9336-e8e9-429e-b7bc-785402e31bfb",
                 "key": "actual",
                 "value": {
                     "rules": [

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/pyproject.toml

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

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

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/src/vellum/workflows/nodes/core/inline_subworkflow_node/node.py

@@ -75,7 +75,9 @@ class InlineSubworkflowNode(
         with execution_context(parent_context=get_parent_context()):
             subworkflow = self.subworkflow(
                 parent_state=self.state,
-                context=WorkflowContext(vellum_client=self._context.vellum_client),
+                context=WorkflowContext(
+                    vellum_client=self._context.vellum_client, generated_files=self._context.generated_files
+                ),
             )
             subworkflow_stream = subworkflow.stream(
                 inputs=self._compile_subworkflow_inputs(),

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/src/vellum/workflows/nodes/core/map_node/node.py

@@ -171,7 +171,9 @@ class MapNode(BaseAdornmentNode[StateType], Generic[StateType, MapNodeItemType])
             self._run_subworkflow(item=item, index=index)
 
     def _run_subworkflow(self, *, item: MapNodeItemType, index: int) -> None:
-        context = WorkflowContext(vellum_client=self._context.vellum_client)
+        context = WorkflowContext(
+            vellum_client=self._context.vellum_client, generated_files=self._context.generated_files
+        )
         subworkflow = self.subworkflow(
             parent_state=self.state,
             context=context,

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/src/vellum/workflows/nodes/core/retry_node/node.py

@@ -45,7 +45,9 @@ class RetryNode(BaseAdornmentNode[StateType], Generic[StateType]):
             with execution_context(parent_context=parent_context):
                 subworkflow = self.subworkflow(
                     parent_state=self.state,
-                    context=WorkflowContext(vellum_client=self._context.vellum_client),
+                    context=WorkflowContext(
+                        vellum_client=self._context.vellum_client, generated_files=self._context.generated_files
+                    ),
                 )
                 subworkflow_stream = subworkflow.stream(
                     inputs=self.SubworkflowInputs(attempt_number=attempt_number),

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/src/vellum/workflows/nodes/core/try_node/node.py

@@ -32,7 +32,9 @@ class TryNode(BaseAdornmentNode[StateType], Generic[StateType]):
         with execution_context(parent_context=parent_context):
             subworkflow = self.subworkflow(
                 parent_state=self.state,
-                context=WorkflowContext(vellum_client=self._context.vellum_client),
+                context=WorkflowContext(
+                    vellum_client=self._context.vellum_client, generated_files=self._context.generated_files
+                ),
             )
             subworkflow_stream = subworkflow.stream(
                 event_filter=all_workflow_event_filter,

{vellum_ai-0.14.32 → vellum_ai-0.14.33}/src/vellum/workflows/nodes/displayable/guardrail_node/node.py

@@ -24,7 +24,7 @@ class GuardrailNode(BaseNode[StateType], Generic[StateType]):
 
     metric_definition: ClassVar[Union[UUID, str]]
 
-    metric_inputs: ClassVar[EntityInputsInterface]
+    metric_inputs: ClassVar[EntityInputsInterface] = {}
     release_tag: str = LATEST_RELEASE_TAG
 
     request_options: Optional[RequestOptions] = None