vellum-ai 0.13.19__py3-none-any.whl → 0.13.20__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 (33) hide show
  1. vellum/__init__.py +2 -0
  2. vellum/client/core/client_wrapper.py +1 -1
  3. vellum/client/resources/test_suites/client.py +44 -8
  4. vellum/client/types/__init__.py +2 -0
  5. vellum/client/types/container_image_container_image_tag.py +21 -0
  6. vellum/client/types/container_image_read.py +2 -1
  7. vellum/types/container_image_container_image_tag.py +3 -0
  8. vellum/workflows/nodes/core/inline_subworkflow_node/node.py +48 -3
  9. {vellum_ai-0.13.19.dist-info → vellum_ai-0.13.20.dist-info}/METADATA +1 -1
  10. {vellum_ai-0.13.19.dist-info → vellum_ai-0.13.20.dist-info}/RECORD +33 -31
  11. vellum_ee/workflows/display/nodes/vellum/subworkflow_deployment_node.py +2 -3
  12. vellum_ee/workflows/display/nodes/vellum/utils.py +1 -1
  13. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py +9 -30
  14. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py +22 -36
  15. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py +35 -70
  16. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_error_node_serialization.py +2 -2
  17. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_generic_node_serialization.py +2 -2
  18. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_guardrail_node_serialization.py +2 -2
  19. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_inline_subworkflow_serialization.py +24 -57
  20. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_map_node_serialization.py +4 -4
  21. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_merge_node_serialization.py +2 -2
  22. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_prompt_deployment_serialization.py +2 -2
  23. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_search_node_serialization.py +2 -2
  24. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_subworkflow_deployment_serialization.py +4 -4
  25. vellum_ee/workflows/display/tests/workflow_serialization/test_basic_templating_node_serialization.py +2 -2
  26. vellum_ee/workflows/display/tests/workflow_serialization/test_complex_terminal_node_serialization.py +2 -2
  27. vellum_ee/workflows/display/utils/vellum.py +0 -2
  28. vellum_ee/workflows/display/vellum.py +0 -2
  29. vellum_ee/workflows/display/workflows/vellum_workflow_display.py +1 -8
  30. vellum_ee/workflows/tests/local_workflow/display/workflow.py +0 -2
  31. {vellum_ai-0.13.19.dist-info → vellum_ai-0.13.20.dist-info}/LICENSE +0 -0
  32. {vellum_ai-0.13.19.dist-info → vellum_ai-0.13.20.dist-info}/WHEEL +0 -0
  33. {vellum_ai-0.13.19.dist-info → vellum_ai-0.13.20.dist-info}/entry_points.txt +0 -0
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py CHANGED
@@ -125,11 +125,11 @@ def test_serialize_workflow_with_filepath():
125
125
  "target_handle_id": "30fb0f4a-61c3-49de-a0aa-7dfdcee6ea07",
126
126
  "output_id": "f6a3e3e0-f83f-4491-8b7a-b20fddd7160c",
127
127
  "output_type": "NUMBER",
128
- "node_input_id": "ae302487-ff2a-457a-81ed-9e0348e91833",
128
+ "node_input_id": "529bdd20-0985-4c99-87dc-590907937c1d",
129
129
  },
130
130
  "inputs": [
131
131
  {
132
- "id": "ae302487-ff2a-457a-81ed-9e0348e91833",
132
+ "id": "529bdd20-0985-4c99-87dc-590907937c1d",
133
133
  "key": "node_input",
134
134
  "value": {
135
135
  "rules": [
@@ -161,11 +161,11 @@ def test_serialize_workflow_with_filepath():
161
161
  "target_handle_id": "1e126004-9de7-42c0-b1e1-87f9eb0642e2",
162
162
  "output_id": "1cee930f-342f-421c-89fc-ff212b3764bb",
163
163
  "output_type": "STRING",
164
- "node_input_id": "c6593516-ffc5-49a8-8a65-1038cccec3f8",
164
+ "node_input_id": "09501b65-d9b3-4920-81d4-96f93c840667",
165
165
  },
166
166
  "inputs": [
167
167
  {
168
- "id": "c6593516-ffc5-49a8-8a65-1038cccec3f8",
168
+ "id": "09501b65-d9b3-4920-81d4-96f93c840667",
169
169
  "key": "node_input",
170
170
  "value": {
171
171
  "rules": [
@@ -358,11 +358,11 @@ def test_serialize_workflow_with_code():
358
358
  "target_handle_id": "de8f2cc2-8c32-4782-87d5-4eb5afcd42e3",
359
359
  "output_id": "283d6849-f3ed-4beb-b261-cf70f90e8d10",
360
360
  "output_type": "NUMBER",
361
- "node_input_id": "b38ba7a8-0b2a-4146-8d58-9fa0bcba8cd5",
361
+ "node_input_id": "101d5222-8578-413a-be80-b1acf7559a0d",
362
362
  },
363
363
  "inputs": [
364
364
  {
365
- "id": "b38ba7a8-0b2a-4146-8d58-9fa0bcba8cd5",
365
+ "id": "101d5222-8578-413a-be80-b1acf7559a0d",
366
366
  "key": "node_input",
367
367
  "value": {
368
368
  "rules": [
@@ -394,11 +394,11 @@ def test_serialize_workflow_with_code():
394
394
  "target_handle_id": "6b7d7f2c-5cc8-4005-9e66-cdb2c97b1998",
395
395
  "output_id": "4c136180-050b-4422-a7a4-2a1c6729042c",
396
396
  "output_type": "STRING",
397
- "node_input_id": "76d49710-1ed0-4105-a1d7-9190c0408558",
397
+ "node_input_id": "de7ea5c4-da2f-4cfd-9192-0257d0f9ba53",
398
398
  },
399
399
  "inputs": [
400
400
  {
401
- "id": "76d49710-1ed0-4105-a1d7-9190c0408558",
401
+ "id": "de7ea5c4-da2f-4cfd-9192-0257d0f9ba53",
402
402
  "key": "node_input",
403
403
  "value": {
404
404
  "rules": [
@@ -601,29 +601,17 @@ def test_serialize_workflow__try_wrapped():
601
601
  {
602
602
  "id": "af4fc1ef-7701-43df-b5e7-4f354f707db2",
603
603
  "type": "TERMINAL",
604
- "base": {
605
- "module": [
606
- "vellum",
607
- "workflows",
608
- "nodes",
609
- "displayable",
610
- "final_output_node",
611
- "node",
612
- ],
613
- "name": "FinalOutputNode",
614
- },
615
- "definition": None,
616
604
  "data": {
617
605
  "label": "Final Output",
618
606
  "name": "log",
619
607
  "target_handle_id": "d243df8d-46f6-4928-ac31-7c775c5d73a9",
620
608
  "output_id": "5fbd27a0-9831-49c7-93c8-9c2a28c78696",
621
609
  "output_type": "STRING",
622
- "node_input_id": "569e1c05-53fa-413b-abf0-0353eaa44208",
610
+ "node_input_id": "a6ffaacf-1284-4e70-907e-042bb281e6fa",
623
611
  },
624
612
  "inputs": [
625
613
  {
626
- "id": "569e1c05-53fa-413b-abf0-0353eaa44208",
614
+ "id": "a6ffaacf-1284-4e70-907e-042bb281e6fa",
627
615
  "key": "node_input",
628
616
  "value": {
629
617
  "rules": [
@@ -640,33 +628,26 @@ def test_serialize_workflow__try_wrapped():
640
628
  }
641
629
  ],
642
630
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
643
- },
644
- {
645
- "id": "4cbfa5f7-fc12-4ab2-81cb-168c5caef4f0",
646
- "type": "TERMINAL",
647
631
  "base": {
648
- "module": [
649
- "vellum",
650
- "workflows",
651
- "nodes",
652
- "displayable",
653
- "final_output_node",
654
- "node",
655
- ],
656
632
  "name": "FinalOutputNode",
633
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
657
634
  },
658
635
  "definition": None,
636
+ },
637
+ {
638
+ "id": "4cbfa5f7-fc12-4ab2-81cb-168c5caef4f0",
639
+ "type": "TERMINAL",
659
640
  "data": {
660
641
  "label": "Final Output",
661
642
  "name": "result",
662
643
  "target_handle_id": "9c43709e-25cb-4548-b840-3fcf6a1c9f3e",
663
644
  "output_id": "400f9ffe-e700-4204-a810-e06123565947",
664
645
  "output_type": "NUMBER",
665
- "node_input_id": "3d6cb7ef-985f-48f1-ad23-de49be60666a",
646
+ "node_input_id": "2cbb7351-a3e8-4a6b-b33e-204b04da7ad6",
666
647
  },
667
648
  "inputs": [
668
649
  {
669
- "id": "3d6cb7ef-985f-48f1-ad23-de49be60666a",
650
+ "id": "2cbb7351-a3e8-4a6b-b33e-204b04da7ad6",
670
651
  "key": "node_input",
671
652
  "value": {
672
653
  "rules": [
@@ -683,6 +664,11 @@ def test_serialize_workflow__try_wrapped():
683
664
  }
684
665
  ],
685
666
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
667
+ "base": {
668
+ "name": "FinalOutputNode",
669
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
670
+ },
671
+ "definition": None,
686
672
  },
687
673
  ],
688
674
  sorted(final_output_nodes, key=lambda x: x["id"], reverse=True),
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py CHANGED
@@ -438,29 +438,17 @@ def test_serialize_workflow():
438
438
  {
439
439
  "id": "9c22ee47-01da-4e4e-863d-b4a6874bed66",
440
440
  "type": "TERMINAL",
441
- "base": {
442
- "module": [
443
- "vellum",
444
- "workflows",
445
- "nodes",
446
- "displayable",
447
- "final_output_node",
448
- "node",
449
- ],
450
- "name": "FinalOutputNode",
451
- },
452
- "definition": None,
453
441
  "data": {
454
442
  "label": "Final Output",
455
443
  "name": "statement",
456
444
  "target_handle_id": "f02a8971-e9a4-4716-bfb4-d08f5614b5d8",
457
445
  "output_id": "cdbe2adf-9951-409a-b9a8-b8b349037f4f",
458
446
  "output_type": "STRING",
459
- "node_input_id": "bed69f8a-3a83-4e52-beba-8eb14f4b0ca9",
447
+ "node_input_id": "2e742a40-cbee-4e19-9269-c62dc4a9204e",
460
448
  },
461
449
  "inputs": [
462
450
  {
463
- "id": "bed69f8a-3a83-4e52-beba-8eb14f4b0ca9",
451
+ "id": "2e742a40-cbee-4e19-9269-c62dc4a9204e",
464
452
  "key": "node_input",
465
453
  "value": {
466
454
  "rules": [
@@ -477,33 +465,26 @@ def test_serialize_workflow():
477
465
  }
478
466
  ],
479
467
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
480
- },
481
- {
482
- "id": "47f0931c-41f6-4b84-bf39-0c486941f599",
483
- "type": "TERMINAL",
484
468
  "base": {
485
- "module": [
486
- "vellum",
487
- "workflows",
488
- "nodes",
489
- "displayable",
490
- "final_output_node",
491
- "node",
492
- ],
493
469
  "name": "FinalOutputNode",
470
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
494
471
  },
495
472
  "definition": None,
473
+ },
474
+ {
475
+ "id": "47f0931c-41f6-4b84-bf39-0c486941f599",
476
+ "type": "TERMINAL",
496
477
  "data": {
497
478
  "label": "Final Output",
498
479
  "name": "compliment",
499
480
  "target_handle_id": "a4d57adc-58c1-40c6-810b-ee5fd923bfc5",
500
481
  "output_id": "93f2cb75-6fa2-4e46-9488-c0bcd29153c0",
501
482
  "output_type": "STRING",
502
- "node_input_id": "e3f2d793-31a7-4670-8155-bd034d9f25e2",
483
+ "node_input_id": "9ba792e6-55e3-4a14-8768-e8ef6955c934",
503
484
  },
504
485
  "inputs": [
505
486
  {
506
- "id": "e3f2d793-31a7-4670-8155-bd034d9f25e2",
487
+ "id": "9ba792e6-55e3-4a14-8768-e8ef6955c934",
507
488
  "key": "node_input",
508
489
  "value": {
509
490
  "rules": [
@@ -520,33 +501,26 @@ def test_serialize_workflow():
520
501
  }
521
502
  ],
522
503
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
523
- },
524
- {
525
- "id": "e3d29229-f746-4125-819e-f847acbed307",
526
- "type": "TERMINAL",
527
504
  "base": {
528
- "module": [
529
- "vellum",
530
- "workflows",
531
- "nodes",
532
- "displayable",
533
- "final_output_node",
534
- "node",
535
- ],
536
505
  "name": "FinalOutputNode",
506
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
537
507
  },
538
508
  "definition": None,
509
+ },
510
+ {
511
+ "id": "e3d29229-f746-4125-819e-f847acbed307",
512
+ "type": "TERMINAL",
539
513
  "data": {
540
514
  "label": "Final Output",
541
515
  "name": "complaint",
542
516
  "target_handle_id": "c5dd9bf5-9e18-4dbc-8c20-2c0baf969ebe",
543
517
  "output_id": "f936ae31-ba15-4864-8961-86231022a4d7",
544
518
  "output_type": "STRING",
545
- "node_input_id": "9b35cc61-5887-478a-814d-9693ead5932f",
519
+ "node_input_id": "47f426a4-8770-4f30-a285-5d21849063a5",
546
520
  },
547
521
  "inputs": [
548
522
  {
549
- "id": "9b35cc61-5887-478a-814d-9693ead5932f",
523
+ "id": "47f426a4-8770-4f30-a285-5d21849063a5",
550
524
  "key": "node_input",
551
525
  "value": {
552
526
  "rules": [
@@ -563,33 +537,26 @@ def test_serialize_workflow():
563
537
  }
564
538
  ],
565
539
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
566
- },
567
- {
568
- "id": "6efa7b45-0580-406d-85aa-439117ba8021",
569
- "type": "TERMINAL",
570
540
  "base": {
571
- "module": [
572
- "vellum",
573
- "workflows",
574
- "nodes",
575
- "displayable",
576
- "final_output_node",
577
- "node",
578
- ],
579
541
  "name": "FinalOutputNode",
542
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
580
543
  },
581
544
  "definition": None,
545
+ },
546
+ {
547
+ "id": "6efa7b45-0580-406d-85aa-439117ba8021",
548
+ "type": "TERMINAL",
582
549
  "data": {
583
550
  "label": "Final Output",
584
551
  "name": "fallthrough",
585
552
  "target_handle_id": "2283cd2c-b077-4b5d-a96f-aa2cd6023eda",
586
553
  "output_id": "62ad462f-f819-4940-99ab-b3f145507f57",
587
554
  "output_type": "STRING",
588
- "node_input_id": "cee5378a-9011-4d03-bd3c-a32421bd093f",
555
+ "node_input_id": "d2d3e7cc-f6d6-4ac1-ad19-e6f52a75b38f",
589
556
  },
590
557
  "inputs": [
591
558
  {
592
- "id": "cee5378a-9011-4d03-bd3c-a32421bd093f",
559
+ "id": "d2d3e7cc-f6d6-4ac1-ad19-e6f52a75b38f",
593
560
  "key": "node_input",
594
561
  "value": {
595
562
  "rules": [
@@ -606,33 +573,26 @@ def test_serialize_workflow():
606
573
  }
607
574
  ],
608
575
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
609
- },
610
- {
611
- "id": "fa11b84b-1d76-4adc-ab28-cbbaa933c267",
612
- "type": "TERMINAL",
613
576
  "base": {
614
- "module": [
615
- "vellum",
616
- "workflows",
617
- "nodes",
618
- "displayable",
619
- "final_output_node",
620
- "node",
621
- ],
622
577
  "name": "FinalOutputNode",
578
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
623
579
  },
624
580
  "definition": None,
581
+ },
582
+ {
583
+ "id": "fa11b84b-1d76-4adc-ab28-cbbaa933c267",
584
+ "type": "TERMINAL",
625
585
  "data": {
626
586
  "label": "Final Output",
627
587
  "name": "question",
628
588
  "target_handle_id": "e1a6da28-02c5-40d7-8ac5-9fb07e2e3e1d",
629
589
  "output_id": "c05f7d96-59a0-4d58-93d7-d451afd3f630",
630
590
  "output_type": "STRING",
631
- "node_input_id": "1c1d34d9-0cda-471f-ac9a-40ac351c9aca",
591
+ "node_input_id": "a13377c4-ae1c-4be4-b695-f9da590486ef",
632
592
  },
633
593
  "inputs": [
634
594
  {
635
- "id": "1c1d34d9-0cda-471f-ac9a-40ac351c9aca",
595
+ "id": "a13377c4-ae1c-4be4-b695-f9da590486ef",
636
596
  "key": "node_input",
637
597
  "value": {
638
598
  "rules": [
@@ -649,6 +609,11 @@ def test_serialize_workflow():
649
609
  }
650
610
  ],
651
611
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
612
+ "base": {
613
+ "name": "FinalOutputNode",
614
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
615
+ },
616
+ "definition": None,
652
617
  },
653
618
  ],
654
619
  workflow_raw_data["nodes"][7:12],
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_error_node_serialization.py CHANGED
@@ -138,11 +138,11 @@ def test_serialize_workflow():
138
138
  "target_handle_id": "b070e9bc-e9b7-46d3-8f5b-0b646bd25cf0",
139
139
  "output_id": "04c5c6be-f5e1-41b8-b668-39e179790d9e",
140
140
  "output_type": "NUMBER",
141
- "node_input_id": "39ff42c9-eae8-432e-ad41-e208fba77027",
141
+ "node_input_id": "c191a5c1-8912-49ad-bf4b-c6d4d58ed482",
142
142
  },
143
143
  "inputs": [
144
144
  {
145
- "id": "39ff42c9-eae8-432e-ad41-e208fba77027",
145
+ "id": "c191a5c1-8912-49ad-bf4b-c6d4d58ed482",
146
146
  "key": "node_input",
147
147
  "value": {
148
148
  "rules": [
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_generic_node_serialization.py CHANGED
@@ -83,11 +83,11 @@ def test_serialize_workflow(vellum_client):
83
83
  "target_handle_id": "8bd9f4f3-9f66-4d95-8e84-529b0002c531",
84
84
  "output_id": "2b6389d0-266a-4be4-843e-4e543dd3d727",
85
85
  "output_type": "STRING",
86
- "node_input_id": "7a9f2d3a-0b23-4bd4-b567-e9493135b727",
86
+ "node_input_id": "545d6001-cfb5-4ccc-bcdf-3b03ccd67d90",
87
87
  },
88
88
  "inputs": [
89
89
  {
90
- "id": "7a9f2d3a-0b23-4bd4-b567-e9493135b727",
90
+ "id": "545d6001-cfb5-4ccc-bcdf-3b03ccd67d90",
91
91
  "key": "node_input",
92
92
  "value": {
93
93
  "rules": [
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_guardrail_node_serialization.py CHANGED
@@ -140,11 +140,11 @@ def test_serialize_workflow():
140
140
  "target_handle_id": "001b97f6-2bc8-4d1e-9572-028dcf17df4e",
141
141
  "output_id": "2abd2b3b-c301-4834-a43f-5db3604f8422",
142
142
  "output_type": "NUMBER",
143
- "node_input_id": "6580eb90-2e99-4010-bebe-4b9fc4cf0dfd",
143
+ "node_input_id": "6321442a-0d0d-4e25-965d-c24ff24712c5",
144
144
  },
145
145
  "inputs": [
146
146
  {
147
- "id": "6580eb90-2e99-4010-bebe-4b9fc4cf0dfd",
147
+ "id": "6321442a-0d0d-4e25-965d-c24ff24712c5",
148
148
  "key": "node_input",
149
149
  "value": {
150
150
  "rules": [
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_inline_subworkflow_serialization.py CHANGED
@@ -145,7 +145,7 @@ def test_serialize_workflow():
145
145
  "merge_behavior": "AWAIT_ATTRIBUTES",
146
146
  },
147
147
  "ports": [
148
- {"id": "1e739e86-a285-4438-9725-a152c15a63e3", "type": "DEFAULT", "name": "default"}
148
+ {"id": "1e739e86-a285-4438-9725-a152c15a63e3", "name": "default", "type": "DEFAULT"}
149
149
  ],
150
150
  "adornments": None,
151
151
  "attributes": [
@@ -190,11 +190,11 @@ def test_serialize_workflow():
190
190
  "target_handle_id": "804bb543-9cf4-457f-acf1-fb4b8b7d9259",
191
191
  "output_id": "2fc57139-7420-49e5-96a6-dcbb3ff5d622",
192
192
  "output_type": "NUMBER",
193
- "node_input_id": "a9dc6e1e-ecbd-4308-93f4-7429483dd07c",
193
+ "node_input_id": "712eaeec-9e1e-41bd-9217-9caec8b6ade7",
194
194
  },
195
195
  "inputs": [
196
196
  {
197
- "id": "a9dc6e1e-ecbd-4308-93f4-7429483dd07c",
197
+ "id": "712eaeec-9e1e-41bd-9217-9caec8b6ade7",
198
198
  "key": "node_input",
199
199
  "value": {
200
200
  "rules": [
@@ -213,14 +213,7 @@ def test_serialize_workflow():
213
213
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
214
214
  "base": {
215
215
  "name": "FinalOutputNode",
216
- "module": [
217
- "vellum",
218
- "workflows",
219
- "nodes",
220
- "displayable",
221
- "final_output_node",
222
- "node",
223
- ],
216
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
224
217
  },
225
218
  "definition": None,
226
219
  },
@@ -233,11 +226,11 @@ def test_serialize_workflow():
233
226
  "target_handle_id": "6d4c4a14-c388-4c7a-b223-eb39baf5c080",
234
227
  "output_id": "fad5dd9f-3328-4e70-ad55-65a5325a4a82",
235
228
  "output_type": "STRING",
236
- "node_input_id": "39c0f9c2-1805-4617-9771-4eb23c76d88e",
229
+ "node_input_id": "8fd4279a-4f13-4257-9577-1b55e964cdf1",
237
230
  },
238
231
  "inputs": [
239
232
  {
240
- "id": "39c0f9c2-1805-4617-9771-4eb23c76d88e",
233
+ "id": "8fd4279a-4f13-4257-9577-1b55e964cdf1",
241
234
  "key": "node_input",
242
235
  "value": {
243
236
  "rules": [
@@ -256,14 +249,7 @@ def test_serialize_workflow():
256
249
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
257
250
  "base": {
258
251
  "name": "FinalOutputNode",
259
- "module": [
260
- "vellum",
261
- "workflows",
262
- "nodes",
263
- "displayable",
264
- "final_output_node",
265
- "node",
266
- ],
252
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
267
253
  },
268
254
  "definition": None,
269
255
  },
@@ -297,12 +283,7 @@ def test_serialize_workflow():
297
283
  "display_data": {"viewport": {"x": 0.0, "y": 0.0, "zoom": 1.0}},
298
284
  "definition": {
299
285
  "name": "NestedWorkflow",
300
- "module": [
301
- "tests",
302
- "workflows",
303
- "basic_inline_subworkflow",
304
- "workflow",
305
- ],
286
+ "module": ["tests", "workflows", "basic_inline_subworkflow", "workflow"],
306
287
  },
307
288
  },
308
289
  "input_variables": [{"id": "704c4640-bfda-44f0-8da3-e9cfc4f21cf2", "key": "metro", "type": "STRING"}],
@@ -336,29 +317,17 @@ def test_serialize_workflow():
336
317
  {
337
318
  "id": "31b74695-3f1c-47cf-8be8-a4d86cc589e8",
338
319
  "type": "TERMINAL",
339
- "base": {
340
- "module": [
341
- "vellum",
342
- "workflows",
343
- "nodes",
344
- "displayable",
345
- "final_output_node",
346
- "node",
347
- ],
348
- "name": "FinalOutputNode",
349
- },
350
- "definition": None,
351
320
  "data": {
352
321
  "label": "Final Output",
353
322
  "name": "reasoning",
354
323
  "target_handle_id": "8b525943-6c27-414b-a329-e29c0b217f72",
355
324
  "output_id": "7444a019-081a-4e10-a528-3249299159f7",
356
325
  "output_type": "STRING",
357
- "node_input_id": "736473c8-b0b4-4cdd-b743-6453dd5306fc",
326
+ "node_input_id": "c1833b54-95b6-4365-8e57-51b09c8e2606",
358
327
  },
359
328
  "inputs": [
360
329
  {
361
- "id": "736473c8-b0b4-4cdd-b743-6453dd5306fc",
330
+ "id": "c1833b54-95b6-4365-8e57-51b09c8e2606",
362
331
  "key": "node_input",
363
332
  "value": {
364
333
  "rules": [
@@ -366,7 +335,7 @@ def test_serialize_workflow():
366
335
  "type": "NODE_OUTPUT",
367
336
  "data": {
368
337
  "node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
369
- "output_id": "fad5dd9f-3328-4e70-ad55-65a5325a4a82",
338
+ "output_id": "0a7192da-5576-4933-bba4-de8adf5d7996",
370
339
  },
371
340
  }
372
341
  ],
@@ -375,6 +344,11 @@ def test_serialize_workflow():
375
344
  }
376
345
  ],
377
346
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
347
+ "base": {
348
+ "name": "FinalOutputNode",
349
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
350
+ },
351
+ "definition": None,
378
352
  },
379
353
  reasoning_terminal_node,
380
354
  ignore_order=True,
@@ -387,29 +361,17 @@ def test_serialize_workflow():
387
361
  {
388
362
  "id": "0779b232-82ab-4dbe-a340-6a85e6ab3368",
389
363
  "type": "TERMINAL",
390
- "base": {
391
- "module": [
392
- "vellum",
393
- "workflows",
394
- "nodes",
395
- "displayable",
396
- "final_output_node",
397
- "node",
398
- ],
399
- "name": "FinalOutputNode",
400
- },
401
- "definition": None,
402
364
  "data": {
403
365
  "label": "Final Output",
404
366
  "name": "temperature",
405
367
  "target_handle_id": "9e077063-c394-4c7b-b0c6-e6686df67984",
406
368
  "output_id": "99afb757-2782-465d-ab55-80ccf50552b9",
407
369
  "output_type": "NUMBER",
408
- "node_input_id": "df0d7e0c-2b37-4059-91c1-0419f950b7fe",
370
+ "node_input_id": "7761c5e1-cc2e-43ab-bfd2-f66c3d47b3b9",
409
371
  },
410
372
  "inputs": [
411
373
  {
412
- "id": "df0d7e0c-2b37-4059-91c1-0419f950b7fe",
374
+ "id": "7761c5e1-cc2e-43ab-bfd2-f66c3d47b3b9",
413
375
  "key": "node_input",
414
376
  "value": {
415
377
  "rules": [
@@ -417,7 +379,7 @@ def test_serialize_workflow():
417
379
  "type": "NODE_OUTPUT",
418
380
  "data": {
419
381
  "node_id": "080e4343-c7ce-4f82-b9dd-e94c8cc92239",
420
- "output_id": "2fc57139-7420-49e5-96a6-dcbb3ff5d622",
382
+ "output_id": "86dd0202-c141-48a3-8382-2da60372e77c",
421
383
  },
422
384
  }
423
385
  ],
@@ -426,6 +388,11 @@ def test_serialize_workflow():
426
388
  }
427
389
  ],
428
390
  "display_data": {"position": {"x": 0.0, "y": 0.0}},
391
+ "base": {
392
+ "name": "FinalOutputNode",
393
+ "module": ["vellum", "workflows", "nodes", "displayable", "final_output_node", "node"],
394
+ },
395
+ "definition": None,
429
396
  },
430
397
  temperature_terminal_node,
431
398
  ignore_order=True,
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_map_node_serialization.py CHANGED
@@ -173,11 +173,11 @@ def test_serialize_workflow():
173
173
  "target_handle_id": "9d74571f-b7f5-4c1d-8b7c-b9c648738a4d",
174
174
  "output_id": "2a957315-fae0-4366-8a35-f0b315c5eade",
175
175
  "output_type": "NUMBER",
176
- "node_input_id": "ae65cf7e-3db6-410f-a556-75ee11ce7e84",
176
+ "node_input_id": "05f2e4cc-0e36-4c6b-a535-be38281e5280",
177
177
  },
178
178
  "inputs": [
179
179
  {
180
- "id": "ae65cf7e-3db6-410f-a556-75ee11ce7e84",
180
+ "id": "05f2e4cc-0e36-4c6b-a535-be38281e5280",
181
181
  "key": "node_input",
182
182
  "value": {
183
183
  "rules": [
@@ -293,11 +293,11 @@ def test_serialize_workflow():
293
293
  "target_handle_id": "720dd872-2f3d-47b9-8245-89387f04f300",
294
294
  "output_id": "145b0b68-224b-4f83-90e6-eea3457e6c3e",
295
295
  "output_type": "JSON",
296
- "node_input_id": "d1b01eac-23a9-43ce-beb3-e13f83bd7d18",
296
+ "node_input_id": "57fe72e9-cc9f-48f2-b5ea-25ac52637a33",
297
297
  },
298
298
  "inputs": [
299
299
  {
300
- "id": "d1b01eac-23a9-43ce-beb3-e13f83bd7d18",
300
+ "id": "57fe72e9-cc9f-48f2-b5ea-25ac52637a33",
301
301
  "key": "node_input",
302
302
  "value": {
303
303
  "rules": [
vellum_ee/workflows/display/tests/workflow_serialization/test_basic_merge_node_serialization.py CHANGED
@@ -101,7 +101,7 @@ def test_serialize_workflow__await_all():
101
101
  "target_handle_id": "ff55701c-16d3-4348-a633-6a298e71377d",
102
102
  "output_id": "959ba00d-d30b-402e-8676-76efc4c3d2ae",
103
103
  "output_type": "STRING",
104
- "node_input_id": "7f950be4-2fab-44e0-87a3-b1631aadd0e3",
104
+ "node_input_id": "fed9d343-6504-460c-968b-d3f9658193d0",
105
105
  },
106
106
  "base": {
107
107
  "module": [
@@ -117,7 +117,7 @@ def test_serialize_workflow__await_all():
117
117
  "definition": None,
118
118
  "inputs": [
119
119
  {
120
- "id": "7f950be4-2fab-44e0-87a3-b1631aadd0e3",
120
+ "id": "fed9d343-6504-460c-968b-d3f9658193d0",
121
121
  "key": "node_input",
122
122
  "value": {
123
123
  "rules": [