vellum-ai 0.7.3__py3-none-any.whl → 0.7.5__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (483) hide show
  1. vellum/__init__.py +142 -94
  2. vellum/client.py +1051 -1097
  3. vellum/core/__init__.py +4 -1
  4. vellum/core/client_wrapper.py +7 -3
  5. vellum/core/http_client.py +365 -20
  6. vellum/core/jsonable_encoder.py +3 -0
  7. vellum/core/pydantic_utilities.py +16 -0
  8. vellum/core/query_encoder.py +33 -0
  9. vellum/core/remove_none_from_dict.py +2 -2
  10. vellum/core/request_options.py +2 -2
  11. vellum/resources/__init__.py +2 -0
  12. vellum/resources/deployments/client.py +349 -377
  13. vellum/resources/document_indexes/client.py +542 -610
  14. vellum/resources/documents/client.py +327 -399
  15. vellum/resources/folder_entities/client.py +56 -72
  16. vellum/{types/map_enum.py → resources/ml_models/__init__.py} +0 -3
  17. vellum/resources/ml_models/client.py +837 -0
  18. vellum/resources/sandboxes/client.py +204 -242
  19. vellum/resources/test_suite_runs/client.py +183 -213
  20. vellum/resources/test_suites/client.py +307 -311
  21. vellum/resources/workflow_deployments/client.py +230 -268
  22. vellum/resources/workflow_sandboxes/client.py +82 -96
  23. vellum/terraform/_jsii/vellum-ai_vellum@0.0.0.jsii.tgz +0 -0
  24. vellum/terraform/data_vellum_document_index/__init__.py +10 -10
  25. vellum/terraform/document_index/__init__.py +17 -17
  26. vellum/terraform/provider/__init__.py +57 -12
  27. vellum/terraform/versions.json +1 -1
  28. vellum/types/__init__.py +148 -96
  29. vellum/types/api_node_result.py +7 -3
  30. vellum/types/api_node_result_data.py +7 -3
  31. vellum/types/array_chat_message_content.py +7 -3
  32. vellum/types/array_chat_message_content_item.py +53 -12
  33. vellum/types/array_chat_message_content_item_request.py +53 -12
  34. vellum/types/array_chat_message_content_request.py +7 -3
  35. vellum/types/array_parameter_config.py +50 -0
  36. vellum/types/array_parameter_config_request.py +50 -0
  37. vellum/types/array_variable_value_item.py +102 -24
  38. vellum/types/array_vellum_value_item.py +102 -24
  39. vellum/types/array_vellum_value_item_request.py +102 -24
  40. vellum/types/basic_vectorizer_intfloat_multilingual_e_5_large.py +7 -3
  41. vellum/types/basic_vectorizer_intfloat_multilingual_e_5_large_request.py +7 -3
  42. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1.py +7 -3
  43. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request.py +7 -3
  44. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1.py +7 -3
  45. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request.py +7 -3
  46. vellum/types/boolean_parameter_config.py +35 -0
  47. vellum/types/boolean_parameter_config_request.py +35 -0
  48. vellum/types/chat_history_input_request.py +7 -3
  49. vellum/types/chat_message.py +7 -3
  50. vellum/types/chat_message_content.py +70 -16
  51. vellum/types/chat_message_content_request.py +70 -16
  52. vellum/types/chat_message_request.py +7 -3
  53. vellum/types/code_execution_node_array_result.py +7 -3
  54. vellum/types/code_execution_node_chat_history_result.py +7 -3
  55. vellum/types/code_execution_node_error_result.py +7 -3
  56. vellum/types/code_execution_node_function_call_result.py +7 -3
  57. vellum/types/code_execution_node_json_result.py +7 -3
  58. vellum/types/code_execution_node_number_result.py +7 -3
  59. vellum/types/code_execution_node_result.py +7 -3
  60. vellum/types/code_execution_node_result_data.py +7 -3
  61. vellum/types/code_execution_node_result_output.py +144 -32
  62. vellum/types/code_execution_node_search_results_result.py +7 -3
  63. vellum/types/code_execution_node_string_result.py +7 -3
  64. vellum/types/compile_prompt_deployment_expand_meta_request.py +7 -3
  65. vellum/types/compile_prompt_meta.py +7 -3
  66. vellum/types/conditional_node_result.py +7 -3
  67. vellum/types/conditional_node_result_data.py +7 -3
  68. vellum/types/const_parameter_config.py +35 -0
  69. vellum/types/const_parameter_config_request.py +35 -0
  70. vellum/types/create_test_suite_test_case_request.py +7 -3
  71. vellum/types/deployment_provider_payload_response.py +7 -3
  72. vellum/types/deployment_read.py +12 -3
  73. vellum/types/deployment_release_tag_deployment_history_item.py +7 -3
  74. vellum/types/deployment_release_tag_read.py +7 -3
  75. vellum/types/document_document_to_document_index.py +7 -3
  76. vellum/types/document_index_chunking.py +54 -12
  77. vellum/types/document_index_chunking_request.py +54 -12
  78. vellum/types/document_index_indexing_config.py +7 -3
  79. vellum/types/document_index_indexing_config_request.py +7 -3
  80. vellum/types/document_index_read.py +7 -3
  81. vellum/types/document_read.py +7 -3
  82. vellum/types/enriched_normalized_completion.py +7 -3
  83. vellum/types/error_variable_value.py +7 -3
  84. vellum/types/error_vellum_value.py +7 -3
  85. vellum/types/error_vellum_value_request.py +7 -3
  86. vellum/types/execute_prompt_event.py +82 -16
  87. vellum/types/execute_prompt_response.py +44 -8
  88. vellum/types/execute_workflow_response.py +7 -3
  89. vellum/types/execute_workflow_workflow_result_event.py +41 -8
  90. vellum/types/execution_array_vellum_value.py +7 -3
  91. vellum/types/execution_chat_history_vellum_value.py +7 -3
  92. vellum/types/execution_error_vellum_value.py +7 -3
  93. vellum/types/execution_function_call_vellum_value.py +7 -3
  94. vellum/types/execution_json_vellum_value.py +7 -3
  95. vellum/types/execution_number_vellum_value.py +7 -3
  96. vellum/types/execution_search_results_vellum_value.py +7 -3
  97. vellum/types/execution_string_vellum_value.py +7 -3
  98. vellum/types/execution_vellum_value.py +152 -32
  99. vellum/types/external_test_case_execution.py +7 -3
  100. vellum/types/external_test_case_execution_request.py +7 -3
  101. vellum/types/fulfilled_execute_prompt_event.py +7 -3
  102. vellum/types/fulfilled_execute_prompt_response.py +7 -3
  103. vellum/types/fulfilled_execute_workflow_workflow_result_event.py +7 -3
  104. vellum/types/fulfilled_prompt_execution_meta.py +7 -3
  105. vellum/types/fulfilled_workflow_node_result_event.py +7 -3
  106. vellum/types/function_call.py +7 -3
  107. vellum/types/function_call_chat_message_content.py +7 -3
  108. vellum/types/function_call_chat_message_content_request.py +7 -3
  109. vellum/types/function_call_chat_message_content_value.py +7 -3
  110. vellum/types/function_call_chat_message_content_value_request.py +7 -3
  111. vellum/types/function_call_request.py +7 -3
  112. vellum/types/function_call_variable_value.py +7 -3
  113. vellum/types/function_call_vellum_value.py +7 -3
  114. vellum/types/function_call_vellum_value_request.py +7 -3
  115. vellum/types/generate_options_request.py +7 -3
  116. vellum/types/generate_request.py +7 -3
  117. vellum/types/generate_result.py +7 -3
  118. vellum/types/generate_result_data.py +7 -3
  119. vellum/types/generate_result_error.py +7 -3
  120. vellum/types/generate_stream_response.py +7 -3
  121. vellum/types/generate_stream_result.py +7 -3
  122. vellum/types/generate_stream_result_data.py +7 -3
  123. vellum/types/hkunlp_instructor_xl_vectorizer.py +7 -3
  124. vellum/types/hkunlp_instructor_xl_vectorizer_request.py +7 -3
  125. vellum/types/hosted_by_enum.py +26 -0
  126. vellum/types/hugging_face_tokenizer_config.py +34 -0
  127. vellum/types/hugging_face_tokenizer_config_request.py +34 -0
  128. vellum/types/image_chat_message_content.py +7 -3
  129. vellum/types/image_chat_message_content_request.py +7 -3
  130. vellum/types/image_variable_value.py +7 -3
  131. vellum/types/image_vellum_value.py +7 -3
  132. vellum/types/image_vellum_value_request.py +7 -3
  133. vellum/types/indexing_config_vectorizer.py +117 -36
  134. vellum/types/indexing_config_vectorizer_request.py +117 -36
  135. vellum/types/initiated_execute_prompt_event.py +7 -3
  136. vellum/types/initiated_prompt_execution_meta.py +7 -3
  137. vellum/types/initiated_workflow_node_result_event.py +7 -3
  138. vellum/types/instructor_vectorizer_config.py +7 -3
  139. vellum/types/instructor_vectorizer_config_request.py +7 -3
  140. vellum/types/integer_parameter_config.py +39 -0
  141. vellum/types/integer_parameter_config_request.py +39 -0
  142. vellum/types/json_input_request.py +7 -3
  143. vellum/types/json_variable_value.py +7 -3
  144. vellum/types/json_vellum_value.py +7 -3
  145. vellum/types/json_vellum_value_request.py +7 -3
  146. vellum/types/map_node_result.py +7 -3
  147. vellum/types/map_node_result_data.py +7 -3
  148. vellum/types/merge_node_result.py +7 -3
  149. vellum/types/merge_node_result_data.py +7 -3
  150. vellum/types/metadata_filter_config_request.py +7 -3
  151. vellum/types/metadata_filter_rule_request.py +7 -3
  152. vellum/types/metric_node_result.py +7 -3
  153. vellum/types/{execute_workflow_stream_error_response.py → ml_model_build_config.py} +9 -8
  154. vellum/types/{execute_prompt_api_error_response.py → ml_model_build_config_request.py} +9 -8
  155. vellum/types/ml_model_developer.py +27 -0
  156. vellum/types/ml_model_developer_enum_value_label.py +31 -0
  157. vellum/types/ml_model_display_config_labelled.py +33 -0
  158. vellum/types/ml_model_display_config_request.py +33 -0
  159. vellum/types/ml_model_display_tag.py +7 -0
  160. vellum/types/ml_model_display_tag_enum_value_label.py +31 -0
  161. vellum/types/ml_model_exec_config.py +40 -0
  162. vellum/types/ml_model_exec_config_request.py +40 -0
  163. vellum/types/ml_model_family.py +29 -0
  164. vellum/types/ml_model_family_enum_value_label.py +31 -0
  165. vellum/types/ml_model_feature.py +18 -0
  166. vellum/types/ml_model_parameter_config.py +42 -0
  167. vellum/types/ml_model_parameter_config_request.py +42 -0
  168. vellum/types/ml_model_read.py +108 -0
  169. vellum/types/ml_model_request_authorization_config.py +30 -0
  170. vellum/types/ml_model_request_authorization_config_request.py +30 -0
  171. vellum/types/ml_model_request_authorization_config_type_enum.py +5 -0
  172. vellum/types/ml_model_request_config.py +32 -0
  173. vellum/types/ml_model_request_config_request.py +32 -0
  174. vellum/types/ml_model_response_config.py +31 -0
  175. vellum/types/ml_model_response_config_request.py +31 -0
  176. vellum/types/ml_model_tokenizer_config.py +59 -0
  177. vellum/types/ml_model_tokenizer_config_request.py +61 -0
  178. vellum/types/ml_model_usage.py +7 -3
  179. vellum/types/named_scenario_input_chat_history_variable_value_request.py +7 -3
  180. vellum/types/named_scenario_input_request.py +38 -8
  181. vellum/types/named_scenario_input_string_variable_value_request.py +7 -3
  182. vellum/types/named_test_case_array_variable_value.py +7 -3
  183. vellum/types/named_test_case_array_variable_value_request.py +7 -3
  184. vellum/types/named_test_case_chat_history_variable_value.py +7 -3
  185. vellum/types/named_test_case_chat_history_variable_value_request.py +7 -3
  186. vellum/types/named_test_case_error_variable_value.py +7 -3
  187. vellum/types/named_test_case_error_variable_value_request.py +7 -3
  188. vellum/types/named_test_case_function_call_variable_value.py +7 -3
  189. vellum/types/named_test_case_function_call_variable_value_request.py +7 -3
  190. vellum/types/named_test_case_json_variable_value.py +7 -3
  191. vellum/types/named_test_case_json_variable_value_request.py +7 -3
  192. vellum/types/named_test_case_number_variable_value.py +7 -3
  193. vellum/types/named_test_case_number_variable_value_request.py +7 -3
  194. vellum/types/named_test_case_search_results_variable_value.py +7 -3
  195. vellum/types/named_test_case_search_results_variable_value_request.py +7 -3
  196. vellum/types/named_test_case_string_variable_value.py +7 -3
  197. vellum/types/named_test_case_string_variable_value_request.py +7 -3
  198. vellum/types/named_test_case_variable_value.py +144 -32
  199. vellum/types/named_test_case_variable_value_request.py +144 -32
  200. vellum/types/node_input_compiled_array_value.py +7 -3
  201. vellum/types/node_input_compiled_chat_history_value.py +7 -3
  202. vellum/types/node_input_compiled_error_value.py +7 -3
  203. vellum/types/node_input_compiled_function_call.py +7 -3
  204. vellum/types/node_input_compiled_json_value.py +7 -3
  205. vellum/types/node_input_compiled_number_value.py +7 -3
  206. vellum/types/node_input_compiled_search_results_value.py +7 -3
  207. vellum/types/node_input_compiled_string_value.py +7 -3
  208. vellum/types/node_input_variable_compiled_value.py +152 -32
  209. vellum/types/node_output_compiled_array_value.py +7 -3
  210. vellum/types/node_output_compiled_chat_history_value.py +7 -3
  211. vellum/types/node_output_compiled_error_value.py +7 -3
  212. vellum/types/node_output_compiled_function_call_value.py +7 -3
  213. vellum/types/node_output_compiled_json_value.py +7 -3
  214. vellum/types/node_output_compiled_number_value.py +7 -3
  215. vellum/types/node_output_compiled_search_results_value.py +7 -3
  216. vellum/types/node_output_compiled_string_value.py +7 -3
  217. vellum/types/node_output_compiled_value.py +153 -32
  218. vellum/types/normalized_log_probs.py +7 -3
  219. vellum/types/normalized_token_log_probs.py +7 -3
  220. vellum/types/number_parameter_config.py +40 -0
  221. vellum/types/number_parameter_config_request.py +40 -0
  222. vellum/types/number_variable_value.py +7 -3
  223. vellum/types/number_vellum_value.py +7 -3
  224. vellum/types/number_vellum_value_request.py +7 -3
  225. vellum/types/object_parameter_config.py +49 -0
  226. vellum/types/object_parameter_config_request.py +49 -0
  227. vellum/types/one_of_parameter_config.py +44 -0
  228. vellum/types/one_of_parameter_config_request.py +44 -0
  229. vellum/types/open_ai_vectorizer_config.py +7 -3
  230. vellum/types/open_ai_vectorizer_config_request.py +7 -3
  231. vellum/types/open_ai_vectorizer_text_embedding_3_large.py +7 -3
  232. vellum/types/open_ai_vectorizer_text_embedding_3_large_request.py +7 -3
  233. vellum/types/open_ai_vectorizer_text_embedding_3_small.py +7 -3
  234. vellum/types/open_ai_vectorizer_text_embedding_3_small_request.py +7 -3
  235. vellum/types/open_ai_vectorizer_text_embedding_ada_002.py +7 -3
  236. vellum/types/open_ai_vectorizer_text_embedding_ada_002_request.py +7 -3
  237. vellum/types/paginated_document_index_read_list.py +7 -3
  238. vellum/types/paginated_ml_model_read_list.py +33 -0
  239. vellum/types/paginated_slim_deployment_read_list.py +7 -3
  240. vellum/types/paginated_slim_document_list.py +7 -3
  241. vellum/types/paginated_slim_workflow_deployment_list.py +7 -3
  242. vellum/types/paginated_test_suite_run_execution_list.py +7 -3
  243. vellum/types/paginated_test_suite_test_case_list.py +7 -3
  244. vellum/types/parameter_config.py +251 -0
  245. vellum/types/parameter_config_request.py +251 -0
  246. vellum/types/pdf_search_result_meta_source.py +7 -3
  247. vellum/types/pdf_search_result_meta_source_request.py +7 -3
  248. vellum/types/prompt_deployment_expand_meta_request_request.py +7 -3
  249. vellum/types/prompt_deployment_input_request.py +55 -12
  250. vellum/types/prompt_execution_meta.py +7 -3
  251. vellum/types/prompt_node_execution_meta.py +7 -3
  252. vellum/types/prompt_node_result.py +7 -3
  253. vellum/types/prompt_node_result_data.py +7 -3
  254. vellum/types/prompt_output.py +69 -16
  255. vellum/types/raw_prompt_execution_overrides_request.py +7 -3
  256. vellum/types/reducto_chunker_config.py +7 -3
  257. vellum/types/reducto_chunker_config_request.py +7 -3
  258. vellum/types/reducto_chunking.py +7 -3
  259. vellum/types/reducto_chunking_request.py +7 -3
  260. vellum/types/rejected_execute_prompt_event.py +7 -3
  261. vellum/types/rejected_execute_prompt_response.py +7 -3
  262. vellum/types/rejected_execute_workflow_workflow_result_event.py +7 -3
  263. vellum/types/rejected_prompt_execution_meta.py +7 -3
  264. vellum/types/rejected_workflow_node_result_event.py +7 -3
  265. vellum/types/replace_test_suite_test_case_request.py +7 -3
  266. vellum/types/sandbox_scenario.py +7 -3
  267. vellum/types/scenario_input.py +38 -8
  268. vellum/types/scenario_input_chat_history_variable_value.py +7 -3
  269. vellum/types/scenario_input_string_variable_value.py +7 -3
  270. vellum/types/search_filters_request.py +7 -3
  271. vellum/types/search_node_result.py +7 -3
  272. vellum/types/search_node_result_data.py +7 -3
  273. vellum/types/search_request_options_request.py +7 -3
  274. vellum/types/search_response.py +7 -3
  275. vellum/types/search_result.py +7 -3
  276. vellum/types/search_result_document.py +7 -3
  277. vellum/types/search_result_document_request.py +7 -3
  278. vellum/types/search_result_merging_request.py +7 -3
  279. vellum/types/search_result_meta.py +7 -3
  280. vellum/types/search_result_meta_request.py +7 -3
  281. vellum/types/search_result_meta_source.py +21 -5
  282. vellum/types/search_result_meta_source_request.py +21 -5
  283. vellum/types/search_result_request.py +7 -3
  284. vellum/types/search_weights_request.py +7 -3
  285. vellum/types/sentence_chunker_config.py +7 -3
  286. vellum/types/sentence_chunker_config_request.py +7 -3
  287. vellum/types/sentence_chunking.py +7 -3
  288. vellum/types/sentence_chunking_request.py +7 -3
  289. vellum/types/slim_deployment_read.py +11 -3
  290. vellum/types/slim_document.py +7 -3
  291. vellum/types/slim_workflow_deployment.py +12 -3
  292. vellum/types/streaming_execute_prompt_event.py +7 -3
  293. vellum/types/streaming_prompt_execution_meta.py +7 -3
  294. vellum/types/streaming_workflow_node_result_event.py +7 -3
  295. vellum/types/string_chat_message_content.py +7 -3
  296. vellum/types/string_chat_message_content_request.py +7 -3
  297. vellum/types/string_input_request.py +7 -3
  298. vellum/types/string_parameter_config.py +39 -0
  299. vellum/types/string_parameter_config_request.py +39 -0
  300. vellum/types/string_variable_value.py +7 -3
  301. vellum/types/string_vellum_value.py +7 -3
  302. vellum/types/string_vellum_value_request.py +7 -3
  303. vellum/types/submit_completion_actual_request.py +7 -3
  304. vellum/types/submit_workflow_execution_actual_request.py +67 -12
  305. vellum/types/subworkflow_node_result.py +7 -3
  306. vellum/types/subworkflow_node_result_data.py +7 -3
  307. vellum/types/templating_node_array_result.py +7 -3
  308. vellum/types/templating_node_chat_history_result.py +7 -3
  309. vellum/types/templating_node_error_result.py +7 -3
  310. vellum/types/templating_node_function_call_result.py +7 -3
  311. vellum/types/templating_node_json_result.py +7 -3
  312. vellum/types/templating_node_number_result.py +7 -3
  313. vellum/types/templating_node_result.py +7 -3
  314. vellum/types/templating_node_result_data.py +7 -3
  315. vellum/types/templating_node_result_output.py +144 -32
  316. vellum/types/templating_node_search_results_result.py +7 -3
  317. vellum/types/templating_node_string_result.py +7 -3
  318. vellum/types/terminal_node_array_result.py +7 -3
  319. vellum/types/terminal_node_chat_history_result.py +7 -3
  320. vellum/types/terminal_node_error_result.py +7 -3
  321. vellum/types/terminal_node_function_call_result.py +7 -3
  322. vellum/types/terminal_node_json_result.py +7 -3
  323. vellum/types/terminal_node_number_result.py +7 -3
  324. vellum/types/terminal_node_result.py +7 -3
  325. vellum/types/terminal_node_result_data.py +7 -3
  326. vellum/types/terminal_node_result_output.py +152 -32
  327. vellum/types/terminal_node_search_results_result.py +7 -3
  328. vellum/types/terminal_node_string_result.py +7 -3
  329. vellum/types/test_case_array_variable_value.py +7 -3
  330. vellum/types/test_case_chat_history_variable_value.py +7 -3
  331. vellum/types/test_case_error_variable_value.py +7 -3
  332. vellum/types/test_case_function_call_variable_value.py +7 -3
  333. vellum/types/test_case_json_variable_value.py +8 -4
  334. vellum/types/test_case_number_variable_value.py +7 -3
  335. vellum/types/test_case_search_results_variable_value.py +7 -3
  336. vellum/types/test_case_string_variable_value.py +7 -3
  337. vellum/types/test_case_variable_value.py +152 -32
  338. vellum/types/test_suite_run_deployment_release_tag_exec_config.py +7 -3
  339. vellum/types/test_suite_run_deployment_release_tag_exec_config_data.py +7 -3
  340. vellum/types/test_suite_run_deployment_release_tag_exec_config_data_request.py +7 -3
  341. vellum/types/test_suite_run_deployment_release_tag_exec_config_request.py +7 -3
  342. vellum/types/test_suite_run_exec_config.py +57 -12
  343. vellum/types/test_suite_run_exec_config_request.py +61 -12
  344. vellum/types/test_suite_run_execution.py +7 -3
  345. vellum/types/test_suite_run_execution_array_output.py +7 -3
  346. vellum/types/test_suite_run_execution_chat_history_output.py +7 -3
  347. vellum/types/test_suite_run_execution_error_output.py +7 -3
  348. vellum/types/test_suite_run_execution_function_call_output.py +7 -3
  349. vellum/types/test_suite_run_execution_json_output.py +7 -3
  350. vellum/types/test_suite_run_execution_metric_definition.py +7 -3
  351. vellum/types/test_suite_run_execution_metric_result.py +7 -3
  352. vellum/types/test_suite_run_execution_number_output.py +7 -3
  353. vellum/types/test_suite_run_execution_output.py +152 -32
  354. vellum/types/test_suite_run_execution_search_results_output.py +7 -3
  355. vellum/types/test_suite_run_execution_string_output.py +7 -3
  356. vellum/types/test_suite_run_external_exec_config.py +7 -3
  357. vellum/types/test_suite_run_external_exec_config_data.py +7 -3
  358. vellum/types/test_suite_run_external_exec_config_data_request.py +7 -3
  359. vellum/types/test_suite_run_external_exec_config_request.py +7 -3
  360. vellum/types/test_suite_run_metric_error_output.py +7 -3
  361. vellum/types/test_suite_run_metric_number_output.py +7 -3
  362. vellum/types/test_suite_run_metric_output.py +55 -12
  363. vellum/types/test_suite_run_metric_string_output.py +7 -3
  364. vellum/types/test_suite_run_read.py +7 -3
  365. vellum/types/test_suite_run_test_suite.py +7 -3
  366. vellum/types/test_suite_run_workflow_release_tag_exec_config.py +7 -3
  367. vellum/types/test_suite_run_workflow_release_tag_exec_config_data.py +7 -3
  368. vellum/types/test_suite_run_workflow_release_tag_exec_config_data_request.py +7 -3
  369. vellum/types/test_suite_run_workflow_release_tag_exec_config_request.py +7 -3
  370. vellum/types/test_suite_test_case.py +7 -3
  371. vellum/types/test_suite_test_case_bulk_operation_request.py +75 -16
  372. vellum/types/test_suite_test_case_bulk_result.py +74 -16
  373. vellum/types/test_suite_test_case_create_bulk_operation_request.py +7 -3
  374. vellum/types/test_suite_test_case_created_bulk_result.py +7 -3
  375. vellum/types/test_suite_test_case_created_bulk_result_data.py +7 -3
  376. vellum/types/test_suite_test_case_delete_bulk_operation_data_request.py +7 -3
  377. vellum/types/test_suite_test_case_delete_bulk_operation_request.py +7 -3
  378. vellum/types/test_suite_test_case_deleted_bulk_result.py +7 -3
  379. vellum/types/test_suite_test_case_deleted_bulk_result_data.py +7 -3
  380. vellum/types/test_suite_test_case_rejected_bulk_result.py +7 -3
  381. vellum/types/test_suite_test_case_replace_bulk_operation_request.py +7 -3
  382. vellum/types/test_suite_test_case_replaced_bulk_result.py +7 -3
  383. vellum/types/test_suite_test_case_replaced_bulk_result_data.py +7 -3
  384. vellum/types/test_suite_test_case_upsert_bulk_operation_request.py +7 -3
  385. vellum/types/{execute_workflow_error_response.py → tik_token_tokenizer_config.py} +11 -6
  386. vellum/types/{generate_error_response.py → tik_token_tokenizer_config_request.py} +11 -6
  387. vellum/types/token_overlapping_window_chunker_config.py +7 -3
  388. vellum/types/token_overlapping_window_chunker_config_request.py +7 -3
  389. vellum/types/token_overlapping_window_chunking.py +7 -3
  390. vellum/types/token_overlapping_window_chunking_request.py +7 -3
  391. vellum/types/upload_document_response.py +7 -3
  392. vellum/types/upsert_test_suite_test_case_request.py +7 -3
  393. vellum/types/vellum_error.py +7 -3
  394. vellum/types/vellum_error_request.py +7 -3
  395. vellum/types/vellum_image.py +7 -3
  396. vellum/types/vellum_image_request.py +7 -3
  397. vellum/types/vellum_variable.py +7 -3
  398. vellum/types/visibility_enum.py +5 -0
  399. vellum/types/workflow_deployment_read.py +12 -3
  400. vellum/types/workflow_event_error.py +7 -3
  401. vellum/types/workflow_execution_actual_chat_history_request.py +7 -3
  402. vellum/types/workflow_execution_actual_json_request.py +7 -3
  403. vellum/types/workflow_execution_actual_string_request.py +7 -3
  404. vellum/types/workflow_execution_node_result_event.py +7 -3
  405. vellum/types/workflow_execution_workflow_result_event.py +7 -3
  406. vellum/types/workflow_expand_meta_request.py +7 -3
  407. vellum/types/workflow_node_result_data.py +189 -45
  408. vellum/types/workflow_node_result_event.py +101 -20
  409. vellum/types/workflow_output.py +171 -36
  410. vellum/types/workflow_output_array.py +7 -3
  411. vellum/types/workflow_output_chat_history.py +7 -3
  412. vellum/types/workflow_output_error.py +7 -3
  413. vellum/types/workflow_output_function_call.py +7 -3
  414. vellum/types/workflow_output_image.py +7 -3
  415. vellum/types/workflow_output_json.py +7 -3
  416. vellum/types/workflow_output_number.py +7 -3
  417. vellum/types/workflow_output_search_results.py +7 -3
  418. vellum/types/workflow_output_string.py +7 -3
  419. vellum/types/workflow_release_tag_read.py +7 -3
  420. vellum/types/workflow_release_tag_workflow_deployment_history_item.py +7 -3
  421. vellum/types/workflow_request_chat_history_input_request.py +7 -3
  422. vellum/types/workflow_request_input_request.py +72 -16
  423. vellum/types/workflow_request_json_input_request.py +7 -3
  424. vellum/types/workflow_request_number_input_request.py +7 -3
  425. vellum/types/workflow_request_string_input_request.py +7 -3
  426. vellum/types/workflow_result_event.py +7 -3
  427. vellum/types/workflow_result_event_output_data.py +178 -33
  428. vellum/types/workflow_result_event_output_data_array.py +7 -3
  429. vellum/types/workflow_result_event_output_data_chat_history.py +7 -3
  430. vellum/types/workflow_result_event_output_data_error.py +7 -3
  431. vellum/types/workflow_result_event_output_data_function_call.py +7 -3
  432. vellum/types/workflow_result_event_output_data_json.py +7 -3
  433. vellum/types/workflow_result_event_output_data_number.py +7 -3
  434. vellum/types/workflow_result_event_output_data_search_results.py +7 -3
  435. vellum/types/workflow_result_event_output_data_string.py +7 -3
  436. vellum/types/workflow_stream_event.py +43 -8
  437. {vellum_ai-0.7.3.dist-info → vellum_ai-0.7.5.dist-info}/METADATA +13 -1
  438. vellum_ai-0.7.5.dist-info/RECORD +502 -0
  439. vellum/types/array_enum.py +0 -5
  440. vellum/types/chat_history_enum.py +0 -5
  441. vellum/types/create_enum.py +0 -5
  442. vellum/types/created_enum.py +0 -5
  443. vellum/types/delete_enum.py +0 -5
  444. vellum/types/deleted_enum.py +0 -5
  445. vellum/types/error_enum.py +0 -5
  446. vellum/types/function_call_enum.py +0 -5
  447. vellum/types/hkunlp_instructor_xl_enum.py +0 -5
  448. vellum/types/image_enum.py +0 -5
  449. vellum/types/initiated_enum.py +0 -5
  450. vellum/types/intfloat_multilingual_e_5_large_enum.py +0 -5
  451. vellum/types/json_enum.py +0 -5
  452. vellum/types/merge_enum.py +0 -5
  453. vellum/types/metric_enum.py +0 -5
  454. vellum/types/number_enum.py +0 -5
  455. vellum/types/pdf_enum.py +0 -5
  456. vellum/types/reducto_chunker_enum.py +0 -5
  457. vellum/types/rejected_enum.py +0 -5
  458. vellum/types/replace_enum.py +0 -5
  459. vellum/types/replaced_enum.py +0 -5
  460. vellum/types/search_error_response.py +0 -28
  461. vellum/types/search_results_enum.py +0 -5
  462. vellum/types/sentence_chunker_enum.py +0 -5
  463. vellum/types/sentence_transformers_multi_qa_mpnet_base_cos_v_1_enum.py +0 -5
  464. vellum/types/sentence_transformers_multi_qa_mpnet_base_dot_v_1_enum.py +0 -5
  465. vellum/types/streaming_enum.py +0 -5
  466. vellum/types/string_enum.py +0 -5
  467. vellum/types/submit_completion_actuals_error_response.py +0 -25
  468. vellum/types/subworkflow_enum.py +0 -5
  469. vellum/types/test_suite_run_deployment_release_tag_exec_config_type_enum.py +0 -5
  470. vellum/types/test_suite_run_external_exec_config_type_enum.py +0 -5
  471. vellum/types/test_suite_run_metric_error_output_type_enum.py +0 -5
  472. vellum/types/test_suite_run_metric_number_output_type_enum.py +0 -5
  473. vellum/types/test_suite_run_metric_string_output_type_enum.py +0 -5
  474. vellum/types/test_suite_run_workflow_release_tag_exec_config_type_enum.py +0 -5
  475. vellum/types/text_embedding_3_large_enum.py +0 -5
  476. vellum/types/text_embedding_3_small_enum.py +0 -5
  477. vellum/types/text_embedding_ada_002_enum.py +0 -5
  478. vellum/types/token_overlapping_window_chunker_enum.py +0 -5
  479. vellum/types/upload_document_error_response.py +0 -25
  480. vellum/types/upsert_enum.py +0 -5
  481. vellum_ai-0.7.3.dist-info/RECORD +0 -496
  482. {vellum_ai-0.7.3.dist-info → vellum_ai-0.7.5.dist-info}/LICENSE +0 -0
  483. {vellum_ai-0.7.3.dist-info → vellum_ai-0.7.5.dist-info}/WHEEL +0 -0
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class WorkflowExecutionActualJsonRequest(pydantic_v1.BaseModel):
@@ -43,8 +43,12 @@ class WorkflowExecutionActualJsonRequest(pydantic_v1.BaseModel):
43
43
  return super().json(**kwargs_with_defaults)
44
44
 
45
45
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
46
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
47
- return super().dict(**kwargs_with_defaults)
46
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
47
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
48
+
49
+ return deep_union_pydantic_dicts(
50
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
51
+ )
48
52
 
49
53
  class Config:
50
54
  frozen = True
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class WorkflowExecutionActualStringRequest(pydantic_v1.BaseModel):
@@ -43,8 +43,12 @@ class WorkflowExecutionActualStringRequest(pydantic_v1.BaseModel):
43
43
  return super().json(**kwargs_with_defaults)
44
44
 
45
45
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
46
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
47
- return super().dict(**kwargs_with_defaults)
46
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
47
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
48
+
49
+ return deep_union_pydantic_dicts(
50
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
51
+ )
48
52
 
49
53
  class Config:
50
54
  frozen = True
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
  from .workflow_node_result_event import WorkflowNodeResultEvent
9
9
 
10
10
 
@@ -23,8 +23,12 @@ class WorkflowExecutionNodeResultEvent(pydantic_v1.BaseModel):
23
23
  return super().json(**kwargs_with_defaults)
24
24
 
25
25
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
26
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
27
- return super().dict(**kwargs_with_defaults)
26
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
27
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
28
+
29
+ return deep_union_pydantic_dicts(
30
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
31
+ )
28
32
 
29
33
  class Config:
30
34
  frozen = True
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
  from .workflow_result_event import WorkflowResultEvent
9
9
 
10
10
 
@@ -23,8 +23,12 @@ class WorkflowExecutionWorkflowResultEvent(pydantic_v1.BaseModel):
23
23
  return super().json(**kwargs_with_defaults)
24
24
 
25
25
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
26
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
27
- return super().dict(**kwargs_with_defaults)
26
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
27
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
28
+
29
+ return deep_union_pydantic_dicts(
30
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
31
+ )
28
32
 
29
33
  class Config:
30
34
  frozen = True
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class WorkflowExpandMetaRequest(pydantic_v1.BaseModel):
@@ -18,8 +18,12 @@ class WorkflowExpandMetaRequest(pydantic_v1.BaseModel):
18
18
  return super().json(**kwargs_with_defaults)
19
19
 
20
20
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
21
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
- return super().dict(**kwargs_with_defaults)
21
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
23
+
24
+ return deep_union_pydantic_dicts(
25
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
26
+ )
23
27
 
24
28
  class Config:
25
29
  frozen = True
@@ -2,129 +2,273 @@
2
2
 
3
3
  from __future__ import annotations
4
4
 
5
+ import datetime as dt
5
6
  import typing
6
7
 
7
- from .api_node_result import ApiNodeResult
8
- from .code_execution_node_result import CodeExecutionNodeResult
9
- from .conditional_node_result import ConditionalNodeResult
10
- from .map_node_result import MapNodeResult
11
- from .merge_node_result import MergeNodeResult
12
- from .metric_node_result import MetricNodeResult
13
- from .prompt_node_result import PromptNodeResult
14
- from .search_node_result import SearchNodeResult
15
- from .subworkflow_node_result import SubworkflowNodeResult
16
- from .templating_node_result import TemplatingNodeResult
17
- from .terminal_node_result import TerminalNodeResult
8
+ from ..core.datetime_utils import serialize_datetime
9
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
10
+ from .api_node_result_data import ApiNodeResultData
11
+ from .code_execution_node_result_data import CodeExecutionNodeResultData
12
+ from .conditional_node_result_data import ConditionalNodeResultData
13
+ from .map_node_result_data import MapNodeResultData
14
+ from .merge_node_result_data import MergeNodeResultData
15
+ from .prompt_node_result_data import PromptNodeResultData
16
+ from .search_node_result_data import SearchNodeResultData
17
+ from .subworkflow_node_result_data import SubworkflowNodeResultData
18
+ from .templating_node_result_data import TemplatingNodeResultData
19
+ from .terminal_node_result_data import TerminalNodeResultData
20
+
21
+
22
+ class WorkflowNodeResultData_Prompt(pydantic_v1.BaseModel):
23
+ data: PromptNodeResultData
24
+ type: typing.Literal["PROMPT"] = "PROMPT"
18
25
 
26
+ def json(self, **kwargs: typing.Any) -> str:
27
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
28
+ return super().json(**kwargs_with_defaults)
19
29
 
20
- class WorkflowNodeResultData_Prompt(PromptNodeResult):
21
- type: typing.Literal["PROMPT"] = "PROMPT"
30
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
31
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
32
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
33
+
34
+ return deep_union_pydantic_dicts(
35
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
36
+ )
22
37
 
23
38
  class Config:
24
39
  frozen = True
25
40
  smart_union = True
26
- allow_population_by_field_name = True
27
- populate_by_name = True
41
+ extra = pydantic_v1.Extra.allow
42
+ json_encoders = {dt.datetime: serialize_datetime}
28
43
 
29
44
 
30
- class WorkflowNodeResultData_Search(SearchNodeResult):
45
+ class WorkflowNodeResultData_Search(pydantic_v1.BaseModel):
46
+ data: SearchNodeResultData
31
47
  type: typing.Literal["SEARCH"] = "SEARCH"
32
48
 
49
+ def json(self, **kwargs: typing.Any) -> str:
50
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
51
+ return super().json(**kwargs_with_defaults)
52
+
53
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
54
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
55
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
56
+
57
+ return deep_union_pydantic_dicts(
58
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
59
+ )
60
+
33
61
  class Config:
34
62
  frozen = True
35
63
  smart_union = True
36
- allow_population_by_field_name = True
37
- populate_by_name = True
64
+ extra = pydantic_v1.Extra.allow
65
+ json_encoders = {dt.datetime: serialize_datetime}
38
66
 
39
67
 
40
- class WorkflowNodeResultData_Templating(TemplatingNodeResult):
68
+ class WorkflowNodeResultData_Templating(pydantic_v1.BaseModel):
69
+ data: TemplatingNodeResultData
41
70
  type: typing.Literal["TEMPLATING"] = "TEMPLATING"
42
71
 
72
+ def json(self, **kwargs: typing.Any) -> str:
73
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
74
+ return super().json(**kwargs_with_defaults)
75
+
76
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
77
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
78
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
79
+
80
+ return deep_union_pydantic_dicts(
81
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
82
+ )
83
+
43
84
  class Config:
44
85
  frozen = True
45
86
  smart_union = True
46
- allow_population_by_field_name = True
47
- populate_by_name = True
87
+ extra = pydantic_v1.Extra.allow
88
+ json_encoders = {dt.datetime: serialize_datetime}
48
89
 
49
90
 
50
- class WorkflowNodeResultData_CodeExecution(CodeExecutionNodeResult):
91
+ class WorkflowNodeResultData_CodeExecution(pydantic_v1.BaseModel):
92
+ data: CodeExecutionNodeResultData
51
93
  type: typing.Literal["CODE_EXECUTION"] = "CODE_EXECUTION"
52
94
 
95
+ def json(self, **kwargs: typing.Any) -> str:
96
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
97
+ return super().json(**kwargs_with_defaults)
98
+
99
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
100
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
101
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
102
+
103
+ return deep_union_pydantic_dicts(
104
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
105
+ )
106
+
53
107
  class Config:
54
108
  frozen = True
55
109
  smart_union = True
56
- allow_population_by_field_name = True
57
- populate_by_name = True
110
+ extra = pydantic_v1.Extra.allow
111
+ json_encoders = {dt.datetime: serialize_datetime}
58
112
 
59
113
 
60
- class WorkflowNodeResultData_Conditional(ConditionalNodeResult):
114
+ class WorkflowNodeResultData_Conditional(pydantic_v1.BaseModel):
115
+ data: ConditionalNodeResultData
61
116
  type: typing.Literal["CONDITIONAL"] = "CONDITIONAL"
62
117
 
118
+ def json(self, **kwargs: typing.Any) -> str:
119
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
120
+ return super().json(**kwargs_with_defaults)
121
+
122
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
123
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
124
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
125
+
126
+ return deep_union_pydantic_dicts(
127
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
128
+ )
129
+
63
130
  class Config:
64
131
  frozen = True
65
132
  smart_union = True
66
- allow_population_by_field_name = True
67
- populate_by_name = True
133
+ extra = pydantic_v1.Extra.allow
134
+ json_encoders = {dt.datetime: serialize_datetime}
68
135
 
69
136
 
70
- class WorkflowNodeResultData_Api(ApiNodeResult):
137
+ class WorkflowNodeResultData_Api(pydantic_v1.BaseModel):
138
+ data: ApiNodeResultData
71
139
  type: typing.Literal["API"] = "API"
72
140
 
141
+ def json(self, **kwargs: typing.Any) -> str:
142
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
143
+ return super().json(**kwargs_with_defaults)
144
+
145
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
146
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
147
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
148
+
149
+ return deep_union_pydantic_dicts(
150
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
151
+ )
152
+
73
153
  class Config:
74
154
  frozen = True
75
155
  smart_union = True
76
- allow_population_by_field_name = True
77
- populate_by_name = True
156
+ extra = pydantic_v1.Extra.allow
157
+ json_encoders = {dt.datetime: serialize_datetime}
78
158
 
79
159
 
80
- class WorkflowNodeResultData_Terminal(TerminalNodeResult):
160
+ class WorkflowNodeResultData_Terminal(pydantic_v1.BaseModel):
161
+ data: TerminalNodeResultData
81
162
  type: typing.Literal["TERMINAL"] = "TERMINAL"
82
163
 
164
+ def json(self, **kwargs: typing.Any) -> str:
165
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
166
+ return super().json(**kwargs_with_defaults)
167
+
168
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
169
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
170
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
171
+
172
+ return deep_union_pydantic_dicts(
173
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
174
+ )
175
+
83
176
  class Config:
84
177
  frozen = True
85
178
  smart_union = True
86
- allow_population_by_field_name = True
87
- populate_by_name = True
179
+ extra = pydantic_v1.Extra.allow
180
+ json_encoders = {dt.datetime: serialize_datetime}
88
181
 
89
182
 
90
- class WorkflowNodeResultData_Merge(MergeNodeResult):
183
+ class WorkflowNodeResultData_Merge(pydantic_v1.BaseModel):
184
+ data: MergeNodeResultData
91
185
  type: typing.Literal["MERGE"] = "MERGE"
92
186
 
187
+ def json(self, **kwargs: typing.Any) -> str:
188
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
189
+ return super().json(**kwargs_with_defaults)
190
+
191
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
192
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
193
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
194
+
195
+ return deep_union_pydantic_dicts(
196
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
197
+ )
198
+
93
199
  class Config:
94
200
  frozen = True
95
201
  smart_union = True
96
- allow_population_by_field_name = True
97
- populate_by_name = True
202
+ extra = pydantic_v1.Extra.allow
203
+ json_encoders = {dt.datetime: serialize_datetime}
98
204
 
99
205
 
100
- class WorkflowNodeResultData_Subworkflow(SubworkflowNodeResult):
206
+ class WorkflowNodeResultData_Subworkflow(pydantic_v1.BaseModel):
207
+ data: typing.Optional[SubworkflowNodeResultData] = None
101
208
  type: typing.Literal["SUBWORKFLOW"] = "SUBWORKFLOW"
102
209
 
210
+ def json(self, **kwargs: typing.Any) -> str:
211
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
212
+ return super().json(**kwargs_with_defaults)
213
+
214
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
215
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
216
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
217
+
218
+ return deep_union_pydantic_dicts(
219
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
220
+ )
221
+
103
222
  class Config:
104
223
  frozen = True
105
224
  smart_union = True
106
- allow_population_by_field_name = True
107
- populate_by_name = True
225
+ extra = pydantic_v1.Extra.allow
226
+ json_encoders = {dt.datetime: serialize_datetime}
108
227
 
109
228
 
110
- class WorkflowNodeResultData_Metric(MetricNodeResult):
229
+ class WorkflowNodeResultData_Metric(pydantic_v1.BaseModel):
111
230
  type: typing.Literal["METRIC"] = "METRIC"
112
231
 
232
+ def json(self, **kwargs: typing.Any) -> str:
233
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
234
+ return super().json(**kwargs_with_defaults)
235
+
236
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
237
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
238
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
239
+
240
+ return deep_union_pydantic_dicts(
241
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
242
+ )
243
+
113
244
  class Config:
114
245
  frozen = True
115
246
  smart_union = True
116
- allow_population_by_field_name = True
117
- populate_by_name = True
247
+ extra = pydantic_v1.Extra.allow
248
+ json_encoders = {dt.datetime: serialize_datetime}
118
249
 
119
250
 
120
- class WorkflowNodeResultData_Map(MapNodeResult):
251
+ class WorkflowNodeResultData_Map(pydantic_v1.BaseModel):
252
+ data: typing.Optional[MapNodeResultData] = None
121
253
  type: typing.Literal["MAP"] = "MAP"
122
254
 
255
+ def json(self, **kwargs: typing.Any) -> str:
256
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
257
+ return super().json(**kwargs_with_defaults)
258
+
259
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
260
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
261
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
262
+
263
+ return deep_union_pydantic_dicts(
264
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
265
+ )
266
+
123
267
  class Config:
124
268
  frozen = True
125
269
  smart_union = True
126
- allow_population_by_field_name = True
127
- populate_by_name = True
270
+ extra = pydantic_v1.Extra.allow
271
+ json_encoders = {dt.datetime: serialize_datetime}
128
272
 
129
273
 
130
274
  WorkflowNodeResultData = typing.Union[
@@ -2,52 +2,133 @@
2
2
 
3
3
  from __future__ import annotations
4
4
 
5
+ import datetime as dt
5
6
  import typing
6
7
 
7
- from .fulfilled_workflow_node_result_event import FulfilledWorkflowNodeResultEvent
8
- from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEvent
9
- from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
10
- from .streaming_workflow_node_result_event import StreamingWorkflowNodeResultEvent
8
+ from ..core.datetime_utils import serialize_datetime
9
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
10
+ from .node_input_variable_compiled_value import NodeInputVariableCompiledValue
11
+ from .node_output_compiled_value import NodeOutputCompiledValue
12
+ from .workflow_event_error import WorkflowEventError
13
+ from .workflow_node_result_data import WorkflowNodeResultData
14
+
15
+
16
+ class WorkflowNodeResultEvent_Initiated(pydantic_v1.BaseModel):
17
+ id: str
18
+ node_id: str
19
+ node_result_id: str
20
+ ts: typing.Optional[dt.datetime] = None
21
+ data: typing.Optional[WorkflowNodeResultData] = None
22
+ source_execution_id: typing.Optional[str] = None
23
+ input_values: typing.Optional[typing.List[NodeInputVariableCompiledValue]] = None
24
+ state: typing.Literal["INITIATED"] = "INITIATED"
11
25
 
26
+ def json(self, **kwargs: typing.Any) -> str:
27
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
28
+ return super().json(**kwargs_with_defaults)
12
29
 
13
- class WorkflowNodeResultEvent_Initiated(InitiatedWorkflowNodeResultEvent):
14
- state: typing.Literal["INITIATED"] = "INITIATED"
30
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
31
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
32
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
33
+
34
+ return deep_union_pydantic_dicts(
35
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
36
+ )
15
37
 
16
38
  class Config:
17
39
  frozen = True
18
40
  smart_union = True
19
- allow_population_by_field_name = True
20
- populate_by_name = True
41
+ extra = pydantic_v1.Extra.allow
42
+ json_encoders = {dt.datetime: serialize_datetime}
43
+
44
+
45
+ class WorkflowNodeResultEvent_Streaming(pydantic_v1.BaseModel):
46
+ id: str
47
+ node_id: str
48
+ node_result_id: str
49
+ ts: typing.Optional[dt.datetime] = None
50
+ data: typing.Optional[WorkflowNodeResultData] = None
51
+ source_execution_id: typing.Optional[str] = None
52
+ output: typing.Optional[NodeOutputCompiledValue] = None
53
+ output_index: typing.Optional[int] = None
54
+ state: typing.Literal["STREAMING"] = "STREAMING"
21
55
 
56
+ def json(self, **kwargs: typing.Any) -> str:
57
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
58
+ return super().json(**kwargs_with_defaults)
22
59
 
23
- class WorkflowNodeResultEvent_Streaming(StreamingWorkflowNodeResultEvent):
24
- state: typing.Literal["STREAMING"] = "STREAMING"
60
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
61
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
62
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
63
+
64
+ return deep_union_pydantic_dicts(
65
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
66
+ )
25
67
 
26
68
  class Config:
27
69
  frozen = True
28
70
  smart_union = True
29
- allow_population_by_field_name = True
30
- populate_by_name = True
71
+ extra = pydantic_v1.Extra.allow
72
+ json_encoders = {dt.datetime: serialize_datetime}
73
+
74
+
75
+ class WorkflowNodeResultEvent_Fulfilled(pydantic_v1.BaseModel):
76
+ id: str
77
+ node_id: str
78
+ node_result_id: str
79
+ ts: typing.Optional[dt.datetime] = None
80
+ data: typing.Optional[WorkflowNodeResultData] = None
81
+ source_execution_id: typing.Optional[str] = None
82
+ output_values: typing.List[NodeOutputCompiledValue]
83
+ mocked: typing.Optional[bool] = None
84
+ state: typing.Literal["FULFILLED"] = "FULFILLED"
31
85
 
86
+ def json(self, **kwargs: typing.Any) -> str:
87
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
88
+ return super().json(**kwargs_with_defaults)
32
89
 
33
- class WorkflowNodeResultEvent_Fulfilled(FulfilledWorkflowNodeResultEvent):
34
- state: typing.Literal["FULFILLED"] = "FULFILLED"
90
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
91
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
92
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
93
+
94
+ return deep_union_pydantic_dicts(
95
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
96
+ )
35
97
 
36
98
  class Config:
37
99
  frozen = True
38
100
  smart_union = True
39
- allow_population_by_field_name = True
40
- populate_by_name = True
101
+ extra = pydantic_v1.Extra.allow
102
+ json_encoders = {dt.datetime: serialize_datetime}
103
+
104
+
105
+ class WorkflowNodeResultEvent_Rejected(pydantic_v1.BaseModel):
106
+ id: str
107
+ node_id: str
108
+ node_result_id: str
109
+ ts: typing.Optional[dt.datetime] = None
110
+ data: typing.Optional[WorkflowNodeResultData] = None
111
+ source_execution_id: typing.Optional[str] = None
112
+ error: WorkflowEventError
113
+ state: typing.Literal["REJECTED"] = "REJECTED"
41
114
 
115
+ def json(self, **kwargs: typing.Any) -> str:
116
+ kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
117
+ return super().json(**kwargs_with_defaults)
42
118
 
43
- class WorkflowNodeResultEvent_Rejected(RejectedWorkflowNodeResultEvent):
44
- state: typing.Literal["REJECTED"] = "REJECTED"
119
+ def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
120
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
121
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
122
+
123
+ return deep_union_pydantic_dicts(
124
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
125
+ )
45
126
 
46
127
  class Config:
47
128
  frozen = True
48
129
  smart_union = True
49
- allow_population_by_field_name = True
50
- populate_by_name = True
130
+ extra = pydantic_v1.Extra.allow
131
+ json_encoders = {dt.datetime: serialize_datetime}
51
132
 
52
133
 
53
134
  WorkflowNodeResultEvent = typing.Union[