vellum-ai 0.5.1__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.
- vellum/__init__.py +170 -20
- vellum/core/client_wrapper.py +1 -1
- vellum/resources/document_indexes/client.py +61 -52
- vellum/resources/documents/client.py +8 -4
- vellum/types/__init__.py +189 -19
- vellum/types/add_openai_api_key_enum.py +3 -0
- vellum/types/array_variable_value_item.py +0 -24
- vellum/types/array_vellum_value_item.py +82 -0
- vellum/types/basic_vectorizer_intfloat_multilingual_e_5_large.py +29 -0
- vellum/types/basic_vectorizer_intfloat_multilingual_e_5_large_request.py +29 -0
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1.py +29 -0
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request.py +29 -0
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1.py +29 -0
- vellum/types/basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request.py +29 -0
- vellum/types/document_index_chunking.py +46 -0
- vellum/types/document_index_chunking_request.py +46 -0
- vellum/types/document_index_indexing_config.py +28 -0
- vellum/types/document_index_indexing_config_request.py +28 -0
- vellum/types/document_index_read.py +2 -4
- vellum/types/{rejected_function_call.py → error_vellum_value.py} +3 -5
- vellum/types/function_call.py +20 -18
- vellum/types/{fulfilled_function_call_request.py → function_call_request.py} +2 -2
- vellum/types/function_call_variable_value.py +1 -1
- vellum/types/{node_output_compiled_function_value.py → function_call_vellum_value.py} +5 -2
- vellum/types/hkunlp_instructor_xl_enum.py +5 -0
- vellum/types/hkunlp_instructor_xl_vectorizer.py +30 -0
- vellum/types/hkunlp_instructor_xl_vectorizer_request.py +30 -0
- vellum/types/{fulfilled_function_call.py → image_vellum_value.py} +4 -5
- vellum/types/indexing_config_vectorizer.py +106 -0
- vellum/types/indexing_config_vectorizer_request.py +106 -0
- vellum/types/instructor_vectorizer_config.py +31 -0
- vellum/types/instructor_vectorizer_config_request.py +31 -0
- vellum/types/intfloat_multilingual_e_5_large_enum.py +5 -0
- vellum/types/json_vellum_value.py +29 -0
- vellum/types/metric_enum.py +5 -0
- vellum/types/{chat_history_variable_value.py → metric_node_result.py} +4 -3
- vellum/types/named_test_case_function_call_variable_value.py +2 -2
- vellum/types/named_test_case_function_call_variable_value_request.py +2 -2
- vellum/types/node_output_compiled_array_value.py +8 -2
- vellum/types/node_output_compiled_chat_history_value.py +7 -1
- vellum/types/node_output_compiled_error_value.py +7 -1
- vellum/types/node_output_compiled_function_call_value.py +33 -0
- vellum/types/node_output_compiled_json_value.py +7 -1
- vellum/types/node_output_compiled_number_value.py +7 -1
- vellum/types/node_output_compiled_search_results_value.py +7 -1
- vellum/types/node_output_compiled_string_value.py +7 -1
- vellum/types/node_output_compiled_value.py +2 -2
- vellum/types/{search_results_variable_value.py → number_vellum_value.py} +6 -3
- vellum/types/open_ai_vectorizer_config.py +30 -0
- vellum/types/open_ai_vectorizer_config_request.py +30 -0
- vellum/types/open_ai_vectorizer_text_embedding_3_large.py +30 -0
- vellum/types/open_ai_vectorizer_text_embedding_3_large_request.py +30 -0
- vellum/types/open_ai_vectorizer_text_embedding_3_small.py +30 -0
- vellum/types/open_ai_vectorizer_text_embedding_3_small_request.py +30 -0
- vellum/types/open_ai_vectorizer_text_embedding_ada_002.py +30 -0
- vellum/types/open_ai_vectorizer_text_embedding_ada_002_request.py +30 -0
- vellum/types/prompt_output.py +8 -8
- vellum/types/reducto_chunker_config.py +29 -0
- vellum/types/reducto_chunker_config_request.py +29 -0
- vellum/types/reducto_chunker_enum.py +5 -0
- vellum/types/reducto_chunking.py +30 -0
- vellum/types/reducto_chunking_request.py +30 -0
- vellum/types/sentence_chunker_config.py +30 -0
- vellum/types/sentence_chunker_config_request.py +30 -0
- vellum/types/sentence_chunker_enum.py +5 -0
- vellum/types/sentence_chunking.py +30 -0
- vellum/types/sentence_chunking_request.py +30 -0
- vellum/types/sentence_transformers_multi_qa_mpnet_base_cos_v_1_enum.py +5 -0
- vellum/types/sentence_transformers_multi_qa_mpnet_base_dot_v_1_enum.py +5 -0
- vellum/types/string_vellum_value.py +29 -0
- vellum/types/test_case_function_call_variable_value.py +2 -2
- vellum/types/test_suite_run_execution_function_call_output.py +2 -2
- vellum/types/text_embedding_3_large_enum.py +5 -0
- vellum/types/text_embedding_3_small_enum.py +5 -0
- vellum/types/text_embedding_ada_002_enum.py +5 -0
- vellum/types/token_overlapping_window_chunker_config.py +30 -0
- vellum/types/token_overlapping_window_chunker_config_request.py +30 -0
- vellum/types/token_overlapping_window_chunker_enum.py +5 -0
- vellum/types/token_overlapping_window_chunking.py +30 -0
- vellum/types/token_overlapping_window_chunking_request.py +30 -0
- vellum/types/workflow_execution_actual_chat_history_request.py +5 -0
- vellum/types/workflow_execution_actual_json_request.py +5 -0
- vellum/types/workflow_execution_actual_string_request.py +5 -0
- vellum/types/workflow_node_result_data.py +12 -0
- vellum/types/workflow_output_array.py +2 -2
- {vellum_ai-0.5.1.dist-info → vellum_ai-0.6.0.dist-info}/METADATA +1 -1
- {vellum_ai-0.5.1.dist-info → vellum_ai-0.6.0.dist-info}/RECORD +89 -37
- {vellum_ai-0.5.1.dist-info → vellum_ai-0.6.0.dist-info}/LICENSE +0 -0
- {vellum_ai-0.5.1.dist-info → vellum_ai-0.6.0.dist-info}/WHEEL +0 -0
vellum/types/__init__.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1
1
|
# This file was auto-generated by Fern from our API Definition.
|
2
2
|
|
3
|
+
from .add_openai_api_key_enum import AddOpenaiApiKeyEnum
|
3
4
|
from .api_node_result import ApiNodeResult
|
4
5
|
from .api_node_result_data import ApiNodeResultData
|
5
6
|
from .array_chat_message_content import ArrayChatMessageContent
|
@@ -19,18 +20,38 @@ from .array_chat_message_content_request import ArrayChatMessageContentRequest
|
|
19
20
|
from .array_enum import ArrayEnum
|
20
21
|
from .array_variable_value_item import (
|
21
22
|
ArrayVariableValueItem,
|
22
|
-
ArrayVariableValueItem_ChatHistory,
|
23
23
|
ArrayVariableValueItem_Error,
|
24
24
|
ArrayVariableValueItem_FunctionCall,
|
25
25
|
ArrayVariableValueItem_Image,
|
26
26
|
ArrayVariableValueItem_Json,
|
27
27
|
ArrayVariableValueItem_Number,
|
28
|
-
ArrayVariableValueItem_SearchResults,
|
29
28
|
ArrayVariableValueItem_String,
|
30
29
|
)
|
30
|
+
from .array_vellum_value_item import (
|
31
|
+
ArrayVellumValueItem,
|
32
|
+
ArrayVellumValueItem_Error,
|
33
|
+
ArrayVellumValueItem_FunctionCall,
|
34
|
+
ArrayVellumValueItem_Image,
|
35
|
+
ArrayVellumValueItem_Json,
|
36
|
+
ArrayVellumValueItem_Number,
|
37
|
+
ArrayVellumValueItem_String,
|
38
|
+
)
|
39
|
+
from .basic_vectorizer_intfloat_multilingual_e_5_large import BasicVectorizerIntfloatMultilingualE5Large
|
40
|
+
from .basic_vectorizer_intfloat_multilingual_e_5_large_request import BasicVectorizerIntfloatMultilingualE5LargeRequest
|
41
|
+
from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1 import (
|
42
|
+
BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1,
|
43
|
+
)
|
44
|
+
from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_cos_v_1_request import (
|
45
|
+
BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request,
|
46
|
+
)
|
47
|
+
from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1 import (
|
48
|
+
BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1,
|
49
|
+
)
|
50
|
+
from .basic_vectorizer_sentence_transformers_multi_qa_mpnet_base_dot_v_1_request import (
|
51
|
+
BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request,
|
52
|
+
)
|
31
53
|
from .chat_history_enum import ChatHistoryEnum
|
32
54
|
from .chat_history_input_request import ChatHistoryInputRequest
|
33
|
-
from .chat_history_variable_value import ChatHistoryVariableValue
|
34
55
|
from .chat_message import ChatMessage
|
35
56
|
from .chat_message_content import (
|
36
57
|
ChatMessageContent,
|
@@ -74,6 +95,20 @@ from .conditional_node_result_data import ConditionalNodeResultData
|
|
74
95
|
from .deployment_provider_payload_response import DeploymentProviderPayloadResponse
|
75
96
|
from .deployment_read import DeploymentRead
|
76
97
|
from .document_document_to_document_index import DocumentDocumentToDocumentIndex
|
98
|
+
from .document_index_chunking import (
|
99
|
+
DocumentIndexChunking,
|
100
|
+
DocumentIndexChunking_ReductoChunker,
|
101
|
+
DocumentIndexChunking_SentenceChunker,
|
102
|
+
DocumentIndexChunking_TokenOverlappingWindowChunker,
|
103
|
+
)
|
104
|
+
from .document_index_chunking_request import (
|
105
|
+
DocumentIndexChunkingRequest,
|
106
|
+
DocumentIndexChunkingRequest_ReductoChunker,
|
107
|
+
DocumentIndexChunkingRequest_SentenceChunker,
|
108
|
+
DocumentIndexChunkingRequest_TokenOverlappingWindowChunker,
|
109
|
+
)
|
110
|
+
from .document_index_indexing_config import DocumentIndexIndexingConfig
|
111
|
+
from .document_index_indexing_config_request import DocumentIndexIndexingConfigRequest
|
77
112
|
from .document_index_read import DocumentIndexRead
|
78
113
|
from .document_read import DocumentRead
|
79
114
|
from .document_status import DocumentStatus
|
@@ -82,6 +117,7 @@ from .entity_status import EntityStatus
|
|
82
117
|
from .environment_enum import EnvironmentEnum
|
83
118
|
from .error_enum import ErrorEnum
|
84
119
|
from .error_variable_value import ErrorVariableValue
|
120
|
+
from .error_vellum_value import ErrorVellumValue
|
85
121
|
from .execute_prompt_api_error_response import ExecutePromptApiErrorResponse
|
86
122
|
from .execute_prompt_event import (
|
87
123
|
ExecutePromptEvent,
|
@@ -129,17 +165,17 @@ from .fulfilled_enum import FulfilledEnum
|
|
129
165
|
from .fulfilled_execute_prompt_event import FulfilledExecutePromptEvent
|
130
166
|
from .fulfilled_execute_prompt_response import FulfilledExecutePromptResponse
|
131
167
|
from .fulfilled_execute_workflow_workflow_result_event import FulfilledExecuteWorkflowWorkflowResultEvent
|
132
|
-
from .fulfilled_function_call import FulfilledFunctionCall
|
133
|
-
from .fulfilled_function_call_request import FulfilledFunctionCallRequest
|
134
168
|
from .fulfilled_prompt_execution_meta import FulfilledPromptExecutionMeta
|
135
169
|
from .fulfilled_workflow_node_result_event import FulfilledWorkflowNodeResultEvent
|
136
|
-
from .function_call import FunctionCall
|
170
|
+
from .function_call import FunctionCall
|
137
171
|
from .function_call_chat_message_content import FunctionCallChatMessageContent
|
138
172
|
from .function_call_chat_message_content_request import FunctionCallChatMessageContentRequest
|
139
173
|
from .function_call_chat_message_content_value import FunctionCallChatMessageContentValue
|
140
174
|
from .function_call_chat_message_content_value_request import FunctionCallChatMessageContentValueRequest
|
141
175
|
from .function_call_enum import FunctionCallEnum
|
176
|
+
from .function_call_request import FunctionCallRequest
|
142
177
|
from .function_call_variable_value import FunctionCallVariableValue
|
178
|
+
from .function_call_vellum_value import FunctionCallVellumValue
|
143
179
|
from .generate_error_response import GenerateErrorResponse
|
144
180
|
from .generate_options_request import GenerateOptionsRequest
|
145
181
|
from .generate_request import GenerateRequest
|
@@ -150,23 +186,53 @@ from .generate_result_error import GenerateResultError
|
|
150
186
|
from .generate_stream_response import GenerateStreamResponse
|
151
187
|
from .generate_stream_result import GenerateStreamResult
|
152
188
|
from .generate_stream_result_data import GenerateStreamResultData
|
189
|
+
from .hkunlp_instructor_xl_enum import HkunlpInstructorXlEnum
|
190
|
+
from .hkunlp_instructor_xl_vectorizer import HkunlpInstructorXlVectorizer
|
191
|
+
from .hkunlp_instructor_xl_vectorizer_request import HkunlpInstructorXlVectorizerRequest
|
153
192
|
from .image_chat_message_content import ImageChatMessageContent
|
154
193
|
from .image_chat_message_content_request import ImageChatMessageContentRequest
|
155
194
|
from .image_enum import ImageEnum
|
156
195
|
from .image_variable_value import ImageVariableValue
|
196
|
+
from .image_vellum_value import ImageVellumValue
|
197
|
+
from .indexing_config_vectorizer import (
|
198
|
+
IndexingConfigVectorizer,
|
199
|
+
IndexingConfigVectorizer_HkunlpInstructorXl,
|
200
|
+
IndexingConfigVectorizer_IntfloatMultilingualE5Large,
|
201
|
+
IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseCosV1,
|
202
|
+
IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseDotV1,
|
203
|
+
IndexingConfigVectorizer_TextEmbedding3Large,
|
204
|
+
IndexingConfigVectorizer_TextEmbedding3Small,
|
205
|
+
IndexingConfigVectorizer_TextEmbeddingAda002,
|
206
|
+
)
|
207
|
+
from .indexing_config_vectorizer_request import (
|
208
|
+
IndexingConfigVectorizerRequest,
|
209
|
+
IndexingConfigVectorizerRequest_HkunlpInstructorXl,
|
210
|
+
IndexingConfigVectorizerRequest_IntfloatMultilingualE5Large,
|
211
|
+
IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseCosV1,
|
212
|
+
IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseDotV1,
|
213
|
+
IndexingConfigVectorizerRequest_TextEmbedding3Large,
|
214
|
+
IndexingConfigVectorizerRequest_TextEmbedding3Small,
|
215
|
+
IndexingConfigVectorizerRequest_TextEmbeddingAda002,
|
216
|
+
)
|
157
217
|
from .indexing_state_enum import IndexingStateEnum
|
158
218
|
from .initiated_enum import InitiatedEnum
|
159
219
|
from .initiated_execute_prompt_event import InitiatedExecutePromptEvent
|
160
220
|
from .initiated_prompt_execution_meta import InitiatedPromptExecutionMeta
|
161
221
|
from .initiated_workflow_node_result_event import InitiatedWorkflowNodeResultEvent
|
222
|
+
from .instructor_vectorizer_config import InstructorVectorizerConfig
|
223
|
+
from .instructor_vectorizer_config_request import InstructorVectorizerConfigRequest
|
224
|
+
from .intfloat_multilingual_e_5_large_enum import IntfloatMultilingualE5LargeEnum
|
162
225
|
from .json_enum import JsonEnum
|
163
226
|
from .json_input_request import JsonInputRequest
|
164
227
|
from .json_variable_value import JsonVariableValue
|
228
|
+
from .json_vellum_value import JsonVellumValue
|
165
229
|
from .logical_operator import LogicalOperator
|
166
230
|
from .logprobs_enum import LogprobsEnum
|
167
231
|
from .metadata_filter_config_request import MetadataFilterConfigRequest
|
168
232
|
from .metadata_filter_rule_combinator import MetadataFilterRuleCombinator
|
169
233
|
from .metadata_filter_rule_request import MetadataFilterRuleRequest
|
234
|
+
from .metric_enum import MetricEnum
|
235
|
+
from .metric_node_result import MetricNodeResult
|
170
236
|
from .ml_model_usage import MlModelUsage
|
171
237
|
from .named_scenario_input_chat_history_variable_value_request import NamedScenarioInputChatHistoryVariableValueRequest
|
172
238
|
from .named_scenario_input_request import (
|
@@ -231,7 +297,7 @@ from .node_input_variable_compiled_value import (
|
|
231
297
|
from .node_output_compiled_array_value import NodeOutputCompiledArrayValue
|
232
298
|
from .node_output_compiled_chat_history_value import NodeOutputCompiledChatHistoryValue
|
233
299
|
from .node_output_compiled_error_value import NodeOutputCompiledErrorValue
|
234
|
-
from .
|
300
|
+
from .node_output_compiled_function_call_value import NodeOutputCompiledFunctionCallValue
|
235
301
|
from .node_output_compiled_json_value import NodeOutputCompiledJsonValue
|
236
302
|
from .node_output_compiled_number_value import NodeOutputCompiledNumberValue
|
237
303
|
from .node_output_compiled_search_results_value import NodeOutputCompiledSearchResultsValue
|
@@ -251,6 +317,15 @@ from .normalized_log_probs import NormalizedLogProbs
|
|
251
317
|
from .normalized_token_log_probs import NormalizedTokenLogProbs
|
252
318
|
from .number_enum import NumberEnum
|
253
319
|
from .number_variable_value import NumberVariableValue
|
320
|
+
from .number_vellum_value import NumberVellumValue
|
321
|
+
from .open_ai_vectorizer_config import OpenAiVectorizerConfig
|
322
|
+
from .open_ai_vectorizer_config_request import OpenAiVectorizerConfigRequest
|
323
|
+
from .open_ai_vectorizer_text_embedding_3_large import OpenAiVectorizerTextEmbedding3Large
|
324
|
+
from .open_ai_vectorizer_text_embedding_3_large_request import OpenAiVectorizerTextEmbedding3LargeRequest
|
325
|
+
from .open_ai_vectorizer_text_embedding_3_small import OpenAiVectorizerTextEmbedding3Small
|
326
|
+
from .open_ai_vectorizer_text_embedding_3_small_request import OpenAiVectorizerTextEmbedding3SmallRequest
|
327
|
+
from .open_ai_vectorizer_text_embedding_ada_002 import OpenAiVectorizerTextEmbeddingAda002
|
328
|
+
from .open_ai_vectorizer_text_embedding_ada_002_request import OpenAiVectorizerTextEmbeddingAda002Request
|
254
329
|
from .paginated_document_index_read_list import PaginatedDocumentIndexReadList
|
255
330
|
from .paginated_slim_deployment_read_list import PaginatedSlimDeploymentReadList
|
256
331
|
from .paginated_slim_document_list import PaginatedSlimDocumentList
|
@@ -277,11 +352,15 @@ from .prompt_output import (
|
|
277
352
|
PromptOutput_String,
|
278
353
|
)
|
279
354
|
from .raw_prompt_execution_overrides_request import RawPromptExecutionOverridesRequest
|
355
|
+
from .reducto_chunker_config import ReductoChunkerConfig
|
356
|
+
from .reducto_chunker_config_request import ReductoChunkerConfigRequest
|
357
|
+
from .reducto_chunker_enum import ReductoChunkerEnum
|
358
|
+
from .reducto_chunking import ReductoChunking
|
359
|
+
from .reducto_chunking_request import ReductoChunkingRequest
|
280
360
|
from .rejected_enum import RejectedEnum
|
281
361
|
from .rejected_execute_prompt_event import RejectedExecutePromptEvent
|
282
362
|
from .rejected_execute_prompt_response import RejectedExecutePromptResponse
|
283
363
|
from .rejected_execute_workflow_workflow_result_event import RejectedExecuteWorkflowWorkflowResultEvent
|
284
|
-
from .rejected_function_call import RejectedFunctionCall
|
285
364
|
from .rejected_prompt_execution_meta import RejectedPromptExecutionMeta
|
286
365
|
from .rejected_workflow_node_result_event import RejectedWorkflowNodeResultEvent
|
287
366
|
from .sandbox_scenario import SandboxScenario
|
@@ -300,8 +379,14 @@ from .search_result_document_request import SearchResultDocumentRequest
|
|
300
379
|
from .search_result_merging_request import SearchResultMergingRequest
|
301
380
|
from .search_result_request import SearchResultRequest
|
302
381
|
from .search_results_enum import SearchResultsEnum
|
303
|
-
from .search_results_variable_value import SearchResultsVariableValue
|
304
382
|
from .search_weights_request import SearchWeightsRequest
|
383
|
+
from .sentence_chunker_config import SentenceChunkerConfig
|
384
|
+
from .sentence_chunker_config_request import SentenceChunkerConfigRequest
|
385
|
+
from .sentence_chunker_enum import SentenceChunkerEnum
|
386
|
+
from .sentence_chunking import SentenceChunking
|
387
|
+
from .sentence_chunking_request import SentenceChunkingRequest
|
388
|
+
from .sentence_transformers_multi_qa_mpnet_base_cos_v_1_enum import SentenceTransformersMultiQaMpnetBaseCosV1Enum
|
389
|
+
from .sentence_transformers_multi_qa_mpnet_base_dot_v_1_enum import SentenceTransformersMultiQaMpnetBaseDotV1Enum
|
305
390
|
from .slim_deployment_read import SlimDeploymentRead
|
306
391
|
from .slim_document import SlimDocument
|
307
392
|
from .slim_workflow_deployment import SlimWorkflowDeployment
|
@@ -314,6 +399,7 @@ from .string_chat_message_content_request import StringChatMessageContentRequest
|
|
314
399
|
from .string_enum import StringEnum
|
315
400
|
from .string_input_request import StringInputRequest
|
316
401
|
from .string_variable_value import StringVariableValue
|
402
|
+
from .string_vellum_value import StringVellumValue
|
317
403
|
from .submit_completion_actual_request import SubmitCompletionActualRequest
|
318
404
|
from .submit_completion_actuals_error_response import SubmitCompletionActualsErrorResponse
|
319
405
|
from .submit_workflow_execution_actual_request import (
|
@@ -452,6 +538,14 @@ from .test_suite_run_workflow_release_tag_exec_config_data_request import (
|
|
452
538
|
from .test_suite_run_workflow_release_tag_exec_config_request import TestSuiteRunWorkflowReleaseTagExecConfigRequest
|
453
539
|
from .test_suite_run_workflow_release_tag_exec_config_type_enum import TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum
|
454
540
|
from .test_suite_test_case import TestSuiteTestCase
|
541
|
+
from .text_embedding_3_large_enum import TextEmbedding3LargeEnum
|
542
|
+
from .text_embedding_3_small_enum import TextEmbedding3SmallEnum
|
543
|
+
from .text_embedding_ada_002_enum import TextEmbeddingAda002Enum
|
544
|
+
from .token_overlapping_window_chunker_config import TokenOverlappingWindowChunkerConfig
|
545
|
+
from .token_overlapping_window_chunker_config_request import TokenOverlappingWindowChunkerConfigRequest
|
546
|
+
from .token_overlapping_window_chunker_enum import TokenOverlappingWindowChunkerEnum
|
547
|
+
from .token_overlapping_window_chunking import TokenOverlappingWindowChunking
|
548
|
+
from .token_overlapping_window_chunking_request import TokenOverlappingWindowChunkingRequest
|
455
549
|
from .upload_document_error_response import UploadDocumentErrorResponse
|
456
550
|
from .upload_document_response import UploadDocumentResponse
|
457
551
|
from .vellum_error import VellumError
|
@@ -475,6 +569,7 @@ from .workflow_node_result_data import (
|
|
475
569
|
WorkflowNodeResultData_Api,
|
476
570
|
WorkflowNodeResultData_CodeExecution,
|
477
571
|
WorkflowNodeResultData_Conditional,
|
572
|
+
WorkflowNodeResultData_Metric,
|
478
573
|
WorkflowNodeResultData_Prompt,
|
479
574
|
WorkflowNodeResultData_Search,
|
480
575
|
WorkflowNodeResultData_Subworkflow,
|
@@ -544,6 +639,7 @@ from .workflow_result_event_output_data_string import WorkflowResultEventOutputD
|
|
544
639
|
from .workflow_stream_event import WorkflowStreamEvent, WorkflowStreamEvent_Node, WorkflowStreamEvent_Workflow
|
545
640
|
|
546
641
|
__all__ = [
|
642
|
+
"AddOpenaiApiKeyEnum",
|
547
643
|
"ApiNodeResult",
|
548
644
|
"ApiNodeResultData",
|
549
645
|
"ArrayChatMessageContent",
|
@@ -558,17 +654,27 @@ __all__ = [
|
|
558
654
|
"ArrayChatMessageContentRequest",
|
559
655
|
"ArrayEnum",
|
560
656
|
"ArrayVariableValueItem",
|
561
|
-
"ArrayVariableValueItem_ChatHistory",
|
562
657
|
"ArrayVariableValueItem_Error",
|
563
658
|
"ArrayVariableValueItem_FunctionCall",
|
564
659
|
"ArrayVariableValueItem_Image",
|
565
660
|
"ArrayVariableValueItem_Json",
|
566
661
|
"ArrayVariableValueItem_Number",
|
567
|
-
"ArrayVariableValueItem_SearchResults",
|
568
662
|
"ArrayVariableValueItem_String",
|
663
|
+
"ArrayVellumValueItem",
|
664
|
+
"ArrayVellumValueItem_Error",
|
665
|
+
"ArrayVellumValueItem_FunctionCall",
|
666
|
+
"ArrayVellumValueItem_Image",
|
667
|
+
"ArrayVellumValueItem_Json",
|
668
|
+
"ArrayVellumValueItem_Number",
|
669
|
+
"ArrayVellumValueItem_String",
|
670
|
+
"BasicVectorizerIntfloatMultilingualE5Large",
|
671
|
+
"BasicVectorizerIntfloatMultilingualE5LargeRequest",
|
672
|
+
"BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1",
|
673
|
+
"BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1Request",
|
674
|
+
"BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1",
|
675
|
+
"BasicVectorizerSentenceTransformersMultiQaMpnetBaseDotV1Request",
|
569
676
|
"ChatHistoryEnum",
|
570
677
|
"ChatHistoryInputRequest",
|
571
|
-
"ChatHistoryVariableValue",
|
572
678
|
"ChatMessage",
|
573
679
|
"ChatMessageContent",
|
574
680
|
"ChatMessageContentRequest",
|
@@ -606,6 +712,16 @@ __all__ = [
|
|
606
712
|
"DeploymentProviderPayloadResponse",
|
607
713
|
"DeploymentRead",
|
608
714
|
"DocumentDocumentToDocumentIndex",
|
715
|
+
"DocumentIndexChunking",
|
716
|
+
"DocumentIndexChunkingRequest",
|
717
|
+
"DocumentIndexChunkingRequest_ReductoChunker",
|
718
|
+
"DocumentIndexChunkingRequest_SentenceChunker",
|
719
|
+
"DocumentIndexChunkingRequest_TokenOverlappingWindowChunker",
|
720
|
+
"DocumentIndexChunking_ReductoChunker",
|
721
|
+
"DocumentIndexChunking_SentenceChunker",
|
722
|
+
"DocumentIndexChunking_TokenOverlappingWindowChunker",
|
723
|
+
"DocumentIndexIndexingConfig",
|
724
|
+
"DocumentIndexIndexingConfigRequest",
|
609
725
|
"DocumentIndexRead",
|
610
726
|
"DocumentRead",
|
611
727
|
"DocumentStatus",
|
@@ -614,6 +730,7 @@ __all__ = [
|
|
614
730
|
"EnvironmentEnum",
|
615
731
|
"ErrorEnum",
|
616
732
|
"ErrorVariableValue",
|
733
|
+
"ErrorVellumValue",
|
617
734
|
"ExecutePromptApiErrorResponse",
|
618
735
|
"ExecutePromptEvent",
|
619
736
|
"ExecutePromptEvent_Fulfilled",
|
@@ -653,8 +770,6 @@ __all__ = [
|
|
653
770
|
"FulfilledExecutePromptEvent",
|
654
771
|
"FulfilledExecutePromptResponse",
|
655
772
|
"FulfilledExecuteWorkflowWorkflowResultEvent",
|
656
|
-
"FulfilledFunctionCall",
|
657
|
-
"FulfilledFunctionCallRequest",
|
658
773
|
"FulfilledPromptExecutionMeta",
|
659
774
|
"FulfilledWorkflowNodeResultEvent",
|
660
775
|
"FunctionCall",
|
@@ -663,9 +778,9 @@ __all__ = [
|
|
663
778
|
"FunctionCallChatMessageContentValue",
|
664
779
|
"FunctionCallChatMessageContentValueRequest",
|
665
780
|
"FunctionCallEnum",
|
781
|
+
"FunctionCallRequest",
|
666
782
|
"FunctionCallVariableValue",
|
667
|
-
"
|
668
|
-
"FunctionCall_Rejected",
|
783
|
+
"FunctionCallVellumValue",
|
669
784
|
"GenerateErrorResponse",
|
670
785
|
"GenerateOptionsRequest",
|
671
786
|
"GenerateRequest",
|
@@ -676,23 +791,49 @@ __all__ = [
|
|
676
791
|
"GenerateStreamResponse",
|
677
792
|
"GenerateStreamResult",
|
678
793
|
"GenerateStreamResultData",
|
794
|
+
"HkunlpInstructorXlEnum",
|
795
|
+
"HkunlpInstructorXlVectorizer",
|
796
|
+
"HkunlpInstructorXlVectorizerRequest",
|
679
797
|
"ImageChatMessageContent",
|
680
798
|
"ImageChatMessageContentRequest",
|
681
799
|
"ImageEnum",
|
682
800
|
"ImageVariableValue",
|
801
|
+
"ImageVellumValue",
|
802
|
+
"IndexingConfigVectorizer",
|
803
|
+
"IndexingConfigVectorizerRequest",
|
804
|
+
"IndexingConfigVectorizerRequest_HkunlpInstructorXl",
|
805
|
+
"IndexingConfigVectorizerRequest_IntfloatMultilingualE5Large",
|
806
|
+
"IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseCosV1",
|
807
|
+
"IndexingConfigVectorizerRequest_SentenceTransformersMultiQaMpnetBaseDotV1",
|
808
|
+
"IndexingConfigVectorizerRequest_TextEmbedding3Large",
|
809
|
+
"IndexingConfigVectorizerRequest_TextEmbedding3Small",
|
810
|
+
"IndexingConfigVectorizerRequest_TextEmbeddingAda002",
|
811
|
+
"IndexingConfigVectorizer_HkunlpInstructorXl",
|
812
|
+
"IndexingConfigVectorizer_IntfloatMultilingualE5Large",
|
813
|
+
"IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseCosV1",
|
814
|
+
"IndexingConfigVectorizer_SentenceTransformersMultiQaMpnetBaseDotV1",
|
815
|
+
"IndexingConfigVectorizer_TextEmbedding3Large",
|
816
|
+
"IndexingConfigVectorizer_TextEmbedding3Small",
|
817
|
+
"IndexingConfigVectorizer_TextEmbeddingAda002",
|
683
818
|
"IndexingStateEnum",
|
684
819
|
"InitiatedEnum",
|
685
820
|
"InitiatedExecutePromptEvent",
|
686
821
|
"InitiatedPromptExecutionMeta",
|
687
822
|
"InitiatedWorkflowNodeResultEvent",
|
823
|
+
"InstructorVectorizerConfig",
|
824
|
+
"InstructorVectorizerConfigRequest",
|
825
|
+
"IntfloatMultilingualE5LargeEnum",
|
688
826
|
"JsonEnum",
|
689
827
|
"JsonInputRequest",
|
690
828
|
"JsonVariableValue",
|
829
|
+
"JsonVellumValue",
|
691
830
|
"LogicalOperator",
|
692
831
|
"LogprobsEnum",
|
693
832
|
"MetadataFilterConfigRequest",
|
694
833
|
"MetadataFilterRuleCombinator",
|
695
834
|
"MetadataFilterRuleRequest",
|
835
|
+
"MetricEnum",
|
836
|
+
"MetricNodeResult",
|
696
837
|
"MlModelUsage",
|
697
838
|
"NamedScenarioInputChatHistoryVariableValueRequest",
|
698
839
|
"NamedScenarioInputRequest",
|
@@ -749,7 +890,7 @@ __all__ = [
|
|
749
890
|
"NodeOutputCompiledArrayValue",
|
750
891
|
"NodeOutputCompiledChatHistoryValue",
|
751
892
|
"NodeOutputCompiledErrorValue",
|
752
|
-
"
|
893
|
+
"NodeOutputCompiledFunctionCallValue",
|
753
894
|
"NodeOutputCompiledJsonValue",
|
754
895
|
"NodeOutputCompiledNumberValue",
|
755
896
|
"NodeOutputCompiledSearchResultsValue",
|
@@ -767,6 +908,15 @@ __all__ = [
|
|
767
908
|
"NormalizedTokenLogProbs",
|
768
909
|
"NumberEnum",
|
769
910
|
"NumberVariableValue",
|
911
|
+
"NumberVellumValue",
|
912
|
+
"OpenAiVectorizerConfig",
|
913
|
+
"OpenAiVectorizerConfigRequest",
|
914
|
+
"OpenAiVectorizerTextEmbedding3Large",
|
915
|
+
"OpenAiVectorizerTextEmbedding3LargeRequest",
|
916
|
+
"OpenAiVectorizerTextEmbedding3Small",
|
917
|
+
"OpenAiVectorizerTextEmbedding3SmallRequest",
|
918
|
+
"OpenAiVectorizerTextEmbeddingAda002",
|
919
|
+
"OpenAiVectorizerTextEmbeddingAda002Request",
|
770
920
|
"PaginatedDocumentIndexReadList",
|
771
921
|
"PaginatedSlimDeploymentReadList",
|
772
922
|
"PaginatedSlimDocumentList",
|
@@ -789,11 +939,15 @@ __all__ = [
|
|
789
939
|
"PromptOutput_Json",
|
790
940
|
"PromptOutput_String",
|
791
941
|
"RawPromptExecutionOverridesRequest",
|
942
|
+
"ReductoChunkerConfig",
|
943
|
+
"ReductoChunkerConfigRequest",
|
944
|
+
"ReductoChunkerEnum",
|
945
|
+
"ReductoChunking",
|
946
|
+
"ReductoChunkingRequest",
|
792
947
|
"RejectedEnum",
|
793
948
|
"RejectedExecutePromptEvent",
|
794
949
|
"RejectedExecutePromptResponse",
|
795
950
|
"RejectedExecuteWorkflowWorkflowResultEvent",
|
796
|
-
"RejectedFunctionCall",
|
797
951
|
"RejectedPromptExecutionMeta",
|
798
952
|
"RejectedWorkflowNodeResultEvent",
|
799
953
|
"SandboxScenario",
|
@@ -814,8 +968,14 @@ __all__ = [
|
|
814
968
|
"SearchResultMergingRequest",
|
815
969
|
"SearchResultRequest",
|
816
970
|
"SearchResultsEnum",
|
817
|
-
"SearchResultsVariableValue",
|
818
971
|
"SearchWeightsRequest",
|
972
|
+
"SentenceChunkerConfig",
|
973
|
+
"SentenceChunkerConfigRequest",
|
974
|
+
"SentenceChunkerEnum",
|
975
|
+
"SentenceChunking",
|
976
|
+
"SentenceChunkingRequest",
|
977
|
+
"SentenceTransformersMultiQaMpnetBaseCosV1Enum",
|
978
|
+
"SentenceTransformersMultiQaMpnetBaseDotV1Enum",
|
819
979
|
"SlimDeploymentRead",
|
820
980
|
"SlimDocument",
|
821
981
|
"SlimWorkflowDeployment",
|
@@ -828,6 +988,7 @@ __all__ = [
|
|
828
988
|
"StringEnum",
|
829
989
|
"StringInputRequest",
|
830
990
|
"StringVariableValue",
|
991
|
+
"StringVellumValue",
|
831
992
|
"SubmitCompletionActualRequest",
|
832
993
|
"SubmitCompletionActualsErrorResponse",
|
833
994
|
"SubmitWorkflowExecutionActualRequest",
|
@@ -944,6 +1105,14 @@ __all__ = [
|
|
944
1105
|
"TestSuiteRunWorkflowReleaseTagExecConfigRequest",
|
945
1106
|
"TestSuiteRunWorkflowReleaseTagExecConfigTypeEnum",
|
946
1107
|
"TestSuiteTestCase",
|
1108
|
+
"TextEmbedding3LargeEnum",
|
1109
|
+
"TextEmbedding3SmallEnum",
|
1110
|
+
"TextEmbeddingAda002Enum",
|
1111
|
+
"TokenOverlappingWindowChunkerConfig",
|
1112
|
+
"TokenOverlappingWindowChunkerConfigRequest",
|
1113
|
+
"TokenOverlappingWindowChunkerEnum",
|
1114
|
+
"TokenOverlappingWindowChunking",
|
1115
|
+
"TokenOverlappingWindowChunkingRequest",
|
947
1116
|
"UploadDocumentErrorResponse",
|
948
1117
|
"UploadDocumentResponse",
|
949
1118
|
"VellumError",
|
@@ -966,6 +1135,7 @@ __all__ = [
|
|
966
1135
|
"WorkflowNodeResultData_Api",
|
967
1136
|
"WorkflowNodeResultData_CodeExecution",
|
968
1137
|
"WorkflowNodeResultData_Conditional",
|
1138
|
+
"WorkflowNodeResultData_Metric",
|
969
1139
|
"WorkflowNodeResultData_Prompt",
|
970
1140
|
"WorkflowNodeResultData_Search",
|
971
1141
|
"WorkflowNodeResultData_Subworkflow",
|
@@ -4,13 +4,11 @@ from __future__ import annotations
|
|
4
4
|
|
5
5
|
import typing
|
6
6
|
|
7
|
-
from .chat_history_variable_value import ChatHistoryVariableValue
|
8
7
|
from .error_variable_value import ErrorVariableValue
|
9
8
|
from .function_call_variable_value import FunctionCallVariableValue
|
10
9
|
from .image_variable_value import ImageVariableValue
|
11
10
|
from .json_variable_value import JsonVariableValue
|
12
11
|
from .number_variable_value import NumberVariableValue
|
13
|
-
from .search_results_variable_value import SearchResultsVariableValue
|
14
12
|
from .string_variable_value import StringVariableValue
|
15
13
|
|
16
14
|
|
@@ -44,26 +42,6 @@ class ArrayVariableValueItem_Json(JsonVariableValue):
|
|
44
42
|
populate_by_name = True
|
45
43
|
|
46
44
|
|
47
|
-
class ArrayVariableValueItem_ChatHistory(ChatHistoryVariableValue):
|
48
|
-
type: typing.Literal["CHAT_HISTORY"] = "CHAT_HISTORY"
|
49
|
-
|
50
|
-
class Config:
|
51
|
-
frozen = True
|
52
|
-
smart_union = True
|
53
|
-
allow_population_by_field_name = True
|
54
|
-
populate_by_name = True
|
55
|
-
|
56
|
-
|
57
|
-
class ArrayVariableValueItem_SearchResults(SearchResultsVariableValue):
|
58
|
-
type: typing.Literal["SEARCH_RESULTS"] = "SEARCH_RESULTS"
|
59
|
-
|
60
|
-
class Config:
|
61
|
-
frozen = True
|
62
|
-
smart_union = True
|
63
|
-
allow_population_by_field_name = True
|
64
|
-
populate_by_name = True
|
65
|
-
|
66
|
-
|
67
45
|
class ArrayVariableValueItem_Error(ErrorVariableValue):
|
68
46
|
type: typing.Literal["ERROR"] = "ERROR"
|
69
47
|
|
@@ -98,8 +76,6 @@ ArrayVariableValueItem = typing.Union[
|
|
98
76
|
ArrayVariableValueItem_String,
|
99
77
|
ArrayVariableValueItem_Number,
|
100
78
|
ArrayVariableValueItem_Json,
|
101
|
-
ArrayVariableValueItem_ChatHistory,
|
102
|
-
ArrayVariableValueItem_SearchResults,
|
103
79
|
ArrayVariableValueItem_Error,
|
104
80
|
ArrayVariableValueItem_FunctionCall,
|
105
81
|
ArrayVariableValueItem_Image,
|
@@ -0,0 +1,82 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
from __future__ import annotations
|
4
|
+
|
5
|
+
import typing
|
6
|
+
|
7
|
+
from .error_vellum_value import ErrorVellumValue
|
8
|
+
from .function_call_vellum_value import FunctionCallVellumValue
|
9
|
+
from .image_vellum_value import ImageVellumValue
|
10
|
+
from .json_vellum_value import JsonVellumValue
|
11
|
+
from .number_vellum_value import NumberVellumValue
|
12
|
+
from .string_vellum_value import StringVellumValue
|
13
|
+
|
14
|
+
|
15
|
+
class ArrayVellumValueItem_String(StringVellumValue):
|
16
|
+
type: typing.Literal["STRING"] = "STRING"
|
17
|
+
|
18
|
+
class Config:
|
19
|
+
frozen = True
|
20
|
+
smart_union = True
|
21
|
+
allow_population_by_field_name = True
|
22
|
+
populate_by_name = True
|
23
|
+
|
24
|
+
|
25
|
+
class ArrayVellumValueItem_Number(NumberVellumValue):
|
26
|
+
type: typing.Literal["NUMBER"] = "NUMBER"
|
27
|
+
|
28
|
+
class Config:
|
29
|
+
frozen = True
|
30
|
+
smart_union = True
|
31
|
+
allow_population_by_field_name = True
|
32
|
+
populate_by_name = True
|
33
|
+
|
34
|
+
|
35
|
+
class ArrayVellumValueItem_Json(JsonVellumValue):
|
36
|
+
type: typing.Literal["JSON"] = "JSON"
|
37
|
+
|
38
|
+
class Config:
|
39
|
+
frozen = True
|
40
|
+
smart_union = True
|
41
|
+
allow_population_by_field_name = True
|
42
|
+
populate_by_name = True
|
43
|
+
|
44
|
+
|
45
|
+
class ArrayVellumValueItem_Image(ImageVellumValue):
|
46
|
+
type: typing.Literal["IMAGE"] = "IMAGE"
|
47
|
+
|
48
|
+
class Config:
|
49
|
+
frozen = True
|
50
|
+
smart_union = True
|
51
|
+
allow_population_by_field_name = True
|
52
|
+
populate_by_name = True
|
53
|
+
|
54
|
+
|
55
|
+
class ArrayVellumValueItem_FunctionCall(FunctionCallVellumValue):
|
56
|
+
type: typing.Literal["FUNCTION_CALL"] = "FUNCTION_CALL"
|
57
|
+
|
58
|
+
class Config:
|
59
|
+
frozen = True
|
60
|
+
smart_union = True
|
61
|
+
allow_population_by_field_name = True
|
62
|
+
populate_by_name = True
|
63
|
+
|
64
|
+
|
65
|
+
class ArrayVellumValueItem_Error(ErrorVellumValue):
|
66
|
+
type: typing.Literal["ERROR"] = "ERROR"
|
67
|
+
|
68
|
+
class Config:
|
69
|
+
frozen = True
|
70
|
+
smart_union = True
|
71
|
+
allow_population_by_field_name = True
|
72
|
+
populate_by_name = True
|
73
|
+
|
74
|
+
|
75
|
+
ArrayVellumValueItem = typing.Union[
|
76
|
+
ArrayVellumValueItem_String,
|
77
|
+
ArrayVellumValueItem_Number,
|
78
|
+
ArrayVellumValueItem_Json,
|
79
|
+
ArrayVellumValueItem_Image,
|
80
|
+
ArrayVellumValueItem_FunctionCall,
|
81
|
+
ArrayVellumValueItem_Error,
|
82
|
+
]
|
@@ -0,0 +1,29 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
import datetime as dt
|
4
|
+
import typing
|
5
|
+
|
6
|
+
from ..core.datetime_utils import serialize_datetime
|
7
|
+
from ..core.pydantic_utilities import pydantic_v1
|
8
|
+
|
9
|
+
|
10
|
+
class BasicVectorizerIntfloatMultilingualE5Large(pydantic_v1.BaseModel):
|
11
|
+
"""
|
12
|
+
Basic vectorizer for intfloat/multilingual-e5-large.
|
13
|
+
"""
|
14
|
+
|
15
|
+
config: typing.Optional[typing.Dict[str, typing.Any]] = None
|
16
|
+
|
17
|
+
def json(self, **kwargs: typing.Any) -> str:
|
18
|
+
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
19
|
+
return super().json(**kwargs_with_defaults)
|
20
|
+
|
21
|
+
def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
|
22
|
+
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
23
|
+
return super().dict(**kwargs_with_defaults)
|
24
|
+
|
25
|
+
class Config:
|
26
|
+
frozen = True
|
27
|
+
smart_union = True
|
28
|
+
extra = pydantic_v1.Extra.allow
|
29
|
+
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -0,0 +1,29 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
import datetime as dt
|
4
|
+
import typing
|
5
|
+
|
6
|
+
from ..core.datetime_utils import serialize_datetime
|
7
|
+
from ..core.pydantic_utilities import pydantic_v1
|
8
|
+
|
9
|
+
|
10
|
+
class BasicVectorizerIntfloatMultilingualE5LargeRequest(pydantic_v1.BaseModel):
|
11
|
+
"""
|
12
|
+
Basic vectorizer for intfloat/multilingual-e5-large.
|
13
|
+
"""
|
14
|
+
|
15
|
+
config: typing.Optional[typing.Dict[str, typing.Any]] = None
|
16
|
+
|
17
|
+
def json(self, **kwargs: typing.Any) -> str:
|
18
|
+
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
19
|
+
return super().json(**kwargs_with_defaults)
|
20
|
+
|
21
|
+
def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
|
22
|
+
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
23
|
+
return super().dict(**kwargs_with_defaults)
|
24
|
+
|
25
|
+
class Config:
|
26
|
+
frozen = True
|
27
|
+
smart_union = True
|
28
|
+
extra = pydantic_v1.Extra.allow
|
29
|
+
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -0,0 +1,29 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
import datetime as dt
|
4
|
+
import typing
|
5
|
+
|
6
|
+
from ..core.datetime_utils import serialize_datetime
|
7
|
+
from ..core.pydantic_utilities import pydantic_v1
|
8
|
+
|
9
|
+
|
10
|
+
class BasicVectorizerSentenceTransformersMultiQaMpnetBaseCosV1(pydantic_v1.BaseModel):
|
11
|
+
"""
|
12
|
+
Basic vectorizer for sentence-transformers/multi-qa-mpnet-base-cos-v1.
|
13
|
+
"""
|
14
|
+
|
15
|
+
config: typing.Optional[typing.Dict[str, typing.Any]] = None
|
16
|
+
|
17
|
+
def json(self, **kwargs: typing.Any) -> str:
|
18
|
+
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
19
|
+
return super().json(**kwargs_with_defaults)
|
20
|
+
|
21
|
+
def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
|
22
|
+
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
23
|
+
return super().dict(**kwargs_with_defaults)
|
24
|
+
|
25
|
+
class Config:
|
26
|
+
frozen = True
|
27
|
+
smart_union = True
|
28
|
+
extra = pydantic_v1.Extra.allow
|
29
|
+
json_encoders = {dt.datetime: serialize_datetime}
|