llama-cloud 0.1.30__py3-none-any.whl → 0.1.32__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/organizations/client.py

@@ -301,7 +301,7 @@ class OrganizationsClient:
         self, organization_id: typing.Optional[str], *, get_current_invoice_total: typing.Optional[bool] = None
     ) -> UsageAndPlan:
         """
-        Get usage for a project
+        Get usage for a specific organization.
 
         Parameters:
             - organization_id: typing.Optional[str].
@@ -1007,7 +1007,7 @@ class AsyncOrganizationsClient:
         self, organization_id: typing.Optional[str], *, get_current_invoice_total: typing.Optional[bool] = None
     ) -> UsageAndPlan:
         """
-        Get usage for a project
+        Get usage for a specific organization.
 
         Parameters:
             - organization_id: typing.Optional[str].

llama_cloud/resources/parsing/client.py

@@ -233,6 +233,7 @@ class ParsingClient:
         language: typing.List[ParserLanguages],
         extract_layout: bool,
         max_pages: typing.Optional[int] = OMIT,
+        merge_tables_across_pages_in_markdown: bool,
         outlined_table_extraction: bool,
         output_pdf_of_document: bool,
         output_s_3_path_prefix: str,
@@ -370,6 +371,8 @@ class ParsingClient:
 
             - max_pages: typing.Optional[int].
 
+            - merge_tables_across_pages_in_markdown: bool.
+
             - outlined_table_extraction: bool.
 
             - output_pdf_of_document: bool.
@@ -518,6 +521,7 @@ class ParsingClient:
             "invalidate_cache": invalidate_cache,
             "language": language,
             "extract_layout": extract_layout,
+            "merge_tables_across_pages_in_markdown": merge_tables_across_pages_in_markdown,
             "outlined_table_extraction": outlined_table_extraction,
             "output_pdf_of_document": output_pdf_of_document,
             "output_s3_path_prefix": output_s_3_path_prefix,
@@ -1389,6 +1393,7 @@ class AsyncParsingClient:
         language: typing.List[ParserLanguages],
         extract_layout: bool,
         max_pages: typing.Optional[int] = OMIT,
+        merge_tables_across_pages_in_markdown: bool,
         outlined_table_extraction: bool,
         output_pdf_of_document: bool,
         output_s_3_path_prefix: str,
@@ -1526,6 +1531,8 @@ class AsyncParsingClient:
 
             - max_pages: typing.Optional[int].
 
+            - merge_tables_across_pages_in_markdown: bool.
+
             - outlined_table_extraction: bool.
 
             - output_pdf_of_document: bool.
@@ -1674,6 +1681,7 @@ class AsyncParsingClient:
             "invalidate_cache": invalidate_cache,
             "language": language,
             "extract_layout": extract_layout,
+            "merge_tables_across_pages_in_markdown": merge_tables_across_pages_in_markdown,
             "outlined_table_extraction": outlined_table_extraction,
             "output_pdf_of_document": output_pdf_of_document,
             "output_s3_path_prefix": output_s_3_path_prefix,