vellum-ai 0.3.13__py3-none-any.whl → 0.3.15__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.13.dist-info → vellum_ai-0.3.15.dist-info}/METADATA +4 -4
- vellum_ai-0.3.15.dist-info/RECORD +365 -0
- vellum_ai-0.3.13.dist-info/RECORD +0 -356
- {vellum_ai-0.3.13.dist-info → vellum_ai-0.3.15.dist-info}/LICENSE +0 -0
- {vellum_ai-0.3.13.dist-info → vellum_ai-0.3.15.dist-info}/WHEEL +0 -0
@@ -20,7 +20,7 @@ class FulfilledExecutePromptEvent(pydantic.BaseModel):
|
|
20
20
|
|
21
21
|
outputs: typing.List[PromptOutput]
|
22
22
|
execution_id: str
|
23
|
-
meta: typing.Optional[FulfilledPromptExecutionMeta]
|
23
|
+
meta: typing.Optional[FulfilledPromptExecutionMeta] = None
|
24
24
|
|
25
25
|
def json(self, **kwargs: typing.Any) -> str:
|
26
26
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -33,4 +33,5 @@ class FulfilledExecutePromptEvent(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,11 +18,17 @@ class FulfilledExecutePromptResponse(pydantic.BaseModel):
|
|
18
18
|
The successful response from the model containing all of the resolved values generated by the prompt.
|
19
19
|
"""
|
20
20
|
|
21
|
-
meta: typing.Optional[PromptExecutionMeta]
|
22
|
-
raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
|
23
|
-
|
24
|
-
|
25
|
-
|
21
|
+
meta: typing.Optional[PromptExecutionMeta] = None
|
22
|
+
raw: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(default=None)
|
23
|
+
"""
|
24
|
+
The subset of the raw response from the model that the request opted into with `expand_raw`.
|
25
|
+
"""
|
26
|
+
|
27
|
+
execution_id: str = pydantic.Field()
|
28
|
+
"""
|
29
|
+
The ID of the execution.
|
30
|
+
"""
|
31
|
+
|
26
32
|
outputs: typing.List[PromptOutput]
|
27
33
|
|
28
34
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -36,4 +42,5 @@ class FulfilledExecutePromptResponse(pydantic.BaseModel):
|
|
36
42
|
class Config:
|
37
43
|
frozen = True
|
38
44
|
smart_union = True
|
45
|
+
extra = pydantic.Extra.allow
|
39
46
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -17,7 +17,7 @@ class FulfilledFunctionCall(pydantic.BaseModel):
|
|
17
17
|
"""
|
18
18
|
|
19
19
|
arguments: typing.Dict[str, typing.Any]
|
20
|
-
id: typing.Optional[str]
|
20
|
+
id: typing.Optional[str] = None
|
21
21
|
name: str
|
22
22
|
|
23
23
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -31,4 +31,5 @@ class FulfilledFunctionCall(pydantic.BaseModel):
|
|
31
31
|
class Config:
|
32
32
|
frozen = True
|
33
33
|
smart_union = True
|
34
|
+
extra = pydantic.Extra.allow
|
34
35
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -17,8 +17,8 @@ class FulfilledPromptExecutionMeta(pydantic.BaseModel):
|
|
17
17
|
The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
|
18
18
|
"""
|
19
19
|
|
20
|
-
latency: typing.Optional[int]
|
21
|
-
finish_reason: typing.Optional[FinishReasonEnum]
|
20
|
+
latency: typing.Optional[int] = None
|
21
|
+
finish_reason: typing.Optional[FinishReasonEnum] = None
|
22
22
|
|
23
23
|
def json(self, **kwargs: typing.Any) -> str:
|
24
24
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -31,4 +31,5 @@ class FulfilledPromptExecutionMeta(pydantic.BaseModel):
|
|
31
31
|
class Config:
|
32
32
|
frozen = True
|
33
33
|
smart_union = True
|
34
|
+
extra = pydantic.Extra.allow
|
34
35
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -21,11 +21,11 @@ class FulfilledWorkflowNodeResultEvent(pydantic.BaseModel):
|
|
21
21
|
id: str
|
22
22
|
node_id: str
|
23
23
|
node_result_id: str
|
24
|
-
ts: typing.Optional[dt.datetime]
|
25
|
-
data: typing.Optional[WorkflowNodeResultData]
|
26
|
-
source_execution_id: typing.Optional[str]
|
27
|
-
output_values: typing.Optional[typing.List[NodeOutputCompiledValue]]
|
28
|
-
mocked: typing.Optional[bool]
|
24
|
+
ts: typing.Optional[dt.datetime] = None
|
25
|
+
data: typing.Optional[WorkflowNodeResultData] = None
|
26
|
+
source_execution_id: typing.Optional[str] = None
|
27
|
+
output_values: typing.Optional[typing.List[NodeOutputCompiledValue]] = None
|
28
|
+
mocked: typing.Optional[bool] = None
|
29
29
|
|
30
30
|
def json(self, **kwargs: typing.Any) -> str:
|
31
31
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -38,4 +38,5 @@ class FulfilledWorkflowNodeResultEvent(pydantic.BaseModel):
|
|
38
38
|
class Config:
|
39
39
|
frozen = True
|
40
40
|
smart_union = True
|
41
|
+
extra = pydantic.Extra.allow
|
41
42
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/function_call.py
CHANGED
@@ -4,28 +4,28 @@ from __future__ import annotations
|
|
4
4
|
|
5
5
|
import typing
|
6
6
|
|
7
|
-
import typing_extensions
|
8
|
-
|
9
7
|
from .fulfilled_function_call import FulfilledFunctionCall
|
10
8
|
from .rejected_function_call import RejectedFunctionCall
|
11
9
|
|
12
10
|
|
13
11
|
class FunctionCall_Fulfilled(FulfilledFunctionCall):
|
14
|
-
state:
|
12
|
+
state: typing.Literal["FULFILLED"] = "FULFILLED"
|
15
13
|
|
16
14
|
class Config:
|
17
15
|
frozen = True
|
18
16
|
smart_union = True
|
19
17
|
allow_population_by_field_name = True
|
18
|
+
populate_by_name = True
|
20
19
|
|
21
20
|
|
22
21
|
class FunctionCall_Rejected(RejectedFunctionCall):
|
23
|
-
state:
|
22
|
+
state: typing.Literal["REJECTED"] = "REJECTED"
|
24
23
|
|
25
24
|
class Config:
|
26
25
|
frozen = True
|
27
26
|
smart_union = True
|
28
27
|
allow_population_by_field_name = True
|
28
|
+
populate_by_name = True
|
29
29
|
|
30
30
|
|
31
31
|
FunctionCall = typing.Union[FunctionCall_Fulfilled, FunctionCall_Rejected]
|
@@ -18,7 +18,7 @@ class FunctionCallChatMessageContentValue(pydantic.BaseModel):
|
|
18
18
|
|
19
19
|
name: str
|
20
20
|
arguments: typing.Dict[str, typing.Any]
|
21
|
-
id: typing.Optional[str]
|
21
|
+
id: typing.Optional[str] = None
|
22
22
|
|
23
23
|
def json(self, **kwargs: typing.Any) -> str:
|
24
24
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -31,4 +31,5 @@ class FunctionCallChatMessageContentValue(pydantic.BaseModel):
|
|
31
31
|
class Config:
|
32
32
|
frozen = True
|
33
33
|
smart_union = True
|
34
|
+
extra = pydantic.Extra.allow
|
34
35
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -18,7 +18,7 @@ class FunctionCallChatMessageContentValueRequest(pydantic.BaseModel):
|
|
18
18
|
|
19
19
|
name: str
|
20
20
|
arguments: typing.Dict[str, typing.Any]
|
21
|
-
id: typing.Optional[str]
|
21
|
+
id: typing.Optional[str] = None
|
22
22
|
|
23
23
|
def json(self, **kwargs: typing.Any) -> str:
|
24
24
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -31,4 +31,5 @@ class FunctionCallChatMessageContentValueRequest(pydantic.BaseModel):
|
|
31
31
|
class Config:
|
32
32
|
frozen = True
|
33
33
|
smart_union = True
|
34
|
+
extra = pydantic.Extra.allow
|
34
35
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -12,7 +12,10 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class GenerateErrorResponse(pydantic.BaseModel):
|
15
|
-
detail: str = pydantic.Field(
|
15
|
+
detail: str = pydantic.Field()
|
16
|
+
"""
|
17
|
+
Details about why the request failed.
|
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 GenerateErrorResponse(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}
|
@@ -13,11 +13,13 @@ except ImportError:
|
|
13
13
|
|
14
14
|
|
15
15
|
class GenerateOptionsRequest(pydantic.BaseModel):
|
16
|
-
logprobs: typing.Optional[LogprobsEnum] = pydantic.Field(
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
16
|
+
logprobs: typing.Optional[LogprobsEnum] = pydantic.Field(default=None)
|
17
|
+
"""
|
18
|
+
Which logprobs to include, if any. Defaults to NONE.
|
19
|
+
|
20
|
+
- `ALL` - ALL
|
21
|
+
- `NONE` - NONE
|
22
|
+
"""
|
21
23
|
|
22
24
|
def json(self, **kwargs: typing.Any) -> str:
|
23
25
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -30,4 +32,5 @@ class GenerateOptionsRequest(pydantic.BaseModel):
|
|
30
32
|
class Config:
|
31
33
|
frozen = True
|
32
34
|
smart_union = True
|
35
|
+
extra = pydantic.Extra.allow
|
33
36
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/generate_request.py
CHANGED
@@ -13,15 +13,20 @@ except ImportError:
|
|
13
13
|
|
14
14
|
|
15
15
|
class GenerateRequest(pydantic.BaseModel):
|
16
|
-
input_values: typing.Dict[str, typing.Any] = pydantic.Field(
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
)
|
22
|
-
|
23
|
-
|
24
|
-
|
16
|
+
input_values: typing.Dict[str, typing.Any] = pydantic.Field()
|
17
|
+
"""
|
18
|
+
Key/value pairs for each template variable defined in the deployment's prompt.
|
19
|
+
"""
|
20
|
+
|
21
|
+
chat_history: typing.Optional[typing.List[ChatMessageRequest]] = pydantic.Field(default=None)
|
22
|
+
"""
|
23
|
+
Optionally provide a list of chat messages that'll be used in place of the special chat_history variable, if included in the prompt.
|
24
|
+
"""
|
25
|
+
|
26
|
+
external_ids: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
|
27
|
+
"""
|
28
|
+
Optionally include a unique identifier for each generation, as represented outside of Vellum. Note that this should generally be a list of length one.
|
29
|
+
"""
|
25
30
|
|
26
31
|
def json(self, **kwargs: typing.Any) -> str:
|
27
32
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -34,4 +39,5 @@ class GenerateRequest(pydantic.BaseModel):
|
|
34
39
|
class Config:
|
35
40
|
frozen = True
|
36
41
|
smart_union = True
|
42
|
+
extra = pydantic.Extra.allow
|
37
43
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/generate_result.py
CHANGED
@@ -14,12 +14,15 @@ except ImportError:
|
|
14
14
|
|
15
15
|
|
16
16
|
class GenerateResult(pydantic.BaseModel):
|
17
|
-
data: typing.Optional[GenerateResultData] = pydantic.Field(
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
|
22
|
-
)
|
17
|
+
data: typing.Optional[GenerateResultData] = pydantic.Field(default=None)
|
18
|
+
"""
|
19
|
+
An object containing the resulting generation. This key will be absent if the LLM provider experienced an error.
|
20
|
+
"""
|
21
|
+
|
22
|
+
error: typing.Optional[GenerateResultError] = pydantic.Field(default=None)
|
23
|
+
"""
|
24
|
+
An object containing details about the error that occurred. This key will be absent if the LLM provider did not experience an error.
|
25
|
+
"""
|
23
26
|
|
24
27
|
def json(self, **kwargs: typing.Any) -> str:
|
25
28
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -32,4 +35,5 @@ class GenerateResult(pydantic.BaseModel):
|
|
32
35
|
class Config:
|
33
36
|
frozen = True
|
34
37
|
smart_union = True
|
38
|
+
extra = pydantic.Extra.allow
|
35
39
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -13,9 +13,10 @@ except ImportError:
|
|
13
13
|
|
14
14
|
|
15
15
|
class GenerateResultData(pydantic.BaseModel):
|
16
|
-
completions: typing.List[EnrichedNormalizedCompletion] = pydantic.Field(
|
17
|
-
|
18
|
-
|
16
|
+
completions: typing.List[EnrichedNormalizedCompletion] = pydantic.Field()
|
17
|
+
"""
|
18
|
+
The generated completions. This will generally be a list of length one.
|
19
|
+
"""
|
19
20
|
|
20
21
|
def json(self, **kwargs: typing.Any) -> str:
|
21
22
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -28,4 +29,5 @@ class GenerateResultData(pydantic.BaseModel):
|
|
28
29
|
class Config:
|
29
30
|
frozen = True
|
30
31
|
smart_union = True
|
32
|
+
extra = pydantic.Extra.allow
|
31
33
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -12,7 +12,10 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class GenerateResultError(pydantic.BaseModel):
|
15
|
-
message: str = pydantic.Field(
|
15
|
+
message: str = pydantic.Field()
|
16
|
+
"""
|
17
|
+
The error message returned by the LLM provider.
|
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 GenerateResultError(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}
|
@@ -15,8 +15,8 @@ except ImportError:
|
|
15
15
|
|
16
16
|
class GenerateStreamResult(pydantic.BaseModel):
|
17
17
|
request_index: int
|
18
|
-
data: typing.Optional[GenerateStreamResultData]
|
19
|
-
error: typing.Optional[GenerateResultError]
|
18
|
+
data: typing.Optional[GenerateStreamResultData] = None
|
19
|
+
error: typing.Optional[GenerateResultError] = None
|
20
20
|
|
21
21
|
def json(self, **kwargs: typing.Any) -> str:
|
22
22
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -29,4 +29,5 @@ class GenerateStreamResult(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}
|
vellum/types/image_enum.py
CHANGED
@@ -17,7 +17,7 @@ class ImageVariableValue(pydantic.BaseModel):
|
|
17
17
|
A base Vellum primitive value representing an image.
|
18
18
|
"""
|
19
19
|
|
20
|
-
value: typing.Optional[VellumImage]
|
20
|
+
value: typing.Optional[VellumImage] = None
|
21
21
|
|
22
22
|
def json(self, **kwargs: typing.Any) -> str:
|
23
23
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -30,4 +30,5 @@ class ImageVariableValue(pydantic.BaseModel):
|
|
30
30
|
class Config:
|
31
31
|
frozen = True
|
32
32
|
smart_union = True
|
33
|
+
extra = pydantic.Extra.allow
|
33
34
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/initiated_enum.py
CHANGED
@@ -17,7 +17,7 @@ class InitiatedExecutePromptEvent(pydantic.BaseModel):
|
|
17
17
|
The initial data returned indicating that the response from the model has returned and begun streaming.
|
18
18
|
"""
|
19
19
|
|
20
|
-
meta: typing.Optional[InitiatedPromptExecutionMeta]
|
20
|
+
meta: typing.Optional[InitiatedPromptExecutionMeta] = None
|
21
21
|
execution_id: str
|
22
22
|
|
23
23
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -31,4 +31,5 @@ class InitiatedExecutePromptEvent(pydantic.BaseModel):
|
|
31
31
|
class Config:
|
32
32
|
frozen = True
|
33
33
|
smart_union = True
|
34
|
+
extra = pydantic.Extra.allow
|
34
35
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -16,10 +16,10 @@ class InitiatedPromptExecutionMeta(pydantic.BaseModel):
|
|
16
16
|
The subset of the metadata tracked by Vellum during prompt execution that the request opted into with `expand_meta`.
|
17
17
|
"""
|
18
18
|
|
19
|
-
model_name: typing.Optional[str]
|
20
|
-
latency: typing.Optional[int]
|
21
|
-
deployment_release_tag: typing.Optional[str]
|
22
|
-
prompt_version_id: typing.Optional[str]
|
19
|
+
model_name: typing.Optional[str] = None
|
20
|
+
latency: typing.Optional[int] = None
|
21
|
+
deployment_release_tag: typing.Optional[str] = None
|
22
|
+
prompt_version_id: typing.Optional[str] = None
|
23
23
|
|
24
24
|
def json(self, **kwargs: typing.Any) -> str:
|
25
25
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -32,4 +32,5 @@ class InitiatedPromptExecutionMeta(pydantic.BaseModel):
|
|
32
32
|
class Config:
|
33
33
|
frozen = True
|
34
34
|
smart_union = True
|
35
|
+
extra = pydantic.Extra.allow
|
35
36
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -21,10 +21,10 @@ class InitiatedWorkflowNodeResultEvent(pydantic.BaseModel):
|
|
21
21
|
id: str
|
22
22
|
node_id: str
|
23
23
|
node_result_id: str
|
24
|
-
ts: typing.Optional[dt.datetime]
|
25
|
-
data: typing.Optional[WorkflowNodeResultData]
|
26
|
-
source_execution_id: typing.Optional[str]
|
27
|
-
input_values: typing.Optional[typing.List[NodeInputVariableCompiledValue]]
|
24
|
+
ts: typing.Optional[dt.datetime] = None
|
25
|
+
data: typing.Optional[WorkflowNodeResultData] = None
|
26
|
+
source_execution_id: typing.Optional[str] = None
|
27
|
+
input_values: typing.Optional[typing.List[NodeInputVariableCompiledValue]] = None
|
28
28
|
|
29
29
|
def json(self, **kwargs: typing.Any) -> str:
|
30
30
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -37,4 +37,5 @@ class InitiatedWorkflowNodeResultEvent(pydantic.BaseModel):
|
|
37
37
|
class Config:
|
38
38
|
frozen = True
|
39
39
|
smart_union = True
|
40
|
+
extra = pydantic.Extra.allow
|
40
41
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/json_enum.py
CHANGED
@@ -16,7 +16,11 @@ class JsonInputRequest(pydantic.BaseModel):
|
|
16
16
|
A user input representing a JSON object
|
17
17
|
"""
|
18
18
|
|
19
|
-
name: str = pydantic.Field(
|
19
|
+
name: str = pydantic.Field()
|
20
|
+
"""
|
21
|
+
The variable's name, as defined in the deployment.
|
22
|
+
"""
|
23
|
+
|
20
24
|
value: typing.Dict[str, typing.Any]
|
21
25
|
|
22
26
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -30,4 +34,5 @@ class JsonInputRequest(pydantic.BaseModel):
|
|
30
34
|
class Config:
|
31
35
|
frozen = True
|
32
36
|
smart_union = True
|
37
|
+
extra = pydantic.Extra.allow
|
33
38
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -12,7 +12,7 @@ except ImportError:
|
|
12
12
|
|
13
13
|
|
14
14
|
class JsonVariableValue(pydantic.BaseModel):
|
15
|
-
value: typing.Optional[typing.Dict[str, typing.Any]]
|
15
|
+
value: typing.Optional[typing.Dict[str, typing.Any]] = None
|
16
16
|
|
17
17
|
def json(self, **kwargs: typing.Any) -> str:
|
18
18
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -25,4 +25,5 @@ class JsonVariableValue(pydantic.BaseModel):
|
|
25
25
|
class Config:
|
26
26
|
frozen = True
|
27
27
|
smart_union = True
|
28
|
+
extra = pydantic.Extra.allow
|
28
29
|
json_encoders = {dt.datetime: serialize_datetime}
|
vellum/types/logprobs_enum.py
CHANGED
@@ -15,8 +15,8 @@ class LogprobsEnum(str, enum.Enum):
|
|
15
15
|
ALL = "ALL"
|
16
16
|
NONE = "NONE"
|
17
17
|
|
18
|
-
def visit(self,
|
18
|
+
def visit(self, all_: typing.Callable[[], T_Result], none: typing.Callable[[], T_Result]) -> T_Result:
|
19
19
|
if self is LogprobsEnum.ALL:
|
20
|
-
return
|
20
|
+
return all_()
|
21
21
|
if self is LogprobsEnum.NONE:
|
22
22
|
return none()
|
@@ -15,12 +15,12 @@ except ImportError:
|
|
15
15
|
|
16
16
|
|
17
17
|
class MetadataFilterConfigRequest(pydantic.BaseModel):
|
18
|
-
combinator: typing.Optional[MetadataFilterRuleCombinator]
|
19
|
-
negated: typing.Optional[bool]
|
20
|
-
rules: typing.Optional[typing.List[MetadataFilterRuleRequest]]
|
21
|
-
field: typing.Optional[str]
|
22
|
-
operator: typing.Optional[LogicalOperator]
|
23
|
-
value: typing.Optional[str]
|
18
|
+
combinator: typing.Optional[MetadataFilterRuleCombinator] = None
|
19
|
+
negated: typing.Optional[bool] = None
|
20
|
+
rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
|
21
|
+
field: typing.Optional[str] = None
|
22
|
+
operator: typing.Optional[LogicalOperator] = None
|
23
|
+
value: typing.Optional[str] = None
|
24
24
|
|
25
25
|
def json(self, **kwargs: typing.Any) -> str:
|
26
26
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -33,4 +33,5 @@ class MetadataFilterConfigRequest(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}
|
@@ -16,12 +16,12 @@ except ImportError:
|
|
16
16
|
|
17
17
|
|
18
18
|
class MetadataFilterRuleRequest(pydantic.BaseModel):
|
19
|
-
combinator: typing.Optional[MetadataFilterRuleCombinator]
|
20
|
-
negated: typing.Optional[bool]
|
21
|
-
rules: typing.Optional[typing.List[MetadataFilterRuleRequest]]
|
22
|
-
field: typing.Optional[str]
|
23
|
-
operator: typing.Optional[LogicalOperator]
|
24
|
-
value: typing.Optional[str]
|
19
|
+
combinator: typing.Optional[MetadataFilterRuleCombinator] = None
|
20
|
+
negated: typing.Optional[bool] = None
|
21
|
+
rules: typing.Optional[typing.List[MetadataFilterRuleRequest]] = None
|
22
|
+
field: typing.Optional[str] = None
|
23
|
+
operator: typing.Optional[LogicalOperator] = None
|
24
|
+
value: typing.Optional[str] = None
|
25
25
|
|
26
26
|
def json(self, **kwargs: typing.Any) -> str:
|
27
27
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -34,6 +34,7 @@ class MetadataFilterRuleRequest(pydantic.BaseModel):
|
|
34
34
|
class Config:
|
35
35
|
frozen = True
|
36
36
|
smart_union = True
|
37
|
+
extra = pydantic.Extra.allow
|
37
38
|
json_encoders = {dt.datetime: serialize_datetime}
|
38
39
|
|
39
40
|
|
@@ -13,13 +13,17 @@ except ImportError:
|
|
13
13
|
|
14
14
|
|
15
15
|
class ModelVersionBuildConfig(pydantic.BaseModel):
|
16
|
-
base_model: str = pydantic.Field(
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
)
|
22
|
-
|
16
|
+
base_model: str = pydantic.Field()
|
17
|
+
"""
|
18
|
+
The name of the base model used to create this model version, as identified by the LLM provider.
|
19
|
+
"""
|
20
|
+
|
21
|
+
sandbox_snapshot: typing.Optional[ModelVersionSandboxSnapshot] = pydantic.Field(default=None)
|
22
|
+
"""
|
23
|
+
Information about the sandbox snapshot that was used to create this model version, if applicable.
|
24
|
+
"""
|
25
|
+
|
26
|
+
prompt_version_id: typing.Optional[str] = None
|
23
27
|
|
24
28
|
def json(self, **kwargs: typing.Any) -> str:
|
25
29
|
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
@@ -32,4 +36,5 @@ class ModelVersionBuildConfig(pydantic.BaseModel):
|
|
32
36
|
class Config:
|
33
37
|
frozen = True
|
34
38
|
smart_union = True
|
39
|
+
extra = pydantic.Extra.allow
|
35
40
|
json_encoders = {dt.datetime: serialize_datetime}
|