vellum-ai 0.3.23__py3-none-any.whl → 0.5.0__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (327) hide show
  1. vellum/__init__.py +37 -60
  2. vellum/client.py +139 -137
  3. vellum/core/__init__.py +2 -0
  4. vellum/core/client_wrapper.py +1 -1
  5. vellum/core/jsonable_encoder.py +5 -9
  6. vellum/core/pydantic_utilities.py +12 -0
  7. vellum/errors/__init__.py +1 -2
  8. vellum/lib/__init__.py +5 -0
  9. vellum/lib/test_suites/__init__.py +5 -0
  10. vellum/lib/test_suites/constants.py +2 -0
  11. vellum/lib/test_suites/exceptions.py +2 -0
  12. vellum/lib/test_suites/resources.py +253 -0
  13. vellum/lib/utils/__init__.py +0 -0
  14. vellum/lib/utils/env.py +11 -0
  15. vellum/lib/utils/exceptions.py +2 -0
  16. vellum/lib/utils/paginator.py +28 -0
  17. vellum/resources/__init__.py +0 -4
  18. vellum/resources/deployments/client.py +27 -31
  19. vellum/resources/deployments/types/deployments_list_request_status.py +1 -13
  20. vellum/resources/document_indexes/client.py +35 -39
  21. vellum/resources/document_indexes/types/document_indexes_list_request_status.py +1 -13
  22. vellum/resources/documents/client.py +131 -35
  23. vellum/resources/folder_entities/client.py +4 -4
  24. vellum/resources/sandboxes/client.py +18 -46
  25. vellum/resources/test_suite_runs/client.py +19 -23
  26. vellum/resources/test_suites/client.py +17 -21
  27. vellum/resources/workflow_deployments/client.py +13 -17
  28. vellum/resources/workflow_deployments/types/workflow_deployments_list_request_status.py +1 -13
  29. vellum/types/__init__.py +37 -55
  30. vellum/types/api_node_result.py +3 -7
  31. vellum/types/api_node_result_data.py +4 -8
  32. vellum/types/array_chat_message_content.py +3 -7
  33. vellum/types/array_chat_message_content_request.py +3 -7
  34. vellum/types/chat_history_input_request.py +4 -8
  35. vellum/types/chat_history_variable_value.py +3 -7
  36. vellum/types/chat_message.py +4 -8
  37. vellum/types/chat_message_request.py +4 -8
  38. vellum/types/chat_message_role.py +1 -32
  39. vellum/types/code_execution_node_array_result.py +3 -7
  40. vellum/types/code_execution_node_chat_history_result.py +3 -7
  41. vellum/types/code_execution_node_error_result.py +3 -7
  42. vellum/types/code_execution_node_function_call_result.py +3 -7
  43. vellum/types/code_execution_node_json_result.py +3 -7
  44. vellum/types/code_execution_node_number_result.py +3 -7
  45. vellum/types/code_execution_node_result.py +3 -7
  46. vellum/types/code_execution_node_result_data.py +3 -7
  47. vellum/types/code_execution_node_search_results_result.py +3 -7
  48. vellum/types/code_execution_node_string_result.py +3 -7
  49. vellum/types/conditional_node_result.py +3 -7
  50. vellum/types/conditional_node_result_data.py +3 -7
  51. vellum/types/deployment_provider_payload_response.py +3 -7
  52. vellum/types/deployment_read.py +8 -12
  53. vellum/types/document_document_to_document_index.py +6 -10
  54. vellum/types/document_index_read.py +8 -12
  55. vellum/types/document_read.py +8 -12
  56. vellum/types/enriched_normalized_completion.py +9 -13
  57. vellum/types/entity_status.py +1 -18
  58. vellum/types/environment_enum.py +1 -27
  59. vellum/types/error_variable_value.py +3 -7
  60. vellum/types/execute_prompt_api_error_response.py +4 -8
  61. vellum/types/execute_workflow_error_response.py +4 -8
  62. vellum/types/execute_workflow_response.py +3 -7
  63. vellum/types/execute_workflow_stream_error_response.py +4 -8
  64. vellum/types/execution_array_vellum_value.py +4 -8
  65. vellum/types/execution_chat_history_vellum_value.py +4 -8
  66. vellum/types/execution_error_vellum_value.py +4 -8
  67. vellum/types/execution_function_call_vellum_value.py +4 -8
  68. vellum/types/execution_json_vellum_value.py +4 -8
  69. vellum/types/execution_number_vellum_value.py +4 -8
  70. vellum/types/execution_search_results_vellum_value.py +4 -8
  71. vellum/types/execution_string_vellum_value.py +4 -8
  72. vellum/types/external_test_case_execution.py +4 -8
  73. vellum/types/external_test_case_execution_request.py +4 -8
  74. vellum/types/finish_reason_enum.py +1 -27
  75. vellum/types/fulfilled_execute_prompt_event.py +3 -7
  76. vellum/types/fulfilled_execute_prompt_response.py +5 -9
  77. vellum/types/fulfilled_execute_workflow_workflow_result_event.py +3 -7
  78. vellum/types/fulfilled_function_call.py +3 -7
  79. vellum/types/{registered_prompt_input_variable_request.py → fulfilled_function_call_request.py} +10 -9
  80. vellum/types/fulfilled_prompt_execution_meta.py +3 -7
  81. vellum/types/fulfilled_workflow_node_result_event.py +3 -7
  82. vellum/types/function_call_chat_message_content.py +3 -7
  83. vellum/types/function_call_chat_message_content_request.py +3 -7
  84. vellum/types/function_call_chat_message_content_value.py +3 -7
  85. vellum/types/function_call_chat_message_content_value_request.py +3 -7
  86. vellum/types/function_call_variable_value.py +3 -7
  87. vellum/types/generate_error_response.py +4 -8
  88. vellum/types/generate_options_request.py +4 -8
  89. vellum/types/generate_request.py +6 -10
  90. vellum/types/generate_result.py +5 -9
  91. vellum/types/generate_result_data.py +4 -8
  92. vellum/types/generate_result_error.py +4 -8
  93. vellum/types/generate_stream_response.py +3 -7
  94. vellum/types/generate_stream_result.py +3 -7
  95. vellum/types/generate_stream_result_data.py +3 -7
  96. vellum/types/image_chat_message_content.py +3 -7
  97. vellum/types/image_chat_message_content_request.py +3 -7
  98. vellum/types/image_variable_value.py +3 -7
  99. vellum/types/indexing_state_enum.py +3 -37
  100. vellum/types/initiated_execute_prompt_event.py +3 -7
  101. vellum/types/initiated_prompt_execution_meta.py +3 -7
  102. vellum/types/initiated_workflow_node_result_event.py +3 -7
  103. vellum/types/json_input_request.py +4 -8
  104. vellum/types/json_variable_value.py +3 -7
  105. vellum/types/logical_operator.py +23 -173
  106. vellum/types/logprobs_enum.py +1 -18
  107. vellum/types/metadata_filter_config_request.py +3 -7
  108. vellum/types/metadata_filter_rule_combinator.py +1 -18
  109. vellum/types/metadata_filter_rule_request.py +3 -7
  110. vellum/types/ml_model_usage.py +3 -7
  111. vellum/types/named_scenario_input_chat_history_variable_value_request.py +31 -0
  112. vellum/types/named_scenario_input_request.py +31 -0
  113. vellum/types/{registered_prompt_sandbox_snapshot.py → named_scenario_input_string_variable_value_request.py} +7 -9
  114. vellum/types/named_test_case_chat_history_variable_value.py +3 -7
  115. vellum/types/named_test_case_chat_history_variable_value_request.py +3 -7
  116. vellum/types/named_test_case_error_variable_value.py +3 -7
  117. vellum/types/named_test_case_error_variable_value_request.py +3 -7
  118. vellum/types/{register_prompt_prompt.py → named_test_case_function_call_variable_value.py} +7 -13
  119. vellum/types/named_test_case_function_call_variable_value_request.py +31 -0
  120. vellum/types/named_test_case_json_variable_value.py +3 -7
  121. vellum/types/named_test_case_json_variable_value_request.py +3 -7
  122. vellum/types/named_test_case_number_variable_value.py +3 -7
  123. vellum/types/named_test_case_number_variable_value_request.py +3 -7
  124. vellum/types/named_test_case_search_results_variable_value.py +3 -7
  125. vellum/types/named_test_case_search_results_variable_value_request.py +3 -7
  126. vellum/types/named_test_case_string_variable_value.py +3 -7
  127. vellum/types/named_test_case_string_variable_value_request.py +3 -7
  128. vellum/types/named_test_case_variable_value.py +12 -0
  129. vellum/types/named_test_case_variable_value_request.py +12 -0
  130. vellum/types/node_input_compiled_array_value.py +3 -7
  131. vellum/types/node_input_compiled_chat_history_value.py +3 -7
  132. vellum/types/node_input_compiled_error_value.py +3 -7
  133. vellum/types/node_input_compiled_function_call.py +3 -7
  134. vellum/types/node_input_compiled_json_value.py +3 -7
  135. vellum/types/node_input_compiled_number_value.py +3 -7
  136. vellum/types/node_input_compiled_search_results_value.py +3 -7
  137. vellum/types/node_input_compiled_string_value.py +3 -7
  138. vellum/types/node_output_compiled_array_value.py +3 -7
  139. vellum/types/node_output_compiled_chat_history_value.py +3 -7
  140. vellum/types/node_output_compiled_error_value.py +3 -7
  141. vellum/types/node_output_compiled_function_value.py +3 -7
  142. vellum/types/node_output_compiled_json_value.py +3 -7
  143. vellum/types/node_output_compiled_number_value.py +3 -7
  144. vellum/types/node_output_compiled_search_results_value.py +3 -7
  145. vellum/types/node_output_compiled_string_value.py +3 -7
  146. vellum/types/normalized_log_probs.py +3 -7
  147. vellum/types/normalized_token_log_probs.py +3 -7
  148. vellum/types/number_variable_value.py +3 -7
  149. vellum/types/paginated_document_index_read_list.py +3 -7
  150. vellum/types/paginated_slim_deployment_read_list.py +3 -7
  151. vellum/types/paginated_slim_document_list.py +3 -7
  152. vellum/types/paginated_slim_workflow_deployment_list.py +3 -7
  153. vellum/types/paginated_test_suite_run_execution_list.py +3 -7
  154. vellum/types/paginated_test_suite_test_case_list.py +3 -7
  155. vellum/types/processing_failure_reason_enum.py +1 -20
  156. vellum/types/processing_state_enum.py +1 -32
  157. vellum/types/prompt_deployment_expand_meta_request_request.py +9 -13
  158. vellum/types/prompt_execution_meta.py +3 -7
  159. vellum/types/prompt_node_result.py +3 -7
  160. vellum/types/prompt_node_result_data.py +3 -7
  161. vellum/types/raw_prompt_execution_overrides_request.py +5 -9
  162. vellum/types/rejected_execute_prompt_event.py +3 -7
  163. vellum/types/rejected_execute_prompt_response.py +5 -9
  164. vellum/types/rejected_execute_workflow_workflow_result_event.py +3 -7
  165. vellum/types/rejected_function_call.py +3 -7
  166. vellum/types/rejected_prompt_execution_meta.py +3 -7
  167. vellum/types/rejected_workflow_node_result_event.py +3 -7
  168. vellum/types/sandbox_scenario.py +5 -9
  169. vellum/types/scenario_input.py +18 -21
  170. vellum/types/{registered_prompt_sandbox.py → scenario_input_chat_history_variable_value.py} +7 -13
  171. vellum/types/{register_prompt_error_response.py → scenario_input_string_variable_value.py} +7 -9
  172. vellum/types/search_error_response.py +4 -8
  173. vellum/types/search_filters_request.py +5 -9
  174. vellum/types/search_node_result.py +3 -7
  175. vellum/types/search_node_result_data.py +4 -8
  176. vellum/types/search_request_options_request.py +7 -11
  177. vellum/types/search_response.py +4 -8
  178. vellum/types/search_result.py +6 -10
  179. vellum/types/search_result_document.py +7 -11
  180. vellum/types/search_result_document_request.py +6 -10
  181. vellum/types/search_result_merging_request.py +4 -8
  182. vellum/types/search_result_request.py +6 -10
  183. vellum/types/search_results_variable_value.py +3 -7
  184. vellum/types/search_weights_request.py +5 -9
  185. vellum/types/slim_deployment_read.py +7 -11
  186. vellum/types/slim_document.py +12 -16
  187. vellum/types/slim_workflow_deployment.py +9 -13
  188. vellum/types/streaming_execute_prompt_event.py +4 -8
  189. vellum/types/streaming_prompt_execution_meta.py +3 -7
  190. vellum/types/streaming_workflow_node_result_event.py +3 -7
  191. vellum/types/string_chat_message_content.py +3 -7
  192. vellum/types/string_chat_message_content_request.py +3 -7
  193. vellum/types/string_input_request.py +4 -8
  194. vellum/types/string_variable_value.py +3 -7
  195. vellum/types/submit_completion_actual_request.py +8 -12
  196. vellum/types/submit_completion_actuals_error_response.py +3 -7
  197. vellum/types/subworkflow_node_result.py +3 -7
  198. vellum/types/templating_node_array_result.py +3 -7
  199. vellum/types/templating_node_chat_history_result.py +3 -7
  200. vellum/types/templating_node_error_result.py +3 -7
  201. vellum/types/templating_node_function_call_result.py +3 -7
  202. vellum/types/templating_node_json_result.py +3 -7
  203. vellum/types/templating_node_number_result.py +3 -7
  204. vellum/types/templating_node_result.py +3 -7
  205. vellum/types/templating_node_result_data.py +3 -7
  206. vellum/types/templating_node_search_results_result.py +3 -7
  207. vellum/types/templating_node_string_result.py +3 -7
  208. vellum/types/terminal_node_array_result.py +4 -8
  209. vellum/types/terminal_node_chat_history_result.py +4 -8
  210. vellum/types/terminal_node_error_result.py +4 -8
  211. vellum/types/terminal_node_function_call_result.py +4 -8
  212. vellum/types/terminal_node_json_result.py +4 -8
  213. vellum/types/terminal_node_number_result.py +4 -8
  214. vellum/types/terminal_node_result.py +3 -7
  215. vellum/types/terminal_node_result_data.py +3 -7
  216. vellum/types/terminal_node_search_results_result.py +4 -8
  217. vellum/types/terminal_node_string_result.py +4 -8
  218. vellum/types/test_case_chat_history_variable_value.py +3 -7
  219. vellum/types/test_case_error_variable_value.py +3 -7
  220. vellum/types/test_case_function_call_variable_value.py +32 -0
  221. vellum/types/test_case_json_variable_value.py +3 -7
  222. vellum/types/test_case_number_variable_value.py +3 -7
  223. vellum/types/test_case_search_results_variable_value.py +3 -7
  224. vellum/types/test_case_string_variable_value.py +3 -7
  225. vellum/types/test_case_variable_value.py +12 -0
  226. vellum/types/test_suite_run_deployment_release_tag_exec_config.py +4 -8
  227. vellum/types/test_suite_run_deployment_release_tag_exec_config_data.py +5 -9
  228. vellum/types/test_suite_run_deployment_release_tag_exec_config_data_request.py +5 -9
  229. vellum/types/test_suite_run_deployment_release_tag_exec_config_request.py +4 -8
  230. vellum/types/test_suite_run_execution.py +3 -7
  231. vellum/types/test_suite_run_execution_chat_history_output.py +3 -7
  232. vellum/types/test_suite_run_execution_error_output.py +3 -7
  233. vellum/types/{registered_prompt_model_version.py → test_suite_run_execution_function_call_output.py} +8 -13
  234. vellum/types/test_suite_run_execution_json_output.py +3 -7
  235. vellum/types/test_suite_run_execution_metric_definition.py +3 -7
  236. vellum/types/test_suite_run_execution_metric_result.py +3 -7
  237. vellum/types/test_suite_run_execution_number_output.py +3 -7
  238. vellum/types/test_suite_run_execution_output.py +12 -0
  239. vellum/types/test_suite_run_execution_search_results_output.py +3 -7
  240. vellum/types/test_suite_run_execution_string_output.py +3 -7
  241. vellum/types/test_suite_run_external_exec_config.py +4 -8
  242. vellum/types/test_suite_run_external_exec_config_data.py +4 -8
  243. vellum/types/test_suite_run_external_exec_config_data_request.py +4 -8
  244. vellum/types/test_suite_run_external_exec_config_request.py +4 -8
  245. vellum/types/test_suite_run_metric_error_output.py +3 -7
  246. vellum/types/test_suite_run_metric_number_output.py +3 -7
  247. vellum/types/test_suite_run_metric_string_output.py +3 -7
  248. vellum/types/test_suite_run_read.py +5 -9
  249. vellum/types/test_suite_run_state.py +1 -37
  250. vellum/types/test_suite_run_test_suite.py +3 -7
  251. vellum/types/test_suite_run_workflow_release_tag_exec_config.py +4 -8
  252. vellum/types/test_suite_run_workflow_release_tag_exec_config_data.py +5 -9
  253. vellum/types/test_suite_run_workflow_release_tag_exec_config_data_request.py +5 -9
  254. vellum/types/test_suite_run_workflow_release_tag_exec_config_request.py +4 -8
  255. vellum/types/test_suite_test_case.py +3 -7
  256. vellum/types/upload_document_error_response.py +3 -7
  257. vellum/types/upload_document_response.py +4 -8
  258. vellum/types/vellum_error.py +3 -7
  259. vellum/types/vellum_error_code_enum.py +3 -32
  260. vellum/types/vellum_error_request.py +3 -7
  261. vellum/types/vellum_image.py +3 -7
  262. vellum/types/vellum_image_request.py +3 -7
  263. vellum/types/vellum_variable.py +3 -7
  264. vellum/types/vellum_variable_type.py +6 -57
  265. vellum/types/workflow_deployment_read.py +9 -13
  266. vellum/types/workflow_event_error.py +3 -7
  267. vellum/types/workflow_execution_actual_chat_history_request.py +8 -12
  268. vellum/types/workflow_execution_actual_json_request.py +8 -12
  269. vellum/types/workflow_execution_actual_string_request.py +8 -12
  270. vellum/types/workflow_execution_event_error_code.py +13 -52
  271. vellum/types/workflow_execution_event_type.py +1 -18
  272. vellum/types/workflow_execution_node_result_event.py +3 -7
  273. vellum/types/workflow_execution_workflow_result_event.py +3 -7
  274. vellum/types/workflow_node_result_event_state.py +3 -32
  275. vellum/types/workflow_output_array.py +4 -8
  276. vellum/types/workflow_output_chat_history.py +4 -8
  277. vellum/types/workflow_output_error.py +4 -8
  278. vellum/types/workflow_output_function_call.py +4 -8
  279. vellum/types/workflow_output_image.py +4 -8
  280. vellum/types/workflow_output_json.py +4 -8
  281. vellum/types/workflow_output_number.py +4 -8
  282. vellum/types/workflow_output_search_results.py +4 -8
  283. vellum/types/workflow_output_string.py +4 -8
  284. vellum/types/workflow_request_chat_history_input_request.py +4 -8
  285. vellum/types/workflow_request_json_input_request.py +4 -8
  286. vellum/types/workflow_request_number_input_request.py +4 -8
  287. vellum/types/workflow_request_string_input_request.py +4 -8
  288. vellum/types/workflow_result_event.py +3 -7
  289. vellum/types/workflow_result_event_output_data_array.py +4 -8
  290. vellum/types/workflow_result_event_output_data_chat_history.py +4 -8
  291. vellum/types/workflow_result_event_output_data_error.py +4 -8
  292. vellum/types/workflow_result_event_output_data_function_call.py +4 -8
  293. vellum/types/workflow_result_event_output_data_json.py +4 -8
  294. vellum/types/workflow_result_event_output_data_number.py +4 -8
  295. vellum/types/workflow_result_event_output_data_search_results.py +4 -8
  296. vellum/types/workflow_result_event_output_data_string.py +5 -9
  297. {vellum_ai-0.3.23.dist-info → vellum_ai-0.5.0.dist-info}/METADATA +1 -2
  298. vellum_ai-0.5.0.dist-info/RECORD +382 -0
  299. vellum/errors/conflict_error.py +0 -9
  300. vellum/resources/model_versions/__init__.py +0 -2
  301. vellum/resources/model_versions/client.py +0 -123
  302. vellum/resources/registered_prompts/__init__.py +0 -2
  303. vellum/resources/registered_prompts/client.py +0 -327
  304. vellum/types/block_type_enum.py +0 -36
  305. vellum/types/model_version_build_config.py +0 -40
  306. vellum/types/model_version_exec_config.py +0 -48
  307. vellum/types/model_version_exec_config_parameters.py +0 -37
  308. vellum/types/model_version_read.py +0 -82
  309. vellum/types/model_version_read_status_enum.py +0 -36
  310. vellum/types/model_version_sandbox_snapshot.py +0 -44
  311. vellum/types/prompt_template_block.py +0 -41
  312. vellum/types/prompt_template_block_data.py +0 -31
  313. vellum/types/prompt_template_block_data_request.py +0 -31
  314. vellum/types/prompt_template_block_properties.py +0 -47
  315. vellum/types/prompt_template_block_properties_request.py +0 -47
  316. vellum/types/prompt_template_block_request.py +0 -41
  317. vellum/types/prompt_template_block_state.py +0 -22
  318. vellum/types/provider_enum.py +0 -81
  319. vellum/types/register_prompt_model_parameters_request.py +0 -37
  320. vellum/types/register_prompt_prompt_info_request.py +0 -35
  321. vellum/types/register_prompt_response.py +0 -62
  322. vellum/types/registered_prompt_deployment.py +0 -42
  323. vellum/types/scenario_input_request.py +0 -34
  324. vellum/types/scenario_input_type_enum.py +0 -22
  325. vellum_ai-0.3.23.dist-info/RECORD +0 -394
  326. {vellum_ai-0.3.23.dist-info → vellum_ai-0.5.0.dist-info}/LICENSE +0 -0
  327. {vellum_ai-0.3.23.dist-info → vellum_ai-0.5.0.dist-info}/WHEEL +0 -0
@@ -4,35 +4,31 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class SubmitCompletionActualRequest(pydantic.BaseModel):
15
- id: typing.Optional[str] = pydantic.Field(default=None)
10
+ class SubmitCompletionActualRequest(pydantic_v1.BaseModel):
11
+ id: typing.Optional[str] = pydantic_v1.Field(default=None)
16
12
  """
17
13
  The Vellum-generated ID of a previously generated completion. Must provide either this or external_id.
18
14
  """
19
15
 
20
- external_id: typing.Optional[str] = pydantic.Field(default=None)
16
+ external_id: typing.Optional[str] = pydantic_v1.Field(default=None)
21
17
  """
22
18
  The external ID that was originally provided when generating the completion that you'd now like to submit actuals for. Must provide either this or id.
23
19
  """
24
20
 
25
- text: typing.Optional[str] = pydantic.Field(default=None)
21
+ text: typing.Optional[str] = pydantic_v1.Field(default=None)
26
22
  """
27
23
  Text representing what the completion _should_ have been.
28
24
  """
29
25
 
30
- quality: typing.Optional[float] = pydantic.Field(default=None)
26
+ quality: typing.Optional[float] = pydantic_v1.Field(default=None)
31
27
  """
32
28
  A number between 0 and 1 representing the quality of the completion. 0 is the worst, 1 is the best.
33
29
  """
34
30
 
35
- timestamp: typing.Optional[dt.datetime] = pydantic.Field(default=None)
31
+ timestamp: typing.Optional[dt.datetime] = pydantic_v1.Field(default=None)
36
32
  """
37
33
  Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
38
34
  """
@@ -48,5 +44,5 @@ class SubmitCompletionActualRequest(pydantic.BaseModel):
48
44
  class Config:
49
45
  frozen = True
50
46
  smart_union = True
51
- extra = pydantic.Extra.allow
47
+ extra = pydantic_v1.Extra.allow
52
48
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,14 +4,10 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class SubmitCompletionActualsErrorResponse(pydantic.BaseModel):
10
+ class SubmitCompletionActualsErrorResponse(pydantic_v1.BaseModel):
15
11
  detail: str
16
12
 
17
13
  def json(self, **kwargs: typing.Any) -> str:
@@ -25,5 +21,5 @@ class SubmitCompletionActualsErrorResponse(pydantic.BaseModel):
25
21
  class Config:
26
22
  frozen = True
27
23
  smart_union = True
28
- extra = pydantic.Extra.allow
24
+ extra = pydantic_v1.Extra.allow
29
25
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,14 +4,10 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class SubworkflowNodeResult(pydantic.BaseModel):
10
+ class SubworkflowNodeResult(pydantic_v1.BaseModel):
15
11
  """
16
12
  A Node Result Event emitted from a Subworkflow Node.
17
13
  """
@@ -27,5 +23,5 @@ class SubworkflowNodeResult(pydantic.BaseModel):
27
23
  class Config:
28
24
  frozen = True
29
25
  smart_union = True
30
- extra = pydantic.Extra.allow
26
+ extra = pydantic_v1.Extra.allow
31
27
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .array_variable_value_item import ArrayVariableValueItem
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TemplatingNodeArrayResult(pydantic.BaseModel):
11
+ class TemplatingNodeArrayResult(pydantic_v1.BaseModel):
16
12
  id: str
17
13
  value: typing.Optional[typing.List[ArrayVariableValueItem]] = None
18
14
 
@@ -27,5 +23,5 @@ class TemplatingNodeArrayResult(pydantic.BaseModel):
27
23
  class Config:
28
24
  frozen = True
29
25
  smart_union = True
30
- extra = pydantic.Extra.allow
26
+ extra = pydantic_v1.Extra.allow
31
27
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .chat_message import ChatMessage
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TemplatingNodeChatHistoryResult(pydantic.BaseModel):
11
+ class TemplatingNodeChatHistoryResult(pydantic_v1.BaseModel):
16
12
  id: str
17
13
  value: typing.Optional[typing.List[ChatMessage]] = None
18
14
 
@@ -27,5 +23,5 @@ class TemplatingNodeChatHistoryResult(pydantic.BaseModel):
27
23
  class Config:
28
24
  frozen = True
29
25
  smart_union = True
30
- extra = pydantic.Extra.allow
26
+ extra = pydantic_v1.Extra.allow
31
27
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .vellum_error import VellumError
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TemplatingNodeErrorResult(pydantic.BaseModel):
11
+ class TemplatingNodeErrorResult(pydantic_v1.BaseModel):
16
12
  id: str
17
13
  value: typing.Optional[VellumError] = None
18
14
 
@@ -27,5 +23,5 @@ class TemplatingNodeErrorResult(pydantic.BaseModel):
27
23
  class Config:
28
24
  frozen = True
29
25
  smart_union = True
30
- extra = pydantic.Extra.allow
26
+ extra = pydantic_v1.Extra.allow
31
27
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .function_call import FunctionCall
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TemplatingNodeFunctionCallResult(pydantic.BaseModel):
11
+ class TemplatingNodeFunctionCallResult(pydantic_v1.BaseModel):
16
12
  id: str
17
13
  value: typing.Optional[FunctionCall] = None
18
14
 
@@ -27,5 +23,5 @@ class TemplatingNodeFunctionCallResult(pydantic.BaseModel):
27
23
  class Config:
28
24
  frozen = True
29
25
  smart_union = True
30
- extra = pydantic.Extra.allow
26
+ extra = pydantic_v1.Extra.allow
31
27
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,14 +4,10 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class TemplatingNodeJsonResult(pydantic.BaseModel):
10
+ class TemplatingNodeJsonResult(pydantic_v1.BaseModel):
15
11
  id: str
16
12
  value: typing.Optional[typing.Dict[str, typing.Any]] = None
17
13
 
@@ -26,5 +22,5 @@ class TemplatingNodeJsonResult(pydantic.BaseModel):
26
22
  class Config:
27
23
  frozen = True
28
24
  smart_union = True
29
- extra = pydantic.Extra.allow
25
+ extra = pydantic_v1.Extra.allow
30
26
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,14 +4,10 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class TemplatingNodeNumberResult(pydantic.BaseModel):
10
+ class TemplatingNodeNumberResult(pydantic_v1.BaseModel):
15
11
  id: str
16
12
  value: typing.Optional[float] = None
17
13
 
@@ -26,5 +22,5 @@ class TemplatingNodeNumberResult(pydantic.BaseModel):
26
22
  class Config:
27
23
  frozen = True
28
24
  smart_union = True
29
- extra = pydantic.Extra.allow
25
+ extra = pydantic_v1.Extra.allow
30
26
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .templating_node_result_data import TemplatingNodeResultData
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TemplatingNodeResult(pydantic.BaseModel):
11
+ class TemplatingNodeResult(pydantic_v1.BaseModel):
16
12
  """
17
13
  A Node Result Event emitted from a Templating Node.
18
14
  """
@@ -30,5 +26,5 @@ class TemplatingNodeResult(pydantic.BaseModel):
30
26
  class Config:
31
27
  frozen = True
32
28
  smart_union = True
33
- extra = pydantic.Extra.allow
29
+ extra = pydantic_v1.Extra.allow
34
30
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .templating_node_result_output import TemplatingNodeResultOutput
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TemplatingNodeResultData(pydantic.BaseModel):
11
+ class TemplatingNodeResultData(pydantic_v1.BaseModel):
16
12
  output: TemplatingNodeResultOutput
17
13
 
18
14
  def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class TemplatingNodeResultData(pydantic.BaseModel):
26
22
  class Config:
27
23
  frozen = True
28
24
  smart_union = True
29
- extra = pydantic.Extra.allow
25
+ extra = pydantic_v1.Extra.allow
30
26
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .search_result import SearchResult
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TemplatingNodeSearchResultsResult(pydantic.BaseModel):
11
+ class TemplatingNodeSearchResultsResult(pydantic_v1.BaseModel):
16
12
  id: str
17
13
  value: typing.Optional[typing.List[SearchResult]] = None
18
14
 
@@ -27,5 +23,5 @@ class TemplatingNodeSearchResultsResult(pydantic.BaseModel):
27
23
  class Config:
28
24
  frozen = True
29
25
  smart_union = True
30
- extra = pydantic.Extra.allow
26
+ extra = pydantic_v1.Extra.allow
31
27
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,14 +4,10 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class TemplatingNodeStringResult(pydantic.BaseModel):
10
+ class TemplatingNodeStringResult(pydantic_v1.BaseModel):
15
11
  id: str
16
12
  value: typing.Optional[str] = None
17
13
 
@@ -26,5 +22,5 @@ class TemplatingNodeStringResult(pydantic.BaseModel):
26
22
  class Config:
27
23
  frozen = True
28
24
  smart_union = True
29
- extra = pydantic.Extra.allow
25
+ extra = pydantic_v1.Extra.allow
30
26
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,17 +4,13 @@ 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
8
  from .array_variable_value_item import ArrayVariableValueItem
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TerminalNodeArrayResult(pydantic.BaseModel):
11
+ class TerminalNodeArrayResult(pydantic_v1.BaseModel):
16
12
  id: typing.Optional[str] = None
17
- name: str = pydantic.Field()
13
+ name: str = pydantic_v1.Field()
18
14
  """
19
15
  The unique name given to the terminal node that produced this output.
20
16
  """
@@ -32,5 +28,5 @@ class TerminalNodeArrayResult(pydantic.BaseModel):
32
28
  class Config:
33
29
  frozen = True
34
30
  smart_union = True
35
- extra = pydantic.Extra.allow
31
+ extra = pydantic_v1.Extra.allow
36
32
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,17 +4,13 @@ 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
8
  from .chat_message import ChatMessage
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TerminalNodeChatHistoryResult(pydantic.BaseModel):
11
+ class TerminalNodeChatHistoryResult(pydantic_v1.BaseModel):
16
12
  id: typing.Optional[str] = None
17
- name: str = pydantic.Field()
13
+ name: str = pydantic_v1.Field()
18
14
  """
19
15
  The unique name given to the terminal node that produced this output.
20
16
  """
@@ -32,5 +28,5 @@ class TerminalNodeChatHistoryResult(pydantic.BaseModel):
32
28
  class Config:
33
29
  frozen = True
34
30
  smart_union = True
35
- extra = pydantic.Extra.allow
31
+ extra = pydantic_v1.Extra.allow
36
32
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,17 +4,13 @@ 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
8
  from .vellum_error import VellumError
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TerminalNodeErrorResult(pydantic.BaseModel):
11
+ class TerminalNodeErrorResult(pydantic_v1.BaseModel):
16
12
  id: typing.Optional[str] = None
17
- name: str = pydantic.Field()
13
+ name: str = pydantic_v1.Field()
18
14
  """
19
15
  The unique name given to the terminal node that produced this output.
20
16
  """
@@ -32,5 +28,5 @@ class TerminalNodeErrorResult(pydantic.BaseModel):
32
28
  class Config:
33
29
  frozen = True
34
30
  smart_union = True
35
- extra = pydantic.Extra.allow
31
+ extra = pydantic_v1.Extra.allow
36
32
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,17 +4,13 @@ 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
8
  from .function_call import FunctionCall
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TerminalNodeFunctionCallResult(pydantic.BaseModel):
11
+ class TerminalNodeFunctionCallResult(pydantic_v1.BaseModel):
16
12
  id: typing.Optional[str] = None
17
- name: str = pydantic.Field()
13
+ name: str = pydantic_v1.Field()
18
14
  """
19
15
  The unique name given to the terminal node that produced this output.
20
16
  """
@@ -32,5 +28,5 @@ class TerminalNodeFunctionCallResult(pydantic.BaseModel):
32
28
  class Config:
33
29
  frozen = True
34
30
  smart_union = True
35
- extra = pydantic.Extra.allow
31
+ extra = pydantic_v1.Extra.allow
36
32
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,16 +4,12 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class TerminalNodeJsonResult(pydantic.BaseModel):
10
+ class TerminalNodeJsonResult(pydantic_v1.BaseModel):
15
11
  id: typing.Optional[str] = None
16
- name: str = pydantic.Field()
12
+ name: str = pydantic_v1.Field()
17
13
  """
18
14
  The unique name given to the terminal node that produced this output.
19
15
  """
@@ -31,5 +27,5 @@ class TerminalNodeJsonResult(pydantic.BaseModel):
31
27
  class Config:
32
28
  frozen = True
33
29
  smart_union = True
34
- extra = pydantic.Extra.allow
30
+ extra = pydantic_v1.Extra.allow
35
31
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,16 +4,12 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class TerminalNodeNumberResult(pydantic.BaseModel):
10
+ class TerminalNodeNumberResult(pydantic_v1.BaseModel):
15
11
  id: typing.Optional[str] = None
16
- name: str = pydantic.Field()
12
+ name: str = pydantic_v1.Field()
17
13
  """
18
14
  The unique name given to the terminal node that produced this output.
19
15
  """
@@ -31,5 +27,5 @@ class TerminalNodeNumberResult(pydantic.BaseModel):
31
27
  class Config:
32
28
  frozen = True
33
29
  smart_union = True
34
- extra = pydantic.Extra.allow
30
+ extra = pydantic_v1.Extra.allow
35
31
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .terminal_node_result_data import TerminalNodeResultData
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TerminalNodeResult(pydantic.BaseModel):
11
+ class TerminalNodeResult(pydantic_v1.BaseModel):
16
12
  """
17
13
  A Node Result Event emitted from a Terminal Node.
18
14
  """
@@ -30,5 +26,5 @@ class TerminalNodeResult(pydantic.BaseModel):
30
26
  class Config:
31
27
  frozen = True
32
28
  smart_union = True
33
- extra = pydantic.Extra.allow
29
+ extra = pydantic_v1.Extra.allow
34
30
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .terminal_node_result_output import TerminalNodeResultOutput
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TerminalNodeResultData(pydantic.BaseModel):
11
+ class TerminalNodeResultData(pydantic_v1.BaseModel):
16
12
  output: TerminalNodeResultOutput
17
13
 
18
14
  def json(self, **kwargs: typing.Any) -> str:
@@ -26,5 +22,5 @@ class TerminalNodeResultData(pydantic.BaseModel):
26
22
  class Config:
27
23
  frozen = True
28
24
  smart_union = True
29
- extra = pydantic.Extra.allow
25
+ extra = pydantic_v1.Extra.allow
30
26
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,17 +4,13 @@ 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
8
  from .search_result import SearchResult
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TerminalNodeSearchResultsResult(pydantic.BaseModel):
11
+ class TerminalNodeSearchResultsResult(pydantic_v1.BaseModel):
16
12
  id: typing.Optional[str] = None
17
- name: str = pydantic.Field()
13
+ name: str = pydantic_v1.Field()
18
14
  """
19
15
  The unique name given to the terminal node that produced this output.
20
16
  """
@@ -32,5 +28,5 @@ class TerminalNodeSearchResultsResult(pydantic.BaseModel):
32
28
  class Config:
33
29
  frozen = True
34
30
  smart_union = True
35
- extra = pydantic.Extra.allow
31
+ extra = pydantic_v1.Extra.allow
36
32
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,16 +4,12 @@ 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
8
 
8
- try:
9
- import pydantic.v1 as pydantic # type: ignore
10
- except ImportError:
11
- import pydantic # type: ignore
12
9
 
13
-
14
- class TerminalNodeStringResult(pydantic.BaseModel):
10
+ class TerminalNodeStringResult(pydantic_v1.BaseModel):
15
11
  id: typing.Optional[str] = None
16
- name: str = pydantic.Field()
12
+ name: str = pydantic_v1.Field()
17
13
  """
18
14
  The unique name given to the terminal node that produced this output.
19
15
  """
@@ -31,5 +27,5 @@ class TerminalNodeStringResult(pydantic.BaseModel):
31
27
  class Config:
32
28
  frozen = True
33
29
  smart_union = True
34
- extra = pydantic.Extra.allow
30
+ extra = pydantic_v1.Extra.allow
35
31
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .chat_message import ChatMessage
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TestCaseChatHistoryVariableValue(pydantic.BaseModel):
11
+ class TestCaseChatHistoryVariableValue(pydantic_v1.BaseModel):
16
12
  """
17
13
  A chat history value for a variable in a Test Case.
18
14
  """
@@ -32,5 +28,5 @@ class TestCaseChatHistoryVariableValue(pydantic.BaseModel):
32
28
  class Config:
33
29
  frozen = True
34
30
  smart_union = True
35
- extra = pydantic.Extra.allow
31
+ extra = pydantic_v1.Extra.allow
36
32
  json_encoders = {dt.datetime: serialize_datetime}
@@ -4,15 +4,11 @@ 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
8
  from .vellum_error import VellumError
8
9
 
9
- try:
10
- import pydantic.v1 as pydantic # type: ignore
11
- except ImportError:
12
- import pydantic # type: ignore
13
10
 
14
-
15
- class TestCaseErrorVariableValue(pydantic.BaseModel):
11
+ class TestCaseErrorVariableValue(pydantic_v1.BaseModel):
16
12
  """
17
13
  An error value for a variable in a Test Case.
18
14
  """
@@ -32,5 +28,5 @@ class TestCaseErrorVariableValue(pydantic.BaseModel):
32
28
  class Config:
33
29
  frozen = True
34
30
  smart_union = True
35
- extra = pydantic.Extra.allow
31
+ extra = pydantic_v1.Extra.allow
36
32
  json_encoders = {dt.datetime: serialize_datetime}