llama-cloud 0.1.20__py3-none-any.whl → 0.1.21__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.

Files changed (30) hide show
  1. llama_cloud/__init__.py +2 -124
  2. llama_cloud/resources/__init__.py +2 -41
  3. llama_cloud/resources/data_sinks/__init__.py +2 -18
  4. llama_cloud/resources/data_sinks/client.py +94 -2
  5. llama_cloud/resources/data_sinks/types/__init__.py +2 -18
  6. llama_cloud/resources/data_sinks/types/data_sink_update_component.py +7 -65
  7. llama_cloud/resources/data_sources/__init__.py +2 -30
  8. llama_cloud/resources/data_sources/types/__init__.py +1 -28
  9. llama_cloud/resources/data_sources/types/data_source_update_component.py +23 -2
  10. llama_cloud/resources/pipelines/client.py +14 -0
  11. llama_cloud/types/__init__.py +4 -102
  12. llama_cloud/types/cloud_jira_data_source.py +4 -0
  13. llama_cloud/types/data_sink_component.py +7 -65
  14. llama_cloud/types/data_sink_create_component.py +7 -65
  15. llama_cloud/types/data_source_component.py +23 -2
  16. llama_cloud/types/data_source_create_component.py +23 -2
  17. llama_cloud/types/extract_config.py +3 -0
  18. llama_cloud/types/pipeline.py +2 -0
  19. llama_cloud/types/pipeline_data_source_component.py +23 -2
  20. llama_cloud/types/pipeline_status.py +17 -0
  21. llama_cloud/types/prompt_conf.py +1 -0
  22. llama_cloud/types/supported_llm_model.py +1 -2
  23. {llama_cloud-0.1.20.dist-info → llama_cloud-0.1.21.dist-info}/METADATA +5 -3
  24. {llama_cloud-0.1.20.dist-info → llama_cloud-0.1.21.dist-info}/RECORD +26 -29
  25. {llama_cloud-0.1.20.dist-info → llama_cloud-0.1.21.dist-info}/WHEEL +1 -1
  26. llama_cloud/resources/data_sources/types/data_source_update_component_one.py +0 -122
  27. llama_cloud/types/data_source_component_one.py +0 -122
  28. llama_cloud/types/data_source_create_component_one.py +0 -122
  29. llama_cloud/types/pipeline_data_source_component_one.py +0 -122
  30. {llama_cloud-0.1.20.dist-info → llama_cloud-0.1.21.dist-info}/LICENSE +0 -0
llama_cloud/__init__.py CHANGED
@@ -65,47 +65,13 @@ from .types import (
65
65
  CreditType,
66
66
  DataSink,
67
67
  DataSinkComponent,
68
- DataSinkComponent_AzureAiSearch,
69
- DataSinkComponent_Milvus,
70
- DataSinkComponent_MongodbAtlas,
71
- DataSinkComponent_Pinecone,
72
- DataSinkComponent_Postgres,
73
- DataSinkComponent_Qdrant,
74
68
  DataSinkCreate,
75
69
  DataSinkCreateComponent,
76
- DataSinkCreateComponent_AzureAiSearch,
77
- DataSinkCreateComponent_Milvus,
78
- DataSinkCreateComponent_MongodbAtlas,
79
- DataSinkCreateComponent_Pinecone,
80
- DataSinkCreateComponent_Postgres,
81
- DataSinkCreateComponent_Qdrant,
82
70
  DataSinkDefinition,
83
71
  DataSource,
84
72
  DataSourceComponent,
85
- DataSourceComponentOne,
86
- DataSourceComponentOne_AzureStorageBlob,
87
- DataSourceComponentOne_Box,
88
- DataSourceComponentOne_Confluence,
89
- DataSourceComponentOne_GoogleDrive,
90
- DataSourceComponentOne_Jira,
91
- DataSourceComponentOne_MicrosoftOnedrive,
92
- DataSourceComponentOne_MicrosoftSharepoint,
93
- DataSourceComponentOne_NotionPage,
94
- DataSourceComponentOne_S3,
95
- DataSourceComponentOne_Slack,
96
73
  DataSourceCreate,
97
74
  DataSourceCreateComponent,
98
- DataSourceCreateComponentOne,
99
- DataSourceCreateComponentOne_AzureStorageBlob,
100
- DataSourceCreateComponentOne_Box,
101
- DataSourceCreateComponentOne_Confluence,
102
- DataSourceCreateComponentOne_GoogleDrive,
103
- DataSourceCreateComponentOne_Jira,
104
- DataSourceCreateComponentOne_MicrosoftOnedrive,
105
- DataSourceCreateComponentOne_MicrosoftSharepoint,
106
- DataSourceCreateComponentOne_NotionPage,
107
- DataSourceCreateComponentOne_S3,
108
- DataSourceCreateComponentOne_Slack,
109
75
  DataSourceCreateCustomMetadataValue,
110
76
  DataSourceCustomMetadataValue,
111
77
  DataSourceDefinition,
@@ -256,17 +222,6 @@ from .types import (
256
222
  PipelineCreateTransformConfig,
257
223
  PipelineDataSource,
258
224
  PipelineDataSourceComponent,
259
- PipelineDataSourceComponentOne,
260
- PipelineDataSourceComponentOne_AzureStorageBlob,
261
- PipelineDataSourceComponentOne_Box,
262
- PipelineDataSourceComponentOne_Confluence,
263
- PipelineDataSourceComponentOne_GoogleDrive,
264
- PipelineDataSourceComponentOne_Jira,
265
- PipelineDataSourceComponentOne_MicrosoftOnedrive,
266
- PipelineDataSourceComponentOne_MicrosoftSharepoint,
267
- PipelineDataSourceComponentOne_NotionPage,
268
- PipelineDataSourceComponentOne_S3,
269
- PipelineDataSourceComponentOne_Slack,
270
225
  PipelineDataSourceCreate,
271
226
  PipelineDataSourceCustomMetadataValue,
272
227
  PipelineDataSourceStatus,
@@ -291,6 +246,7 @@ from .types import (
291
246
  PipelineFileUpdaterConfig,
292
247
  PipelineManagedIngestionJobParams,
293
248
  PipelineMetadataConfig,
249
+ PipelineStatus,
294
250
  PipelineTransformConfig,
295
251
  PipelineTransformConfig_Advanced,
296
252
  PipelineTransformConfig_Auto,
@@ -367,24 +323,7 @@ from .types import (
367
323
  from .errors import UnprocessableEntityError
368
324
  from .resources import (
369
325
  DataSinkUpdateComponent,
370
- DataSinkUpdateComponent_AzureAiSearch,
371
- DataSinkUpdateComponent_Milvus,
372
- DataSinkUpdateComponent_MongodbAtlas,
373
- DataSinkUpdateComponent_Pinecone,
374
- DataSinkUpdateComponent_Postgres,
375
- DataSinkUpdateComponent_Qdrant,
376
326
  DataSourceUpdateComponent,
377
- DataSourceUpdateComponentOne,
378
- DataSourceUpdateComponentOne_AzureStorageBlob,
379
- DataSourceUpdateComponentOne_Box,
380
- DataSourceUpdateComponentOne_Confluence,
381
- DataSourceUpdateComponentOne_GoogleDrive,
382
- DataSourceUpdateComponentOne_Jira,
383
- DataSourceUpdateComponentOne_MicrosoftOnedrive,
384
- DataSourceUpdateComponentOne_MicrosoftSharepoint,
385
- DataSourceUpdateComponentOne_NotionPage,
386
- DataSourceUpdateComponentOne_S3,
387
- DataSourceUpdateComponentOne_Slack,
388
327
  DataSourceUpdateCustomMetadataValue,
389
328
  EmbeddingModelConfigCreateEmbeddingConfig,
390
329
  EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding,
@@ -500,69 +439,18 @@ __all__ = [
500
439
  "CreditType",
501
440
  "DataSink",
502
441
  "DataSinkComponent",
503
- "DataSinkComponent_AzureAiSearch",
504
- "DataSinkComponent_Milvus",
505
- "DataSinkComponent_MongodbAtlas",
506
- "DataSinkComponent_Pinecone",
507
- "DataSinkComponent_Postgres",
508
- "DataSinkComponent_Qdrant",
509
442
  "DataSinkCreate",
510
443
  "DataSinkCreateComponent",
511
- "DataSinkCreateComponent_AzureAiSearch",
512
- "DataSinkCreateComponent_Milvus",
513
- "DataSinkCreateComponent_MongodbAtlas",
514
- "DataSinkCreateComponent_Pinecone",
515
- "DataSinkCreateComponent_Postgres",
516
- "DataSinkCreateComponent_Qdrant",
517
444
  "DataSinkDefinition",
518
445
  "DataSinkUpdateComponent",
519
- "DataSinkUpdateComponent_AzureAiSearch",
520
- "DataSinkUpdateComponent_Milvus",
521
- "DataSinkUpdateComponent_MongodbAtlas",
522
- "DataSinkUpdateComponent_Pinecone",
523
- "DataSinkUpdateComponent_Postgres",
524
- "DataSinkUpdateComponent_Qdrant",
525
446
  "DataSource",
526
447
  "DataSourceComponent",
527
- "DataSourceComponentOne",
528
- "DataSourceComponentOne_AzureStorageBlob",
529
- "DataSourceComponentOne_Box",
530
- "DataSourceComponentOne_Confluence",
531
- "DataSourceComponentOne_GoogleDrive",
532
- "DataSourceComponentOne_Jira",
533
- "DataSourceComponentOne_MicrosoftOnedrive",
534
- "DataSourceComponentOne_MicrosoftSharepoint",
535
- "DataSourceComponentOne_NotionPage",
536
- "DataSourceComponentOne_S3",
537
- "DataSourceComponentOne_Slack",
538
448
  "DataSourceCreate",
539
449
  "DataSourceCreateComponent",
540
- "DataSourceCreateComponentOne",
541
- "DataSourceCreateComponentOne_AzureStorageBlob",
542
- "DataSourceCreateComponentOne_Box",
543
- "DataSourceCreateComponentOne_Confluence",
544
- "DataSourceCreateComponentOne_GoogleDrive",
545
- "DataSourceCreateComponentOne_Jira",
546
- "DataSourceCreateComponentOne_MicrosoftOnedrive",
547
- "DataSourceCreateComponentOne_MicrosoftSharepoint",
548
- "DataSourceCreateComponentOne_NotionPage",
549
- "DataSourceCreateComponentOne_S3",
550
- "DataSourceCreateComponentOne_Slack",
551
450
  "DataSourceCreateCustomMetadataValue",
552
451
  "DataSourceCustomMetadataValue",
553
452
  "DataSourceDefinition",
554
453
  "DataSourceUpdateComponent",
555
- "DataSourceUpdateComponentOne",
556
- "DataSourceUpdateComponentOne_AzureStorageBlob",
557
- "DataSourceUpdateComponentOne_Box",
558
- "DataSourceUpdateComponentOne_Confluence",
559
- "DataSourceUpdateComponentOne_GoogleDrive",
560
- "DataSourceUpdateComponentOne_Jira",
561
- "DataSourceUpdateComponentOne_MicrosoftOnedrive",
562
- "DataSourceUpdateComponentOne_MicrosoftSharepoint",
563
- "DataSourceUpdateComponentOne_NotionPage",
564
- "DataSourceUpdateComponentOne_S3",
565
- "DataSourceUpdateComponentOne_Slack",
566
454
  "DataSourceUpdateCustomMetadataValue",
567
455
  "DataSourceUpdateDispatcherConfig",
568
456
  "DeleteParams",
@@ -731,17 +619,6 @@ __all__ = [
731
619
  "PipelineCreateTransformConfig",
732
620
  "PipelineDataSource",
733
621
  "PipelineDataSourceComponent",
734
- "PipelineDataSourceComponentOne",
735
- "PipelineDataSourceComponentOne_AzureStorageBlob",
736
- "PipelineDataSourceComponentOne_Box",
737
- "PipelineDataSourceComponentOne_Confluence",
738
- "PipelineDataSourceComponentOne_GoogleDrive",
739
- "PipelineDataSourceComponentOne_Jira",
740
- "PipelineDataSourceComponentOne_MicrosoftOnedrive",
741
- "PipelineDataSourceComponentOne_MicrosoftSharepoint",
742
- "PipelineDataSourceComponentOne_NotionPage",
743
- "PipelineDataSourceComponentOne_S3",
744
- "PipelineDataSourceComponentOne_Slack",
745
622
  "PipelineDataSourceCreate",
746
623
  "PipelineDataSourceCustomMetadataValue",
747
624
  "PipelineDataSourceStatus",
@@ -767,6 +644,7 @@ __all__ = [
767
644
  "PipelineFileUpdaterConfig",
768
645
  "PipelineManagedIngestionJobParams",
769
646
  "PipelineMetadataConfig",
647
+ "PipelineStatus",
770
648
  "PipelineTransformConfig",
771
649
  "PipelineTransformConfig_Advanced",
772
650
  "PipelineTransformConfig_Auto",
@@ -18,30 +18,8 @@ from . import (
18
18
  reports,
19
19
  retrievers,
20
20
  )
21
- from .data_sinks import (
22
- DataSinkUpdateComponent,
23
- DataSinkUpdateComponent_AzureAiSearch,
24
- DataSinkUpdateComponent_Milvus,
25
- DataSinkUpdateComponent_MongodbAtlas,
26
- DataSinkUpdateComponent_Pinecone,
27
- DataSinkUpdateComponent_Postgres,
28
- DataSinkUpdateComponent_Qdrant,
29
- )
30
- from .data_sources import (
31
- DataSourceUpdateComponent,
32
- DataSourceUpdateComponentOne,
33
- DataSourceUpdateComponentOne_AzureStorageBlob,
34
- DataSourceUpdateComponentOne_Box,
35
- DataSourceUpdateComponentOne_Confluence,
36
- DataSourceUpdateComponentOne_GoogleDrive,
37
- DataSourceUpdateComponentOne_Jira,
38
- DataSourceUpdateComponentOne_MicrosoftOnedrive,
39
- DataSourceUpdateComponentOne_MicrosoftSharepoint,
40
- DataSourceUpdateComponentOne_NotionPage,
41
- DataSourceUpdateComponentOne_S3,
42
- DataSourceUpdateComponentOne_Slack,
43
- DataSourceUpdateCustomMetadataValue,
44
- )
21
+ from .data_sinks import DataSinkUpdateComponent
22
+ from .data_sources import DataSourceUpdateComponent, DataSourceUpdateCustomMetadataValue
45
23
  from .embedding_model_configs import (
46
24
  EmbeddingModelConfigCreateEmbeddingConfig,
47
25
  EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding,
@@ -79,24 +57,7 @@ from .reports import UpdateReportPlanApiV1ReportsReportIdPlanPatchRequestAction
79
57
 
80
58
  __all__ = [
81
59
  "DataSinkUpdateComponent",
82
- "DataSinkUpdateComponent_AzureAiSearch",
83
- "DataSinkUpdateComponent_Milvus",
84
- "DataSinkUpdateComponent_MongodbAtlas",
85
- "DataSinkUpdateComponent_Pinecone",
86
- "DataSinkUpdateComponent_Postgres",
87
- "DataSinkUpdateComponent_Qdrant",
88
60
  "DataSourceUpdateComponent",
89
- "DataSourceUpdateComponentOne",
90
- "DataSourceUpdateComponentOne_AzureStorageBlob",
91
- "DataSourceUpdateComponentOne_Box",
92
- "DataSourceUpdateComponentOne_Confluence",
93
- "DataSourceUpdateComponentOne_GoogleDrive",
94
- "DataSourceUpdateComponentOne_Jira",
95
- "DataSourceUpdateComponentOne_MicrosoftOnedrive",
96
- "DataSourceUpdateComponentOne_MicrosoftSharepoint",
97
- "DataSourceUpdateComponentOne_NotionPage",
98
- "DataSourceUpdateComponentOne_S3",
99
- "DataSourceUpdateComponentOne_Slack",
100
61
  "DataSourceUpdateCustomMetadataValue",
101
62
  "EmbeddingModelConfigCreateEmbeddingConfig",
102
63
  "EmbeddingModelConfigCreateEmbeddingConfig_AzureEmbedding",
@@ -1,21 +1,5 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
- from .types import (
4
- DataSinkUpdateComponent,
5
- DataSinkUpdateComponent_AzureAiSearch,
6
- DataSinkUpdateComponent_Milvus,
7
- DataSinkUpdateComponent_MongodbAtlas,
8
- DataSinkUpdateComponent_Pinecone,
9
- DataSinkUpdateComponent_Postgres,
10
- DataSinkUpdateComponent_Qdrant,
11
- )
3
+ from .types import DataSinkUpdateComponent
12
4
 
13
- __all__ = [
14
- "DataSinkUpdateComponent",
15
- "DataSinkUpdateComponent_AzureAiSearch",
16
- "DataSinkUpdateComponent_Milvus",
17
- "DataSinkUpdateComponent_MongodbAtlas",
18
- "DataSinkUpdateComponent_Pinecone",
19
- "DataSinkUpdateComponent_Postgres",
20
- "DataSinkUpdateComponent_Qdrant",
21
- ]
5
+ __all__ = ["DataSinkUpdateComponent"]
@@ -82,6 +82,19 @@ class DataSinksClient:
82
82
  - organization_id: typing.Optional[str].
83
83
 
84
84
  - request: DataSinkCreate.
85
+ ---
86
+ from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
87
+ from llama_cloud.client import LlamaCloud
88
+
89
+ client = LlamaCloud(
90
+ token="YOUR_TOKEN",
91
+ )
92
+ client.data_sinks.create_data_sink(
93
+ request=DataSinkCreate(
94
+ name="string",
95
+ sink_type=ConfigurableDataSinkNames.PINECONE,
96
+ ),
97
+ )
85
98
  """
86
99
  _response = self._client_wrapper.httpx_client.request(
87
100
  "POST",
@@ -118,6 +131,19 @@ class DataSinksClient:
118
131
  - organization_id: typing.Optional[str].
119
132
 
120
133
  - request: DataSinkCreate.
134
+ ---
135
+ from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
136
+ from llama_cloud.client import LlamaCloud
137
+
138
+ client = LlamaCloud(
139
+ token="YOUR_TOKEN",
140
+ )
141
+ client.data_sinks.upsert_data_sink(
142
+ request=DataSinkCreate(
143
+ name="string",
144
+ sink_type=ConfigurableDataSinkNames.PINECONE,
145
+ ),
146
+ )
121
147
  """
122
148
  _response = self._client_wrapper.httpx_client.request(
123
149
  "PUT",
@@ -143,6 +169,15 @@ class DataSinksClient:
143
169
 
144
170
  Parameters:
145
171
  - data_sink_id: str.
172
+ ---
173
+ from llama_cloud.client import LlamaCloud
174
+
175
+ client = LlamaCloud(
176
+ token="YOUR_TOKEN",
177
+ )
178
+ client.data_sinks.get_data_sink(
179
+ data_sink_id="string",
180
+ )
146
181
  """
147
182
  _response = self._client_wrapper.httpx_client.request(
148
183
  "GET",
@@ -178,7 +213,18 @@ class DataSinksClient:
178
213
 
179
214
  - sink_type: ConfigurableDataSinkNames.
180
215
 
181
- - component: typing.Optional[DataSinkUpdateComponent].
216
+ - component: typing.Optional[DataSinkUpdateComponent]. Component that implements the data sink
217
+ ---
218
+ from llama_cloud import ConfigurableDataSinkNames
219
+ from llama_cloud.client import LlamaCloud
220
+
221
+ client = LlamaCloud(
222
+ token="YOUR_TOKEN",
223
+ )
224
+ client.data_sinks.update_data_sink(
225
+ data_sink_id="string",
226
+ sink_type=ConfigurableDataSinkNames.PINECONE,
227
+ )
182
228
  """
183
229
  _request: typing.Dict[str, typing.Any] = {"sink_type": sink_type}
184
230
  if name is not OMIT:
@@ -290,6 +336,19 @@ class AsyncDataSinksClient:
290
336
  - organization_id: typing.Optional[str].
291
337
 
292
338
  - request: DataSinkCreate.
339
+ ---
340
+ from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
341
+ from llama_cloud.client import AsyncLlamaCloud
342
+
343
+ client = AsyncLlamaCloud(
344
+ token="YOUR_TOKEN",
345
+ )
346
+ await client.data_sinks.create_data_sink(
347
+ request=DataSinkCreate(
348
+ name="string",
349
+ sink_type=ConfigurableDataSinkNames.PINECONE,
350
+ ),
351
+ )
293
352
  """
294
353
  _response = await self._client_wrapper.httpx_client.request(
295
354
  "POST",
@@ -326,6 +385,19 @@ class AsyncDataSinksClient:
326
385
  - organization_id: typing.Optional[str].
327
386
 
328
387
  - request: DataSinkCreate.
388
+ ---
389
+ from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
390
+ from llama_cloud.client import AsyncLlamaCloud
391
+
392
+ client = AsyncLlamaCloud(
393
+ token="YOUR_TOKEN",
394
+ )
395
+ await client.data_sinks.upsert_data_sink(
396
+ request=DataSinkCreate(
397
+ name="string",
398
+ sink_type=ConfigurableDataSinkNames.PINECONE,
399
+ ),
400
+ )
329
401
  """
330
402
  _response = await self._client_wrapper.httpx_client.request(
331
403
  "PUT",
@@ -351,6 +423,15 @@ class AsyncDataSinksClient:
351
423
 
352
424
  Parameters:
353
425
  - data_sink_id: str.
426
+ ---
427
+ from llama_cloud.client import AsyncLlamaCloud
428
+
429
+ client = AsyncLlamaCloud(
430
+ token="YOUR_TOKEN",
431
+ )
432
+ await client.data_sinks.get_data_sink(
433
+ data_sink_id="string",
434
+ )
354
435
  """
355
436
  _response = await self._client_wrapper.httpx_client.request(
356
437
  "GET",
@@ -386,7 +467,18 @@ class AsyncDataSinksClient:
386
467
 
387
468
  - sink_type: ConfigurableDataSinkNames.
388
469
 
389
- - component: typing.Optional[DataSinkUpdateComponent].
470
+ - component: typing.Optional[DataSinkUpdateComponent]. Component that implements the data sink
471
+ ---
472
+ from llama_cloud import ConfigurableDataSinkNames
473
+ from llama_cloud.client import AsyncLlamaCloud
474
+
475
+ client = AsyncLlamaCloud(
476
+ token="YOUR_TOKEN",
477
+ )
478
+ await client.data_sinks.update_data_sink(
479
+ data_sink_id="string",
480
+ sink_type=ConfigurableDataSinkNames.PINECONE,
481
+ )
390
482
  """
391
483
  _request: typing.Dict[str, typing.Any] = {"sink_type": sink_type}
392
484
  if name is not OMIT:
@@ -1,21 +1,5 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
- from .data_sink_update_component import (
4
- DataSinkUpdateComponent,
5
- DataSinkUpdateComponent_AzureAiSearch,
6
- DataSinkUpdateComponent_Milvus,
7
- DataSinkUpdateComponent_MongodbAtlas,
8
- DataSinkUpdateComponent_Pinecone,
9
- DataSinkUpdateComponent_Postgres,
10
- DataSinkUpdateComponent_Qdrant,
11
- )
3
+ from .data_sink_update_component import DataSinkUpdateComponent
12
4
 
13
- __all__ = [
14
- "DataSinkUpdateComponent",
15
- "DataSinkUpdateComponent_AzureAiSearch",
16
- "DataSinkUpdateComponent_Milvus",
17
- "DataSinkUpdateComponent_MongodbAtlas",
18
- "DataSinkUpdateComponent_Pinecone",
19
- "DataSinkUpdateComponent_Postgres",
20
- "DataSinkUpdateComponent_Qdrant",
21
- ]
5
+ __all__ = ["DataSinkUpdateComponent"]
@@ -1,11 +1,7 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
- from __future__ import annotations
4
-
5
3
  import typing
6
4
 
7
- import typing_extensions
8
-
9
5
  from ....types.cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
10
6
  from ....types.cloud_milvus_vector_store import CloudMilvusVectorStore
11
7
  from ....types.cloud_mongo_db_atlas_vector_search import CloudMongoDbAtlasVectorSearch
@@ -13,66 +9,12 @@ from ....types.cloud_pinecone_vector_store import CloudPineconeVectorStore
13
9
  from ....types.cloud_postgres_vector_store import CloudPostgresVectorStore
14
10
  from ....types.cloud_qdrant_vector_store import CloudQdrantVectorStore
15
11
 
16
-
17
- class DataSinkUpdateComponent_AzureAiSearch(CloudAzureAiSearchVectorStore):
18
- type: typing_extensions.Literal["azure_ai_search"]
19
-
20
- class Config:
21
- frozen = True
22
- smart_union = True
23
- allow_population_by_field_name = True
24
-
25
-
26
- class DataSinkUpdateComponent_Milvus(CloudMilvusVectorStore):
27
- type: typing_extensions.Literal["milvus"]
28
-
29
- class Config:
30
- frozen = True
31
- smart_union = True
32
- allow_population_by_field_name = True
33
-
34
-
35
- class DataSinkUpdateComponent_MongodbAtlas(CloudMongoDbAtlasVectorSearch):
36
- type: typing_extensions.Literal["mongodb_atlas"]
37
-
38
- class Config:
39
- frozen = True
40
- smart_union = True
41
- allow_population_by_field_name = True
42
-
43
-
44
- class DataSinkUpdateComponent_Pinecone(CloudPineconeVectorStore):
45
- type: typing_extensions.Literal["pinecone"]
46
-
47
- class Config:
48
- frozen = True
49
- smart_union = True
50
- allow_population_by_field_name = True
51
-
52
-
53
- class DataSinkUpdateComponent_Postgres(CloudPostgresVectorStore):
54
- type: typing_extensions.Literal["postgres"]
55
-
56
- class Config:
57
- frozen = True
58
- smart_union = True
59
- allow_population_by_field_name = True
60
-
61
-
62
- class DataSinkUpdateComponent_Qdrant(CloudQdrantVectorStore):
63
- type: typing_extensions.Literal["qdrant"]
64
-
65
- class Config:
66
- frozen = True
67
- smart_union = True
68
- allow_population_by_field_name = True
69
-
70
-
71
12
  DataSinkUpdateComponent = typing.Union[
72
- DataSinkUpdateComponent_AzureAiSearch,
73
- DataSinkUpdateComponent_Milvus,
74
- DataSinkUpdateComponent_MongodbAtlas,
75
- DataSinkUpdateComponent_Pinecone,
76
- DataSinkUpdateComponent_Postgres,
77
- DataSinkUpdateComponent_Qdrant,
13
+ typing.Dict[str, typing.Any],
14
+ CloudPineconeVectorStore,
15
+ CloudPostgresVectorStore,
16
+ CloudQdrantVectorStore,
17
+ CloudAzureAiSearchVectorStore,
18
+ CloudMongoDbAtlasVectorSearch,
19
+ CloudMilvusVectorStore,
78
20
  ]
@@ -1,33 +1,5 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
- from .types import (
4
- DataSourceUpdateComponent,
5
- DataSourceUpdateComponentOne,
6
- DataSourceUpdateComponentOne_AzureStorageBlob,
7
- DataSourceUpdateComponentOne_Box,
8
- DataSourceUpdateComponentOne_Confluence,
9
- DataSourceUpdateComponentOne_GoogleDrive,
10
- DataSourceUpdateComponentOne_Jira,
11
- DataSourceUpdateComponentOne_MicrosoftOnedrive,
12
- DataSourceUpdateComponentOne_MicrosoftSharepoint,
13
- DataSourceUpdateComponentOne_NotionPage,
14
- DataSourceUpdateComponentOne_S3,
15
- DataSourceUpdateComponentOne_Slack,
16
- DataSourceUpdateCustomMetadataValue,
17
- )
3
+ from .types import DataSourceUpdateComponent, DataSourceUpdateCustomMetadataValue
18
4
 
19
- __all__ = [
20
- "DataSourceUpdateComponent",
21
- "DataSourceUpdateComponentOne",
22
- "DataSourceUpdateComponentOne_AzureStorageBlob",
23
- "DataSourceUpdateComponentOne_Box",
24
- "DataSourceUpdateComponentOne_Confluence",
25
- "DataSourceUpdateComponentOne_GoogleDrive",
26
- "DataSourceUpdateComponentOne_Jira",
27
- "DataSourceUpdateComponentOne_MicrosoftOnedrive",
28
- "DataSourceUpdateComponentOne_MicrosoftSharepoint",
29
- "DataSourceUpdateComponentOne_NotionPage",
30
- "DataSourceUpdateComponentOne_S3",
31
- "DataSourceUpdateComponentOne_Slack",
32
- "DataSourceUpdateCustomMetadataValue",
33
- ]
5
+ __all__ = ["DataSourceUpdateComponent", "DataSourceUpdateCustomMetadataValue"]
@@ -1,33 +1,6 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
3
  from .data_source_update_component import DataSourceUpdateComponent
4
- from .data_source_update_component_one import (
5
- DataSourceUpdateComponentOne,
6
- DataSourceUpdateComponentOne_AzureStorageBlob,
7
- DataSourceUpdateComponentOne_Box,
8
- DataSourceUpdateComponentOne_Confluence,
9
- DataSourceUpdateComponentOne_GoogleDrive,
10
- DataSourceUpdateComponentOne_Jira,
11
- DataSourceUpdateComponentOne_MicrosoftOnedrive,
12
- DataSourceUpdateComponentOne_MicrosoftSharepoint,
13
- DataSourceUpdateComponentOne_NotionPage,
14
- DataSourceUpdateComponentOne_S3,
15
- DataSourceUpdateComponentOne_Slack,
16
- )
17
4
  from .data_source_update_custom_metadata_value import DataSourceUpdateCustomMetadataValue
18
5
 
19
- __all__ = [
20
- "DataSourceUpdateComponent",
21
- "DataSourceUpdateComponentOne",
22
- "DataSourceUpdateComponentOne_AzureStorageBlob",
23
- "DataSourceUpdateComponentOne_Box",
24
- "DataSourceUpdateComponentOne_Confluence",
25
- "DataSourceUpdateComponentOne_GoogleDrive",
26
- "DataSourceUpdateComponentOne_Jira",
27
- "DataSourceUpdateComponentOne_MicrosoftOnedrive",
28
- "DataSourceUpdateComponentOne_MicrosoftSharepoint",
29
- "DataSourceUpdateComponentOne_NotionPage",
30
- "DataSourceUpdateComponentOne_S3",
31
- "DataSourceUpdateComponentOne_Slack",
32
- "DataSourceUpdateCustomMetadataValue",
33
- ]
6
+ __all__ = ["DataSourceUpdateComponent", "DataSourceUpdateCustomMetadataValue"]
@@ -2,6 +2,27 @@
2
2
 
3
3
  import typing
4
4
 
5
- from .data_source_update_component_one import DataSourceUpdateComponentOne
5
+ from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
6
+ from ....types.cloud_box_data_source import CloudBoxDataSource
7
+ from ....types.cloud_confluence_data_source import CloudConfluenceDataSource
8
+ from ....types.cloud_google_drive_data_source import CloudGoogleDriveDataSource
9
+ from ....types.cloud_jira_data_source import CloudJiraDataSource
10
+ from ....types.cloud_notion_page_data_source import CloudNotionPageDataSource
11
+ from ....types.cloud_one_drive_data_source import CloudOneDriveDataSource
12
+ from ....types.cloud_s_3_data_source import CloudS3DataSource
13
+ from ....types.cloud_sharepoint_data_source import CloudSharepointDataSource
14
+ from ....types.cloud_slack_data_source import CloudSlackDataSource
6
15
 
7
- DataSourceUpdateComponent = typing.Union[typing.Dict[str, typing.Any], DataSourceUpdateComponentOne]
16
+ DataSourceUpdateComponent = typing.Union[
17
+ typing.Dict[str, typing.Any],
18
+ CloudS3DataSource,
19
+ CloudAzStorageBlobDataSource,
20
+ CloudGoogleDriveDataSource,
21
+ CloudOneDriveDataSource,
22
+ CloudSharepointDataSource,
23
+ CloudSlackDataSource,
24
+ CloudNotionPageDataSource,
25
+ CloudConfluenceDataSource,
26
+ CloudJiraDataSource,
27
+ CloudBoxDataSource,
28
+ ]