vellum-ai 0.0.18__py3-none-any.whl → 0.0.25__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (99) hide show
  1. vellum/__init__.py +119 -16
  2. vellum/client.py +341 -77
  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/errors/__init__.py +2 -1
  7. vellum/errors/forbidden_error.py +9 -0
  8. vellum/resources/deployments/client.py +35 -15
  9. vellum/resources/document_indexes/client.py +64 -16
  10. vellum/resources/documents/client.py +110 -35
  11. vellum/resources/model_versions/client.py +67 -25
  12. vellum/resources/registered_prompts/client.py +80 -16
  13. vellum/resources/sandboxes/client.py +90 -25
  14. vellum/resources/test_suites/client.py +90 -25
  15. vellum/types/__init__.py +108 -0
  16. vellum/types/conditional_node_result.py +25 -0
  17. vellum/types/conditional_node_result_data.py +24 -0
  18. vellum/types/deployment_node_result.py +25 -0
  19. vellum/types/deployment_node_result_data.py +26 -0
  20. vellum/types/deployment_read.py +2 -6
  21. vellum/types/document.py +3 -7
  22. vellum/types/document_document_to_document_index.py +2 -2
  23. vellum/types/document_index_read.py +3 -7
  24. vellum/types/enriched_normalized_completion.py +5 -9
  25. vellum/types/evaluation_params.py +1 -3
  26. vellum/types/evaluation_params_request.py +1 -3
  27. vellum/types/execute_workflow_stream_error_response.py +24 -0
  28. vellum/types/generate_error_response.py +1 -1
  29. vellum/types/generate_request.py +3 -7
  30. vellum/types/generate_result.py +2 -6
  31. vellum/types/generate_result_data.py +1 -1
  32. vellum/types/generate_result_error.py +1 -1
  33. vellum/types/model_version_build_config.py +2 -6
  34. vellum/types/model_version_compile_prompt_response.py +1 -1
  35. vellum/types/model_version_compiled_prompt.py +2 -4
  36. vellum/types/model_version_exec_config.py +3 -3
  37. vellum/types/model_version_read.py +7 -10
  38. vellum/types/model_version_sandbox_snapshot.py +3 -5
  39. vellum/types/prompt_node_result.py +25 -0
  40. vellum/types/prompt_node_result_data.py +26 -0
  41. vellum/types/prompt_template_block_properties.py +1 -0
  42. vellum/types/prompt_template_block_properties_request.py +3 -2
  43. vellum/types/prompt_template_block_request.py +1 -1
  44. vellum/types/prompt_template_input_variable.py +1 -1
  45. vellum/types/prompt_template_input_variable_request.py +1 -1
  46. vellum/types/provider_enum.py +5 -0
  47. vellum/types/register_prompt_error_response.py +1 -1
  48. vellum/types/register_prompt_prompt.py +2 -2
  49. vellum/types/register_prompt_prompt_info_request.py +1 -1
  50. vellum/types/register_prompt_response.py +5 -7
  51. vellum/types/registered_prompt_deployment.py +3 -3
  52. vellum/types/registered_prompt_model_version.py +2 -2
  53. vellum/types/registered_prompt_sandbox.py +2 -2
  54. vellum/types/registered_prompt_sandbox_snapshot.py +1 -1
  55. vellum/types/sandbox_node_result.py +25 -0
  56. vellum/types/sandbox_node_result_data.py +26 -0
  57. vellum/types/sandbox_scenario.py +2 -2
  58. vellum/types/scenario_input_request.py +1 -1
  59. vellum/types/search_error_response.py +1 -1
  60. vellum/types/search_filters_request.py +1 -1
  61. vellum/types/search_node_result.py +25 -0
  62. vellum/types/search_node_result_data.py +27 -0
  63. vellum/types/search_request_options_request.py +4 -6
  64. vellum/types/search_response.py +1 -1
  65. vellum/types/search_result.py +3 -3
  66. vellum/types/search_result_merging_request.py +1 -1
  67. vellum/types/search_weights_request.py +2 -2
  68. vellum/types/slim_document.py +5 -9
  69. vellum/types/submit_completion_actual_request.py +5 -15
  70. vellum/types/terminal_node_chat_history_result.py +26 -0
  71. vellum/types/terminal_node_json_result.py +25 -0
  72. vellum/types/terminal_node_result.py +25 -0
  73. vellum/types/terminal_node_result_data.py +25 -0
  74. vellum/types/terminal_node_result_output.py +40 -0
  75. vellum/types/terminal_node_string_result.py +25 -0
  76. vellum/types/test_suite_test_case.py +4 -8
  77. vellum/types/upload_document_response.py +1 -1
  78. vellum/types/workflow_event_error.py +26 -0
  79. vellum/types/workflow_execution_event_error_code.py +31 -0
  80. vellum/types/workflow_execution_node_result_event.py +27 -0
  81. vellum/types/workflow_execution_workflow_result_event.py +27 -0
  82. vellum/types/workflow_node_result_data.py +72 -0
  83. vellum/types/workflow_node_result_event.py +33 -0
  84. vellum/types/workflow_node_result_event_state.py +36 -0
  85. vellum/types/workflow_request_chat_history_input_request.py +28 -0
  86. vellum/types/workflow_request_input_request.py +40 -0
  87. vellum/types/workflow_request_json_input_request.py +27 -0
  88. vellum/types/workflow_request_string_input_request.py +27 -0
  89. vellum/types/workflow_result_event.py +31 -0
  90. vellum/types/workflow_result_event_output_data.py +40 -0
  91. vellum/types/workflow_result_event_output_data_chat_history.py +32 -0
  92. vellum/types/workflow_result_event_output_data_json.py +31 -0
  93. vellum/types/workflow_result_event_output_data_string.py +33 -0
  94. vellum/types/workflow_stream_event.py +29 -0
  95. {vellum_ai-0.0.18.dist-info → vellum_ai-0.0.25.dist-info}/METADATA +1 -1
  96. vellum_ai-0.0.25.dist-info/RECORD +149 -0
  97. vellum/core/remove_none_from_headers.py +0 -11
  98. vellum_ai-0.0.18.dist-info/RECORD +0 -113
  99. {vellum_ai-0.0.18.dist-info → vellum_ai-0.0.25.dist-info}/WHEEL +0 -0
vellum/__init__.py CHANGED
@@ -1,22 +1,15 @@
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, 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,
17
6
  ChatMessageRequest,
18
7
  ChatMessageRole,
8
+ ConditionalNodeResult,
9
+ ConditionalNodeResultData,
19
10
  ContentType,
11
+ DeploymentNodeResult,
12
+ DeploymentNodeResultData,
20
13
  DeploymentRead,
21
14
  DeploymentReadStatusEnum,
22
15
  Document,
@@ -27,6 +20,7 @@ from .types import (
27
20
  EnvironmentEnum,
28
21
  EvaluationParams,
29
22
  EvaluationParamsRequest,
23
+ ExecuteWorkflowStreamErrorResponse,
30
24
  FinishReasonEnum,
31
25
  GenerateErrorResponse,
32
26
  GenerateOptionsRequest,
@@ -42,8 +36,8 @@ from .types import (
42
36
  LogprobsEnum,
43
37
  ModelTypeEnum,
44
38
  ModelVersionBuildConfig,
45
- ModelVersionCompiledPrompt,
46
39
  ModelVersionCompilePromptResponse,
40
+ ModelVersionCompiledPrompt,
47
41
  ModelVersionExecConfig,
48
42
  ModelVersionExecConfigParameters,
49
43
  ModelVersionRead,
@@ -54,6 +48,8 @@ from .types import (
54
48
  PaginatedSlimDocumentList,
55
49
  ProcessingFailureReasonEnum,
56
50
  ProcessingStateEnum,
51
+ PromptNodeResult,
52
+ PromptNodeResultData,
57
53
  PromptTemplateBlock,
58
54
  PromptTemplateBlockData,
59
55
  PromptTemplateBlockDataRequest,
@@ -63,23 +59,27 @@ from .types import (
63
59
  PromptTemplateInputVariable,
64
60
  PromptTemplateInputVariableRequest,
65
61
  ProviderEnum,
66
- RegisteredPromptDeployment,
67
- RegisteredPromptModelVersion,
68
- RegisteredPromptSandbox,
69
- RegisteredPromptSandboxSnapshot,
70
62
  RegisterPromptErrorResponse,
71
63
  RegisterPromptModelParametersRequest,
72
64
  RegisterPromptPrompt,
73
65
  RegisterPromptPromptInfoRequest,
74
66
  RegisterPromptResponse,
67
+ RegisteredPromptDeployment,
68
+ RegisteredPromptModelVersion,
69
+ RegisteredPromptSandbox,
70
+ RegisteredPromptSandboxSnapshot,
75
71
  SandboxMetricInputParams,
76
72
  SandboxMetricInputParamsRequest,
73
+ SandboxNodeResult,
74
+ SandboxNodeResultData,
77
75
  SandboxScenario,
78
76
  ScenarioInput,
79
77
  ScenarioInputRequest,
80
78
  ScenarioInputTypeEnum,
81
79
  SearchErrorResponse,
82
80
  SearchFiltersRequest,
81
+ SearchNodeResult,
82
+ SearchNodeResultData,
83
83
  SearchRequestOptionsRequest,
84
84
  SearchResponse,
85
85
  SearchResult,
@@ -89,10 +89,61 @@ from .types import (
89
89
  SlimDocumentStatusEnum,
90
90
  SubmitCompletionActualRequest,
91
91
  SubmitCompletionActualsErrorResponse,
92
+ TerminalNodeChatHistoryResult,
93
+ TerminalNodeJsonResult,
94
+ TerminalNodeResult,
95
+ TerminalNodeResultData,
96
+ TerminalNodeResultOutput,
97
+ TerminalNodeResultOutput_ChatHistory,
98
+ TerminalNodeResultOutput_Json,
99
+ TerminalNodeResultOutput_String,
100
+ TerminalNodeStringResult,
92
101
  TestSuiteTestCase,
93
102
  UploadDocumentErrorResponse,
94
103
  UploadDocumentResponse,
104
+ WorkflowEventError,
105
+ WorkflowExecutionEventErrorCode,
106
+ WorkflowExecutionNodeResultEvent,
107
+ WorkflowExecutionWorkflowResultEvent,
108
+ WorkflowNodeResultData,
109
+ WorkflowNodeResultData_Conditional,
110
+ WorkflowNodeResultData_Deployment,
111
+ WorkflowNodeResultData_Prompt,
112
+ WorkflowNodeResultData_Sandbox,
113
+ WorkflowNodeResultData_Search,
114
+ WorkflowNodeResultData_Terminal,
115
+ WorkflowNodeResultEvent,
116
+ WorkflowNodeResultEventState,
117
+ WorkflowRequestChatHistoryInputRequest,
118
+ WorkflowRequestInputRequest,
119
+ WorkflowRequestInputRequest_ChatHistory,
120
+ WorkflowRequestInputRequest_Json,
121
+ WorkflowRequestInputRequest_String,
122
+ WorkflowRequestJsonInputRequest,
123
+ WorkflowRequestStringInputRequest,
124
+ WorkflowResultEvent,
125
+ WorkflowResultEventOutputData,
126
+ WorkflowResultEventOutputDataChatHistory,
127
+ WorkflowResultEventOutputDataJson,
128
+ WorkflowResultEventOutputDataString,
129
+ WorkflowResultEventOutputData_ChatHistory,
130
+ WorkflowResultEventOutputData_Json,
131
+ WorkflowResultEventOutputData_String,
132
+ WorkflowStreamEvent,
133
+ WorkflowStreamEvent_Node,
134
+ WorkflowStreamEvent_Workflow,
95
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
96
147
 
97
148
  __all__ = [
98
149
  "BadRequestError",
@@ -100,8 +151,12 @@ __all__ = [
100
151
  "ChatMessage",
101
152
  "ChatMessageRequest",
102
153
  "ChatMessageRole",
154
+ "ConditionalNodeResult",
155
+ "ConditionalNodeResultData",
103
156
  "ConflictError",
104
157
  "ContentType",
158
+ "DeploymentNodeResult",
159
+ "DeploymentNodeResultData",
105
160
  "DeploymentRead",
106
161
  "DeploymentReadStatusEnum",
107
162
  "Document",
@@ -112,7 +167,9 @@ __all__ = [
112
167
  "EnvironmentEnum",
113
168
  "EvaluationParams",
114
169
  "EvaluationParamsRequest",
170
+ "ExecuteWorkflowStreamErrorResponse",
115
171
  "FinishReasonEnum",
172
+ "ForbiddenError",
116
173
  "GenerateErrorResponse",
117
174
  "GenerateOptionsRequest",
118
175
  "GenerateRequest",
@@ -141,6 +198,8 @@ __all__ = [
141
198
  "PaginatedSlimDocumentList",
142
199
  "ProcessingFailureReasonEnum",
143
200
  "ProcessingStateEnum",
201
+ "PromptNodeResult",
202
+ "PromptNodeResultData",
144
203
  "PromptTemplateBlock",
145
204
  "PromptTemplateBlockData",
146
205
  "PromptTemplateBlockDataRequest",
@@ -161,12 +220,16 @@ __all__ = [
161
220
  "RegisteredPromptSandboxSnapshot",
162
221
  "SandboxMetricInputParams",
163
222
  "SandboxMetricInputParamsRequest",
223
+ "SandboxNodeResult",
224
+ "SandboxNodeResultData",
164
225
  "SandboxScenario",
165
226
  "ScenarioInput",
166
227
  "ScenarioInputRequest",
167
228
  "ScenarioInputTypeEnum",
168
229
  "SearchErrorResponse",
169
230
  "SearchFiltersRequest",
231
+ "SearchNodeResult",
232
+ "SearchNodeResultData",
170
233
  "SearchRequestOptionsRequest",
171
234
  "SearchResponse",
172
235
  "SearchResult",
@@ -176,10 +239,50 @@ __all__ = [
176
239
  "SlimDocumentStatusEnum",
177
240
  "SubmitCompletionActualRequest",
178
241
  "SubmitCompletionActualsErrorResponse",
242
+ "TerminalNodeChatHistoryResult",
243
+ "TerminalNodeJsonResult",
244
+ "TerminalNodeResult",
245
+ "TerminalNodeResultData",
246
+ "TerminalNodeResultOutput",
247
+ "TerminalNodeResultOutput_ChatHistory",
248
+ "TerminalNodeResultOutput_Json",
249
+ "TerminalNodeResultOutput_String",
250
+ "TerminalNodeStringResult",
179
251
  "TestSuiteTestCase",
180
252
  "UploadDocumentErrorResponse",
181
253
  "UploadDocumentResponse",
182
254
  "VellumEnvironment",
255
+ "WorkflowEventError",
256
+ "WorkflowExecutionEventErrorCode",
257
+ "WorkflowExecutionNodeResultEvent",
258
+ "WorkflowExecutionWorkflowResultEvent",
259
+ "WorkflowNodeResultData",
260
+ "WorkflowNodeResultData_Conditional",
261
+ "WorkflowNodeResultData_Deployment",
262
+ "WorkflowNodeResultData_Prompt",
263
+ "WorkflowNodeResultData_Sandbox",
264
+ "WorkflowNodeResultData_Search",
265
+ "WorkflowNodeResultData_Terminal",
266
+ "WorkflowNodeResultEvent",
267
+ "WorkflowNodeResultEventState",
268
+ "WorkflowRequestChatHistoryInputRequest",
269
+ "WorkflowRequestInputRequest",
270
+ "WorkflowRequestInputRequest_ChatHistory",
271
+ "WorkflowRequestInputRequest_Json",
272
+ "WorkflowRequestInputRequest_String",
273
+ "WorkflowRequestJsonInputRequest",
274
+ "WorkflowRequestStringInputRequest",
275
+ "WorkflowResultEvent",
276
+ "WorkflowResultEventOutputData",
277
+ "WorkflowResultEventOutputDataChatHistory",
278
+ "WorkflowResultEventOutputDataJson",
279
+ "WorkflowResultEventOutputDataString",
280
+ "WorkflowResultEventOutputData_ChatHistory",
281
+ "WorkflowResultEventOutputData_Json",
282
+ "WorkflowResultEventOutputData_String",
283
+ "WorkflowStreamEvent",
284
+ "WorkflowStreamEvent_Node",
285
+ "WorkflowStreamEvent_Workflow",
183
286
  "deployments",
184
287
  "document_indexes",
185
288
  "documents",