vellum-ai 0.3.23__py3-none-any.whl → 0.5.1__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- vellum/__init__.py +37 -60
- vellum/client.py +139 -137
- vellum/core/__init__.py +2 -0
- vellum/core/client_wrapper.py +1 -1
- vellum/core/jsonable_encoder.py +5 -9
- vellum/core/pydantic_utilities.py +12 -0
- vellum/errors/__init__.py +1 -2
- vellum/lib/__init__.py +5 -0
- vellum/lib/test_suites/__init__.py +5 -0
- vellum/lib/test_suites/constants.py +2 -0
- vellum/lib/test_suites/exceptions.py +2 -0
- vellum/lib/test_suites/resources.py +331 -0
- vellum/lib/utils/__init__.py +0 -0
- vellum/lib/utils/env.py +13 -0
- vellum/lib/utils/exceptions.py +2 -0
- vellum/lib/utils/paginator.py +29 -0
- vellum/resources/__init__.py +0 -4
- vellum/resources/deployments/client.py +27 -31
- vellum/resources/deployments/types/deployments_list_request_status.py +1 -13
- vellum/resources/document_indexes/client.py +35 -39
- vellum/resources/document_indexes/types/document_indexes_list_request_status.py +1 -13
- vellum/resources/documents/client.py +131 -35
- vellum/resources/folder_entities/client.py +4 -4
- vellum/resources/sandboxes/client.py +18 -46
- vellum/resources/test_suite_runs/client.py +19 -23
- vellum/resources/test_suites/client.py +17 -21
- vellum/resources/workflow_deployments/client.py +13 -17
- vellum/resources/workflow_deployments/types/workflow_deployments_list_request_status.py +1 -13
- vellum/types/__init__.py +37 -55
- vellum/types/api_node_result.py +3 -7
- vellum/types/api_node_result_data.py +4 -8
- vellum/types/array_chat_message_content.py +3 -7
- vellum/types/array_chat_message_content_request.py +3 -7
- vellum/types/chat_history_input_request.py +4 -8
- vellum/types/chat_history_variable_value.py +3 -7
- vellum/types/chat_message.py +4 -8
- vellum/types/chat_message_request.py +4 -8
- vellum/types/chat_message_role.py +1 -32
- vellum/types/code_execution_node_array_result.py +3 -7
- vellum/types/code_execution_node_chat_history_result.py +3 -7
- vellum/types/code_execution_node_error_result.py +3 -7
- vellum/types/code_execution_node_function_call_result.py +3 -7
- vellum/types/code_execution_node_json_result.py +3 -7
- vellum/types/code_execution_node_number_result.py +3 -7
- vellum/types/code_execution_node_result.py +3 -7
- vellum/types/code_execution_node_result_data.py +3 -7
- vellum/types/code_execution_node_search_results_result.py +3 -7
- vellum/types/code_execution_node_string_result.py +3 -7
- vellum/types/conditional_node_result.py +3 -7
- vellum/types/conditional_node_result_data.py +3 -7
- vellum/types/deployment_provider_payload_response.py +3 -7
- vellum/types/deployment_read.py +8 -12
- vellum/types/document_document_to_document_index.py +6 -10
- vellum/types/document_index_read.py +8 -12
- vellum/types/document_read.py +8 -12
- vellum/types/enriched_normalized_completion.py +9 -13
- vellum/types/entity_status.py +1 -18
- vellum/types/environment_enum.py +1 -27
- vellum/types/error_variable_value.py +3 -7
- vellum/types/execute_prompt_api_error_response.py +4 -8
- vellum/types/execute_workflow_error_response.py +4 -8
- vellum/types/execute_workflow_response.py +3 -7
- vellum/types/execute_workflow_stream_error_response.py +4 -8
- vellum/types/execution_array_vellum_value.py +4 -8
- vellum/types/execution_chat_history_vellum_value.py +4 -8
- vellum/types/execution_error_vellum_value.py +4 -8
- vellum/types/execution_function_call_vellum_value.py +4 -8
- vellum/types/execution_json_vellum_value.py +4 -8
- vellum/types/execution_number_vellum_value.py +4 -8
- vellum/types/execution_search_results_vellum_value.py +4 -8
- vellum/types/execution_string_vellum_value.py +4 -8
- vellum/types/external_test_case_execution.py +4 -8
- vellum/types/external_test_case_execution_request.py +4 -8
- vellum/types/finish_reason_enum.py +1 -27
- vellum/types/fulfilled_execute_prompt_event.py +3 -7
- vellum/types/fulfilled_execute_prompt_response.py +5 -9
- vellum/types/fulfilled_execute_workflow_workflow_result_event.py +3 -7
- vellum/types/fulfilled_function_call.py +3 -7
- vellum/types/{registered_prompt_input_variable_request.py → fulfilled_function_call_request.py} +10 -9
- vellum/types/fulfilled_prompt_execution_meta.py +3 -7
- vellum/types/fulfilled_workflow_node_result_event.py +3 -7
- vellum/types/function_call_chat_message_content.py +3 -7
- vellum/types/function_call_chat_message_content_request.py +3 -7
- vellum/types/function_call_chat_message_content_value.py +3 -7
- vellum/types/function_call_chat_message_content_value_request.py +3 -7
- vellum/types/function_call_variable_value.py +3 -7
- vellum/types/generate_error_response.py +4 -8
- vellum/types/generate_options_request.py +4 -8
- vellum/types/generate_request.py +6 -10
- vellum/types/generate_result.py +5 -9
- vellum/types/generate_result_data.py +4 -8
- vellum/types/generate_result_error.py +4 -8
- vellum/types/generate_stream_response.py +3 -7
- vellum/types/generate_stream_result.py +3 -7
- vellum/types/generate_stream_result_data.py +3 -7
- vellum/types/image_chat_message_content.py +3 -7
- vellum/types/image_chat_message_content_request.py +3 -7
- vellum/types/image_variable_value.py +3 -7
- vellum/types/indexing_state_enum.py +3 -37
- vellum/types/initiated_execute_prompt_event.py +3 -7
- vellum/types/initiated_prompt_execution_meta.py +3 -7
- vellum/types/initiated_workflow_node_result_event.py +3 -7
- vellum/types/json_input_request.py +4 -8
- vellum/types/json_variable_value.py +3 -7
- vellum/types/logical_operator.py +23 -173
- vellum/types/logprobs_enum.py +1 -18
- vellum/types/metadata_filter_config_request.py +3 -7
- vellum/types/metadata_filter_rule_combinator.py +1 -18
- vellum/types/metadata_filter_rule_request.py +3 -7
- vellum/types/ml_model_usage.py +3 -7
- vellum/types/named_scenario_input_chat_history_variable_value_request.py +31 -0
- vellum/types/named_scenario_input_request.py +31 -0
- vellum/types/{registered_prompt_sandbox_snapshot.py → named_scenario_input_string_variable_value_request.py} +7 -9
- vellum/types/named_test_case_chat_history_variable_value.py +3 -7
- vellum/types/named_test_case_chat_history_variable_value_request.py +3 -7
- vellum/types/named_test_case_error_variable_value.py +3 -7
- vellum/types/named_test_case_error_variable_value_request.py +3 -7
- vellum/types/{register_prompt_prompt.py → named_test_case_function_call_variable_value.py} +7 -13
- vellum/types/named_test_case_function_call_variable_value_request.py +31 -0
- vellum/types/named_test_case_json_variable_value.py +3 -7
- vellum/types/named_test_case_json_variable_value_request.py +3 -7
- vellum/types/named_test_case_number_variable_value.py +3 -7
- vellum/types/named_test_case_number_variable_value_request.py +3 -7
- vellum/types/named_test_case_search_results_variable_value.py +3 -7
- vellum/types/named_test_case_search_results_variable_value_request.py +3 -7
- vellum/types/named_test_case_string_variable_value.py +3 -7
- vellum/types/named_test_case_string_variable_value_request.py +3 -7
- vellum/types/named_test_case_variable_value.py +12 -0
- vellum/types/named_test_case_variable_value_request.py +12 -0
- vellum/types/node_input_compiled_array_value.py +3 -7
- vellum/types/node_input_compiled_chat_history_value.py +3 -7
- vellum/types/node_input_compiled_error_value.py +3 -7
- vellum/types/node_input_compiled_function_call.py +3 -7
- vellum/types/node_input_compiled_json_value.py +3 -7
- vellum/types/node_input_compiled_number_value.py +3 -7
- vellum/types/node_input_compiled_search_results_value.py +3 -7
- vellum/types/node_input_compiled_string_value.py +3 -7
- vellum/types/node_output_compiled_array_value.py +3 -7
- vellum/types/node_output_compiled_chat_history_value.py +3 -7
- vellum/types/node_output_compiled_error_value.py +3 -7
- vellum/types/node_output_compiled_function_value.py +3 -7
- vellum/types/node_output_compiled_json_value.py +3 -7
- vellum/types/node_output_compiled_number_value.py +3 -7
- vellum/types/node_output_compiled_search_results_value.py +3 -7
- vellum/types/node_output_compiled_string_value.py +3 -7
- vellum/types/normalized_log_probs.py +3 -7
- vellum/types/normalized_token_log_probs.py +3 -7
- vellum/types/number_variable_value.py +3 -7
- vellum/types/paginated_document_index_read_list.py +3 -7
- vellum/types/paginated_slim_deployment_read_list.py +3 -7
- vellum/types/paginated_slim_document_list.py +3 -7
- vellum/types/paginated_slim_workflow_deployment_list.py +3 -7
- vellum/types/paginated_test_suite_run_execution_list.py +3 -7
- vellum/types/paginated_test_suite_test_case_list.py +3 -7
- vellum/types/processing_failure_reason_enum.py +1 -20
- vellum/types/processing_state_enum.py +1 -32
- vellum/types/prompt_deployment_expand_meta_request_request.py +9 -13
- vellum/types/prompt_execution_meta.py +3 -7
- vellum/types/prompt_node_result.py +3 -7
- vellum/types/prompt_node_result_data.py +3 -7
- vellum/types/raw_prompt_execution_overrides_request.py +5 -9
- vellum/types/rejected_execute_prompt_event.py +3 -7
- vellum/types/rejected_execute_prompt_response.py +5 -9
- vellum/types/rejected_execute_workflow_workflow_result_event.py +3 -7
- vellum/types/rejected_function_call.py +3 -7
- vellum/types/rejected_prompt_execution_meta.py +3 -7
- vellum/types/rejected_workflow_node_result_event.py +3 -7
- vellum/types/sandbox_scenario.py +5 -9
- vellum/types/scenario_input.py +18 -21
- vellum/types/{registered_prompt_sandbox.py → scenario_input_chat_history_variable_value.py} +7 -13
- vellum/types/{register_prompt_error_response.py → scenario_input_string_variable_value.py} +7 -9
- vellum/types/search_error_response.py +4 -8
- vellum/types/search_filters_request.py +5 -9
- vellum/types/search_node_result.py +3 -7
- vellum/types/search_node_result_data.py +4 -8
- vellum/types/search_request_options_request.py +7 -11
- vellum/types/search_response.py +4 -8
- vellum/types/search_result.py +6 -10
- vellum/types/search_result_document.py +7 -11
- vellum/types/search_result_document_request.py +6 -10
- vellum/types/search_result_merging_request.py +4 -8
- vellum/types/search_result_request.py +6 -10
- vellum/types/search_results_variable_value.py +3 -7
- vellum/types/search_weights_request.py +5 -9
- vellum/types/slim_deployment_read.py +7 -11
- vellum/types/slim_document.py +12 -16
- vellum/types/slim_workflow_deployment.py +9 -13
- vellum/types/streaming_execute_prompt_event.py +4 -8
- vellum/types/streaming_prompt_execution_meta.py +3 -7
- vellum/types/streaming_workflow_node_result_event.py +3 -7
- vellum/types/string_chat_message_content.py +3 -7
- vellum/types/string_chat_message_content_request.py +3 -7
- vellum/types/string_input_request.py +4 -8
- vellum/types/string_variable_value.py +3 -7
- vellum/types/submit_completion_actual_request.py +8 -12
- vellum/types/submit_completion_actuals_error_response.py +3 -7
- vellum/types/subworkflow_node_result.py +3 -7
- vellum/types/templating_node_array_result.py +3 -7
- vellum/types/templating_node_chat_history_result.py +3 -7
- vellum/types/templating_node_error_result.py +3 -7
- vellum/types/templating_node_function_call_result.py +3 -7
- vellum/types/templating_node_json_result.py +3 -7
- vellum/types/templating_node_number_result.py +3 -7
- vellum/types/templating_node_result.py +3 -7
- vellum/types/templating_node_result_data.py +3 -7
- vellum/types/templating_node_search_results_result.py +3 -7
- vellum/types/templating_node_string_result.py +3 -7
- vellum/types/terminal_node_array_result.py +4 -8
- vellum/types/terminal_node_chat_history_result.py +4 -8
- vellum/types/terminal_node_error_result.py +4 -8
- vellum/types/terminal_node_function_call_result.py +4 -8
- vellum/types/terminal_node_json_result.py +4 -8
- vellum/types/terminal_node_number_result.py +4 -8
- vellum/types/terminal_node_result.py +3 -7
- vellum/types/terminal_node_result_data.py +3 -7
- vellum/types/terminal_node_search_results_result.py +4 -8
- vellum/types/terminal_node_string_result.py +4 -8
- vellum/types/test_case_chat_history_variable_value.py +3 -7
- vellum/types/test_case_error_variable_value.py +3 -7
- vellum/types/test_case_function_call_variable_value.py +32 -0
- vellum/types/test_case_json_variable_value.py +3 -7
- vellum/types/test_case_number_variable_value.py +3 -7
- vellum/types/test_case_search_results_variable_value.py +3 -7
- vellum/types/test_case_string_variable_value.py +3 -7
- vellum/types/test_case_variable_value.py +12 -0
- vellum/types/test_suite_run_deployment_release_tag_exec_config.py +4 -8
- vellum/types/test_suite_run_deployment_release_tag_exec_config_data.py +5 -9
- vellum/types/test_suite_run_deployment_release_tag_exec_config_data_request.py +5 -9
- vellum/types/test_suite_run_deployment_release_tag_exec_config_request.py +4 -8
- vellum/types/test_suite_run_execution.py +3 -7
- vellum/types/test_suite_run_execution_chat_history_output.py +3 -7
- vellum/types/test_suite_run_execution_error_output.py +3 -7
- vellum/types/{registered_prompt_model_version.py → test_suite_run_execution_function_call_output.py} +8 -13
- vellum/types/test_suite_run_execution_json_output.py +3 -7
- vellum/types/test_suite_run_execution_metric_definition.py +3 -7
- vellum/types/test_suite_run_execution_metric_result.py +3 -7
- vellum/types/test_suite_run_execution_number_output.py +3 -7
- vellum/types/test_suite_run_execution_output.py +12 -0
- vellum/types/test_suite_run_execution_search_results_output.py +3 -7
- vellum/types/test_suite_run_execution_string_output.py +3 -7
- vellum/types/test_suite_run_external_exec_config.py +4 -8
- vellum/types/test_suite_run_external_exec_config_data.py +4 -8
- vellum/types/test_suite_run_external_exec_config_data_request.py +4 -8
- vellum/types/test_suite_run_external_exec_config_request.py +4 -8
- vellum/types/test_suite_run_metric_error_output.py +3 -7
- vellum/types/test_suite_run_metric_number_output.py +3 -7
- vellum/types/test_suite_run_metric_string_output.py +3 -7
- vellum/types/test_suite_run_read.py +5 -9
- vellum/types/test_suite_run_state.py +1 -37
- vellum/types/test_suite_run_test_suite.py +3 -7
- vellum/types/test_suite_run_workflow_release_tag_exec_config.py +4 -8
- vellum/types/test_suite_run_workflow_release_tag_exec_config_data.py +5 -9
- vellum/types/test_suite_run_workflow_release_tag_exec_config_data_request.py +5 -9
- vellum/types/test_suite_run_workflow_release_tag_exec_config_request.py +4 -8
- vellum/types/test_suite_test_case.py +3 -7
- vellum/types/upload_document_error_response.py +3 -7
- vellum/types/upload_document_response.py +4 -8
- vellum/types/vellum_error.py +3 -7
- vellum/types/vellum_error_code_enum.py +3 -32
- vellum/types/vellum_error_request.py +3 -7
- vellum/types/vellum_image.py +3 -7
- vellum/types/vellum_image_request.py +3 -7
- vellum/types/vellum_variable.py +3 -7
- vellum/types/vellum_variable_type.py +6 -57
- vellum/types/workflow_deployment_read.py +9 -13
- vellum/types/workflow_event_error.py +3 -7
- vellum/types/workflow_execution_actual_chat_history_request.py +8 -12
- vellum/types/workflow_execution_actual_json_request.py +8 -12
- vellum/types/workflow_execution_actual_string_request.py +8 -12
- vellum/types/workflow_execution_event_error_code.py +13 -52
- vellum/types/workflow_execution_event_type.py +1 -18
- vellum/types/workflow_execution_node_result_event.py +3 -7
- vellum/types/workflow_execution_workflow_result_event.py +3 -7
- vellum/types/workflow_node_result_event_state.py +3 -32
- vellum/types/workflow_output_array.py +4 -8
- vellum/types/workflow_output_chat_history.py +4 -8
- vellum/types/workflow_output_error.py +4 -8
- vellum/types/workflow_output_function_call.py +4 -8
- vellum/types/workflow_output_image.py +4 -8
- vellum/types/workflow_output_json.py +4 -8
- vellum/types/workflow_output_number.py +4 -8
- vellum/types/workflow_output_search_results.py +4 -8
- vellum/types/workflow_output_string.py +4 -8
- vellum/types/workflow_request_chat_history_input_request.py +4 -8
- vellum/types/workflow_request_json_input_request.py +4 -8
- vellum/types/workflow_request_number_input_request.py +4 -8
- vellum/types/workflow_request_string_input_request.py +4 -8
- vellum/types/workflow_result_event.py +3 -7
- vellum/types/workflow_result_event_output_data_array.py +4 -8
- vellum/types/workflow_result_event_output_data_chat_history.py +4 -8
- vellum/types/workflow_result_event_output_data_error.py +4 -8
- vellum/types/workflow_result_event_output_data_function_call.py +4 -8
- vellum/types/workflow_result_event_output_data_json.py +4 -8
- vellum/types/workflow_result_event_output_data_number.py +4 -8
- vellum/types/workflow_result_event_output_data_search_results.py +4 -8
- vellum/types/workflow_result_event_output_data_string.py +5 -9
- {vellum_ai-0.3.23.dist-info → vellum_ai-0.5.1.dist-info}/METADATA +1 -2
- vellum_ai-0.5.1.dist-info/RECORD +382 -0
- vellum/errors/conflict_error.py +0 -9
- vellum/resources/model_versions/__init__.py +0 -2
- vellum/resources/model_versions/client.py +0 -123
- vellum/resources/registered_prompts/__init__.py +0 -2
- vellum/resources/registered_prompts/client.py +0 -327
- vellum/types/block_type_enum.py +0 -36
- vellum/types/model_version_build_config.py +0 -40
- vellum/types/model_version_exec_config.py +0 -48
- vellum/types/model_version_exec_config_parameters.py +0 -37
- vellum/types/model_version_read.py +0 -82
- vellum/types/model_version_read_status_enum.py +0 -36
- vellum/types/model_version_sandbox_snapshot.py +0 -44
- vellum/types/prompt_template_block.py +0 -41
- vellum/types/prompt_template_block_data.py +0 -31
- vellum/types/prompt_template_block_data_request.py +0 -31
- vellum/types/prompt_template_block_properties.py +0 -47
- vellum/types/prompt_template_block_properties_request.py +0 -47
- vellum/types/prompt_template_block_request.py +0 -41
- vellum/types/prompt_template_block_state.py +0 -22
- vellum/types/provider_enum.py +0 -81
- vellum/types/register_prompt_model_parameters_request.py +0 -37
- vellum/types/register_prompt_prompt_info_request.py +0 -35
- vellum/types/register_prompt_response.py +0 -62
- vellum/types/registered_prompt_deployment.py +0 -42
- vellum/types/scenario_input_request.py +0 -34
- vellum/types/scenario_input_type_enum.py +0 -22
- vellum_ai-0.3.23.dist-info/RECORD +0 -394
- {vellum_ai-0.3.23.dist-info → vellum_ai-0.5.1.dist-info}/LICENSE +0 -0
- {vellum_ai-0.3.23.dist-info → vellum_ai-0.5.1.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
|
-
|
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] =
|
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] =
|
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] =
|
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] =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
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 =
|
31
|
+
extra = pydantic_v1.Extra.allow
|
36
32
|
json_encoders = {dt.datetime: serialize_datetime}
|