vellum-ai 0.13.19__py3-none-any.whl → 0.13.20__py3-none-any.whl

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

@@ -178,7 +178,7 @@ def test_serialize_workflow(vellum_client):
         "definition": None,
         "inputs": [
             {
-                "id": "5f0fdd0f-63fe-4437-bc1b-cd6a84bb84c5",
+                "id": "78aeb65b-3491-4d2a-8c47-401d4cb3d560",
                 "key": "node_input",
                 "value": {
                     "combinator": "OR",
@@ -197,7 +197,7 @@ def test_serialize_workflow(vellum_client):
         "data": {
             "label": "Final Output",
             "name": "text",
-            "node_input_id": "5f0fdd0f-63fe-4437-bc1b-cd6a84bb84c5",
+            "node_input_id": "78aeb65b-3491-4d2a-8c47-401d4cb3d560",
             "output_id": "a609ab19-db1b-4cd0-bdb0-aee5ed31dc28",
             "output_type": "STRING",
             "target_handle_id": "dced939a-9122-4290-8482-7daa9525dad6",

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

@@ -249,11 +249,11 @@ def test_serialize_workflow():
             "target_handle_id": "cd8c736f-1b77-493d-b857-d8feb5c03b15",
             "output_id": "27424f7d-9767-4059-bdcf-c2be8b798fd7",
             "output_type": "STRING",
-            "node_input_id": "008eaf1d-98c0-4098-8839-5b94121394d7",
+            "node_input_id": "39c7f674-a794-4525-8a04-f22a40ed0914",
         },
         "inputs": [
             {
-                "id": "008eaf1d-98c0-4098-8839-5b94121394d7",
+                "id": "39c7f674-a794-4525-8a04-f22a40ed0914",
                 "key": "node_input",
                 "value": {
                     "rules": [

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

@@ -181,11 +181,11 @@ def test_serialize_workflow(vellum_client):
                     "target_handle_id": "23117248-df28-4519-bebc-abcb24f966b3",
                     "output_id": "3f487916-126f-4d6c-95b4-fa72d875b793",
                     "output_type": "NUMBER",
-                    "node_input_id": "29e10bd7-eb39-4e0f-a406-c3a55b834d6f",
+                    "node_input_id": "6f4955d8-8a3f-4db7-8293-7affc5877dcd",
                 },
                 "inputs": [
                     {
-                        "id": "29e10bd7-eb39-4e0f-a406-c3a55b834d6f",
+                        "id": "6f4955d8-8a3f-4db7-8293-7affc5877dcd",
                         "key": "node_input",
                         "value": {
                             "rules": [
@@ -224,11 +224,11 @@ def test_serialize_workflow(vellum_client):
                     "target_handle_id": "c3aeba92-4faf-4814-9842-eec7436ee555",
                     "output_id": "45d53a1e-26e8-4c43-a010-80d141acc249",
                     "output_type": "STRING",
-                    "node_input_id": "daaf0664-0aca-4619-af30-07caa1486b90",
+                    "node_input_id": "74337307-3fcb-42c5-9aed-98bd4a79caef",
                 },
                 "inputs": [
                     {
-                        "id": "daaf0664-0aca-4619-af30-07caa1486b90",
+                        "id": "74337307-3fcb-42c5-9aed-98bd4a79caef",
                         "key": "node_input",
                         "value": {
                             "rules": [

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

@@ -130,11 +130,11 @@ def test_serialize_workflow():
                 "target_handle_id": "16ba108e-61a8-4338-8a5b-4f1278d7fd7b",
                 "output_id": "62ec9b08-6437-4f8d-bc20-983d317bc348",
                 "output_type": "JSON",
-                "node_input_id": "310072db-6d07-4d29-8e5c-9c6909300925",
+                "node_input_id": "da51bb91-227c-481c-84d7-69c64322b485",
             },
             "inputs": [
                 {
-                    "id": "310072db-6d07-4d29-8e5c-9c6909300925",
+                    "id": "da51bb91-227c-481c-84d7-69c64322b485",
                     "key": "node_input",
                     "value": {
                         "rules": [

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

@@ -132,11 +132,11 @@ def test_serialize_workflow__missing_final_output_node():
                     "target_handle_id": "5e337b19-cef6-45af-802b-46da4ad7e794",
                     "output_id": "5e6d3ea6-ef91-4937-8fff-f33e07446e6a",
                     "output_type": "STRING",
-                    "node_input_id": "47ba0ee9-4725-4065-a178-c929ac556be9",
+                    "node_input_id": "590161f1-20ed-4339-9ce3-61aade3a142a",
                 },
                 "inputs": [
                     {
-                        "id": "47ba0ee9-4725-4065-a178-c929ac556be9",
+                        "id": "590161f1-20ed-4339-9ce3-61aade3a142a",
                         "key": "node_input",
                         "value": {
                             "rules": [

vellum_ee/workflows/display/utils/vellum.py

@@ -85,8 +85,6 @@ def create_node_input_value_pointer_rule(
         workflow_input_display = display_context.global_workflow_input_displays[value]
         return InputVariablePointer(data=InputVariableData(input_variable_id=str(workflow_input_display.id)))
     if isinstance(value, VellumSecretReference):
-        # TODO: Pass through the name instead of retrieving the ID
-        # https://app.shortcut.com/vellum/story/5072
         vellum_client = create_vellum_client()
         workspace_secret = vellum_client.workspace_secrets.retrieve(
             id=value.name,

vellum_ee/workflows/display/vellum.py

@@ -113,10 +113,8 @@ class WorkflowOutputVellumDisplayOverrides(WorkflowOutputDisplay, WorkflowOutput
     name: str
     label: str
     node_id: UUID
-    node_input_id: UUID
     target_handle_id: UUID
     display_data: NodeDisplayData
-    edge_id: UUID
 
 
 @dataclass

vellum_ee/workflows/display/workflows/vellum_workflow_display.py

@@ -135,7 +135,6 @@ class VellumWorkflowDisplay(
                     # This is currently the wrapper node's output, but we want the wrapped node
                     workflow_output.instance,
                     self.display_context,
-                    workflow_output_display.node_input_id,
                 )
 
                 source_node_display: Optional[BaseNodeDisplay]
@@ -182,7 +181,7 @@ class VellumWorkflowDisplay(
 
                     synthetic_output_edges.append(
                         {
-                            "id": str(workflow_output_display.edge_id),
+                            "id": str(uuid4_from_hash(f"{self.workflow_id}|edge_id|{workflow_output_display.name}")),
                             "source_node_id": str(source_node_display.node_id),
                             "source_handle_id": str(source_handle_id),
                             "target_node_id": str(workflow_output_display.node_id),
@@ -323,26 +322,20 @@ class VellumWorkflowDisplay(
                 name=overrides.name,
                 label=overrides.label,
                 node_id=overrides.node_id,
-                node_input_id=overrides.node_input_id,
                 target_handle_id=overrides.target_handle_id,
-                edge_id=overrides.edge_id,
                 display_data=overrides.display_data,
             )
 
         output_id = uuid4_from_hash(f"{self.workflow_id}|id|{output.name}")
-        edge_id = uuid4_from_hash(f"{self.workflow_id}|edge_id|{output.name}")
         node_id = uuid4_from_hash(f"{self.workflow_id}|node_id|{output.name}")
-        node_input_id = uuid4_from_hash(f"{self.workflow_id}|node_input_id|{output.name}")
         target_handle_id = uuid4_from_hash(f"{self.workflow_id}|target_handle_id|{output.name}")
 
         return WorkflowOutputVellumDisplay(
             id=output_id,
             node_id=node_id,
-            node_input_id=node_input_id,
             name=output.name,
             label="Final Output",
             target_handle_id=target_handle_id,
-            edge_id=edge_id,
             display_data=NodeDisplayData(),
         )
 

vellum_ee/workflows/tests/local_workflow/display/workflow.py

@@ -48,11 +48,9 @@ class WorkflowDisplay(VellumWorkflowDisplay[Workflow]):
         Workflow.Outputs.final_output: WorkflowOutputVellumDisplayOverrides(
             id=UUID("5469b810-6ea6-4362-9e79-e360d44a1405"),
             node_id=UUID("f3ef4b2b-fec9-4026-9cc6-e5eac295307f"),
-            node_input_id=UUID("fe6cba85-2423-4b5e-8f85-06311a8be5fb"),
             name="final-output",
             label="Final Output",
             target_handle_id=UUID("3ec34f6e-da48-40d5-a65b-a48fefa75763"),
             display_data=NodeDisplayData(position=NodeDisplayPosition(x=2750, y=210), width=459, height=234),
-            edge_id=UUID("417c56a4-cdc1-4f9d-a10c-b535163f51e8"),
         )
     }