vellum-ai 0.1.13__py3-none-any.whl → 0.2.1__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (86) hide show
  1. vellum/__init__.py +172 -4
  2. vellum/client.py +214 -10
  3. vellum/core/client_wrapper.py +1 -1
  4. vellum/resources/__init__.py +15 -1
  5. vellum/resources/deployments/__init__.py +3 -0
  6. vellum/resources/deployments/client.py +113 -2
  7. vellum/resources/deployments/types/__init__.py +5 -0
  8. vellum/resources/deployments/types/deployments_list_request_status.py +17 -0
  9. vellum/resources/document_indexes/client.py +55 -9
  10. vellum/resources/documents/client.py +22 -2
  11. vellum/resources/model_versions/client.py +18 -0
  12. vellum/resources/registered_prompts/client.py +104 -0
  13. vellum/resources/sandboxes/client.py +66 -4
  14. vellum/resources/test_suites/client.py +6 -6
  15. vellum/resources/workflow_deployments/__init__.py +5 -0
  16. vellum/resources/workflow_deployments/client.py +116 -0
  17. vellum/resources/workflow_deployments/types/__init__.py +5 -0
  18. vellum/resources/workflow_deployments/types/workflow_deployments_list_request_status.py +17 -0
  19. vellum/types/__init__.py +180 -4
  20. vellum/types/array_chat_message_content.py +33 -0
  21. vellum/types/array_chat_message_content_item.py +43 -0
  22. vellum/types/array_chat_message_content_item_request.py +45 -0
  23. vellum/types/array_chat_message_content_request.py +33 -0
  24. vellum/types/array_enum.py +5 -0
  25. vellum/types/array_variable_value.py +35 -0
  26. vellum/types/chat_history_enum.py +5 -0
  27. vellum/types/chat_history_input_request.py +4 -0
  28. vellum/types/chat_history_variable_value.py +29 -0
  29. vellum/types/chat_message.py +3 -1
  30. vellum/types/chat_message_content.py +53 -0
  31. vellum/types/chat_message_content_request.py +56 -0
  32. vellum/types/chat_message_request.py +3 -1
  33. vellum/types/deployment_read.py +5 -11
  34. vellum/types/document_index_read.py +2 -2
  35. vellum/types/{document_index_status.py → entity_status.py} +3 -3
  36. vellum/types/error_enum.py +5 -0
  37. vellum/types/execute_workflow_error_response.py +28 -0
  38. vellum/types/execute_workflow_response.py +32 -0
  39. vellum/types/execute_workflow_workflow_result_event.py +33 -0
  40. vellum/types/fulfilled_execute_workflow_workflow_result_event.py +35 -0
  41. vellum/types/function_call_chat_message_content.py +33 -0
  42. vellum/types/function_call_chat_message_content_request.py +33 -0
  43. vellum/types/function_call_chat_message_content_value.py +34 -0
  44. vellum/types/function_call_chat_message_content_value_request.py +34 -0
  45. vellum/types/function_call_enum.py +5 -0
  46. vellum/types/image_chat_message_content.py +33 -0
  47. vellum/types/image_chat_message_content_request.py +33 -0
  48. vellum/types/image_enum.py +5 -0
  49. vellum/types/json_enum.py +5 -0
  50. vellum/types/json_input_request.py +4 -0
  51. vellum/types/model_version_exec_config_parameters.py +1 -0
  52. vellum/types/number_enum.py +5 -0
  53. vellum/types/number_variable_value.py +28 -0
  54. vellum/types/paginated_slim_deployment_read_list.py +32 -0
  55. vellum/types/paginated_slim_workflow_deployment_list.py +32 -0
  56. vellum/types/register_prompt_model_parameters_request.py +1 -0
  57. vellum/types/rejected_execute_workflow_workflow_result_event.py +35 -0
  58. vellum/types/scenario_input_type_enum.py +2 -2
  59. vellum/types/search_results_enum.py +5 -0
  60. vellum/types/search_results_variable_value.py +29 -0
  61. vellum/types/slim_deployment_read.py +48 -0
  62. vellum/types/slim_workflow_deployment.py +57 -0
  63. vellum/types/string_chat_message_content.py +32 -0
  64. vellum/types/string_chat_message_content_request.py +32 -0
  65. vellum/types/string_enum.py +5 -0
  66. vellum/types/string_input_request.py +4 -0
  67. vellum/types/variable_value.py +102 -0
  68. vellum/types/vellum_image.py +29 -0
  69. vellum/types/vellum_image_request.py +29 -0
  70. vellum/types/vellum_variable_type.py +5 -0
  71. vellum/types/workflow_output.py +111 -0
  72. vellum/types/workflow_output_array.py +35 -0
  73. vellum/types/workflow_output_chat_history.py +35 -0
  74. vellum/types/workflow_output_error.py +35 -0
  75. vellum/types/workflow_output_function_call.py +35 -0
  76. vellum/types/workflow_output_image.py +35 -0
  77. vellum/types/workflow_output_json.py +34 -0
  78. vellum/types/workflow_output_number.py +34 -0
  79. vellum/types/workflow_output_search_results.py +35 -0
  80. vellum/types/workflow_output_string.py +34 -0
  81. vellum/types/workflow_result_event.py +2 -0
  82. {vellum_ai-0.1.13.dist-info → vellum_ai-0.2.1.dist-info}/METADATA +1 -1
  83. {vellum_ai-0.1.13.dist-info → vellum_ai-0.2.1.dist-info}/RECORD +85 -31
  84. vellum/types/deployment_status.py +0 -31
  85. {vellum_ai-0.1.13.dist-info → vellum_ai-0.2.1.dist-info}/LICENSE +0 -0
  86. {vellum_ai-0.1.13.dist-info → vellum_ai-0.2.1.dist-info}/WHEEL +0 -0
vellum/__init__.py CHANGED
@@ -3,9 +3,33 @@
3
3
  from .types import (
4
4
  ApiNodeResult,
5
5
  ApiNodeResultData,
6
+ ArrayChatMessageContent,
7
+ ArrayChatMessageContentItem,
8
+ ArrayChatMessageContentItemRequest,
9
+ ArrayChatMessageContentItemRequest_FunctionCall,
10
+ ArrayChatMessageContentItemRequest_Image,
11
+ ArrayChatMessageContentItemRequest_String,
12
+ ArrayChatMessageContentItem_FunctionCall,
13
+ ArrayChatMessageContentItem_Image,
14
+ ArrayChatMessageContentItem_String,
15
+ ArrayChatMessageContentRequest,
16
+ ArrayEnum,
17
+ ArrayVariableValue,
6
18
  BlockTypeEnum,
19
+ ChatHistoryEnum,
7
20
  ChatHistoryInputRequest,
21
+ ChatHistoryVariableValue,
8
22
  ChatMessage,
23
+ ChatMessageContent,
24
+ ChatMessageContentRequest,
25
+ ChatMessageContentRequest_Array,
26
+ ChatMessageContentRequest_FunctionCall,
27
+ ChatMessageContentRequest_Image,
28
+ ChatMessageContentRequest_String,
29
+ ChatMessageContent_Array,
30
+ ChatMessageContent_FunctionCall,
31
+ ChatMessageContent_Image,
32
+ ChatMessageContent_String,
9
33
  ChatMessageRequest,
10
34
  ChatMessageRole,
11
35
  CodeExecutionNodeChatHistoryResult,
@@ -27,14 +51,14 @@ from .types import (
27
51
  ConditionalNodeResultData,
28
52
  DeploymentProviderPayloadResponse,
29
53
  DeploymentRead,
30
- DeploymentStatus,
31
54
  DocumentDocumentToDocumentIndex,
32
55
  DocumentIndexRead,
33
- DocumentIndexStatus,
34
56
  DocumentRead,
35
57
  DocumentStatus,
36
58
  EnrichedNormalizedCompletion,
59
+ EntityStatus,
37
60
  EnvironmentEnum,
61
+ ErrorEnum,
38
62
  ErrorVariableValue,
39
63
  EvaluationParams,
40
64
  EvaluationParamsRequest,
@@ -47,14 +71,25 @@ from .types import (
47
71
  ExecutePromptResponse,
48
72
  ExecutePromptResponse_Fulfilled,
49
73
  ExecutePromptResponse_Rejected,
74
+ ExecuteWorkflowErrorResponse,
75
+ ExecuteWorkflowResponse,
50
76
  ExecuteWorkflowStreamErrorResponse,
77
+ ExecuteWorkflowWorkflowResultEvent,
78
+ ExecuteWorkflowWorkflowResultEvent_Fulfilled,
79
+ ExecuteWorkflowWorkflowResultEvent_Rejected,
51
80
  FinishReasonEnum,
52
81
  FulfilledEnum,
53
82
  FulfilledExecutePromptEvent,
54
83
  FulfilledExecutePromptResponse,
84
+ FulfilledExecuteWorkflowWorkflowResultEvent,
55
85
  FulfilledFunctionCall,
56
86
  FulfilledPromptExecutionMeta,
57
87
  FunctionCall,
88
+ FunctionCallChatMessageContent,
89
+ FunctionCallChatMessageContentRequest,
90
+ FunctionCallChatMessageContentValue,
91
+ FunctionCallChatMessageContentValueRequest,
92
+ FunctionCallEnum,
58
93
  FunctionCallVariableValue,
59
94
  FunctionCall_Fulfilled,
60
95
  FunctionCall_Rejected,
@@ -68,10 +103,14 @@ from .types import (
68
103
  GenerateStreamResponse,
69
104
  GenerateStreamResult,
70
105
  GenerateStreamResultData,
106
+ ImageChatMessageContent,
107
+ ImageChatMessageContentRequest,
108
+ ImageEnum,
71
109
  IndexingStateEnum,
72
110
  InitiatedEnum,
73
111
  InitiatedExecutePromptEvent,
74
112
  InitiatedPromptExecutionMeta,
113
+ JsonEnum,
75
114
  JsonInputRequest,
76
115
  JsonVariableValue,
77
116
  LogicalOperator,
@@ -113,7 +152,11 @@ from .types import (
113
152
  NodeInputVariableCompiledValue_String,
114
153
  NormalizedLogProbs,
115
154
  NormalizedTokenLogProbs,
155
+ NumberEnum,
156
+ NumberVariableValue,
157
+ PaginatedSlimDeploymentReadList,
116
158
  PaginatedSlimDocumentList,
159
+ PaginatedSlimWorkflowDeploymentList,
117
160
  ProcessingFailureReasonEnum,
118
161
  ProcessingStateEnum,
119
162
  PromptDeploymentExpandMetaRequestRequest,
@@ -150,6 +193,7 @@ from .types import (
150
193
  RejectedEnum,
151
194
  RejectedExecutePromptEvent,
152
195
  RejectedExecutePromptResponse,
196
+ RejectedExecuteWorkflowWorkflowResultEvent,
153
197
  RejectedFunctionCall,
154
198
  RejectedPromptExecutionMeta,
155
199
  SandboxMetricInputParams,
@@ -169,11 +213,18 @@ from .types import (
169
213
  SearchResultDocumentRequest,
170
214
  SearchResultMergingRequest,
171
215
  SearchResultRequest,
216
+ SearchResultsEnum,
217
+ SearchResultsVariableValue,
172
218
  SearchWeightsRequest,
219
+ SlimDeploymentRead,
173
220
  SlimDocument,
221
+ SlimWorkflowDeployment,
174
222
  StreamingEnum,
175
223
  StreamingExecutePromptEvent,
176
224
  StreamingPromptExecutionMeta,
225
+ StringChatMessageContent,
226
+ StringChatMessageContentRequest,
227
+ StringEnum,
177
228
  StringInputRequest,
178
229
  StringVariableValue,
179
230
  SubmitCompletionActualRequest,
@@ -228,9 +279,20 @@ from .types import (
228
279
  TestSuiteTestCase,
229
280
  UploadDocumentErrorResponse,
230
281
  UploadDocumentResponse,
282
+ VariableValue,
283
+ VariableValue_Array,
284
+ VariableValue_ChatHistory,
285
+ VariableValue_Error,
286
+ VariableValue_FunctionCall,
287
+ VariableValue_Json,
288
+ VariableValue_Number,
289
+ VariableValue_SearchResults,
290
+ VariableValue_String,
231
291
  VellumError,
232
292
  VellumErrorCodeEnum,
233
293
  VellumErrorRequest,
294
+ VellumImage,
295
+ VellumImageRequest,
234
296
  VellumVariable,
235
297
  VellumVariableType,
236
298
  WorkflowEventError,
@@ -251,6 +313,25 @@ from .types import (
251
313
  WorkflowNodeResultData_Terminal,
252
314
  WorkflowNodeResultEvent,
253
315
  WorkflowNodeResultEventState,
316
+ WorkflowOutput,
317
+ WorkflowOutputArray,
318
+ WorkflowOutputChatHistory,
319
+ WorkflowOutputError,
320
+ WorkflowOutputFunctionCall,
321
+ WorkflowOutputImage,
322
+ WorkflowOutputJson,
323
+ WorkflowOutputNumber,
324
+ WorkflowOutputSearchResults,
325
+ WorkflowOutputString,
326
+ WorkflowOutput_Array,
327
+ WorkflowOutput_ChatHistory,
328
+ WorkflowOutput_Error,
329
+ WorkflowOutput_FunctionCall,
330
+ WorkflowOutput_Image,
331
+ WorkflowOutput_Json,
332
+ WorkflowOutput_Number,
333
+ WorkflowOutput_SearchResults,
334
+ WorkflowOutput_String,
254
335
  WorkflowRequestChatHistoryInputRequest,
255
336
  WorkflowRequestInputRequest,
256
337
  WorkflowRequestInputRequest_ChatHistory,
@@ -280,6 +361,8 @@ from .types import (
280
361
  )
281
362
  from .errors import BadRequestError, ConflictError, ForbiddenError, InternalServerError, NotFoundError
282
363
  from .resources import (
364
+ DeploymentsListRequestStatus,
365
+ WorkflowDeploymentsListRequestStatus,
283
366
  deployments,
284
367
  document_indexes,
285
368
  documents,
@@ -287,16 +370,41 @@ from .resources import (
287
370
  registered_prompts,
288
371
  sandboxes,
289
372
  test_suites,
373
+ workflow_deployments,
290
374
  )
291
375
  from .environment import VellumEnvironment
292
376
 
293
377
  __all__ = [
294
378
  "ApiNodeResult",
295
379
  "ApiNodeResultData",
380
+ "ArrayChatMessageContent",
381
+ "ArrayChatMessageContentItem",
382
+ "ArrayChatMessageContentItemRequest",
383
+ "ArrayChatMessageContentItemRequest_FunctionCall",
384
+ "ArrayChatMessageContentItemRequest_Image",
385
+ "ArrayChatMessageContentItemRequest_String",
386
+ "ArrayChatMessageContentItem_FunctionCall",
387
+ "ArrayChatMessageContentItem_Image",
388
+ "ArrayChatMessageContentItem_String",
389
+ "ArrayChatMessageContentRequest",
390
+ "ArrayEnum",
391
+ "ArrayVariableValue",
296
392
  "BadRequestError",
297
393
  "BlockTypeEnum",
394
+ "ChatHistoryEnum",
298
395
  "ChatHistoryInputRequest",
396
+ "ChatHistoryVariableValue",
299
397
  "ChatMessage",
398
+ "ChatMessageContent",
399
+ "ChatMessageContentRequest",
400
+ "ChatMessageContentRequest_Array",
401
+ "ChatMessageContentRequest_FunctionCall",
402
+ "ChatMessageContentRequest_Image",
403
+ "ChatMessageContentRequest_String",
404
+ "ChatMessageContent_Array",
405
+ "ChatMessageContent_FunctionCall",
406
+ "ChatMessageContent_Image",
407
+ "ChatMessageContent_String",
300
408
  "ChatMessageRequest",
301
409
  "ChatMessageRole",
302
410
  "CodeExecutionNodeChatHistoryResult",
@@ -319,14 +427,15 @@ __all__ = [
319
427
  "ConflictError",
320
428
  "DeploymentProviderPayloadResponse",
321
429
  "DeploymentRead",
322
- "DeploymentStatus",
430
+ "DeploymentsListRequestStatus",
323
431
  "DocumentDocumentToDocumentIndex",
324
432
  "DocumentIndexRead",
325
- "DocumentIndexStatus",
326
433
  "DocumentRead",
327
434
  "DocumentStatus",
328
435
  "EnrichedNormalizedCompletion",
436
+ "EntityStatus",
329
437
  "EnvironmentEnum",
438
+ "ErrorEnum",
330
439
  "ErrorVariableValue",
331
440
  "EvaluationParams",
332
441
  "EvaluationParamsRequest",
@@ -339,15 +448,26 @@ __all__ = [
339
448
  "ExecutePromptResponse",
340
449
  "ExecutePromptResponse_Fulfilled",
341
450
  "ExecutePromptResponse_Rejected",
451
+ "ExecuteWorkflowErrorResponse",
452
+ "ExecuteWorkflowResponse",
342
453
  "ExecuteWorkflowStreamErrorResponse",
454
+ "ExecuteWorkflowWorkflowResultEvent",
455
+ "ExecuteWorkflowWorkflowResultEvent_Fulfilled",
456
+ "ExecuteWorkflowWorkflowResultEvent_Rejected",
343
457
  "FinishReasonEnum",
344
458
  "ForbiddenError",
345
459
  "FulfilledEnum",
346
460
  "FulfilledExecutePromptEvent",
347
461
  "FulfilledExecutePromptResponse",
462
+ "FulfilledExecuteWorkflowWorkflowResultEvent",
348
463
  "FulfilledFunctionCall",
349
464
  "FulfilledPromptExecutionMeta",
350
465
  "FunctionCall",
466
+ "FunctionCallChatMessageContent",
467
+ "FunctionCallChatMessageContentRequest",
468
+ "FunctionCallChatMessageContentValue",
469
+ "FunctionCallChatMessageContentValueRequest",
470
+ "FunctionCallEnum",
351
471
  "FunctionCallVariableValue",
352
472
  "FunctionCall_Fulfilled",
353
473
  "FunctionCall_Rejected",
@@ -361,11 +481,15 @@ __all__ = [
361
481
  "GenerateStreamResponse",
362
482
  "GenerateStreamResult",
363
483
  "GenerateStreamResultData",
484
+ "ImageChatMessageContent",
485
+ "ImageChatMessageContentRequest",
486
+ "ImageEnum",
364
487
  "IndexingStateEnum",
365
488
  "InitiatedEnum",
366
489
  "InitiatedExecutePromptEvent",
367
490
  "InitiatedPromptExecutionMeta",
368
491
  "InternalServerError",
492
+ "JsonEnum",
369
493
  "JsonInputRequest",
370
494
  "JsonVariableValue",
371
495
  "LogicalOperator",
@@ -408,7 +532,11 @@ __all__ = [
408
532
  "NormalizedLogProbs",
409
533
  "NormalizedTokenLogProbs",
410
534
  "NotFoundError",
535
+ "NumberEnum",
536
+ "NumberVariableValue",
537
+ "PaginatedSlimDeploymentReadList",
411
538
  "PaginatedSlimDocumentList",
539
+ "PaginatedSlimWorkflowDeploymentList",
412
540
  "ProcessingFailureReasonEnum",
413
541
  "ProcessingStateEnum",
414
542
  "PromptDeploymentExpandMetaRequestRequest",
@@ -445,6 +573,7 @@ __all__ = [
445
573
  "RejectedEnum",
446
574
  "RejectedExecutePromptEvent",
447
575
  "RejectedExecutePromptResponse",
576
+ "RejectedExecuteWorkflowWorkflowResultEvent",
448
577
  "RejectedFunctionCall",
449
578
  "RejectedPromptExecutionMeta",
450
579
  "SandboxMetricInputParams",
@@ -464,11 +593,18 @@ __all__ = [
464
593
  "SearchResultDocumentRequest",
465
594
  "SearchResultMergingRequest",
466
595
  "SearchResultRequest",
596
+ "SearchResultsEnum",
597
+ "SearchResultsVariableValue",
467
598
  "SearchWeightsRequest",
599
+ "SlimDeploymentRead",
468
600
  "SlimDocument",
601
+ "SlimWorkflowDeployment",
469
602
  "StreamingEnum",
470
603
  "StreamingExecutePromptEvent",
471
604
  "StreamingPromptExecutionMeta",
605
+ "StringChatMessageContent",
606
+ "StringChatMessageContentRequest",
607
+ "StringEnum",
472
608
  "StringInputRequest",
473
609
  "StringVariableValue",
474
610
  "SubmitCompletionActualRequest",
@@ -523,12 +659,24 @@ __all__ = [
523
659
  "TestSuiteTestCase",
524
660
  "UploadDocumentErrorResponse",
525
661
  "UploadDocumentResponse",
662
+ "VariableValue",
663
+ "VariableValue_Array",
664
+ "VariableValue_ChatHistory",
665
+ "VariableValue_Error",
666
+ "VariableValue_FunctionCall",
667
+ "VariableValue_Json",
668
+ "VariableValue_Number",
669
+ "VariableValue_SearchResults",
670
+ "VariableValue_String",
526
671
  "VellumEnvironment",
527
672
  "VellumError",
528
673
  "VellumErrorCodeEnum",
529
674
  "VellumErrorRequest",
675
+ "VellumImage",
676
+ "VellumImageRequest",
530
677
  "VellumVariable",
531
678
  "VellumVariableType",
679
+ "WorkflowDeploymentsListRequestStatus",
532
680
  "WorkflowEventError",
533
681
  "WorkflowExecutionActualChatHistoryRequest",
534
682
  "WorkflowExecutionActualJsonRequest",
@@ -547,6 +695,25 @@ __all__ = [
547
695
  "WorkflowNodeResultData_Terminal",
548
696
  "WorkflowNodeResultEvent",
549
697
  "WorkflowNodeResultEventState",
698
+ "WorkflowOutput",
699
+ "WorkflowOutputArray",
700
+ "WorkflowOutputChatHistory",
701
+ "WorkflowOutputError",
702
+ "WorkflowOutputFunctionCall",
703
+ "WorkflowOutputImage",
704
+ "WorkflowOutputJson",
705
+ "WorkflowOutputNumber",
706
+ "WorkflowOutputSearchResults",
707
+ "WorkflowOutputString",
708
+ "WorkflowOutput_Array",
709
+ "WorkflowOutput_ChatHistory",
710
+ "WorkflowOutput_Error",
711
+ "WorkflowOutput_FunctionCall",
712
+ "WorkflowOutput_Image",
713
+ "WorkflowOutput_Json",
714
+ "WorkflowOutput_Number",
715
+ "WorkflowOutput_SearchResults",
716
+ "WorkflowOutput_String",
550
717
  "WorkflowRequestChatHistoryInputRequest",
551
718
  "WorkflowRequestInputRequest",
552
719
  "WorkflowRequestInputRequest_ChatHistory",
@@ -580,4 +747,5 @@ __all__ = [
580
747
  "registered_prompts",
581
748
  "sandboxes",
582
749
  "test_suites",
750
+ "workflow_deployments",
583
751
  ]