vellum-ai 0.9.16__py3-none-any.whl → 0.9.16rc4__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- vellum/client/README.md +242 -0
- vellum/client/core/__init__.py +47 -0
- vellum/client/core/api_error.py +15 -0
- vellum/client/core/client_wrapper.py +59 -0
- vellum/client/core/datetime_utils.py +28 -0
- vellum/client/core/file.py +62 -0
- vellum/client/core/http_client.py +487 -0
- vellum/client/core/jsonable_encoder.py +101 -0
- vellum/client/core/pydantic_utilities.py +296 -0
- vellum/client/core/query_encoder.py +58 -0
- vellum/client/core/remove_none_from_dict.py +11 -0
- vellum/client/core/request_options.py +32 -0
- vellum/client/core/serialization.py +272 -0
- vellum/client/environment.py +17 -0
- vellum/client/errors/__init__.py +8 -0
- vellum/client/errors/bad_request_error.py +9 -0
- vellum/client/errors/forbidden_error.py +9 -0
- vellum/client/errors/internal_server_error.py +9 -0
- vellum/client/errors/not_found_error.py +9 -0
- vellum/client/resources/__init__.py +47 -0
- vellum/client/resources/ad_hoc/__init__.py +2 -0
- vellum/client/resources/ad_hoc/client.py +459 -0
- vellum/client/resources/container_images/__init__.py +2 -0
- vellum/client/resources/container_images/client.py +502 -0
- vellum/client/resources/deployments/__init__.py +5 -0
- vellum/client/resources/deployments/client.py +975 -0
- vellum/client/resources/deployments/types/__init__.py +6 -0
- vellum/client/resources/deployments/types/deployments_list_request_status.py +5 -0
- vellum/client/resources/deployments/types/list_deployment_release_tags_request_source.py +5 -0
- vellum/client/resources/document_indexes/__init__.py +5 -0
- vellum/client/resources/document_indexes/client.py +1154 -0
- vellum/client/resources/document_indexes/types/__init__.py +5 -0
- vellum/client/resources/document_indexes/types/document_indexes_list_request_status.py +5 -0
- vellum/client/resources/documents/__init__.py +2 -0
- vellum/client/resources/documents/client.py +781 -0
- vellum/client/resources/folder_entities/__init__.py +5 -0
- vellum/client/resources/folder_entities/client.py +333 -0
- vellum/client/resources/folder_entities/types/__init__.py +5 -0
- vellum/client/resources/folder_entities/types/folder_entities_list_request_entity_status.py +5 -0
- vellum/client/resources/metric_definitions/__init__.py +2 -0
- vellum/client/resources/metric_definitions/client.py +178 -0
- vellum/client/resources/sandboxes/__init__.py +2 -0
- vellum/client/resources/sandboxes/client.py +481 -0
- vellum/client/resources/test_suite_runs/__init__.py +2 -0
- vellum/client/resources/test_suite_runs/client.py +445 -0
- vellum/client/resources/test_suites/__init__.py +2 -0
- vellum/client/resources/test_suites/client.py +688 -0
- vellum/client/resources/workflow_deployments/__init__.py +5 -0
- vellum/client/resources/workflow_deployments/client.py +693 -0
- vellum/client/resources/workflow_deployments/types/__init__.py +6 -0
- vellum/client/resources/workflow_deployments/types/list_workflow_release_tags_request_source.py +5 -0
- vellum/client/resources/workflow_deployments/types/workflow_deployments_list_request_status.py +5 -0
- vellum/client/resources/workflow_sandboxes/__init__.py +2 -0
- vellum/client/resources/workflow_sandboxes/client.py +190 -0
- vellum/client/resources/workflows/__init__.py +5 -0
- vellum/client/resources/workflows/client.py +333 -0
- vellum/client/resources/workflows/types/__init__.py +5 -0
- vellum/client/resources/workflows/types/workflows_pull_request_format.py +5 -0
- vellum/client/resources/workspace_secrets/__init__.py +2 -0
- vellum/client/resources/workspace_secrets/client.py +266 -0
- vellum/client/types/__init__.py +979 -0
- vellum/client/types/ad_hoc_execute_prompt_event.py +14 -0
- vellum/client/types/ad_hoc_expand_meta.py +37 -0
- vellum/client/types/ad_hoc_fulfilled_prompt_execution_meta.py +27 -0
- vellum/client/types/ad_hoc_initiated_prompt_execution_meta.py +24 -0
- vellum/client/types/ad_hoc_rejected_prompt_execution_meta.py +25 -0
- vellum/client/types/ad_hoc_streaming_prompt_execution_meta.py +23 -0
- vellum/client/types/add_openai_api_key_enum.py +3 -0
- vellum/client/types/api_node_result.py +25 -0
- vellum/client/types/api_node_result_data.py +28 -0
- vellum/client/types/array_chat_message_content.py +25 -0
- vellum/client/types/array_chat_message_content_item.py +11 -0
- vellum/client/types/array_chat_message_content_item_request.py +14 -0
- vellum/client/types/array_chat_message_content_request.py +25 -0
- vellum/client/types/array_input.py +36 -0
- vellum/client/types/array_variable_value.py +27 -0
- vellum/client/types/array_variable_value_item.py +29 -0
- vellum/client/types/array_vellum_value.py +31 -0
- vellum/client/types/array_vellum_value_request.py +31 -0
- vellum/client/types/audio_chat_message_content.py +25 -0
- vellum/client/types/audio_chat_message_content_request.py +25 -0
- vellum/client/types/audio_variable_value.py +25 -0
- vellum/client/types/audio_vellum_value.py +25 -0
- vellum/client/types/audio_vellum_value_request.py +25 -0
- vellum/client/types/basic_vectorizer_intfloat_multilingual_e_5_large.py +24 -0
- vellum/client/types/basic_vectorizer_intfloat_multilingual_e_5_large_request.py +24 -0
- vellum/client/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1.py +26 -0
- vellum/client/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request.py +26 -0
- vellum/client/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1.py +26 -0
- vellum/client/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request.py +26 -0
- vellum/client/types/chat_history_input.py +30 -0
- vellum/client/types/chat_history_input_request.py +30 -0
- vellum/client/types/chat_history_variable_value.py +21 -0
- vellum/client/types/chat_history_vellum_value.py +25 -0
- vellum/client/types/chat_history_vellum_value_request.py +25 -0
- vellum/client/types/chat_message.py +27 -0
- vellum/client/types/chat_message_content.py +16 -0
- vellum/client/types/chat_message_content_request.py +16 -0
- vellum/client/types/chat_message_prompt_block.py +39 -0
- vellum/client/types/chat_message_request.py +27 -0
- vellum/client/types/chat_message_role.py +5 -0
- vellum/client/types/code_execution_node_array_result.py +28 -0
- vellum/client/types/code_execution_node_chat_history_result.py +22 -0
- vellum/client/types/code_execution_node_error_result.py +22 -0
- vellum/client/types/code_execution_node_function_call_result.py +22 -0
- vellum/client/types/code_execution_node_json_result.py +21 -0
- vellum/client/types/code_execution_node_number_result.py +21 -0
- vellum/client/types/code_execution_node_result.py +31 -0
- vellum/client/types/code_execution_node_result_data.py +27 -0
- vellum/client/types/code_execution_node_result_output.py +22 -0
- vellum/client/types/code_execution_node_search_results_result.py +22 -0
- vellum/client/types/code_execution_node_string_result.py +21 -0
- vellum/client/types/code_execution_package.py +20 -0
- vellum/client/types/code_execution_runtime.py +5 -0
- vellum/client/types/code_executor_input.py +24 -0
- vellum/client/types/code_executor_response.py +27 -0
- vellum/client/types/code_executor_secret_input.py +29 -0
- vellum/client/types/compile_prompt_deployment_expand_meta_request.py +32 -0
- vellum/client/types/compile_prompt_meta.py +25 -0
- vellum/client/types/components_schemas_pdf_search_result_meta_source.py +5 -0
- vellum/client/types/components_schemas_pdf_search_result_meta_source_request.py +5 -0
- vellum/client/types/condition_combinator.py +5 -0
- vellum/client/types/conditional_node_result.py +25 -0
- vellum/client/types/conditional_node_result_data.py +19 -0
- vellum/client/types/container_image_read.py +28 -0
- vellum/client/types/create_test_suite_test_case_request.py +48 -0
- vellum/client/types/deployment_provider_payload_response.py +22 -0
- vellum/client/types/deployment_provider_payload_response_payload.py +5 -0
- vellum/client/types/deployment_read.py +73 -0
- vellum/client/types/deployment_release_tag_deployment_history_item.py +21 -0
- vellum/client/types/deployment_release_tag_read.py +37 -0
- vellum/client/types/docker_service_token.py +21 -0
- vellum/client/types/document_document_to_document_index.py +39 -0
- vellum/client/types/document_index_chunking.py +8 -0
- vellum/client/types/document_index_chunking_request.py +10 -0
- vellum/client/types/document_index_indexing_config.py +22 -0
- vellum/client/types/document_index_indexing_config_request.py +22 -0
- vellum/client/types/document_index_read.py +52 -0
- vellum/client/types/document_processing_state.py +7 -0
- vellum/client/types/document_read.py +49 -0
- vellum/client/types/document_status.py +5 -0
- vellum/client/types/enriched_normalized_completion.py +59 -0
- vellum/client/types/entity_status.py +5 -0
- vellum/client/types/entity_visibility.py +5 -0
- vellum/client/types/environment_enum.py +5 -0
- vellum/client/types/ephemeral_prompt_cache_config.py +20 -0
- vellum/client/types/ephemeral_prompt_cache_config_type_enum.py +5 -0
- vellum/client/types/error_input.py +30 -0
- vellum/client/types/error_variable_value.py +21 -0
- vellum/client/types/error_vellum_value.py +25 -0
- vellum/client/types/error_vellum_value_request.py +25 -0
- vellum/client/types/execute_prompt_event.py +11 -0
- vellum/client/types/execute_prompt_response.py +7 -0
- vellum/client/types/execute_workflow_response.py +29 -0
- vellum/client/types/execute_workflow_workflow_result_event.py +9 -0
- vellum/client/types/execution_array_vellum_value.py +37 -0
- vellum/client/types/execution_chat_history_vellum_value.py +31 -0
- vellum/client/types/execution_error_vellum_value.py +31 -0
- vellum/client/types/execution_function_call_vellum_value.py +31 -0
- vellum/client/types/execution_json_vellum_value.py +30 -0
- vellum/client/types/execution_number_vellum_value.py +30 -0
- vellum/client/types/execution_search_results_vellum_value.py +31 -0
- vellum/client/types/execution_string_vellum_value.py +30 -0
- vellum/client/types/execution_vellum_value.py +22 -0
- vellum/client/types/external_test_case_execution.py +31 -0
- vellum/client/types/external_test_case_execution_request.py +31 -0
- vellum/client/types/finish_reason_enum.py +5 -0
- vellum/client/types/folder_entity.py +16 -0
- vellum/client/types/folder_entity_document_index.py +26 -0
- vellum/client/types/folder_entity_document_index_data.py +25 -0
- vellum/client/types/folder_entity_folder.py +26 -0
- vellum/client/types/folder_entity_folder_data.py +24 -0
- vellum/client/types/folder_entity_prompt_sandbox.py +26 -0
- vellum/client/types/folder_entity_prompt_sandbox_data.py +26 -0
- vellum/client/types/folder_entity_test_suite.py +26 -0
- vellum/client/types/folder_entity_test_suite_data.py +25 -0
- vellum/client/types/folder_entity_workflow_sandbox.py +26 -0
- vellum/client/types/folder_entity_workflow_sandbox_data.py +26 -0
- vellum/client/types/fulfilled_ad_hoc_execute_prompt_event.py +28 -0
- vellum/client/types/fulfilled_enum.py +5 -0
- vellum/client/types/fulfilled_execute_prompt_event.py +28 -0
- vellum/client/types/fulfilled_execute_prompt_response.py +37 -0
- vellum/client/types/fulfilled_execute_workflow_workflow_result_event.py +36 -0
- vellum/client/types/fulfilled_prompt_execution_meta.py +27 -0
- vellum/client/types/fulfilled_workflow_node_result_event.py +40 -0
- vellum/client/types/function_call.py +27 -0
- vellum/client/types/function_call_chat_message_content.py +25 -0
- vellum/client/types/function_call_chat_message_content_request.py +25 -0
- vellum/client/types/function_call_chat_message_content_value.py +25 -0
- vellum/client/types/function_call_chat_message_content_value_request.py +25 -0
- vellum/client/types/function_call_input.py +30 -0
- vellum/client/types/function_call_request.py +27 -0
- vellum/client/types/function_call_variable_value.py +21 -0
- vellum/client/types/function_call_vellum_value.py +25 -0
- vellum/client/types/function_call_vellum_value_request.py +25 -0
- vellum/client/types/function_definition.py +31 -0
- vellum/client/types/generate_options_request.py +26 -0
- vellum/client/types/generate_request.py +33 -0
- vellum/client/types/generate_response.py +38 -0
- vellum/client/types/generate_result.py +29 -0
- vellum/client/types/generate_result_data.py +23 -0
- vellum/client/types/generate_result_error.py +22 -0
- vellum/client/types/generate_stream_response.py +20 -0
- vellum/client/types/generate_stream_result.py +23 -0
- vellum/client/types/generate_stream_result_data.py +21 -0
- vellum/client/types/google_vertex_ai_vectorizer_config.py +20 -0
- vellum/client/types/google_vertex_ai_vectorizer_config_request.py +20 -0
- vellum/client/types/google_vertex_ai_vectorizer_text_embedding_004.py +21 -0
- vellum/client/types/google_vertex_ai_vectorizer_text_embedding_004_request.py +21 -0
- vellum/client/types/google_vertex_ai_vectorizer_text_multilingual_embedding_002.py +21 -0
- vellum/client/types/google_vertex_ai_vectorizer_text_multilingual_embedding_002_request.py +21 -0
- vellum/client/types/hkunlp_instructor_xl_vectorizer.py +25 -0
- vellum/client/types/hkunlp_instructor_xl_vectorizer_request.py +25 -0
- vellum/client/types/image_chat_message_content.py +25 -0
- vellum/client/types/image_chat_message_content_request.py +25 -0
- vellum/client/types/image_variable_value.py +25 -0
- vellum/client/types/image_vellum_value.py +25 -0
- vellum/client/types/image_vellum_value_request.py +25 -0
- vellum/client/types/indexing_config_vectorizer.py +30 -0
- vellum/client/types/indexing_config_vectorizer_request.py +30 -0
- vellum/client/types/indexing_state_enum.py +7 -0
- vellum/client/types/initiated_ad_hoc_execute_prompt_event.py +26 -0
- vellum/client/types/initiated_execute_prompt_event.py +26 -0
- vellum/client/types/initiated_prompt_execution_meta.py +26 -0
- vellum/client/types/initiated_workflow_node_result_event.py +39 -0
- vellum/client/types/instructor_vectorizer_config.py +25 -0
- vellum/client/types/instructor_vectorizer_config_request.py +25 -0
- vellum/client/types/iteration_state_enum.py +5 -0
- vellum/client/types/jinja_prompt_block.py +28 -0
- vellum/client/types/json_input.py +29 -0
- vellum/client/types/json_input_request.py +29 -0
- vellum/client/types/json_variable_value.py +20 -0
- vellum/client/types/json_vellum_value.py +24 -0
- vellum/client/types/json_vellum_value_request.py +24 -0
- vellum/client/types/logical_operator.py +29 -0
- vellum/client/types/logprobs_enum.py +5 -0
- vellum/client/types/map_node_result.py +25 -0
- vellum/client/types/map_node_result_data.py +21 -0
- vellum/client/types/merge_node_result.py +25 -0
- vellum/client/types/merge_node_result_data.py +19 -0
- vellum/client/types/metadata_filter_config_request.py +36 -0
- vellum/client/types/metadata_filter_rule_combinator.py +5 -0
- vellum/client/types/metadata_filter_rule_request.py +31 -0
- vellum/client/types/metadata_filters_request.py +7 -0
- vellum/client/types/metric_definition_execution.py +20 -0
- vellum/client/types/metric_definition_input.py +9 -0
- vellum/client/types/metric_node_result.py +23 -0
- vellum/client/types/ml_model_usage.py +25 -0
- vellum/client/types/named_scenario_input_chat_history_variable_value_request.py +26 -0
- vellum/client/types/named_scenario_input_json_variable_value_request.py +25 -0
- vellum/client/types/named_scenario_input_request.py +12 -0
- vellum/client/types/named_scenario_input_string_variable_value_request.py +25 -0
- vellum/client/types/named_test_case_array_variable_value.py +32 -0
- vellum/client/types/named_test_case_array_variable_value_request.py +34 -0
- vellum/client/types/named_test_case_chat_history_variable_value.py +26 -0
- vellum/client/types/named_test_case_chat_history_variable_value_request.py +26 -0
- vellum/client/types/named_test_case_error_variable_value.py +26 -0
- vellum/client/types/named_test_case_error_variable_value_request.py +26 -0
- vellum/client/types/named_test_case_function_call_variable_value.py +26 -0
- vellum/client/types/named_test_case_function_call_variable_value_request.py +26 -0
- vellum/client/types/named_test_case_json_variable_value.py +25 -0
- vellum/client/types/named_test_case_json_variable_value_request.py +25 -0
- vellum/client/types/named_test_case_number_variable_value.py +25 -0
- vellum/client/types/named_test_case_number_variable_value_request.py +25 -0
- vellum/client/types/named_test_case_search_results_variable_value.py +26 -0
- vellum/client/types/named_test_case_search_results_variable_value_request.py +26 -0
- vellum/client/types/named_test_case_string_variable_value.py +25 -0
- vellum/client/types/named_test_case_string_variable_value_request.py +25 -0
- vellum/client/types/named_test_case_variable_value.py +22 -0
- vellum/client/types/named_test_case_variable_value_request.py +22 -0
- vellum/client/types/node_input_compiled_array_value.py +29 -0
- vellum/client/types/node_input_compiled_chat_history_value.py +23 -0
- vellum/client/types/node_input_compiled_error_value.py +23 -0
- vellum/client/types/node_input_compiled_function_call_value.py +23 -0
- vellum/client/types/node_input_compiled_json_value.py +22 -0
- vellum/client/types/node_input_compiled_number_value.py +22 -0
- vellum/client/types/node_input_compiled_search_results_value.py +23 -0
- vellum/client/types/node_input_compiled_string_value.py +22 -0
- vellum/client/types/node_input_variable_compiled_value.py +22 -0
- vellum/client/types/node_output_compiled_array_value.py +34 -0
- vellum/client/types/node_output_compiled_chat_history_value.py +28 -0
- vellum/client/types/node_output_compiled_error_value.py +28 -0
- vellum/client/types/node_output_compiled_function_call_value.py +28 -0
- vellum/client/types/node_output_compiled_json_value.py +27 -0
- vellum/client/types/node_output_compiled_number_value.py +27 -0
- vellum/client/types/node_output_compiled_search_results_value.py +28 -0
- vellum/client/types/node_output_compiled_string_value.py +27 -0
- vellum/client/types/node_output_compiled_value.py +22 -0
- vellum/client/types/normalized_log_probs.py +21 -0
- vellum/client/types/normalized_token_log_probs.py +22 -0
- vellum/client/types/number_input.py +29 -0
- vellum/client/types/number_variable_value.py +20 -0
- vellum/client/types/number_vellum_value.py +24 -0
- vellum/client/types/number_vellum_value_request.py +24 -0
- vellum/client/types/open_ai_vectorizer_config.py +24 -0
- vellum/client/types/open_ai_vectorizer_config_request.py +24 -0
- vellum/client/types/open_ai_vectorizer_text_embedding_3_large.py +25 -0
- vellum/client/types/open_ai_vectorizer_text_embedding_3_large_request.py +25 -0
- vellum/client/types/open_ai_vectorizer_text_embedding_3_small.py +25 -0
- vellum/client/types/open_ai_vectorizer_text_embedding_3_small_request.py +25 -0
- vellum/client/types/open_ai_vectorizer_text_embedding_ada_002.py +25 -0
- vellum/client/types/open_ai_vectorizer_text_embedding_ada_002_request.py +25 -0
- vellum/client/types/paginated_container_image_read_list.py +23 -0
- vellum/client/types/paginated_deployment_release_tag_read_list.py +23 -0
- vellum/client/types/paginated_document_index_read_list.py +23 -0
- vellum/client/types/paginated_folder_entity_list.py +23 -0
- vellum/client/types/paginated_slim_deployment_read_list.py +29 -0
- vellum/client/types/paginated_slim_document_list.py +23 -0
- vellum/client/types/paginated_slim_workflow_deployment_list.py +29 -0
- vellum/client/types/paginated_test_suite_run_execution_list.py +29 -0
- vellum/client/types/paginated_test_suite_test_case_list.py +29 -0
- vellum/client/types/paginated_workflow_release_tag_read_list.py +23 -0
- vellum/client/types/pdf_search_result_meta_source.py +32 -0
- vellum/client/types/pdf_search_result_meta_source_request.py +32 -0
- vellum/client/types/plain_text_prompt_block.py +28 -0
- vellum/client/types/price.py +21 -0
- vellum/client/types/processing_failure_reason_enum.py +5 -0
- vellum/client/types/prompt_block.py +12 -0
- vellum/client/types/prompt_block_state.py +5 -0
- vellum/client/types/prompt_deployment_expand_meta_request.py +52 -0
- vellum/client/types/prompt_deployment_input_request.py +8 -0
- vellum/client/types/prompt_execution_meta.py +32 -0
- vellum/client/types/prompt_node_execution_meta.py +24 -0
- vellum/client/types/prompt_node_result.py +25 -0
- vellum/client/types/prompt_node_result_data.py +25 -0
- vellum/client/types/prompt_output.py +9 -0
- vellum/client/types/prompt_parameters.py +27 -0
- vellum/client/types/prompt_request_chat_history_input.py +26 -0
- vellum/client/types/prompt_request_input.py +8 -0
- vellum/client/types/prompt_request_json_input.py +25 -0
- vellum/client/types/prompt_request_string_input.py +25 -0
- vellum/client/types/prompt_settings.py +19 -0
- vellum/client/types/raw_prompt_execution_overrides_request.py +28 -0
- vellum/client/types/reducto_chunker_config.py +23 -0
- vellum/client/types/reducto_chunker_config_request.py +23 -0
- vellum/client/types/reducto_chunking.py +25 -0
- vellum/client/types/reducto_chunking_request.py +25 -0
- vellum/client/types/rejected_ad_hoc_execute_prompt_event.py +28 -0
- vellum/client/types/rejected_execute_prompt_event.py +28 -0
- vellum/client/types/rejected_execute_prompt_response.py +37 -0
- vellum/client/types/rejected_execute_workflow_workflow_result_event.py +28 -0
- vellum/client/types/rejected_prompt_execution_meta.py +25 -0
- vellum/client/types/rejected_workflow_node_result_event.py +39 -0
- vellum/client/types/release_tag_source.py +5 -0
- vellum/client/types/replace_test_suite_test_case_request.py +53 -0
- vellum/client/types/rich_text_child_block.py +7 -0
- vellum/client/types/rich_text_prompt_block.py +29 -0
- vellum/client/types/sandbox_scenario.py +33 -0
- vellum/client/types/scenario_input.py +10 -0
- vellum/client/types/scenario_input_chat_history_variable_value.py +26 -0
- vellum/client/types/scenario_input_json_variable_value.py +25 -0
- vellum/client/types/scenario_input_string_variable_value.py +25 -0
- vellum/client/types/search_filters_request.py +38 -0
- vellum/client/types/search_node_result.py +25 -0
- vellum/client/types/search_node_result_data.py +27 -0
- vellum/client/types/search_request_options_request.py +50 -0
- vellum/client/types/search_response.py +23 -0
- vellum/client/types/search_result.py +40 -0
- vellum/client/types/search_result_document.py +37 -0
- vellum/client/types/search_result_document_request.py +37 -0
- vellum/client/types/search_result_merging_request.py +22 -0
- vellum/client/types/search_result_meta.py +20 -0
- vellum/client/types/search_result_meta_request.py +20 -0
- vellum/client/types/search_result_request.py +40 -0
- vellum/client/types/search_results_input.py +30 -0
- vellum/client/types/search_results_variable_value.py +21 -0
- vellum/client/types/search_results_vellum_value.py +25 -0
- vellum/client/types/search_results_vellum_value_request.py +25 -0
- vellum/client/types/search_weights_request.py +27 -0
- vellum/client/types/secret_type_enum.py +5 -0
- vellum/client/types/sentence_chunker_config.py +24 -0
- vellum/client/types/sentence_chunker_config_request.py +24 -0
- vellum/client/types/sentence_chunking.py +25 -0
- vellum/client/types/sentence_chunking_request.py +25 -0
- vellum/client/types/slim_deployment_read.py +63 -0
- vellum/client/types/slim_document.py +70 -0
- vellum/client/types/slim_workflow_deployment.py +72 -0
- vellum/client/types/streaming_ad_hoc_execute_prompt_event.py +33 -0
- vellum/client/types/streaming_execute_prompt_event.py +33 -0
- vellum/client/types/streaming_prompt_execution_meta.py +23 -0
- vellum/client/types/streaming_workflow_node_result_event.py +40 -0
- vellum/client/types/string_chat_message_content.py +24 -0
- vellum/client/types/string_chat_message_content_request.py +24 -0
- vellum/client/types/string_input.py +29 -0
- vellum/client/types/string_input_request.py +29 -0
- vellum/client/types/string_variable_value.py +20 -0
- vellum/client/types/string_vellum_value.py +24 -0
- vellum/client/types/string_vellum_value_request.py +24 -0
- vellum/client/types/submit_completion_actual_request.py +48 -0
- vellum/client/types/submit_workflow_execution_actual_request.py +10 -0
- vellum/client/types/subworkflow_node_result.py +25 -0
- vellum/client/types/subworkflow_node_result_data.py +19 -0
- vellum/client/types/templating_node_array_result.py +28 -0
- vellum/client/types/templating_node_chat_history_result.py +22 -0
- vellum/client/types/templating_node_error_result.py +22 -0
- vellum/client/types/templating_node_function_call_result.py +22 -0
- vellum/client/types/templating_node_json_result.py +21 -0
- vellum/client/types/templating_node_number_result.py +21 -0
- vellum/client/types/templating_node_result.py +31 -0
- vellum/client/types/templating_node_result_data.py +26 -0
- vellum/client/types/templating_node_result_output.py +22 -0
- vellum/client/types/templating_node_search_results_result.py +22 -0
- vellum/client/types/templating_node_string_result.py +21 -0
- vellum/client/types/terminal_node_array_result.py +33 -0
- vellum/client/types/terminal_node_chat_history_result.py +27 -0
- vellum/client/types/terminal_node_error_result.py +27 -0
- vellum/client/types/terminal_node_function_call_result.py +27 -0
- vellum/client/types/terminal_node_json_result.py +26 -0
- vellum/client/types/terminal_node_number_result.py +26 -0
- vellum/client/types/terminal_node_result.py +31 -0
- vellum/client/types/terminal_node_result_data.py +26 -0
- vellum/client/types/terminal_node_result_output.py +22 -0
- vellum/client/types/terminal_node_search_results_result.py +27 -0
- vellum/client/types/terminal_node_string_result.py +26 -0
- vellum/client/types/test_case_array_variable_value.py +33 -0
- vellum/client/types/test_case_chat_history_variable_value.py +27 -0
- vellum/client/types/test_case_error_variable_value.py +27 -0
- vellum/client/types/test_case_function_call_variable_value.py +27 -0
- vellum/client/types/test_case_json_variable_value.py +26 -0
- vellum/client/types/test_case_number_variable_value.py +26 -0
- vellum/client/types/test_case_search_results_variable_value.py +27 -0
- vellum/client/types/test_case_string_variable_value.py +26 -0
- vellum/client/types/test_case_variable_value.py +22 -0
- vellum/client/types/test_suite_run_deployment_release_tag_exec_config.py +29 -0
- vellum/client/types/test_suite_run_deployment_release_tag_exec_config_data.py +27 -0
- vellum/client/types/test_suite_run_deployment_release_tag_exec_config_data_request.py +27 -0
- vellum/client/types/test_suite_run_deployment_release_tag_exec_config_request.py +31 -0
- vellum/client/types/test_suite_run_exec_config.py +10 -0
- vellum/client/types/test_suite_run_exec_config_request.py +12 -0
- vellum/client/types/test_suite_run_execution.py +30 -0
- vellum/client/types/test_suite_run_execution_array_output.py +33 -0
- vellum/client/types/test_suite_run_execution_chat_history_output.py +27 -0
- vellum/client/types/test_suite_run_execution_error_output.py +27 -0
- vellum/client/types/test_suite_run_execution_function_call_output.py +27 -0
- vellum/client/types/test_suite_run_execution_json_output.py +26 -0
- vellum/client/types/test_suite_run_execution_metric_definition.py +21 -0
- vellum/client/types/test_suite_run_execution_metric_result.py +24 -0
- vellum/client/types/test_suite_run_execution_number_output.py +26 -0
- vellum/client/types/test_suite_run_execution_output.py +22 -0
- vellum/client/types/test_suite_run_execution_search_results_output.py +27 -0
- vellum/client/types/test_suite_run_execution_string_output.py +26 -0
- vellum/client/types/test_suite_run_external_exec_config.py +36 -0
- vellum/client/types/test_suite_run_external_exec_config_data.py +29 -0
- vellum/client/types/test_suite_run_external_exec_config_data_request.py +31 -0
- vellum/client/types/test_suite_run_external_exec_config_request.py +38 -0
- vellum/client/types/test_suite_run_metric_error_output.py +26 -0
- vellum/client/types/test_suite_run_metric_json_output.py +25 -0
- vellum/client/types/test_suite_run_metric_number_output.py +25 -0
- vellum/client/types/test_suite_run_metric_output.py +14 -0
- vellum/client/types/test_suite_run_metric_string_output.py +25 -0
- vellum/client/types/test_suite_run_read.py +46 -0
- vellum/client/types/test_suite_run_state.py +5 -0
- vellum/client/types/test_suite_run_test_suite.py +21 -0
- vellum/client/types/test_suite_run_workflow_release_tag_exec_config.py +29 -0
- vellum/client/types/test_suite_run_workflow_release_tag_exec_config_data.py +27 -0
- vellum/client/types/test_suite_run_workflow_release_tag_exec_config_data_request.py +27 -0
- vellum/client/types/test_suite_run_workflow_release_tag_exec_config_request.py +31 -0
- vellum/client/types/test_suite_test_case.py +30 -0
- vellum/client/types/test_suite_test_case_bulk_operation_request.py +14 -0
- vellum/client/types/test_suite_test_case_bulk_result.py +14 -0
- vellum/client/types/test_suite_test_case_create_bulk_operation_request.py +38 -0
- vellum/client/types/test_suite_test_case_created_bulk_result.py +26 -0
- vellum/client/types/test_suite_test_case_created_bulk_result_data.py +23 -0
- vellum/client/types/test_suite_test_case_delete_bulk_operation_data_request.py +19 -0
- vellum/client/types/test_suite_test_case_delete_bulk_operation_request.py +33 -0
- vellum/client/types/test_suite_test_case_deleted_bulk_result.py +30 -0
- vellum/client/types/test_suite_test_case_deleted_bulk_result_data.py +23 -0
- vellum/client/types/test_suite_test_case_rejected_bulk_result.py +32 -0
- vellum/client/types/test_suite_test_case_replace_bulk_operation_request.py +38 -0
- vellum/client/types/test_suite_test_case_replaced_bulk_result.py +30 -0
- vellum/client/types/test_suite_test_case_replaced_bulk_result_data.py +23 -0
- vellum/client/types/test_suite_test_case_upsert_bulk_operation_request.py +38 -0
- vellum/client/types/token_overlapping_window_chunker_config.py +24 -0
- vellum/client/types/token_overlapping_window_chunker_config_request.py +24 -0
- vellum/client/types/token_overlapping_window_chunking.py +25 -0
- vellum/client/types/token_overlapping_window_chunking_request.py +25 -0
- vellum/client/types/unit_enum.py +5 -0
- vellum/client/types/upload_document_response.py +22 -0
- vellum/client/types/upsert_test_suite_test_case_request.py +49 -0
- vellum/client/types/variable_prompt_block.py +28 -0
- vellum/client/types/vellum_audio.py +24 -0
- vellum/client/types/vellum_audio_request.py +24 -0
- vellum/client/types/vellum_error.py +21 -0
- vellum/client/types/vellum_error_code_enum.py +15 -0
- vellum/client/types/vellum_error_request.py +21 -0
- vellum/client/types/vellum_image.py +20 -0
- vellum/client/types/vellum_image_request.py +20 -0
- vellum/client/types/vellum_value.py +29 -0
- vellum/client/types/vellum_value_logical_condition_group_request.py +38 -0
- vellum/client/types/vellum_value_logical_condition_request.py +34 -0
- vellum/client/types/vellum_value_logical_expression_request.py +12 -0
- vellum/client/types/vellum_value_request.py +29 -0
- vellum/client/types/vellum_variable.py +33 -0
- vellum/client/types/vellum_variable_extensions.py +23 -0
- vellum/client/types/vellum_variable_type.py +20 -0
- vellum/client/types/workflow_deployment_read.py +77 -0
- vellum/client/types/workflow_event_error.py +21 -0
- vellum/client/types/workflow_execution_actual_chat_history_request.py +49 -0
- vellum/client/types/workflow_execution_actual_json_request.py +48 -0
- vellum/client/types/workflow_execution_actual_string_request.py +48 -0
- vellum/client/types/workflow_execution_event_error_code.py +17 -0
- vellum/client/types/workflow_execution_event_type.py +5 -0
- vellum/client/types/workflow_execution_node_result_event.py +34 -0
- vellum/client/types/workflow_execution_workflow_result_event.py +36 -0
- vellum/client/types/workflow_expand_meta_request.py +22 -0
- vellum/client/types/workflow_node_result_data.py +28 -0
- vellum/client/types/workflow_node_result_event.py +14 -0
- vellum/client/types/workflow_node_result_event_state.py +7 -0
- vellum/client/types/workflow_output.py +24 -0
- vellum/client/types/workflow_output_array.py +37 -0
- vellum/client/types/workflow_output_chat_history.py +31 -0
- vellum/client/types/workflow_output_error.py +31 -0
- vellum/client/types/workflow_output_function_call.py +31 -0
- vellum/client/types/workflow_output_image.py +31 -0
- vellum/client/types/workflow_output_json.py +30 -0
- vellum/client/types/workflow_output_number.py +30 -0
- vellum/client/types/workflow_output_search_results.py +31 -0
- vellum/client/types/workflow_output_string.py +30 -0
- vellum/client/types/workflow_push_deployment_config_request.py +22 -0
- vellum/client/types/workflow_push_exec_config.py +3 -0
- vellum/client/types/workflow_push_response.py +20 -0
- vellum/client/types/workflow_release_tag_read.py +37 -0
- vellum/client/types/workflow_release_tag_workflow_deployment_history_item.py +28 -0
- vellum/client/types/workflow_request_chat_history_input_request.py +30 -0
- vellum/client/types/workflow_request_input_request.py +14 -0
- vellum/client/types/workflow_request_json_input_request.py +29 -0
- vellum/client/types/workflow_request_number_input_request.py +29 -0
- vellum/client/types/workflow_request_string_input_request.py +29 -0
- vellum/client/types/workflow_result_event.py +39 -0
- vellum/client/types/workflow_result_event_output_data.py +22 -0
- vellum/client/types/workflow_result_event_output_data_array.py +41 -0
- vellum/client/types/workflow_result_event_output_data_chat_history.py +35 -0
- vellum/client/types/workflow_result_event_output_data_error.py +35 -0
- vellum/client/types/workflow_result_event_output_data_function_call.py +35 -0
- vellum/client/types/workflow_result_event_output_data_json.py +34 -0
- vellum/client/types/workflow_result_event_output_data_number.py +34 -0
- vellum/client/types/workflow_result_event_output_data_search_results.py +35 -0
- vellum/client/types/workflow_result_event_output_data_string.py +37 -0
- vellum/client/types/workflow_stream_event.py +7 -0
- vellum/client/types/workspace_secret_read.py +25 -0
- vellum/core/__init__.py +2 -46
- vellum/core/api_error.py +2 -14
- vellum/core/client_wrapper.py +2 -58
- vellum/core/datetime_utils.py +2 -27
- vellum/core/file.py +2 -61
- vellum/core/http_client.py +2 -486
- vellum/core/jsonable_encoder.py +2 -100
- vellum/core/pydantic_utilities.py +2 -295
- vellum/core/query_encoder.py +2 -57
- vellum/core/remove_none_from_dict.py +2 -10
- vellum/core/request_options.py +2 -31
- vellum/core/serialization.py +2 -271
- vellum/environment.py +2 -16
- vellum/errors/__init__.py +2 -7
- vellum/errors/bad_request_error.py +2 -8
- vellum/errors/forbidden_error.py +2 -8
- vellum/errors/internal_server_error.py +2 -8
- vellum/errors/not_found_error.py +2 -8
- vellum/{lib/test_suites → evaluations}/resources.py +20 -51
- vellum/plugins/__init__.py +0 -0
- vellum/plugins/pydantic.py +74 -0
- vellum/plugins/utils.py +19 -0
- vellum/plugins/vellum_mypy.py +647 -0
- vellum/resources/__init__.py +2 -46
- vellum/resources/ad_hoc/__init__.py +2 -1
- vellum/resources/ad_hoc/client.py +2 -458
- vellum/resources/container_images/__init__.py +2 -1
- vellum/resources/container_images/client.py +2 -501
- vellum/resources/deployments/__init__.py +2 -4
- vellum/resources/deployments/client.py +2 -974
- vellum/resources/deployments/types/__init__.py +2 -5
- vellum/resources/deployments/types/deployments_list_request_status.py +2 -4
- vellum/resources/deployments/types/list_deployment_release_tags_request_source.py +2 -4
- vellum/resources/document_indexes/__init__.py +2 -4
- vellum/resources/document_indexes/client.py +2 -1153
- vellum/resources/document_indexes/types/__init__.py +2 -4
- vellum/resources/document_indexes/types/document_indexes_list_request_status.py +2 -4
- vellum/resources/documents/__init__.py +2 -1
- vellum/resources/documents/client.py +2 -780
- vellum/resources/folder_entities/__init__.py +2 -4
- vellum/resources/folder_entities/client.py +2 -332
- vellum/resources/folder_entities/types/__init__.py +2 -4
- vellum/resources/folder_entities/types/folder_entities_list_request_entity_status.py +2 -4
- vellum/resources/metric_definitions/__init__.py +2 -1
- vellum/resources/metric_definitions/client.py +2 -177
- vellum/resources/sandboxes/__init__.py +2 -1
- vellum/resources/sandboxes/client.py +2 -480
- vellum/resources/test_suite_runs/__init__.py +2 -1
- vellum/resources/test_suite_runs/client.py +2 -444
- vellum/resources/test_suites/__init__.py +2 -1
- vellum/resources/test_suites/client.py +2 -687
- vellum/resources/workflow_deployments/__init__.py +2 -4
- vellum/resources/workflow_deployments/client.py +2 -692
- vellum/resources/workflow_deployments/types/__init__.py +2 -5
- vellum/resources/workflow_deployments/types/list_workflow_release_tags_request_source.py +2 -4
- vellum/resources/workflow_deployments/types/workflow_deployments_list_request_status.py +2 -4
- vellum/resources/workflow_sandboxes/__init__.py +2 -1
- vellum/resources/workflow_sandboxes/client.py +2 -189
- vellum/resources/workflows/__init__.py +2 -4
- vellum/resources/workflows/client.py +2 -332
- vellum/resources/workflows/types/__init__.py +2 -4
- vellum/resources/workflows/types/workflows_pull_request_format.py +2 -4
- vellum/resources/workspace_secrets/__init__.py +2 -1
- vellum/resources/workspace_secrets/client.py +2 -265
- vellum/types/__init__.py +2 -978
- vellum/types/ad_hoc_execute_prompt_event.py +2 -13
- vellum/types/ad_hoc_expand_meta.py +2 -36
- vellum/types/ad_hoc_fulfilled_prompt_execution_meta.py +2 -26
- vellum/types/ad_hoc_initiated_prompt_execution_meta.py +2 -23
- vellum/types/ad_hoc_rejected_prompt_execution_meta.py +2 -24
- vellum/types/ad_hoc_streaming_prompt_execution_meta.py +2 -22
- vellum/types/add_openai_api_key_enum.py +2 -2
- vellum/types/api_node_result.py +2 -24
- vellum/types/api_node_result_data.py +2 -27
- vellum/types/array_chat_message_content.py +2 -24
- vellum/types/array_chat_message_content_item.py +2 -10
- vellum/types/array_chat_message_content_item_request.py +2 -13
- vellum/types/array_chat_message_content_request.py +2 -24
- vellum/types/array_input.py +2 -35
- vellum/types/array_variable_value.py +2 -26
- vellum/types/array_variable_value_item.py +2 -28
- vellum/types/array_vellum_value.py +2 -30
- vellum/types/array_vellum_value_request.py +2 -30
- vellum/types/audio_chat_message_content.py +2 -24
- vellum/types/audio_chat_message_content_request.py +2 -24
- vellum/types/audio_variable_value.py +2 -24
- vellum/types/audio_vellum_value.py +2 -24
- vellum/types/audio_vellum_value_request.py +2 -24
- vellum/types/basic_vectorizer_intfloat_multilingual_e_5_large.py +2 -23
- vellum/types/basic_vectorizer_intfloat_multilingual_e_5_large_request.py +2 -23
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1.py +2 -25
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request.py +2 -25
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1.py +2 -25
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request.py +2 -25
- vellum/types/chat_history_input.py +2 -29
- vellum/types/chat_history_input_request.py +2 -29
- vellum/types/chat_history_variable_value.py +2 -20
- vellum/types/chat_history_vellum_value.py +2 -24
- vellum/types/chat_history_vellum_value_request.py +2 -24
- vellum/types/chat_message.py +2 -26
- vellum/types/chat_message_content.py +2 -15
- vellum/types/chat_message_content_request.py +2 -15
- vellum/types/chat_message_prompt_block.py +2 -38
- vellum/types/chat_message_request.py +2 -26
- vellum/types/chat_message_role.py +2 -4
- vellum/types/code_execution_node_array_result.py +2 -27
- vellum/types/code_execution_node_chat_history_result.py +2 -21
- vellum/types/code_execution_node_error_result.py +2 -21
- vellum/types/code_execution_node_function_call_result.py +2 -21
- vellum/types/code_execution_node_json_result.py +2 -20
- vellum/types/code_execution_node_number_result.py +2 -20
- vellum/types/code_execution_node_result.py +2 -30
- vellum/types/code_execution_node_result_data.py +2 -26
- vellum/types/code_execution_node_result_output.py +2 -21
- vellum/types/code_execution_node_search_results_result.py +2 -21
- vellum/types/code_execution_node_string_result.py +2 -20
- vellum/types/code_execution_package.py +2 -19
- vellum/types/code_execution_runtime.py +2 -4
- vellum/types/code_executor_input.py +2 -23
- vellum/types/code_executor_response.py +2 -26
- vellum/types/code_executor_secret_input.py +2 -28
- vellum/types/compile_prompt_deployment_expand_meta_request.py +2 -31
- vellum/types/compile_prompt_meta.py +2 -24
- vellum/types/components_schemas_pdf_search_result_meta_source.py +2 -4
- vellum/types/components_schemas_pdf_search_result_meta_source_request.py +2 -4
- vellum/types/condition_combinator.py +2 -4
- vellum/types/conditional_node_result.py +2 -24
- vellum/types/conditional_node_result_data.py +2 -18
- vellum/types/container_image_read.py +2 -27
- vellum/types/create_test_suite_test_case_request.py +2 -47
- vellum/types/deployment_provider_payload_response.py +2 -21
- vellum/types/deployment_provider_payload_response_payload.py +2 -4
- vellum/types/deployment_read.py +2 -72
- vellum/types/deployment_release_tag_deployment_history_item.py +2 -20
- vellum/types/deployment_release_tag_read.py +2 -36
- vellum/types/docker_service_token.py +2 -20
- vellum/types/document_document_to_document_index.py +2 -38
- vellum/types/document_index_chunking.py +2 -7
- vellum/types/document_index_chunking_request.py +2 -9
- vellum/types/document_index_indexing_config.py +2 -21
- vellum/types/document_index_indexing_config_request.py +2 -21
- vellum/types/document_index_read.py +2 -51
- vellum/types/document_processing_state.py +2 -6
- vellum/types/document_read.py +2 -48
- vellum/types/document_status.py +2 -4
- vellum/types/enriched_normalized_completion.py +2 -58
- vellum/types/entity_status.py +2 -4
- vellum/types/entity_visibility.py +2 -4
- vellum/types/environment_enum.py +2 -4
- vellum/types/ephemeral_prompt_cache_config.py +2 -19
- vellum/types/ephemeral_prompt_cache_config_type_enum.py +2 -4
- vellum/types/error_input.py +2 -29
- vellum/types/error_variable_value.py +2 -20
- vellum/types/error_vellum_value.py +2 -24
- vellum/types/error_vellum_value_request.py +2 -24
- vellum/types/execute_prompt_event.py +2 -10
- vellum/types/execute_prompt_response.py +2 -6
- vellum/types/execute_workflow_response.py +2 -28
- vellum/types/execute_workflow_workflow_result_event.py +2 -8
- vellum/types/execution_array_vellum_value.py +2 -36
- vellum/types/execution_chat_history_vellum_value.py +2 -30
- vellum/types/execution_error_vellum_value.py +2 -30
- vellum/types/execution_function_call_vellum_value.py +2 -30
- vellum/types/execution_json_vellum_value.py +2 -29
- vellum/types/execution_number_vellum_value.py +2 -29
- vellum/types/execution_search_results_vellum_value.py +2 -30
- vellum/types/execution_string_vellum_value.py +2 -29
- vellum/types/execution_vellum_value.py +2 -21
- vellum/types/external_test_case_execution.py +2 -30
- vellum/types/external_test_case_execution_request.py +2 -30
- vellum/types/finish_reason_enum.py +2 -4
- vellum/types/folder_entity.py +2 -15
- vellum/types/folder_entity_document_index.py +2 -25
- vellum/types/folder_entity_document_index_data.py +2 -24
- vellum/types/folder_entity_folder.py +2 -25
- vellum/types/folder_entity_folder_data.py +2 -23
- vellum/types/folder_entity_prompt_sandbox.py +2 -25
- vellum/types/folder_entity_prompt_sandbox_data.py +2 -25
- vellum/types/folder_entity_test_suite.py +2 -25
- vellum/types/folder_entity_test_suite_data.py +2 -24
- vellum/types/folder_entity_workflow_sandbox.py +2 -25
- vellum/types/folder_entity_workflow_sandbox_data.py +2 -25
- vellum/types/fulfilled_ad_hoc_execute_prompt_event.py +2 -27
- vellum/types/fulfilled_enum.py +2 -4
- vellum/types/fulfilled_execute_prompt_event.py +2 -27
- vellum/types/fulfilled_execute_prompt_response.py +2 -36
- vellum/types/fulfilled_execute_workflow_workflow_result_event.py +2 -35
- vellum/types/fulfilled_prompt_execution_meta.py +2 -26
- vellum/types/fulfilled_workflow_node_result_event.py +2 -39
- vellum/types/function_call.py +2 -26
- vellum/types/function_call_chat_message_content.py +2 -24
- vellum/types/function_call_chat_message_content_request.py +2 -24
- vellum/types/function_call_chat_message_content_value.py +2 -24
- vellum/types/function_call_chat_message_content_value_request.py +2 -24
- vellum/types/function_call_input.py +2 -29
- vellum/types/function_call_request.py +2 -26
- vellum/types/function_call_variable_value.py +2 -20
- vellum/types/function_call_vellum_value.py +2 -24
- vellum/types/function_call_vellum_value_request.py +2 -24
- vellum/types/function_definition.py +2 -30
- vellum/types/generate_options_request.py +2 -25
- vellum/types/generate_request.py +2 -32
- vellum/types/generate_response.py +2 -24
- vellum/types/generate_result.py +2 -28
- vellum/types/generate_result_data.py +2 -22
- vellum/types/generate_result_error.py +2 -21
- vellum/types/generate_stream_response.py +2 -19
- vellum/types/generate_stream_result.py +2 -22
- vellum/types/generate_stream_result_data.py +2 -20
- vellum/types/google_vertex_ai_vectorizer_config.py +2 -19
- vellum/types/google_vertex_ai_vectorizer_config_request.py +2 -19
- vellum/types/google_vertex_ai_vectorizer_text_embedding_004.py +2 -20
- vellum/types/google_vertex_ai_vectorizer_text_embedding_004_request.py +2 -20
- vellum/types/google_vertex_ai_vectorizer_text_multilingual_embedding_002.py +2 -20
- vellum/types/google_vertex_ai_vectorizer_text_multilingual_embedding_002_request.py +2 -20
- vellum/types/hkunlp_instructor_xl_vectorizer.py +2 -24
- vellum/types/hkunlp_instructor_xl_vectorizer_request.py +2 -24
- vellum/types/image_chat_message_content.py +2 -24
- vellum/types/image_chat_message_content_request.py +2 -24
- vellum/types/image_variable_value.py +2 -24
- vellum/types/image_vellum_value.py +2 -24
- vellum/types/image_vellum_value_request.py +2 -24
- vellum/types/indexing_config_vectorizer.py +2 -29
- vellum/types/indexing_config_vectorizer_request.py +2 -29
- vellum/types/indexing_state_enum.py +2 -6
- vellum/types/initiated_ad_hoc_execute_prompt_event.py +2 -25
- vellum/types/initiated_execute_prompt_event.py +2 -25
- vellum/types/initiated_prompt_execution_meta.py +2 -25
- vellum/types/initiated_workflow_node_result_event.py +2 -38
- vellum/types/instructor_vectorizer_config.py +2 -24
- vellum/types/instructor_vectorizer_config_request.py +2 -24
- vellum/types/iteration_state_enum.py +2 -4
- vellum/types/jinja_prompt_block.py +2 -27
- vellum/types/json_input.py +2 -28
- vellum/types/json_input_request.py +2 -28
- vellum/types/json_variable_value.py +2 -19
- vellum/types/json_vellum_value.py +2 -23
- vellum/types/json_vellum_value_request.py +2 -23
- vellum/types/logical_operator.py +2 -28
- vellum/types/logprobs_enum.py +2 -4
- vellum/types/map_node_result.py +2 -24
- vellum/types/map_node_result_data.py +2 -20
- vellum/types/merge_node_result.py +2 -24
- vellum/types/merge_node_result_data.py +2 -18
- vellum/types/metadata_filter_config_request.py +2 -35
- vellum/types/metadata_filter_rule_combinator.py +2 -4
- vellum/types/metadata_filter_rule_request.py +2 -30
- vellum/types/metadata_filters_request.py +2 -6
- vellum/types/metric_definition_execution.py +2 -19
- vellum/types/metric_definition_input.py +2 -8
- vellum/types/metric_node_result.py +2 -22
- vellum/types/ml_model_usage.py +2 -24
- vellum/types/named_scenario_input_chat_history_variable_value_request.py +2 -25
- vellum/types/named_scenario_input_json_variable_value_request.py +2 -24
- vellum/types/named_scenario_input_request.py +2 -11
- vellum/types/named_scenario_input_string_variable_value_request.py +2 -24
- vellum/types/named_test_case_array_variable_value.py +2 -31
- vellum/types/named_test_case_array_variable_value_request.py +2 -33
- vellum/types/named_test_case_chat_history_variable_value.py +2 -25
- vellum/types/named_test_case_chat_history_variable_value_request.py +2 -25
- vellum/types/named_test_case_error_variable_value.py +2 -25
- vellum/types/named_test_case_error_variable_value_request.py +2 -25
- vellum/types/named_test_case_function_call_variable_value.py +2 -25
- vellum/types/named_test_case_function_call_variable_value_request.py +2 -25
- vellum/types/named_test_case_json_variable_value.py +2 -24
- vellum/types/named_test_case_json_variable_value_request.py +2 -24
- vellum/types/named_test_case_number_variable_value.py +2 -24
- vellum/types/named_test_case_number_variable_value_request.py +2 -24
- vellum/types/named_test_case_search_results_variable_value.py +2 -25
- vellum/types/named_test_case_search_results_variable_value_request.py +2 -25
- vellum/types/named_test_case_string_variable_value.py +2 -24
- vellum/types/named_test_case_string_variable_value_request.py +2 -24
- vellum/types/named_test_case_variable_value.py +2 -21
- vellum/types/named_test_case_variable_value_request.py +2 -21
- vellum/types/node_input_compiled_array_value.py +2 -28
- vellum/types/node_input_compiled_chat_history_value.py +2 -22
- vellum/types/node_input_compiled_error_value.py +2 -22
- vellum/types/node_input_compiled_function_call_value.py +2 -22
- vellum/types/node_input_compiled_json_value.py +2 -21
- vellum/types/node_input_compiled_number_value.py +2 -21
- vellum/types/node_input_compiled_search_results_value.py +2 -22
- vellum/types/node_input_compiled_string_value.py +2 -21
- vellum/types/node_input_variable_compiled_value.py +2 -21
- vellum/types/node_output_compiled_array_value.py +2 -33
- vellum/types/node_output_compiled_chat_history_value.py +2 -27
- vellum/types/node_output_compiled_error_value.py +2 -27
- vellum/types/node_output_compiled_function_call_value.py +2 -27
- vellum/types/node_output_compiled_json_value.py +2 -26
- vellum/types/node_output_compiled_number_value.py +2 -26
- vellum/types/node_output_compiled_search_results_value.py +2 -27
- vellum/types/node_output_compiled_string_value.py +2 -26
- vellum/types/node_output_compiled_value.py +2 -21
- vellum/types/normalized_log_probs.py +2 -20
- vellum/types/normalized_token_log_probs.py +2 -21
- vellum/types/number_input.py +2 -28
- vellum/types/number_variable_value.py +2 -19
- vellum/types/number_vellum_value.py +2 -23
- vellum/types/number_vellum_value_request.py +2 -23
- vellum/types/open_ai_vectorizer_config.py +2 -23
- vellum/types/open_ai_vectorizer_config_request.py +2 -23
- vellum/types/open_ai_vectorizer_text_embedding_3_large.py +2 -24
- vellum/types/open_ai_vectorizer_text_embedding_3_large_request.py +2 -24
- vellum/types/open_ai_vectorizer_text_embedding_3_small.py +2 -24
- vellum/types/open_ai_vectorizer_text_embedding_3_small_request.py +2 -24
- vellum/types/open_ai_vectorizer_text_embedding_ada_002.py +2 -24
- vellum/types/open_ai_vectorizer_text_embedding_ada_002_request.py +2 -24
- vellum/types/paginated_container_image_read_list.py +2 -22
- vellum/types/paginated_deployment_release_tag_read_list.py +2 -22
- vellum/types/paginated_document_index_read_list.py +2 -22
- vellum/types/paginated_folder_entity_list.py +2 -22
- vellum/types/paginated_slim_deployment_read_list.py +2 -28
- vellum/types/paginated_slim_document_list.py +2 -22
- vellum/types/paginated_slim_workflow_deployment_list.py +2 -28
- vellum/types/paginated_test_suite_run_execution_list.py +2 -28
- vellum/types/paginated_test_suite_test_case_list.py +2 -28
- vellum/types/paginated_workflow_release_tag_read_list.py +2 -22
- vellum/types/pdf_search_result_meta_source.py +2 -31
- vellum/types/pdf_search_result_meta_source_request.py +2 -31
- vellum/types/plain_text_prompt_block.py +2 -27
- vellum/types/price.py +2 -20
- vellum/types/processing_failure_reason_enum.py +2 -4
- vellum/types/prompt_block.py +2 -11
- vellum/types/prompt_block_state.py +2 -4
- vellum/types/prompt_deployment_expand_meta_request.py +2 -51
- vellum/types/prompt_deployment_input_request.py +2 -7
- vellum/types/prompt_execution_meta.py +2 -31
- vellum/types/prompt_node_execution_meta.py +2 -23
- vellum/types/prompt_node_result.py +2 -24
- vellum/types/prompt_node_result_data.py +2 -24
- vellum/types/prompt_output.py +2 -8
- vellum/types/prompt_parameters.py +2 -26
- vellum/types/prompt_request_chat_history_input.py +2 -25
- vellum/types/prompt_request_input.py +2 -7
- vellum/types/prompt_request_json_input.py +2 -24
- vellum/types/prompt_request_string_input.py +2 -24
- vellum/types/prompt_settings.py +2 -18
- vellum/types/raw_prompt_execution_overrides_request.py +2 -27
- vellum/types/reducto_chunker_config.py +2 -22
- vellum/types/reducto_chunker_config_request.py +2 -22
- vellum/types/reducto_chunking.py +2 -24
- vellum/types/reducto_chunking_request.py +2 -24
- vellum/types/rejected_ad_hoc_execute_prompt_event.py +2 -27
- vellum/types/rejected_execute_prompt_event.py +2 -27
- vellum/types/rejected_execute_prompt_response.py +2 -36
- vellum/types/rejected_execute_workflow_workflow_result_event.py +2 -27
- vellum/types/rejected_prompt_execution_meta.py +2 -24
- vellum/types/rejected_workflow_node_result_event.py +2 -38
- vellum/types/release_tag_source.py +2 -4
- vellum/types/replace_test_suite_test_case_request.py +2 -52
- vellum/types/rich_text_child_block.py +2 -6
- vellum/types/rich_text_prompt_block.py +2 -28
- vellum/types/sandbox_scenario.py +2 -32
- vellum/types/scenario_input.py +2 -9
- vellum/types/scenario_input_chat_history_variable_value.py +2 -25
- vellum/types/scenario_input_json_variable_value.py +2 -24
- vellum/types/scenario_input_string_variable_value.py +2 -24
- vellum/types/search_filters_request.py +2 -37
- vellum/types/search_node_result.py +2 -24
- vellum/types/search_node_result_data.py +2 -26
- vellum/types/search_request_options_request.py +2 -49
- vellum/types/search_response.py +2 -22
- vellum/types/search_result.py +2 -39
- vellum/types/search_result_document.py +2 -36
- vellum/types/search_result_document_request.py +2 -36
- vellum/types/search_result_merging_request.py +2 -21
- vellum/types/search_result_meta.py +2 -19
- vellum/types/search_result_meta_request.py +2 -19
- vellum/types/search_result_request.py +2 -39
- vellum/types/search_results_input.py +2 -29
- vellum/types/search_results_variable_value.py +2 -20
- vellum/types/search_results_vellum_value.py +2 -24
- vellum/types/search_results_vellum_value_request.py +2 -24
- vellum/types/search_weights_request.py +2 -26
- vellum/types/secret_type_enum.py +2 -4
- vellum/types/sentence_chunker_config.py +2 -23
- vellum/types/sentence_chunker_config_request.py +2 -23
- vellum/types/sentence_chunking.py +2 -24
- vellum/types/sentence_chunking_request.py +2 -24
- vellum/types/slim_deployment_read.py +2 -62
- vellum/types/slim_document.py +2 -69
- vellum/types/slim_workflow_deployment.py +2 -71
- vellum/types/streaming_ad_hoc_execute_prompt_event.py +2 -32
- vellum/types/streaming_execute_prompt_event.py +2 -32
- vellum/types/streaming_prompt_execution_meta.py +2 -22
- vellum/types/streaming_workflow_node_result_event.py +2 -39
- vellum/types/string_chat_message_content.py +2 -23
- vellum/types/string_chat_message_content_request.py +2 -23
- vellum/types/string_input.py +2 -28
- vellum/types/string_input_request.py +2 -28
- vellum/types/string_variable_value.py +2 -19
- vellum/types/string_vellum_value.py +2 -23
- vellum/types/string_vellum_value_request.py +2 -23
- vellum/types/submit_completion_actual_request.py +2 -47
- vellum/types/submit_workflow_execution_actual_request.py +2 -9
- vellum/types/subworkflow_node_result.py +2 -24
- vellum/types/subworkflow_node_result_data.py +2 -18
- vellum/types/templating_node_array_result.py +2 -27
- vellum/types/templating_node_chat_history_result.py +2 -21
- vellum/types/templating_node_error_result.py +2 -21
- vellum/types/templating_node_function_call_result.py +2 -21
- vellum/types/templating_node_json_result.py +2 -20
- vellum/types/templating_node_number_result.py +2 -20
- vellum/types/templating_node_result.py +2 -30
- vellum/types/templating_node_result_data.py +2 -25
- vellum/types/templating_node_result_output.py +2 -21
- vellum/types/templating_node_search_results_result.py +2 -21
- vellum/types/templating_node_string_result.py +2 -20
- vellum/types/terminal_node_array_result.py +2 -32
- vellum/types/terminal_node_chat_history_result.py +2 -26
- vellum/types/terminal_node_error_result.py +2 -26
- vellum/types/terminal_node_function_call_result.py +2 -26
- vellum/types/terminal_node_json_result.py +2 -25
- vellum/types/terminal_node_number_result.py +2 -25
- vellum/types/terminal_node_result.py +2 -30
- vellum/types/terminal_node_result_data.py +2 -25
- vellum/types/terminal_node_result_output.py +2 -21
- vellum/types/terminal_node_search_results_result.py +2 -26
- vellum/types/terminal_node_string_result.py +2 -25
- vellum/types/test_case_array_variable_value.py +2 -32
- vellum/types/test_case_chat_history_variable_value.py +2 -26
- vellum/types/test_case_error_variable_value.py +2 -26
- vellum/types/test_case_function_call_variable_value.py +2 -26
- vellum/types/test_case_json_variable_value.py +2 -25
- vellum/types/test_case_number_variable_value.py +2 -25
- vellum/types/test_case_search_results_variable_value.py +2 -26
- vellum/types/test_case_string_variable_value.py +2 -25
- vellum/types/test_case_variable_value.py +2 -21
- vellum/types/test_suite_run_deployment_release_tag_exec_config.py +2 -28
- vellum/types/test_suite_run_deployment_release_tag_exec_config_data.py +2 -26
- vellum/types/test_suite_run_deployment_release_tag_exec_config_data_request.py +2 -26
- vellum/types/test_suite_run_deployment_release_tag_exec_config_request.py +2 -30
- vellum/types/test_suite_run_exec_config.py +2 -9
- vellum/types/test_suite_run_exec_config_request.py +2 -11
- vellum/types/test_suite_run_execution.py +2 -29
- vellum/types/test_suite_run_execution_array_output.py +2 -32
- vellum/types/test_suite_run_execution_chat_history_output.py +2 -26
- vellum/types/test_suite_run_execution_error_output.py +2 -26
- vellum/types/test_suite_run_execution_function_call_output.py +2 -26
- vellum/types/test_suite_run_execution_json_output.py +2 -25
- vellum/types/test_suite_run_execution_metric_definition.py +2 -20
- vellum/types/test_suite_run_execution_metric_result.py +2 -23
- vellum/types/test_suite_run_execution_number_output.py +2 -25
- vellum/types/test_suite_run_execution_output.py +2 -21
- vellum/types/test_suite_run_execution_search_results_output.py +2 -26
- vellum/types/test_suite_run_execution_string_output.py +2 -25
- vellum/types/test_suite_run_external_exec_config.py +2 -35
- vellum/types/test_suite_run_external_exec_config_data.py +2 -28
- vellum/types/test_suite_run_external_exec_config_data_request.py +2 -30
- vellum/types/test_suite_run_external_exec_config_request.py +2 -37
- vellum/types/test_suite_run_metric_error_output.py +2 -25
- vellum/types/test_suite_run_metric_json_output.py +2 -24
- vellum/types/test_suite_run_metric_number_output.py +2 -24
- vellum/types/test_suite_run_metric_output.py +2 -13
- vellum/types/test_suite_run_metric_string_output.py +2 -24
- vellum/types/test_suite_run_read.py +2 -45
- vellum/types/test_suite_run_state.py +2 -4
- vellum/types/test_suite_run_test_suite.py +2 -20
- vellum/types/test_suite_run_workflow_release_tag_exec_config.py +2 -28
- vellum/types/test_suite_run_workflow_release_tag_exec_config_data.py +2 -26
- vellum/types/test_suite_run_workflow_release_tag_exec_config_data_request.py +2 -26
- vellum/types/test_suite_run_workflow_release_tag_exec_config_request.py +2 -30
- vellum/types/test_suite_test_case.py +2 -29
- vellum/types/test_suite_test_case_bulk_operation_request.py +2 -13
- vellum/types/test_suite_test_case_bulk_result.py +2 -13
- vellum/types/test_suite_test_case_create_bulk_operation_request.py +2 -37
- vellum/types/test_suite_test_case_created_bulk_result.py +2 -25
- vellum/types/test_suite_test_case_created_bulk_result_data.py +2 -22
- vellum/types/test_suite_test_case_delete_bulk_operation_data_request.py +2 -18
- vellum/types/test_suite_test_case_delete_bulk_operation_request.py +2 -32
- vellum/types/test_suite_test_case_deleted_bulk_result.py +2 -29
- vellum/types/test_suite_test_case_deleted_bulk_result_data.py +2 -22
- vellum/types/test_suite_test_case_rejected_bulk_result.py +2 -31
- vellum/types/test_suite_test_case_replace_bulk_operation_request.py +2 -37
- vellum/types/test_suite_test_case_replaced_bulk_result.py +2 -29
- vellum/types/test_suite_test_case_replaced_bulk_result_data.py +2 -22
- vellum/types/test_suite_test_case_upsert_bulk_operation_request.py +2 -37
- vellum/types/token_overlapping_window_chunker_config.py +2 -23
- vellum/types/token_overlapping_window_chunker_config_request.py +2 -23
- vellum/types/token_overlapping_window_chunking.py +2 -24
- vellum/types/token_overlapping_window_chunking_request.py +2 -24
- vellum/types/unit_enum.py +2 -4
- vellum/types/upload_document_response.py +2 -21
- vellum/types/upsert_test_suite_test_case_request.py +2 -48
- vellum/types/variable_prompt_block.py +2 -27
- vellum/types/vellum_audio.py +2 -23
- vellum/types/vellum_audio_request.py +2 -23
- vellum/types/vellum_error.py +2 -20
- vellum/types/vellum_error_code_enum.py +2 -14
- vellum/types/vellum_error_request.py +2 -20
- vellum/types/vellum_image.py +2 -19
- vellum/types/vellum_image_request.py +2 -19
- vellum/types/vellum_value.py +2 -28
- vellum/types/vellum_value_logical_condition_group_request.py +2 -37
- vellum/types/vellum_value_logical_condition_request.py +2 -33
- vellum/types/vellum_value_logical_expression_request.py +2 -11
- vellum/types/vellum_value_request.py +2 -28
- vellum/types/vellum_variable.py +2 -32
- vellum/types/vellum_variable_extensions.py +2 -22
- vellum/types/vellum_variable_type.py +2 -19
- vellum/types/workflow_deployment_read.py +2 -76
- vellum/types/workflow_event_error.py +2 -20
- vellum/types/workflow_execution_actual_chat_history_request.py +2 -48
- vellum/types/workflow_execution_actual_json_request.py +2 -47
- vellum/types/workflow_execution_actual_string_request.py +2 -47
- vellum/types/workflow_execution_event_error_code.py +2 -16
- vellum/types/workflow_execution_event_type.py +2 -4
- vellum/types/workflow_execution_node_result_event.py +2 -33
- vellum/types/workflow_execution_workflow_result_event.py +2 -35
- vellum/types/workflow_expand_meta_request.py +2 -21
- vellum/types/workflow_node_result_data.py +2 -27
- vellum/types/workflow_node_result_event.py +2 -13
- vellum/types/workflow_node_result_event_state.py +2 -6
- vellum/types/workflow_output.py +2 -23
- vellum/types/workflow_output_array.py +2 -36
- vellum/types/workflow_output_chat_history.py +2 -30
- vellum/types/workflow_output_error.py +2 -30
- vellum/types/workflow_output_function_call.py +2 -30
- vellum/types/workflow_output_image.py +2 -30
- vellum/types/workflow_output_json.py +2 -29
- vellum/types/workflow_output_number.py +2 -29
- vellum/types/workflow_output_search_results.py +2 -30
- vellum/types/workflow_output_string.py +2 -29
- vellum/types/workflow_push_deployment_config_request.py +2 -21
- vellum/types/workflow_push_exec_config.py +2 -2
- vellum/types/workflow_push_response.py +2 -19
- vellum/types/workflow_release_tag_read.py +2 -36
- vellum/types/workflow_release_tag_workflow_deployment_history_item.py +2 -27
- vellum/types/workflow_request_chat_history_input_request.py +2 -29
- vellum/types/workflow_request_input_request.py +2 -13
- vellum/types/workflow_request_json_input_request.py +2 -28
- vellum/types/workflow_request_number_input_request.py +2 -28
- vellum/types/workflow_request_string_input_request.py +2 -28
- vellum/types/workflow_result_event.py +2 -38
- vellum/types/workflow_result_event_output_data.py +2 -21
- vellum/types/workflow_result_event_output_data_array.py +2 -40
- vellum/types/workflow_result_event_output_data_chat_history.py +2 -34
- vellum/types/workflow_result_event_output_data_error.py +2 -34
- vellum/types/workflow_result_event_output_data_function_call.py +2 -34
- vellum/types/workflow_result_event_output_data_json.py +2 -33
- vellum/types/workflow_result_event_output_data_number.py +2 -33
- vellum/types/workflow_result_event_output_data_search_results.py +2 -34
- vellum/types/workflow_result_event_output_data_string.py +2 -36
- vellum/types/workflow_stream_event.py +2 -6
- vellum/types/workspace_secret_read.py +2 -24
- vellum/workflows/README.md +90 -0
- vellum/workflows/__init__.py +5 -0
- vellum/workflows/constants.py +43 -0
- vellum/workflows/descriptors/__init__.py +0 -0
- vellum/workflows/descriptors/base.py +339 -0
- vellum/workflows/descriptors/tests/test_utils.py +83 -0
- vellum/workflows/descriptors/utils.py +90 -0
- vellum/workflows/edges/__init__.py +5 -0
- vellum/workflows/edges/edge.py +23 -0
- vellum/workflows/emitters/__init__.py +5 -0
- vellum/workflows/emitters/base.py +14 -0
- vellum/workflows/environment/__init__.py +5 -0
- vellum/workflows/environment/environment.py +7 -0
- vellum/workflows/errors/__init__.py +6 -0
- vellum/workflows/errors/types.py +20 -0
- vellum/workflows/events/__init__.py +31 -0
- vellum/workflows/events/node.py +125 -0
- vellum/workflows/events/tests/__init__.py +0 -0
- vellum/workflows/events/tests/test_event.py +216 -0
- vellum/workflows/events/types.py +52 -0
- vellum/workflows/events/utils.py +5 -0
- vellum/workflows/events/workflow.py +139 -0
- vellum/workflows/exceptions.py +15 -0
- vellum/workflows/expressions/__init__.py +0 -0
- vellum/workflows/expressions/accessor.py +52 -0
- vellum/workflows/expressions/and_.py +32 -0
- vellum/workflows/expressions/begins_with.py +31 -0
- vellum/workflows/expressions/between.py +38 -0
- vellum/workflows/expressions/coalesce_expression.py +41 -0
- vellum/workflows/expressions/contains.py +30 -0
- vellum/workflows/expressions/does_not_begin_with.py +31 -0
- vellum/workflows/expressions/does_not_contain.py +30 -0
- vellum/workflows/expressions/does_not_end_with.py +31 -0
- vellum/workflows/expressions/does_not_equal.py +25 -0
- vellum/workflows/expressions/ends_with.py +31 -0
- vellum/workflows/expressions/equals.py +25 -0
- vellum/workflows/expressions/greater_than.py +33 -0
- vellum/workflows/expressions/greater_than_or_equal_to.py +33 -0
- vellum/workflows/expressions/in_.py +31 -0
- vellum/workflows/expressions/is_blank.py +24 -0
- vellum/workflows/expressions/is_not_blank.py +24 -0
- vellum/workflows/expressions/is_not_null.py +21 -0
- vellum/workflows/expressions/is_not_undefined.py +22 -0
- vellum/workflows/expressions/is_null.py +21 -0
- vellum/workflows/expressions/is_undefined.py +22 -0
- vellum/workflows/expressions/less_than.py +33 -0
- vellum/workflows/expressions/less_than_or_equal_to.py +33 -0
- vellum/workflows/expressions/not_between.py +38 -0
- vellum/workflows/expressions/not_in.py +31 -0
- vellum/workflows/expressions/or_.py +32 -0
- vellum/workflows/graph/__init__.py +3 -0
- vellum/workflows/graph/graph.py +131 -0
- vellum/workflows/graph/tests/__init__.py +0 -0
- vellum/workflows/graph/tests/test_graph.py +437 -0
- vellum/workflows/inputs/__init__.py +5 -0
- vellum/workflows/inputs/base.py +55 -0
- vellum/workflows/logging.py +14 -0
- vellum/workflows/nodes/__init__.py +46 -0
- vellum/workflows/nodes/bases/__init__.py +7 -0
- vellum/workflows/nodes/bases/base.py +332 -0
- vellum/workflows/nodes/bases/base_subworkflow_node/__init__.py +5 -0
- vellum/workflows/nodes/bases/base_subworkflow_node/node.py +10 -0
- vellum/workflows/nodes/bases/tests/__init__.py +0 -0
- vellum/workflows/nodes/bases/tests/test_base_node.py +125 -0
- vellum/workflows/nodes/core/__init__.py +16 -0
- vellum/workflows/nodes/core/error_node/__init__.py +5 -0
- vellum/workflows/nodes/core/error_node/node.py +26 -0
- vellum/workflows/nodes/core/inline_subworkflow_node/__init__.py +5 -0
- vellum/workflows/nodes/core/inline_subworkflow_node/node.py +73 -0
- vellum/workflows/nodes/core/map_node/__init__.py +5 -0
- vellum/workflows/nodes/core/map_node/node.py +147 -0
- vellum/workflows/nodes/core/map_node/tests/__init__.py +0 -0
- vellum/workflows/nodes/core/map_node/tests/test_node.py +65 -0
- vellum/workflows/nodes/core/retry_node/__init__.py +5 -0
- vellum/workflows/nodes/core/retry_node/node.py +106 -0
- vellum/workflows/nodes/core/retry_node/tests/__init__.py +0 -0
- vellum/workflows/nodes/core/retry_node/tests/test_node.py +93 -0
- vellum/workflows/nodes/core/templating_node/__init__.py +5 -0
- vellum/workflows/nodes/core/templating_node/custom_filters.py +12 -0
- vellum/workflows/nodes/core/templating_node/exceptions.py +2 -0
- vellum/workflows/nodes/core/templating_node/node.py +123 -0
- vellum/workflows/nodes/core/templating_node/render.py +55 -0
- vellum/workflows/nodes/core/templating_node/tests/test_templating_node.py +21 -0
- vellum/workflows/nodes/core/try_node/__init__.py +5 -0
- vellum/workflows/nodes/core/try_node/node.py +110 -0
- vellum/workflows/nodes/core/try_node/tests/__init__.py +0 -0
- vellum/workflows/nodes/core/try_node/tests/test_node.py +82 -0
- vellum/workflows/nodes/displayable/__init__.py +31 -0
- vellum/workflows/nodes/displayable/api_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/api_node/node.py +44 -0
- vellum/workflows/nodes/displayable/bases/__init__.py +11 -0
- vellum/workflows/nodes/displayable/bases/api_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/bases/api_node/node.py +70 -0
- vellum/workflows/nodes/displayable/bases/base_prompt_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/bases/base_prompt_node/node.py +60 -0
- vellum/workflows/nodes/displayable/bases/inline_prompt_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/bases/inline_prompt_node/constants.py +13 -0
- vellum/workflows/nodes/displayable/bases/inline_prompt_node/node.py +118 -0
- vellum/workflows/nodes/displayable/bases/prompt_deployment_node.py +98 -0
- vellum/workflows/nodes/displayable/bases/search_node.py +90 -0
- vellum/workflows/nodes/displayable/code_execution_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/code_execution_node/node.py +197 -0
- vellum/workflows/nodes/displayable/code_execution_node/tests/__init__.py +0 -0
- vellum/workflows/nodes/displayable/code_execution_node/tests/fixtures/__init__.py +0 -0
- vellum/workflows/nodes/displayable/code_execution_node/tests/fixtures/main.py +3 -0
- vellum/workflows/nodes/displayable/code_execution_node/tests/test_code_execution_node.py +111 -0
- vellum/workflows/nodes/displayable/code_execution_node/utils.py +10 -0
- vellum/workflows/nodes/displayable/conditional_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/conditional_node/node.py +25 -0
- vellum/workflows/nodes/displayable/final_output_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/final_output_node/node.py +43 -0
- vellum/workflows/nodes/displayable/guardrail_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/guardrail_node/node.py +97 -0
- vellum/workflows/nodes/displayable/inline_prompt_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/inline_prompt_node/node.py +41 -0
- vellum/workflows/nodes/displayable/merge_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/merge_node/node.py +10 -0
- vellum/workflows/nodes/displayable/prompt_deployment_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/prompt_deployment_node/node.py +45 -0
- vellum/workflows/nodes/displayable/search_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/search_node/node.py +26 -0
- vellum/workflows/nodes/displayable/subworkflow_deployment_node/__init__.py +5 -0
- vellum/workflows/nodes/displayable/subworkflow_deployment_node/node.py +156 -0
- vellum/workflows/nodes/displayable/tests/__init__.py +0 -0
- vellum/workflows/nodes/displayable/tests/test_inline_text_prompt_node.py +148 -0
- vellum/workflows/nodes/displayable/tests/test_search_node_wth_text_output.py +134 -0
- vellum/workflows/nodes/displayable/tests/test_text_prompt_deployment_node.py +80 -0
- vellum/workflows/nodes/utils.py +27 -0
- vellum/workflows/outputs/__init__.py +6 -0
- vellum/workflows/outputs/base.py +196 -0
- vellum/workflows/ports/__init__.py +7 -0
- vellum/workflows/ports/node_ports.py +75 -0
- vellum/workflows/ports/port.py +75 -0
- vellum/workflows/ports/utils.py +40 -0
- vellum/workflows/references/__init__.py +17 -0
- vellum/workflows/references/environment_variable.py +20 -0
- vellum/workflows/references/execution_count.py +20 -0
- vellum/workflows/references/external_input.py +49 -0
- vellum/workflows/references/input.py +7 -0
- vellum/workflows/references/lazy.py +55 -0
- vellum/workflows/references/node.py +43 -0
- vellum/workflows/references/output.py +78 -0
- vellum/workflows/references/state_value.py +23 -0
- vellum/workflows/references/vellum_secret.py +15 -0
- vellum/workflows/references/workflow_input.py +41 -0
- vellum/workflows/resolvers/__init__.py +5 -0
- vellum/workflows/resolvers/base.py +15 -0
- vellum/workflows/runner/__init__.py +5 -0
- vellum/workflows/runner/runner.py +588 -0
- vellum/workflows/runner/types.py +18 -0
- vellum/workflows/state/__init__.py +5 -0
- vellum/workflows/state/base.py +327 -0
- vellum/workflows/state/context.py +18 -0
- vellum/workflows/state/encoder.py +57 -0
- vellum/workflows/state/store.py +28 -0
- vellum/workflows/state/tests/__init__.py +0 -0
- vellum/workflows/state/tests/test_state.py +113 -0
- vellum/workflows/types/__init__.py +0 -0
- vellum/workflows/types/core.py +91 -0
- vellum/workflows/types/generics.py +14 -0
- vellum/workflows/types/stack.py +39 -0
- vellum/workflows/types/tests/__init__.py +0 -0
- vellum/workflows/types/tests/test_utils.py +76 -0
- vellum/workflows/types/utils.py +164 -0
- vellum/workflows/utils/__init__.py +0 -0
- vellum/workflows/utils/names.py +13 -0
- vellum/workflows/utils/tests/__init__.py +0 -0
- vellum/workflows/utils/tests/test_names.py +15 -0
- vellum/workflows/utils/tests/test_vellum_variables.py +25 -0
- vellum/workflows/utils/vellum_variables.py +81 -0
- vellum/workflows/vellum_client.py +18 -0
- vellum/workflows/workflows/__init__.py +5 -0
- vellum/workflows/workflows/base.py +365 -0
- {vellum_ai-0.9.16.dist-info → vellum_ai-0.9.16rc4.dist-info}/LICENSE +11 -3
- vellum_ai-0.9.16rc4.dist-info/METADATA +120 -0
- vellum_ai-0.9.16rc4.dist-info/RECORD +1339 -0
- vellum_ai-0.9.16rc4.dist-info/entry_points.txt +3 -0
- vellum_cli/__init__.py +72 -0
- vellum_cli/aliased_group.py +103 -0
- vellum_cli/config.py +96 -0
- vellum_cli/image_push.py +112 -0
- vellum_cli/logger.py +36 -0
- vellum_cli/pull.py +73 -0
- vellum_cli/push.py +121 -0
- vellum_cli/tests/test_config.py +100 -0
- vellum_cli/tests/test_pull.py +152 -0
- vellum_ee/__init__.py +0 -0
- vellum_ee/workflows/__init__.py +0 -0
- vellum_ee/workflows/display/__init__.py +0 -0
- vellum_ee/workflows/display/base.py +73 -0
- vellum_ee/workflows/display/nodes/__init__.py +4 -0
- vellum_ee/workflows/display/nodes/base_node_display.py +116 -0
- vellum_ee/workflows/display/nodes/base_node_vellum_display.py +36 -0
- vellum_ee/workflows/display/nodes/get_node_display_class.py +25 -0
- vellum_ee/workflows/display/nodes/tests/__init__.py +0 -0
- vellum_ee/workflows/display/nodes/tests/test_base_node_display.py +47 -0
- vellum_ee/workflows/display/nodes/types.py +18 -0
- vellum_ee/workflows/display/nodes/utils.py +33 -0
- vellum_ee/workflows/display/nodes/vellum/__init__.py +32 -0
- vellum_ee/workflows/display/nodes/vellum/api_node.py +205 -0
- vellum_ee/workflows/display/nodes/vellum/code_execution_node.py +71 -0
- vellum_ee/workflows/display/nodes/vellum/conditional_node.py +217 -0
- vellum_ee/workflows/display/nodes/vellum/final_output_node.py +61 -0
- vellum_ee/workflows/display/nodes/vellum/guardrail_node.py +49 -0
- vellum_ee/workflows/display/nodes/vellum/inline_prompt_node.py +170 -0
- vellum_ee/workflows/display/nodes/vellum/inline_subworkflow_node.py +99 -0
- vellum_ee/workflows/display/nodes/vellum/map_node.py +100 -0
- vellum_ee/workflows/display/nodes/vellum/merge_node.py +48 -0
- vellum_ee/workflows/display/nodes/vellum/prompt_deployment_node.py +68 -0
- vellum_ee/workflows/display/nodes/vellum/search_node.py +193 -0
- vellum_ee/workflows/display/nodes/vellum/subworkflow_deployment_node.py +58 -0
- vellum_ee/workflows/display/nodes/vellum/templating_node.py +67 -0
- vellum_ee/workflows/display/nodes/vellum/tests/__init__.py +0 -0
- vellum_ee/workflows/display/nodes/vellum/tests/test_utils.py +106 -0
- vellum_ee/workflows/display/nodes/vellum/try_node.py +38 -0
- vellum_ee/workflows/display/nodes/vellum/utils.py +76 -0
- vellum_ee/workflows/display/tests/__init__.py +0 -0
- vellum_ee/workflows/display/tests/workflow_serialization/__init__.py +0 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_api_node_serialization.py +426 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_code_execution_node_serialization.py +607 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_conditional_node_serialization.py +1175 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_guardrail_node_serialization.py +235 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_inline_subworkflow_serialization.py +511 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_map_node_serialization.py +372 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_merge_node_serialization.py +272 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_prompt_deployment_serialization.py +289 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_subworkflow_deployment_serialization.py +354 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_terminal_node_serialization.py +123 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_basic_try_node_serialization.py +84 -0
- vellum_ee/workflows/display/tests/workflow_serialization/test_complex_terminal_node_serialization.py +233 -0
- vellum_ee/workflows/display/types.py +46 -0
- vellum_ee/workflows/display/utils/__init__.py +0 -0
- vellum_ee/workflows/display/utils/tests/__init__.py +0 -0
- vellum_ee/workflows/display/utils/tests/test_uuids.py +16 -0
- vellum_ee/workflows/display/utils/uuids.py +24 -0
- vellum_ee/workflows/display/utils/vellum.py +121 -0
- vellum_ee/workflows/display/vellum.py +357 -0
- vellum_ee/workflows/display/workflows/__init__.py +5 -0
- vellum_ee/workflows/display/workflows/base_workflow_display.py +302 -0
- vellum_ee/workflows/display/workflows/get_vellum_workflow_display_class.py +32 -0
- vellum_ee/workflows/display/workflows/vellum_workflow_display.py +386 -0
- vellum/lib/__init__.py +0 -5
- vellum_ai-0.9.16.dist-info/METADATA +0 -148
- vellum_ai-0.9.16.dist-info/RECORD +0 -558
- /vellum/{client.py → client/__init__.py} +0 -0
- /vellum/{lib/test_suites → evaluations}/__init__.py +0 -0
- /vellum/{lib/test_suites → evaluations}/constants.py +0 -0
- /vellum/{lib/test_suites → evaluations}/exceptions.py +0 -0
- /vellum/{lib → evaluations}/utils/__init__.py +0 -0
- /vellum/{lib → evaluations}/utils/env.py +0 -0
- /vellum/{lib → evaluations}/utils/exceptions.py +0 -0
- /vellum/{lib → evaluations}/utils/paginator.py +0 -0
- /vellum/{lib → evaluations}/utils/typing.py +0 -0
- /vellum/{lib → evaluations}/utils/uuid.py +0 -0
- {vellum_ai-0.9.16.dist-info → vellum_ai-0.9.16rc4.dist-info}/WHEEL +0 -0
@@ -0,0 +1,588 @@
|
|
1
|
+
from collections import defaultdict
|
2
|
+
from copy import deepcopy
|
3
|
+
import logging
|
4
|
+
from queue import Empty, Queue
|
5
|
+
from threading import Event as ThreadingEvent, Thread
|
6
|
+
from uuid import UUID, uuid4
|
7
|
+
from typing import TYPE_CHECKING, Any, Dict, Generic, Iterable, Iterator, Optional, Sequence, Set, Type, Union
|
8
|
+
|
9
|
+
from vellum.workflows.constants import UNDEF
|
10
|
+
from vellum.workflows.descriptors.base import BaseDescriptor
|
11
|
+
from vellum.workflows.edges.edge import Edge
|
12
|
+
from vellum.workflows.errors import VellumError, VellumErrorCode
|
13
|
+
from vellum.workflows.events import (
|
14
|
+
NodeExecutionFulfilledEvent,
|
15
|
+
NodeExecutionInitiatedEvent,
|
16
|
+
NodeExecutionRejectedEvent,
|
17
|
+
NodeExecutionStreamingEvent,
|
18
|
+
WorkflowEvent,
|
19
|
+
WorkflowEventStream,
|
20
|
+
WorkflowExecutionFulfilledEvent,
|
21
|
+
WorkflowExecutionInitiatedEvent,
|
22
|
+
WorkflowExecutionRejectedEvent,
|
23
|
+
WorkflowExecutionStreamingEvent,
|
24
|
+
)
|
25
|
+
from vellum.workflows.events.node import (
|
26
|
+
NodeExecutionFulfilledBody,
|
27
|
+
NodeExecutionInitiatedBody,
|
28
|
+
NodeExecutionRejectedBody,
|
29
|
+
NodeExecutionStreamingBody,
|
30
|
+
)
|
31
|
+
from vellum.workflows.events.types import BaseEvent
|
32
|
+
from vellum.workflows.events.utils import is_terminal_event
|
33
|
+
from vellum.workflows.events.workflow import (
|
34
|
+
WorkflowExecutionFulfilledBody,
|
35
|
+
WorkflowExecutionInitiatedBody,
|
36
|
+
WorkflowExecutionPausedBody,
|
37
|
+
WorkflowExecutionPausedEvent,
|
38
|
+
WorkflowExecutionRejectedBody,
|
39
|
+
WorkflowExecutionResumedBody,
|
40
|
+
WorkflowExecutionResumedEvent,
|
41
|
+
WorkflowExecutionStreamingBody,
|
42
|
+
)
|
43
|
+
from vellum.workflows.exceptions import NodeException
|
44
|
+
from vellum.workflows.nodes.bases import BaseNode
|
45
|
+
from vellum.workflows.outputs import BaseOutputs
|
46
|
+
from vellum.workflows.outputs.base import BaseOutput
|
47
|
+
from vellum.workflows.ports.port import Port
|
48
|
+
from vellum.workflows.references import ExternalInputReference, OutputReference
|
49
|
+
from vellum.workflows.runner.types import WorkItemEvent
|
50
|
+
from vellum.workflows.state.base import BaseState
|
51
|
+
from vellum.workflows.types.generics import OutputsType, StateType, WorkflowInputsType
|
52
|
+
|
53
|
+
if TYPE_CHECKING:
|
54
|
+
from vellum.workflows import BaseWorkflow
|
55
|
+
|
56
|
+
logger = logging.getLogger(__name__)
|
57
|
+
|
58
|
+
RunFromNodeArg = Sequence[Type[BaseNode]]
|
59
|
+
ExternalInputsArg = Dict[ExternalInputReference, Any]
|
60
|
+
BackgroundThreadItem = Union[BaseState, WorkflowEvent, None]
|
61
|
+
|
62
|
+
|
63
|
+
class WorkflowRunner(Generic[StateType]):
|
64
|
+
_entrypoints: Iterable[Type[BaseNode]]
|
65
|
+
|
66
|
+
def __init__(
|
67
|
+
self,
|
68
|
+
workflow: "BaseWorkflow[WorkflowInputsType, StateType]",
|
69
|
+
inputs: Optional[WorkflowInputsType] = None,
|
70
|
+
state: Optional[StateType] = None,
|
71
|
+
entrypoint_nodes: Optional[RunFromNodeArg] = None,
|
72
|
+
external_inputs: Optional[ExternalInputsArg] = None,
|
73
|
+
cancel_signal: Optional[ThreadingEvent] = None,
|
74
|
+
):
|
75
|
+
if state and external_inputs:
|
76
|
+
raise ValueError("Can only run a Workflow providing one of state or external inputs, not both")
|
77
|
+
|
78
|
+
self.workflow = workflow
|
79
|
+
if entrypoint_nodes:
|
80
|
+
if len(list(entrypoint_nodes)) > 1:
|
81
|
+
raise ValueError("Cannot resume from multiple nodes")
|
82
|
+
|
83
|
+
# TODO: Support resuming from multiple nodes
|
84
|
+
# https://app.shortcut.com/vellum/story/4408
|
85
|
+
node = next(iter(entrypoint_nodes))
|
86
|
+
if state:
|
87
|
+
self._initial_state = state
|
88
|
+
else:
|
89
|
+
self._initial_state = self.workflow.get_state_at_node(node)
|
90
|
+
self._entrypoints = entrypoint_nodes
|
91
|
+
elif external_inputs:
|
92
|
+
self._initial_state = self.workflow.get_most_recent_state()
|
93
|
+
for descriptor, value in external_inputs.items():
|
94
|
+
self._initial_state.meta.external_inputs[descriptor] = value
|
95
|
+
|
96
|
+
self._entrypoints = [
|
97
|
+
ei.inputs_class.__parent_class__
|
98
|
+
for ei in external_inputs
|
99
|
+
if issubclass(ei.inputs_class.__parent_class__, BaseNode)
|
100
|
+
]
|
101
|
+
else:
|
102
|
+
normalized_inputs = deepcopy(inputs) if inputs else self.workflow.get_default_inputs()
|
103
|
+
if state:
|
104
|
+
self._initial_state = deepcopy(state)
|
105
|
+
self._initial_state.meta.workflow_inputs = normalized_inputs
|
106
|
+
else:
|
107
|
+
self._initial_state = self.workflow.get_default_state(normalized_inputs)
|
108
|
+
self._entrypoints = self.workflow.get_entrypoints()
|
109
|
+
|
110
|
+
self._work_item_event_queue: Queue[WorkItemEvent[StateType]] = Queue()
|
111
|
+
self._workflow_event_queue: Queue[WorkflowEvent] = Queue()
|
112
|
+
self._background_thread_queue: Queue[BackgroundThreadItem] = Queue()
|
113
|
+
self._dependencies: Dict[Type[BaseNode], Set[Type[BaseNode]]] = defaultdict(set)
|
114
|
+
self._state_forks: Set[StateType] = {self._initial_state}
|
115
|
+
|
116
|
+
self._active_nodes_by_execution_id: Dict[UUID, BaseNode[StateType]] = {}
|
117
|
+
self._cancel_signal = cancel_signal
|
118
|
+
|
119
|
+
setattr(
|
120
|
+
self._initial_state,
|
121
|
+
"__snapshot_callback__",
|
122
|
+
lambda s: self._snapshot_state(s),
|
123
|
+
)
|
124
|
+
|
125
|
+
def _snapshot_state(self, state: StateType) -> StateType:
|
126
|
+
self.workflow._store.append_state_snapshot(state)
|
127
|
+
self._background_thread_queue.put(state)
|
128
|
+
return state
|
129
|
+
|
130
|
+
def _emit_event(self, event: WorkflowEvent) -> WorkflowEvent:
|
131
|
+
self.workflow._store.append_event(event)
|
132
|
+
self._background_thread_queue.put(event)
|
133
|
+
return event
|
134
|
+
|
135
|
+
def _run_work_item(self, node: BaseNode[StateType], span_id: UUID) -> None:
|
136
|
+
self._work_item_event_queue.put(
|
137
|
+
WorkItemEvent(
|
138
|
+
node=node,
|
139
|
+
event=NodeExecutionInitiatedEvent(
|
140
|
+
trace_id=node.state.meta.trace_id,
|
141
|
+
span_id=span_id,
|
142
|
+
body=NodeExecutionInitiatedBody(
|
143
|
+
node_definition=node.__class__,
|
144
|
+
inputs=node._inputs,
|
145
|
+
),
|
146
|
+
),
|
147
|
+
)
|
148
|
+
)
|
149
|
+
|
150
|
+
logger.debug(f"Started running node: {node.__class__.__name__}")
|
151
|
+
|
152
|
+
try:
|
153
|
+
node_run_response = node.run()
|
154
|
+
ports = node.Ports()
|
155
|
+
if not isinstance(node_run_response, (BaseOutputs, Iterator)):
|
156
|
+
raise NodeException(
|
157
|
+
message=f"Node {node.__class__.__name__} did not return a valid node run response",
|
158
|
+
code=VellumErrorCode.INVALID_OUTPUTS,
|
159
|
+
)
|
160
|
+
|
161
|
+
if isinstance(node_run_response, BaseOutputs):
|
162
|
+
if not isinstance(node_run_response, node.Outputs):
|
163
|
+
raise NodeException(
|
164
|
+
message=f"Node {node.__class__.__name__} did not return a valid outputs object",
|
165
|
+
code=VellumErrorCode.INVALID_OUTPUTS,
|
166
|
+
)
|
167
|
+
|
168
|
+
outputs = node_run_response
|
169
|
+
else:
|
170
|
+
streaming_output_queues: Dict[str, Queue] = {}
|
171
|
+
outputs = node.Outputs()
|
172
|
+
|
173
|
+
for output in node_run_response:
|
174
|
+
invoked_ports = output > ports
|
175
|
+
if not output.is_fulfilled:
|
176
|
+
if output.name not in streaming_output_queues:
|
177
|
+
streaming_output_queues[output.name] = Queue()
|
178
|
+
output_descriptor = OutputReference(
|
179
|
+
name=output.name,
|
180
|
+
types=(type(output.delta),),
|
181
|
+
instance=None,
|
182
|
+
outputs_class=node.Outputs,
|
183
|
+
)
|
184
|
+
node.state.meta.node_outputs[output_descriptor] = streaming_output_queues[output.name]
|
185
|
+
self._work_item_event_queue.put(
|
186
|
+
WorkItemEvent(
|
187
|
+
node=node,
|
188
|
+
event=NodeExecutionStreamingEvent(
|
189
|
+
trace_id=node.state.meta.trace_id,
|
190
|
+
span_id=span_id,
|
191
|
+
body=NodeExecutionStreamingBody(
|
192
|
+
node_definition=node.__class__,
|
193
|
+
output=BaseOutput(name=output.name),
|
194
|
+
),
|
195
|
+
),
|
196
|
+
invoked_ports=invoked_ports,
|
197
|
+
)
|
198
|
+
)
|
199
|
+
|
200
|
+
streaming_output_queues[output.name].put(output.delta)
|
201
|
+
self._work_item_event_queue.put(
|
202
|
+
WorkItemEvent(
|
203
|
+
node=node,
|
204
|
+
event=NodeExecutionStreamingEvent(
|
205
|
+
trace_id=node.state.meta.trace_id,
|
206
|
+
span_id=span_id,
|
207
|
+
body=NodeExecutionStreamingBody(
|
208
|
+
node_definition=node.__class__,
|
209
|
+
output=output,
|
210
|
+
),
|
211
|
+
),
|
212
|
+
invoked_ports=invoked_ports,
|
213
|
+
)
|
214
|
+
)
|
215
|
+
else:
|
216
|
+
if output.name in streaming_output_queues:
|
217
|
+
streaming_output_queues[output.name].put(UNDEF)
|
218
|
+
|
219
|
+
setattr(outputs, output.name, output.value)
|
220
|
+
self._work_item_event_queue.put(
|
221
|
+
WorkItemEvent(
|
222
|
+
node=node,
|
223
|
+
event=NodeExecutionStreamingEvent(
|
224
|
+
trace_id=node.state.meta.trace_id,
|
225
|
+
span_id=span_id,
|
226
|
+
body=NodeExecutionStreamingBody(
|
227
|
+
node_definition=node.__class__,
|
228
|
+
output=output,
|
229
|
+
),
|
230
|
+
),
|
231
|
+
invoked_ports=invoked_ports,
|
232
|
+
)
|
233
|
+
)
|
234
|
+
|
235
|
+
for descriptor, output_value in outputs:
|
236
|
+
node.state.meta.node_outputs[descriptor] = output_value
|
237
|
+
|
238
|
+
invoked_ports = ports(outputs, node.state)
|
239
|
+
node.state.meta.node_execution_cache.fulfill_node_execution(node.__class__, span_id)
|
240
|
+
|
241
|
+
self._work_item_event_queue.put(
|
242
|
+
WorkItemEvent(
|
243
|
+
node=node,
|
244
|
+
event=NodeExecutionFulfilledEvent(
|
245
|
+
trace_id=node.state.meta.trace_id,
|
246
|
+
span_id=span_id,
|
247
|
+
body=NodeExecutionFulfilledBody(
|
248
|
+
node_definition=node.__class__,
|
249
|
+
outputs=outputs,
|
250
|
+
),
|
251
|
+
),
|
252
|
+
invoked_ports=invoked_ports,
|
253
|
+
)
|
254
|
+
)
|
255
|
+
except NodeException as e:
|
256
|
+
self._work_item_event_queue.put(
|
257
|
+
WorkItemEvent(
|
258
|
+
node=node,
|
259
|
+
event=NodeExecutionRejectedEvent(
|
260
|
+
trace_id=node.state.meta.trace_id,
|
261
|
+
span_id=span_id,
|
262
|
+
body=NodeExecutionRejectedBody(
|
263
|
+
node_definition=node.__class__,
|
264
|
+
error=e.error,
|
265
|
+
),
|
266
|
+
),
|
267
|
+
)
|
268
|
+
)
|
269
|
+
except Exception as e:
|
270
|
+
logger.exception(f"An unexpected error occurred while running node {node.__class__.__name__}")
|
271
|
+
|
272
|
+
self._work_item_event_queue.put(
|
273
|
+
WorkItemEvent(
|
274
|
+
node=node,
|
275
|
+
event=NodeExecutionRejectedEvent(
|
276
|
+
trace_id=node.state.meta.trace_id,
|
277
|
+
span_id=span_id,
|
278
|
+
body=NodeExecutionRejectedBody(
|
279
|
+
node_definition=node.__class__,
|
280
|
+
error=VellumError(
|
281
|
+
message=str(e),
|
282
|
+
code=VellumErrorCode.INTERNAL_ERROR,
|
283
|
+
),
|
284
|
+
),
|
285
|
+
),
|
286
|
+
)
|
287
|
+
)
|
288
|
+
|
289
|
+
logger.debug(f"Finished running node: {node.__class__.__name__}")
|
290
|
+
|
291
|
+
def _handle_invoked_ports(self, state: StateType, ports: Optional[Iterable[Port]]) -> None:
|
292
|
+
if not ports:
|
293
|
+
return
|
294
|
+
|
295
|
+
for port in ports:
|
296
|
+
for edge in port.edges:
|
297
|
+
if port.fork_state:
|
298
|
+
next_state = deepcopy(state)
|
299
|
+
self._state_forks.add(next_state)
|
300
|
+
else:
|
301
|
+
next_state = state
|
302
|
+
|
303
|
+
self._run_node_if_ready(next_state, edge.to_node, edge)
|
304
|
+
|
305
|
+
def _run_node_if_ready(
|
306
|
+
self, state: StateType, node_class: Type[BaseNode], invoked_by: Optional[Edge] = None
|
307
|
+
) -> None:
|
308
|
+
with state.__lock__:
|
309
|
+
for descriptor in node_class.ExternalInputs:
|
310
|
+
if not isinstance(descriptor, ExternalInputReference):
|
311
|
+
continue
|
312
|
+
|
313
|
+
if state.meta.external_inputs.get(descriptor, UNDEF) is UNDEF:
|
314
|
+
state.meta.external_inputs[descriptor] = UNDEF
|
315
|
+
return
|
316
|
+
|
317
|
+
all_deps = self._dependencies[node_class]
|
318
|
+
if not node_class.Trigger.should_initiate(state, all_deps, invoked_by):
|
319
|
+
return
|
320
|
+
|
321
|
+
node = node_class(state=state, context=self.workflow.context)
|
322
|
+
node_span_id = uuid4()
|
323
|
+
state.meta.node_execution_cache.initiate_node_execution(node_class, node_span_id)
|
324
|
+
self._active_nodes_by_execution_id[node_span_id] = node
|
325
|
+
|
326
|
+
worker_thread = Thread(target=self._run_work_item, kwargs={"node": node, "span_id": node_span_id})
|
327
|
+
worker_thread.start()
|
328
|
+
|
329
|
+
def _handle_work_item_event(self, work_item_event: WorkItemEvent[StateType]) -> Optional[VellumError]:
|
330
|
+
node = work_item_event.node
|
331
|
+
event = work_item_event.event
|
332
|
+
invoked_ports = work_item_event.invoked_ports
|
333
|
+
|
334
|
+
if event.name == "node.execution.initiated":
|
335
|
+
return None
|
336
|
+
|
337
|
+
if event.name == "node.execution.rejected":
|
338
|
+
self._active_nodes_by_execution_id.pop(event.span_id)
|
339
|
+
return event.error
|
340
|
+
|
341
|
+
if event.name == "node.execution.streaming":
|
342
|
+
for workflow_output_descriptor in self.workflow.Outputs:
|
343
|
+
node_output_descriptor = workflow_output_descriptor.instance
|
344
|
+
if not isinstance(node_output_descriptor, OutputReference):
|
345
|
+
continue
|
346
|
+
if node_output_descriptor.outputs_class != event.node_definition.Outputs:
|
347
|
+
continue
|
348
|
+
if node_output_descriptor.name != event.output.name:
|
349
|
+
continue
|
350
|
+
|
351
|
+
self._workflow_event_queue.put(
|
352
|
+
self._stream_workflow_event(
|
353
|
+
BaseOutput(
|
354
|
+
name=workflow_output_descriptor.name,
|
355
|
+
value=event.output.value,
|
356
|
+
delta=event.output.delta,
|
357
|
+
)
|
358
|
+
)
|
359
|
+
)
|
360
|
+
|
361
|
+
self._handle_invoked_ports(node.state, invoked_ports)
|
362
|
+
|
363
|
+
return None
|
364
|
+
|
365
|
+
if event.name == "node.execution.fulfilled":
|
366
|
+
self._active_nodes_by_execution_id.pop(event.span_id)
|
367
|
+
self._handle_invoked_ports(node.state, invoked_ports)
|
368
|
+
|
369
|
+
return None
|
370
|
+
|
371
|
+
raise ValueError(f"Invalid event name: {event.name}")
|
372
|
+
|
373
|
+
def _initiate_workflow_event(self) -> WorkflowExecutionInitiatedEvent:
|
374
|
+
return WorkflowExecutionInitiatedEvent(
|
375
|
+
trace_id=self._initial_state.meta.trace_id,
|
376
|
+
span_id=self._initial_state.meta.span_id,
|
377
|
+
body=WorkflowExecutionInitiatedBody(
|
378
|
+
workflow_definition=self.workflow.__class__,
|
379
|
+
inputs=self._initial_state.meta.workflow_inputs,
|
380
|
+
),
|
381
|
+
)
|
382
|
+
|
383
|
+
def _stream_workflow_event(self, output: BaseOutput) -> WorkflowExecutionStreamingEvent:
|
384
|
+
return WorkflowExecutionStreamingEvent(
|
385
|
+
trace_id=self._initial_state.meta.trace_id,
|
386
|
+
span_id=self._initial_state.meta.span_id,
|
387
|
+
body=WorkflowExecutionStreamingBody(
|
388
|
+
workflow_definition=self.workflow.__class__,
|
389
|
+
output=output,
|
390
|
+
),
|
391
|
+
)
|
392
|
+
|
393
|
+
def _fulfill_workflow_event(self, outputs: OutputsType) -> WorkflowExecutionFulfilledEvent:
|
394
|
+
return WorkflowExecutionFulfilledEvent(
|
395
|
+
trace_id=self._initial_state.meta.trace_id,
|
396
|
+
span_id=self._initial_state.meta.span_id,
|
397
|
+
body=WorkflowExecutionFulfilledBody(
|
398
|
+
workflow_definition=self.workflow.__class__,
|
399
|
+
outputs=outputs,
|
400
|
+
),
|
401
|
+
)
|
402
|
+
|
403
|
+
def _reject_workflow_event(self, error: VellumError) -> WorkflowExecutionRejectedEvent:
|
404
|
+
return WorkflowExecutionRejectedEvent(
|
405
|
+
trace_id=self._initial_state.meta.trace_id,
|
406
|
+
span_id=self._initial_state.meta.span_id,
|
407
|
+
body=WorkflowExecutionRejectedBody(
|
408
|
+
workflow_definition=self.workflow.__class__,
|
409
|
+
error=error,
|
410
|
+
),
|
411
|
+
)
|
412
|
+
|
413
|
+
def _resume_workflow_event(self) -> WorkflowExecutionResumedEvent:
|
414
|
+
return WorkflowExecutionResumedEvent(
|
415
|
+
trace_id=self._initial_state.meta.trace_id,
|
416
|
+
span_id=self._initial_state.meta.span_id,
|
417
|
+
body=WorkflowExecutionResumedBody(
|
418
|
+
workflow_definition=self.workflow.__class__,
|
419
|
+
),
|
420
|
+
)
|
421
|
+
|
422
|
+
def _pause_workflow_event(self, external_inputs: Iterable[ExternalInputReference]) -> WorkflowExecutionPausedEvent:
|
423
|
+
return WorkflowExecutionPausedEvent(
|
424
|
+
trace_id=self._initial_state.meta.trace_id,
|
425
|
+
span_id=self._initial_state.meta.span_id,
|
426
|
+
body=WorkflowExecutionPausedBody(
|
427
|
+
workflow_definition=self.workflow.__class__,
|
428
|
+
external_inputs=external_inputs,
|
429
|
+
),
|
430
|
+
)
|
431
|
+
|
432
|
+
def _stream(self) -> None:
|
433
|
+
# TODO: We should likely handle this during initialization
|
434
|
+
# https://app.shortcut.com/vellum/story/4327
|
435
|
+
if not self._entrypoints:
|
436
|
+
self._workflow_event_queue.put(
|
437
|
+
self._reject_workflow_event(
|
438
|
+
VellumError(message="No entrypoints defined", code=VellumErrorCode.INVALID_WORKFLOW)
|
439
|
+
)
|
440
|
+
)
|
441
|
+
return
|
442
|
+
|
443
|
+
for edge in self.workflow.get_edges():
|
444
|
+
self._dependencies[edge.to_node].add(edge.from_port.node_class)
|
445
|
+
|
446
|
+
for node_cls in self._entrypoints:
|
447
|
+
try:
|
448
|
+
self._run_node_if_ready(self._initial_state, node_cls)
|
449
|
+
except NodeException as e:
|
450
|
+
self._workflow_event_queue.put(self._reject_workflow_event(e.error))
|
451
|
+
return
|
452
|
+
except Exception:
|
453
|
+
err_message = f"An unexpected error occurred while initializing node {node_cls.__name__}"
|
454
|
+
logger.exception(err_message)
|
455
|
+
self._workflow_event_queue.put(
|
456
|
+
self._reject_workflow_event(
|
457
|
+
VellumError(code=VellumErrorCode.INTERNAL_ERROR, message=err_message),
|
458
|
+
)
|
459
|
+
)
|
460
|
+
return
|
461
|
+
|
462
|
+
rejection_error: Optional[VellumError] = None
|
463
|
+
|
464
|
+
while True:
|
465
|
+
if not self._active_nodes_by_execution_id:
|
466
|
+
break
|
467
|
+
|
468
|
+
work_item_event = self._work_item_event_queue.get()
|
469
|
+
event = work_item_event.event
|
470
|
+
|
471
|
+
self._workflow_event_queue.put(event)
|
472
|
+
|
473
|
+
rejection_error = self._handle_work_item_event(work_item_event)
|
474
|
+
if rejection_error:
|
475
|
+
break
|
476
|
+
|
477
|
+
# Handle any remaining events
|
478
|
+
try:
|
479
|
+
while work_item_event := self._work_item_event_queue.get_nowait():
|
480
|
+
self._workflow_event_queue.put(work_item_event.event)
|
481
|
+
|
482
|
+
rejection_error = self._handle_work_item_event(work_item_event)
|
483
|
+
if rejection_error:
|
484
|
+
break
|
485
|
+
except Empty:
|
486
|
+
pass
|
487
|
+
|
488
|
+
final_state = self._state_forks.pop()
|
489
|
+
for other_state in self._state_forks:
|
490
|
+
final_state += other_state
|
491
|
+
|
492
|
+
unresolved_external_inputs = {
|
493
|
+
descriptor
|
494
|
+
for descriptor, node_input_value in final_state.meta.external_inputs.items()
|
495
|
+
if node_input_value is UNDEF
|
496
|
+
}
|
497
|
+
if unresolved_external_inputs:
|
498
|
+
self._workflow_event_queue.put(
|
499
|
+
self._pause_workflow_event(unresolved_external_inputs),
|
500
|
+
)
|
501
|
+
return
|
502
|
+
|
503
|
+
final_state.meta.is_terminated = True
|
504
|
+
if rejection_error:
|
505
|
+
self._workflow_event_queue.put(self._reject_workflow_event(rejection_error))
|
506
|
+
return
|
507
|
+
|
508
|
+
fulfilled_outputs = self.workflow.Outputs()
|
509
|
+
for descriptor, value in fulfilled_outputs:
|
510
|
+
if isinstance(value, BaseDescriptor):
|
511
|
+
setattr(fulfilled_outputs, descriptor.name, value.resolve(final_state))
|
512
|
+
elif isinstance(descriptor.instance, BaseDescriptor):
|
513
|
+
setattr(fulfilled_outputs, descriptor.name, descriptor.instance.resolve(final_state))
|
514
|
+
|
515
|
+
self._workflow_event_queue.put(self._fulfill_workflow_event(fulfilled_outputs))
|
516
|
+
|
517
|
+
def _run_background_thread(self) -> None:
|
518
|
+
state_class = self.workflow.get_state_class()
|
519
|
+
while True:
|
520
|
+
item = self._background_thread_queue.get()
|
521
|
+
if item is None:
|
522
|
+
break
|
523
|
+
|
524
|
+
if isinstance(item, state_class):
|
525
|
+
for emitter in self.workflow.emitters:
|
526
|
+
emitter.snapshot_state(item)
|
527
|
+
elif isinstance(item, BaseEvent):
|
528
|
+
for emitter in self.workflow.emitters:
|
529
|
+
emitter.emit_event(item)
|
530
|
+
|
531
|
+
def _run_cancel_thread(self) -> None:
|
532
|
+
if not self._cancel_signal:
|
533
|
+
return
|
534
|
+
|
535
|
+
self._cancel_signal.wait()
|
536
|
+
self._workflow_event_queue.put(
|
537
|
+
self._reject_workflow_event(
|
538
|
+
VellumError(code=VellumErrorCode.WORKFLOW_CANCELLED, message="Workflow run cancelled")
|
539
|
+
)
|
540
|
+
)
|
541
|
+
|
542
|
+
def stream(self) -> WorkflowEventStream:
|
543
|
+
background_thread = Thread(target=self._run_background_thread)
|
544
|
+
background_thread.start()
|
545
|
+
|
546
|
+
if self._cancel_signal:
|
547
|
+
cancel_thread = Thread(target=self._run_cancel_thread)
|
548
|
+
cancel_thread.start()
|
549
|
+
|
550
|
+
event: WorkflowEvent
|
551
|
+
if self._initial_state.meta.is_terminated or self._initial_state.meta.is_terminated is None:
|
552
|
+
event = self._initiate_workflow_event()
|
553
|
+
else:
|
554
|
+
event = self._resume_workflow_event()
|
555
|
+
|
556
|
+
yield self._emit_event(event)
|
557
|
+
self._initial_state.meta.is_terminated = False
|
558
|
+
|
559
|
+
# The extra level of indirection prevents the runner from waiting on the caller to consume the event stream
|
560
|
+
stream_thread = Thread(target=self._stream)
|
561
|
+
stream_thread.start()
|
562
|
+
|
563
|
+
while stream_thread.is_alive():
|
564
|
+
try:
|
565
|
+
event = self._workflow_event_queue.get(timeout=0.1)
|
566
|
+
except Empty:
|
567
|
+
continue
|
568
|
+
|
569
|
+
yield self._emit_event(event)
|
570
|
+
|
571
|
+
if is_terminal_event(event):
|
572
|
+
break
|
573
|
+
|
574
|
+
try:
|
575
|
+
while event := self._workflow_event_queue.get_nowait():
|
576
|
+
yield self._emit_event(event)
|
577
|
+
except Empty:
|
578
|
+
pass
|
579
|
+
|
580
|
+
if not is_terminal_event(event):
|
581
|
+
yield self._reject_workflow_event(
|
582
|
+
VellumError(
|
583
|
+
code=VellumErrorCode.INTERNAL_ERROR,
|
584
|
+
message="An unexpected error occurred while streaming Workflow events",
|
585
|
+
)
|
586
|
+
)
|
587
|
+
|
588
|
+
self._background_thread_queue.put(None)
|
@@ -0,0 +1,18 @@
|
|
1
|
+
"""Only intenral types and enums for WorkflowRunner should be defined in this module."""
|
2
|
+
|
3
|
+
from dataclasses import dataclass
|
4
|
+
from typing import TYPE_CHECKING, Generic, Iterable, Optional
|
5
|
+
|
6
|
+
from vellum.workflows.types.generics import StateType
|
7
|
+
|
8
|
+
if TYPE_CHECKING:
|
9
|
+
from vellum.workflows.events import NodeEvent
|
10
|
+
from vellum.workflows.nodes.bases import BaseNode
|
11
|
+
from vellum.workflows.ports import Port
|
12
|
+
|
13
|
+
|
14
|
+
@dataclass(frozen=True)
|
15
|
+
class WorkItemEvent(Generic[StateType]):
|
16
|
+
node: "BaseNode[StateType]"
|
17
|
+
event: "NodeEvent"
|
18
|
+
invoked_ports: Optional[Iterable["Port"]] = None
|