vellum-ai 0.0.20__py3-none-any.whl → 0.0.22__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (82) hide show
  1. vellum/__init__.py +36 -20
  2. vellum/client.py +270 -101
  3. vellum/core/__init__.py +11 -2
  4. vellum/core/client_wrapper.py +27 -0
  5. vellum/core/remove_none_from_dict.py +11 -0
  6. vellum/resources/deployments/client.py +35 -15
  7. vellum/resources/document_indexes/client.py +64 -16
  8. vellum/resources/documents/client.py +110 -35
  9. vellum/resources/model_versions/client.py +67 -25
  10. vellum/resources/registered_prompts/client.py +80 -16
  11. vellum/resources/sandboxes/client.py +90 -25
  12. vellum/resources/test_suites/client.py +90 -25
  13. vellum/types/__init__.py +22 -4
  14. vellum/types/deployment_read.py +2 -6
  15. vellum/types/document.py +3 -7
  16. vellum/types/document_document_to_document_index.py +2 -2
  17. vellum/types/document_index_read.py +3 -7
  18. vellum/types/enriched_normalized_completion.py +5 -9
  19. vellum/types/evaluation_params.py +1 -3
  20. vellum/types/evaluation_params_request.py +1 -3
  21. vellum/types/generate_error_response.py +1 -1
  22. vellum/types/generate_request.py +3 -7
  23. vellum/types/generate_result.py +2 -6
  24. vellum/types/generate_result_data.py +1 -1
  25. vellum/types/generate_result_error.py +1 -1
  26. vellum/types/model_version_build_config.py +2 -6
  27. vellum/types/model_version_compile_prompt_response.py +1 -1
  28. vellum/types/model_version_compiled_prompt.py +2 -4
  29. vellum/types/model_version_exec_config.py +3 -3
  30. vellum/types/model_version_read.py +7 -10
  31. vellum/types/model_version_sandbox_snapshot.py +3 -5
  32. vellum/types/prompt_template_block_properties_request.py +2 -2
  33. vellum/types/prompt_template_block_request.py +1 -1
  34. vellum/types/prompt_template_input_variable.py +1 -1
  35. vellum/types/prompt_template_input_variable_request.py +1 -1
  36. vellum/types/provider_enum.py +5 -0
  37. vellum/types/register_prompt_error_response.py +1 -1
  38. vellum/types/register_prompt_prompt.py +2 -2
  39. vellum/types/register_prompt_prompt_info_request.py +1 -1
  40. vellum/types/register_prompt_response.py +5 -7
  41. vellum/types/registered_prompt_deployment.py +3 -3
  42. vellum/types/registered_prompt_model_version.py +2 -2
  43. vellum/types/registered_prompt_sandbox.py +2 -2
  44. vellum/types/registered_prompt_sandbox_snapshot.py +1 -1
  45. vellum/types/sandbox_scenario.py +2 -2
  46. vellum/types/scenario_input_request.py +1 -1
  47. vellum/types/search_error_response.py +1 -1
  48. vellum/types/search_filters_request.py +1 -1
  49. vellum/types/search_request_options_request.py +4 -6
  50. vellum/types/search_response.py +1 -1
  51. vellum/types/search_result.py +3 -3
  52. vellum/types/search_result_merging_request.py +1 -1
  53. vellum/types/search_weights_request.py +2 -2
  54. vellum/types/slim_document.py +5 -9
  55. vellum/types/submit_completion_actual_request.py +5 -15
  56. vellum/types/terminal_node_chat_history_result.py +1 -1
  57. vellum/types/terminal_node_json_result.py +1 -1
  58. vellum/types/terminal_node_result_output.py +2 -4
  59. vellum/types/terminal_node_string_result.py +1 -1
  60. vellum/types/test_suite_test_case.py +4 -8
  61. vellum/types/upload_document_response.py +1 -1
  62. vellum/types/workflow_event_error.py +26 -0
  63. vellum/types/workflow_execution_event_error_code.py +24 -0
  64. vellum/types/workflow_node_result_data.py +7 -11
  65. vellum/types/workflow_node_result_event.py +4 -3
  66. vellum/types/{workflow_node_result_event_state_enum.py → workflow_node_result_event_state.py} +5 -5
  67. vellum/types/workflow_request_chat_history_input_request.py +1 -3
  68. vellum/types/workflow_request_input_request.py +2 -6
  69. vellum/types/workflow_request_json_input_request.py +1 -3
  70. vellum/types/workflow_request_string_input_request.py +1 -3
  71. vellum/types/workflow_result_event.py +6 -3
  72. vellum/types/workflow_result_event_output_data.py +40 -0
  73. vellum/types/workflow_result_event_output_data_chat_history.py +32 -0
  74. vellum/types/workflow_result_event_output_data_json.py +31 -0
  75. vellum/types/workflow_result_event_output_data_string.py +33 -0
  76. vellum/types/workflow_stream_event.py +1 -4
  77. {vellum_ai-0.0.20.dist-info → vellum_ai-0.0.22.dist-info}/METADATA +1 -1
  78. vellum_ai-0.0.22.dist-info/RECORD +148 -0
  79. vellum/core/remove_none_from_headers.py +0 -11
  80. vellum/types/workflow_result_event_state_enum.py +0 -31
  81. vellum_ai-0.0.20.dist-info/RECORD +0 -142
  82. {vellum_ai-0.0.20.dist-info → vellum_ai-0.0.22.dist-info}/WHEEL +0 -0
vellum/__init__.py CHANGED
@@ -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",