vellum-ai 0.14.15__py3-none-any.whl → 0.14.16__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (17) hide show
  1. vellum/client/core/client_wrapper.py +1 -1
  2. vellum/client/resources/document_indexes/client.py +0 -55
  3. vellum/client/types/document_index_read.py +0 -10
  4. vellum/plugins/pydantic.py +14 -4
  5. vellum/workflows/vellum_client.py +9 -5
  6. {vellum_ai-0.14.15.dist-info → vellum_ai-0.14.16.dist-info}/METADATA +1 -1
  7. {vellum_ai-0.14.15.dist-info → vellum_ai-0.14.16.dist-info}/RECORD +17 -16
  8. vellum_cli/image_push.py +76 -42
  9. vellum_cli/tests/test_image_push.py +56 -0
  10. vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_adornments_serialization.py +29 -33
  11. vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_attributes_serialization.py +91 -106
  12. vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_outputs_serialization.py +33 -38
  13. vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_ports_serialization.py +138 -153
  14. vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_trigger_serialization.py +23 -26
  15. {vellum_ai-0.14.15.dist-info → vellum_ai-0.14.16.dist-info}/LICENSE +0 -0
  16. {vellum_ai-0.14.15.dist-info → vellum_ai-0.14.16.dist-info}/WHEEL +0 -0
  17. {vellum_ai-0.14.15.dist-info → vellum_ai-0.14.16.dist-info}/entry_points.txt +0 -0
vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_trigger_serialization.py CHANGED
@@ -9,16 +9,15 @@ class Inputs(BaseInputs):
9
9
  input: str
10
10
 
11
11
 
12
- class BasicGenericNode(BaseNode):
13
- pass
14
-
15
-
16
12
  def test_serialize_node__basic(serialize_node):
13
+ class BasicGenericNode(BaseNode):
14
+ pass
15
+
17
16
  serialized_node = serialize_node(BasicGenericNode)
18
17
  assert not DeepDiff(
19
18
  {
20
- "id": "c2ed23f7-f6cb-4a56-a91c-2e5f9d8fda7f",
21
- "label": "BasicGenericNode",
19
+ "id": "8d7cbfe4-72ca-4367-a401-8d28723d2f00",
20
+ "label": "test_serialize_node__basic.<locals>.BasicGenericNode",
22
21
  "type": "GENERIC",
23
22
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
24
23
  "base": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"]},
@@ -34,10 +33,10 @@ def test_serialize_node__basic(serialize_node):
34
33
  "test_trigger_serialization",
35
34
  ],
36
35
  },
37
- "trigger": {"id": "9d3a1b3d-4a38-4f2e-bbf1-dd8be152bce8", "merge_behavior": "AWAIT_ATTRIBUTES"},
36
+ "trigger": {"id": "be19c63b-3492-46b1-be9d-16f8d2e6410b", "merge_behavior": "AWAIT_ATTRIBUTES"},
38
37
  "ports": [
39
38
  {
40
- "id": "89dccfa5-cc1a-4612-bd87-86cb444f6dd4",
39
+ "id": "8bec8d0c-113f-4110-afcb-4a6e566e7236",
41
40
  "name": "default",
42
41
  "type": "DEFAULT",
43
42
  }
@@ -51,17 +50,16 @@ def test_serialize_node__basic(serialize_node):
51
50
  )
52
51
 
53
52
 
54
- class AwaitAnyGenericNode(BaseNode):
55
- class Trigger(BaseNode.Trigger):
56
- merge_behavior = MergeBehavior.AWAIT_ANY
57
-
58
-
59
53
  def test_serialize_node__await_any(serialize_node):
54
+ class AwaitAnyGenericNode(BaseNode):
55
+ class Trigger(BaseNode.Trigger):
56
+ merge_behavior = MergeBehavior.AWAIT_ANY
57
+
60
58
  serialized_node = serialize_node(AwaitAnyGenericNode)
61
59
  assert not DeepDiff(
62
60
  {
63
- "id": "0ba67f76-aaff-4bd4-a20f-73a32ef5810d",
64
- "label": "AwaitAnyGenericNode",
61
+ "id": "42e17f0e-8496-415f-9c72-f85250ba6f0b",
62
+ "label": "test_serialize_node__await_any.<locals>.AwaitAnyGenericNode",
65
63
  "type": "GENERIC",
66
64
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
67
65
  "base": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"]},
@@ -77,10 +75,10 @@ def test_serialize_node__await_any(serialize_node):
77
75
  "test_trigger_serialization",
78
76
  ],
79
77
  },
80
- "trigger": {"id": "ffa72187-9a18-453f-ae55-b77aad332630", "merge_behavior": "AWAIT_ANY"},
78
+ "trigger": {"id": "5bb6bb4c-4374-44c8-a7b5-7bb6c1060a5b", "merge_behavior": "AWAIT_ANY"},
81
79
  "ports": [
82
80
  {
83
- "id": "38b83138-cb07-40cb-82d2-83982ded6883",
81
+ "id": "d9a84db7-8bd6-4a15-9e3c-c2e898c26d16",
84
82
  "name": "default",
85
83
  "type": "DEFAULT",
86
84
  }
@@ -94,17 +92,16 @@ def test_serialize_node__await_any(serialize_node):
94
92
  )
95
93
 
96
94
 
97
- class AwaitAllGenericNode(BaseNode):
98
- class Trigger(BaseNode.Trigger):
99
- merge_behavior = MergeBehavior.AWAIT_ALL
100
-
101
-
102
95
  def test_serialize_node__await_all(serialize_node):
96
+ class AwaitAllGenericNode(BaseNode):
97
+ class Trigger(BaseNode.Trigger):
98
+ merge_behavior = MergeBehavior.AWAIT_ALL
99
+
103
100
  serialized_node = serialize_node(AwaitAllGenericNode)
104
101
  assert not DeepDiff(
105
102
  {
106
- "id": "09d06cd3-06ea-40cc-afd8-17ad88542271",
107
- "label": "AwaitAllGenericNode",
103
+ "id": "b3e1145a-5f41-456b-9382-6d0a1e828c2f",
104
+ "label": "test_serialize_node__await_all.<locals>.AwaitAllGenericNode",
108
105
  "type": "GENERIC",
109
106
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
110
107
  "base": {"name": "BaseNode", "module": ["vellum", "workflows", "nodes", "bases", "base"]},
@@ -120,10 +117,10 @@ def test_serialize_node__await_all(serialize_node):
120
117
  "test_trigger_serialization",
121
118
  ],
122
119
  },
123
- "trigger": {"id": "62074276-c817-476d-b59d-da523ae3f218", "merge_behavior": "AWAIT_ALL"},
120
+ "trigger": {"id": "124ba9cf-a30e-41ef-81bf-143708f8b1c3", "merge_behavior": "AWAIT_ALL"},
124
121
  "ports": [
125
122
  {
126
- "id": "9edef179-a1c2-4624-b185-593bb84f08a0",
123
+ "id": "fa73da35-0bf9-4f02-bf5b-0b0d1a6f1494",
127
124
  "name": "default",
128
125
  "type": "DEFAULT",
129
126
  }