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

Sign up to get free protection for your applications and to get access to all the features.
Files changed (76) hide show
  1. vellum/__init__.py +160 -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 +179 -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/metric_enum.py +5 -0
  33. vellum/types/{chat_history_variable_value.py → metric_node_result.py} +4 -3
  34. vellum/types/named_test_case_function_call_variable_value.py +2 -2
  35. vellum/types/named_test_case_function_call_variable_value_request.py +2 -2
  36. vellum/types/node_output_compiled_array_value.py +2 -2
  37. vellum/types/{search_results_variable_value.py → number_vellum_value.py} +6 -3
  38. vellum/types/open_ai_vectorizer_config.py +30 -0
  39. vellum/types/open_ai_vectorizer_config_request.py +30 -0
  40. vellum/types/open_ai_vectorizer_text_embedding_3_large.py +30 -0
  41. vellum/types/open_ai_vectorizer_text_embedding_3_large_request.py +30 -0
  42. vellum/types/open_ai_vectorizer_text_embedding_3_small.py +30 -0
  43. vellum/types/open_ai_vectorizer_text_embedding_3_small_request.py +30 -0
  44. vellum/types/open_ai_vectorizer_text_embedding_ada_002.py +30 -0
  45. vellum/types/open_ai_vectorizer_text_embedding_ada_002_request.py +30 -0
  46. vellum/types/reducto_chunker_config.py +29 -0
  47. vellum/types/reducto_chunker_config_request.py +29 -0
  48. vellum/types/reducto_chunker_enum.py +5 -0
  49. vellum/types/reducto_chunking.py +30 -0
  50. vellum/types/reducto_chunking_request.py +30 -0
  51. vellum/types/sentence_chunker_config.py +30 -0
  52. vellum/types/sentence_chunker_config_request.py +30 -0
  53. vellum/types/sentence_chunker_enum.py +5 -0
  54. vellum/types/sentence_chunking.py +30 -0
  55. vellum/types/sentence_chunking_request.py +30 -0
  56. vellum/types/sentence_transformers_multi_qa_mpnet_base_cos_v_1_enum.py +5 -0
  57. vellum/types/sentence_transformers_multi_qa_mpnet_base_dot_v_1_enum.py +5 -0
  58. vellum/types/test_case_function_call_variable_value.py +2 -2
  59. vellum/types/test_suite_run_execution_function_call_output.py +2 -2
  60. vellum/types/text_embedding_3_large_enum.py +5 -0
  61. vellum/types/text_embedding_3_small_enum.py +5 -0
  62. vellum/types/text_embedding_ada_002_enum.py +5 -0
  63. vellum/types/token_overlapping_window_chunker_config.py +30 -0
  64. vellum/types/token_overlapping_window_chunker_config_request.py +30 -0
  65. vellum/types/token_overlapping_window_chunker_enum.py +5 -0
  66. vellum/types/token_overlapping_window_chunking.py +30 -0
  67. vellum/types/token_overlapping_window_chunking_request.py +30 -0
  68. vellum/types/workflow_execution_actual_chat_history_request.py +5 -0
  69. vellum/types/workflow_execution_actual_json_request.py +5 -0
  70. vellum/types/workflow_execution_actual_string_request.py +5 -0
  71. vellum/types/workflow_node_result_data.py +12 -0
  72. vellum/types/workflow_output_array.py +2 -2
  73. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.0.dist-info}/METADATA +1 -1
  74. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.0.dist-info}/RECORD +76 -28
  75. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.0.dist-info}/LICENSE +0 -0
  76. {vellum_ai-0.5.2.dist-info → vellum_ai-0.6.0.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,15 +153,38 @@ 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,
@@ -153,6 +194,8 @@ from .types import (
153
194
  MetadataFilterConfigRequest,
154
195
  MetadataFilterRuleCombinator,
155
196
  MetadataFilterRuleRequest,
197
+ MetricEnum,
198
+ MetricNodeResult,
156
199
  MlModelUsage,
157
200
  NamedScenarioInputChatHistoryVariableValueRequest,
158
201
  NamedScenarioInputRequest,
@@ -227,6 +270,15 @@ from .types import (
227
270
  NormalizedTokenLogProbs,
228
271
  NumberEnum,
229
272
  NumberVariableValue,
273
+ NumberVellumValue,
274
+ OpenAiVectorizerConfig,
275
+ OpenAiVectorizerConfigRequest,
276
+ OpenAiVectorizerTextEmbedding3Large,
277
+ OpenAiVectorizerTextEmbedding3LargeRequest,
278
+ OpenAiVectorizerTextEmbedding3Small,
279
+ OpenAiVectorizerTextEmbedding3SmallRequest,
280
+ OpenAiVectorizerTextEmbeddingAda002,
281
+ OpenAiVectorizerTextEmbeddingAda002Request,
230
282
  PaginatedDocumentIndexReadList,
231
283
  PaginatedSlimDeploymentReadList,
232
284
  PaginatedSlimDocumentList,
@@ -249,11 +301,15 @@ from .types import (
249
301
  PromptOutput_Json,
250
302
  PromptOutput_String,
251
303
  RawPromptExecutionOverridesRequest,
304
+ ReductoChunkerConfig,
305
+ ReductoChunkerConfigRequest,
306
+ ReductoChunkerEnum,
307
+ ReductoChunking,
308
+ ReductoChunkingRequest,
252
309
  RejectedEnum,
253
310
  RejectedExecutePromptEvent,
254
311
  RejectedExecutePromptResponse,
255
312
  RejectedExecuteWorkflowWorkflowResultEvent,
256
- RejectedFunctionCall,
257
313
  RejectedPromptExecutionMeta,
258
314
  RejectedWorkflowNodeResultEvent,
259
315
  SandboxScenario,
@@ -274,8 +330,14 @@ from .types import (
274
330
  SearchResultMergingRequest,
275
331
  SearchResultRequest,
276
332
  SearchResultsEnum,
277
- SearchResultsVariableValue,
278
333
  SearchWeightsRequest,
334
+ SentenceChunkerConfig,
335
+ SentenceChunkerConfigRequest,
336
+ SentenceChunkerEnum,
337
+ SentenceChunking,
338
+ SentenceChunkingRequest,
339
+ SentenceTransformersMultiQaMpnetBaseCosV1Enum,
340
+ SentenceTransformersMultiQaMpnetBaseDotV1Enum,
279
341
  SlimDeploymentRead,
280
342
  SlimDocument,
281
343
  SlimWorkflowDeployment,
@@ -405,6 +467,14 @@ from .types import (
405
467
  TestSuiteRunWorkflowReleaseTagExecConfigRequest,
406
468
  TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum,
407
469
  TestSuiteTestCase,
470
+ TextEmbedding3LargeEnum,
471
+ TextEmbedding3SmallEnum,
472
+ TextEmbeddingAda002Enum,
473
+ TokenOverlappingWindowChunkerConfig,
474
+ TokenOverlappingWindowChunkerConfigRequest,
475
+ TokenOverlappingWindowChunkerEnum,
476
+ TokenOverlappingWindowChunking,
477
+ TokenOverlappingWindowChunkingRequest,
408
478
  UploadDocumentErrorResponse,
409
479
  UploadDocumentResponse,
410
480
  VellumError,
@@ -427,6 +497,7 @@ from .types import (
427
497
  WorkflowNodeResultData_Api,
428
498
  WorkflowNodeResultData_CodeExecution,
429
499
  WorkflowNodeResultData_Conditional,
500
+ WorkflowNodeResultData_Metric,
430
501
  WorkflowNodeResultData_Prompt,
431
502
  WorkflowNodeResultData_Search,
432
503
  WorkflowNodeResultData_Subworkflow,
@@ -506,6 +577,7 @@ from .environment import VellumEnvironment
506
577
  from .version import __version__
507
578
 
508
579
  __all__ = [
580
+ "AddOpenaiApiKeyEnum",
509
581
  "ApiNodeResult",
510
582
  "ApiNodeResultData",
511
583
  "ArrayChatMessageContent",
@@ -520,18 +592,28 @@ __all__ = [
520
592
  "ArrayChatMessageContentRequest",
521
593
  "ArrayEnum",
522
594
  "ArrayVariableValueItem",
523
- "ArrayVariableValueItem_ChatHistory",
524
595
  "ArrayVariableValueItem_Error",
525
596
  "ArrayVariableValueItem_FunctionCall",
526
597
  "ArrayVariableValueItem_Image",
527
598
  "ArrayVariableValueItem_Json",
528
599
  "ArrayVariableValueItem_Number",
529
- "ArrayVariableValueItem_SearchResults",
530
600
  "ArrayVariableValueItem_String",
601
+ "ArrayVellumValueItem",
602
+ "ArrayVellumValueItem_Error",
603
+ "ArrayVellumValueItem_FunctionCall",
604
+ "ArrayVellumValueItem_Image",
605
+ "ArrayVellumValueItem_Json",
606
+ "ArrayVellumValueItem_Number",
607
+ "ArrayVellumValueItem_String",
531
608
  "BadRequestError",
609
+ "BasicVectorizerIntfloatMultilingualE5Large",
610
+ "BasicVectorizerIntfloatMultilingualE5LargeRequest",
611
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1",
612
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
613
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
614
+ "BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
532
615
  "ChatHistoryEnum",
533
616
  "ChatHistoryInputRequest",
534
- "ChatHistoryVariableValue",
535
617
  "ChatMessage",
536
618
  "ChatMessageContent",
537
619
  "ChatMessageContentRequest",
@@ -570,6 +652,16 @@ __all__ = [
570
652
  "DeploymentRead",
571
653
  "DeploymentsListRequestStatus",
572
654
  "DocumentDocumentToDocumentIndex",
655
+ "DocumentIndexChunking",
656
+ "DocumentIndexChunkingRequest",
657
+ "DocumentIndexChunkingRequest_ReductoChunker",
658
+ "DocumentIndexChunkingRequest_SentenceChunker",
659
+ "DocumentIndexChunkingRequest_TokenOverlappingWindowChunker",
660
+ "DocumentIndexChunking_ReductoChunker",
661
+ "DocumentIndexChunking_SentenceChunker",
662
+ "DocumentIndexChunking_TokenOverlappingWindowChunker",
663
+ "DocumentIndexIndexingConfig",
664
+ "DocumentIndexIndexingConfigRequest",
573
665
  "DocumentIndexRead",
574
666
  "DocumentIndexesListRequestStatus",
575
667
  "DocumentRead",
@@ -620,8 +712,6 @@ __all__ = [
620
712
  "FulfilledExecutePromptEvent",
621
713
  "FulfilledExecutePromptResponse",
622
714
  "FulfilledExecuteWorkflowWorkflowResultEvent",
623
- "FulfilledFunctionCall",
624
- "FulfilledFunctionCallRequest",
625
715
  "FulfilledPromptExecutionMeta",
626
716
  "FulfilledWorkflowNodeResultEvent",
627
717
  "FunctionCall",
@@ -630,10 +720,9 @@ __all__ = [
630
720
  "FunctionCallChatMessageContentValue",
631
721
  "FunctionCallChatMessageContentValueRequest",
632
722
  "FunctionCallEnum",
723
+ "FunctionCallRequest",
633
724
  "FunctionCallVariableValue",
634
725
  "FunctionCallVellumValue",
635
- "FunctionCall_Fulfilled",
636
- "FunctionCall_Rejected",
637
726
  "GenerateErrorResponse",
638
727
  "GenerateOptionsRequest",
639
728
  "GenerateRequest",
@@ -644,16 +733,39 @@ __all__ = [
644
733
  "GenerateStreamResponse",
645
734
  "GenerateStreamResult",
646
735
  "GenerateStreamResultData",
736
+ "HkunlpInstructorXlEnum",
737
+ "HkunlpInstructorXlVectorizer",
738
+ "HkunlpInstructorXlVectorizerRequest",
647
739
  "ImageChatMessageContent",
648
740
  "ImageChatMessageContentRequest",
649
741
  "ImageEnum",
650
742
  "ImageVariableValue",
743
+ "ImageVellumValue",
744
+ "IndexingConfigVectorizer",
745
+ "IndexingConfigVectorizerRequest",
746
+ "IndexingConfigVectorizerRequest_HkunlpInstructorXl",
747
+ "IndexingConfigVectorizerRequest_IntfloatMultilingualE5Large",
748
+ "IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseCosV1",
749
+ "IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseDotV1",
750
+ "IndexingConfigVectorizerRequest_TextEmbedding3Large",
751
+ "IndexingConfigVectorizerRequest_TextEmbedding3Small",
752
+ "IndexingConfigVectorizerRequest_TextEmbeddingAda002",
753
+ "IndexingConfigVectorizer_HkunlpInstructorXl",
754
+ "IndexingConfigVectorizer_IntfloatMultilingualE5Large",
755
+ "IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseCosV1",
756
+ "IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseDotV1",
757
+ "IndexingConfigVectorizer_TextEmbedding3Large",
758
+ "IndexingConfigVectorizer_TextEmbedding3Small",
759
+ "IndexingConfigVectorizer_TextEmbeddingAda002",
651
760
  "IndexingStateEnum",
652
761
  "InitiatedEnum",
653
762
  "InitiatedExecutePromptEvent",
654
763
  "InitiatedPromptExecutionMeta",
655
764
  "InitiatedWorkflowNodeResultEvent",
765
+ "InstructorVectorizerConfig",
766
+ "InstructorVectorizerConfigRequest",
656
767
  "InternalServerError",
768
+ "IntfloatMultilingualE5LargeEnum",
657
769
  "JsonEnum",
658
770
  "JsonInputRequest",
659
771
  "JsonVariableValue",
@@ -663,6 +775,8 @@ __all__ = [
663
775
  "MetadataFilterConfigRequest",
664
776
  "MetadataFilterRuleCombinator",
665
777
  "MetadataFilterRuleRequest",
778
+ "MetricEnum",
779
+ "MetricNodeResult",
666
780
  "MlModelUsage",
667
781
  "NamedScenarioInputChatHistoryVariableValueRequest",
668
782
  "NamedScenarioInputRequest",
@@ -738,6 +852,15 @@ __all__ = [
738
852
  "NotFoundError",
739
853
  "NumberEnum",
740
854
  "NumberVariableValue",
855
+ "NumberVellumValue",
856
+ "OpenAiVectorizerConfig",
857
+ "OpenAiVectorizerConfigRequest",
858
+ "OpenAiVectorizerTextEmbedding3Large",
859
+ "OpenAiVectorizerTextEmbedding3LargeRequest",
860
+ "OpenAiVectorizerTextEmbedding3Small",
861
+ "OpenAiVectorizerTextEmbedding3SmallRequest",
862
+ "OpenAiVectorizerTextEmbeddingAda002",
863
+ "OpenAiVectorizerTextEmbeddingAda002Request",
741
864
  "PaginatedDocumentIndexReadList",
742
865
  "PaginatedSlimDeploymentReadList",
743
866
  "PaginatedSlimDocumentList",
@@ -760,11 +883,15 @@ __all__ = [
760
883
  "PromptOutput_Json",
761
884
  "PromptOutput_String",
762
885
  "RawPromptExecutionOverridesRequest",
886
+ "ReductoChunkerConfig",
887
+ "ReductoChunkerConfigRequest",
888
+ "ReductoChunkerEnum",
889
+ "ReductoChunking",
890
+ "ReductoChunkingRequest",
763
891
  "RejectedEnum",
764
892
  "RejectedExecutePromptEvent",
765
893
  "RejectedExecutePromptResponse",
766
894
  "RejectedExecuteWorkflowWorkflowResultEvent",
767
- "RejectedFunctionCall",
768
895
  "RejectedPromptExecutionMeta",
769
896
  "RejectedWorkflowNodeResultEvent",
770
897
  "SandboxScenario",
@@ -785,8 +912,14 @@ __all__ = [
785
912
  "SearchResultMergingRequest",
786
913
  "SearchResultRequest",
787
914
  "SearchResultsEnum",
788
- "SearchResultsVariableValue",
789
915
  "SearchWeightsRequest",
916
+ "SentenceChunkerConfig",
917
+ "SentenceChunkerConfigRequest",
918
+ "SentenceChunkerEnum",
919
+ "SentenceChunking",
920
+ "SentenceChunkingRequest",
921
+ "SentenceTransformersMultiQaMpnetBaseCosV1Enum",
922
+ "SentenceTransformersMultiQaMpnetBaseDotV1Enum",
790
923
  "SlimDeploymentRead",
791
924
  "SlimDocument",
792
925
  "SlimWorkflowDeployment",
@@ -916,6 +1049,14 @@ __all__ = [
916
1049
  "TestSuiteRunWorkflowReleaseTagExecConfigRequest",
917
1050
  "TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum",
918
1051
  "TestSuiteTestCase",
1052
+ "TextEmbedding3LargeEnum",
1053
+ "TextEmbedding3SmallEnum",
1054
+ "TextEmbeddingAda002Enum",
1055
+ "TokenOverlappingWindowChunkerConfig",
1056
+ "TokenOverlappingWindowChunkerConfigRequest",
1057
+ "TokenOverlappingWindowChunkerEnum",
1058
+ "TokenOverlappingWindowChunking",
1059
+ "TokenOverlappingWindowChunkingRequest",
919
1060
  "UploadDocumentErrorResponse",
920
1061
  "UploadDocumentResponse",
921
1062
  "VellumEnvironment",
@@ -940,6 +1081,7 @@ __all__ = [
940
1081
  "WorkflowNodeResultData_Api",
941
1082
  "WorkflowNodeResultData_CodeExecution",
942
1083
  "WorkflowNodeResultData_Conditional",
1084
+ "WorkflowNodeResultData_Metric",
943
1085
  "WorkflowNodeResultData_Prompt",
944
1086
  "WorkflowNodeResultData_Search",
945
1087
  "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.0",
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] = {}