vellum-ai 0.5.2__py3-none-any.whl → 0.6.1__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (81) hide show
  1. vellum/__init__.py +166 -18
  2. vellum/core/client_wrapper.py +1 -1
  3. vellum/resources/document_indexes/client.py +61 -52
  4. vellum/resources/documents/client.py +8 -4
  5. vellum/types/__init__.py +185 -17
  6. vellum/types/add_openai_api_key_enum.py +3 -0
  7. vellum/types/array_variable_value_item.py +0 -24
  8. vellum/types/array_vellum_value_item.py +82 -0
  9. vellum/types/{rejected_function_call.py → basic_vectorizer_intfloat_multilingual_e_5_large.py} +3 -6
  10. vellum/types/basic_vectorizer_intfloat_multilingual_e_5_large_request.py +29 -0
  11. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1.py +29 -0
  12. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request.py +29 -0
  13. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1.py +29 -0
  14. vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request.py +29 -0
  15. vellum/types/document_index_chunking.py +46 -0
  16. vellum/types/document_index_chunking_request.py +46 -0
  17. vellum/types/document_index_indexing_config.py +28 -0
  18. vellum/types/document_index_indexing_config_request.py +28 -0
  19. vellum/types/document_index_read.py +2 -4
  20. vellum/types/function_call.py +20 -18
  21. vellum/types/{fulfilled_function_call_request.py → function_call_request.py} +2 -2
  22. vellum/types/function_call_vellum_value.py +1 -1
  23. vellum/types/hkunlp_instructor_xl_enum.py +5 -0
  24. vellum/types/hkunlp_instructor_xl_vectorizer.py +30 -0
  25. vellum/types/hkunlp_instructor_xl_vectorizer_request.py +30 -0
  26. vellum/types/{fulfilled_function_call.py → image_vellum_value.py} +4 -5
  27. vellum/types/indexing_config_vectorizer.py +106 -0
  28. vellum/types/indexing_config_vectorizer_request.py +106 -0
  29. vellum/types/instructor_vectorizer_config.py +31 -0
  30. vellum/types/instructor_vectorizer_config_request.py +31 -0
  31. vellum/types/intfloat_multilingual_e_5_large_enum.py +5 -0
  32. vellum/types/merge_enum.py +5 -0
  33. vellum/types/{chat_history_variable_value.py → merge_node_result.py} +4 -3
  34. vellum/types/metric_enum.py +5 -0
  35. vellum/types/{search_results_variable_value.py → metric_node_result.py} +4 -3
  36. vellum/types/named_test_case_function_call_variable_value.py +2 -2
  37. vellum/types/named_test_case_function_call_variable_value_request.py +2 -2
  38. vellum/types/node_output_compiled_array_value.py +2 -2
  39. vellum/types/number_vellum_value.py +29 -0
  40. vellum/types/open_ai_vectorizer_config.py +30 -0
  41. vellum/types/open_ai_vectorizer_config_request.py +30 -0
  42. vellum/types/open_ai_vectorizer_text_embedding_3_large.py +30 -0
  43. vellum/types/open_ai_vectorizer_text_embedding_3_large_request.py +30 -0
  44. vellum/types/open_ai_vectorizer_text_embedding_3_small.py +30 -0
  45. vellum/types/open_ai_vectorizer_text_embedding_3_small_request.py +30 -0
  46. vellum/types/open_ai_vectorizer_text_embedding_ada_002.py +30 -0
  47. vellum/types/open_ai_vectorizer_text_embedding_ada_002_request.py +30 -0
  48. vellum/types/reducto_chunker_config.py +29 -0
  49. vellum/types/reducto_chunker_config_request.py +29 -0
  50. vellum/types/reducto_chunker_enum.py +5 -0
  51. vellum/types/reducto_chunking.py +30 -0
  52. vellum/types/reducto_chunking_request.py +30 -0
  53. vellum/types/search_result_document.py +1 -1
  54. vellum/types/search_result_document_request.py +5 -0
  55. vellum/types/sentence_chunker_config.py +30 -0
  56. vellum/types/sentence_chunker_config_request.py +30 -0
  57. vellum/types/sentence_chunker_enum.py +5 -0
  58. vellum/types/sentence_chunking.py +30 -0
  59. vellum/types/sentence_chunking_request.py +30 -0
  60. vellum/types/sentence_transformers_multi_qa_mpnet_base_cos_v_1_enum.py +5 -0
  61. vellum/types/sentence_transformers_multi_qa_mpnet_base_dot_v_1_enum.py +5 -0
  62. vellum/types/submit_completion_actual_request.py +5 -0
  63. vellum/types/test_case_function_call_variable_value.py +2 -2
  64. vellum/types/test_suite_run_execution_function_call_output.py +2 -2
  65. vellum/types/text_embedding_3_large_enum.py +5 -0
  66. vellum/types/text_embedding_3_small_enum.py +5 -0
  67. vellum/types/text_embedding_ada_002_enum.py +5 -0
  68. vellum/types/token_overlapping_window_chunker_config.py +30 -0
  69. vellum/types/token_overlapping_window_chunker_config_request.py +30 -0
  70. vellum/types/token_overlapping_window_chunker_enum.py +5 -0
  71. vellum/types/token_overlapping_window_chunking.py +30 -0
  72. vellum/types/token_overlapping_window_chunking_request.py +30 -0
  73. vellum/types/workflow_execution_actual_chat_history_request.py +5 -0
  74. vellum/types/workflow_execution_actual_json_request.py +5 -0
  75. vellum/types/workflow_execution_actual_string_request.py +5 -0
  76. vellum/types/workflow_node_result_data.py +24 -0
  77. vellum/types/workflow_output_array.py +2 -2
  78. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.1.dist-info}/METADATA +2 -3
  79. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.1.dist-info}/RECORD +81 -31
  80. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.1.dist-info}/LICENSE +0 -0
  81. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.1.dist-info}/WHEEL +0 -0
vellum/__init__.py CHANGED
@@ -1,6 +1,7 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
3
  from .types import (
4
+ AddOpenaiApiKeyEnum,
4
5
  ApiNodeResult,
5
6
  ApiNodeResultData,
6
7
  ArrayChatMessageContent,
@@ -15,17 +16,27 @@ from .types import (
15
16
  ArrayChatMessageContentRequest,
16
17
  ArrayEnum,
17
18
  ArrayVariableValueItem,
18
- ArrayVariableValueItem_ChatHistory,
19
19
  ArrayVariableValueItem_Error,
20
20
  ArrayVariableValueItem_FunctionCall,
21
21
  ArrayVariableValueItem_Image,
22
22
  ArrayVariableValueItem_Json,
23
23
  ArrayVariableValueItem_Number,
24
- ArrayVariableValueItem_SearchResults,
25
24
  ArrayVariableValueItem_String,
25
+ ArrayVellumValueItem,
26
+ ArrayVellumValueItem_Error,
27
+ ArrayVellumValueItem_FunctionCall,
28
+ ArrayVellumValueItem_Image,
29
+ ArrayVellumValueItem_Json,
30
+ ArrayVellumValueItem_Number,
31
+ ArrayVellumValueItem_String,
32
+ BasicVectorizerIntfloatMultilingualE5Large,
33
+ BasicVectorizerIntfloatMultilingualE5LargeRequest,
34
+ BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1,
35
+ BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request,
36
+ BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1,
37
+ BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
26
38
  ChatHistoryEnum,
27
39
  ChatHistoryInputRequest,
28
- ChatHistoryVariableValue,
29
40
  ChatMessage,
30
41
  ChatMessageContent,
31
42
  ChatMessageContentRequest,
@@ -63,6 +74,16 @@ from .types import (
63
74
  DeploymentProviderPayloadResponse,
64
75
  DeploymentRead,
65
76
  DocumentDocumentToDocumentIndex,
77
+ DocumentIndexChunking,
78
+ DocumentIndexChunkingRequest,
79
+ DocumentIndexChunkingRequest_ReductoChunker,
80
+ DocumentIndexChunkingRequest_SentenceChunker,
81
+ DocumentIndexChunkingRequest_TokenOverlappingWindowChunker,
82
+ DocumentIndexChunking_ReductoChunker,
83
+ DocumentIndexChunking_SentenceChunker,
84
+ DocumentIndexChunking_TokenOverlappingWindowChunker,
85
+ DocumentIndexIndexingConfig,
86
+ DocumentIndexIndexingConfigRequest,
66
87
  DocumentIndexRead,
67
88
  DocumentRead,
68
89
  DocumentStatus,
@@ -111,8 +132,6 @@ from .types import (
111
132
  FulfilledExecutePromptEvent,
112
133
  FulfilledExecutePromptResponse,
113
134
  FulfilledExecuteWorkflowWorkflowResultEvent,
114
- FulfilledFunctionCall,
115
- FulfilledFunctionCallRequest,
116
135
  FulfilledPromptExecutionMeta,
117
136
  FulfilledWorkflowNodeResultEvent,
118
137
  FunctionCall,
@@ -121,10 +140,9 @@ from .types import (
121
140
  FunctionCallChatMessageContentValue,
122
141
  FunctionCallChatMessageContentValueRequest,
123
142
  FunctionCallEnum,
143
+ FunctionCallRequest,
124
144
  FunctionCallVariableValue,
125
145
  FunctionCallVellumValue,
126
- FunctionCall_Fulfilled,
127
- FunctionCall_Rejected,
128
146
  GenerateErrorResponse,
129
147
  GenerateOptionsRequest,
130
148
  GenerateRequest,
@@ -135,24 +153,51 @@ from .types import (
135
153
  GenerateStreamResponse,
136
154
  GenerateStreamResult,
137
155
  GenerateStreamResultData,
156
+ HkunlpInstructorXlEnum,
157
+ HkunlpInstructorXlVectorizer,
158
+ HkunlpInstructorXlVectorizerRequest,
138
159
  ImageChatMessageContent,
139
160
  ImageChatMessageContentRequest,
140
161
  ImageEnum,
141
162
  ImageVariableValue,
163
+ ImageVellumValue,
164
+ IndexingConfigVectorizer,
165
+ IndexingConfigVectorizerRequest,
166
+ IndexingConfigVectorizerRequest_HkunlpInstructorXl,
167
+ IndexingConfigVectorizerRequest_IntfloatMultilingualE5Large,
168
+ IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseCosV1,
169
+ IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseDotV1,
170
+ IndexingConfigVectorizerRequest_TextEmbedding3Large,
171
+ IndexingConfigVectorizerRequest_TextEmbedding3Small,
172
+ IndexingConfigVectorizerRequest_TextEmbeddingAda002,
173
+ IndexingConfigVectorizer_HkunlpInstructorXl,
174
+ IndexingConfigVectorizer_IntfloatMultilingualE5Large,
175
+ IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseCosV1,
176
+ IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseDotV1,
177
+ IndexingConfigVectorizer_TextEmbedding3Large,
178
+ IndexingConfigVectorizer_TextEmbedding3Small,
179
+ IndexingConfigVectorizer_TextEmbeddingAda002,
142
180
  IndexingStateEnum,
143
181
  InitiatedEnum,
144
182
  InitiatedExecutePromptEvent,
145
183
  InitiatedPromptExecutionMeta,
146
184
  InitiatedWorkflowNodeResultEvent,
185
+ InstructorVectorizerConfig,
186
+ InstructorVectorizerConfigRequest,
187
+ IntfloatMultilingualE5LargeEnum,
147
188
  JsonEnum,
148
189
  JsonInputRequest,
149
190
  JsonVariableValue,
150
191
  JsonVellumValue,
151
192
  LogicalOperator,
152
193
  LogprobsEnum,
194
+ MergeEnum,
195
+ MergeNodeResult,
153
196
  MetadataFilterConfigRequest,
154
197
  MetadataFilterRuleCombinator,
155
198
  MetadataFilterRuleRequest,
199
+ MetricEnum,
200
+ MetricNodeResult,
156
201
  MlModelUsage,
157
202
  NamedScenarioInputChatHistoryVariableValueRequest,
158
203
  NamedScenarioInputRequest,
@@ -227,6 +272,15 @@ from .types import (
227
272
  NormalizedTokenLogProbs,
228
273
  NumberEnum,
229
274
  NumberVariableValue,
275
+ NumberVellumValue,
276
+ OpenAiVectorizerConfig,
277
+ OpenAiVectorizerConfigRequest,
278
+ OpenAiVectorizerTextEmbedding3Large,
279
+ OpenAiVectorizerTextEmbedding3LargeRequest,
280
+ OpenAiVectorizerTextEmbedding3Small,
281
+ OpenAiVectorizerTextEmbedding3SmallRequest,
282
+ OpenAiVectorizerTextEmbeddingAda002,
283
+ OpenAiVectorizerTextEmbeddingAda002Request,
230
284
  PaginatedDocumentIndexReadList,
231
285
  PaginatedSlimDeploymentReadList,
232
286
  PaginatedSlimDocumentList,
@@ -249,11 +303,15 @@ from .types import (
249
303
  PromptOutput_Json,
250
304
  PromptOutput_String,
251
305
  RawPromptExecutionOverridesRequest,
306
+ ReductoChunkerConfig,
307
+ ReductoChunkerConfigRequest,
308
+ ReductoChunkerEnum,
309
+ ReductoChunking,
310
+ ReductoChunkingRequest,
252
311
  RejectedEnum,
253
312
  RejectedExecutePromptEvent,
254
313
  RejectedExecutePromptResponse,
255
314
  RejectedExecuteWorkflowWorkflowResultEvent,
256
- RejectedFunctionCall,
257
315
  RejectedPromptExecutionMeta,
258
316
  RejectedWorkflowNodeResultEvent,
259
317
  SandboxScenario,
@@ -274,8 +332,14 @@ from .types import (
274
332
  SearchResultMergingRequest,
275
333
  SearchResultRequest,
276
334
  SearchResultsEnum,
277
- SearchResultsVariableValue,
278
335
  SearchWeightsRequest,
336
+ SentenceChunkerConfig,
337
+ SentenceChunkerConfigRequest,
338
+ SentenceChunkerEnum,
339
+ SentenceChunking,
340
+ SentenceChunkingRequest,
341
+ SentenceTransformersMultiQaMpnetBaseCosV1Enum,
342
+ SentenceTransformersMultiQaMpnetBaseDotV1Enum,
279
343
  SlimDeploymentRead,
280
344
  SlimDocument,
281
345
  SlimWorkflowDeployment,
@@ -405,6 +469,14 @@ from .types import (
405
469
  TestSuiteRunWorkflowReleaseTagExecConfigRequest,
406
470
  TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum,
407
471
  TestSuiteTestCase,
472
+ TextEmbedding3LargeEnum,
473
+ TextEmbedding3SmallEnum,
474
+ TextEmbeddingAda002Enum,
475
+ TokenOverlappingWindowChunkerConfig,
476
+ TokenOverlappingWindowChunkerConfigRequest,
477
+ TokenOverlappingWindowChunkerEnum,
478
+ TokenOverlappingWindowChunking,
479
+ TokenOverlappingWindowChunkingRequest,
408
480
  UploadDocumentErrorResponse,
409
481
  UploadDocumentResponse,
410
482
  VellumError,
@@ -427,6 +499,8 @@ from .types import (
427
499
  WorkflowNodeResultData_Api,
428
500
  WorkflowNodeResultData_CodeExecution,
429
501
  WorkflowNodeResultData_Conditional,
502
+ WorkflowNodeResultData_Merge,
503
+ WorkflowNodeResultData_Metric,
430
504
  WorkflowNodeResultData_Prompt,
431
505
  WorkflowNodeResultData_Search,
432
506
  WorkflowNodeResultData_Subworkflow,
@@ -506,6 +580,7 @@ from .environment import VellumEnvironment
506
580
  from .version import __version__
507
581
 
508
582
  __all__ = [
583
+ "AddOpenaiApiKeyEnum",
509
584
  "ApiNodeResult",
510
585
  "ApiNodeResultData",
511
586
  "ArrayChatMessageContent",
@@ -520,18 +595,28 @@ __all__ = [
520
595
  "ArrayChatMessageContentRequest",
521
596
  "ArrayEnum",
522
597
  "ArrayVariableValueItem",
523
- "ArrayVariableValueItem_ChatHistory",
524
598
  "ArrayVariableValueItem_Error",
525
599
  "ArrayVariableValueItem_FunctionCall",
526
600
  "ArrayVariableValueItem_Image",
527
601
  "ArrayVariableValueItem_Json",
528
602
  "ArrayVariableValueItem_Number",
529
- "ArrayVariableValueItem_SearchResults",
530
603
  "ArrayVariableValueItem_String",
604
+ "ArrayVellumValueItem",
605
+ "ArrayVellumValueItem_Error",
606
+ "ArrayVellumValueItem_FunctionCall",
607
+ "ArrayVellumValueItem_Image",
608
+ "ArrayVellumValueItem_Json",
609
+ "ArrayVellumValueItem_Number",
610
+ "ArrayVellumValueItem_String",
531
611
  "BadRequestError",
612
+ "BasicVectorizerIntfloatMultilingualE5Large",
613
+ "BasicVectorizerIntfloatMultilingualE5LargeRequest",
614
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1",
615
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
616
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
617
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
532
618
  "ChatHistoryEnum",
533
619
  "ChatHistoryInputRequest",
534
- "ChatHistoryVariableValue",
535
620
  "ChatMessage",
536
621
  "ChatMessageContent",
537
622
  "ChatMessageContentRequest",
@@ -570,6 +655,16 @@ __all__ = [
570
655
  "DeploymentRead",
571
656
  "DeploymentsListRequestStatus",
572
657
  "DocumentDocumentToDocumentIndex",
658
+ "DocumentIndexChunking",
659
+ "DocumentIndexChunkingRequest",
660
+ "DocumentIndexChunkingRequest_ReductoChunker",
661
+ "DocumentIndexChunkingRequest_SentenceChunker",
662
+ "DocumentIndexChunkingRequest_TokenOverlappingWindowChunker",
663
+ "DocumentIndexChunking_ReductoChunker",
664
+ "DocumentIndexChunking_SentenceChunker",
665
+ "DocumentIndexChunking_TokenOverlappingWindowChunker",
666
+ "DocumentIndexIndexingConfig",
667
+ "DocumentIndexIndexingConfigRequest",
573
668
  "DocumentIndexRead",
574
669
  "DocumentIndexesListRequestStatus",
575
670
  "DocumentRead",
@@ -620,8 +715,6 @@ __all__ = [
620
715
  "FulfilledExecutePromptEvent",
621
716
  "FulfilledExecutePromptResponse",
622
717
  "FulfilledExecuteWorkflowWorkflowResultEvent",
623
- "FulfilledFunctionCall",
624
- "FulfilledFunctionCallRequest",
625
718
  "FulfilledPromptExecutionMeta",
626
719
  "FulfilledWorkflowNodeResultEvent",
627
720
  "FunctionCall",
@@ -630,10 +723,9 @@ __all__ = [
630
723
  "FunctionCallChatMessageContentValue",
631
724
  "FunctionCallChatMessageContentValueRequest",
632
725
  "FunctionCallEnum",
726
+ "FunctionCallRequest",
633
727
  "FunctionCallVariableValue",
634
728
  "FunctionCallVellumValue",
635
- "FunctionCall_Fulfilled",
636
- "FunctionCall_Rejected",
637
729
  "GenerateErrorResponse",
638
730
  "GenerateOptionsRequest",
639
731
  "GenerateRequest",
@@ -644,25 +736,52 @@ __all__ = [
644
736
  "GenerateStreamResponse",
645
737
  "GenerateStreamResult",
646
738
  "GenerateStreamResultData",
739
+ "HkunlpInstructorXlEnum",
740
+ "HkunlpInstructorXlVectorizer",
741
+ "HkunlpInstructorXlVectorizerRequest",
647
742
  "ImageChatMessageContent",
648
743
  "ImageChatMessageContentRequest",
649
744
  "ImageEnum",
650
745
  "ImageVariableValue",
746
+ "ImageVellumValue",
747
+ "IndexingConfigVectorizer",
748
+ "IndexingConfigVectorizerRequest",
749
+ "IndexingConfigVectorizerRequest_HkunlpInstructorXl",
750
+ "IndexingConfigVectorizerRequest_IntfloatMultilingualE5Large",
751
+ "IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseCosV1",
752
+ "IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseDotV1",
753
+ "IndexingConfigVectorizerRequest_TextEmbedding3Large",
754
+ "IndexingConfigVectorizerRequest_TextEmbedding3Small",
755
+ "IndexingConfigVectorizerRequest_TextEmbeddingAda002",
756
+ "IndexingConfigVectorizer_HkunlpInstructorXl",
757
+ "IndexingConfigVectorizer_IntfloatMultilingualE5Large",
758
+ "IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseCosV1",
759
+ "IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseDotV1",
760
+ "IndexingConfigVectorizer_TextEmbedding3Large",
761
+ "IndexingConfigVectorizer_TextEmbedding3Small",
762
+ "IndexingConfigVectorizer_TextEmbeddingAda002",
651
763
  "IndexingStateEnum",
652
764
  "InitiatedEnum",
653
765
  "InitiatedExecutePromptEvent",
654
766
  "InitiatedPromptExecutionMeta",
655
767
  "InitiatedWorkflowNodeResultEvent",
768
+ "InstructorVectorizerConfig",
769
+ "InstructorVectorizerConfigRequest",
656
770
  "InternalServerError",
771
+ "IntfloatMultilingualE5LargeEnum",
657
772
  "JsonEnum",
658
773
  "JsonInputRequest",
659
774
  "JsonVariableValue",
660
775
  "JsonVellumValue",
661
776
  "LogicalOperator",
662
777
  "LogprobsEnum",
778
+ "MergeEnum",
779
+ "MergeNodeResult",
663
780
  "MetadataFilterConfigRequest",
664
781
  "MetadataFilterRuleCombinator",
665
782
  "MetadataFilterRuleRequest",
783
+ "MetricEnum",
784
+ "MetricNodeResult",
666
785
  "MlModelUsage",
667
786
  "NamedScenarioInputChatHistoryVariableValueRequest",
668
787
  "NamedScenarioInputRequest",
@@ -738,6 +857,15 @@ __all__ = [
738
857
  "NotFoundError",
739
858
  "NumberEnum",
740
859
  "NumberVariableValue",
860
+ "NumberVellumValue",
861
+ "OpenAiVectorizerConfig",
862
+ "OpenAiVectorizerConfigRequest",
863
+ "OpenAiVectorizerTextEmbedding3Large",
864
+ "OpenAiVectorizerTextEmbedding3LargeRequest",
865
+ "OpenAiVectorizerTextEmbedding3Small",
866
+ "OpenAiVectorizerTextEmbedding3SmallRequest",
867
+ "OpenAiVectorizerTextEmbeddingAda002",
868
+ "OpenAiVectorizerTextEmbeddingAda002Request",
741
869
  "PaginatedDocumentIndexReadList",
742
870
  "PaginatedSlimDeploymentReadList",
743
871
  "PaginatedSlimDocumentList",
@@ -760,11 +888,15 @@ __all__ = [
760
888
  "PromptOutput_Json",
761
889
  "PromptOutput_String",
762
890
  "RawPromptExecutionOverridesRequest",
891
+ "ReductoChunkerConfig",
892
+ "ReductoChunkerConfigRequest",
893
+ "ReductoChunkerEnum",
894
+ "ReductoChunking",
895
+ "ReductoChunkingRequest",
763
896
  "RejectedEnum",
764
897
  "RejectedExecutePromptEvent",
765
898
  "RejectedExecutePromptResponse",
766
899
  "RejectedExecuteWorkflowWorkflowResultEvent",
767
- "RejectedFunctionCall",
768
900
  "RejectedPromptExecutionMeta",
769
901
  "RejectedWorkflowNodeResultEvent",
770
902
  "SandboxScenario",
@@ -785,8 +917,14 @@ __all__ = [
785
917
  "SearchResultMergingRequest",
786
918
  "SearchResultRequest",
787
919
  "SearchResultsEnum",
788
- "SearchResultsVariableValue",
789
920
  "SearchWeightsRequest",
921
+ "SentenceChunkerConfig",
922
+ "SentenceChunkerConfigRequest",
923
+ "SentenceChunkerEnum",
924
+ "SentenceChunking",
925
+ "SentenceChunkingRequest",
926
+ "SentenceTransformersMultiQaMpnetBaseCosV1Enum",
927
+ "SentenceTransformersMultiQaMpnetBaseDotV1Enum",
790
928
  "SlimDeploymentRead",
791
929
  "SlimDocument",
792
930
  "SlimWorkflowDeployment",
@@ -916,6 +1054,14 @@ __all__ = [
916
1054
  "TestSuiteRunWorkflowReleaseTagExecConfigRequest",
917
1055
  "TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum",
918
1056
  "TestSuiteTestCase",
1057
+ "TextEmbedding3LargeEnum",
1058
+ "TextEmbedding3SmallEnum",
1059
+ "TextEmbeddingAda002Enum",
1060
+ "TokenOverlappingWindowChunkerConfig",
1061
+ "TokenOverlappingWindowChunkerConfigRequest",
1062
+ "TokenOverlappingWindowChunkerEnum",
1063
+ "TokenOverlappingWindowChunking",
1064
+ "TokenOverlappingWindowChunkingRequest",
919
1065
  "UploadDocumentErrorResponse",
920
1066
  "UploadDocumentResponse",
921
1067
  "VellumEnvironment",
@@ -940,6 +1086,8 @@ __all__ = [
940
1086
  "WorkflowNodeResultData_Api",
941
1087
  "WorkflowNodeResultData_CodeExecution",
942
1088
  "WorkflowNodeResultData_Conditional",
1089
+ "WorkflowNodeResultData_Merge",
1090
+ "WorkflowNodeResultData_Metric",
943
1091
  "WorkflowNodeResultData_Prompt",
944
1092
  "WorkflowNodeResultData_Search",
945
1093
  "WorkflowNodeResultData_Subworkflow",
@@ -18,7 +18,7 @@ class BaseClientWrapper:
18
18
  headers: typing.Dict[str, str] = {
19
19
  "X-Fern-Language": "Python",
20
20
  "X-Fern-SDK-Name": "vellum-ai",
21
- "X-Fern-SDK-Version": "0.5.2",
21
+ "X-Fern-SDK-Version": "0.6.1",
22
22
  }
23
23
  headers["X_API_KEY"] = self.api_key
24
24
  return headers
@@ -10,6 +10,7 @@ from ...core.jsonable_encoder import jsonable_encoder
10
10
  from ...core.pydantic_utilities import pydantic_v1
11
11
  from ...core.remove_none_from_dict import remove_none_from_dict
12
12
  from ...core.request_options import RequestOptions
13
+ from ...types.document_index_indexing_config_request import DocumentIndexIndexingConfigRequest
13
14
  from ...types.document_index_read import DocumentIndexRead
14
15
  from ...types.entity_status import EntityStatus
15
16
  from ...types.environment_enum import EnvironmentEnum
@@ -30,6 +31,7 @@ class DocumentIndexesClient:
30
31
  limit: typing.Optional[int] = None,
31
32
  offset: typing.Optional[int] = None,
32
33
  ordering: typing.Optional[str] = None,
34
+ search: typing.Optional[str] = None,
33
35
  status: typing.Optional[DocumentIndexesListRequestStatus] = None,
34
36
  request_options: typing.Optional[RequestOptions] = None,
35
37
  ) -> PaginatedDocumentIndexReadList:
@@ -43,7 +45,9 @@ class DocumentIndexesClient:
43
45
 
44
46
  - ordering: typing.Optional[str]. Which field to use when ordering the results.
45
47
 
46
- - status: typing.Optional[DocumentIndexesListRequestStatus]. The current status of the document index
48
+ - search: typing.Optional[str]. Search for document indices by name or label
49
+
50
+ - status: typing.Optional[DocumentIndexesListRequestStatus]. Filter down to only document indices that have a status matching the status specified
47
51
 
48
52
  - `ACTIVE` - Active
49
53
  - `ARCHIVED` - Archived
@@ -65,6 +69,7 @@ class DocumentIndexesClient:
65
69
  "limit": limit,
66
70
  "offset": offset,
67
71
  "ordering": ordering,
72
+ "search": search,
68
73
  "status": status,
69
74
  **(
70
75
  request_options.get("additional_query_parameters", {})
@@ -103,7 +108,7 @@ class DocumentIndexesClient:
103
108
  name: str,
104
109
  status: typing.Optional[EntityStatus] = OMIT,
105
110
  environment: typing.Optional[EnvironmentEnum] = OMIT,
106
- indexing_config: typing.Dict[str, typing.Any],
111
+ indexing_config: DocumentIndexIndexingConfigRequest,
107
112
  copy_documents_from_index_id: typing.Optional[str] = OMIT,
108
113
  request_options: typing.Optional[RequestOptions] = None,
109
114
  ) -> DocumentIndexRead:
@@ -124,37 +129,32 @@ class DocumentIndexesClient:
124
129
  * `DEVELOPMENT` - Development
125
130
  * `STAGING` - Staging
126
131
  * `PRODUCTION` - Production
127
- - indexing_config: typing.Dict[str, typing.Any]. Configuration representing how documents should be indexed
132
+ - indexing_config: DocumentIndexIndexingConfigRequest.
128
133
 
129
134
  - copy_documents_from_index_id: typing.Optional[str]. Optionally specify the id of a document index from which you'd like to copy and re-index its documents into this newly created index
130
135
 
131
136
  - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
132
137
  ---
138
+ from vellum import (
139
+ DocumentIndexChunkingRequest_ReductoChunker,
140
+ DocumentIndexIndexingConfigRequest,
141
+ IndexingConfigVectorizerRequest_TextEmbedding3Small,
142
+ )
133
143
  from vellum.client import Vellum
134
144
 
135
145
  client = Vellum(
136
146
  api_key="YOUR_API_KEY",
137
147
  )
138
148
  client.document_indexes.create(
139
- label="My Document Index",
140
- name="my-document-index",
141
- indexing_config={
142
- "chunking": {
143
- "chunker_name": "sentence-chunker",
144
- "chunker_config": {
145
- "character_limit": 1000,
146
- "min_overlap_ratio": 0.5,
147
- },
148
- },
149
- "vectorizer": {
150
- "model_name": "hkunlp/instructor-xl",
151
- "config": {
152
- "instruction_domain": "",
153
- "instruction_document_text_type": "plain_text",
154
- "instruction_query_text_type": "plain_text",
155
- },
156
- },
157
- },
149
+ label="string",
150
+ name="string",
151
+ status="ACTIVE",
152
+ environment="DEVELOPMENT",
153
+ indexing_config=DocumentIndexIndexingConfigRequest(
154
+ vectorizer=IndexingConfigVectorizerRequest_TextEmbedding3Small(),
155
+ chunking=DocumentIndexChunkingRequest_ReductoChunker(),
156
+ ),
157
+ copy_documents_from_index_id="string",
158
158
  )
159
159
  """
160
160
  _request: typing.Dict[str, typing.Any] = {"label": label, "name": name, "indexing_config": indexing_config}
@@ -213,7 +213,7 @@ class DocumentIndexesClient:
213
213
  api_key="YOUR_API_KEY",
214
214
  )
215
215
  client.document_indexes.retrieve(
216
- id="id",
216
+ id="string",
217
217
  )
218
218
  """
219
219
  _response = self._client_wrapper.httpx_client.request(
@@ -280,8 +280,10 @@ class DocumentIndexesClient:
280
280
  api_key="YOUR_API_KEY",
281
281
  )
282
282
  client.document_indexes.update(
283
- id="id",
284
- label="label",
283
+ id="string",
284
+ label="string",
285
+ status="ACTIVE",
286
+ environment="DEVELOPMENT",
285
287
  )
286
288
  """
287
289
  _request: typing.Dict[str, typing.Any] = {"label": label}
@@ -407,7 +409,10 @@ class DocumentIndexesClient:
407
409
  api_key="YOUR_API_KEY",
408
410
  )
409
411
  client.document_indexes.partial_update(
410
- id="id",
412
+ id="string",
413
+ label="string",
414
+ status="ACTIVE",
415
+ environment="DEVELOPMENT",
411
416
  )
412
417
  """
413
418
  _request: typing.Dict[str, typing.Any] = {}
@@ -464,6 +469,7 @@ class AsyncDocumentIndexesClient:
464
469
  limit: typing.Optional[int] = None,
465
470
  offset: typing.Optional[int] = None,
466
471
  ordering: typing.Optional[str] = None,
472
+ search: typing.Optional[str] = None,
467
473
  status: typing.Optional[DocumentIndexesListRequestStatus] = None,
468
474
  request_options: typing.Optional[RequestOptions] = None,
469
475
  ) -> PaginatedDocumentIndexReadList:
@@ -477,7 +483,9 @@ class AsyncDocumentIndexesClient:
477
483
 
478
484
  - ordering: typing.Optional[str]. Which field to use when ordering the results.
479
485
 
480
- - status: typing.Optional[DocumentIndexesListRequestStatus]. The current status of the document index
486
+ - search: typing.Optional[str]. Search for document indices by name or label
487
+
488
+ - status: typing.Optional[DocumentIndexesListRequestStatus]. Filter down to only document indices that have a status matching the status specified
481
489
 
482
490
  - `ACTIVE` - Active
483
491
  - `ARCHIVED` - Archived
@@ -499,6 +507,7 @@ class AsyncDocumentIndexesClient:
499
507
  "limit": limit,
500
508
  "offset": offset,
501
509
  "ordering": ordering,
510
+ "search": search,
502
511
  "status": status,
503
512
  **(
504
513
  request_options.get("additional_query_parameters", {})
@@ -537,7 +546,7 @@ class AsyncDocumentIndexesClient:
537
546
  name: str,
538
547
  status: typing.Optional[EntityStatus] = OMIT,
539
548
  environment: typing.Optional[EnvironmentEnum] = OMIT,
540
- indexing_config: typing.Dict[str, typing.Any],
549
+ indexing_config: DocumentIndexIndexingConfigRequest,
541
550
  copy_documents_from_index_id: typing.Optional[str] = OMIT,
542
551
  request_options: typing.Optional[RequestOptions] = None,
543
552
  ) -> DocumentIndexRead:
@@ -558,37 +567,32 @@ class AsyncDocumentIndexesClient:
558
567
  * `DEVELOPMENT` - Development
559
568
  * `STAGING` - Staging
560
569
  * `PRODUCTION` - Production
561
- - indexing_config: typing.Dict[str, typing.Any]. Configuration representing how documents should be indexed
570
+ - indexing_config: DocumentIndexIndexingConfigRequest.
562
571
 
563
572
  - copy_documents_from_index_id: typing.Optional[str]. Optionally specify the id of a document index from which you'd like to copy and re-index its documents into this newly created index
564
573
 
565
574
  - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
566
575
  ---
576
+ from vellum import (
577
+ DocumentIndexChunkingRequest_ReductoChunker,
578
+ DocumentIndexIndexingConfigRequest,
579
+ IndexingConfigVectorizerRequest_TextEmbedding3Small,
580
+ )
567
581
  from vellum.client import AsyncVellum
568
582
 
569
583
  client = AsyncVellum(
570
584
  api_key="YOUR_API_KEY",
571
585
  )
572
586
  await client.document_indexes.create(
573
- label="My Document Index",
574
- name="my-document-index",
575
- indexing_config={
576
- "chunking": {
577
- "chunker_name": "sentence-chunker",
578
- "chunker_config": {
579
- "character_limit": 1000,
580
- "min_overlap_ratio": 0.5,
581
- },
582
- },
583
- "vectorizer": {
584
- "model_name": "hkunlp/instructor-xl",
585
- "config": {
586
- "instruction_domain": "",
587
- "instruction_document_text_type": "plain_text",
588
- "instruction_query_text_type": "plain_text",
589
- },
590
- },
591
- },
587
+ label="string",
588
+ name="string",
589
+ status="ACTIVE",
590
+ environment="DEVELOPMENT",
591
+ indexing_config=DocumentIndexIndexingConfigRequest(
592
+ vectorizer=IndexingConfigVectorizerRequest_TextEmbedding3Small(),
593
+ chunking=DocumentIndexChunkingRequest_ReductoChunker(),
594
+ ),
595
+ copy_documents_from_index_id="string",
592
596
  )
593
597
  """
594
598
  _request: typing.Dict[str, typing.Any] = {"label": label, "name": name, "indexing_config": indexing_config}
@@ -647,7 +651,7 @@ class AsyncDocumentIndexesClient:
647
651
  api_key="YOUR_API_KEY",
648
652
  )
649
653
  await client.document_indexes.retrieve(
650
- id="id",
654
+ id="string",
651
655
  )
652
656
  """
653
657
  _response = await self._client_wrapper.httpx_client.request(
@@ -714,8 +718,10 @@ class AsyncDocumentIndexesClient:
714
718
  api_key="YOUR_API_KEY",
715
719
  )
716
720
  await client.document_indexes.update(
717
- id="id",
718
- label="label",
721
+ id="string",
722
+ label="string",
723
+ status="ACTIVE",
724
+ environment="DEVELOPMENT",
719
725
  )
720
726
  """
721
727
  _request: typing.Dict[str, typing.Any] = {"label": label}
@@ -841,7 +847,10 @@ class AsyncDocumentIndexesClient:
841
847
  api_key="YOUR_API_KEY",
842
848
  )
843
849
  await client.document_indexes.partial_update(
844
- id="id",
850
+ id="string",
851
+ label="string",
852
+ status="ACTIVE",
853
+ environment="DEVELOPMENT",
845
854
  )
846
855
  """
847
856
  _request: typing.Dict[str, typing.Any] = {}