vellum-ai 0.0.20__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 (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",