vellum-ai 1.1.0__tar.gz → 1.1.1__tar.gz

{vellum_ai-1.1.0 → vellum_ai-1.1.1}/PKG-INFO

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

vellum_ai-1.1.1/ee/vellum_ee/workflows/display/exceptions.py

@@ -0,0 +1,7 @@
+class NodeValidationError(Exception):
+    """Exception raised when a node fails validation during workflow display serialization."""
+
+    def __init__(self, message: str, node_class_name: str):
+        self.message = message
+        self.node_class_name = node_class_name
+        super().__init__(f"Node validation error in {node_class_name}: {message}")

{vellum_ai-1.1.0 → vellum_ai-1.1.1}/ee/vellum_ee/workflows/display/nodes/vellum/code_execution_node.py

@@ -6,6 +6,7 @@ from vellum.workflows.nodes.displayable.code_execution_node import CodeExecution
 from vellum.workflows.nodes.displayable.code_execution_node.utils import read_file_from_path
 from vellum.workflows.types.core import JsonObject
 from vellum.workflows.utils.vellum_variables import primitive_type_to_vellum_variable_type
+from vellum_ee.workflows.display.exceptions import NodeValidationError
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.utils import raise_if_descriptor
 from vellum_ee.workflows.display.nodes.vellum.utils import create_node_input
@@ -41,7 +42,7 @@ class BaseCodeExecutionNodeDisplay(BaseNodeDisplay[_CodeExecutionNodeType], Gene
                 script_filepath=filepath,
             )
             if not file_code:
-                raise Exception(f"Filepath '{filepath}' of node {node.__name__} does not exist")
+                raise NodeValidationError(f"Filepath '{filepath}' does not exist", node.__name__)
             code_value = file_code
         else:
             code_value = ""

{vellum_ai-1.1.0 → vellum_ai-1.1.1}/ee/vellum_ee/workflows/display/nodes/vellum/tests/test_code_execution_node.py

@@ -6,6 +6,7 @@ from vellum.client.core.api_error import ApiError
 from vellum.workflows.nodes.displayable.code_execution_node.node import CodeExecutionNode
 from vellum.workflows.references.vellum_secret import VellumSecretReference
 from vellum.workflows.workflows.base import BaseWorkflow
+from vellum_ee.workflows.display.exceptions import NodeValidationError
 from vellum_ee.workflows.display.nodes.vellum.code_execution_node import BaseCodeExecutionNodeDisplay
 from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
 
@@ -166,6 +167,56 @@ def test_serialize_node__with_non_exist_code_input_path():
 
     # WHEN we serialize the workflow
     workflow_display = get_workflow_display(workflow_class=Workflow)
-    with pytest.raises(Exception) as exc_info:
+    with pytest.raises(NodeValidationError) as exc_info:
         workflow_display.serialize()
-    assert "Filepath 'non_existent_file.py' of node MyNode does not exist" in str(exc_info.value)
+    assert "Filepath 'non_existent_file.py' does not exist" in str(exc_info.value)
+
+
+def test_serialize_node__with_non_exist_code_input_path_with_dry_run():
+    # 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 with dry_run=True
+    workflow_display = get_workflow_display(workflow_class=Workflow, dry_run=True)
+    data: dict = workflow_display.serialize()
+
+    # THEN the workflow should not raise an error
+    assert data == {
+        "workflow_raw_data": {
+            "nodes": [
+                {
+                    "id": "9b9e2a5d-01a4-46b2-80a3-d9484b2c0e08",
+                    "type": "ENTRYPOINT",
+                    "inputs": [],
+                    "data": {"label": "Entrypoint Node", "source_handle_id": "3e2a3f52-5047-4e2e-9a21-37bd43c63250"},
+                    "display_data": {"position": {"x": 0.0, "y": -50.0}},
+                    "base": None,
+                    "definition": None,
+                }
+            ],
+            "edges": [
+                {
+                    "id": "ab6ef06e-df2c-4877-9c3e-9d7261b39748",
+                    "source_node_id": "9b9e2a5d-01a4-46b2-80a3-d9484b2c0e08",
+                    "source_handle_id": "3e2a3f52-5047-4e2e-9a21-37bd43c63250",
+                    "target_node_id": "ac90c0ce-f393-438c-a24f-e5e9a9286182",
+                    "target_handle_id": "3a39ea63-9f86-4891-a902-0216a7190720",
+                    "type": "DEFAULT",
+                }
+            ],
+            "display_data": {"viewport": {"x": 0.0, "y": 0.0, "zoom": 1.0}},
+            "definition": {
+                "name": "Workflow",
+                "module": ["vellum_ee", "workflows", "display", "nodes", "vellum", "tests", "test_code_execution_node"],
+            },
+            "output_values": [],
+        },
+        "input_variables": [],
+        "state_variables": [],
+        "output_variables": [],
+    }

{vellum_ai-1.1.0 → vellum_ai-1.1.1}/ee/vellum_ee/workflows/display/workflows/base_workflow_display.py

@@ -35,6 +35,7 @@ from vellum_ee.workflows.display.base import (
     WorkflowOutputDisplay,
 )
 from vellum_ee.workflows.display.editor.types import NodeDisplayData, NodeDisplayPosition
+from vellum_ee.workflows.display.exceptions import NodeValidationError
 from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
 from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
 from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplay
@@ -194,7 +195,7 @@ class BaseWorkflowDisplay(Generic[WorkflowType]):
 
             try:
                 serialized_node = node_display.serialize(self.display_context)
-            except NotImplementedError as e:
+            except (NotImplementedError, NodeValidationError) as e:
                 self.add_error(e)
                 continue
 

{vellum_ai-1.1.0 → vellum_ai-1.1.1}/pyproject.toml

@@ -3,7 +3,7 @@ name = "vellum-ai"
 
 [tool.poetry]
 name = "vellum-ai"
-version = "1.1.0"
+version = "1.1.1"
 description = ""
 readme = "README.md"
 authors = []
@@ -113,7 +113,7 @@ build-backend = "poetry.core.masonry.api"
 vellum = "vellum_cli:main"
 
 [tool.isort]
-skip = ["src/vellum/client", "tests/client", ".venv", "src/vellum/__init__.py", "examples/workflows"]
+skip = ["src/vellum/client", "tests/client", ".venv", "src/vellum/__init__.py", "examples/workflows", "src/vellum/client/raw_client.py"]
 multi_line_output = 3
 force_grid_wrap = 0
 line_length = 120
@@ -139,7 +139,7 @@ exclude = "(.git|venv|src/vellum/client|tests/client|scripts|examples/workflows)
 
 [tool.flake8]
 max-line-length = 120
-exclude = ".git,.venv,ee/codegen/node_modules,ee/codegen_integration/fixtures,scripts,src/vellum/__init__.py,src/vellum/client,src/vellum/core,src/vellum/environment.py,src/vellum/errors,src/vellum/resources,src/vellum/types,tests/client,examples/workflows"
+exclude = ".git,.venv,ee/codegen/node_modules,ee/codegen_integration/fixtures,scripts,src/vellum/__init__.py,src/vellum/client,src/vellum/core,src/vellum/environment.py,src/vellum/errors,src/vellum/raw_client.py,src/vellum/resources,src/vellum/types,tests/client,examples/workflows"
 ignore = "E203, W503, E704, U100, U101"
 
 [tool.coverage.run]

{vellum_ai-1.1.0 → vellum_ai-1.1.1}/src/vellum/client/README.md

@@ -95,61 +95,6 @@ except ApiError as e:
     print(e.body)
 ```
 
-## Streaming
-
-The SDK supports streaming responses, as well, the response will be a generator that you can loop over.
-
-```python
-from vellum import (
-    JinjaPromptBlock,
-    PromptParameters,
-    PromptRequestStringInput,
-    Vellum,
-    VellumVariable,
-)
-
-client = Vellum(
-    api_version="YOUR_API_VERSION",
-    api_key="YOUR_API_KEY",
-)
-response = client.ad_hoc.adhoc_execute_prompt_stream(
-    ml_model="x",
-    input_values=[
-        PromptRequestStringInput(
-            key="x",
-            value="value",
-        ),
-        PromptRequestStringInput(
-            key="x",
-            value="value",
-        ),
-    ],
-    input_variables=[
-        VellumVariable(
-            id="x",
-            key="key",
-            type="STRING",
-        ),
-        VellumVariable(
-            id="x",
-            key="key",
-            type="STRING",
-        ),
-    ],
-    parameters=PromptParameters(),
-    blocks=[
-        JinjaPromptBlock(
-            template="template",
-        ),
-        JinjaPromptBlock(
-            template="template",
-        ),
-    ],
-)
-for chunk in response:
-    yield chunk
-```
-
 ## Advanced
 
 ### Retries