llama-cloud 0.0.13__py3-none-any.whl → 0.0.14__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of llama-cloud might be problematic. Click here for more details.
- llama_cloud/__init__.py +76 -14
- llama_cloud/resources/__init__.py +14 -0
- llama_cloud/resources/data_sources/types/data_source_update_component_one.py +2 -0
- llama_cloud/resources/evals/client.py +5 -5
- llama_cloud/resources/parsing/client.py +8 -0
- llama_cloud/resources/pipelines/__init__.py +14 -0
- llama_cloud/resources/pipelines/client.py +115 -66
- llama_cloud/resources/pipelines/types/__init__.py +16 -0
- llama_cloud/resources/pipelines/types/pipeline_update_embedding_config.py +78 -0
- llama_cloud/types/__init__.py +68 -14
- llama_cloud/types/{embedding_config.py → azure_open_ai_embedding_config.py} +4 -6
- llama_cloud/types/bedrock_embedding_config.py +34 -0
- llama_cloud/types/box_auth_mechanism.py +21 -0
- llama_cloud/types/chat_data.py +1 -1
- llama_cloud/types/chat_message.py +14 -4
- llama_cloud/types/cloud_azure_ai_search_vector_store.py +3 -0
- llama_cloud/types/cloud_box_data_source.py +51 -0
- llama_cloud/types/cloud_document.py +3 -0
- llama_cloud/types/cloud_document_create.py +3 -0
- llama_cloud/types/cloud_sharepoint_data_source.py +2 -1
- llama_cloud/types/cohere_embedding_config.py +34 -0
- llama_cloud/types/configurable_data_source_names.py +4 -0
- llama_cloud/types/custom_claims.py +0 -3
- llama_cloud/types/data_source_component_one.py +2 -0
- llama_cloud/types/data_source_create_component_one.py +2 -0
- llama_cloud/types/eval_execution_params.py +2 -2
- llama_cloud/types/eval_execution_params_override.py +2 -2
- llama_cloud/types/filter_operator.py +4 -0
- llama_cloud/types/gemini_embedding_config.py +34 -0
- llama_cloud/types/hugging_face_inference_api_embedding_config.py +34 -0
- llama_cloud/types/input_message.py +42 -0
- llama_cloud/types/llama_parse_parameters.py +4 -1
- llama_cloud/types/{eval_llm_model_data.py → llm_model_data.py} +1 -1
- llama_cloud/types/llm_parameters.py +2 -2
- llama_cloud/types/{supported_eval_llm_model.py → message_annotation.py} +6 -6
- llama_cloud/types/metadata_filter.py +1 -1
- llama_cloud/types/open_ai_embedding_config.py +34 -0
- llama_cloud/types/page_segmentation_config.py +2 -0
- llama_cloud/types/pipeline.py +11 -1
- llama_cloud/types/pipeline_create.py +3 -3
- llama_cloud/types/pipeline_create_embedding_config.py +78 -0
- llama_cloud/types/pipeline_data_source_component_one.py +2 -0
- llama_cloud/types/pipeline_embedding_config.py +78 -0
- llama_cloud/types/pipeline_transform_config.py +31 -0
- llama_cloud/types/playground_session.py +51 -0
- llama_cloud/types/supported_llm_model.py +41 -0
- llama_cloud/types/{supported_eval_llm_model_names.py → supported_llm_model_names.py} +10 -6
- {llama_cloud-0.0.13.dist-info → llama_cloud-0.0.14.dist-info}/METADATA +1 -1
- {llama_cloud-0.0.13.dist-info → llama_cloud-0.0.14.dist-info}/RECORD +51 -40
- llama_cloud/types/embedding_config_component.py +0 -7
- llama_cloud/types/embedding_config_component_one.py +0 -19
- llama_cloud/types/embedding_config_type.py +0 -41
- {llama_cloud-0.0.13.dist-info → llama_cloud-0.0.14.dist-info}/LICENSE +0 -0
- {llama_cloud-0.0.13.dist-info → llama_cloud-0.0.14.dist-info}/WHEEL +0 -0
llama_cloud/__init__.py
CHANGED
|
@@ -14,15 +14,19 @@ from .types import (
|
|
|
14
14
|
AdvancedModeTransformConfigSegmentationConfig_Page,
|
|
15
15
|
AutoTransformConfig,
|
|
16
16
|
AzureOpenAiEmbedding,
|
|
17
|
+
AzureOpenAiEmbeddingConfig,
|
|
17
18
|
Base,
|
|
18
19
|
BasePromptTemplate,
|
|
19
20
|
BedrockEmbedding,
|
|
21
|
+
BedrockEmbeddingConfig,
|
|
22
|
+
BoxAuthMechanism,
|
|
20
23
|
CharacterChunkingConfig,
|
|
21
24
|
CharacterSplitter,
|
|
22
25
|
ChatData,
|
|
23
26
|
ChatMessage,
|
|
24
27
|
CloudAzStorageBlobDataSource,
|
|
25
28
|
CloudAzureAiSearchVectorStore,
|
|
29
|
+
CloudBoxDataSource,
|
|
26
30
|
CloudChromaVectorStore,
|
|
27
31
|
CloudConfluenceDataSource,
|
|
28
32
|
CloudDocument,
|
|
@@ -41,6 +45,7 @@ from .types import (
|
|
|
41
45
|
CloudWeaviateVectorStore,
|
|
42
46
|
CodeSplitter,
|
|
43
47
|
CohereEmbedding,
|
|
48
|
+
CohereEmbeddingConfig,
|
|
44
49
|
ConfigurableDataSinkNames,
|
|
45
50
|
ConfigurableDataSourceNames,
|
|
46
51
|
ConfigurableTransformationDefinition,
|
|
@@ -66,16 +71,11 @@ from .types import (
|
|
|
66
71
|
DataSourceCustomMetadataValue,
|
|
67
72
|
DataSourceDefinition,
|
|
68
73
|
ElementSegmentationConfig,
|
|
69
|
-
EmbeddingConfig,
|
|
70
|
-
EmbeddingConfigComponent,
|
|
71
|
-
EmbeddingConfigComponentOne,
|
|
72
|
-
EmbeddingConfigType,
|
|
73
74
|
EvalDataset,
|
|
74
75
|
EvalDatasetJobParams,
|
|
75
76
|
EvalDatasetJobRecord,
|
|
76
77
|
EvalExecutionParams,
|
|
77
78
|
EvalExecutionParamsOverride,
|
|
78
|
-
EvalLlmModelData,
|
|
79
79
|
EvalQuestion,
|
|
80
80
|
EvalQuestionCreate,
|
|
81
81
|
EvalQuestionResult,
|
|
@@ -89,14 +89,18 @@ from .types import (
|
|
|
89
89
|
FilterCondition,
|
|
90
90
|
FilterOperator,
|
|
91
91
|
GeminiEmbedding,
|
|
92
|
+
GeminiEmbeddingConfig,
|
|
92
93
|
HttpValidationError,
|
|
93
94
|
HuggingFaceInferenceApiEmbedding,
|
|
95
|
+
HuggingFaceInferenceApiEmbeddingConfig,
|
|
94
96
|
HuggingFaceInferenceApiEmbeddingToken,
|
|
95
97
|
IngestionErrorResponse,
|
|
98
|
+
InputMessage,
|
|
96
99
|
JobNameMapping,
|
|
97
100
|
LlamaParseParameters,
|
|
98
101
|
LlamaParseSupportedFileExtensions,
|
|
99
102
|
Llm,
|
|
103
|
+
LlmModelData,
|
|
100
104
|
LlmParameters,
|
|
101
105
|
LocalEval,
|
|
102
106
|
LocalEvalResults,
|
|
@@ -105,6 +109,7 @@ from .types import (
|
|
|
105
109
|
ManagedIngestionStatusResponse,
|
|
106
110
|
MarkdownElementNodeParser,
|
|
107
111
|
MarkdownNodeParser,
|
|
112
|
+
MessageAnnotation,
|
|
108
113
|
MessageRole,
|
|
109
114
|
MetadataFilter,
|
|
110
115
|
MetadataFilterValue,
|
|
@@ -116,6 +121,7 @@ from .types import (
|
|
|
116
121
|
NoneSegmentationConfig,
|
|
117
122
|
ObjectType,
|
|
118
123
|
OpenAiEmbedding,
|
|
124
|
+
OpenAiEmbeddingConfig,
|
|
119
125
|
Organization,
|
|
120
126
|
OrganizationCreate,
|
|
121
127
|
PageSegmentationConfig,
|
|
@@ -129,6 +135,13 @@ from .types import (
|
|
|
129
135
|
ParsingUsage,
|
|
130
136
|
Pipeline,
|
|
131
137
|
PipelineCreate,
|
|
138
|
+
PipelineCreateEmbeddingConfig,
|
|
139
|
+
PipelineCreateEmbeddingConfig_AzureEmbedding,
|
|
140
|
+
PipelineCreateEmbeddingConfig_BedrockEmbedding,
|
|
141
|
+
PipelineCreateEmbeddingConfig_CohereEmbedding,
|
|
142
|
+
PipelineCreateEmbeddingConfig_GeminiEmbedding,
|
|
143
|
+
PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding,
|
|
144
|
+
PipelineCreateEmbeddingConfig_OpenaiEmbedding,
|
|
132
145
|
PipelineCreateTransformConfig,
|
|
133
146
|
PipelineCreateTransformConfig_Advanced,
|
|
134
147
|
PipelineCreateTransformConfig_Auto,
|
|
@@ -138,12 +151,23 @@ from .types import (
|
|
|
138
151
|
PipelineDataSourceCreate,
|
|
139
152
|
PipelineDataSourceCustomMetadataValue,
|
|
140
153
|
PipelineDeployment,
|
|
154
|
+
PipelineEmbeddingConfig,
|
|
155
|
+
PipelineEmbeddingConfig_AzureEmbedding,
|
|
156
|
+
PipelineEmbeddingConfig_BedrockEmbedding,
|
|
157
|
+
PipelineEmbeddingConfig_CohereEmbedding,
|
|
158
|
+
PipelineEmbeddingConfig_GeminiEmbedding,
|
|
159
|
+
PipelineEmbeddingConfig_HuggingfaceApiEmbedding,
|
|
160
|
+
PipelineEmbeddingConfig_OpenaiEmbedding,
|
|
141
161
|
PipelineFile,
|
|
142
162
|
PipelineFileCreate,
|
|
143
163
|
PipelineFileCreateCustomMetadataValue,
|
|
144
164
|
PipelineFileCustomMetadataValue,
|
|
145
165
|
PipelineFileResourceInfoValue,
|
|
166
|
+
PipelineTransformConfig,
|
|
167
|
+
PipelineTransformConfig_Advanced,
|
|
168
|
+
PipelineTransformConfig_Auto,
|
|
146
169
|
PipelineType,
|
|
170
|
+
PlaygroundSession,
|
|
147
171
|
Pooling,
|
|
148
172
|
PresetRetrievalParams,
|
|
149
173
|
PresignedUrl,
|
|
@@ -159,8 +183,8 @@ from .types import (
|
|
|
159
183
|
SentenceChunkingConfig,
|
|
160
184
|
SentenceSplitter,
|
|
161
185
|
StatusEnum,
|
|
162
|
-
|
|
163
|
-
|
|
186
|
+
SupportedLlmModel,
|
|
187
|
+
SupportedLlmModelNames,
|
|
164
188
|
TextNode,
|
|
165
189
|
TextNodeRelationshipsValue,
|
|
166
190
|
TextNodeWithScore,
|
|
@@ -185,6 +209,13 @@ from .resources import (
|
|
|
185
209
|
ExtractionSchemaUpdateDataSchemaValue,
|
|
186
210
|
FileCreateResourceInfoValue,
|
|
187
211
|
PipelineFileUpdateCustomMetadataValue,
|
|
212
|
+
PipelineUpdateEmbeddingConfig,
|
|
213
|
+
PipelineUpdateEmbeddingConfig_AzureEmbedding,
|
|
214
|
+
PipelineUpdateEmbeddingConfig_BedrockEmbedding,
|
|
215
|
+
PipelineUpdateEmbeddingConfig_CohereEmbedding,
|
|
216
|
+
PipelineUpdateEmbeddingConfig_GeminiEmbedding,
|
|
217
|
+
PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
|
|
218
|
+
PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
|
|
188
219
|
PipelineUpdateTransformConfig,
|
|
189
220
|
PipelineUpdateTransformConfig_Advanced,
|
|
190
221
|
PipelineUpdateTransformConfig_Auto,
|
|
@@ -216,15 +247,19 @@ __all__ = [
|
|
|
216
247
|
"AdvancedModeTransformConfigSegmentationConfig_Page",
|
|
217
248
|
"AutoTransformConfig",
|
|
218
249
|
"AzureOpenAiEmbedding",
|
|
250
|
+
"AzureOpenAiEmbeddingConfig",
|
|
219
251
|
"Base",
|
|
220
252
|
"BasePromptTemplate",
|
|
221
253
|
"BedrockEmbedding",
|
|
254
|
+
"BedrockEmbeddingConfig",
|
|
255
|
+
"BoxAuthMechanism",
|
|
222
256
|
"CharacterChunkingConfig",
|
|
223
257
|
"CharacterSplitter",
|
|
224
258
|
"ChatData",
|
|
225
259
|
"ChatMessage",
|
|
226
260
|
"CloudAzStorageBlobDataSource",
|
|
227
261
|
"CloudAzureAiSearchVectorStore",
|
|
262
|
+
"CloudBoxDataSource",
|
|
228
263
|
"CloudChromaVectorStore",
|
|
229
264
|
"CloudConfluenceDataSource",
|
|
230
265
|
"CloudDocument",
|
|
@@ -243,6 +278,7 @@ __all__ = [
|
|
|
243
278
|
"CloudWeaviateVectorStore",
|
|
244
279
|
"CodeSplitter",
|
|
245
280
|
"CohereEmbedding",
|
|
281
|
+
"CohereEmbeddingConfig",
|
|
246
282
|
"ConfigurableDataSinkNames",
|
|
247
283
|
"ConfigurableDataSourceNames",
|
|
248
284
|
"ConfigurableTransformationDefinition",
|
|
@@ -273,16 +309,11 @@ __all__ = [
|
|
|
273
309
|
"DataSourceUpdateComponentOne",
|
|
274
310
|
"DataSourceUpdateCustomMetadataValue",
|
|
275
311
|
"ElementSegmentationConfig",
|
|
276
|
-
"EmbeddingConfig",
|
|
277
|
-
"EmbeddingConfigComponent",
|
|
278
|
-
"EmbeddingConfigComponentOne",
|
|
279
|
-
"EmbeddingConfigType",
|
|
280
312
|
"EvalDataset",
|
|
281
313
|
"EvalDatasetJobParams",
|
|
282
314
|
"EvalDatasetJobRecord",
|
|
283
315
|
"EvalExecutionParams",
|
|
284
316
|
"EvalExecutionParamsOverride",
|
|
285
|
-
"EvalLlmModelData",
|
|
286
317
|
"EvalQuestion",
|
|
287
318
|
"EvalQuestionCreate",
|
|
288
319
|
"EvalQuestionResult",
|
|
@@ -299,15 +330,19 @@ __all__ = [
|
|
|
299
330
|
"FilterCondition",
|
|
300
331
|
"FilterOperator",
|
|
301
332
|
"GeminiEmbedding",
|
|
333
|
+
"GeminiEmbeddingConfig",
|
|
302
334
|
"HttpValidationError",
|
|
303
335
|
"HuggingFaceInferenceApiEmbedding",
|
|
336
|
+
"HuggingFaceInferenceApiEmbeddingConfig",
|
|
304
337
|
"HuggingFaceInferenceApiEmbeddingToken",
|
|
305
338
|
"IngestionErrorResponse",
|
|
339
|
+
"InputMessage",
|
|
306
340
|
"JobNameMapping",
|
|
307
341
|
"LlamaCloudEnvironment",
|
|
308
342
|
"LlamaParseParameters",
|
|
309
343
|
"LlamaParseSupportedFileExtensions",
|
|
310
344
|
"Llm",
|
|
345
|
+
"LlmModelData",
|
|
311
346
|
"LlmParameters",
|
|
312
347
|
"LocalEval",
|
|
313
348
|
"LocalEvalResults",
|
|
@@ -316,6 +351,7 @@ __all__ = [
|
|
|
316
351
|
"ManagedIngestionStatusResponse",
|
|
317
352
|
"MarkdownElementNodeParser",
|
|
318
353
|
"MarkdownNodeParser",
|
|
354
|
+
"MessageAnnotation",
|
|
319
355
|
"MessageRole",
|
|
320
356
|
"MetadataFilter",
|
|
321
357
|
"MetadataFilterValue",
|
|
@@ -327,6 +363,7 @@ __all__ = [
|
|
|
327
363
|
"NoneSegmentationConfig",
|
|
328
364
|
"ObjectType",
|
|
329
365
|
"OpenAiEmbedding",
|
|
366
|
+
"OpenAiEmbeddingConfig",
|
|
330
367
|
"Organization",
|
|
331
368
|
"OrganizationCreate",
|
|
332
369
|
"PageSegmentationConfig",
|
|
@@ -340,6 +377,13 @@ __all__ = [
|
|
|
340
377
|
"ParsingUsage",
|
|
341
378
|
"Pipeline",
|
|
342
379
|
"PipelineCreate",
|
|
380
|
+
"PipelineCreateEmbeddingConfig",
|
|
381
|
+
"PipelineCreateEmbeddingConfig_AzureEmbedding",
|
|
382
|
+
"PipelineCreateEmbeddingConfig_BedrockEmbedding",
|
|
383
|
+
"PipelineCreateEmbeddingConfig_CohereEmbedding",
|
|
384
|
+
"PipelineCreateEmbeddingConfig_GeminiEmbedding",
|
|
385
|
+
"PipelineCreateEmbeddingConfig_HuggingfaceApiEmbedding",
|
|
386
|
+
"PipelineCreateEmbeddingConfig_OpenaiEmbedding",
|
|
343
387
|
"PipelineCreateTransformConfig",
|
|
344
388
|
"PipelineCreateTransformConfig_Advanced",
|
|
345
389
|
"PipelineCreateTransformConfig_Auto",
|
|
@@ -349,16 +393,34 @@ __all__ = [
|
|
|
349
393
|
"PipelineDataSourceCreate",
|
|
350
394
|
"PipelineDataSourceCustomMetadataValue",
|
|
351
395
|
"PipelineDeployment",
|
|
396
|
+
"PipelineEmbeddingConfig",
|
|
397
|
+
"PipelineEmbeddingConfig_AzureEmbedding",
|
|
398
|
+
"PipelineEmbeddingConfig_BedrockEmbedding",
|
|
399
|
+
"PipelineEmbeddingConfig_CohereEmbedding",
|
|
400
|
+
"PipelineEmbeddingConfig_GeminiEmbedding",
|
|
401
|
+
"PipelineEmbeddingConfig_HuggingfaceApiEmbedding",
|
|
402
|
+
"PipelineEmbeddingConfig_OpenaiEmbedding",
|
|
352
403
|
"PipelineFile",
|
|
353
404
|
"PipelineFileCreate",
|
|
354
405
|
"PipelineFileCreateCustomMetadataValue",
|
|
355
406
|
"PipelineFileCustomMetadataValue",
|
|
356
407
|
"PipelineFileResourceInfoValue",
|
|
357
408
|
"PipelineFileUpdateCustomMetadataValue",
|
|
409
|
+
"PipelineTransformConfig",
|
|
410
|
+
"PipelineTransformConfig_Advanced",
|
|
411
|
+
"PipelineTransformConfig_Auto",
|
|
358
412
|
"PipelineType",
|
|
413
|
+
"PipelineUpdateEmbeddingConfig",
|
|
414
|
+
"PipelineUpdateEmbeddingConfig_AzureEmbedding",
|
|
415
|
+
"PipelineUpdateEmbeddingConfig_BedrockEmbedding",
|
|
416
|
+
"PipelineUpdateEmbeddingConfig_CohereEmbedding",
|
|
417
|
+
"PipelineUpdateEmbeddingConfig_GeminiEmbedding",
|
|
418
|
+
"PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
|
|
419
|
+
"PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
|
|
359
420
|
"PipelineUpdateTransformConfig",
|
|
360
421
|
"PipelineUpdateTransformConfig_Advanced",
|
|
361
422
|
"PipelineUpdateTransformConfig_Auto",
|
|
423
|
+
"PlaygroundSession",
|
|
362
424
|
"Pooling",
|
|
363
425
|
"PresetRetrievalParams",
|
|
364
426
|
"PresignedUrl",
|
|
@@ -374,8 +436,8 @@ __all__ = [
|
|
|
374
436
|
"SentenceChunkingConfig",
|
|
375
437
|
"SentenceSplitter",
|
|
376
438
|
"StatusEnum",
|
|
377
|
-
"
|
|
378
|
-
"
|
|
439
|
+
"SupportedLlmModel",
|
|
440
|
+
"SupportedLlmModelNames",
|
|
379
441
|
"TextNode",
|
|
380
442
|
"TextNodeRelationshipsValue",
|
|
381
443
|
"TextNodeWithScore",
|
|
@@ -19,6 +19,13 @@ from .extraction import ExtractionSchemaCreateDataSchemaValue, ExtractionSchemaU
|
|
|
19
19
|
from .files import FileCreateResourceInfoValue
|
|
20
20
|
from .pipelines import (
|
|
21
21
|
PipelineFileUpdateCustomMetadataValue,
|
|
22
|
+
PipelineUpdateEmbeddingConfig,
|
|
23
|
+
PipelineUpdateEmbeddingConfig_AzureEmbedding,
|
|
24
|
+
PipelineUpdateEmbeddingConfig_BedrockEmbedding,
|
|
25
|
+
PipelineUpdateEmbeddingConfig_CohereEmbedding,
|
|
26
|
+
PipelineUpdateEmbeddingConfig_GeminiEmbedding,
|
|
27
|
+
PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
|
|
28
|
+
PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
|
|
22
29
|
PipelineUpdateTransformConfig,
|
|
23
30
|
PipelineUpdateTransformConfig_Advanced,
|
|
24
31
|
PipelineUpdateTransformConfig_Auto,
|
|
@@ -34,6 +41,13 @@ __all__ = [
|
|
|
34
41
|
"ExtractionSchemaUpdateDataSchemaValue",
|
|
35
42
|
"FileCreateResourceInfoValue",
|
|
36
43
|
"PipelineFileUpdateCustomMetadataValue",
|
|
44
|
+
"PipelineUpdateEmbeddingConfig",
|
|
45
|
+
"PipelineUpdateEmbeddingConfig_AzureEmbedding",
|
|
46
|
+
"PipelineUpdateEmbeddingConfig_BedrockEmbedding",
|
|
47
|
+
"PipelineUpdateEmbeddingConfig_CohereEmbedding",
|
|
48
|
+
"PipelineUpdateEmbeddingConfig_GeminiEmbedding",
|
|
49
|
+
"PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
|
|
50
|
+
"PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
|
|
37
51
|
"PipelineUpdateTransformConfig",
|
|
38
52
|
"PipelineUpdateTransformConfig_Advanced",
|
|
39
53
|
"PipelineUpdateTransformConfig_Auto",
|
|
@@ -3,6 +3,7 @@
|
|
|
3
3
|
import typing
|
|
4
4
|
|
|
5
5
|
from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
|
|
6
|
+
from ....types.cloud_box_data_source import CloudBoxDataSource
|
|
6
7
|
from ....types.cloud_confluence_data_source import CloudConfluenceDataSource
|
|
7
8
|
from ....types.cloud_jira_data_source import CloudJiraDataSource
|
|
8
9
|
from ....types.cloud_notion_page_data_source import CloudNotionPageDataSource
|
|
@@ -20,4 +21,5 @@ DataSourceUpdateComponentOne = typing.Union[
|
|
|
20
21
|
CloudNotionPageDataSource,
|
|
21
22
|
CloudConfluenceDataSource,
|
|
22
23
|
CloudJiraDataSource,
|
|
24
|
+
CloudBoxDataSource,
|
|
23
25
|
]
|
|
@@ -12,7 +12,7 @@ from ...types.eval_dataset import EvalDataset
|
|
|
12
12
|
from ...types.eval_question import EvalQuestion
|
|
13
13
|
from ...types.eval_question_create import EvalQuestionCreate
|
|
14
14
|
from ...types.http_validation_error import HttpValidationError
|
|
15
|
-
from ...types.
|
|
15
|
+
from ...types.supported_llm_model import SupportedLlmModel
|
|
16
16
|
|
|
17
17
|
try:
|
|
18
18
|
import pydantic
|
|
@@ -348,7 +348,7 @@ class EvalsClient:
|
|
|
348
348
|
raise ApiError(status_code=_response.status_code, body=_response.text)
|
|
349
349
|
raise ApiError(status_code=_response.status_code, body=_response_json)
|
|
350
350
|
|
|
351
|
-
def list_supported_models(self) -> typing.List[
|
|
351
|
+
def list_supported_models(self) -> typing.List[SupportedLlmModel]:
|
|
352
352
|
"""
|
|
353
353
|
List supported models.
|
|
354
354
|
|
|
@@ -367,7 +367,7 @@ class EvalsClient:
|
|
|
367
367
|
timeout=60,
|
|
368
368
|
)
|
|
369
369
|
if 200 <= _response.status_code < 300:
|
|
370
|
-
return pydantic.parse_obj_as(typing.List[
|
|
370
|
+
return pydantic.parse_obj_as(typing.List[SupportedLlmModel], _response.json()) # type: ignore
|
|
371
371
|
if _response.status_code == 422:
|
|
372
372
|
raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
|
|
373
373
|
try:
|
|
@@ -699,7 +699,7 @@ class AsyncEvalsClient:
|
|
|
699
699
|
raise ApiError(status_code=_response.status_code, body=_response.text)
|
|
700
700
|
raise ApiError(status_code=_response.status_code, body=_response_json)
|
|
701
701
|
|
|
702
|
-
async def list_supported_models(self) -> typing.List[
|
|
702
|
+
async def list_supported_models(self) -> typing.List[SupportedLlmModel]:
|
|
703
703
|
"""
|
|
704
704
|
List supported models.
|
|
705
705
|
|
|
@@ -718,7 +718,7 @@ class AsyncEvalsClient:
|
|
|
718
718
|
timeout=60,
|
|
719
719
|
)
|
|
720
720
|
if 200 <= _response.status_code < 300:
|
|
721
|
-
return pydantic.parse_obj_as(typing.List[
|
|
721
|
+
return pydantic.parse_obj_as(typing.List[SupportedLlmModel], _response.json()) # type: ignore
|
|
722
722
|
if _response.status_code == 422:
|
|
723
723
|
raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json())) # type: ignore
|
|
724
724
|
try:
|
|
@@ -118,6 +118,7 @@ class ParsingClient:
|
|
|
118
118
|
vendor_multimodal_api_key: str,
|
|
119
119
|
page_prefix: str,
|
|
120
120
|
page_suffix: str,
|
|
121
|
+
take_screenshot: bool,
|
|
121
122
|
file: typing.IO,
|
|
122
123
|
) -> ParsingJob:
|
|
123
124
|
"""
|
|
@@ -158,6 +159,8 @@ class ParsingClient:
|
|
|
158
159
|
|
|
159
160
|
- page_suffix: str.
|
|
160
161
|
|
|
162
|
+
- take_screenshot: bool.
|
|
163
|
+
|
|
161
164
|
- file: typing.IO.
|
|
162
165
|
"""
|
|
163
166
|
_response = self._client_wrapper.httpx_client.request(
|
|
@@ -182,6 +185,7 @@ class ParsingClient:
|
|
|
182
185
|
"vendor_multimodal_api_key": vendor_multimodal_api_key,
|
|
183
186
|
"page_prefix": page_prefix,
|
|
184
187
|
"page_suffix": page_suffix,
|
|
188
|
+
"take_screenshot": take_screenshot,
|
|
185
189
|
}
|
|
186
190
|
),
|
|
187
191
|
files={"file": file},
|
|
@@ -643,6 +647,7 @@ class AsyncParsingClient:
|
|
|
643
647
|
vendor_multimodal_api_key: str,
|
|
644
648
|
page_prefix: str,
|
|
645
649
|
page_suffix: str,
|
|
650
|
+
take_screenshot: bool,
|
|
646
651
|
file: typing.IO,
|
|
647
652
|
) -> ParsingJob:
|
|
648
653
|
"""
|
|
@@ -683,6 +688,8 @@ class AsyncParsingClient:
|
|
|
683
688
|
|
|
684
689
|
- page_suffix: str.
|
|
685
690
|
|
|
691
|
+
- take_screenshot: bool.
|
|
692
|
+
|
|
686
693
|
- file: typing.IO.
|
|
687
694
|
"""
|
|
688
695
|
_response = await self._client_wrapper.httpx_client.request(
|
|
@@ -707,6 +714,7 @@ class AsyncParsingClient:
|
|
|
707
714
|
"vendor_multimodal_api_key": vendor_multimodal_api_key,
|
|
708
715
|
"page_prefix": page_prefix,
|
|
709
716
|
"page_suffix": page_suffix,
|
|
717
|
+
"take_screenshot": take_screenshot,
|
|
710
718
|
}
|
|
711
719
|
),
|
|
712
720
|
files={"file": file},
|
|
@@ -2,6 +2,13 @@
|
|
|
2
2
|
|
|
3
3
|
from .types import (
|
|
4
4
|
PipelineFileUpdateCustomMetadataValue,
|
|
5
|
+
PipelineUpdateEmbeddingConfig,
|
|
6
|
+
PipelineUpdateEmbeddingConfig_AzureEmbedding,
|
|
7
|
+
PipelineUpdateEmbeddingConfig_BedrockEmbedding,
|
|
8
|
+
PipelineUpdateEmbeddingConfig_CohereEmbedding,
|
|
9
|
+
PipelineUpdateEmbeddingConfig_GeminiEmbedding,
|
|
10
|
+
PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
|
|
11
|
+
PipelineUpdateEmbeddingConfig_OpenaiEmbedding,
|
|
5
12
|
PipelineUpdateTransformConfig,
|
|
6
13
|
PipelineUpdateTransformConfig_Advanced,
|
|
7
14
|
PipelineUpdateTransformConfig_Auto,
|
|
@@ -9,6 +16,13 @@ from .types import (
|
|
|
9
16
|
|
|
10
17
|
__all__ = [
|
|
11
18
|
"PipelineFileUpdateCustomMetadataValue",
|
|
19
|
+
"PipelineUpdateEmbeddingConfig",
|
|
20
|
+
"PipelineUpdateEmbeddingConfig_AzureEmbedding",
|
|
21
|
+
"PipelineUpdateEmbeddingConfig_BedrockEmbedding",
|
|
22
|
+
"PipelineUpdateEmbeddingConfig_CohereEmbedding",
|
|
23
|
+
"PipelineUpdateEmbeddingConfig_GeminiEmbedding",
|
|
24
|
+
"PipelineUpdateEmbeddingConfig_HuggingfaceApiEmbedding",
|
|
25
|
+
"PipelineUpdateEmbeddingConfig_OpenaiEmbedding",
|
|
12
26
|
"PipelineUpdateTransformConfig",
|
|
13
27
|
"PipelineUpdateTransformConfig_Advanced",
|
|
14
28
|
"PipelineUpdateTransformConfig_Auto",
|