llama-cloud 0.1.29__py3-none-any.whl → 0.1.31__py3-none-any.whl

llama_cloud/resources/beta/client.py

@@ -9,11 +9,15 @@ from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.remove_none_from_dict import remove_none_from_dict
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.agent_data import AgentData
 from ...types.batch import Batch
 from ...types.batch_paginated_list import BatchPaginatedList
 from ...types.batch_public_output import BatchPublicOutput
+from ...types.filter_operation import FilterOperation
 from ...types.http_validation_error import HttpValidationError
 from ...types.llama_parse_parameters import LlamaParseParameters
+from ...types.paginated_response_agent_data import PaginatedResponseAgentData
+from ...types.paginated_response_aggregate_group import PaginatedResponseAggregateGroup
 
 try:
     import pydantic
@@ -204,6 +208,305 @@ class BetaClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_agent_data(self, item_id: str) -> AgentData:
+        """
+        Get agent data by ID.
+
+        Parameters:
+            - item_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.get_agent_data(
+            item_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(AgentData, _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 update_agent_data(self, item_id: str, *, data: typing.Dict[str, typing.Any]) -> AgentData:
+        """
+        Update agent data by ID (overwrites).
+
+        Parameters:
+            - item_id: str.
+
+            - data: typing.Dict[str, typing.Any].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.update_agent_data(
+            item_id="string",
+            data={"string": {}},
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
+            json=jsonable_encoder({"data": data}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(AgentData, _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 delete_agent_data(self, item_id: str) -> typing.Dict[str, str]:
+        """
+        Delete agent data by ID.
+
+        Parameters:
+            - item_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.delete_agent_data(
+            item_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Dict[str, str], _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_agent_data_api_v_1_beta_agent_data_post(
+        self, *, agent_slug: str, collection: typing.Optional[str] = OMIT, data: typing.Dict[str, typing.Any]
+    ) -> AgentData:
+        """
+        Create new agent data.
+
+        Parameters:
+            - agent_slug: str.
+
+            - collection: typing.Optional[str].
+
+            - data: typing.Dict[str, typing.Any].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.create_agent_data_api_v_1_beta_agent_data_post(
+            agent_slug="string",
+            data={"string": {}},
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"agent_slug": agent_slug, "data": data}
+        if collection is not OMIT:
+            _request["collection"] = collection
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(AgentData, _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 search_agent_data_api_v_1_beta_agent_data_search_post(
+        self,
+        *,
+        page_size: typing.Optional[int] = OMIT,
+        page_token: typing.Optional[str] = OMIT,
+        filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]] = OMIT,
+        order_by: typing.Optional[str] = OMIT,
+        agent_slug: str,
+        collection: typing.Optional[str] = OMIT,
+        include_total: typing.Optional[bool] = OMIT,
+        offset: typing.Optional[int] = OMIT,
+    ) -> PaginatedResponseAgentData:
+        """
+        Search agent data with filtering, sorting, and pagination.
+
+        Parameters:
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].
+
+            - order_by: typing.Optional[str].
+
+            - agent_slug: str. The agent deployment's agent_slug to search within
+
+            - collection: typing.Optional[str]. The logical agent data collection to search within
+
+            - include_total: typing.Optional[bool]. Whether to include the total number of items in the response
+
+            - offset: typing.Optional[int].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.search_agent_data_api_v_1_beta_agent_data_search_post(
+            agent_slug="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"agent_slug": agent_slug}
+        if page_size is not OMIT:
+            _request["page_size"] = page_size
+        if page_token is not OMIT:
+            _request["page_token"] = page_token
+        if filter is not OMIT:
+            _request["filter"] = filter
+        if order_by is not OMIT:
+            _request["order_by"] = order_by
+        if collection is not OMIT:
+            _request["collection"] = collection
+        if include_total is not OMIT:
+            _request["include_total"] = include_total
+        if offset is not OMIT:
+            _request["offset"] = offset
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:search"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PaginatedResponseAgentData, _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 aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post(
+        self,
+        *,
+        page_size: typing.Optional[int] = OMIT,
+        page_token: typing.Optional[str] = OMIT,
+        filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]] = OMIT,
+        order_by: typing.Optional[str] = OMIT,
+        agent_slug: str,
+        collection: typing.Optional[str] = OMIT,
+        group_by: typing.Optional[typing.List[str]] = OMIT,
+        count: typing.Optional[bool] = OMIT,
+        first: typing.Optional[bool] = OMIT,
+        offset: typing.Optional[int] = OMIT,
+    ) -> PaginatedResponseAggregateGroup:
+        """
+        Aggregate agent data with grouping and optional counting/first item retrieval.
+
+        Parameters:
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].
+
+            - order_by: typing.Optional[str].
+
+            - agent_slug: str. The agent deployment's agent_slug to aggregate data for
+
+            - collection: typing.Optional[str]. The logical agent data collection to aggregate data for
+
+            - group_by: typing.Optional[typing.List[str]].
+
+            - count: typing.Optional[bool].
+
+            - first: typing.Optional[bool].
+
+            - offset: typing.Optional[int].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post(
+            agent_slug="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"agent_slug": agent_slug}
+        if page_size is not OMIT:
+            _request["page_size"] = page_size
+        if page_token is not OMIT:
+            _request["page_token"] = page_token
+        if filter is not OMIT:
+            _request["filter"] = filter
+        if order_by is not OMIT:
+            _request["order_by"] = order_by
+        if collection is not OMIT:
+            _request["collection"] = collection
+        if group_by is not OMIT:
+            _request["group_by"] = group_by
+        if count is not OMIT:
+            _request["count"] = count
+        if first is not OMIT:
+            _request["first"] = first
+        if offset is not OMIT:
+            _request["offset"] = offset
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:aggregate"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PaginatedResponseAggregateGroup, _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)
+
 
 class AsyncBetaClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -381,3 +684,302 @@ class AsyncBetaClient:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def get_agent_data(self, item_id: str) -> AgentData:
+        """
+        Get agent data by ID.
+
+        Parameters:
+            - item_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.get_agent_data(
+            item_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(AgentData, _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 update_agent_data(self, item_id: str, *, data: typing.Dict[str, typing.Any]) -> AgentData:
+        """
+        Update agent data by ID (overwrites).
+
+        Parameters:
+            - item_id: str.
+
+            - data: typing.Dict[str, typing.Any].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.update_agent_data(
+            item_id="string",
+            data={"string": {}},
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
+            json=jsonable_encoder({"data": data}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(AgentData, _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 delete_agent_data(self, item_id: str) -> typing.Dict[str, str]:
+        """
+        Delete agent data by ID.
+
+        Parameters:
+            - item_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.delete_agent_data(
+            item_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Dict[str, str], _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_agent_data_api_v_1_beta_agent_data_post(
+        self, *, agent_slug: str, collection: typing.Optional[str] = OMIT, data: typing.Dict[str, typing.Any]
+    ) -> AgentData:
+        """
+        Create new agent data.
+
+        Parameters:
+            - agent_slug: str.
+
+            - collection: typing.Optional[str].
+
+            - data: typing.Dict[str, typing.Any].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.create_agent_data_api_v_1_beta_agent_data_post(
+            agent_slug="string",
+            data={"string": {}},
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"agent_slug": agent_slug, "data": data}
+        if collection is not OMIT:
+            _request["collection"] = collection
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(AgentData, _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 search_agent_data_api_v_1_beta_agent_data_search_post(
+        self,
+        *,
+        page_size: typing.Optional[int] = OMIT,
+        page_token: typing.Optional[str] = OMIT,
+        filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]] = OMIT,
+        order_by: typing.Optional[str] = OMIT,
+        agent_slug: str,
+        collection: typing.Optional[str] = OMIT,
+        include_total: typing.Optional[bool] = OMIT,
+        offset: typing.Optional[int] = OMIT,
+    ) -> PaginatedResponseAgentData:
+        """
+        Search agent data with filtering, sorting, and pagination.
+
+        Parameters:
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].
+
+            - order_by: typing.Optional[str].
+
+            - agent_slug: str. The agent deployment's agent_slug to search within
+
+            - collection: typing.Optional[str]. The logical agent data collection to search within
+
+            - include_total: typing.Optional[bool]. Whether to include the total number of items in the response
+
+            - offset: typing.Optional[int].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.search_agent_data_api_v_1_beta_agent_data_search_post(
+            agent_slug="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"agent_slug": agent_slug}
+        if page_size is not OMIT:
+            _request["page_size"] = page_size
+        if page_token is not OMIT:
+            _request["page_token"] = page_token
+        if filter is not OMIT:
+            _request["filter"] = filter
+        if order_by is not OMIT:
+            _request["order_by"] = order_by
+        if collection is not OMIT:
+            _request["collection"] = collection
+        if include_total is not OMIT:
+            _request["include_total"] = include_total
+        if offset is not OMIT:
+            _request["offset"] = offset
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:search"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PaginatedResponseAgentData, _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 aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post(
+        self,
+        *,
+        page_size: typing.Optional[int] = OMIT,
+        page_token: typing.Optional[str] = OMIT,
+        filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]] = OMIT,
+        order_by: typing.Optional[str] = OMIT,
+        agent_slug: str,
+        collection: typing.Optional[str] = OMIT,
+        group_by: typing.Optional[typing.List[str]] = OMIT,
+        count: typing.Optional[bool] = OMIT,
+        first: typing.Optional[bool] = OMIT,
+        offset: typing.Optional[int] = OMIT,
+    ) -> PaginatedResponseAggregateGroup:
+        """
+        Aggregate agent data with grouping and optional counting/first item retrieval.
+
+        Parameters:
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - filter: typing.Optional[typing.Dict[str, typing.Optional[FilterOperation]]].
+
+            - order_by: typing.Optional[str].
+
+            - agent_slug: str. The agent deployment's agent_slug to aggregate data for
+
+            - collection: typing.Optional[str]. The logical agent data collection to aggregate data for
+
+            - group_by: typing.Optional[typing.List[str]].
+
+            - count: typing.Optional[bool].
+
+            - first: typing.Optional[bool].
+
+            - offset: typing.Optional[int].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.aggregate_agent_data_api_v_1_beta_agent_data_aggregate_post(
+            agent_slug="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {"agent_slug": agent_slug}
+        if page_size is not OMIT:
+            _request["page_size"] = page_size
+        if page_token is not OMIT:
+            _request["page_token"] = page_token
+        if filter is not OMIT:
+            _request["filter"] = filter
+        if order_by is not OMIT:
+            _request["order_by"] = order_by
+        if collection is not OMIT:
+            _request["collection"] = collection
+        if group_by is not OMIT:
+            _request["group_by"] = group_by
+        if count is not OMIT:
+            _request["count"] = count
+        if first is not OMIT:
+            _request["first"] = first
+        if offset is not OMIT:
+            _request["offset"] = offset
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:aggregate"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(PaginatedResponseAggregateGroup, _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)

llama_cloud/resources/data_sources/types/data_source_update_component.py

@@ -5,7 +5,6 @@ import typing
 from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
 from ....types.cloud_box_data_source import CloudBoxDataSource
 from ....types.cloud_confluence_data_source import CloudConfluenceDataSource
-from ....types.cloud_google_drive_data_source import CloudGoogleDriveDataSource
 from ....types.cloud_jira_data_source import CloudJiraDataSource
 from ....types.cloud_notion_page_data_source import CloudNotionPageDataSource
 from ....types.cloud_one_drive_data_source import CloudOneDriveDataSource
@@ -17,7 +16,6 @@ DataSourceUpdateComponent = typing.Union[
     typing.Dict[str, typing.Any],
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
-    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
     CloudSlackDataSource,