vellum-ai 0.0.20__tar.gz → 0.0.22__tar.gz

Sign up to get free protection for your applications and to get access to all the features.
Files changed (155) hide show
  1. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/PKG-INFO +1 -1
  2. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/pyproject.toml +2 -1
  3. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/__init__.py +36 -20
  4. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/client.py +270 -101
  5. vellum_ai-0.0.22/src/vellum/core/__init__.py +17 -0
  6. vellum_ai-0.0.22/src/vellum/core/client_wrapper.py +27 -0
  7. vellum_ai-0.0.22/src/vellum/core/remove_none_from_dict.py +11 -0
  8. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/deployments/client.py +35 -15
  9. vellum_ai-0.0.22/src/vellum/resources/document_indexes/client.py +144 -0
  10. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/documents/client.py +110 -35
  11. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/model_versions/client.py +67 -25
  12. vellum_ai-0.0.22/src/vellum/resources/registered_prompts/client.py +175 -0
  13. vellum_ai-0.0.22/src/vellum/resources/sandboxes/client.py +190 -0
  14. vellum_ai-0.0.22/src/vellum/resources/test_suites/client.py +185 -0
  15. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/__init__.py +22 -4
  16. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/deployment_read.py +2 -6
  17. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/document.py +3 -7
  18. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/document_document_to_document_index.py +2 -2
  19. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/document_index_read.py +3 -7
  20. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/enriched_normalized_completion.py +5 -9
  21. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/evaluation_params.py +1 -3
  22. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/evaluation_params_request.py +1 -3
  23. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_error_response.py +1 -1
  24. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_request.py +3 -7
  25. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_result.py +2 -6
  26. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_result_data.py +1 -1
  27. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_result_error.py +1 -1
  28. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_build_config.py +2 -6
  29. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_compile_prompt_response.py +1 -1
  30. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_compiled_prompt.py +2 -4
  31. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_exec_config.py +3 -3
  32. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_read.py +7 -10
  33. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_sandbox_snapshot.py +3 -5
  34. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_block_properties_request.py +2 -2
  35. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_block_request.py +1 -1
  36. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_input_variable.py +1 -1
  37. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_input_variable_request.py +1 -1
  38. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/provider_enum.py +5 -0
  39. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/register_prompt_error_response.py +1 -1
  40. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/register_prompt_prompt.py +2 -2
  41. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/register_prompt_prompt_info_request.py +1 -1
  42. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/register_prompt_response.py +5 -7
  43. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/registered_prompt_deployment.py +3 -3
  44. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/registered_prompt_model_version.py +2 -2
  45. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/registered_prompt_sandbox.py +2 -2
  46. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/registered_prompt_sandbox_snapshot.py +1 -1
  47. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/sandbox_scenario.py +2 -2
  48. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/scenario_input_request.py +1 -1
  49. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_error_response.py +1 -1
  50. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_filters_request.py +1 -1
  51. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_request_options_request.py +4 -6
  52. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_response.py +1 -1
  53. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_result.py +3 -3
  54. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_result_merging_request.py +1 -1
  55. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_weights_request.py +2 -2
  56. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/slim_document.py +5 -9
  57. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/submit_completion_actual_request.py +5 -15
  58. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/terminal_node_chat_history_result.py +1 -1
  59. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/terminal_node_json_result.py +1 -1
  60. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/terminal_node_result_output.py +2 -4
  61. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/terminal_node_string_result.py +1 -1
  62. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/test_suite_test_case.py +4 -8
  63. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/upload_document_response.py +1 -1
  64. vellum_ai-0.0.22/src/vellum/types/workflow_event_error.py +26 -0
  65. vellum_ai-0.0.22/src/vellum/types/workflow_execution_event_error_code.py +24 -0
  66. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_node_result_data.py +7 -11
  67. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_node_result_event.py +4 -3
  68. vellum_ai-0.0.20/src/vellum/types/workflow_node_result_event_state_enum.py → vellum_ai-0.0.22/src/vellum/types/workflow_node_result_event_state.py +5 -5
  69. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_request_chat_history_input_request.py +1 -3
  70. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_request_input_request.py +2 -6
  71. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_request_json_input_request.py +1 -3
  72. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_request_string_input_request.py +1 -3
  73. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_result_event.py +6 -3
  74. vellum_ai-0.0.22/src/vellum/types/workflow_result_event_output_data.py +40 -0
  75. vellum_ai-0.0.22/src/vellum/types/workflow_result_event_output_data_chat_history.py +32 -0
  76. vellum_ai-0.0.22/src/vellum/types/workflow_result_event_output_data_json.py +31 -0
  77. vellum_ai-0.0.22/src/vellum/types/workflow_result_event_output_data_string.py +33 -0
  78. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_stream_event.py +1 -4
  79. vellum_ai-0.0.20/src/vellum/core/__init__.py +0 -8
  80. vellum_ai-0.0.20/src/vellum/core/remove_none_from_headers.py +0 -11
  81. vellum_ai-0.0.20/src/vellum/resources/document_indexes/client.py +0 -96
  82. vellum_ai-0.0.20/src/vellum/resources/registered_prompts/client.py +0 -111
  83. vellum_ai-0.0.20/src/vellum/resources/sandboxes/client.py +0 -125
  84. vellum_ai-0.0.20/src/vellum/resources/test_suites/client.py +0 -120
  85. vellum_ai-0.0.20/src/vellum/types/workflow_result_event_state_enum.py +0 -31
  86. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/README.md +0 -0
  87. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/core/api_error.py +0 -0
  88. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/core/datetime_utils.py +0 -0
  89. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/core/jsonable_encoder.py +0 -0
  90. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/environment.py +0 -0
  91. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/errors/__init__.py +0 -0
  92. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/errors/bad_request_error.py +0 -0
  93. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/errors/conflict_error.py +0 -0
  94. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/errors/forbidden_error.py +0 -0
  95. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/errors/internal_server_error.py +0 -0
  96. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/errors/not_found_error.py +0 -0
  97. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/py.typed +0 -0
  98. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/__init__.py +0 -0
  99. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/deployments/__init__.py +0 -0
  100. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/document_indexes/__init__.py +0 -0
  101. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/documents/__init__.py +0 -0
  102. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/model_versions/__init__.py +0 -0
  103. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/registered_prompts/__init__.py +0 -0
  104. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/sandboxes/__init__.py +0 -0
  105. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/resources/test_suites/__init__.py +0 -0
  106. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/block_type_enum.py +0 -0
  107. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/chat_message.py +0 -0
  108. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/chat_message_request.py +0 -0
  109. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/chat_message_role.py +0 -0
  110. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/conditional_node_result.py +0 -0
  111. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/conditional_node_result_data.py +0 -0
  112. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/content_type.py +0 -0
  113. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/deployment_node_result.py +0 -0
  114. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/deployment_node_result_data.py +0 -0
  115. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/deployment_read_status_enum.py +0 -0
  116. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/document_index_status.py +0 -0
  117. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/environment_enum.py +0 -0
  118. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/finish_reason_enum.py +0 -0
  119. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_options_request.py +0 -0
  120. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_response.py +0 -0
  121. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_stream_response.py +0 -0
  122. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_stream_result.py +0 -0
  123. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/generate_stream_result_data.py +0 -0
  124. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/indexing_state_enum.py +0 -0
  125. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/logprobs_enum.py +0 -0
  126. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_type_enum.py +0 -0
  127. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_exec_config_parameters.py +0 -0
  128. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/model_version_read_status_enum.py +0 -0
  129. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/normalized_log_probs.py +0 -0
  130. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/normalized_token_log_probs.py +0 -0
  131. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/paginated_slim_document_list.py +0 -0
  132. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/processing_failure_reason_enum.py +0 -0
  133. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/processing_state_enum.py +0 -0
  134. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_node_result.py +0 -0
  135. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_node_result_data.py +0 -0
  136. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_block.py +0 -0
  137. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_block_data.py +0 -0
  138. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_block_data_request.py +0 -0
  139. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/prompt_template_block_properties.py +0 -0
  140. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/register_prompt_model_parameters_request.py +0 -0
  141. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/sandbox_metric_input_params.py +0 -0
  142. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/sandbox_metric_input_params_request.py +0 -0
  143. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/sandbox_node_result.py +0 -0
  144. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/sandbox_node_result_data.py +0 -0
  145. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/scenario_input.py +0 -0
  146. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/scenario_input_type_enum.py +0 -0
  147. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_node_result.py +0 -0
  148. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/search_node_result_data.py +0 -0
  149. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/slim_document_status_enum.py +0 -0
  150. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/submit_completion_actuals_error_response.py +0 -0
  151. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/terminal_node_result.py +0 -0
  152. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/terminal_node_result_data.py +0 -0
  153. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/upload_document_error_response.py +0 -0
  154. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_execution_node_result_event.py +0 -0
  155. {vellum_ai-0.0.20 → vellum_ai-0.0.22}/src/vellum/types/workflow_execution_workflow_result_event.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: vellum-ai
3
- Version: 0.0.20
3
+ Version: 0.0.22
4
4
  Summary:
5
5
  Requires-Python: >=3.7,<4.0
6
6
  Classifier: Programming Language :: Python :: 3
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "vellum-ai"
3
- version = "v0.0.20"
3
+ version = "v0.0.22"
4
4
  description = ""
5
5
  readme = "README.md"
6
6
  authors = []
@@ -15,6 +15,7 @@ httpx = "0.23.3"
15
15
 
16
16
  [tool.poetry.dev-dependencies]
17
17
  mypy = "0.971"
18
+ pytest = "^7.4.0"
18
19
 
19
20
  [build-system]
20
21
  requires = ["poetry-core"]
@@ -1,16 +1,5 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
- from .environment import VellumEnvironment
4
- from .errors import BadRequestError, ConflictError, ForbiddenError, InternalServerError, NotFoundError
5
- from .resources import (
6
- deployments,
7
- document_indexes,
8
- documents,
9
- model_versions,
10
- registered_prompts,
11
- sandboxes,
12
- test_suites,
13
- )
14
3
  from .types import (
15
4
  BlockTypeEnum,
16
5
  ChatMessage,
@@ -46,8 +35,8 @@ from .types import (
46
35
  LogprobsEnum,
47
36
  ModelTypeEnum,
48
37
  ModelVersionBuildConfig,
49
- ModelVersionCompiledPrompt,
50
38
  ModelVersionCompilePromptResponse,
39
+ ModelVersionCompiledPrompt,
51
40
  ModelVersionExecConfig,
52
41
  ModelVersionExecConfigParameters,
53
42
  ModelVersionRead,
@@ -69,15 +58,15 @@ from .types import (
69
58
  PromptTemplateInputVariable,
70
59
  PromptTemplateInputVariableRequest,
71
60
  ProviderEnum,
72
- RegisteredPromptDeployment,
73
- RegisteredPromptModelVersion,
74
- RegisteredPromptSandbox,
75
- RegisteredPromptSandboxSnapshot,
76
61
  RegisterPromptErrorResponse,
77
62
  RegisterPromptModelParametersRequest,
78
63
  RegisterPromptPrompt,
79
64
  RegisterPromptPromptInfoRequest,
80
65
  RegisterPromptResponse,
66
+ RegisteredPromptDeployment,
67
+ RegisteredPromptModelVersion,
68
+ RegisteredPromptSandbox,
69
+ RegisteredPromptSandboxSnapshot,
81
70
  SandboxMetricInputParams,
82
71
  SandboxMetricInputParamsRequest,
83
72
  SandboxNodeResult,
@@ -111,6 +100,8 @@ from .types import (
111
100
  TestSuiteTestCase,
112
101
  UploadDocumentErrorResponse,
113
102
  UploadDocumentResponse,
103
+ WorkflowEventError,
104
+ WorkflowExecutionEventErrorCode,
114
105
  WorkflowExecutionNodeResultEvent,
115
106
  WorkflowExecutionWorkflowResultEvent,
116
107
  WorkflowNodeResultData,
@@ -121,7 +112,7 @@ from .types import (
121
112
  WorkflowNodeResultData_Search,
122
113
  WorkflowNodeResultData_Terminal,
123
114
  WorkflowNodeResultEvent,
124
- WorkflowNodeResultEventStateEnum,
115
+ WorkflowNodeResultEventState,
125
116
  WorkflowRequestChatHistoryInputRequest,
126
117
  WorkflowRequestInputRequest,
127
118
  WorkflowRequestInputRequest_ChatHistory,
@@ -130,11 +121,28 @@ from .types import (
130
121
  WorkflowRequestJsonInputRequest,
131
122
  WorkflowRequestStringInputRequest,
132
123
  WorkflowResultEvent,
133
- WorkflowResultEventStateEnum,
124
+ WorkflowResultEventOutputData,
125
+ WorkflowResultEventOutputDataChatHistory,
126
+ WorkflowResultEventOutputDataJson,
127
+ WorkflowResultEventOutputDataString,
128
+ WorkflowResultEventOutputData_ChatHistory,
129
+ WorkflowResultEventOutputData_Json,
130
+ WorkflowResultEventOutputData_String,
134
131
  WorkflowStreamEvent,
135
132
  WorkflowStreamEvent_Node,
136
133
  WorkflowStreamEvent_Workflow,
137
134
  )
135
+ from .errors import BadRequestError, ConflictError, ForbiddenError, InternalServerError, NotFoundError
136
+ from .resources import (
137
+ deployments,
138
+ document_indexes,
139
+ documents,
140
+ model_versions,
141
+ registered_prompts,
142
+ sandboxes,
143
+ test_suites,
144
+ )
145
+ from .environment import VellumEnvironment
138
146
 
139
147
  __all__ = [
140
148
  "BadRequestError",
@@ -242,6 +250,8 @@ __all__ = [
242
250
  "UploadDocumentErrorResponse",
243
251
  "UploadDocumentResponse",
244
252
  "VellumEnvironment",
253
+ "WorkflowEventError",
254
+ "WorkflowExecutionEventErrorCode",
245
255
  "WorkflowExecutionNodeResultEvent",
246
256
  "WorkflowExecutionWorkflowResultEvent",
247
257
  "WorkflowNodeResultData",
@@ -252,7 +262,7 @@ __all__ = [
252
262
  "WorkflowNodeResultData_Search",
253
263
  "WorkflowNodeResultData_Terminal",
254
264
  "WorkflowNodeResultEvent",
255
- "WorkflowNodeResultEventStateEnum",
265
+ "WorkflowNodeResultEventState",
256
266
  "WorkflowRequestChatHistoryInputRequest",
257
267
  "WorkflowRequestInputRequest",
258
268
  "WorkflowRequestInputRequest_ChatHistory",
@@ -261,7 +271,13 @@ __all__ = [
261
271
  "WorkflowRequestJsonInputRequest",
262
272
  "WorkflowRequestStringInputRequest",
263
273
  "WorkflowResultEvent",
264
- "WorkflowResultEventStateEnum",
274
+ "WorkflowResultEventOutputData",
275
+ "WorkflowResultEventOutputDataChatHistory",
276
+ "WorkflowResultEventOutputDataJson",
277
+ "WorkflowResultEventOutputDataString",
278
+ "WorkflowResultEventOutputData_ChatHistory",
279
+ "WorkflowResultEventOutputData_Json",
280
+ "WorkflowResultEventOutputData_String",
265
281
  "WorkflowStreamEvent",
266
282
  "WorkflowStreamEvent_Node",
267
283
  "WorkflowStreamEvent_Workflow",