llama-cloud 0.0.10__py3-none-any.whl → 0.0.12__py3-none-any.whl

llama_cloud/resources/extraction/client.py

@@ -13,6 +13,7 @@ from ...types.extraction_job import ExtractionJob
 from ...types.extraction_result import ExtractionResult
 from ...types.extraction_schema import ExtractionSchema
 from ...types.http_validation_error import HttpValidationError
+from .types.extraction_schema_create_data_schema_value import ExtractionSchemaCreateDataSchemaValue
 from .types.extraction_schema_update_data_schema_value import ExtractionSchemaUpdateDataSchemaValue
 
 try:
@@ -31,47 +32,66 @@ class ExtractionClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def infer_schema(
+    def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.extraction.list_schemas()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[ExtractionSchema], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def create_schema(
         self,
         *,
-        schema_id: typing.Optional[str] = OMIT,
         name: str,
         project_id: typing.Optional[str] = OMIT,
-        file_ids: typing.List[str],
-        stream: typing.Optional[bool] = OMIT,
+        data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue],
     ) -> ExtractionSchema:
         """
         Parameters:
-            - schema_id: typing.Optional[str]. The ID of a schema to update with the new schema
-
             - name: str. The name of the extraction schema
 
             - project_id: typing.Optional[str]. The ID of the project that the extraction schema belongs to
 
-            - file_ids: typing.List[str]. The IDs of the files that the extraction schema contains
-
-            - stream: typing.Optional[bool]. Whether to stream the results of the extraction schema
+            - data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue]. The schema of the data
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.extraction.infer_schema(
+        client.extraction.create_schema(
             name="string",
-            file_ids=[],
+            data_schema={},
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
-        if schema_id is not OMIT:
-            _request["schema_id"] = schema_id
+        _request: typing.Dict[str, typing.Any] = {"name": name, "data_schema": data_schema}
         if project_id is not OMIT:
             _request["project_id"] = project_id
-        if stream is not OMIT:
-            _request["stream"] = stream
         _response = self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -86,27 +106,53 @@ class ExtractionClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+    def infer_schema(
+        self,
+        *,
+        schema_id: typing.Optional[str] = OMIT,
+        name: str,
+        project_id: typing.Optional[str] = OMIT,
+        file_ids: typing.List[str],
+        stream: typing.Optional[bool] = OMIT,
+    ) -> ExtractionSchema:
         """
         Parameters:
-            - project_id: typing.Optional[str].
+            - schema_id: typing.Optional[str]. The ID of a schema to update with the new schema
+
+            - name: str. The name of the extraction schema
+
+            - project_id: typing.Optional[str]. The ID of the project that the extraction schema belongs to
+
+            - file_ids: typing.List[str]. The IDs of the files that the extraction schema contains
+
+            - stream: typing.Optional[bool]. Whether to stream the results of the extraction schema
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.extraction.list_schemas()
+        client.extraction.infer_schema(
+            name="string",
+            file_ids=[],
+        )
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
+        if schema_id is not OMIT:
+            _request["schema_id"] = schema_id
+        if project_id is not OMIT:
+            _request["project_id"] = project_id
+        if stream is not OMIT:
+            _request["stream"] = stream
         _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionSchema], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionSchema, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -350,47 +396,66 @@ class AsyncExtractionClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def infer_schema(
+    async def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.extraction.list_schemas()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
+            params=remove_none_from_dict({"project_id": project_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[ExtractionSchema], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create_schema(
         self,
         *,
-        schema_id: typing.Optional[str] = OMIT,
         name: str,
         project_id: typing.Optional[str] = OMIT,
-        file_ids: typing.List[str],
-        stream: typing.Optional[bool] = OMIT,
+        data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue],
     ) -> ExtractionSchema:
         """
         Parameters:
-            - schema_id: typing.Optional[str]. The ID of a schema to update with the new schema
-
             - name: str. The name of the extraction schema
 
             - project_id: typing.Optional[str]. The ID of the project that the extraction schema belongs to
 
-            - file_ids: typing.List[str]. The IDs of the files that the extraction schema contains
-
-            - stream: typing.Optional[bool]. Whether to stream the results of the extraction schema
+            - data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue]. The schema of the data
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.extraction.infer_schema(
+        await client.extraction.create_schema(
             name="string",
-            file_ids=[],
+            data_schema={},
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
-        if schema_id is not OMIT:
-            _request["schema_id"] = schema_id
+        _request: typing.Dict[str, typing.Any] = {"name": name, "data_schema": data_schema}
         if project_id is not OMIT:
             _request["project_id"] = project_id
-        if stream is not OMIT:
-            _request["stream"] = stream
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -405,27 +470,53 @@ class AsyncExtractionClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+    async def infer_schema(
+        self,
+        *,
+        schema_id: typing.Optional[str] = OMIT,
+        name: str,
+        project_id: typing.Optional[str] = OMIT,
+        file_ids: typing.List[str],
+        stream: typing.Optional[bool] = OMIT,
+    ) -> ExtractionSchema:
         """
         Parameters:
-            - project_id: typing.Optional[str].
+            - schema_id: typing.Optional[str]. The ID of a schema to update with the new schema
+
+            - name: str. The name of the extraction schema
+
+            - project_id: typing.Optional[str]. The ID of the project that the extraction schema belongs to
+
+            - file_ids: typing.List[str]. The IDs of the files that the extraction schema contains
+
+            - stream: typing.Optional[bool]. Whether to stream the results of the extraction schema
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.extraction.list_schemas()
+        await client.extraction.infer_schema(
+            name="string",
+            file_ids=[],
+        )
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
+        if schema_id is not OMIT:
+            _request["schema_id"] = schema_id
+        if project_id is not OMIT:
+            _request["project_id"] = project_id
+        if stream is not OMIT:
+            _request["stream"] = stream
         _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionSchema], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionSchema, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:

llama_cloud/resources/extraction/types/__init__.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .extraction_schema_create_data_schema_value import ExtractionSchemaCreateDataSchemaValue
 from .extraction_schema_update_data_schema_value import ExtractionSchemaUpdateDataSchemaValue
 
-__all__ = ["ExtractionSchemaUpdateDataSchemaValue"]
+__all__ = ["ExtractionSchemaCreateDataSchemaValue", "ExtractionSchemaUpdateDataSchemaValue"]

llama_cloud/resources/extraction/types/extraction_schema_create_data_schema_value.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ExtractionSchemaCreateDataSchemaValue = typing.Union[
+    typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
+]

llama_cloud/resources/pipelines/__init__.py

@@ -1,5 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import PipelineFileUpdateCustomMetadataValue
+from .types import (
+    PipelineFileUpdateCustomMetadataValue,
+    PipelineUpdateTransformConfig,
+    PipelineUpdateTransformConfig_Advanced,
+    PipelineUpdateTransformConfig_Auto,
+)
 
-__all__ = ["PipelineFileUpdateCustomMetadataValue"]
+__all__ = [
+    "PipelineFileUpdateCustomMetadataValue",
+    "PipelineUpdateTransformConfig",
+    "PipelineUpdateTransformConfig_Advanced",
+    "PipelineUpdateTransformConfig_Auto",
+]

llama_cloud/resources/pipelines/client.py

@@ -15,6 +15,7 @@ from ...types.cloud_document import CloudDocument
 from ...types.cloud_document_create import CloudDocumentCreate
 from ...types.configured_transformation_item import ConfiguredTransformationItem
 from ...types.data_sink_create import DataSinkCreate
+from ...types.embedding_config import EmbeddingConfig
 from ...types.eval_dataset_job_record import EvalDatasetJobRecord
 from ...types.eval_execution_params import EvalExecutionParams
 from ...types.eval_execution_params_override import EvalExecutionParamsOverride
@@ -36,6 +37,7 @@ from ...types.retrieval_mode import RetrievalMode
 from ...types.retrieve_results import RetrieveResults
 from ...types.text_node import TextNode
 from .types.pipeline_file_update_custom_metadata_value import PipelineFileUpdateCustomMetadataValue
+from .types.pipeline_update_transform_config import PipelineUpdateTransformConfig
 
 try:
     import pydantic
@@ -119,6 +121,8 @@ class PipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -136,6 +140,9 @@ class PipelinesClient:
         )
         client.pipelines.create_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -187,6 +194,8 @@ class PipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -204,6 +213,9 @@ class PipelinesClient:
         )
         client.pipelines.upsert_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -278,6 +290,8 @@ class PipelinesClient:
         self,
         pipeline_id: str,
         *,
+        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
+        transform_config: typing.Optional[PipelineUpdateTransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
         data_sink: typing.Optional[DataSinkCreate] = OMIT,
@@ -293,7 +307,11 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
 
-            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]].
+            - embedding_config: typing.Optional[EmbeddingConfig]. Configuration for the embedding model.
+
+            - transform_config: typing.Optional[PipelineUpdateTransformConfig]. Configuration for the transformation.
+
+            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. List of configured transformations.
 
             - data_sink_id: typing.Optional[str]. Data sink ID. When provided instead of data_sink, the data sink will be looked up by ID.
 
@@ -312,6 +330,8 @@ class PipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -327,6 +347,9 @@ class PipelinesClient:
         )
         client.pipelines.update_existing_pipeline(
             pipeline_id="string",
+            embedding_config=EmbeddingConfig(
+                type=EmbeddingConfigType.OPENAI_EMBEDDING,
+            ),
             data_sink=DataSinkCreate(
                 name="string",
                 sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -345,6 +368,10 @@ class PipelinesClient:
         )
         """
         _request: typing.Dict[str, typing.Any] = {}
+        if embedding_config is not OMIT:
+            _request["embedding_config"] = embedding_config
+        if transform_config is not OMIT:
+            _request["transform_config"] = transform_config
         if configured_transformations is not OMIT:
             _request["configured_transformations"] = configured_transformations
         if data_sink_id is not OMIT:
@@ -1281,6 +1308,7 @@ class PipelinesClient:
         from llama_cloud import (
             ChatData,
             FilterCondition,
+            LlmParameters,
             MetadataFilters,
             PresetRetrievalParams,
             RetrievalMode,
@@ -1301,6 +1329,7 @@ class PipelinesClient:
                     ),
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
+                llm_parameters=LlmParameters(),
             ),
         )
         """
@@ -1678,6 +1707,8 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1695,6 +1726,9 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.create_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1746,6 +1780,8 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1763,6 +1799,9 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.upsert_pipeline(
             request=PipelineCreate(
+                embedding_config=EmbeddingConfig(
+                    type=EmbeddingConfigType.OPENAI_EMBEDDING,
+                ),
                 data_sink=DataSinkCreate(
                     name="string",
                     sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1837,6 +1876,8 @@ class AsyncPipelinesClient:
         self,
         pipeline_id: str,
         *,
+        embedding_config: typing.Optional[EmbeddingConfig] = OMIT,
+        transform_config: typing.Optional[PipelineUpdateTransformConfig] = OMIT,
         configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]] = OMIT,
         data_sink_id: typing.Optional[str] = OMIT,
         data_sink: typing.Optional[DataSinkCreate] = OMIT,
@@ -1852,7 +1893,11 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
 
-            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]].
+            - embedding_config: typing.Optional[EmbeddingConfig]. Configuration for the embedding model.
+
+            - transform_config: typing.Optional[PipelineUpdateTransformConfig]. Configuration for the transformation.
+
+            - configured_transformations: typing.Optional[typing.List[ConfiguredTransformationItem]]. List of configured transformations.
 
             - data_sink_id: typing.Optional[str]. Data sink ID. When provided instead of data_sink, the data sink will be looked up by ID.
 
@@ -1871,6 +1916,8 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ConfigurableDataSinkNames,
             DataSinkCreate,
+            EmbeddingConfig,
+            EmbeddingConfigType,
             EvalExecutionParams,
             FilterCondition,
             LlamaParseParameters,
@@ -1886,6 +1933,9 @@ class AsyncPipelinesClient:
         )
         await client.pipelines.update_existing_pipeline(
             pipeline_id="string",
+            embedding_config=EmbeddingConfig(
+                type=EmbeddingConfigType.OPENAI_EMBEDDING,
+            ),
             data_sink=DataSinkCreate(
                 name="string",
                 sink_type=ConfigurableDataSinkNames.CHROMA,
@@ -1904,6 +1954,10 @@ class AsyncPipelinesClient:
         )
         """
         _request: typing.Dict[str, typing.Any] = {}
+        if embedding_config is not OMIT:
+            _request["embedding_config"] = embedding_config
+        if transform_config is not OMIT:
+            _request["transform_config"] = transform_config
         if configured_transformations is not OMIT:
             _request["configured_transformations"] = configured_transformations
         if data_sink_id is not OMIT:
@@ -2842,6 +2896,7 @@ class AsyncPipelinesClient:
         from llama_cloud import (
             ChatData,
             FilterCondition,
+            LlmParameters,
             MetadataFilters,
             PresetRetrievalParams,
             RetrievalMode,
@@ -2862,6 +2917,7 @@ class AsyncPipelinesClient:
                     ),
                     retrieval_mode=RetrievalMode.CHUNKS,
                 ),
+                llm_parameters=LlmParameters(),
             ),
         )
         """

llama_cloud/resources/pipelines/types/__init__.py

@@ -1,5 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .pipeline_file_update_custom_metadata_value import PipelineFileUpdateCustomMetadataValue
+from .pipeline_update_transform_config import (
+    PipelineUpdateTransformConfig,
+    PipelineUpdateTransformConfig_Advanced,
+    PipelineUpdateTransformConfig_Auto,
+)
 
-__all__ = ["PipelineFileUpdateCustomMetadataValue"]
+__all__ = [
+    "PipelineFileUpdateCustomMetadataValue",
+    "PipelineUpdateTransformConfig",
+    "PipelineUpdateTransformConfig_Advanced",
+    "PipelineUpdateTransformConfig_Auto",
+]

llama_cloud/resources/pipelines/types/pipeline_update_transform_config.py

@@ -0,0 +1,31 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from ....types.advanced_mode_transform_config import AdvancedModeTransformConfig
+from ....types.auto_transform_config import AutoTransformConfig
+
+
+class PipelineUpdateTransformConfig_Auto(AutoTransformConfig):
+    mode: typing_extensions.Literal["auto"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class PipelineUpdateTransformConfig_Advanced(AdvancedModeTransformConfig):
+    mode: typing_extensions.Literal["advanced"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+PipelineUpdateTransformConfig = typing.Union[PipelineUpdateTransformConfig_Auto, PipelineUpdateTransformConfig_Advanced]