vellum-ai 1.2.5__py3-none-any.whl → 1.3.1__py3-none-any.whl

{vellum_ai-1.2.5.dist-info → vellum_ai-1.3.1.dist-info}/RECORD

@@ -14,10 +14,10 @@ vellum_cli/tests/conftest.py,sha256=wx3PlJjVB0HRf5dr2b_idOIw27WPPl0J0FNbhIJJaVk,
 vellum_cli/tests/test_config.py,sha256=uvKGDc8BoVyT9_H0Z-g8469zVxomn6Oi3Zj-vK7O_wU,2631
 vellum_cli/tests/test_image_push.py,sha256=X0YOPdoaAnWtK9IQTxaN_wWpw08-5G3v76k1Wu53JpU,12801
 vellum_cli/tests/test_image_push_error_handling.py,sha256=QRH0eYNEEIkD2-EVFQWYexOKlhMB6puh1GouovrE-VU,7647
-vellum_cli/tests/test_init.py,sha256=8UOc_ThfouR4ja5cCl_URuLk7ohr9JXfCnG4yka1OUQ,18754
+vellum_cli/tests/test_init.py,sha256=C_rV4lu-ab5iFoLRizs1XAUnSPdMf7oFuc1i4N4udOU,18285
 vellum_cli/tests/test_main.py,sha256=qDZG-aQauPwBwM6A2DIu1494n47v3pL28XakTbLGZ-k,272
 vellum_cli/tests/test_ping.py,sha256=b3aQLd-N59_8w2rRiWqwpB1rlHaKEYVbAj1Y3hi7A-g,2605
-vellum_cli/tests/test_pull.py,sha256=WcRtp5N27CbfBv7vx2wNhLTsSacYw7TiBVVXcI7ZMW4,50014
+vellum_cli/tests/test_pull.py,sha256=f7dK61y82LlJm-FAv-IpfmscJijIfZZrR-rzN1TKkU0,49421
 vellum_cli/tests/test_push.py,sha256=10G-H88tdiYvg9CaC8OXOf25UEzWkWWB01vNhuxzjog,41944
 vellum_ee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -75,7 +75,7 @@ vellum_ee/workflows/display/tests/workflow_serialization/__init__.py,sha256=47DE
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/conftest.py,sha256=Y-ajeT65b5varmrZCw6L3hir4hJCFq-eO0jZfRcrs7g,1886
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_adornments_serialization.py,sha256=bD9AtBo7Pz1Drbh420NR0VSiHBfExw_UtNRPCiAqki0,13967
-vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_attributes_serialization.py,sha256=iqTbPYVSrMgIls7vKdPgr7ZEbdpztDjlOWvexPU-zqc,26240
+vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_attributes_serialization.py,sha256=xEGh4RXGMR06M5s6rD6GjsloNslsRaH4UhIg7Djyw7Q,25144
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_outputs_serialization.py,sha256=O1CIKMmRTaMaT3IhjwCAiMz1ZThPg9lAUbSiZkx_E8c,6321
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_ports_serialization.py,sha256=fPXcX-tWQ0UMuEyOFAylgi7pWiE7lZWk9vrShlTExik,40053
 vellum_ee/workflows/display/tests/workflow_serialization/generic_nodes/test_trigger_serialization.py,sha256=FLvcD8eoABHUPv08oSpIp_P-65sw2gl4whMXEJJj4f8,6785
@@ -106,7 +106,7 @@ vellum_ee/workflows/display/tests/workflow_serialization/test_workflow_input_par
 vellum_ee/workflows/display/types.py,sha256=cyZruu4sXAdHjwuFc7dydM4DcFNf-pp_CmulXItxac4,3679
 vellum_ee/workflows/display/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum_ee/workflows/display/utils/auto_layout.py,sha256=f4GiLn_LazweupfqTpubcdtdfE_vrOcmZudSsnYIY9E,3906
-vellum_ee/workflows/display/utils/events.py,sha256=HHKXnVy1Zs0YJxTojp2iKQJFTLBqVXiY-YW0J-rRZlI,1673
+vellum_ee/workflows/display/utils/events.py,sha256=KC684ZWODHRGgJtik_KjGXIjM5x4_58_1dGuco_9O8c,1748
 vellum_ee/workflows/display/utils/exceptions.py,sha256=LSwwxCYNxFkf5XMUcFkaZKpQ13OSrI7y_bpEUwbKVk0,169
 vellum_ee/workflows/display/utils/expressions.py,sha256=yPrUIEQKL5fDwUC9NpVO9yjL6emTOdFJw8VjoZgZ0E0,16942
 vellum_ee/workflows/display/utils/registry.py,sha256=1qXiBTdsnro6FeCX0FGBEK7CIf6wa--Jt50iZ_nEp_M,3460
@@ -143,14 +143,14 @@ vellum_ee/workflows/tests/local_workflow/workflow.py,sha256=A4qOzOPNwePYxWbcAgIP
 vellum_ee/workflows/tests/test_display_meta.py,sha256=PkXJVnMZs9GNooDkd59n4YTBAX3XGPQWeSSVbhehVFM,5112
 vellum_ee/workflows/tests/test_registry.py,sha256=B8xRIuEyLWfSqrYoPldNQXhKPfe50PllvtAZoI8-uPs,6066
 vellum_ee/workflows/tests/test_serialize_module.py,sha256=lk-4dVnG2HcxxywBXxDR1ieH8D9RJt4lvchoZhtQPdU,2892
-vellum_ee/workflows/tests/test_server.py,sha256=sqF1qcgBVy9-blGetaea6MZ73KHnm0KaPz5B7s4NT9Q,19687
+vellum_ee/workflows/tests/test_server.py,sha256=dXFBraU99Y6cKp2aBhLFXQTScSRcE9WaWjo1z9piqdU,23344
 vellum_ee/workflows/tests/test_virtual_files.py,sha256=TJEcMR0v2S8CkloXNmCHA0QW0K6pYNGaIjraJz7sFvY,2762
-vellum/__init__.py,sha256=VZhH0RFOGRpsERt-mVzsKJlfopp3j9J4m72dNZU27j4,46934
-vellum/client/README.md,sha256=gxc7JlJRBrBZpN5LHa2ORxYTRHFLPnWmnIugN8pmQh4,5600
+vellum/__init__.py,sha256=8-mW70qyqMe2d4jNe2hgthuXs5n3c0qTlkdV8Nz8qcs,47252
+vellum/client/README.md,sha256=Uwq9qIFZAHOVCgEfatfWe4YoW-QYYy6xVA1bMjPvx3k,5601
 vellum/client/__init__.py,sha256=LOu_TklkJG01SgXIpPWDhOX2QvKDyd2ZbQyr5H0m31I,72349
 vellum/client/core/__init__.py,sha256=lTcqUPXcx4112yLDd70RAPeqq6tu3eFMe1pKOqkW9JQ,1562
 vellum/client/core/api_error.py,sha256=44vPoTyWN59gonCIZMdzw7M1uspygiLnr3GNFOoVL2Q,614
-vellum/client/core/client_wrapper.py,sha256=Hs7PXt1UQaeHfbQYgreOs11HOuvUOU0Oi4YROuf2M08,2840
+vellum/client/core/client_wrapper.py,sha256=yjqDTK-pFWVYr-n4F1kmy-TuBlGzXJBmaztIfr5uBF0,2840
 vellum/client/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 vellum/client/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 vellum/client/core/force_multipart.py,sha256=awxh5MtcRYe74ehY8U76jzv6fYM_w_D3Rur7KQQzSDk,429
@@ -171,7 +171,7 @@ vellum/client/errors/not_found_error.py,sha256=YrqVM0oc3qkQbFbmmm6xr300VGfUNxMSy
 vellum/client/errors/too_many_requests_error.py,sha256=SJJemdgUlQHV_VpxK8UfFNexgZebNT5_MTOeQs6oVgc,397
 vellum/client/errors/unauthorized_error.py,sha256=waPl5Swiqsk3FQK-Lljzx4KDh4RPZ0wL6BLHjM8onQ8,394
 vellum/client/raw_client.py,sha256=IJNaTvVzigyiqHZ9g-nLZzXsV8P7aRUZHdXV0U3ddn8,112401
-vellum/client/reference.md,sha256=oGKMw2vruJgy2hmDYu0adKkhmb_55J3UU8jB5iWjoNk,95172
+vellum/client/reference.md,sha256=Y5xeW_q9baKaA9Dcd9RL8MQLcGEGAQZXbiCmcrA7yr0,95032
 vellum/client/resources/__init__.py,sha256=R10JFtO6U6QRvNubrmDIjncT7e5q4AiOKjXSK4wFuXc,1609
 vellum/client/resources/ad_hoc/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 vellum/client/resources/ad_hoc/client.py,sha256=AbK_g9HPL2xEBHymwQ9rGO7js7X1YpNYcqPa-PxlsyM,14400
@@ -232,8 +232,8 @@ vellum/client/resources/workflow_executions/__init__.py,sha256=_VhToAyIt_5axN6CL
 vellum/client/resources/workflow_executions/client.py,sha256=842zA9XaKSgY0scgGZG0ICerJrJyDfVerSGf_zHXPcY,4177
 vellum/client/resources/workflow_executions/raw_client.py,sha256=y5iJ61fTqe4h6-YcqeIxXXMwChxJF80LFmym5t3oWNY,4728
 vellum/client/resources/workflow_sandboxes/__init__.py,sha256=fWzZgZDNIbJgdYSnL3f0lGa3lbb2bnmaofmodrlVG6o,195
-vellum/client/resources/workflow_sandboxes/client.py,sha256=w8fWQCIFbDLXl4OX1OT_cKoVPA_q65qgOhb-I_zXO1k,10457
-vellum/client/resources/workflow_sandboxes/raw_client.py,sha256=_C2MzCk36NP8Do4pat0Q0aPSSbP6KhOqBOaGtqNxwk0,12643
+vellum/client/resources/workflow_sandboxes/client.py,sha256=tsf2ujDOxZLcjZQp8FKXnNewQlxzv8s-3nIRgVv26uc,10089
+vellum/client/resources/workflow_sandboxes/raw_client.py,sha256=iQpaFR-D96ygLLh4xYbYGK88UH-wttapsPxMc1xkvas,12323
 vellum/client/resources/workflow_sandboxes/types/__init__.py,sha256=mZxwNpOmBfllmNZrhp8e_M62hUFI73B2fNlvgG8vi3Y,232
 vellum/client/resources/workflow_sandboxes/types/list_workflow_sandbox_examples_request_tag.py,sha256=TEwWit20W3X-zWPPLAhmUG05UudG9gaBSJ4Q4-rNJws,188
 vellum/client/resources/workflows/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
@@ -245,7 +245,7 @@ vellum/client/resources/workspace_secrets/raw_client.py,sha256=ZfiNd1NisolmK07QP
 vellum/client/resources/workspaces/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 vellum/client/resources/workspaces/client.py,sha256=36KYa2FDu6h65q2GscUFOJs4qKeiOA6grOYoCc0Gi3E,2936
 vellum/client/resources/workspaces/raw_client.py,sha256=M3Ewk1ZfEZ44EeTvBtBNoNKi5whwfLY-1GR07SyfDTI,3517
-vellum/client/types/__init__.py,sha256=K33OdhOUq5W7oh8cdwsFfo-aRHxPa6Jel9rlBQS2at0,71137
+vellum/client/types/__init__.py,sha256=-PQEF1IO-RdF7X4YeQf2ZO-OCPvy0pYxt11__WpVxso,71638
 vellum/client/types/ad_hoc_execute_prompt_event.py,sha256=B69EesIH6fpNsdoiJaSG9zF1Sl17FnjoTu4CBkUSoHk,608
 vellum/client/types/ad_hoc_expand_meta.py,sha256=Kajcj3dKKed5e7uZibRnE3ZonK_bB2JPM-3aLjLfUp4,1295
 vellum/client/types/ad_hoc_fulfilled_prompt_execution_meta.py,sha256=5kD6ZcbU8P8ynK0lMD8Mh7vHzvQt06ziMyphvWYg6FU,968
@@ -325,7 +325,7 @@ vellum/client/types/delimiter_chunking_request.py,sha256=zZ1Y-hKzM9ie2kzuTjKain2
 vellum/client/types/deployment_history_item.py,sha256=E48hjXMKGQfpf4Shw1_WLk2i9IjKKAh4ntDzKmbLJ6s,1225
 vellum/client/types/deployment_provider_payload_response.py,sha256=VIpzush2vTmhurLs9Jf16mo19je2WoKyLJxEHwVeKtk,782
 vellum/client/types/deployment_provider_payload_response_payload.py,sha256=xHLQnWFN0AZRZdrOiKawwpoKK7BTmnZfp0P7FCc2ZqE,188
-vellum/client/types/deployment_read.py,sha256=CzudxnNjP2uoL3bZrJ0XwrMJzqOTkNJLbYl7hucnfy0,2063
+vellum/client/types/deployment_read.py,sha256=gfkVoVD84glJuUFDX61uMwmhDYg46h2UXcXIhTEA-ls,2040
 vellum/client/types/deployment_release_tag_deployment_history_item.py,sha256=wVzgUnsa4Up9ReH_rP_l5ssfA0J1Nsfd4AN6lRsxhv4,595
 vellum/client/types/deployment_release_tag_read.py,sha256=SV_kKYFIjhfD5CR9Ab35pdJpAJSx-7jhk2x_HUiz-Uk,1242
 vellum/client/types/docker_service_token.py,sha256=rv0F19r2Sgba_Hbz757HFfTV8UioAQTV2xXkWxfn4sg,578
@@ -658,11 +658,11 @@ vellum/client/types/sentence_chunker_config.py,sha256=WQGyc0jUiATYhQ7ezbBN06--ul
 vellum/client/types/sentence_chunker_config_request.py,sha256=Da2ANuD4icLxI74emC4z2EEZjeY2k1w9HnEQt2g1AvI,680
 vellum/client/types/sentence_chunking.py,sha256=MyEImiN0dhWEVFvD76Q6mCqIEjZcyY__Hhu0j9Gjs6Y,747
 vellum/client/types/sentence_chunking_request.py,sha256=itrdqiKJJ7UKWVGTfj9sJu3kMSThkBF_Qj-3KqFunP4,776
-vellum/client/types/slim_deployment_read.py,sha256=yg1T-Ly642S3_ozdZuhMPy0lLJjH7nvHlbg_-IH0Ypc,1744
+vellum/client/types/slim_deployment_read.py,sha256=lEJ8Fx7QSmqjEK-LcR0y_hX2ywcaPbYjxINwPGy_ofw,1733
 vellum/client/types/slim_document.py,sha256=2eSbvk2u8NsFsT3SPmDRvdBJsytJ8zPAVhuTsC-hF8E,2359
 vellum/client/types/slim_document_document_to_document_index.py,sha256=Ogo8OFjtmcHjDmIynB9vckfjSZxFNhKM-cYpqSfREGI,1515
 vellum/client/types/slim_release_review.py,sha256=vWNkPXk5wZ_scHsWHz_77PfMZRDn-4qUkqVbCKqY1zQ,747
-vellum/client/types/slim_workflow_deployment.py,sha256=Gm3zwOXx-7Pwm8bh8lgmNsJ8zsZiGUKvDQkNJx9I5qI,2103
+vellum/client/types/slim_workflow_deployment.py,sha256=jIciGPCW9QNtDRdq3w_zUdrrE4cg1LWkcoyGM-L6cs0,2085
 vellum/client/types/slim_workflow_execution_read.py,sha256=6Ep2iQ6tWTCGxzvAh4t3G_4BjHQ3_JZ9WFdc81XGqkA,2302
 vellum/client/types/span_link.py,sha256=roO93-xRnqZy2hx9WJFNZW46o8l2bGf3ugeaBepegr4,1433
 vellum/client/types/span_link_type_enum.py,sha256=SQsPKp1Jb8PWE2tkCHt9d5fZP8Oz77vG8_VYKF-h9J4,186
@@ -824,7 +824,7 @@ vellum/client/types/video_vellum_value_request.py,sha256=q6dptI5XdJNFwtTRlNyPqU0
 vellum/client/types/workflow_deployment_event_executions_response.py,sha256=KXzRC0iaXsBu9ezA4p7PIHL4-IWIQ-lX6NW6noTHtZ4,1570
 vellum/client/types/workflow_deployment_history_item.py,sha256=RF_ByBLXRo47MhAuyUV4FkxmN_j879-UylfRyZaYAm0,1327
 vellum/client/types/workflow_deployment_parent_context.py,sha256=1JdTw3LL6tgCgiMVrX35SZVu7mZcxz6SbxWPCUBZ09s,1796
-vellum/client/types/workflow_deployment_read.py,sha256=dLot3_qIrajmCdDVKSHm7fq5wCwjcQe7gSn8BgcM0kA,2277
+vellum/client/types/workflow_deployment_read.py,sha256=rQQZmRy92tfXie8kg14dRI9t_lZ1StbXSEtSFPMTV9E,2247
 vellum/client/types/workflow_deployment_release.py,sha256=yd9Okgpdm8Oob9VmIKaksZjmRv2uLHGJtc-H5sO10F8,1520
 vellum/client/types/workflow_deployment_release_workflow_deployment.py,sha256=ZmQhemAQEqARhkSvc9DFdgfTKQfSHvRTS9reW1Wx9dI,550
 vellum/client/types/workflow_deployment_release_workflow_version.py,sha256=VbVrwsDHapwPSsGEt8OM9Gjs31esIqkp8ta5zrZvPTw,885
@@ -883,11 +883,15 @@ vellum/client/types/workflow_push_exec_config.py,sha256=6TaVMVqhSOz4DnY46l8axPDt
 vellum/client/types/workflow_push_response.py,sha256=Tkwpvh_Ar1yl-kyClh2fTIP2e0cqyk1ohbArIp8I9gs,688
 vellum/client/types/workflow_release_tag_read.py,sha256=NIZWhCVQ3UhxkpjuyoZ2nY9WHuUr4WSlFvqTQ4JStR4,1259
 vellum/client/types/workflow_release_tag_workflow_deployment_history_item.py,sha256=we8Qj3udlnbOtIz5hZRpGgtxIxjr_2OSleaNutuB3f0,601
+vellum/client/types/workflow_request_audio_input_request.py,sha256=wKTsNDj4cvvr9mkGlACq1hdiDtffc5IdpvJmBjRyw3s,823
 vellum/client/types/workflow_request_chat_history_input_request.py,sha256=JiiqxniYKIrwOhQAVUHkffNx6YTqWBhMmwg5Pf-6e-Y,862
-vellum/client/types/workflow_request_input_request.py,sha256=REr7j_FA1ugRt-9-2EWm5OQrj7hsReetPaNDMTS57HA,633
+vellum/client/types/workflow_request_document_input_request.py,sha256=MWDi6Ttd62KK2uSkNQdMgf-l8QRlHepA6tTz5YH9BVU,843
+vellum/client/types/workflow_request_image_input_request.py,sha256=NNlkMhYlz3ZzfMx223T9UleMDpeIFrgIJpT2jZAjlWM,823
+vellum/client/types/workflow_request_input_request.py,sha256=CVLyzs3rfFpx_UR-guH2jr-Mp3_PYedTGBZ3F0L6_-Y,1126
 vellum/client/types/workflow_request_json_input_request.py,sha256=LZu_IblkMhk5JWAcfaUYEv4TV2FeAkd4zCjPnZbFUPM,781
 vellum/client/types/workflow_request_number_input_request.py,sha256=1eesPBjUA7wKWLR90x6in3U9H1ZuIt_2_OqIYjy-V9s,760
 vellum/client/types/workflow_request_string_input_request.py,sha256=oqdelrQTkpCH5d6tN6ytaTFVkBHJynl_bF8a_Grokh4,758
+vellum/client/types/workflow_request_video_input_request.py,sha256=SAZ-tX5w7Cfx_u5HueZhN9BUv3K9BLlz0nnUbNc9QrE,822
 vellum/client/types/workflow_result_event.py,sha256=LRA5i2sE9cZYQwZSygp7MnKXvHP8dB9gWVVxWY2w1f0,1343
 vellum/client/types/workflow_result_event_output_data.py,sha256=S_S_h42LMnqJv9FLEEASFS3HsqX2RZHT_kQeeWOSj28,1222
 vellum/client/types/workflow_result_event_output_data_array.py,sha256=pjEc56lhN5uTtB2DLXDkzH6mwj4glGNNXKG_dCAG_Io,1318
@@ -1661,11 +1665,15 @@ vellum/types/workflow_push_exec_config.py,sha256=UCpZQx6no5gyaf-jNhqcKNqGyQ8xoKQ
 vellum/types/workflow_push_response.py,sha256=3R8pxlK0HXaqJFMQ3ImohWmpAfPpLn-jaHmkc1lLA6g,160
 vellum/types/workflow_release_tag_read.py,sha256=n4Rd4Ju2LYtE_ESjaVImf4ybpxzWWkzfuwl9ynW5xZ8,163
 vellum/types/workflow_release_tag_workflow_deployment_history_item.py,sha256=slQ8VEaWoC9JEJUNENmiDaLMLCKYqZa3Oggaw6x7TGI,191
+vellum/types/workflow_request_audio_input_request.py,sha256=PZ0kscIMggI1T4ihIHIZhU9SDlNpt3JLp1CB-NqJasc,174
 vellum/types/workflow_request_chat_history_input_request.py,sha256=eyyTyEIdhYpBzd4boLyonevA0x3QF8bkFQPTNgbYioo,181
+vellum/types/workflow_request_document_input_request.py,sha256=bGKzwRm2a6UgtUvHPtPFrmiscqt_NR2hP-VDFckLvfM,177
+vellum/types/workflow_request_image_input_request.py,sha256=5SPnlRRmRw3QHJsjY-UIkAN_Rd74XN6V1UsxyolbNPs,174
 vellum/types/workflow_request_input_request.py,sha256=SVJ5Zu-uuEpFEqoJwEnuHJz1ldpBm-W40b02HPE9dNg,168
 vellum/types/workflow_request_json_input_request.py,sha256=fkeZTjEns0R1UdA1oX8DWEmIMrwYgSy8ia0rJyOUb74,173
 vellum/types/workflow_request_number_input_request.py,sha256=C5GareD-9BxO_LkcWnLTaUx_O7Z3iGy760gvrT2_Iyo,175
 vellum/types/workflow_request_string_input_request.py,sha256=ivQxu4db3dB7tlDlD8GUiVpn6IbsPVsuxdXeA2Jtk-4,175
+vellum/types/workflow_request_video_input_request.py,sha256=x4f5IMo8EaLGz0gckHBl-1bXHP9I5QrL8A-wBa11FYc,174
 vellum/types/workflow_result_event.py,sha256=JifChSMV7_5gIVYwHlvOxe2JD-HfSWBfTsCQghpO0yU,159
 vellum/types/workflow_result_event_output_data.py,sha256=jfl2_Oo21LmC1HOsZpt2Xkm6NhsW_DY9gt1EUfxOdJU,171
 vellum/types/workflow_result_event_output_data_array.py,sha256=MnUSnSZgUdFSOblnHhV6juyLEDvanINvmXCo7ixwZVw,177
@@ -1718,8 +1726,8 @@ vellum/workflows/events/stream.py,sha256=xhXJTZirFi0xad5neAQNogrIQ4h47fpnKbVC3vC
 vellum/workflows/events/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/events/tests/test_basic_workflow.py,sha256=Pj6orHsXz37jWC5FARi0Sx2Gjf99Owri2Cvr6Chb79k,1765
 vellum/workflows/events/tests/test_event.py,sha256=c2pM8ZOuES_8XjLYP4cU4cChrZUZ1ZZ3HUNIaaPAxXk,18411
-vellum/workflows/events/types.py,sha256=5iU8KmS_fl0l42XBQshiNLPl2yLdPMEQKQ8nYP--0L0,4940
-vellum/workflows/events/workflow.py,sha256=gjg4Dt5i40jaTMK_-AaGtW9lzX1JGn3FUAu1XqhduJ8,8753
+vellum/workflows/events/types.py,sha256=mVrqAH9Hs9SpXm08Hcxdyap_ImQPwGsxRr56rSNMP34,5043
+vellum/workflows/events/workflow.py,sha256=rAj2jH94t-bWqtQRtG8mkxXpQxIfqux06QtKdFK_L54,8960
 vellum/workflows/exceptions.py,sha256=NiBiR3ggfmPxBVqD-H1SqmjI-7mIn0EStSN1BqApvCM,1213
 vellum/workflows/expressions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/expressions/accessor.py,sha256=3lu1-_-dBfZdJvtX-q66jbmRAZtqIfdsh_3_JNuzg1E,4462
@@ -1776,9 +1784,9 @@ vellum/workflows/inputs/tests/test_inputs.py,sha256=lioA8917mFLYq7Ml69UNkqUjcWbb
 vellum/workflows/integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/integrations/composio_service.py,sha256=rSliaZtNiBcDSvDxz9k5i1KkyUIrbxyegu0yU9cDByU,6023
 vellum/workflows/integrations/mcp_service.py,sha256=9DYb8dg2_kgc1UOu830kxhaFlt9yTbhKPhK3L6kb1t4,9831
-vellum/workflows/integrations/tests/test_mcp_service.py,sha256=xQVtcqRbgq6Bx4pyxLdgUtZFNGISPfISvpXcyd0U6Hc,4595
+vellum/workflows/integrations/tests/test_mcp_service.py,sha256=q_DYrDkIqI4sQBNgID4YdbM4e9tneLVWY8YmI4R26d8,8859
 vellum/workflows/logging.py,sha256=_a217XogktV4Ncz6xKFz7WfYmZAzkfVRVuC0rWob8ls,437
-vellum/workflows/nodes/__init__.py,sha256=aVdQVv7Y3Ro3JlqXGpxwaU2zrI06plDHD2aumH5WUIs,1157
+vellum/workflows/nodes/__init__.py,sha256=zymtc3_iW2rFmMR-sayTLuN6ZsAw8VnJweWPsjQk2-Q,1197
 vellum/workflows/nodes/bases/__init__.py,sha256=cniHuz_RXdJ4TQgD8CBzoiKDiPxg62ErdVpCbWICX64,58
 vellum/workflows/nodes/bases/base.py,sha256=9vOPHP6K6oK36bEby1KFRXpMevMwq6slwuSRhC-bWFc,20752
 vellum/workflows/nodes/bases/base_adornment_node.py,sha256=hrgzuTetM4ynPd9YGHoK8Vwwn4XITi3aZZ_OCnQrq4Y,3433
@@ -1807,7 +1815,7 @@ vellum/workflows/nodes/core/try_node/__init__.py,sha256=JVD4DrldTIqFQQFrubs9KtWC
 vellum/workflows/nodes/core/try_node/node.py,sha256=XdyOvlwQ3m4h0-_WNtaBl2t_CdlzPXclulkLOtUcX3E,4388
 vellum/workflows/nodes/core/try_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/nodes/core/try_node/tests/test_node.py,sha256=h6eUc3SggvhzBWlOD0PrPUlkoCSQHwjqYn81VkxSIxU,4948
-vellum/workflows/nodes/displayable/__init__.py,sha256=zH7SFotr4i8sO-r5_k53yPipQwDouDmHxTfCE6dXAoU,1093
+vellum/workflows/nodes/displayable/__init__.py,sha256=7U82xW397jdeo0910ksf_Twx6etn5xtiKcZIsB9TOE8,1157
 vellum/workflows/nodes/displayable/api_node/__init__.py,sha256=MoxdQSnidIj1Nf_d-hTxlOxcZXaZnsWFDbE-PkTK24o,56
 vellum/workflows/nodes/displayable/api_node/node.py,sha256=F7ucsuEmrVYlTKMIoi60fFJ_ELYgGuc7jEmJCEyQczw,2956
 vellum/workflows/nodes/displayable/api_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1880,7 +1888,11 @@ vellum/workflows/nodes/displayable/tool_calling_node/tests/__init__.py,sha256=47
 vellum/workflows/nodes/displayable/tool_calling_node/tests/test_composio_service.py,sha256=in1fbEz5x1tx3uKv9YXdvOncsHucNL8Ro6Go7lBuuOQ,8962
 vellum/workflows/nodes/displayable/tool_calling_node/tests/test_node.py,sha256=GZoeybB9uM7ai8sBLAtUMHrMVgh-WrJDWrKZci6feDs,11892
 vellum/workflows/nodes/displayable/tool_calling_node/tests/test_utils.py,sha256=SIu5GCj4tIE4fz-cAcdULtQfqZIhrcc3Doo6TWLXBws,8804
-vellum/workflows/nodes/displayable/tool_calling_node/utils.py,sha256=7pM6MH3Z7r0k7Kpr3d-26QXUo_4GrxLflKwFTxRPREY,22844
+vellum/workflows/nodes/displayable/tool_calling_node/utils.py,sha256=ACEfl2pY65sO0DaMYxuecx2Kxg32HpSh7XXpp8U1J84,23459
+vellum/workflows/nodes/displayable/web_search_node/__init__.py,sha256=8FOnEP-n-U68cvxTlJW9wphIAGHq5aqjzLM-DoSSXnU,61
+vellum/workflows/nodes/displayable/web_search_node/node.py,sha256=NQYux2bOtuBF5E4tn-fXi5y3btURPRrNqMSM9MAZYI4,5091
+vellum/workflows/nodes/displayable/web_search_node/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vellum/workflows/nodes/displayable/web_search_node/tests/test_node.py,sha256=H3titxCYFsfIJ3EDeRL6Pe9I-tdKMRt1axOwb2qIw6U,10983
 vellum/workflows/nodes/experimental/README.md,sha256=eF6DfIL8t-HbF9-mcofOMymKrraiBHDLKTlnBa51ZiE,284
 vellum/workflows/nodes/experimental/__init__.py,sha256=jCQgvZEknXKfuNhGSOou4XPfrPqZ1_XBj5F0n0fgiWM,106
 vellum/workflows/nodes/experimental/openai_chat_completion_node/__init__.py,sha256=lsyD9laR9p7kx5-BXGH2gUTM242UhKy8SMV0SR6S2iE,90
@@ -1910,15 +1922,16 @@ vellum/workflows/references/tests/test_lazy.py,sha256=0s50-LizMTlSTBQahpK0fg_xqC
 vellum/workflows/references/vellum_secret.py,sha256=Od4d19a5yletWMqNfJR5d_mZQUkVcFzj29mE-T9J7yE,480
 vellum/workflows/references/workflow_input.py,sha256=W3rOK1EPd2gYHb04WJwmNm1CUSdvZ9LKrs8RMKxACBs,1751
 vellum/workflows/resolvers/__init__.py,sha256=eH6hTvZO4IciDaf_cf7aM2vs-DkBDyJPycOQevJxQnI,82
-vellum/workflows/resolvers/base.py,sha256=1jwWJPE0CVJdYJTZBd2ktscl-TK1WNgr4V95Q0ATH0o,1078
-vellum/workflows/resolvers/resolver.py,sha256=4A3e-Kr65VEPQedn-q_Mne-Ln7JvK95rM35K2kUsqB8,1463
-vellum/workflows/resolvers/tests/test_resolver.py,sha256=U4LKQ6-UBU-7nJ39uSkxW9Y5n17tPh5ZTlSYb4jzFjY,2138
+vellum/workflows/resolvers/base.py,sha256=wrQiSC02Bw4-dBwgFjJIHsjpe-4xz4rUJs_1RdErKA0,1164
+vellum/workflows/resolvers/resolver.py,sha256=0Y9ArmN_2Gu62TpgYjFvSzhFMMFy7oGWOJh9zg3jaFk,4085
+vellum/workflows/resolvers/tests/test_resolver.py,sha256=JX6-zjIQShNvt4XYmub6zmY0iUf6z90KZynqvEd33qA,5202
+vellum/workflows/resolvers/types.py,sha256=Hndhlk69g6EKLh_LYg5ILepW5U_h_BYNllfzhS9k8p4,237
 vellum/workflows/runner/__init__.py,sha256=i1iG5sAhtpdsrlvwgH6B-m49JsINkiWyPWs8vyT-bqM,72
-vellum/workflows/runner/runner.py,sha256=sADrp593ia4auJ8fIP6jc5na3b5WaTvvBdjpkTthboY,36929
+vellum/workflows/runner/runner.py,sha256=jdFFDr-MCVzTm5NKEzOgrw_9osuwdiVk97SRo73uLRQ,39366
 vellum/workflows/sandbox.py,sha256=jwlFFQjHDwmbVoBah_Q3i8K_BrzOt-F6TXFauiyVyIk,3021
 vellum/workflows/state/__init__.py,sha256=yUUdR-_Vl7UiixNDYQZ-GEM_kJI9dnOia75TtuNEsnE,60
 vellum/workflows/state/base.py,sha256=m9fCqbZn21GshCVCjJTD1dPZEQjFrsMXqlg7tM9fIwM,24283
-vellum/workflows/state/context.py,sha256=xcDe7FB_omse3lLCZkWigcEQvk-hnhruENItaqXNUGo,6956
+vellum/workflows/state/context.py,sha256=khM30U1iDNts5Xp8LXa_WfpkITNITexrDUUFJ5wZ2W4,8445
 vellum/workflows/state/delta.py,sha256=7h8wR10lRCm15SykaPj-gSEvvsMjCwYLPsOx3nsvBQg,440
 vellum/workflows/state/encoder.py,sha256=HdNlabmBOcFSeY_dgn4LNtQEugyzsw3p4mvn2ChC1Io,3380
 vellum/workflows/state/store.py,sha256=uVe-oN73KwGV6M6YLhwZMMUQhzTQomsVfVnb8V91gVo,1147
@@ -1949,15 +1962,16 @@ vellum/workflows/utils/tests/test_uuids.py,sha256=i77ABQ0M3S-aFLzDXHJq_yr5FPkJEW
 vellum/workflows/utils/tests/test_vellum_variables.py,sha256=vbnKgm41aB5OXlO-ZIPbhQ6xDiZkT8KMxCLqz4zocWY,1791
 vellum/workflows/utils/uuids.py,sha256=IaZQANz7fhF7la0_J1O50Y6D2PIYv_taRDTRzBT9aWw,1284
 vellum/workflows/utils/vellum_variables.py,sha256=-CohqD3AeyCYzqwyrPZ7mt_lt7ibWLyx0MuS484feJk,5503
+vellum/workflows/utils/zip.py,sha256=HVg_YZLmBOTXKaDV3Xhaf3V6sYnfqqZXQ8CpuafkbPY,1181
 vellum/workflows/vellum_client.py,sha256=xkfoucodxNK5JR2-lbRqZx3xzDgExWkP6kySrpi_Ubc,1079
 vellum/workflows/workflows/__init__.py,sha256=KY45TqvavCCvXIkyCFMEc0dc6jTMOUci93U2DUrlZYc,66
-vellum/workflows/workflows/base.py,sha256=-hu_s2LQcV7zeBumMHFwezD4ctV7VUvDI4mpMdNRkCI,27488
+vellum/workflows/workflows/base.py,sha256=PT7Hveate5A4Nt1UQoS6wwFehxgZQ5fuYjS1vLnlkl0,28008
 vellum/workflows/workflows/event_filters.py,sha256=GSxIgwrX26a1Smfd-6yss2abGCnadGsrSZGa7t7LpJA,2008
 vellum/workflows/workflows/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vellum/workflows/workflows/tests/test_base_workflow.py,sha256=ptMntHzVyy8ZuzNgeTuk7hREgKQ5UBdgq8VJFSGaW4Y,20832
 vellum/workflows/workflows/tests/test_context.py,sha256=VJBUcyWVtMa_lE5KxdhgMu0WYNYnUQUDvTF7qm89hJ0,2333
-vellum_ai-1.2.5.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
-vellum_ai-1.2.5.dist-info/METADATA,sha256=pez3o4cs4bFWAYjNMO2FyJR942rNI_Ht_KzWysTXyKU,5547
-vellum_ai-1.2.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-vellum_ai-1.2.5.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
-vellum_ai-1.2.5.dist-info/RECORD,,
+vellum_ai-1.3.1.dist-info/LICENSE,sha256=hOypcdt481qGNISA784bnAGWAE6tyIf9gc2E78mYC3E,1574
+vellum_ai-1.3.1.dist-info/METADATA,sha256=8eP8UQcNY66V3X1QK8GmDgvQnyi_hHztrZQNUMTCT1M,5547
+vellum_ai-1.3.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+vellum_ai-1.3.1.dist-info/entry_points.txt,sha256=HCH4yc_V3J_nDv3qJzZ_nYS8llCHZViCDP1ejgCc5Ak,42
+vellum_ai-1.3.1.dist-info/RECORD,,

vellum_cli/tests/test_init.py

@@ -1,31 +1,14 @@
-import io
 import json
 import os
 from unittest.mock import patch
-import zipfile
 
 from click.testing import CliRunner
 from pydash import snake_case
 
+from vellum.workflows.utils.zip import zip_file_map
 from vellum_cli import main as cli_main
 
 
-def _zip_file_map(file_map: dict[str, str]) -> bytes:
-    # Create an in-memory bytes buffer to store the zip
-    zip_buffer = io.BytesIO()
-
-    # Create zip file and add files from file_map
-    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
-        for filename, content in file_map.items():
-            zip_file.writestr(filename, content)
-
-    # Get the bytes from the buffer
-    zip_bytes = zip_buffer.getvalue()
-    zip_buffer.close()
-
-    return zip_bytes
-
-
 class MockTemplate:
     def __init__(self, id, label):
         self.id = id
@@ -44,7 +27,7 @@ def test_init_command(vellum_client, mock_module):
     vellum_client.workflow_sandboxes.list_workflow_sandbox_examples.return_value.results = fake_templates
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the `init` command and selects the first template
     runner = CliRunner()
@@ -173,7 +156,7 @@ def test_init_command_target_directory_exists(vellum_client, mock_module):
     vellum_client.workflow_sandboxes.list_workflow_sandbox_examples.return_value.results = fake_templates
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the `init` command and selects the template
     runner = CliRunner()
@@ -214,7 +197,7 @@ def test_init_command_with_template_name(vellum_client, mock_module):
     vellum_client.workflow_sandboxes.list_workflow_sandbox_examples.return_value.results = fake_templates
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the `init` command with a specific template name
     template_name = snake_case("Another Workflow")
@@ -309,7 +292,7 @@ def test_init__with_target_dir(vellum_client, mock_module):
     vellum_client.workflow_sandboxes.list_workflow_sandbox_examples.return_value.results = fake_templates
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # AND a target directory
     target_dir = os.path.join(temp_dir, "dir")
@@ -364,7 +347,7 @@ def test_init__with_nested_target_dir(vellum_client, mock_module):
     vellum_client.workflow_sandboxes.list_workflow_sandbox_examples.return_value.results = fake_templates
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # AND a nested target directory that doesn't exist yet
     nested_target_dir = os.path.join(temp_dir, "dir-1", "dir-2")
@@ -422,7 +405,7 @@ def test_init__with_template_name_and_target_dir(vellum_client, mock_module):
     vellum_client.workflow_sandboxes.list_workflow_sandbox_examples.return_value.results = fake_templates
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # AND a target directory
     target_dir = os.path.join(temp_dir, "dir")

vellum_cli/tests/test_pull.py

@@ -1,34 +1,17 @@
 import pytest
-import io
 import json
 import os
 import tempfile
 from unittest import mock
 from uuid import uuid4
-import zipfile
 
 from click.testing import CliRunner
 
 from vellum.client.core.api_error import ApiError
+from vellum.workflows.utils.zip import zip_file_map
 from vellum_cli import main as cli_main
 
 
-def _zip_file_map(file_map: dict[str, str]) -> bytes:
-    # Create an in-memory bytes buffer to store the zip
-    zip_buffer = io.BytesIO()
-
-    # Create zip file and add files from file_map
-    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
-        for filename, content in file_map.items():
-            zip_file.writestr(filename, content)
-
-    # Get the bytes from the buffer
-    zip_bytes = zip_buffer.getvalue()
-    zip_buffer.close()
-
-    return zip_bytes
-
-
 @pytest.mark.parametrize(
     "base_command",
     [
@@ -44,7 +27,7 @@ def test_pull(vellum_client, mock_module, base_command):
     workflow_sandbox_id = mock_module.workflow_sandbox_id
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the pull command
     runner = CliRunner()
@@ -89,7 +72,7 @@ def test_pull__second_module(vellum_client, mock_module):
     set_pyproject_toml = mock_module.set_pyproject_toml
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # AND the module we're about to pull is configured second
     set_pyproject_toml(
@@ -134,7 +117,7 @@ def test_pull__with_target_dir(vellum_client, mock_module, base_command):
     os.makedirs(target_dir, exist_ok=True)
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the pull command with target-dir
     runner = CliRunner()
@@ -195,7 +178,7 @@ def test_pull__with_nested_target_dir(vellum_client, mock_module, base_command):
     nested_target_dir = os.path.join(temp_dir, "dir-1", "dir-2")
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the pull command with nested target-dir
     runner = CliRunner()
@@ -250,11 +233,7 @@ def test_pull__sandbox_id_with_no_config(vellum_client):
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [
-            _zip_file_map(
-                {"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})}
-            )
-        ]
+        [zip_file_map({"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})})]
     )
 
     # AND we are currently in a new directory
@@ -309,11 +288,7 @@ def test_pull__sandbox_id_with_other_workflow_configured(vellum_client, mock_mod
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [
-            _zip_file_map(
-                {"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})}
-            )
-        ]
+        [zip_file_map({"workflow.py": "print('hello')", "metadata.json": json.dumps({"label": "Super Cool Workflow"})})]
     )
 
     # WHEN the user runs the pull command with the new workflow sandbox id
@@ -343,7 +318,7 @@ def test_pull__workflow_deployment_with_no_config(vellum_client):
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -414,7 +389,7 @@ def test_pull__both_workflow_sandbox_id_and_deployment(vellum_client):
     workflow_deployment = "my-deployment"
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # AND we are currently in a new directory
     current_dir = os.getcwd()
@@ -447,7 +422,7 @@ def test_pull__remove_missing_files(vellum_client, mock_module):
     module = mock_module.module
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # AND there is already a different file in the module directory
     other_file_path = os.path.join(temp_dir, *module.split("."), "other_file.py")
@@ -478,7 +453,7 @@ def test_pull__remove_missing_files__ignore_pattern(vellum_client, mock_module):
     set_pyproject_toml = mock_module.set_pyproject_toml
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # AND there is already a different file in the module directory
     other_file_path = os.path.join(temp_dir, *module.split("."), "other_file.py")
@@ -530,7 +505,7 @@ def test_pull__include_json(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
+        [zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
     )
 
     # WHEN the user runs the pull command
@@ -552,7 +527,7 @@ def test_pull__exclude_code(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
+        [zip_file_map({"workflow.py": "print('hello')", "workflow.json": "{}"})]
     )
 
     # WHEN the user runs the pull command
@@ -606,7 +581,7 @@ def test_pull__sandbox_id_with_other_workflow_deployment_in_lock(vellum_client,
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -671,7 +646,7 @@ def test_pull__handle_error_log(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file with an error log
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "error.log": "test error"})]
+        [zip_file_map({"workflow.py": "print('hello')", "error.log": "test error"})]
     )
 
     # WHEN the user runs the pull command with the new workflow sandbox id
@@ -693,7 +668,7 @@ def test_pull__strict__with_error(vellum_client, mock_module):
     workflow_sandbox_id = mock_module.workflow_sandbox_id
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the pull command with the new workflow sandbox id
     runner = CliRunner()
@@ -715,7 +690,7 @@ def test_pull__include_sandbox(vellum_client, mock_module):
 
     # AND the workflow pull API call returns a zip file
     vellum_client.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')", "sandbox.py": "print('hello')"})]
+        [zip_file_map({"workflow.py": "print('hello')", "sandbox.py": "print('hello')"})]
     )
 
     # WHEN the user runs the pull command
@@ -744,7 +719,7 @@ def test_pull__same_pull_twice__one_entry_in_lockfile(vellum_client, mock_module
     workflow_sandbox_id = mock_module.workflow_sandbox_id
 
     # AND the workflow pull API call returns a zip file both times
-    zip_contents = _zip_file_map({"workflow.py": "print('hello')"})
+    zip_contents = zip_file_map({"workflow.py": "print('hello')"})
     responses = iter([zip_contents, zip_contents])
 
     def workflows_pull_side_effect(*_args, **_kwargs):
@@ -780,7 +755,7 @@ def test_pull__module_not_in_config(vellum_client, mock_module):
     set_pyproject_toml({"workflows": []})
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the pull command again with the workflow sandbox id and module
     runner = CliRunner()
@@ -833,7 +808,7 @@ def test_pull__multiple_instances_of_same_module__keep_when_pulling_another_modu
         json.dump(lock_data, f)
 
     # AND the workflow pull API call returns a zip file
-    vellum_client.workflows.pull.return_value = iter([_zip_file_map({"workflow.py": "print('hello')"})])
+    vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
 
     # WHEN the user runs the pull command on the new module
     runner = CliRunner()
@@ -856,7 +831,7 @@ def test_pull__module_name_from_deployment_name(vellum_client):
     deployment_name = "Test Deployment"
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps({"deployment_name": deployment_name, "label": "Some Label"}),
@@ -995,7 +970,7 @@ def test_pull__workflow_deployment_adds_deployment_to_config(vellum_client, work
     # AND the workflow pull API call returns a zip file with metadata
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1082,7 +1057,7 @@ def test_pull__workflow_deployment_name_is_uuid(vellum_client):
     updated_label = "Updated Label"
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1170,7 +1145,7 @@ def test_pull__workflow_deployment_updates_existing_deployment(vellum_client, ge
     updated_label = "Updated Label"
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1218,7 +1193,7 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
     # AND the workflow pull API call returns a zip file with metadata
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1264,7 +1239,7 @@ def test_pull__workflow_deployment_with_name_and_id(vellum_client):
     # AND pull with name will not add a new deployment to the config
     vellum_client.workflows.pull.return_value = iter(
         [
-            _zip_file_map(
+            zip_file_map(
                 {
                     "workflow.py": "print('hello')",
                     "metadata.json": json.dumps(
@@ -1323,7 +1298,7 @@ MY_OTHER_VELLUM_API_KEY=aaabbbcccddd
 
     # AND the workflow pull API call returns a zip file
     vellum_client_class.return_value.workflows.pull.return_value = iter(
-        [_zip_file_map({"workflow.py": "print('hello')"})]
+        [zip_file_map({"workflow.py": "print('hello')"})]
     )
 
     # WHEN calling `vellum pull` with --workspace