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

Sign up to get free protection for your applications and to get access to all the features.
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",