vellum-ai 0.14.41__py3-none-any.whl → 0.14.42__py3-none-any.whl

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

@@ -425,6 +425,87 @@ def test_serialize_workflow():
                 "name": "CategoryConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow"],
             },
+            "ports": [
+                {
+                    "id": "3a45b81f-95e4-4cbd-8997-bfdbe30251e8",
+                    "name": "category_question",
+                    "type": "IF",
+                    "expression": {
+                        "type": "BINARY_EXPRESSION",
+                        "lhs": {"type": "WORKFLOW_INPUT", "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
+                        "operator": "=",
+                        "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "question"}},
+                    },
+                },
+                {
+                    "id": "7202f702-1ebc-4067-ab1e-ec67e49158ee",
+                    "name": "category_complaint",
+                    "type": "ELIF",
+                    "expression": {
+                        "type": "BINARY_EXPRESSION",
+                        "lhs": {"type": "WORKFLOW_INPUT", "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
+                        "operator": "=",
+                        "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "complaint"}},
+                    },
+                },
+                {
+                    "id": "cf45705d-1a47-43a6-9d24-a7fdf78baae0",
+                    "name": "category_compliment",
+                    "type": "ELIF",
+                    "expression": {
+                        "type": "BINARY_EXPRESSION",
+                        "lhs": {"type": "WORKFLOW_INPUT", "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf"},
+                        "operator": "=",
+                        "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "compliment"}},
+                    },
+                },
+                {
+                    "id": "f04610dd-61cf-41b0-b337-2235e101cdb0",
+                    "name": "category_statement",
+                    "type": "ELIF",
+                    "expression": {
+                        "type": "BINARY_EXPRESSION",
+                        "lhs": {
+                            "type": "BINARY_EXPRESSION",
+                            "lhs": {
+                                "type": "WORKFLOW_INPUT",
+                                "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf",
+                            },
+                            "operator": "=",
+                            "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "statement"}},
+                        },
+                        "operator": "and",
+                        "rhs": {
+                            "type": "BINARY_EXPRESSION",
+                            "lhs": {
+                                "type": "BINARY_EXPRESSION",
+                                "lhs": {
+                                    "type": "WORKFLOW_INPUT",
+                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf",
+                                },
+                                "operator": "=",
+                                "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "statement"}},
+                            },
+                            "operator": "and",
+                            "rhs": {
+                                "type": "BINARY_EXPRESSION",
+                                "lhs": {
+                                    "type": "WORKFLOW_INPUT",
+                                    "input_variable_id": "eece050a-432e-4a2c-8c87-9480397e4cbf",
+                                },
+                                "operator": "=",
+                                "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "statement"}},
+                            },
+                        },
+                    },
+                },
+                {
+                    "id": "f9dde637-ea90-465f-a871-caf8380ae377",
+                    "name": "category_fallthrough",
+                    "type": "ELSE",
+                    "expression": None,
+                },
+            ],
         },
         conditional_node,
         ignore_order=True,
@@ -873,6 +954,25 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
                 "name": "SimpleConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
+            "ports": [
+                {
+                    "id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
+                    "name": "text_str",
+                    "type": "IF",
+                    "expression": {
+                        "type": "BINARY_EXPRESSION",
+                        "lhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "123"}},
+                        "operator": f"{operator}",
+                        "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "123"}},
+                    },
+                },
+                {
+                    "id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
+                    "name": "text_fallthrough",
+                    "type": "ELSE",
+                    "expression": None,
+                },
+            ],
         },
         conditional_node,
         ignore_order=True,
@@ -966,6 +1066,24 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
                 "name": "SimpleConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
+            "ports": [
+                {
+                    "id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
+                    "name": "text_str",
+                    "type": "IF",
+                    "expression": {
+                        "type": "UNARY_EXPRESSION",
+                        "lhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "123"}},
+                        "operator": f"{operator}",
+                    },
+                },
+                {
+                    "id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
+                    "name": "text_fallthrough",
+                    "type": "ELSE",
+                    "expression": None,
+                },
+            ],
         },
         conditional_node,
         ignore_order=True,
@@ -1072,6 +1190,26 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
                 "name": "SimpleConditionalNode",
                 "module": ["tests", "workflows", "basic_conditional_node", "workflow_with_only_one_conditional_node"],
             },
+            "ports": [
+                {
+                    "id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
+                    "name": "text_str",
+                    "type": "IF",
+                    "expression": {
+                        "type": "TERNARY_EXPRESSION",
+                        "base": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "123"}},
+                        "operator": f"{operator}",
+                        "lhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "123"}},
+                        "rhs": {"type": "CONSTANT_VALUE", "value": {"type": "STRING", "value": "123"}},
+                    },
+                },
+                {
+                    "id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
+                    "name": "text_fallthrough",
+                    "type": "ELSE",
+                    "expression": None,
+                },
+            ],
         },
         conditional_node,
         ignore_order=True,

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

@@ -118,6 +118,7 @@ def test_serialize_workflow():
             "module": ["tests", "workflows", "basic_guardrail_node", "workflow"],
             "name": "ExampleGuardrailNode",
         },
+        "ports": [{"id": "0ed87407-697e-4ae9-ab9b-6c5cc2e57cf7", "name": "default", "type": "DEFAULT"}],
     }
 
     final_output_node = workflow_raw_data["nodes"][2]

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

@@ -299,6 +299,7 @@ def test_serialize_workflow():
                 "name": "ExampleInlineSubworkflowNode",
                 "module": ["tests", "workflows", "basic_inline_subworkflow", "workflow"],
             },
+            "ports": [{"id": "cfd831bc-ee7f-44d0-8d76-0ba0cd0277dc", "name": "default", "type": "DEFAULT"}],
         },
         subworkflow_node,
         ignore_order=True,

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

@@ -254,7 +254,7 @@ def test_serialize_workflow():
                         "extensions": {"color": None},
                     },
                     {
-                        "id": "d6fc6c7a-235f-4b98-86f3-e258d1198f93",
+                        "id": "0acef1e7-caa1-4d9f-bb69-6981729af18d",
                         "key": "items",
                         "type": "JSON",
                         "required": True,
@@ -264,7 +264,7 @@ def test_serialize_workflow():
                 ],
                 "output_variables": [{"id": "2a957315-fae0-4366-8a35-f0b315c5eade", "key": "count", "type": "NUMBER"}],
                 "concurrency": None,
-                "items_input_id": "d6fc6c7a-235f-4b98-86f3-e258d1198f93",
+                "items_input_id": "0acef1e7-caa1-4d9f-bb69-6981729af18d",
                 "item_input_id": "b29bb546-9bc8-4136-857d-8c7a464ba9d4",
                 "index_input_id": "17e7ca49-668f-450d-a792-e1f97d13db67",
             },
@@ -277,6 +277,7 @@ def test_serialize_workflow():
                 "name": "MapFruitsNode",
                 "module": ["tests", "workflows", "basic_map_node", "workflow"],
             },
+            "ports": [{"id": "a2171a61-0657-43ad-b6d9-cf93ce3270d0", "name": "default", "type": "DEFAULT"}],
         },
         map_node,
         ignore_order=True,

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

@@ -84,6 +84,7 @@ def test_serialize_workflow__await_all():
                 "module": ["tests", "workflows", "basic_merge_node", "await_all_workflow"],
                 "name": "AwaitAllMergeNode",
             },
+            "ports": [{"id": "3bbc469f-0fb0-4b3d-a28b-746fefec2818", "name": "default", "type": "DEFAULT"}],
         },
         merge_node,
         ignore_order_func=lambda x: x.path() == "root['data']['target_handles']",

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

@@ -157,6 +157,7 @@ def test_serialize_workflow(vellum_client):
             "module": ["tests", "workflows", "basic_text_prompt_deployment", "workflow"],
             "name": "ExamplePromptDeploymentNode",
         },
+        "ports": [{"id": "2f26c7e0-283d-4f04-b639-adebb56bc679", "name": "default", "type": "DEFAULT"}],
     }
 
     final_output_node = workflow_raw_data["nodes"][2]

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

@@ -239,6 +239,7 @@ def test_serialize_workflow():
             "name": "SimpleSearchNode",
             "module": ["tests", "workflows", "basic_search_node", "workflow"],
         },
+        "ports": [{"id": "00ae06b3-f8d9-4ae6-9fbf-e4ff4d520e9b", "name": "default", "type": "DEFAULT"}],
     }
 
     final_output_node = workflow_raw_data["nodes"][2]

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

@@ -154,6 +154,7 @@ def test_serialize_workflow(vellum_client):
             "module": ["tests", "workflows", "basic_subworkflow_deployment", "workflow"],
             "name": "ExampleSubworkflowDeploymentNode",
         },
+        "ports": [{"id": "ab0db8a9-7b53-4d88-8667-273b31303273", "name": "default", "type": "DEFAULT"}],
     }
 
     assert not DeepDiff(

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

@@ -113,6 +113,7 @@ def test_serialize_workflow():
                 "name": "ExampleTemplatingNode",
                 "module": ["tests", "workflows", "basic_templating_node", "workflow_with_json_input"],
             },
+            "ports": [{"id": "39317827-df43-4f5a-bfbc-20bffc839748", "name": "default", "type": "DEFAULT"}],
         },
         templating_node,
     )

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

@@ -68,7 +68,7 @@ def test_serialize_workflow():
             "label": "Basic Final Output Node",
             "name": "basic-final-output-node",
             "target_handle_id": "0173d3c6-11d1-44b7-b070-ca9ff5119046",
-            "output_id": "97349956-d228-4b51-a64b-1331f788373f",
+            "output_id": "aa63e3f6-fde3-4d19-84ef-29982d44d709",
             "output_type": "STRING",
             "node_input_id": "5322567a-f40c-400a-96b3-c3b054db543e",
         },
@@ -98,7 +98,7 @@ def test_serialize_workflow():
         },
         "outputs": [
             {
-                "id": "97349956-d228-4b51-a64b-1331f788373f",
+                "id": "aa63e3f6-fde3-4d19-84ef-29982d44d709",
                 "name": "value",
                 "type": "STRING",
                 "value": {"type": "WORKFLOW_INPUT", "input_variable_id": "e39a7b63-de15-490a-ae9b-8112c767aea0"},

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

@@ -92,7 +92,7 @@ def test_serialize_workflow__missing_final_output_node():
                     "label": "First Final Output Node",
                     "name": "first-final-output-node",
                     "target_handle_id": "a0c2eb7a-398e-4f28-b63d-f3bae9b563ee",
-                    "output_id": "5517e50d-7f40-4f7c-acb2-e329d79a25bf",
+                    "output_id": "0cd02933-c5b9-47c9-aede-e97c5870e8aa",
                     "output_type": "STRING",
                     "node_input_id": "16363762-c14a-4162-8fab-525079d3cffe",
                 },
@@ -122,7 +122,7 @@ def test_serialize_workflow__missing_final_output_node():
                 },
                 "outputs": [
                     {
-                        "id": "5517e50d-7f40-4f7c-acb2-e329d79a25bf",
+                        "id": "0cd02933-c5b9-47c9-aede-e97c5870e8aa",
                         "name": "value",
                         "type": "STRING",
                         "value": {

vellum/client/types/node_event_display_context.py

@@ -1,30 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class NodeEventDisplayContext(UniversalBaseModel):
-    input_display: typing.Dict[str, str]
-    output_display: typing.Dict[str, str]
-    port_display: typing.Dict[str, str]
-    subworkflow_display: typing.Optional["WorkflowEventDisplayContext"] = 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
-
-
-from .workflow_event_display_context import WorkflowEventDisplayContext  # noqa: E402
-
-update_forward_refs(WorkflowEventDisplayContext, NodeEventDisplayContext=NodeEventDisplayContext)
-update_forward_refs(NodeEventDisplayContext)

vellum/client/types/workflow_event_display_context.py

@@ -1,28 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class WorkflowEventDisplayContext(UniversalBaseModel):
-    node_displays: typing.Dict[str, "NodeEventDisplayContext"]
-    workflow_inputs: typing.Dict[str, str]
-    workflow_outputs: typing.Dict[str, str]
-
-    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
-
-
-from .node_event_display_context import NodeEventDisplayContext  # noqa: E402
-
-update_forward_refs(WorkflowEventDisplayContext)

vellum/types/workflow_event_display_context.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.workflow_event_display_context import *