vellum-ai 0.3.14__py3-none-any.whl → 0.3.16__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- vellum/__init__.py +22 -0
- vellum/client.py +846 -100
- vellum/core/__init__.py +8 -0
- vellum/core/client_wrapper.py +27 -8
- vellum/core/file.py +38 -0
- vellum/core/http_client.py +130 -0
- vellum/core/jsonable_encoder.py +2 -2
- vellum/core/request_options.py +32 -0
- vellum/resources/deployments/client.py +161 -24
- vellum/resources/document_indexes/client.py +337 -80
- vellum/resources/documents/client.py +278 -58
- vellum/resources/folder_entities/client.py +60 -10
- vellum/resources/model_versions/client.py +48 -8
- vellum/resources/registered_prompts/client.py +52 -6
- vellum/resources/sandboxes/client.py +108 -26
- vellum/resources/test_suite_runs/client.py +221 -38
- vellum/resources/test_suites/client.py +110 -24
- vellum/resources/workflow_deployments/client.py +110 -14
- vellum/types/__init__.py +20 -0
- vellum/types/api_node_result.py +1 -0
- vellum/types/api_node_result_data.py +4 -2
- vellum/types/array_chat_message_content.py +1 -0
- vellum/types/array_chat_message_content_item.py +6 -5
- vellum/types/array_chat_message_content_item_request.py +6 -5
- vellum/types/array_chat_message_content_request.py +1 -0
- vellum/types/array_enum.py +2 -2
- vellum/types/array_variable_value_item.py +16 -10
- vellum/types/chat_history_enum.py +2 -2
- vellum/types/chat_history_input_request.py +6 -1
- vellum/types/chat_history_variable_value.py +2 -1
- vellum/types/chat_message.py +7 -5
- vellum/types/chat_message_content.py +8 -6
- vellum/types/chat_message_content_request.py +8 -6
- vellum/types/chat_message_request.py +7 -5
- vellum/types/code_execution_node_array_result.py +31 -0
- vellum/types/code_execution_node_chat_history_result.py +2 -1
- vellum/types/code_execution_node_error_result.py +2 -1
- vellum/types/code_execution_node_function_call_result.py +31 -0
- vellum/types/code_execution_node_json_result.py +2 -1
- vellum/types/code_execution_node_number_result.py +2 -1
- vellum/types/code_execution_node_result.py +1 -0
- vellum/types/code_execution_node_result_data.py +2 -1
- vellum/types/code_execution_node_result_output.py +36 -8
- vellum/types/code_execution_node_search_results_result.py +2 -1
- vellum/types/code_execution_node_string_result.py +2 -1
- vellum/types/conditional_node_result.py +1 -0
- vellum/types/conditional_node_result_data.py +2 -1
- vellum/types/deployment_provider_payload_response.py +1 -0
- vellum/types/deployment_read.py +32 -17
- vellum/types/document_document_to_document_index.py +21 -15
- vellum/types/document_index_read.py +32 -19
- vellum/types/document_read.py +34 -24
- vellum/types/document_status.py +2 -2
- vellum/types/enriched_normalized_completion.py +36 -19
- vellum/types/error_enum.py +2 -2
- vellum/types/error_variable_value.py +2 -1
- vellum/types/execute_prompt_api_error_response.py +5 -1
- vellum/types/execute_prompt_event.py +8 -6
- vellum/types/execute_prompt_response.py +4 -4
- vellum/types/execute_workflow_error_response.py +5 -1
- vellum/types/execute_workflow_response.py +3 -2
- vellum/types/execute_workflow_stream_error_response.py +5 -1
- vellum/types/execute_workflow_workflow_result_event.py +4 -4
- vellum/types/execution_array_vellum_value.py +7 -2
- vellum/types/execution_chat_history_vellum_value.py +7 -2
- vellum/types/execution_error_vellum_value.py +7 -2
- vellum/types/execution_function_call_vellum_value.py +7 -2
- vellum/types/execution_json_vellum_value.py +7 -2
- vellum/types/execution_number_vellum_value.py +7 -2
- vellum/types/execution_search_results_vellum_value.py +7 -2
- vellum/types/execution_string_vellum_value.py +7 -2
- vellum/types/execution_vellum_value.py +16 -10
- vellum/types/fulfilled_enum.py +2 -2
- vellum/types/fulfilled_execute_prompt_event.py +2 -1
- vellum/types/fulfilled_execute_prompt_response.py +12 -5
- vellum/types/fulfilled_execute_workflow_workflow_result_event.py +1 -0
- vellum/types/fulfilled_function_call.py +2 -1
- vellum/types/fulfilled_prompt_execution_meta.py +3 -2
- vellum/types/fulfilled_workflow_node_result_event.py +6 -5
- vellum/types/function_call.py +4 -4
- vellum/types/function_call_chat_message_content.py +1 -0
- vellum/types/function_call_chat_message_content_request.py +1 -0
- vellum/types/function_call_chat_message_content_value.py +2 -1
- vellum/types/function_call_chat_message_content_value_request.py +2 -1
- vellum/types/function_call_enum.py +2 -2
- vellum/types/function_call_variable_value.py +1 -0
- vellum/types/generate_error_response.py +5 -1
- vellum/types/generate_options_request.py +8 -5
- vellum/types/generate_request.py +15 -9
- vellum/types/generate_result.py +10 -6
- vellum/types/generate_result_data.py +5 -3
- vellum/types/generate_result_error.py +5 -1
- vellum/types/generate_stream_response.py +1 -0
- vellum/types/generate_stream_result.py +3 -2
- vellum/types/generate_stream_result_data.py +1 -0
- vellum/types/image_chat_message_content.py +1 -0
- vellum/types/image_chat_message_content_request.py +1 -0
- vellum/types/image_enum.py +2 -2
- vellum/types/image_variable_value.py +2 -1
- vellum/types/initiated_enum.py +2 -2
- vellum/types/initiated_execute_prompt_event.py +2 -1
- vellum/types/initiated_prompt_execution_meta.py +5 -4
- vellum/types/initiated_workflow_node_result_event.py +5 -4
- vellum/types/json_enum.py +2 -2
- vellum/types/json_input_request.py +6 -1
- vellum/types/json_variable_value.py +2 -1
- vellum/types/logprobs_enum.py +2 -2
- vellum/types/metadata_filter_config_request.py +7 -6
- vellum/types/metadata_filter_rule_request.py +7 -6
- vellum/types/model_version_build_config.py +12 -7
- vellum/types/model_version_exec_config.py +18 -11
- vellum/types/model_version_exec_config_parameters.py +7 -6
- vellum/types/model_version_read.py +51 -32
- vellum/types/model_version_sandbox_snapshot.py +17 -4
- vellum/types/named_test_case_chat_history_variable_value_request.py +2 -1
- vellum/types/named_test_case_error_variable_value_request.py +2 -1
- vellum/types/named_test_case_json_variable_value_request.py +2 -1
- vellum/types/named_test_case_number_variable_value_request.py +2 -1
- vellum/types/named_test_case_search_results_variable_value_request.py +2 -1
- vellum/types/named_test_case_string_variable_value_request.py +2 -1
- vellum/types/named_test_case_variable_value_request.py +12 -8
- vellum/types/node_input_compiled_array_value.py +2 -1
- vellum/types/node_input_compiled_chat_history_value.py +2 -1
- vellum/types/node_input_compiled_error_value.py +2 -1
- vellum/types/node_input_compiled_function_call.py +32 -0
- vellum/types/node_input_compiled_json_value.py +2 -1
- vellum/types/node_input_compiled_number_value.py +2 -1
- vellum/types/node_input_compiled_search_results_value.py +2 -1
- vellum/types/node_input_compiled_string_value.py +2 -1
- vellum/types/node_input_variable_compiled_value.py +26 -9
- vellum/types/node_output_compiled_array_value.py +2 -1
- vellum/types/node_output_compiled_chat_history_value.py +2 -1
- vellum/types/node_output_compiled_error_value.py +2 -1
- vellum/types/node_output_compiled_function_value.py +2 -1
- vellum/types/node_output_compiled_json_value.py +2 -1
- vellum/types/node_output_compiled_number_value.py +2 -1
- vellum/types/node_output_compiled_search_results_value.py +2 -1
- vellum/types/node_output_compiled_string_value.py +2 -1
- vellum/types/node_output_compiled_value.py +16 -10
- vellum/types/normalized_log_probs.py +2 -1
- vellum/types/normalized_token_log_probs.py +3 -2
- vellum/types/number_enum.py +2 -2
- vellum/types/number_variable_value.py +2 -1
- vellum/types/paginated_document_index_read_list.py +5 -4
- vellum/types/paginated_slim_deployment_read_list.py +5 -4
- vellum/types/paginated_slim_document_list.py +5 -4
- vellum/types/paginated_slim_workflow_deployment_list.py +5 -4
- vellum/types/paginated_test_suite_run_execution_list.py +5 -4
- vellum/types/prompt_deployment_expand_meta_request_request.py +25 -15
- vellum/types/prompt_deployment_input_request.py +6 -5
- vellum/types/prompt_execution_meta.py +6 -5
- vellum/types/prompt_node_result.py +1 -0
- vellum/types/prompt_node_result_data.py +4 -3
- vellum/types/prompt_output.py +8 -6
- vellum/types/prompt_template_block.py +1 -0
- vellum/types/prompt_template_block_data.py +1 -0
- vellum/types/prompt_template_block_data_request.py +1 -0
- vellum/types/prompt_template_block_properties.py +11 -10
- vellum/types/prompt_template_block_properties_request.py +11 -10
- vellum/types/prompt_template_block_request.py +1 -0
- vellum/types/raw_prompt_execution_overrides_request.py +11 -5
- vellum/types/register_prompt_error_response.py +5 -1
- vellum/types/register_prompt_model_parameters_request.py +5 -4
- vellum/types/register_prompt_prompt.py +10 -2
- vellum/types/register_prompt_prompt_info_request.py +5 -3
- vellum/types/register_prompt_response.py +30 -10
- vellum/types/registered_prompt_deployment.py +15 -3
- vellum/types/registered_prompt_input_variable_request.py +3 -2
- vellum/types/registered_prompt_model_version.py +10 -2
- vellum/types/registered_prompt_sandbox.py +10 -2
- vellum/types/registered_prompt_sandbox_snapshot.py +5 -1
- vellum/types/rejected_enum.py +2 -2
- vellum/types/rejected_execute_prompt_event.py +2 -1
- vellum/types/rejected_execute_prompt_response.py +12 -5
- vellum/types/rejected_execute_workflow_workflow_result_event.py +1 -0
- vellum/types/rejected_function_call.py +2 -1
- vellum/types/rejected_prompt_execution_meta.py +3 -2
- vellum/types/rejected_workflow_node_result_event.py +4 -3
- vellum/types/sandbox_scenario.py +11 -3
- vellum/types/scenario_input.py +4 -3
- vellum/types/scenario_input_request.py +4 -3
- vellum/types/search_error_response.py +5 -1
- vellum/types/search_filters_request.py +10 -6
- vellum/types/search_node_result.py +1 -0
- vellum/types/search_node_result_data.py +7 -4
- vellum/types/search_request_options_request.py +20 -8
- vellum/types/search_response.py +5 -3
- vellum/types/search_result.py +15 -5
- vellum/types/search_result_document.py +20 -8
- vellum/types/search_result_document_request.py +15 -7
- vellum/types/search_result_merging_request.py +5 -1
- vellum/types/search_result_request.py +15 -5
- vellum/types/search_results_enum.py +2 -2
- vellum/types/search_results_variable_value.py +2 -1
- vellum/types/search_weights_request.py +10 -4
- vellum/types/slim_deployment_read.py +28 -14
- vellum/types/slim_document.py +56 -35
- vellum/types/slim_workflow_deployment.py +37 -24
- vellum/types/streaming_enum.py +2 -2
- vellum/types/streaming_execute_prompt_event.py +6 -4
- vellum/types/streaming_prompt_execution_meta.py +2 -1
- vellum/types/streaming_workflow_node_result_event.py +6 -5
- vellum/types/string_chat_message_content.py +1 -0
- vellum/types/string_chat_message_content_request.py +1 -0
- vellum/types/string_enum.py +2 -2
- vellum/types/string_input_request.py +6 -1
- vellum/types/string_variable_value.py +2 -1
- vellum/types/submit_completion_actual_request.py +25 -13
- vellum/types/submit_completion_actuals_error_response.py +1 -0
- vellum/types/submit_workflow_execution_actual_request.py +6 -5
- vellum/types/subworkflow_enum.py +2 -2
- vellum/types/subworkflow_node_result.py +1 -0
- vellum/types/templating_node_array_result.py +31 -0
- vellum/types/templating_node_chat_history_result.py +2 -1
- vellum/types/templating_node_error_result.py +2 -1
- vellum/types/templating_node_function_call_result.py +31 -0
- vellum/types/templating_node_json_result.py +2 -1
- vellum/types/templating_node_number_result.py +2 -1
- vellum/types/templating_node_result.py +1 -0
- vellum/types/templating_node_result_data.py +1 -0
- vellum/types/templating_node_result_output.py +36 -8
- vellum/types/templating_node_search_results_result.py +2 -1
- vellum/types/templating_node_string_result.py +2 -1
- vellum/types/terminal_node_array_result.py +8 -3
- vellum/types/terminal_node_chat_history_result.py +8 -3
- vellum/types/terminal_node_error_result.py +8 -3
- vellum/types/terminal_node_function_call_result.py +8 -3
- vellum/types/terminal_node_json_result.py +8 -3
- vellum/types/terminal_node_number_result.py +8 -3
- vellum/types/terminal_node_result.py +1 -0
- vellum/types/terminal_node_result_data.py +1 -0
- vellum/types/terminal_node_result_output.py +16 -10
- vellum/types/terminal_node_search_results_result.py +8 -3
- vellum/types/terminal_node_string_result.py +8 -3
- vellum/types/test_case_chat_history_variable_value.py +2 -1
- vellum/types/test_case_error_variable_value.py +2 -1
- vellum/types/test_case_json_variable_value.py +2 -1
- vellum/types/test_case_number_variable_value.py +2 -1
- vellum/types/test_case_search_results_variable_value.py +2 -1
- vellum/types/test_case_string_variable_value.py +2 -1
- vellum/types/test_case_variable_value.py +12 -8
- vellum/types/test_suite_run_deployment_release_tag_exec_config.py +5 -3
- vellum/types/test_suite_run_deployment_release_tag_exec_config_data.py +10 -4
- vellum/types/test_suite_run_deployment_release_tag_exec_config_data_request.py +10 -4
- vellum/types/test_suite_run_deployment_release_tag_exec_config_request.py +5 -3
- vellum/types/test_suite_run_deployment_release_tag_exec_config_type_enum.py +2 -2
- vellum/types/test_suite_run_exec_config.py +4 -4
- vellum/types/test_suite_run_exec_config_request.py +4 -4
- vellum/types/test_suite_run_execution.py +1 -0
- vellum/types/test_suite_run_execution_chat_history_output.py +2 -1
- vellum/types/test_suite_run_execution_error_output.py +2 -1
- vellum/types/test_suite_run_execution_json_output.py +2 -1
- vellum/types/test_suite_run_execution_metric_result.py +1 -0
- vellum/types/test_suite_run_execution_number_output.py +2 -1
- vellum/types/test_suite_run_execution_output.py +12 -8
- vellum/types/test_suite_run_execution_search_results_output.py +2 -1
- vellum/types/test_suite_run_execution_string_output.py +2 -1
- vellum/types/test_suite_run_metric_error_output.py +1 -0
- vellum/types/test_suite_run_metric_error_output_type_enum.py +2 -2
- vellum/types/test_suite_run_metric_number_output.py +1 -0
- vellum/types/test_suite_run_metric_number_output_type_enum.py +2 -2
- vellum/types/test_suite_run_metric_output.py +4 -4
- vellum/types/test_suite_run_read.py +16 -14
- vellum/types/test_suite_run_test_suite.py +1 -0
- vellum/types/test_suite_run_workflow_release_tag_exec_config.py +5 -3
- vellum/types/test_suite_run_workflow_release_tag_exec_config_data.py +10 -6
- vellum/types/test_suite_run_workflow_release_tag_exec_config_data_request.py +10 -6
- vellum/types/test_suite_run_workflow_release_tag_exec_config_request.py +5 -3
- vellum/types/test_suite_run_workflow_release_tag_exec_config_type_enum.py +2 -2
- vellum/types/test_suite_test_case.py +3 -2
- vellum/types/upload_document_error_response.py +1 -0
- vellum/types/upload_document_response.py +5 -1
- vellum/types/vellum_error.py +1 -0
- vellum/types/vellum_error_request.py +1 -0
- vellum/types/vellum_image.py +2 -1
- vellum/types/vellum_image_request.py +2 -1
- vellum/types/vellum_variable.py +1 -0
- vellum/types/workflow_deployment_read.py +37 -24
- vellum/types/workflow_event_error.py +1 -0
- vellum/types/workflow_execution_actual_chat_history_request.py +25 -15
- vellum/types/workflow_execution_actual_json_request.py +25 -15
- vellum/types/workflow_execution_actual_string_request.py +25 -15
- vellum/types/workflow_execution_node_result_event.py +3 -2
- vellum/types/workflow_execution_workflow_result_event.py +3 -2
- vellum/types/workflow_node_result_data.py +16 -10
- vellum/types/workflow_node_result_event.py +8 -6
- vellum/types/workflow_output.py +18 -11
- vellum/types/workflow_output_array.py +7 -2
- vellum/types/workflow_output_chat_history.py +7 -2
- vellum/types/workflow_output_error.py +7 -2
- vellum/types/workflow_output_function_call.py +7 -2
- vellum/types/workflow_output_image.py +7 -2
- vellum/types/workflow_output_json.py +7 -2
- vellum/types/workflow_output_number.py +7 -2
- vellum/types/workflow_output_search_results.py +7 -2
- vellum/types/workflow_output_string.py +7 -2
- vellum/types/workflow_request_chat_history_input_request.py +6 -1
- vellum/types/workflow_request_input_request.py +8 -6
- vellum/types/workflow_request_json_input_request.py +6 -1
- vellum/types/workflow_request_number_input_request.py +6 -1
- vellum/types/workflow_request_string_input_request.py +6 -1
- vellum/types/workflow_result_event.py +5 -4
- vellum/types/workflow_result_event_output_data.py +16 -10
- vellum/types/workflow_result_event_output_data_array.py +8 -5
- vellum/types/workflow_result_event_output_data_chat_history.py +8 -5
- vellum/types/workflow_result_event_output_data_error.py +8 -5
- vellum/types/workflow_result_event_output_data_function_call.py +8 -5
- vellum/types/workflow_result_event_output_data_json.py +8 -5
- vellum/types/workflow_result_event_output_data_number.py +8 -5
- vellum/types/workflow_result_event_output_data_search_results.py +8 -5
- vellum/types/workflow_result_event_output_data_string.py +11 -7
- vellum/types/workflow_stream_event.py +4 -4
- vellum/version.py +4 -0
- {vellum_ai-0.3.14.dist-info → vellum_ai-0.3.16.dist-info}/METADATA +5 -4
- vellum_ai-0.3.16.dist-info/RECORD +365 -0
- vellum_ai-0.3.14.dist-info/RECORD +0 -356
- {vellum_ai-0.3.14.dist-info → vellum_ai-0.3.16.dist-info}/LICENSE +0 -0
- {vellum_ai-0.3.14.dist-info → vellum_ai-0.3.16.dist-info}/WHEEL +0 -0
@@ -12,12 +12,15 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class TestSuiteRunWorkflowReleaseTagExecConfigData(pydantic.BaseModel):
|
15
|
-
workflow_deployment_id: str = pydantic.Field(
|
16
|
-
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
)
|
15
|
+
workflow_deployment_id: str = pydantic.Field()
|
16
|
+
"""
|
17
|
+
The ID of the Workflow Deployment to run the Test Suite against.
|
18
|
+
"""
|
19
|
+
|
20
|
+
tag: typing.Optional[str] = pydantic.Field(default=None)
|
21
|
+
"""
|
22
|
+
A tag identifying which release of the Workflow Deployment to run the Test Suite against. Useful for testing past versions of the Workflow Deployment
|
23
|
+
"""
|
21
24
|
|
22
25
|
def json(self, **kwargs: typing.Any) -> str:
|
23
26
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -30,4 +33,5 @@ class TestSuiteRunWorkflowReleaseTagExecConfigData(pydantic.BaseModel):
|
|
30
33
|
class Config:
|
31
34
|
frozen = True
|
32
35
|
smart_union = True
|
36
|
+
extra = pydantic.Extra.allow
|
33
37
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -12,12 +12,15 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class TestSuiteRunWorkflowReleaseTagExecConfigDataRequest(pydantic.BaseModel):
|
15
|
-
workflow_deployment_id: str = pydantic.Field(
|
16
|
-
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
)
|
15
|
+
workflow_deployment_id: str = pydantic.Field()
|
16
|
+
"""
|
17
|
+
The ID of the Workflow Deployment to run the Test Suite against.
|
18
|
+
"""
|
19
|
+
|
20
|
+
tag: typing.Optional[str] = pydantic.Field(default=None)
|
21
|
+
"""
|
22
|
+
A tag identifying which release of the Workflow Deployment to run the Test Suite against. Useful for testing past versions of the Workflow Deployment
|
23
|
+
"""
|
21
24
|
|
22
25
|
def json(self, **kwargs: typing.Any) -> str:
|
23
26
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -30,4 +33,5 @@ class TestSuiteRunWorkflowReleaseTagExecConfigDataRequest(pydantic.BaseModel):
|
|
30
33
|
class Config:
|
31
34
|
frozen = True
|
32
35
|
smart_union = True
|
36
|
+
extra = pydantic.Extra.allow
|
33
37
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -20,9 +20,10 @@ class TestSuiteRunWorkflowReleaseTagExecConfigRequest(pydantic.BaseModel):
|
|
20
20
|
"""
|
21
21
|
|
22
22
|
data: TestSuiteRunWorkflowReleaseTagExecConfigDataRequest
|
23
|
-
test_case_ids: typing.Optional[typing.List[str]] = pydantic.Field(
|
24
|
-
|
25
|
-
|
23
|
+
test_case_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
|
24
|
+
"""
|
25
|
+
Optionally specify a subset of test case ids to run. If not provided, all test cases within the test suite will be run by default.
|
26
|
+
"""
|
26
27
|
|
27
28
|
def json(self, **kwargs: typing.Any) -> str:
|
28
29
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -35,4 +36,5 @@ class TestSuiteRunWorkflowReleaseTagExecConfigRequest(pydantic.BaseModel):
|
|
35
36
|
class Config:
|
36
37
|
frozen = True
|
37
38
|
smart_union = True
|
39
|
+
extra = pydantic.Extra.allow
|
38
40
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -1,5 +1,5 @@
|
|
1
1
|
# This file was auto-generated by Fern from our API Definition.
|
2
2
|
|
3
|
-
import
|
3
|
+
import typing
|
4
4
|
|
5
|
-
TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum =
|
5
|
+
TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum = typing.Literal["WORKFLOW_RELEASE_TAG"]
|
@@ -13,8 +13,8 @@ except ImportError:
|
|
13
13
|
|
14
14
|
|
15
15
|
class TestSuiteTestCase(pydantic.BaseModel):
|
16
|
-
id: typing.Optional[str]
|
17
|
-
label: typing.Optional[str]
|
16
|
+
id: typing.Optional[str] = None
|
17
|
+
label: typing.Optional[str] = None
|
18
18
|
input_values: typing.List[TestCaseVariableValue]
|
19
19
|
evaluation_values: typing.List[TestCaseVariableValue]
|
20
20
|
|
@@ -29,4 +29,5 @@ class TestSuiteTestCase(pydantic.BaseModel):
|
|
29
29
|
class Config:
|
30
30
|
frozen = True
|
31
31
|
smart_union = True
|
32
|
+
extra = pydantic.Extra.allow
|
32
33
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -12,7 +12,10 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class UploadDocumentResponse(pydantic.BaseModel):
|
15
|
-
document_id: str = pydantic.Field(
|
15
|
+
document_id: str = pydantic.Field()
|
16
|
+
"""
|
17
|
+
The ID of the newly created document.
|
18
|
+
"""
|
16
19
|
|
17
20
|
def json(self, **kwargs: typing.Any) -> str:
|
18
21
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -25,4 +28,5 @@ class UploadDocumentResponse(pydantic.BaseModel):
|
|
25
28
|
class Config:
|
26
29
|
frozen = True
|
27
30
|
smart_union = True
|
31
|
+
extra = pydantic.Extra.allow
|
28
32
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/vellum_error.py
CHANGED
vellum/types/vellum_image.py
CHANGED
@@ -13,7 +13,7 @@ except ImportError:
|
|
13
13
|
|
14
14
|
class VellumImage(pydantic.BaseModel):
|
15
15
|
src: str
|
16
|
-
metadata: typing.Optional[typing.Dict[str, typing.Any]]
|
16
|
+
metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
|
17
17
|
|
18
18
|
def json(self, **kwargs: typing.Any) -> str:
|
19
19
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -26,4 +26,5 @@ class VellumImage(pydantic.BaseModel):
|
|
26
26
|
class Config:
|
27
27
|
frozen = True
|
28
28
|
smart_union = True
|
29
|
+
extra = pydantic.Extra.allow
|
29
30
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -13,7 +13,7 @@ except ImportError:
|
|
13
13
|
|
14
14
|
class VellumImageRequest(pydantic.BaseModel):
|
15
15
|
src: str
|
16
|
-
metadata: typing.Optional[typing.Dict[str, typing.Any]]
|
16
|
+
metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
|
17
17
|
|
18
18
|
def json(self, **kwargs: typing.Any) -> str:
|
19
19
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -26,4 +26,5 @@ class VellumImageRequest(pydantic.BaseModel):
|
|
26
26
|
class Config:
|
27
27
|
frozen = True
|
28
28
|
smart_union = True
|
29
|
+
extra = pydantic.Extra.allow
|
29
30
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/vellum_variable.py
CHANGED
@@ -16,32 +16,44 @@ except ImportError:
|
|
16
16
|
|
17
17
|
class WorkflowDeploymentRead(pydantic.BaseModel):
|
18
18
|
id: str
|
19
|
-
name: str = pydantic.Field(
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
31
|
-
|
32
|
-
|
33
|
-
|
34
|
-
|
35
|
-
|
36
|
-
|
19
|
+
name: str = pydantic.Field()
|
20
|
+
"""
|
21
|
+
A name that uniquely identifies this workflow deployment within its workspace
|
22
|
+
"""
|
23
|
+
|
24
|
+
label: str = pydantic.Field()
|
25
|
+
"""
|
26
|
+
A human-readable label for the workflow deployment
|
27
|
+
"""
|
28
|
+
|
29
|
+
status: typing.Optional[EntityStatus] = pydantic.Field(default=None)
|
30
|
+
"""
|
31
|
+
The current status of the workflow deployment
|
32
|
+
|
33
|
+
- `ACTIVE` - Active
|
34
|
+
- `ARCHIVED` - Archived
|
35
|
+
"""
|
36
|
+
|
37
|
+
environment: typing.Optional[EnvironmentEnum] = pydantic.Field(default=None)
|
38
|
+
"""
|
39
|
+
The environment this workflow deployment is used in
|
40
|
+
|
41
|
+
- `DEVELOPMENT` - Development
|
42
|
+
- `STAGING` - Staging
|
43
|
+
- `PRODUCTION` - Production
|
44
|
+
"""
|
45
|
+
|
37
46
|
created: dt.datetime
|
38
47
|
last_deployed_on: dt.datetime
|
39
|
-
input_variables: typing.List[VellumVariable] = pydantic.Field(
|
40
|
-
|
41
|
-
|
42
|
-
|
43
|
-
|
44
|
-
)
|
48
|
+
input_variables: typing.List[VellumVariable] = pydantic.Field()
|
49
|
+
"""
|
50
|
+
The input variables this Workflow Deployment expects to receive values for when it is executed.
|
51
|
+
"""
|
52
|
+
|
53
|
+
output_variables: typing.List[VellumVariable] = pydantic.Field()
|
54
|
+
"""
|
55
|
+
The output variables this Workflow Deployment produces values for when it's executed.
|
56
|
+
"""
|
45
57
|
|
46
58
|
def json(self, **kwargs: typing.Any) -> str:
|
47
59
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -54,4 +66,5 @@ class WorkflowDeploymentRead(pydantic.BaseModel):
|
|
54
66
|
class Config:
|
55
67
|
frozen = True
|
56
68
|
smart_union = True
|
69
|
+
extra = pydantic.Extra.allow
|
57
70
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -13,21 +13,30 @@ except ImportError:
|
|
13
13
|
|
14
14
|
|
15
15
|
class WorkflowExecutionActualChatHistoryRequest(pydantic.BaseModel):
|
16
|
-
output_id: typing.Optional[str] = pydantic.Field(
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
)
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
16
|
+
output_id: typing.Optional[str] = pydantic.Field(default=None)
|
17
|
+
"""
|
18
|
+
The Vellum-generated ID of a workflow output. Must provide either this or output_key. output_key is typically preferred.
|
19
|
+
"""
|
20
|
+
|
21
|
+
output_key: typing.Optional[str] = pydantic.Field(default=None)
|
22
|
+
"""
|
23
|
+
The user-defined name of a workflow output. Must provide either this or output_id. Should correspond to the `Name` specified in a Final Output Node. Generally preferred over output_id.
|
24
|
+
"""
|
25
|
+
|
26
|
+
quality: typing.Optional[float] = pydantic.Field(default=None)
|
27
|
+
"""
|
28
|
+
Optionally provide a decimal number between 0.0 and 1.0 (inclusive) representing the quality of the output. 0 is the worst, 1 is the best.
|
29
|
+
"""
|
30
|
+
|
31
|
+
timestamp: typing.Optional[float] = pydantic.Field(default=None)
|
32
|
+
"""
|
33
|
+
Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
|
34
|
+
"""
|
35
|
+
|
36
|
+
desired_output_value: typing.Optional[typing.List[ChatMessageRequest]] = pydantic.Field(default=None)
|
37
|
+
"""
|
38
|
+
Optionally provide the value that the output ideally should have been.
|
39
|
+
"""
|
31
40
|
|
32
41
|
def json(self, **kwargs: typing.Any) -> str:
|
33
42
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -40,4 +49,5 @@ class WorkflowExecutionActualChatHistoryRequest(pydantic.BaseModel):
|
|
40
49
|
class Config:
|
41
50
|
frozen = True
|
42
51
|
smart_union = True
|
52
|
+
extra = pydantic.Extra.allow
|
43
53
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -12,21 +12,30 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class WorkflowExecutionActualJsonRequest(pydantic.BaseModel):
|
15
|
-
output_id: typing.Optional[str] = pydantic.Field(
|
16
|
-
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
)
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
15
|
+
output_id: typing.Optional[str] = pydantic.Field(default=None)
|
16
|
+
"""
|
17
|
+
The Vellum-generated ID of a workflow output. Must provide either this or output_key. output_key is typically preferred.
|
18
|
+
"""
|
19
|
+
|
20
|
+
output_key: typing.Optional[str] = pydantic.Field(default=None)
|
21
|
+
"""
|
22
|
+
The user-defined name of a workflow output. Must provide either this or output_id. Should correspond to the `Name` specified in a Final Output Node. Generally preferred over output_id.
|
23
|
+
"""
|
24
|
+
|
25
|
+
quality: typing.Optional[float] = pydantic.Field(default=None)
|
26
|
+
"""
|
27
|
+
Optionally provide a decimal number between 0.0 and 1.0 (inclusive) representing the quality of the output. 0 is the worst, 1 is the best.
|
28
|
+
"""
|
29
|
+
|
30
|
+
timestamp: typing.Optional[float] = pydantic.Field(default=None)
|
31
|
+
"""
|
32
|
+
Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
|
33
|
+
"""
|
34
|
+
|
35
|
+
desired_output_value: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
|
36
|
+
"""
|
37
|
+
Optionally provide the value that the output ideally should have been.
|
38
|
+
"""
|
30
39
|
|
31
40
|
def json(self, **kwargs: typing.Any) -> str:
|
32
41
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -39,4 +48,5 @@ class WorkflowExecutionActualJsonRequest(pydantic.BaseModel):
|
|
39
48
|
class Config:
|
40
49
|
frozen = True
|
41
50
|
smart_union = True
|
51
|
+
extra = pydantic.Extra.allow
|
42
52
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -12,21 +12,30 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class WorkflowExecutionActualStringRequest(pydantic.BaseModel):
|
15
|
-
output_id: typing.Optional[str] = pydantic.Field(
|
16
|
-
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
)
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
15
|
+
output_id: typing.Optional[str] = pydantic.Field(default=None)
|
16
|
+
"""
|
17
|
+
The Vellum-generated ID of a workflow output. Must provide either this or output_key. output_key is typically preferred.
|
18
|
+
"""
|
19
|
+
|
20
|
+
output_key: typing.Optional[str] = pydantic.Field(default=None)
|
21
|
+
"""
|
22
|
+
The user-defined name of a workflow output. Must provide either this or output_id. Should correspond to the `Name` specified in a Final Output Node. Generally preferred over output_id.
|
23
|
+
"""
|
24
|
+
|
25
|
+
quality: typing.Optional[float] = pydantic.Field(default=None)
|
26
|
+
"""
|
27
|
+
Optionally provide a decimal number between 0.0 and 1.0 (inclusive) representing the quality of the output. 0 is the worst, 1 is the best.
|
28
|
+
"""
|
29
|
+
|
30
|
+
timestamp: typing.Optional[float] = pydantic.Field(default=None)
|
31
|
+
"""
|
32
|
+
Optionally provide the timestamp representing when this feedback was collected. Used for reporting purposes.
|
33
|
+
"""
|
34
|
+
|
35
|
+
desired_output_value: typing.Optional[str] = pydantic.Field(default=None)
|
36
|
+
"""
|
37
|
+
Optionally provide the value that the output ideally should have been.
|
38
|
+
"""
|
30
39
|
|
31
40
|
def json(self, **kwargs: typing.Any) -> str:
|
32
41
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -39,4 +48,5 @@ class WorkflowExecutionActualStringRequest(pydantic.BaseModel):
|
|
39
48
|
class Config:
|
40
49
|
frozen = True
|
41
50
|
smart_union = True
|
51
|
+
extra = pydantic.Extra.allow
|
42
52
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -18,8 +18,8 @@ class WorkflowExecutionNodeResultEvent(pydantic.BaseModel):
|
|
18
18
|
"""
|
19
19
|
|
20
20
|
execution_id: str
|
21
|
-
run_id: typing.Optional[str]
|
22
|
-
external_id: typing.Optional[str]
|
21
|
+
run_id: typing.Optional[str] = None
|
22
|
+
external_id: typing.Optional[str] = None
|
23
23
|
data: WorkflowNodeResultEvent
|
24
24
|
|
25
25
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -33,4 +33,5 @@ class WorkflowExecutionNodeResultEvent(pydantic.BaseModel):
|
|
33
33
|
class Config:
|
34
34
|
frozen = True
|
35
35
|
smart_union = True
|
36
|
+
extra = pydantic.Extra.allow
|
36
37
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -18,8 +18,8 @@ class WorkflowExecutionWorkflowResultEvent(pydantic.BaseModel):
|
|
18
18
|
"""
|
19
19
|
|
20
20
|
execution_id: str
|
21
|
-
run_id: typing.Optional[str]
|
22
|
-
external_id: typing.Optional[str]
|
21
|
+
run_id: typing.Optional[str] = None
|
22
|
+
external_id: typing.Optional[str] = None
|
23
23
|
data: WorkflowResultEvent
|
24
24
|
|
25
25
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -33,4 +33,5 @@ class WorkflowExecutionWorkflowResultEvent(pydantic.BaseModel):
|
|
33
33
|
class Config:
|
34
34
|
frozen = True
|
35
35
|
smart_union = True
|
36
|
+
extra = pydantic.Extra.allow
|
36
37
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -4,8 +4,6 @@ from __future__ import annotations
|
|
4
4
|
|
5
5
|
import typing
|
6
6
|
|
7
|
-
import typing_extensions
|
8
|
-
|
9
7
|
from .api_node_result import ApiNodeResult
|
10
8
|
from .code_execution_node_result import CodeExecutionNodeResult
|
11
9
|
from .conditional_node_result import ConditionalNodeResult
|
@@ -17,75 +15,83 @@ from .terminal_node_result import TerminalNodeResult
|
|
17
15
|
|
18
16
|
|
19
17
|
class WorkflowNodeResultData_Prompt(PromptNodeResult):
|
20
|
-
type:
|
18
|
+
type: typing.Literal["PROMPT"] = "PROMPT"
|
21
19
|
|
22
20
|
class Config:
|
23
21
|
frozen = True
|
24
22
|
smart_union = True
|
25
23
|
allow_population_by_field_name = True
|
24
|
+
populate_by_name = True
|
26
25
|
|
27
26
|
|
28
27
|
class WorkflowNodeResultData_Search(SearchNodeResult):
|
29
|
-
type:
|
28
|
+
type: typing.Literal["SEARCH"] = "SEARCH"
|
30
29
|
|
31
30
|
class Config:
|
32
31
|
frozen = True
|
33
32
|
smart_union = True
|
34
33
|
allow_population_by_field_name = True
|
34
|
+
populate_by_name = True
|
35
35
|
|
36
36
|
|
37
37
|
class WorkflowNodeResultData_Templating(TemplatingNodeResult):
|
38
|
-
type:
|
38
|
+
type: typing.Literal["TEMPLATING"] = "TEMPLATING"
|
39
39
|
|
40
40
|
class Config:
|
41
41
|
frozen = True
|
42
42
|
smart_union = True
|
43
43
|
allow_population_by_field_name = True
|
44
|
+
populate_by_name = True
|
44
45
|
|
45
46
|
|
46
47
|
class WorkflowNodeResultData_CodeExecution(CodeExecutionNodeResult):
|
47
|
-
type:
|
48
|
+
type: typing.Literal["CODE_EXECUTION"] = "CODE_EXECUTION"
|
48
49
|
|
49
50
|
class Config:
|
50
51
|
frozen = True
|
51
52
|
smart_union = True
|
52
53
|
allow_population_by_field_name = True
|
54
|
+
populate_by_name = True
|
53
55
|
|
54
56
|
|
55
57
|
class WorkflowNodeResultData_Conditional(ConditionalNodeResult):
|
56
|
-
type:
|
58
|
+
type: typing.Literal["CONDITIONAL"] = "CONDITIONAL"
|
57
59
|
|
58
60
|
class Config:
|
59
61
|
frozen = True
|
60
62
|
smart_union = True
|
61
63
|
allow_population_by_field_name = True
|
64
|
+
populate_by_name = True
|
62
65
|
|
63
66
|
|
64
67
|
class WorkflowNodeResultData_Api(ApiNodeResult):
|
65
|
-
type:
|
68
|
+
type: typing.Literal["API"] = "API"
|
66
69
|
|
67
70
|
class Config:
|
68
71
|
frozen = True
|
69
72
|
smart_union = True
|
70
73
|
allow_population_by_field_name = True
|
74
|
+
populate_by_name = True
|
71
75
|
|
72
76
|
|
73
77
|
class WorkflowNodeResultData_Terminal(TerminalNodeResult):
|
74
|
-
type:
|
78
|
+
type: typing.Literal["TERMINAL"] = "TERMINAL"
|
75
79
|
|
76
80
|
class Config:
|
77
81
|
frozen = True
|
78
82
|
smart_union = True
|
79
83
|
allow_population_by_field_name = True
|
84
|
+
populate_by_name = True
|
80
85
|
|
81
86
|
|
82
87
|
class WorkflowNodeResultData_Subworkflow(SubworkflowNodeResult):
|
83
|
-
type:
|
88
|
+
type: typing.Literal["SUBWORKFLOW"] = "SUBWORKFLOW"
|
84
89
|
|
85
90
|
class Config:
|
86
91
|
frozen = True
|
87
92
|
smart_union = True
|
88
93
|
allow_population_by_field_name = True
|
94
|
+
populate_by_name = True
|
89
95
|
|
90
96
|
|
91
97
|
WorkflowNodeResultData = typing.Union[
|
@@ -4,8 +4,6 @@ from __future__ import annotations
|
|
4
4
|
|
5
5
|
import typing
|
6
6
|
|
7
|
-
import typing_extensions
|
8
|
-
|
9
7
|
from .fulfilled_workflow_node_result_event import FulfilledWorkflowNodeResultEvent
|
10
8
|
from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEvent
|
11
9
|
from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
|
@@ -13,39 +11,43 @@ from .streaming_workflow_node_result_event import StreamingWorkflowNodeResultEve
|
|
13
11
|
|
14
12
|
|
15
13
|
class WorkflowNodeResultEvent_Initiated(InitiatedWorkflowNodeResultEvent):
|
16
|
-
state:
|
14
|
+
state: typing.Literal["INITIATED"] = "INITIATED"
|
17
15
|
|
18
16
|
class Config:
|
19
17
|
frozen = True
|
20
18
|
smart_union = True
|
21
19
|
allow_population_by_field_name = True
|
20
|
+
populate_by_name = True
|
22
21
|
|
23
22
|
|
24
23
|
class WorkflowNodeResultEvent_Streaming(StreamingWorkflowNodeResultEvent):
|
25
|
-
state:
|
24
|
+
state: typing.Literal["STREAMING"] = "STREAMING"
|
26
25
|
|
27
26
|
class Config:
|
28
27
|
frozen = True
|
29
28
|
smart_union = True
|
30
29
|
allow_population_by_field_name = True
|
30
|
+
populate_by_name = True
|
31
31
|
|
32
32
|
|
33
33
|
class WorkflowNodeResultEvent_Fulfilled(FulfilledWorkflowNodeResultEvent):
|
34
|
-
state:
|
34
|
+
state: typing.Literal["FULFILLED"] = "FULFILLED"
|
35
35
|
|
36
36
|
class Config:
|
37
37
|
frozen = True
|
38
38
|
smart_union = True
|
39
39
|
allow_population_by_field_name = True
|
40
|
+
populate_by_name = True
|
40
41
|
|
41
42
|
|
42
43
|
class WorkflowNodeResultEvent_Rejected(RejectedWorkflowNodeResultEvent):
|
43
|
-
state:
|
44
|
+
state: typing.Literal["REJECTED"] = "REJECTED"
|
44
45
|
|
45
46
|
class Config:
|
46
47
|
frozen = True
|
47
48
|
smart_union = True
|
48
49
|
allow_population_by_field_name = True
|
50
|
+
populate_by_name = True
|
49
51
|
|
50
52
|
|
51
53
|
WorkflowNodeResultEvent = typing.Union[
|