vellum-ai 0.0.20__py3-none-any.whl → 0.0.25__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 (84) hide show
  1. vellum/__init__.py +38 -20
  2. vellum/client.py +278 -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 +24 -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/execute_workflow_stream_error_response.py +24 -0
  22. vellum/types/generate_error_response.py +1 -1
  23. vellum/types/generate_request.py +3 -7
  24. vellum/types/generate_result.py +2 -6
  25. vellum/types/generate_result_data.py +1 -1
  26. vellum/types/generate_result_error.py +1 -1
  27. vellum/types/model_version_build_config.py +2 -6
  28. vellum/types/model_version_compile_prompt_response.py +1 -1
  29. vellum/types/model_version_compiled_prompt.py +2 -4
  30. vellum/types/model_version_exec_config.py +3 -3
  31. vellum/types/model_version_read.py +7 -10
  32. vellum/types/model_version_sandbox_snapshot.py +3 -5
  33. vellum/types/prompt_template_block_properties.py +1 -0
  34. vellum/types/prompt_template_block_properties_request.py +3 -2
  35. vellum/types/prompt_template_block_request.py +1 -1
  36. vellum/types/prompt_template_input_variable.py +1 -1
  37. vellum/types/prompt_template_input_variable_request.py +1 -1
  38. vellum/types/provider_enum.py +5 -0
  39. vellum/types/register_prompt_error_response.py +1 -1
  40. vellum/types/register_prompt_prompt.py +2 -2
  41. vellum/types/register_prompt_prompt_info_request.py +1 -1
  42. vellum/types/register_prompt_response.py +5 -7
  43. vellum/types/registered_prompt_deployment.py +3 -3
  44. vellum/types/registered_prompt_model_version.py +2 -2
  45. vellum/types/registered_prompt_sandbox.py +2 -2
  46. vellum/types/registered_prompt_sandbox_snapshot.py +1 -1
  47. vellum/types/sandbox_scenario.py +2 -2
  48. vellum/types/scenario_input_request.py +1 -1
  49. vellum/types/search_error_response.py +1 -1
  50. vellum/types/search_filters_request.py +1 -1
  51. vellum/types/search_request_options_request.py +4 -6
  52. vellum/types/search_response.py +1 -1
  53. vellum/types/search_result.py +3 -3
  54. vellum/types/search_result_merging_request.py +1 -1
  55. vellum/types/search_weights_request.py +2 -2
  56. vellum/types/slim_document.py +5 -9
  57. vellum/types/submit_completion_actual_request.py +5 -15
  58. vellum/types/terminal_node_chat_history_result.py +1 -1
  59. vellum/types/terminal_node_json_result.py +1 -1
  60. vellum/types/terminal_node_result_output.py +2 -4
  61. vellum/types/terminal_node_string_result.py +1 -1
  62. vellum/types/test_suite_test_case.py +4 -8
  63. vellum/types/upload_document_response.py +1 -1
  64. vellum/types/workflow_event_error.py +26 -0
  65. vellum/types/workflow_execution_event_error_code.py +31 -0
  66. vellum/types/workflow_node_result_data.py +7 -11
  67. vellum/types/workflow_node_result_event.py +4 -3
  68. vellum/types/{workflow_node_result_event_state_enum.py → workflow_node_result_event_state.py} +5 -5
  69. vellum/types/workflow_request_chat_history_input_request.py +1 -3
  70. vellum/types/workflow_request_input_request.py +2 -6
  71. vellum/types/workflow_request_json_input_request.py +1 -3
  72. vellum/types/workflow_request_string_input_request.py +1 -3
  73. vellum/types/workflow_result_event.py +6 -3
  74. vellum/types/workflow_result_event_output_data.py +40 -0
  75. vellum/types/workflow_result_event_output_data_chat_history.py +32 -0
  76. vellum/types/workflow_result_event_output_data_json.py +31 -0
  77. vellum/types/workflow_result_event_output_data_string.py +33 -0
  78. vellum/types/workflow_stream_event.py +1 -4
  79. {vellum_ai-0.0.20.dist-info → vellum_ai-0.0.25.dist-info}/METADATA +1 -1
  80. vellum_ai-0.0.25.dist-info/RECORD +149 -0
  81. vellum/core/remove_none_from_headers.py +0 -11
  82. vellum/types/workflow_result_event_state_enum.py +0 -31
  83. vellum_ai-0.0.20.dist-info/RECORD +0 -142
  84. {vellum_ai-0.0.20.dist-info → vellum_ai-0.0.25.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,
@@ -31,6 +20,7 @@ from .types import (
31
20
  EnvironmentEnum,
32
21
  EvaluationParams,
33
22
  EvaluationParamsRequest,
23
+ ExecuteWorkflowStreamErrorResponse,
34
24
  FinishReasonEnum,
35
25
  GenerateErrorResponse,
36
26
  GenerateOptionsRequest,
@@ -46,8 +36,8 @@ from .types import (
46
36
  LogprobsEnum,
47
37
  ModelTypeEnum,
48
38
  ModelVersionBuildConfig,
49
- ModelVersionCompiledPrompt,
50
39
  ModelVersionCompilePromptResponse,
40
+ ModelVersionCompiledPrompt,
51
41
  ModelVersionExecConfig,
52
42
  ModelVersionExecConfigParameters,
53
43
  ModelVersionRead,
@@ -69,15 +59,15 @@ from .types import (
69
59
  PromptTemplateInputVariable,
70
60
  PromptTemplateInputVariableRequest,
71
61
  ProviderEnum,
72
- RegisteredPromptDeployment,
73
- RegisteredPromptModelVersion,
74
- RegisteredPromptSandbox,
75
- RegisteredPromptSandboxSnapshot,
76
62
  RegisterPromptErrorResponse,
77
63
  RegisterPromptModelParametersRequest,
78
64
  RegisterPromptPrompt,
79
65
  RegisterPromptPromptInfoRequest,
80
66
  RegisterPromptResponse,
67
+ RegisteredPromptDeployment,
68
+ RegisteredPromptModelVersion,
69
+ RegisteredPromptSandbox,
70
+ RegisteredPromptSandboxSnapshot,
81
71
  SandboxMetricInputParams,
82
72
  SandboxMetricInputParamsRequest,
83
73
  SandboxNodeResult,
@@ -111,6 +101,8 @@ from .types import (
111
101
  TestSuiteTestCase,
112
102
  UploadDocumentErrorResponse,
113
103
  UploadDocumentResponse,
104
+ WorkflowEventError,
105
+ WorkflowExecutionEventErrorCode,
114
106
  WorkflowExecutionNodeResultEvent,
115
107
  WorkflowExecutionWorkflowResultEvent,
116
108
  WorkflowNodeResultData,
@@ -121,7 +113,7 @@ from .types import (
121
113
  WorkflowNodeResultData_Search,
122
114
  WorkflowNodeResultData_Terminal,
123
115
  WorkflowNodeResultEvent,
124
- WorkflowNodeResultEventStateEnum,
116
+ WorkflowNodeResultEventState,
125
117
  WorkflowRequestChatHistoryInputRequest,
126
118
  WorkflowRequestInputRequest,
127
119
  WorkflowRequestInputRequest_ChatHistory,
@@ -130,11 +122,28 @@ from .types import (
130
122
  WorkflowRequestJsonInputRequest,
131
123
  WorkflowRequestStringInputRequest,
132
124
  WorkflowResultEvent,
133
- WorkflowResultEventStateEnum,
125
+ WorkflowResultEventOutputData,
126
+ WorkflowResultEventOutputDataChatHistory,
127
+ WorkflowResultEventOutputDataJson,
128
+ WorkflowResultEventOutputDataString,
129
+ WorkflowResultEventOutputData_ChatHistory,
130
+ WorkflowResultEventOutputData_Json,
131
+ WorkflowResultEventOutputData_String,
134
132
  WorkflowStreamEvent,
135
133
  WorkflowStreamEvent_Node,
136
134
  WorkflowStreamEvent_Workflow,
137
135
  )
136
+ from .errors import BadRequestError, ConflictError, ForbiddenError, InternalServerError, NotFoundError
137
+ from .resources import (
138
+ deployments,
139
+ document_indexes,
140
+ documents,
141
+ model_versions,
142
+ registered_prompts,
143
+ sandboxes,
144
+ test_suites,
145
+ )
146
+ from .environment import VellumEnvironment
138
147
 
139
148
  __all__ = [
140
149
  "BadRequestError",
@@ -158,6 +167,7 @@ __all__ = [
158
167
  "EnvironmentEnum",
159
168
  "EvaluationParams",
160
169
  "EvaluationParamsRequest",
170
+ "ExecuteWorkflowStreamErrorResponse",
161
171
  "FinishReasonEnum",
162
172
  "ForbiddenError",
163
173
  "GenerateErrorResponse",
@@ -242,6 +252,8 @@ __all__ = [
242
252
  "UploadDocumentErrorResponse",
243
253
  "UploadDocumentResponse",
244
254
  "VellumEnvironment",
255
+ "WorkflowEventError",
256
+ "WorkflowExecutionEventErrorCode",
245
257
  "WorkflowExecutionNodeResultEvent",
246
258
  "WorkflowExecutionWorkflowResultEvent",
247
259
  "WorkflowNodeResultData",
@@ -252,7 +264,7 @@ __all__ = [
252
264
  "WorkflowNodeResultData_Search",
253
265
  "WorkflowNodeResultData_Terminal",
254
266
  "WorkflowNodeResultEvent",
255
- "WorkflowNodeResultEventStateEnum",
267
+ "WorkflowNodeResultEventState",
256
268
  "WorkflowRequestChatHistoryInputRequest",
257
269
  "WorkflowRequestInputRequest",
258
270
  "WorkflowRequestInputRequest_ChatHistory",
@@ -261,7 +273,13 @@ __all__ = [
261
273
  "WorkflowRequestJsonInputRequest",
262
274
  "WorkflowRequestStringInputRequest",
263
275
  "WorkflowResultEvent",
264
- "WorkflowResultEventStateEnum",
276
+ "WorkflowResultEventOutputData",
277
+ "WorkflowResultEventOutputDataChatHistory",
278
+ "WorkflowResultEventOutputDataJson",
279
+ "WorkflowResultEventOutputDataString",
280
+ "WorkflowResultEventOutputData_ChatHistory",
281
+ "WorkflowResultEventOutputData_Json",
282
+ "WorkflowResultEventOutputData_String",
265
283
  "WorkflowStreamEvent",
266
284
  "WorkflowStreamEvent_Node",
267
285
  "WorkflowStreamEvent_Workflow",