vellum-ai 0.14.89__py3-none-any.whl → 1.0.0__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 (39) hide show
  1. vellum/__init__.py +1 -1
  2. vellum/client/__init__.py +2 -2
  3. vellum/client/core/client_wrapper.py +2 -2
  4. vellum/evaluations/resources.py +5 -5
  5. vellum/workflows/events/node.py +1 -1
  6. vellum/workflows/events/types.py +1 -1
  7. vellum/workflows/events/workflow.py +1 -1
  8. vellum/workflows/nodes/bases/tests/test_base_node.py +1 -1
  9. vellum/workflows/nodes/displayable/bases/base_prompt_node/node.py +1 -1
  10. vellum/workflows/nodes/displayable/bases/search_node.py +1 -1
  11. vellum/workflows/nodes/displayable/code_execution_node/node.py +1 -1
  12. vellum/workflows/nodes/displayable/guardrail_node/node.py +1 -1
  13. vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py +1 -1
  14. vellum/workflows/nodes/displayable/tool_calling_node/utils.py +83 -114
  15. vellum/workflows/nodes/experimental/__init__.py +2 -2
  16. vellum/workflows/state/base.py +1 -1
  17. vellum/workflows/state/delta.py +1 -1
  18. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.0.dist-info}/METADATA +1 -1
  19. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.0.dist-info}/RECORD +36 -39
  20. vellum_cli/config.py +1 -1
  21. vellum_cli/push.py +1 -1
  22. vellum_ee/workflows/display/base.py +0 -54
  23. vellum_ee/workflows/display/nodes/vellum/conditional_node.py +16 -2
  24. vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py +3 -3
  25. vellum_ee/workflows/display/tests/test_base_workflow_display.py +2 -4
  26. vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/conftest.py +2 -2
  27. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py +11 -11
  28. vellum_ee/workflows/display/vellum.py +2 -128
  29. vellum_ee/workflows/display/workflows/__init__.py +0 -1
  30. vellum_ee/workflows/display/workflows/base_workflow_display.py +1 -1
  31. vellum_ee/workflows/tests/local_workflow/display/nodes/final_output.py +1 -1
  32. vellum_ee/workflows/tests/local_workflow/display/nodes/templating_node.py +1 -1
  33. vellum_ee/workflows/tests/local_workflow/display/workflow.py +11 -14
  34. vellum/workflows/nodes/experimental/tool_calling_node/__init__.py +0 -3
  35. vellum/workflows/nodes/experimental/tool_calling_node/node.py +0 -3
  36. vellum_ee/workflows/display/workflows/vellum_workflow_display.py +0 -9
  37. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.0.dist-info}/LICENSE +0 -0
  38. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.0.dist-info}/WHEEL +0 -0
  39. {vellum_ai-0.14.89.dist-info → vellum_ai-1.0.0.dist-info}/entry_points.txt +0 -0
@@ -2,13 +2,13 @@ vellum_cli/CONTRIBUTING.md,sha256=FtDC7BGxSeMnwCXAUssFsAIElXtmJE-O5Z7BpolcgvI,29
2
2
  vellum_cli/README.md,sha256=2NudRoLzWxNKqnuVy1JuQ7DerIaxWGYkrH8kMd-asIE,90
3
3
  vellum_cli/__init__.py,sha256=50dYVkBzGUS1i-UKf_9zv1JMif1lLClriNUSs-Z7i4E,12678
4
4
  vellum_cli/aliased_group.py,sha256=ugW498j0yv4ALJ8vS9MsO7ctDW7Jlir9j6nE_uHAP8c,3363
5
- vellum_cli/config.py,sha256=v5BmZ-t_v4Jmqd7KVuQMZF2pRI-rbMspSkVYXIRoTmI,9448
5
+ vellum_cli/config.py,sha256=r92zfnBRbf7xrRSYv0XbV55wAKbgaiablWB8p47dcOE,9455
6
6
  vellum_cli/image_push.py,sha256=eeOBqKtKkPu6Kgm_jQCVCivogzAcdlIlkv61-QxH67c,11006
7
7
  vellum_cli/init.py,sha256=WpnMXPItPmh0f0bBGIer3p-e5gu8DUGwSArT_FuoMEw,5093
8
8
  vellum_cli/logger.py,sha256=dcM_OmgqXLo93vDYswO5ylyUQQcTfnA5GTd5tbIt3wM,1446
9
9
  vellum_cli/ping.py,sha256=p_BCCRjgPhng6JktuECtkDQLbhopt6JpmrtGoLnLJT8,1161
10
10
  vellum_cli/pull.py,sha256=udYyPlJ6VKDdh78rApNJOZgxHl82fcV6iGnRPSdX1LY,14750
11
- vellum_cli/push.py,sha256=sUo0d7lm7saGxQciH-JW2FB_gWfP2gwoEHO1wGnNQR0,10933
11
+ vellum_cli/push.py,sha256=3icrAKRu2ZbOvtVsk-m6pSM9Hau4MrmtMje-fKHMt8A,10940
12
12
  vellum_cli/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
13
13
  vellum_cli/tests/conftest.py,sha256=wx3PlJjVB0HRf5dr2b_idOIw27WPPl0J0FNbhIJJaVk,1689
14
14
  vellum_cli/tests/test_config.py,sha256=uvKGDc8BoVyT9_H0Z-g8469zVxomn6Oi3Zj-vK7O_wU,2631
@@ -23,7 +23,7 @@ vellum_ee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
23
23
  vellum_ee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
24
24
  vellum_ee/workflows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
25
25
  vellum_ee/workflows/display/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
26
- vellum_ee/workflows/display/base.py,sha256=EqlQFD56kpqMY02ZBJBQajzJKh33Dwi60Wo779kJlec,2880
26
+ vellum_ee/workflows/display/base.py,sha256=po1YIXOupglAFi1-_zsgNH-VwIjHfy5GCvii6Wzntu0,1878
27
27
  vellum_ee/workflows/display/editor/__init__.py,sha256=MSAgY91xCEg2neH5d8jXx5wRdR962ftZVa6vO9BGq9k,167
28
28
  vellum_ee/workflows/display/editor/types.py,sha256=x-tOOCJ6CF4HmiKDfCmcc3bOVfc1EBlP5o6u5WEfLoY,567
29
29
  vellum_ee/workflows/display/nodes/__init__.py,sha256=jI1aPBQf8DkmrYoZ4O-wR1duqZByOf5mDFmo_wFJPE4,307
@@ -37,7 +37,7 @@ vellum_ee/workflows/display/nodes/vellum/__init__.py,sha256=nUIgH2s0-7IbQRNrBhLP
37
37
  vellum_ee/workflows/display/nodes/vellum/api_node.py,sha256=5Uq8x08F64yrBcqbfsVeuoGnTa9eoOPumYzZZrDPmr0,8847
38
38
  vellum_ee/workflows/display/nodes/vellum/base_adornment_node.py,sha256=FHhPoGmmny4Xcxi2pm12Sk3ZOREanWEVrOWcjRhncH4,6337
39
39
  vellum_ee/workflows/display/nodes/vellum/code_execution_node.py,sha256=6lavdBw297GwAQqyxjnPUtx5pHv6k5V9Vkuq7s2D0TM,4508
40
- vellum_ee/workflows/display/nodes/vellum/conditional_node.py,sha256=9GtbvSJUNF626tCYxnMxETVZm3Fq84vOZ3Nkdkl3n-M,11146
40
+ vellum_ee/workflows/display/nodes/vellum/conditional_node.py,sha256=OEw8QRPliL4P8J6oEZdQH8Oc-0u7aFa_Jqx0RyL6F-M,11656
41
41
  vellum_ee/workflows/display/nodes/vellum/error_node.py,sha256=YhMsi2TG1zSR8E7IpxzzSncOyVLcvqTuGa3mr4RqHd8,2364
42
42
  vellum_ee/workflows/display/nodes/vellum/final_output_node.py,sha256=zo-nalsuayMqeb2GwR2OB9SFK3y2U5aG-rtwrsjdasQ,3089
43
43
  vellum_ee/workflows/display/nodes/vellum/guardrail_node.py,sha256=IniO5KvO0Rw9zghFg3RFvbXBTv6Zi1iuQhaA1DLazqU,2331
@@ -64,14 +64,14 @@ vellum_ee/workflows/display/nodes/vellum/tests/test_subworkflow_deployment_node.
64
64
  vellum_ee/workflows/display/nodes/vellum/tests/test_templating_node.py,sha256=LSk2gx9TpGXbAqKe8dggQW8yJZqj-Cf0EGJFeGGlEcw,3321
65
65
  vellum_ee/workflows/display/nodes/vellum/tests/test_tool_calling_node.py,sha256=bSMWCwD-niSx3MQPtgsLXjIf6zynQxZdpmeaOM6H3QA,4580
66
66
  vellum_ee/workflows/display/nodes/vellum/tests/test_try_node.py,sha256=Khjsb53PKpZuyhKoRMgKAL45eGp5hZqXvHmVeQWRw4w,2289
67
- vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py,sha256=3LS1O4DGPWit05oj_ubeW8AlHGnoBxdUMferGQuAiZs,4851
67
+ vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py,sha256=rHybfUAWwa0LlstQTNthGb-zYXrUCLADFtn_4SGsbw8,4807
68
68
  vellum_ee/workflows/display/nodes/vellum/try_node.py,sha256=z9Omo676RRc7mQjLoL7hjiHhUj0OWVLhrrb97YTN4QA,4086
69
69
  vellum_ee/workflows/display/nodes/vellum/utils.py,sha256=oICunzyaXPs0tYnW5zH1r93Bx35MSH7mcD-n0DEWRok,4978
70
70
  vellum_ee/workflows/display/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
71
- vellum_ee/workflows/display/tests/test_base_workflow_display.py,sha256=axUOLwpIhhzrk9d3rNmUaFbdfTaebfi1dzfRj9Up6Z4,10679
71
+ vellum_ee/workflows/display/tests/test_base_workflow_display.py,sha256=NTtZltrBqA0BlbDvCOJaBWABMmns74StnisA9icBZ8U,10602
72
72
  vellum_ee/workflows/display/tests/workflow_serialization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
73
73
  vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
74
- vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/conftest.py,sha256=XOQDDRiG46etxTC7-_RUEutoNumXc02fo7oho4GYM0c,1900
74
+ vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/conftest.py,sha256=Y-ajeT65b5varmrZCw6L3hir4hJCFq-eO0jZfRcrs7g,1886
75
75
  vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_adornments_serialization.py,sha256=aEmF6zpmmHdFgiG4V81EweOk5qO6cfukR782dYSE960,14060
76
76
  vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_attributes_serialization.py,sha256=DFNp_iUxlyE-zCJBH9ab3e1jQEK-NSE9M2CQd4NCjY8,24853
77
77
  vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_outputs_serialization.py,sha256=s6_mnk0pkztU59wYpSfOFpMhAJaRjmyfxM6WJGtnD4Y,6456
@@ -79,7 +79,7 @@ vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_port
79
79
  vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_trigger_serialization.py,sha256=dsJr8I9AdPwMOGszirfNDzZP2Ychd94aAKuPXAzknMk,4632
80
80
  vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py,sha256=Q3QOzzF7KE_9EukJ9yNFgYL7Qio_vTqDVd-6kc40wd4,15834
81
81
  vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py,sha256=B4DC2e5nqEk5NZeUwR4vKmBoEyToHkAuq4cCqbsidFo,29708
82
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py,sha256=cBHGuY-DeyqQlLbpRZrrQyDh7em8rPkB7bV9Vi0EFoE,53608
82
+ vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py,sha256=oKO8KGlkl505TSW5v0retJF4FS8TfwxxuAzqY9Adr8k,53608
83
83
  vellum_ee/workflows/display/tests/workflow_serialization/test_basic_default_state_serialization.py,sha256=E1ww97BFoGbnRkxf84TScat5NQhP8nLVrdaOlpGnSKU,8615
84
84
  vellum_ee/workflows/display/tests/workflow_serialization/test_basic_error_node_serialization.py,sha256=w1BYEZXIMjWmcKGQXzhelHnKFlEAXmsgjkI9fcgqKXA,5850
85
85
  vellum_ee/workflows/display/tests/workflow_serialization/test_basic_generic_node_serialization.py,sha256=-T0cd2jx1bC0ZNtAESF78fnYD_0nOqo8zMMLwRHUTRM,6227
@@ -108,12 +108,11 @@ vellum_ee/workflows/display/utils/registry.py,sha256=fWIm5Jj-10gNFjgn34iBu4RWv3V
108
108
  vellum_ee/workflows/display/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
109
109
  vellum_ee/workflows/display/utils/tests/test_auto_layout.py,sha256=vfXI769418s9vda5Gb5NFBH747WMOwSgHRXeLCTLVm8,2356
110
110
  vellum_ee/workflows/display/utils/vellum.py,sha256=mtoXmSYwR7rvrq-d6CzCW_auaJXTct0Mi1F0xpRCiNQ,5627
111
- vellum_ee/workflows/display/vellum.py,sha256=o7mq_vk2Yapu9DDKRz5l76h8EmCAypWGQYe6pryrbB8,3576
112
- vellum_ee/workflows/display/workflows/__init__.py,sha256=kapXsC67VJcgSuiBMa86FdePG5A9kMB5Pi4Uy1O2ob4,207
113
- vellum_ee/workflows/display/workflows/base_workflow_display.py,sha256=_gbMjULwEV6S7rLQKPNuuaD1LoTNf2WWTUkVqxQvG04,41153
111
+ vellum_ee/workflows/display/vellum.py,sha256=J2mdJZ1sdLW535DDUkq_Vm8Z572vhuxHxVZF9deKSdk,391
112
+ vellum_ee/workflows/display/workflows/__init__.py,sha256=JTB9ObEV3l4gGGdtfBHwVJtTTKC22uj-a-XjTVwXCyA,148
113
+ vellum_ee/workflows/display/workflows/base_workflow_display.py,sha256=5FApz_OAqdj9ljZ7xqln8zi9Y-GKGP810y2XcZmfC-g,41160
114
114
  vellum_ee/workflows/display/workflows/get_vellum_workflow_display_class.py,sha256=gxz76AeCqgAZ9D2lZeTiZzxY9eMgn3qOSfVgiqYcOh8,2028
115
115
  vellum_ee/workflows/display/workflows/tests/test_workflow_display.py,sha256=L7SKWJ26Ex-XXTNfHYXux7KP6I-dxE1EMQylap4Mhjs,31762
116
- vellum_ee/workflows/display/workflows/vellum_workflow_display.py,sha256=aaKdmWrgEe5YyV4zuDY_4E3y-l59rIHQnNGiPj2OWxQ,359
117
116
  vellum_ee/workflows/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
118
117
  vellum_ee/workflows/server/virtual_file_loader.py,sha256=7JphJcSO3H85qiC2DpFfBWjC3JjrbRmoynBC6KKHVsA,2710
119
118
  vellum_ee/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -126,9 +125,9 @@ vellum_ee/workflows/tests/local_files/inner_files/inner_class.py,sha256=RUN5k3os
126
125
  vellum_ee/workflows/tests/local_workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
127
126
  vellum_ee/workflows/tests/local_workflow/display/__init__.py,sha256=xo75Uqb4aErOsrgy9mJaQwlzLMPDUWIzoi3p-fYk90I,73
128
127
  vellum_ee/workflows/tests/local_workflow/display/nodes/__init__.py,sha256=szW_mgOUriyZ6v1vlnevBgkzNi8g83-ihS98UOLHVcE,155
129
- vellum_ee/workflows/tests/local_workflow/display/nodes/final_output.py,sha256=Kv92TCREiZsB9531KZYaBIq83uHn7e_ECw_yAbD1qfk,1017
130
- vellum_ee/workflows/tests/local_workflow/display/nodes/templating_node.py,sha256=5cankEe1rDZlXKgILFSPbmN0tUZhIdmcFgz_AguXTJc,1229
131
- vellum_ee/workflows/tests/local_workflow/display/workflow.py,sha256=ZqOwUFaS6vQAOWRsAq02cno5ULXFKsnXFwWPMfqH8uw,1996
128
+ vellum_ee/workflows/tests/local_workflow/display/nodes/final_output.py,sha256=Y5l1lgEdTuVakDXlLNV2x39d1GfgDE80w_UAbQHBMkw,1017
129
+ vellum_ee/workflows/tests/local_workflow/display/nodes/templating_node.py,sha256=QXJvdGzE1Ba2B_rUEGOYzUO_GfXBFII4rcu7TJyy8Bk,1229
130
+ vellum_ee/workflows/tests/local_workflow/display/workflow.py,sha256=rOI8MkfRYqX_X0t-rfSU--XI49cmCMiolApzbXLL0Ts,1835
132
131
  vellum_ee/workflows/tests/local_workflow/inputs.py,sha256=4cgsZBoUbIY0Rs8gknC9yqxQ-sSoULclx_SAm1FT2RA,96
133
132
  vellum_ee/workflows/tests/local_workflow/metadata.json,sha256=rdu3h5qkFZiqhCAMxoyoWyMI0O8QALC5-URvSIW6F00,24
134
133
  vellum_ee/workflows/tests/local_workflow/nodes/__init__.py,sha256=1F6jxUpSKfPXPj4ZZKSbnX6Mg-VwF3euLJSZfGn6xkM,127
@@ -139,12 +138,12 @@ vellum_ee/workflows/tests/test_display_meta.py,sha256=PkXJVnMZs9GNooDkd59n4YTBAX
139
138
  vellum_ee/workflows/tests/test_serialize_module.py,sha256=EVrCRAP0lpvd0GIDlg2tnGfJzDNooNDXPfGFPLAqmbI,1870
140
139
  vellum_ee/workflows/tests/test_server.py,sha256=SsOkS6sGO7uGC4mxvk4iv8AtcXs058P9hgFHzTWmpII,14519
141
140
  vellum_ee/workflows/tests/test_virtual_files.py,sha256=TJEcMR0v2S8CkloXNmCHA0QW0K6pYNGaIjraJz7sFvY,2762
142
- vellum/__init__.py,sha256=4RAS94IJNzJ0I4ywijtwTf9hPuwjHWjOs4ObAioCZF4,42908
141
+ vellum/__init__.py,sha256=S2zlpYoHKXOAP1vcibHB9BPZ2OST32AwXP8AJA0ErUA,42915
143
142
  vellum/client/README.md,sha256=Dle5iytCXxP1pNeNd7uZyhFo0rl7tp7vU7s8gmi10OQ,4863
144
- vellum/client/__init__.py,sha256=EQgVGB7vcK4XzQyTSTV0SHyaF8UlaeuSEtTAKgGv-aU,124904
143
+ vellum/client/__init__.py,sha256=yIMTLkqrMput0mnWPw7OQayz3mW7w478E9RmZsiTtWM,124904
145
144
  vellum/client/core/__init__.py,sha256=SQ85PF84B9MuKnBwHNHWemSGuy-g_515gFYNFhvEE0I,1438
146
145
  vellum/client/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
147
- vellum/client/core/client_wrapper.py,sha256=-g25WBC_j71FLmLSiIoBkcV7S40V_wCKP9Zk-1ZHXNw,2387
146
+ vellum/client/core/client_wrapper.py,sha256=XSSzA5VEGoD2dtUQf284uK6Ujy0yg-FTiuwHz7WdiXk,2383
148
147
  vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
149
148
  vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
150
149
  vellum/client/core/http_client.py,sha256=cKs2w0ybDBk1wHQf-fTALm_MmvaMe3cZKcYJxqmCxkE,19539
@@ -833,7 +832,7 @@ vellum/errors/not_found_error.py,sha256=gC71YBdPyHR46l3RNTs0v9taVvAY0gWRFrcKpKzb
833
832
  vellum/evaluations/__init__.py,sha256=hNsLoHSykqXDJP-MwFvu2lExImxo9KEyEJjt_fdAzpE,77
834
833
  vellum/evaluations/constants.py,sha256=Vteml4_csZsMgo_q3-71E3JRCAoN6308TXLu5nfLhmU,116
835
834
  vellum/evaluations/exceptions.py,sha256=6Xacoyv43fJvVf6Dt6Io5a-f9vF12Tx51jzsQRNSqhY,56
836
- vellum/evaluations/resources.py,sha256=7xS4eYeRLLz44KLMDL0wwIY80wMy2carFREB85k_Avk,12664
835
+ vellum/evaluations/resources.py,sha256=5JxDp8HO5FXEnzsUDcrn9nGEbMrsrCgDW0t_XOBMjiM,12671
837
836
  vellum/evaluations/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
838
837
  vellum/evaluations/utils/env.py,sha256=Xj_nxsoU5ox06EOTjRopR4lrigQI6Le6qbWGltYoEGU,276
839
838
  vellum/evaluations/utils/exceptions.py,sha256=dXMAkzqbHV_AP5FjjbegPlfUE0zQDlpA3qOsoOJUxfg,49
@@ -1533,12 +1532,12 @@ vellum/workflows/environment/environment.py,sha256=Ck3RPKXJvtMGx_toqYQQQF-ZwXm5i
1533
1532
  vellum/workflows/errors/__init__.py,sha256=tWGPu5xyAU8gRb8_bl0fL7OfU3wxQ9UH6qVwy4X4P_Q,113
1534
1533
  vellum/workflows/errors/types.py,sha256=nUWuniEfrhdtb-_2GzoDGlYnSJ_yuNUGjVkaKLNr-rM,4049
1535
1534
  vellum/workflows/events/__init__.py,sha256=V4mh766fyA70WvHelm9kfVZGrUgEKcJ9tJt8EepfQYU,832
1536
- vellum/workflows/events/node.py,sha256=TaawUxKaZG8cv_GkiKnJmjOmC4Ic0wMlsxUduF2Rbpw,5446
1535
+ vellum/workflows/events/node.py,sha256=n3JdlCmeZ6tFTu_x5tu1oSCe5EBE8B1Cy3CzSWK5B08,5453
1537
1536
  vellum/workflows/events/stream.py,sha256=xhXJTZirFi0xad5neAQNogrIQ4h47fpnKbVC3vCM5Js,889
1538
1537
  vellum/workflows/events/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1539
1538
  vellum/workflows/events/tests/test_event.py,sha256=elg5j3MOdWQe-JjshopqVwd716Jrvtr9P_V2JUGZC60,17866
1540
- vellum/workflows/events/types.py,sha256=LwgFlMRbptJvdPtPO1POUtGtbhGw7BSuvgHxNSgS7N8,4652
1541
- vellum/workflows/events/workflow.py,sha256=7rb1evKsTCRUdCnisAM5Anc19ZKGxnsmzwOOv6jwWEI,7761
1539
+ vellum/workflows/events/types.py,sha256=jc6vEcydzYETfK3u2Cx1BfUDmYh3Jxa86sdsrrQ5oBc,4659
1540
+ vellum/workflows/events/workflow.py,sha256=cR2aG6A-JqvLMKFqonDhKjSg496qldEtuks4BXXMTrY,7768
1542
1541
  vellum/workflows/exceptions.py,sha256=NiBiR3ggfmPxBVqD-H1SqmjI-7mIn0EStSN1BqApvCM,1213
1543
1542
  vellum/workflows/expressions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1544
1543
  vellum/workflows/expressions/accessor.py,sha256=4OvF9JQKHSyYLTz7gGprcw3Da5-zA8D0Co0j1eIeer0,4141
@@ -1590,7 +1589,7 @@ vellum/workflows/nodes/bases/base.py,sha256=OLnt140jfqRHxg4xyzdiBd3TciqNPfuNNH2i
1590
1589
  vellum/workflows/nodes/bases/base_adornment_node.py,sha256=hrgzuTetM4ynPd9YGHoK8Vwwn4XITi3aZZ_OCnQrq4Y,3433
1591
1590
  vellum/workflows/nodes/bases/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1592
1591
  vellum/workflows/nodes/bases/tests/test_base_adornment_node.py,sha256=fXZI9KqpS4XMBrBnIEkK3foHaBVvyHwYcQWWDKay7ic,1148
1593
- vellum/workflows/nodes/bases/tests/test_base_node.py,sha256=Qk54NIuNa-eAtkq2AVeyl1AjZbotWlqdxTXrlZ3pIe0,10935
1592
+ vellum/workflows/nodes/bases/tests/test_base_node.py,sha256=igJrcSxSZADk5tLpssAt0uAXe9oJoLbilzueJJvA5p4,10942
1594
1593
  vellum/workflows/nodes/core/__init__.py,sha256=5zDMCmyt1v0HTJzlUBwq3U9L825yZGZhT9JL18-mRR4,455
1595
1594
  vellum/workflows/nodes/core/error_node/__init__.py,sha256=g7RRnlHhqu4qByfLjBwCunmgGA8dI5gNsjS3h6TwlSI,60
1596
1595
  vellum/workflows/nodes/core/error_node/node.py,sha256=MFHU5vITYSK-L9CuMZ49In2ZeNLWnhZD0f8r5dWvb5Y,1270
@@ -1622,20 +1621,20 @@ vellum/workflows/nodes/displayable/bases/__init__.py,sha256=0mWIx3qUrzllV7jqt7wN
1622
1621
  vellum/workflows/nodes/displayable/bases/api_node/__init__.py,sha256=1jwx4WC358CLA1jgzl_UD-rZmdMm2v9Mps39ndwCD7U,64
1623
1622
  vellum/workflows/nodes/displayable/bases/api_node/node.py,sha256=4rRrF8IV9ATMm4i5mxjwArYmvSrlDyDxBMWQTrhY3p8,4606
1624
1623
  vellum/workflows/nodes/displayable/bases/base_prompt_node/__init__.py,sha256=Org3xTvgp1pA0uUXFfnJr29D3HzCey2lEdYF4zbIUgo,70
1625
- vellum/workflows/nodes/displayable/bases/base_prompt_node/node.py,sha256=lGAWxTf02rRBgyxAFJNGldRX6sCRAZUDiUllg6zogHY,4459
1624
+ vellum/workflows/nodes/displayable/bases/base_prompt_node/node.py,sha256=EJsGaz8Umss6-JWGGYbJp93ZHx3IlZQWAySlHAdNYtY,4466
1626
1625
  vellum/workflows/nodes/displayable/bases/inline_prompt_node/__init__.py,sha256=Hl35IAoepRpE-j4cALaXVJIYTYOF3qszyVbxTj4kS1s,82
1627
1626
  vellum/workflows/nodes/displayable/bases/inline_prompt_node/constants.py,sha256=fnjiRWLoRlC4Puo5oQcpZD5Hd-EesxsAo9l5tGAkpZQ,270
1628
1627
  vellum/workflows/nodes/displayable/bases/inline_prompt_node/node.py,sha256=6zzgeBTUwCiYTLRhrZAqmi-nU2r4q0gX-ytnatY5kwY,12896
1629
1628
  vellum/workflows/nodes/displayable/bases/inline_prompt_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1630
1629
  vellum/workflows/nodes/displayable/bases/inline_prompt_node/tests/test_inline_prompt_node.py,sha256=H1xEaL3vgEPAGQaF_rt8kLl46qmVJEEhcORXT7oRdPk,22626
1631
1630
  vellum/workflows/nodes/displayable/bases/prompt_deployment_node.py,sha256=0a40fkkZkFMmZN0CsWf6EP_y1H6x36EGa3WcfVNyOsM,9797
1632
- vellum/workflows/nodes/displayable/bases/search_node.py,sha256=-BxQtuPgq8NZfUf5XJgHNQTni5kGbvAHq7WFqd0Euck,6107
1631
+ vellum/workflows/nodes/displayable/bases/search_node.py,sha256=9TtFn6oNpEkpCL59QdBViUe4WPjcITajbiS7EOjOGag,6114
1633
1632
  vellum/workflows/nodes/displayable/bases/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1634
1633
  vellum/workflows/nodes/displayable/bases/tests/test_utils.py,sha256=eqdqbKNRWVMDPevgwLg1i6YK0g4L4bCy-7xCBN5yYZI,3156
1635
1634
  vellum/workflows/nodes/displayable/bases/types.py,sha256=C37B2Qh2YP7s7pUjd-EYKc2Zl1TbnCgI_mENuUSb8bo,1706
1636
1635
  vellum/workflows/nodes/displayable/bases/utils.py,sha256=7L5wdQVm0ZMLpTJ7IBbTOzPhany7wSYPLYuY-0FBhvg,5490
1637
1636
  vellum/workflows/nodes/displayable/code_execution_node/__init__.py,sha256=0FLWMMktpzSnmBMizQglBpcPrP80fzVsoJwJgf822Cg,76
1638
- vellum/workflows/nodes/displayable/code_execution_node/node.py,sha256=Qh4SPafzdRKbVKQH3h9SgH4vKeQQjvHAfP8ttUxLA2M,10304
1637
+ vellum/workflows/nodes/displayable/code_execution_node/node.py,sha256=cCpZPna1Exw060KlI8j3oMCfru1GCzUFJz9ucWdcZVM,10311
1639
1638
  vellum/workflows/nodes/displayable/code_execution_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1640
1639
  vellum/workflows/nodes/displayable/code_execution_node/tests/fixtures/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1641
1640
  vellum/workflows/nodes/displayable/code_execution_node/tests/fixtures/main.py,sha256=5QsbmkzSlSbcbWTG_JmIqcP-JNJzOPTKxGzdHos19W4,79
@@ -1649,7 +1648,7 @@ vellum/workflows/nodes/displayable/final_output_node/node.py,sha256=AfonKXEoT5kp
1649
1648
  vellum/workflows/nodes/displayable/final_output_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1650
1649
  vellum/workflows/nodes/displayable/final_output_node/tests/test_node.py,sha256=E6LQ74qZjY4Xi4avx2qdOCgGhF8pEcNLBh8cqYRkzMI,709
1651
1650
  vellum/workflows/nodes/displayable/guardrail_node/__init__.py,sha256=Ab5eXmOoBhyV4dMWdzh32HLUmnPIBEK_zFCT38C4Fng,68
1652
- vellum/workflows/nodes/displayable/guardrail_node/node.py,sha256=oBqQ0eAhALkGL64aAqEKP5lmQxvgYMJ2BeDD8cnLJE8,5813
1651
+ vellum/workflows/nodes/displayable/guardrail_node/node.py,sha256=GRWRDCVfPbL03hGbyZP4FAUtAjg_8K3XrU0xJ_lIFeA,5820
1653
1652
  vellum/workflows/nodes/displayable/guardrail_node/test_node.py,sha256=SAGv6hSFcBwQkudn1VxtaKNsXSXWWELl3eK05zM6tS0,5410
1654
1653
  vellum/workflows/nodes/displayable/guardrail_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1655
1654
  vellum/workflows/nodes/displayable/guardrail_node/tests/test_node.py,sha256=X2pd6TI8miYxIa7rgvs1pHTEreyWcf77EyR0_Jsa700,2055
@@ -1670,7 +1669,7 @@ vellum/workflows/nodes/displayable/search_node/node.py,sha256=1dGCB1kb7MvX3fUJ5z
1670
1669
  vellum/workflows/nodes/displayable/search_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1671
1670
  vellum/workflows/nodes/displayable/search_node/tests/test_node.py,sha256=WVZR3BI_CvxBG9hulv0-tcAc_gW5ozs0nH4uVNRJa2U,8863
1672
1671
  vellum/workflows/nodes/displayable/subworkflow_deployment_node/__init__.py,sha256=9yYM6001YZeqI1VOk1QuEM_yrffk_EdsO7qaPzINKds,92
1673
- vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py,sha256=hMyCIe9PoxAu5TLxalgnIc7na4LEKM9UAugd5ReXZnk,10026
1672
+ vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py,sha256=965eN_w54Q9xirRXrz5CUcg5sFA6AcrSs_IgUXVBCvk,10033
1674
1673
  vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1675
1674
  vellum/workflows/nodes/displayable/subworkflow_deployment_node/tests/test_node.py,sha256=c98nMPogZ6iN_pTvVUMTB3J72Hj--H-XVgvvRXhdSQE,19085
1676
1675
  vellum/workflows/nodes/displayable/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1684,13 +1683,11 @@ vellum/workflows/nodes/displayable/tool_calling_node/state.py,sha256=oQg_GAtc349
1684
1683
  vellum/workflows/nodes/displayable/tool_calling_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1685
1684
  vellum/workflows/nodes/displayable/tool_calling_node/tests/test_node.py,sha256=3cCCIz-quneCVl3w8XnGxzyHx-nn_jGTKCbSH9jkYT0,7824
1686
1685
  vellum/workflows/nodes/displayable/tool_calling_node/tests/test_utils.py,sha256=eu6WTyENhGLg9pGp_j69rysZjf_qiQXske1YdZn9PzU,1718
1687
- vellum/workflows/nodes/displayable/tool_calling_node/utils.py,sha256=xdPJ35GvGG26oMKtZZ_KcpaEr7Moz0kEBQRIzCZZXbA,13901
1686
+ vellum/workflows/nodes/displayable/tool_calling_node/utils.py,sha256=K4ouSeqMEBr6X9Df-qxlJ6wTmFBsYGwbiRtjaibuVlI,12528
1688
1687
  vellum/workflows/nodes/experimental/README.md,sha256=eF6DfIL8t-HbF9-mcofOMymKrraiBHDLKTlnBa51ZiE,284
1689
- vellum/workflows/nodes/experimental/__init__.py,sha256=k7VQEyvgEdnrEZ-icXx3fiByPnyMOnMXNGGuqacyyik,91
1688
+ vellum/workflows/nodes/experimental/__init__.py,sha256=jCQgvZEknXKfuNhGSOou4XPfrPqZ1_XBj5F0n0fgiWM,106
1690
1689
  vellum/workflows/nodes/experimental/openai_chat_completion_node/__init__.py,sha256=lsyD9laR9p7kx5-BXGH2gUTM242UhKy8SMV0SR6S2iE,90
1691
1690
  vellum/workflows/nodes/experimental/openai_chat_completion_node/node.py,sha256=cKI2Ls25L-JVt4z4a2ozQa-YBeVy21Z7BQ32Sj7iBPE,10460
1692
- vellum/workflows/nodes/experimental/tool_calling_node/__init__.py,sha256=8R5HQtDDfo0hgFX6VdM45-FiyxK-IMTI9B9LL4TRUbw,112
1693
- vellum/workflows/nodes/experimental/tool_calling_node/node.py,sha256=8R5HQtDDfo0hgFX6VdM45-FiyxK-IMTI9B9LL4TRUbw,112
1694
1691
  vellum/workflows/nodes/mocks.py,sha256=a1FjWEIocseMfjzM-i8DNozpUsaW0IONRpZmXBoWlyc,10455
1695
1692
  vellum/workflows/nodes/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1696
1693
  vellum/workflows/nodes/tests/test_mocks.py,sha256=mfPvrs75PKcsNsbJLQAN6PDFoVqs9TmQxpdyFKDdO60,7837
@@ -1721,9 +1718,9 @@ vellum/workflows/runner/__init__.py,sha256=i1iG5sAhtpdsrlvwgH6B-m49JsINkiWyPWs8v
1721
1718
  vellum/workflows/runner/runner.py,sha256=t9Y3k82YjeML-Z7f1W2sj_1nAjJEpsnJUComDBebagU,36339
1722
1719
  vellum/workflows/sandbox.py,sha256=GVJzVjMuYzOBnSrboB0_6MMRZWBluAyQ2o7syeaeBd0,2235
1723
1720
  vellum/workflows/state/__init__.py,sha256=yUUdR-_Vl7UiixNDYQZ-GEM_kJI9dnOia75TtuNEsnE,60
1724
- vellum/workflows/state/base.py,sha256=F53r9n1tnipe6a7BmI4qcqgBRDkYzuEsN_mb-65Dazw,24276
1721
+ vellum/workflows/state/base.py,sha256=m9fCqbZn21GshCVCjJTD1dPZEQjFrsMXqlg7tM9fIwM,24283
1725
1722
  vellum/workflows/state/context.py,sha256=KOAI1wEGn8dGmhmAemJaf4SZbitP3jpIBcwKfznQaRE,3076
1726
- vellum/workflows/state/delta.py,sha256=7p0jJ3wquaFPGktZZz1vLM1ydv4CI7kuYumIaNbiSF4,433
1723
+ vellum/workflows/state/delta.py,sha256=7h8wR10lRCm15SykaPj-gSEvvsMjCwYLPsOx3nsvBQg,440
1727
1724
  vellum/workflows/state/encoder.py,sha256=sV80h-lAo66isD2ezK1_YLzj75Aa5dAtF2YA1c3BYKo,2883
1728
1725
  vellum/workflows/state/store.py,sha256=uVe-oN73KwGV6M6YLhwZMMUQhzTQomsVfVnb8V91gVo,1147
1729
1726
  vellum/workflows/state/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1759,8 +1756,8 @@ vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnad
1759
1756
  vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
1760
1757
  vellum/workflows/workflows/tests/test_base_workflow.py,sha256=ptMntHzVyy8ZuzNgeTuk7hREgKQ5UBdgq8VJFSGaW4Y,20832
1761
1758
  vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
1762
- vellum_ai-0.14.89.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
1763
- vellum_ai-0.14.89.dist-info/METADATA,sha256=PUlFnfoQJiH5Jo_qs4triR8bNZHxQvEY_0pgGgM1jQA,5556
1764
- vellum_ai-0.14.89.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
1765
- vellum_ai-0.14.89.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
1766
- vellum_ai-0.14.89.dist-info/RECORD,,
1759
+ vellum_ai-1.0.0.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
1760
+ vellum_ai-1.0.0.dist-info/METADATA,sha256=AbpfGbPaocNLUgsmqS-axWjsiIWqYAhiKX7RyA1ZXgo,5554
1761
+ vellum_ai-1.0.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
1762
+ vellum_ai-1.0.0.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
1763
+ vellum_ai-1.0.0.dist-info/RECORD,,
vellum_cli/config.py CHANGED
@@ -7,7 +7,7 @@ from typing import Dict, List, Literal, Optional, Union
7
7
 
8
8
  import tomli
9
9
 
10
- from vellum.core.pydantic_utilities import UniversalBaseModel
10
+ from vellum.client.core.pydantic_utilities import UniversalBaseModel
11
11
  from vellum.workflows.state.encoder import DefaultStateEncoder
12
12
 
13
13
  LOCKFILE_PATH = "vellum.lock.json"
vellum_cli/push.py CHANGED
@@ -11,8 +11,8 @@ from dotenv import load_dotenv
11
11
  from pydantic import ValidationError
12
12
 
13
13
  from vellum.client.core.api_error import ApiError
14
+ from vellum.client.types import WorkflowPushDeploymentConfigRequest
14
15
  from vellum.resources.workflows.client import OMIT
15
- from vellum.types import WorkflowPushDeploymentConfigRequest
16
16
  from vellum.workflows.vellum_client import create_vellum_client
17
17
  from vellum_cli.config import DEFAULT_WORKSPACE_CONFIG, WorkflowConfig, WorkflowDeploymentConfig, load_vellum_cli_config
18
18
  from vellum_cli.logger import handle_cli_error, load_cli_logger
@@ -39,15 +39,6 @@ class WorkflowMetaDisplay:
39
39
  )
40
40
 
41
41
 
42
- @dataclass
43
- class WorkflowMetaDisplayOverrides(WorkflowMetaDisplay):
44
- """
45
- DEPRECATED: Use WorkflowMetaDisplay instead. Will be removed in 0.15.0
46
- """
47
-
48
- pass
49
-
50
-
51
42
  @dataclass
52
43
  class WorkflowInputsDisplay:
53
44
  id: UUID
@@ -55,15 +46,6 @@ class WorkflowInputsDisplay:
55
46
  color: Optional[str] = None
56
47
 
57
48
 
58
- @dataclass
59
- class WorkflowInputsDisplayOverrides(WorkflowInputsDisplay):
60
- """
61
- DEPRECATED: Use WorkflowInputsDisplay instead. Will be removed in 0.15.0
62
- """
63
-
64
- pass
65
-
66
-
67
49
  @dataclass
68
50
  class StateValueDisplay:
69
51
  id: UUID
@@ -71,54 +53,18 @@ class StateValueDisplay:
71
53
  color: Optional[str] = None
72
54
 
73
55
 
74
- @dataclass
75
- class StateValueDisplayOverrides(StateValueDisplay):
76
- """
77
- DEPRECATED: Use StateValueDisplay instead. Will be removed in 0.15.0
78
- """
79
-
80
- pass
81
-
82
-
83
56
  @dataclass
84
57
  class EdgeDisplay:
85
58
  id: UUID
86
59
 
87
60
 
88
- @dataclass
89
- class EdgeDisplayOverrides(EdgeDisplay):
90
- """
91
- DEPRECATED: Use EdgeDisplay instead. Will be removed in 0.15.0
92
- """
93
-
94
- pass
95
-
96
-
97
61
  @dataclass
98
62
  class EntrypointDisplay:
99
63
  id: UUID
100
64
  edge_display: EdgeDisplay
101
65
 
102
66
 
103
- @dataclass
104
- class EntrypointDisplayOverrides(EntrypointDisplay):
105
- """
106
- DEPRECATED: Use EntrypointDisplay instead. Will be removed in 0.15.0
107
- """
108
-
109
- pass
110
-
111
-
112
67
  @dataclass
113
68
  class WorkflowOutputDisplay:
114
69
  id: UUID
115
70
  name: str
116
-
117
-
118
- @dataclass
119
- class WorkflowOutputDisplayOverrides(WorkflowOutputDisplay):
120
- """
121
- DEPRECATED: Use WorkflowOutputDisplay instead. Will be removed in 0.15.0
122
- """
123
-
124
- pass
@@ -1,6 +1,6 @@
1
1
  from dataclasses import dataclass
2
2
  from uuid import UUID
3
- from typing import Any, ClassVar, Dict, Generic, List, Optional, Tuple, TypeVar, Union
3
+ from typing import Any, ClassVar, Dict, Generic, List, Optional, Tuple, TypeVar, Union, cast
4
4
 
5
5
  from vellum.workflows.descriptors.base import BaseDescriptor
6
6
  from vellum.workflows.expressions.and_ import AndExpression
@@ -61,6 +61,9 @@ class BaseConditionalNodeDisplay(BaseNodeDisplay[_ConditionalNodeType], Generic[
61
61
  but the defined conditions have length {len(condition_ids)}"""
62
62
  )
63
63
 
64
+ ports = self.serialize_ports(display_context)
65
+ ports_by_index = {idx: port for idx, port in enumerate(ports)}
66
+
64
67
  def serialize_rule(
65
68
  descriptor: BaseDescriptor, path: List[int], rule_id_map: Optional[RuleIdMap]
66
69
  ) -> Dict[str, Any]:
@@ -166,7 +169,7 @@ but the defined conditions have length {len(condition_ids)}"""
166
169
  rule_group_id = str(
167
170
  condition_ids[idx].rule_group_id if condition_ids else uuid4_from_hash(f"{condition_id}|rule_group")
168
171
  )
169
- source_handle_id = str(source_handle_ids.get(idx) or uuid4_from_hash(f"{node_id}|handles|{idx}"))
172
+ source_handle_id = self._get_source_handle_id(idx, source_handle_ids, ports_by_index)
170
173
 
171
174
  if port._condition is None:
172
175
  if port._condition_type == ConditionType.ELSE:
@@ -258,3 +261,14 @@ but the defined conditions have length {len(condition_ids)}"""
258
261
 
259
262
  def _get_condition_ids(self) -> List[ConditionId]:
260
263
  return self._get_explicit_node_display_attr("condition_ids", List[ConditionId]) or []
264
+
265
+ def _get_source_handle_id(
266
+ self, idx: int, source_handle_ids: Dict[int, UUID], ports_by_index: Dict[int, Any]
267
+ ) -> str:
268
+ if source_handle_ids.get(idx):
269
+ return str(source_handle_ids[idx])
270
+
271
+ if idx in ports_by_index:
272
+ return cast(str, ports_by_index[idx]["id"])
273
+
274
+ return str(uuid4_from_hash(f"{self.node_id}|source_handle|{idx}"))
@@ -8,6 +8,7 @@ from vellum.workflows.inputs import BaseInputs
8
8
  from vellum.workflows.nodes.bases import BaseNode
9
9
  from vellum.workflows.outputs import BaseOutputs
10
10
  from vellum.workflows.references import LazyReference
11
+ from vellum_ee.workflows.display.base import WorkflowInputsDisplay, WorkflowMetaDisplay
11
12
  from vellum_ee.workflows.display.editor.types import NodeDisplayData
12
13
  from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
13
14
  from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay
@@ -21,7 +22,6 @@ from vellum_ee.workflows.display.utils.vellum import (
21
22
  NodeOutputData,
22
23
  NodeOutputPointer,
23
24
  )
24
- from vellum_ee.workflows.display.vellum import WorkflowInputsVellumDisplayOverrides, WorkflowMetaVellumDisplay
25
25
  from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
26
26
 
27
27
 
@@ -110,13 +110,13 @@ def test_create_node_input_value_pointer_rules(
110
110
  descriptor,
111
111
  WorkflowDisplayContext(
112
112
  workflow_display_class=BaseWorkflowDisplay,
113
- workflow_display=WorkflowMetaVellumDisplay(
113
+ workflow_display=WorkflowMetaDisplay(
114
114
  entrypoint_node_id=uuid4(),
115
115
  entrypoint_node_source_handle_id=uuid4(),
116
116
  entrypoint_node_display=NodeDisplayData(),
117
117
  ),
118
118
  global_workflow_input_displays={
119
- Inputs.example_workflow_input: WorkflowInputsVellumDisplayOverrides(
119
+ Inputs.example_workflow_input: WorkflowInputsDisplay(
120
120
  id=UUID("a154c29d-fac0-4cd0-ba88-bc52034f5470"),
121
121
  ),
122
122
  },
@@ -7,7 +7,7 @@ from vellum.workflows.ports.port import Port
7
7
  from vellum.workflows.references.lazy import LazyReference
8
8
  from vellum.workflows.state import BaseState
9
9
  from vellum.workflows.workflows.base import BaseWorkflow
10
- from vellum_ee.workflows.display.vellum import WorkflowInputsVellumDisplayOverrides
10
+ from vellum_ee.workflows.display.base import WorkflowInputsDisplay
11
11
  from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
12
12
  from vellum_ee.workflows.display.workflows.get_vellum_workflow_display_class import get_workflow_display
13
13
 
@@ -64,9 +64,7 @@ def test_vellum_workflow_display__serialize_input_variables_with_capitalized_var
64
64
 
65
65
  class ExampleWorkflowDisplay(BaseWorkflowDisplay[ExampleWorkflow]):
66
66
  inputs_display = {
67
- Inputs.foo: WorkflowInputsVellumDisplayOverrides(
68
- id=UUID("97b63d71-5413-417f-9cf5-49e1b4fd56e4"), name="Foo", required=True
69
- )
67
+ Inputs.foo: WorkflowInputsDisplay(id=UUID("97b63d71-5413-417f-9cf5-49e1b4fd56e4"), name="Foo")
70
68
  }
71
69
 
72
70
  display = get_workflow_display(
@@ -4,6 +4,7 @@ from typing import Type
4
4
 
5
5
  from vellum.workflows.types.core import JsonObject
6
6
  from vellum.workflows.types.generics import NodeType
7
+ from vellum_ee.workflows.display.base import WorkflowMetaDisplay
7
8
  from vellum_ee.workflows.display.editor.types import NodeDisplayData
8
9
  from vellum_ee.workflows.display.nodes.get_node_display_class import get_node_display_class
9
10
  from vellum_ee.workflows.display.types import (
@@ -13,7 +14,6 @@ from vellum_ee.workflows.display.types import (
13
14
  WorkflowDisplayContext,
14
15
  WorkflowInputsDisplays,
15
16
  )
16
- from vellum_ee.workflows.display.vellum import WorkflowMetaVellumDisplay
17
17
  from vellum_ee.workflows.display.workflows.base_workflow_display import BaseWorkflowDisplay
18
18
 
19
19
 
@@ -31,7 +31,7 @@ def serialize_node():
31
31
 
32
32
  context: WorkflowDisplayContext = WorkflowDisplayContext(
33
33
  workflow_display_class=BaseWorkflowDisplay,
34
- workflow_display=WorkflowMetaVellumDisplay(
34
+ workflow_display=WorkflowMetaDisplay(
35
35
  entrypoint_node_id=uuid4(),
36
36
  entrypoint_node_source_handle_id=uuid4(),
37
37
  entrypoint_node_display=NodeDisplayData(),
@@ -285,7 +285,7 @@ def test_serialize_workflow():
285
285
  {
286
286
  "id": "de7b0b4e-7803-4d36-a275-2e7e3f60342b",
287
287
  "type": "IF",
288
- "source_handle_id": "561b4e3a-8db3-448a-8933-1115937082ff",
288
+ "source_handle_id": "3a45b81f-95e4-4cbd-8997-bfdbe30251e8",
289
289
  "data": {
290
290
  "id": "2ccd0730-26d1-4fb4-baa9-1a2a182dd9a0",
291
291
  "rules": [
@@ -309,7 +309,7 @@ def test_serialize_workflow():
309
309
  {
310
310
  "id": "5e783d17-6808-441a-ac6c-33a4e184f4e0",
311
311
  "type": "ELIF",
312
- "source_handle_id": "0644f22e-2680-441a-9554-eedf1d3d22a9",
312
+ "source_handle_id": "7202f702-1ebc-4067-ab1e-ec67e49158ee",
313
313
  "data": {
314
314
  "id": "cc3f0d92-b603-42cc-b2e9-83e3b23b3bcb",
315
315
  "rules": [
@@ -333,7 +333,7 @@ def test_serialize_workflow():
333
333
  {
334
334
  "id": "6bd2f643-9cf5-4e7f-9113-f90e5c8057be",
335
335
  "type": "ELIF",
336
- "source_handle_id": "2a48d274-ecfc-4f40-95ac-bc697663f10c",
336
+ "source_handle_id": "cf45705d-1a47-43a6-9d24-a7fdf78baae0",
337
337
  "data": {
338
338
  "id": "a5a0f391-7052-452f-9fe1-a5781a491591",
339
339
  "rules": [
@@ -357,7 +357,7 @@ def test_serialize_workflow():
357
357
  {
358
358
  "id": "0a058485-18a4-4e20-8a30-6da8196ac46f",
359
359
  "type": "ELIF",
360
- "source_handle_id": "90fc9dc8-0a74-4a98-b6ac-55ffce4a2881",
360
+ "source_handle_id": "f04610dd-61cf-41b0-b337-2235e101cdb0",
361
361
  "data": {
362
362
  "id": "efe7a851-2a67-4189-99ec-bc193242b270",
363
363
  "rules": [
@@ -409,7 +409,7 @@ def test_serialize_workflow():
409
409
  {
410
410
  "id": "c2fa8a44-923b-462a-b0d2-fa800a152e52",
411
411
  "type": "ELSE",
412
- "source_handle_id": "493024f4-8010-4e1a-abae-b6adbc6fb208",
412
+ "source_handle_id": "f9dde637-ea90-465f-a871-caf8380ae377",
413
413
  "data": None,
414
414
  },
415
415
  ],
@@ -913,7 +913,7 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
913
913
  {
914
914
  "id": "a4c32611-fd58-4b98-9d08-313cfd1c214e",
915
915
  "type": "IF",
916
- "source_handle_id": "8124a6cf-4a34-4149-adc0-68696c11bd4e",
916
+ "source_handle_id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
917
917
  "data": {
918
918
  "id": "650e7105-3e76-43ca-858f-b290970b438b",
919
919
  "rules": [
@@ -937,7 +937,7 @@ def test_conditional_node_serialize_all_operators_with_lhs_and_rhs(descriptor, o
937
937
  {
938
938
  "id": "342e5497-ea2b-4e5c-99cf-e6492f133a3c",
939
939
  "type": "ELSE",
940
- "source_handle_id": "4df924c0-7bed-4f4a-9db4-2bfe51841755",
940
+ "source_handle_id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
941
941
  "data": None,
942
942
  },
943
943
  ],
@@ -1024,7 +1024,7 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
1024
1024
  {
1025
1025
  "id": "a4c32611-fd58-4b98-9d08-313cfd1c214e",
1026
1026
  "type": "IF",
1027
- "source_handle_id": "8124a6cf-4a34-4149-adc0-68696c11bd4e",
1027
+ "source_handle_id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
1028
1028
  "data": {
1029
1029
  "id": "650e7105-3e76-43ca-858f-b290970b438b",
1030
1030
  "rules": [
@@ -1048,7 +1048,7 @@ def test_conditional_node_serialize_all_operators_with_expression(descriptor, op
1048
1048
  {
1049
1049
  "id": "342e5497-ea2b-4e5c-99cf-e6492f133a3c",
1050
1050
  "type": "ELSE",
1051
- "source_handle_id": "4df924c0-7bed-4f4a-9db4-2bfe51841755",
1051
+ "source_handle_id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
1052
1052
  "data": None,
1053
1053
  },
1054
1054
  ],
@@ -1147,7 +1147,7 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
1147
1147
  {
1148
1148
  "id": "a4c32611-fd58-4b98-9d08-313cfd1c214e",
1149
1149
  "type": "IF",
1150
- "source_handle_id": "8124a6cf-4a34-4149-adc0-68696c11bd4e",
1150
+ "source_handle_id": "2ff87aa6-37cf-43dd-af9d-13b9198ab70a",
1151
1151
  "data": {
1152
1152
  "id": "650e7105-3e76-43ca-858f-b290970b438b",
1153
1153
  "rules": [
@@ -1171,7 +1171,7 @@ def test_conditional_node_serialize_all_operators_with_value_and_start_and_end(d
1171
1171
  {
1172
1172
  "id": "342e5497-ea2b-4e5c-99cf-e6492f133a3c",
1173
1173
  "type": "ELSE",
1174
- "source_handle_id": "4df924c0-7bed-4f4a-9db4-2bfe51841755",
1174
+ "source_handle_id": "9a7e8c2e-0228-4321-8f74-61cb5778f3df",
1175
1175
  "data": None,
1176
1176
  },
1177
1177
  ],