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,100 @@
|
|
1
|
+
import os
|
2
|
+
import tempfile
|
3
|
+
from uuid import uuid4
|
4
|
+
|
5
|
+
from vellum_cli.config import VellumCliConfig, WorkflowConfig, load_vellum_cli_config
|
6
|
+
|
7
|
+
|
8
|
+
def test_load_vellum_cli_config__pyproject_toml_only():
|
9
|
+
# GIVEN a pyproject.toml file with vellum config
|
10
|
+
cwd = tempfile.mkdtemp()
|
11
|
+
with open(os.path.join(cwd, "pyproject.toml"), "w") as f:
|
12
|
+
f.write(
|
13
|
+
"""[[tool.vellum.workflows]]
|
14
|
+
module = "module1.workflow1"
|
15
|
+
"""
|
16
|
+
)
|
17
|
+
|
18
|
+
# WHEN the config is loaded
|
19
|
+
config = load_vellum_cli_config(cwd)
|
20
|
+
|
21
|
+
# THEN the config is loaded correctly
|
22
|
+
assert config == VellumCliConfig(
|
23
|
+
workflows=[WorkflowConfig(module="module1.workflow1")],
|
24
|
+
version="1.0",
|
25
|
+
)
|
26
|
+
|
27
|
+
|
28
|
+
def test_load_vellum_cli_config__pyproject_toml_and_lockfile():
|
29
|
+
# GIVEN a pyproject.toml file with vellum config
|
30
|
+
cwd = tempfile.mkdtemp()
|
31
|
+
with open(os.path.join(cwd, "pyproject.toml"), "w") as f:
|
32
|
+
f.write(
|
33
|
+
"""[[tool.vellum.workflows]]
|
34
|
+
module = "module1.workflow1"
|
35
|
+
"""
|
36
|
+
)
|
37
|
+
|
38
|
+
# AND a lockfile
|
39
|
+
workflow_sandbox_id = uuid4()
|
40
|
+
with open(os.path.join(cwd, "vellum.lock.json"), "w") as f:
|
41
|
+
f.write(
|
42
|
+
f"""{{
|
43
|
+
"version": "1.0",
|
44
|
+
"workflows": [
|
45
|
+
{{
|
46
|
+
"workflow_sandbox_id": "{workflow_sandbox_id}",
|
47
|
+
"module": "module1.workflow1"
|
48
|
+
}}
|
49
|
+
]
|
50
|
+
}}
|
51
|
+
"""
|
52
|
+
)
|
53
|
+
|
54
|
+
# WHEN the config is loaded
|
55
|
+
config = load_vellum_cli_config(cwd)
|
56
|
+
|
57
|
+
# THEN the config is loaded correctly
|
58
|
+
assert config == VellumCliConfig(
|
59
|
+
workflows=[WorkflowConfig(module="module1.workflow1", workflow_sandbox_id=str(workflow_sandbox_id))],
|
60
|
+
version="1.0",
|
61
|
+
)
|
62
|
+
|
63
|
+
|
64
|
+
def test_load_vellum_cli_config__pyproject_toml_and_lockfile__different_modules():
|
65
|
+
# GIVEN a pyproject.toml file with vellum config
|
66
|
+
cwd = tempfile.mkdtemp()
|
67
|
+
with open(os.path.join(cwd, "pyproject.toml"), "w") as f:
|
68
|
+
f.write(
|
69
|
+
"""[[tool.vellum.workflows]]
|
70
|
+
module = "module1.workflow1"
|
71
|
+
"""
|
72
|
+
)
|
73
|
+
|
74
|
+
# AND a lockfile
|
75
|
+
workflow_sandbox_id = uuid4()
|
76
|
+
with open(os.path.join(cwd, "vellum.lock.json"), "w") as f:
|
77
|
+
f.write(
|
78
|
+
f"""{{
|
79
|
+
"version": "1.0",
|
80
|
+
"workflows": [
|
81
|
+
{{
|
82
|
+
"workflow_sandbox_id": "{workflow_sandbox_id}",
|
83
|
+
"module": "module2.workflow2"
|
84
|
+
}}
|
85
|
+
]
|
86
|
+
}}
|
87
|
+
"""
|
88
|
+
)
|
89
|
+
|
90
|
+
# WHEN the config is loaded
|
91
|
+
config = load_vellum_cli_config(cwd)
|
92
|
+
|
93
|
+
# THEN the config is loaded correctly
|
94
|
+
assert config == VellumCliConfig(
|
95
|
+
workflows=[
|
96
|
+
WorkflowConfig(module="module1.workflow1"),
|
97
|
+
WorkflowConfig(module="module2.workflow2", workflow_sandbox_id=str(workflow_sandbox_id)),
|
98
|
+
],
|
99
|
+
version="1.0",
|
100
|
+
)
|
@@ -0,0 +1,152 @@
|
|
1
|
+
import pytest
|
2
|
+
import io
|
3
|
+
import os
|
4
|
+
import shutil
|
5
|
+
import tempfile
|
6
|
+
from uuid import uuid4
|
7
|
+
import zipfile
|
8
|
+
from typing import Generator, Tuple
|
9
|
+
|
10
|
+
import tomli_w
|
11
|
+
|
12
|
+
from vellum_cli.pull import pull_command
|
13
|
+
|
14
|
+
|
15
|
+
def zip_file_map(file_map: dict[str, str]) -> bytes:
|
16
|
+
# Create an in-memory bytes buffer to store the zip
|
17
|
+
zip_buffer = io.BytesIO()
|
18
|
+
|
19
|
+
# Create zip file and add files from file_map
|
20
|
+
with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
|
21
|
+
for filename, content in file_map.items():
|
22
|
+
zip_file.writestr(filename, content)
|
23
|
+
|
24
|
+
# Get the bytes from the buffer
|
25
|
+
zip_bytes = zip_buffer.getvalue()
|
26
|
+
zip_buffer.close()
|
27
|
+
|
28
|
+
return zip_bytes
|
29
|
+
|
30
|
+
|
31
|
+
@pytest.fixture
|
32
|
+
def mock_module() -> Generator[Tuple[str, str], None, None]:
|
33
|
+
current_dir = os.getcwd()
|
34
|
+
temp_dir = tempfile.mkdtemp()
|
35
|
+
os.chdir(temp_dir)
|
36
|
+
module = "examples.mock"
|
37
|
+
|
38
|
+
with open(os.path.join(temp_dir, "pyproject.toml"), "wb") as f:
|
39
|
+
tomli_w.dump(
|
40
|
+
{
|
41
|
+
"tool": {
|
42
|
+
"vellum": {
|
43
|
+
"workflows": [
|
44
|
+
{
|
45
|
+
"module": module,
|
46
|
+
"workflow_sandbox_id": str(uuid4()),
|
47
|
+
}
|
48
|
+
]
|
49
|
+
}
|
50
|
+
}
|
51
|
+
},
|
52
|
+
f,
|
53
|
+
)
|
54
|
+
|
55
|
+
yield temp_dir, module
|
56
|
+
|
57
|
+
os.chdir(current_dir)
|
58
|
+
shutil.rmtree(temp_dir)
|
59
|
+
|
60
|
+
|
61
|
+
def test_pull(vellum_client, mock_module):
|
62
|
+
# GIVEN a module on the user's filesystem
|
63
|
+
temp_dir, module = mock_module
|
64
|
+
|
65
|
+
# AND the workflow pull API call returns a zip file
|
66
|
+
vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
|
67
|
+
|
68
|
+
# WHEN the user runs the pull command
|
69
|
+
pull_command(module)
|
70
|
+
|
71
|
+
# THEN the workflow.py file is written to the module directory
|
72
|
+
assert os.path.exists(os.path.join(temp_dir, *module.split("."), "workflow.py"))
|
73
|
+
with open(os.path.join(temp_dir, *module.split("."), "workflow.py")) as f:
|
74
|
+
assert f.read() == "print('hello')"
|
75
|
+
|
76
|
+
|
77
|
+
def test_pull__remove_missing_files(vellum_client, mock_module):
|
78
|
+
# GIVEN a module on the user's filesystem
|
79
|
+
temp_dir, module = mock_module
|
80
|
+
|
81
|
+
# AND the workflow pull API call returns a zip file
|
82
|
+
vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
|
83
|
+
|
84
|
+
# AND there is already a different file in the module directory
|
85
|
+
other_file_path = os.path.join(temp_dir, *module.split("."), "other_file.py")
|
86
|
+
os.makedirs(os.path.dirname(other_file_path), exist_ok=True)
|
87
|
+
with open(other_file_path, "w") as f:
|
88
|
+
f.write("print('hello')")
|
89
|
+
|
90
|
+
# WHEN the user runs the pull command
|
91
|
+
pull_command(module)
|
92
|
+
|
93
|
+
# THEN the workflow.py file is written to the module directory
|
94
|
+
assert os.path.exists(os.path.join(temp_dir, *module.split("."), "workflow.py"))
|
95
|
+
with open(os.path.join(temp_dir, *module.split("."), "workflow.py")) as f:
|
96
|
+
assert f.read() == "print('hello')"
|
97
|
+
|
98
|
+
# AND the other_file.py file is deleted
|
99
|
+
assert not os.path.exists(other_file_path)
|
100
|
+
|
101
|
+
|
102
|
+
def test_pull__remove_missing_files__ignore_pattern(vellum_client, mock_module):
|
103
|
+
# GIVEN a module on the user's filesystem
|
104
|
+
temp_dir, module = mock_module
|
105
|
+
|
106
|
+
# AND the workflow pull API call returns a zip file
|
107
|
+
vellum_client.workflows.pull.return_value = iter([zip_file_map({"workflow.py": "print('hello')"})])
|
108
|
+
|
109
|
+
# AND there is already a different file in the module directory
|
110
|
+
other_file_path = os.path.join(temp_dir, *module.split("."), "other_file.py")
|
111
|
+
os.makedirs(os.path.dirname(other_file_path), exist_ok=True)
|
112
|
+
with open(other_file_path, "w") as f:
|
113
|
+
f.write("print('hello')")
|
114
|
+
|
115
|
+
# AND there is already a test file
|
116
|
+
test_file_path = os.path.join(temp_dir, *module.split("."), "tests", "test_workflow.py")
|
117
|
+
os.makedirs(os.path.dirname(test_file_path), exist_ok=True)
|
118
|
+
with open(test_file_path, "w") as f:
|
119
|
+
f.write("print('hello')")
|
120
|
+
|
121
|
+
# AND the ignore pattern is set to tests
|
122
|
+
with open(os.path.join(temp_dir, "pyproject.toml"), "wb") as f:
|
123
|
+
tomli_w.dump(
|
124
|
+
{
|
125
|
+
"tool": {
|
126
|
+
"vellum": {
|
127
|
+
"workflows": [
|
128
|
+
{
|
129
|
+
"module": module,
|
130
|
+
"workflow_sandbox_id": str(uuid4()),
|
131
|
+
"ignore": "tests/*",
|
132
|
+
}
|
133
|
+
]
|
134
|
+
}
|
135
|
+
}
|
136
|
+
},
|
137
|
+
f,
|
138
|
+
)
|
139
|
+
|
140
|
+
# WHEN the user runs the pull command
|
141
|
+
pull_command(module)
|
142
|
+
|
143
|
+
# THEN the workflow.py file is written to the module directory
|
144
|
+
assert os.path.exists(os.path.join(temp_dir, *module.split("."), "workflow.py"))
|
145
|
+
with open(os.path.join(temp_dir, *module.split("."), "workflow.py")) as f:
|
146
|
+
assert f.read() == "print('hello')"
|
147
|
+
|
148
|
+
# AND the other_file.py file is deleted
|
149
|
+
assert not os.path.exists(other_file_path)
|
150
|
+
|
151
|
+
# AND the tests/test_workflow.py file is untouched
|
152
|
+
assert os.path.exists(test_file_path)
|
vellum_ee/__init__.py
ADDED
File without changes
|
File without changes
|
File without changes
|
@@ -0,0 +1,73 @@
|
|
1
|
+
from dataclasses import dataclass
|
2
|
+
from uuid import UUID
|
3
|
+
from typing import TypeVar
|
4
|
+
|
5
|
+
|
6
|
+
@dataclass
|
7
|
+
class WorkflowMetaDisplayOverrides:
|
8
|
+
pass
|
9
|
+
|
10
|
+
|
11
|
+
@dataclass
|
12
|
+
class WorkflowMetaDisplay(WorkflowMetaDisplayOverrides):
|
13
|
+
pass
|
14
|
+
|
15
|
+
|
16
|
+
WorkflowMetaDisplayType = TypeVar("WorkflowMetaDisplayType", bound=WorkflowMetaDisplay)
|
17
|
+
WorkflowMetaDisplayOverridesType = TypeVar("WorkflowMetaDisplayOverridesType", bound=WorkflowMetaDisplayOverrides)
|
18
|
+
|
19
|
+
|
20
|
+
@dataclass
|
21
|
+
class WorkflowInputsDisplayOverrides:
|
22
|
+
id: UUID
|
23
|
+
|
24
|
+
|
25
|
+
@dataclass
|
26
|
+
class WorkflowInputsDisplay(WorkflowInputsDisplayOverrides):
|
27
|
+
pass
|
28
|
+
|
29
|
+
|
30
|
+
WorkflowInputsDisplayType = TypeVar("WorkflowInputsDisplayType", bound=WorkflowInputsDisplay)
|
31
|
+
WorkflowInputsDisplayOverridesType = TypeVar("WorkflowInputsDisplayOverridesType", bound=WorkflowInputsDisplayOverrides)
|
32
|
+
|
33
|
+
|
34
|
+
@dataclass
|
35
|
+
class EdgeDisplayOverrides:
|
36
|
+
id: UUID
|
37
|
+
|
38
|
+
|
39
|
+
@dataclass
|
40
|
+
class EdgeDisplay(EdgeDisplayOverrides):
|
41
|
+
pass
|
42
|
+
|
43
|
+
|
44
|
+
EdgeDisplayType = TypeVar("EdgeDisplayType", bound=EdgeDisplay)
|
45
|
+
EdgeDisplayOverridesType = TypeVar("EdgeDisplayOverridesType", bound=EdgeDisplayOverrides)
|
46
|
+
|
47
|
+
|
48
|
+
@dataclass
|
49
|
+
class EntrypointDisplayOverrides:
|
50
|
+
id: UUID
|
51
|
+
|
52
|
+
|
53
|
+
@dataclass
|
54
|
+
class EntrypointDisplay(EntrypointDisplayOverrides):
|
55
|
+
pass
|
56
|
+
|
57
|
+
|
58
|
+
EntrypointDisplayType = TypeVar("EntrypointDisplayType", bound=EntrypointDisplay)
|
59
|
+
EntrypointDisplayOverridesType = TypeVar("EntrypointDisplayOverridesType", bound=EntrypointDisplayOverrides)
|
60
|
+
|
61
|
+
|
62
|
+
@dataclass
|
63
|
+
class WorkflowOutputDisplayOverrides:
|
64
|
+
id: UUID
|
65
|
+
|
66
|
+
|
67
|
+
@dataclass
|
68
|
+
class WorkflowOutputDisplay(WorkflowOutputDisplayOverrides):
|
69
|
+
pass
|
70
|
+
|
71
|
+
|
72
|
+
WorkflowOutputDisplayType = TypeVar("WorkflowOutputDisplayType", bound=WorkflowOutputDisplay)
|
73
|
+
WorkflowOutputDisplayOverridesType = TypeVar("WorkflowOutputDisplayOverridesType", bound=WorkflowOutputDisplayOverrides)
|
@@ -0,0 +1,116 @@
|
|
1
|
+
from functools import cached_property
|
2
|
+
import inspect
|
3
|
+
from uuid import UUID
|
4
|
+
from typing import TYPE_CHECKING, Any, Dict, Generic, Optional, Type, TypeVar, get_args
|
5
|
+
|
6
|
+
from vellum_ee.workflows.display.nodes.types import NodeOutputDisplay, PortDisplay, PortDisplayOverrides
|
7
|
+
from vellum_ee.workflows.display.utils.uuids import uuid4_from_hash
|
8
|
+
from vellum_ee.workflows.display.vellum import CodeResourceDefinition, NodeDefinition
|
9
|
+
from vellum.workflows.nodes.bases.base import BaseNode
|
10
|
+
from vellum.workflows.nodes.utils import get_wrapped_node, has_wrapped_node
|
11
|
+
from vellum.workflows.ports import Port
|
12
|
+
from vellum.workflows.references import OutputReference
|
13
|
+
from vellum.workflows.types.core import JsonObject
|
14
|
+
from vellum.workflows.types.generics import NodeType
|
15
|
+
from vellum.workflows.types.utils import get_original_base
|
16
|
+
from vellum.workflows.utils.names import pascal_to_title_case
|
17
|
+
|
18
|
+
if TYPE_CHECKING:
|
19
|
+
from vellum_ee.workflows.display.types import WorkflowDisplayContext
|
20
|
+
|
21
|
+
_NodeDisplayAttrType = TypeVar("_NodeDisplayAttrType")
|
22
|
+
|
23
|
+
|
24
|
+
class BaseNodeDisplay(Generic[NodeType]):
|
25
|
+
output_display: Dict[OutputReference, NodeOutputDisplay] = {}
|
26
|
+
port_displays: Dict[Port, PortDisplayOverrides] = {}
|
27
|
+
|
28
|
+
# Used to store the mapping between node types and their display classes
|
29
|
+
_node_display_registry: Dict[Type[NodeType], Type["BaseNodeDisplay"]] = {}
|
30
|
+
|
31
|
+
def __init__(self, node: Type[NodeType]):
|
32
|
+
self._node = node
|
33
|
+
|
34
|
+
def serialize(self, display_context: "WorkflowDisplayContext", **kwargs: Any) -> JsonObject:
|
35
|
+
raise NotImplementedError(f"Serialization for nodes of type {self._node.__name__} is not supported.")
|
36
|
+
|
37
|
+
def get_definition(self) -> NodeDefinition:
|
38
|
+
node = self._node
|
39
|
+
node_definition = NodeDefinition(
|
40
|
+
name=node.__name__,
|
41
|
+
module=node.__module__.split("."),
|
42
|
+
bases=[
|
43
|
+
CodeResourceDefinition(
|
44
|
+
name=base.__name__,
|
45
|
+
module=base.__module__.split("."),
|
46
|
+
)
|
47
|
+
for base in node.__bases__
|
48
|
+
],
|
49
|
+
)
|
50
|
+
return node_definition
|
51
|
+
|
52
|
+
def get_node_output_display(self, output: OutputReference) -> NodeOutputDisplay:
|
53
|
+
explicit_display = self.output_display.get(output)
|
54
|
+
if explicit_display:
|
55
|
+
return explicit_display
|
56
|
+
|
57
|
+
return NodeOutputDisplay(id=uuid4_from_hash(f"{self.node_id}|{output.name}"), name=output.name)
|
58
|
+
|
59
|
+
def get_node_port_display(self, port: Port) -> PortDisplay:
|
60
|
+
overrides = self.port_displays.get(port)
|
61
|
+
|
62
|
+
port_id: UUID
|
63
|
+
if overrides:
|
64
|
+
port_id = overrides.id
|
65
|
+
else:
|
66
|
+
port_id = uuid4_from_hash(f"{self.node_id}|ports|{port.name}")
|
67
|
+
|
68
|
+
return PortDisplay(id=port_id, node_id=self.node_id)
|
69
|
+
|
70
|
+
@classmethod
|
71
|
+
def get_from_node_display_registry(cls, node_class: Type[NodeType]) -> Type["BaseNodeDisplay"]:
|
72
|
+
return cls._node_display_registry[node_class]
|
73
|
+
|
74
|
+
@cached_property
|
75
|
+
def node_id(self) -> UUID:
|
76
|
+
"""Can be overridden as a class attribute to specify a custom node id."""
|
77
|
+
return uuid4_from_hash(self._node.__qualname__)
|
78
|
+
|
79
|
+
@cached_property
|
80
|
+
def label(self) -> str:
|
81
|
+
"""Can be overridden as a class attribute to specify a custom label."""
|
82
|
+
return pascal_to_title_case(self._node.__name__)
|
83
|
+
|
84
|
+
@classmethod
|
85
|
+
def _get_explicit_node_display_attr(
|
86
|
+
cls,
|
87
|
+
attribute: str,
|
88
|
+
attribute_type: Type[_NodeDisplayAttrType],
|
89
|
+
) -> Optional[_NodeDisplayAttrType]:
|
90
|
+
node_display_attribute: Optional[_NodeDisplayAttrType] = getattr(cls, attribute, None)
|
91
|
+
|
92
|
+
if node_display_attribute is None:
|
93
|
+
return None
|
94
|
+
|
95
|
+
if isinstance(node_display_attribute, attribute_type):
|
96
|
+
return node_display_attribute
|
97
|
+
|
98
|
+
raise ValueError(f"Node {cls.__name__} must define an explicit {attribute} of type {attribute_type.__name__}.")
|
99
|
+
|
100
|
+
def __init_subclass__(cls, **kwargs: Any) -> None:
|
101
|
+
super().__init_subclass__(**kwargs)
|
102
|
+
|
103
|
+
original_base = get_original_base(cls)
|
104
|
+
node_class = get_args(original_base)[0]
|
105
|
+
if isinstance(node_class, TypeVar):
|
106
|
+
bounded_class = node_class.__bound__
|
107
|
+
if inspect.isclass(bounded_class) and issubclass(bounded_class, BaseNode):
|
108
|
+
cls._node_display_registry[bounded_class] = cls
|
109
|
+
elif issubclass(node_class, BaseNode):
|
110
|
+
if has_wrapped_node(node_class):
|
111
|
+
wrapped_node = get_wrapped_node(node_class)
|
112
|
+
if wrapped_node._is_wrapped_node:
|
113
|
+
cls._node_display_registry[wrapped_node] = cls
|
114
|
+
return
|
115
|
+
|
116
|
+
cls._node_display_registry[node_class] = cls
|
@@ -0,0 +1,36 @@
|
|
1
|
+
from uuid import UUID
|
2
|
+
from typing import ClassVar, Dict, Optional
|
3
|
+
|
4
|
+
from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
|
5
|
+
from vellum_ee.workflows.display.nodes.types import PortDisplay
|
6
|
+
from vellum_ee.workflows.display.utils.uuids import uuid4_from_hash
|
7
|
+
from vellum_ee.workflows.display.vellum import NodeDisplayData
|
8
|
+
from vellum.workflows.ports import Port
|
9
|
+
from vellum.workflows.types.generics import NodeType
|
10
|
+
|
11
|
+
|
12
|
+
class BaseNodeVellumDisplay(BaseNodeDisplay[NodeType]):
|
13
|
+
# Used to explicitly set display data for a node
|
14
|
+
display_data: ClassVar[Optional[NodeDisplayData]] = None
|
15
|
+
|
16
|
+
# Used to explicitly set the target handle id for a node
|
17
|
+
target_handle_id: ClassVar[Optional[UUID]] = None
|
18
|
+
|
19
|
+
# Used to explicitly set the node input ids by name for a node
|
20
|
+
node_input_ids_by_name: ClassVar[Dict[str, UUID]] = {}
|
21
|
+
|
22
|
+
def _get_node_display_uuid(self, attribute: str) -> UUID:
|
23
|
+
explicit_value = self._get_explicit_node_display_attr(attribute, UUID)
|
24
|
+
return explicit_value if explicit_value else uuid4_from_hash(f"{self.node_id}|{attribute}")
|
25
|
+
|
26
|
+
def get_display_data(self) -> NodeDisplayData:
|
27
|
+
explicit_value = self._get_explicit_node_display_attr("display_data", NodeDisplayData)
|
28
|
+
return explicit_value if explicit_value else NodeDisplayData()
|
29
|
+
|
30
|
+
def get_target_handle_id(self) -> UUID:
|
31
|
+
return self._get_node_display_uuid("target_handle_id")
|
32
|
+
|
33
|
+
def get_source_handle_id(self, port_displays: Dict[Port, PortDisplay]) -> UUID:
|
34
|
+
default_port = self._node.Ports.default
|
35
|
+
default_port_display = port_displays[default_port]
|
36
|
+
return default_port_display.id
|
@@ -0,0 +1,25 @@
|
|
1
|
+
from typing import Optional, Type
|
2
|
+
|
3
|
+
from vellum_ee.workflows.display.types import NodeDisplayType
|
4
|
+
from vellum.workflows.types.generics import NodeType
|
5
|
+
|
6
|
+
|
7
|
+
def get_node_display_class(
|
8
|
+
base_class: Type[NodeDisplayType], node_class: Type[NodeType], root_node_class: Optional[Type[NodeType]] = None
|
9
|
+
) -> Type[NodeDisplayType]:
|
10
|
+
try:
|
11
|
+
node_display_class = base_class.get_from_node_display_registry(node_class)
|
12
|
+
except KeyError:
|
13
|
+
try:
|
14
|
+
return get_node_display_class(
|
15
|
+
base_class, node_class.__bases__[0], node_class if root_node_class is None else root_node_class
|
16
|
+
)
|
17
|
+
except IndexError:
|
18
|
+
return base_class
|
19
|
+
|
20
|
+
if not issubclass(node_display_class, base_class):
|
21
|
+
raise TypeError(
|
22
|
+
f"Expected to find a subclass of '{base_class.__name__}' for node class '{node_class.__name__}'"
|
23
|
+
)
|
24
|
+
|
25
|
+
return node_display_class
|
File without changes
|
@@ -0,0 +1,47 @@
|
|
1
|
+
import pytest
|
2
|
+
from uuid import UUID
|
3
|
+
|
4
|
+
from vellum_ee.workflows.display.nodes.base_node_display import BaseNodeDisplay
|
5
|
+
from vellum.workflows.nodes.bases import BaseNode
|
6
|
+
|
7
|
+
|
8
|
+
@pytest.fixture
|
9
|
+
def node_with_implicit_properties():
|
10
|
+
class MyNode(BaseNode):
|
11
|
+
pass
|
12
|
+
|
13
|
+
class MyNodeDisplay(BaseNodeDisplay[MyNode]):
|
14
|
+
pass
|
15
|
+
|
16
|
+
expected_id = UUID("ace7f746-4fe6-45c7-8207-fc8a4d0c7f6f")
|
17
|
+
|
18
|
+
return MyNode, MyNodeDisplay, expected_id
|
19
|
+
|
20
|
+
|
21
|
+
@pytest.fixture
|
22
|
+
def node_with_explicit_properties():
|
23
|
+
explicit_id = UUID("a422f67a-1d37-43f0-bdfc-1e4618c9496d")
|
24
|
+
|
25
|
+
class MyNode(BaseNode):
|
26
|
+
pass
|
27
|
+
|
28
|
+
class MyNodeDisplay(BaseNodeDisplay[MyNode]):
|
29
|
+
node_id = explicit_id
|
30
|
+
|
31
|
+
return MyNode, MyNodeDisplay, explicit_id
|
32
|
+
|
33
|
+
|
34
|
+
@pytest.fixture(
|
35
|
+
params=[
|
36
|
+
"node_with_implicit_properties",
|
37
|
+
"node_with_explicit_properties",
|
38
|
+
]
|
39
|
+
)
|
40
|
+
def node_info(request):
|
41
|
+
return request.getfixturevalue(request.param)
|
42
|
+
|
43
|
+
|
44
|
+
def test_get_id(node_info):
|
45
|
+
node, node_display, expected_id = node_info
|
46
|
+
|
47
|
+
assert node_display(node).node_id == expected_id
|
@@ -0,0 +1,18 @@
|
|
1
|
+
from dataclasses import dataclass
|
2
|
+
from uuid import UUID
|
3
|
+
|
4
|
+
|
5
|
+
@dataclass
|
6
|
+
class NodeOutputDisplay:
|
7
|
+
id: UUID
|
8
|
+
name: str
|
9
|
+
|
10
|
+
|
11
|
+
@dataclass
|
12
|
+
class PortDisplayOverrides:
|
13
|
+
id: UUID
|
14
|
+
|
15
|
+
|
16
|
+
@dataclass
|
17
|
+
class PortDisplay(PortDisplayOverrides):
|
18
|
+
node_id: UUID
|
@@ -0,0 +1,33 @@
|
|
1
|
+
import re
|
2
|
+
from typing import Optional, TypeVar, Union, overload
|
3
|
+
|
4
|
+
from vellum.workflows.descriptors.base import BaseDescriptor
|
5
|
+
from vellum.workflows.references.node import NodeReference
|
6
|
+
|
7
|
+
_T = TypeVar("_T")
|
8
|
+
|
9
|
+
|
10
|
+
@overload
|
11
|
+
def raise_if_descriptor(node_attr: BaseDescriptor[_T]) -> _T: ...
|
12
|
+
|
13
|
+
|
14
|
+
@overload
|
15
|
+
def raise_if_descriptor(node_attr: _T) -> _T: ...
|
16
|
+
|
17
|
+
|
18
|
+
def raise_if_descriptor(node_attr: Union[NodeReference[_T], _T]) -> Optional[_T]:
|
19
|
+
if not isinstance(node_attr, NodeReference):
|
20
|
+
raise AttributeError(f"Expected to find a Node descriptor, but found '{node_attr.__class__.__name__}'")
|
21
|
+
|
22
|
+
return node_attr.instance
|
23
|
+
|
24
|
+
|
25
|
+
def to_kebab_case(string: str) -> str:
|
26
|
+
"""Converts a string from Capital Case to kebab-case."""
|
27
|
+
string = re.sub(r"(\s|_|-)+", " ", string)
|
28
|
+
string = re.sub(
|
29
|
+
r"[A-Z]{2,}(?=[A-Z][a-z]+[0-9]*|\b)|[A-Z]?[a-z]+[0-9]*|[A-Z]|[0-9]+",
|
30
|
+
lambda mo: " " + mo.group(0).lower(),
|
31
|
+
string,
|
32
|
+
)
|
33
|
+
return "-".join(string.split())
|
@@ -0,0 +1,32 @@
|
|
1
|
+
from .api_node import BaseAPINodeDisplay
|
2
|
+
from .code_execution_node import BaseCodeExecutionNodeDisplay
|
3
|
+
from .conditional_node import BaseConditionalNodeDisplay
|
4
|
+
from .final_output_node import BaseFinalOutputNodeDisplay
|
5
|
+
from .guardrail_node import BaseGuardrailNodeDisplay
|
6
|
+
from .inline_prompt_node import BaseInlinePromptNodeDisplay
|
7
|
+
from .inline_subworkflow_node import BaseInlineSubworkflowNodeDisplay
|
8
|
+
from .map_node import BaseMapNodeDisplay
|
9
|
+
from .merge_node import BaseMergeNodeDisplay
|
10
|
+
from .prompt_deployment_node import BasePromptDeploymentNodeDisplay
|
11
|
+
from .search_node import BaseSearchNodeDisplay
|
12
|
+
from .subworkflow_deployment_node import BaseSubworkflowDeploymentNodeDisplay
|
13
|
+
from .templating_node import BaseTemplatingNodeDisplay
|
14
|
+
from .try_node import BaseTryNodeDisplay
|
15
|
+
|
16
|
+
# All node display classes must be imported here to be registered in BaseNodeDisplay's node display registry
|
17
|
+
__all__ = [
|
18
|
+
"BaseCodeExecutionNodeDisplay",
|
19
|
+
"BaseConditionalNodeDisplay",
|
20
|
+
"BaseGuardrailNodeDisplay",
|
21
|
+
"BaseInlinePromptNodeDisplay",
|
22
|
+
"BaseInlineSubworkflowNodeDisplay",
|
23
|
+
"BaseAPINodeDisplay",
|
24
|
+
"BaseMapNodeDisplay",
|
25
|
+
"BaseMergeNodeDisplay",
|
26
|
+
"BaseSearchNodeDisplay",
|
27
|
+
"BaseSubworkflowDeploymentNodeDisplay",
|
28
|
+
"BaseTemplatingNodeDisplay",
|
29
|
+
"BasePromptDeploymentNodeDisplay",
|
30
|
+
"BaseFinalOutputNodeDisplay",
|
31
|
+
"BaseTryNodeDisplay",
|
32
|
+
]
|