llama-cloud 0.1.36__py3-none-any.whl → 0.1.38__py3-none-any.whl

llama_cloud/resources/beta/client.py

@@ -15,6 +15,10 @@ 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.file import File
+from ...types.file_create import FileCreate
+from ...types.file_filter import FileFilter
+from ...types.file_query_response import FileQueryResponse
 from ...types.filter_operation import FilterOperation
 from ...types.http_validation_error import HttpValidationError
 from ...types.llama_parse_parameters import LlamaParseParameters
@@ -45,6 +49,7 @@ class BetaClient:
         offset: typing.Optional[int] = None,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> BatchPaginatedList:
         """
         Parameters:
@@ -55,6 +60,8 @@ class BetaClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -69,7 +76,7 @@ class BetaClient:
             params=remove_none_from_dict(
                 {"limit": limit, "offset": offset, "project_id": project_id, "organization_id": organization_id}
             ),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -96,6 +103,7 @@ class BetaClient:
         batch_create_project_id: str,
         external_id: str,
         completion_window: typing.Optional[int] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> Batch:
         """
         Parameters:
@@ -120,6 +128,8 @@ class BetaClient:
             - external_id: str. A developer-provided ID for the batch. This ID will be returned in the response.
 
             - completion_window: typing.Optional[int]. The time frame within which the batch should be processed. Currently only 24h is supported.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import (
             FailPageMode,
@@ -165,7 +175,7 @@ class BetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/batches"),
             params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -212,7 +222,12 @@ class BetaClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def get_agent_data(
-        self, item_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        item_id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> AgentData:
         """
         Get agent data by ID.
@@ -223,6 +238,8 @@ class BetaClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -237,7 +254,7 @@ class BetaClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -257,6 +274,7 @@ class BetaClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         data: typing.Dict[str, typing.Any],
+        project_id: typing.Optional[str] = None,
     ) -> AgentData:
         """
         Update agent data by ID (overwrites).
@@ -269,6 +287,8 @@ class BetaClient:
             - organization_id: typing.Optional[str].
 
             - data: typing.Dict[str, typing.Any].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -285,7 +305,7 @@ class BetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"data": data}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -299,7 +319,12 @@ class BetaClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def delete_agent_data(
-        self, item_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        item_id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.Dict[str, str]:
         """
         Delete agent data by ID.
@@ -310,6 +335,8 @@ class BetaClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -324,7 +351,7 @@ class BetaClient:
             "DELETE",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -345,6 +372,7 @@ class BetaClient:
         agent_slug: str,
         collection: typing.Optional[str] = OMIT,
         data: typing.Dict[str, typing.Any],
+        project_id: typing.Optional[str] = None,
     ) -> AgentData:
         """
         Create new agent data.
@@ -359,6 +387,8 @@ class BetaClient:
             - collection: typing.Optional[str].
 
             - data: typing.Dict[str, typing.Any].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -378,7 +408,7 @@ class BetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -404,6 +434,7 @@ class BetaClient:
         collection: typing.Optional[str] = OMIT,
         include_total: typing.Optional[bool] = OMIT,
         offset: typing.Optional[int] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> PaginatedResponseAgentData:
         """
         Search agent data with filtering, sorting, and pagination.
@@ -428,6 +459,8 @@ class BetaClient:
             - include_total: typing.Optional[bool]. Whether to include the total number of items in the response
 
             - offset: typing.Optional[int].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -458,7 +491,7 @@ class BetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:search"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -486,6 +519,7 @@ class BetaClient:
         count: typing.Optional[bool] = OMIT,
         first: typing.Optional[bool] = OMIT,
         offset: typing.Optional[int] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> PaginatedResponseAggregateGroup:
         """
         Aggregate agent data with grouping and optional counting/first item retrieval.
@@ -514,6 +548,8 @@ class BetaClient:
             - first: typing.Optional[bool].
 
             - offset: typing.Optional[int].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -548,7 +584,7 @@ class BetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:aggregate"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -610,6 +646,250 @@ class BetaClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def create_file(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: FileCreate,
+        project_id: typing.Optional[str] = None,
+    ) -> File:
+        """
+        Create a new file in the project.
+
+        Args:
+        file_create: File creation data
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        The created file
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - request: FileCreate.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud import FileCreate
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.create_file(
+            request=FileCreate(
+                name="string",
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/files"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 upsert_file(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: FileCreate,
+        project_id: typing.Optional[str] = None,
+    ) -> File:
+        """
+        Upsert a file (create or update if exists) in the project.
+
+        Args:
+        file_create: File creation/update data
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        The upserted file
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - request: FileCreate.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud import FileCreate
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.upsert_file(
+            request=FileCreate(
+                name="string",
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/files"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 query_files(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        page_size: typing.Optional[int] = OMIT,
+        page_token: typing.Optional[str] = OMIT,
+        filter: typing.Optional[FileFilter] = OMIT,
+        order_by: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = None,
+    ) -> FileQueryResponse:
+        """
+        Query files with flexible filtering and pagination.
+
+        Args:
+        request: The query request with filters and pagination
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        Paginated response with files
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - filter: typing.Optional[FileFilter].
+
+            - order_by: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud import FileFilter
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.query_files(
+            filter=FileFilter(),
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        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
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/files/query"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(_request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(FileQueryResponse, _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_file(
+        self,
+        file_id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+    ) -> None:
+        """
+        Delete a single file from the project.
+
+        Args:
+        file_id: The ID of the file to delete
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        None (204 No Content on success)
+
+        Parameters:
+            - file_id: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.beta.delete_file(
+            file_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/files/{file_id}"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        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):
@@ -622,6 +902,7 @@ class AsyncBetaClient:
         offset: typing.Optional[int] = None,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> BatchPaginatedList:
         """
         Parameters:
@@ -632,6 +913,8 @@ class AsyncBetaClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -646,7 +929,7 @@ class AsyncBetaClient:
             params=remove_none_from_dict(
                 {"limit": limit, "offset": offset, "project_id": project_id, "organization_id": organization_id}
             ),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -673,6 +956,7 @@ class AsyncBetaClient:
         batch_create_project_id: str,
         external_id: str,
         completion_window: typing.Optional[int] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> Batch:
         """
         Parameters:
@@ -697,6 +981,8 @@ class AsyncBetaClient:
             - external_id: str. A developer-provided ID for the batch. This ID will be returned in the response.
 
             - completion_window: typing.Optional[int]. The time frame within which the batch should be processed. Currently only 24h is supported.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import (
             FailPageMode,
@@ -742,7 +1028,7 @@ class AsyncBetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/batches"),
             params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -789,7 +1075,12 @@ class AsyncBetaClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def get_agent_data(
-        self, item_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        item_id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> AgentData:
         """
         Get agent data by ID.
@@ -800,6 +1091,8 @@ class AsyncBetaClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -814,7 +1107,7 @@ class AsyncBetaClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -834,6 +1127,7 @@ class AsyncBetaClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         data: typing.Dict[str, typing.Any],
+        project_id: typing.Optional[str] = None,
     ) -> AgentData:
         """
         Update agent data by ID (overwrites).
@@ -846,6 +1140,8 @@ class AsyncBetaClient:
             - organization_id: typing.Optional[str].
 
             - data: typing.Dict[str, typing.Any].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -862,7 +1158,7 @@ class AsyncBetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"data": data}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -876,7 +1172,12 @@ class AsyncBetaClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def delete_agent_data(
-        self, item_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        item_id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.Dict[str, str]:
         """
         Delete agent data by ID.
@@ -887,6 +1188,8 @@ class AsyncBetaClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -901,7 +1204,7 @@ class AsyncBetaClient:
             "DELETE",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/beta/agent-data/{item_id}"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -922,6 +1225,7 @@ class AsyncBetaClient:
         agent_slug: str,
         collection: typing.Optional[str] = OMIT,
         data: typing.Dict[str, typing.Any],
+        project_id: typing.Optional[str] = None,
     ) -> AgentData:
         """
         Create new agent data.
@@ -936,6 +1240,8 @@ class AsyncBetaClient:
             - collection: typing.Optional[str].
 
             - data: typing.Dict[str, typing.Any].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -955,7 +1261,7 @@ class AsyncBetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -981,6 +1287,7 @@ class AsyncBetaClient:
         collection: typing.Optional[str] = OMIT,
         include_total: typing.Optional[bool] = OMIT,
         offset: typing.Optional[int] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> PaginatedResponseAgentData:
         """
         Search agent data with filtering, sorting, and pagination.
@@ -1005,6 +1312,8 @@ class AsyncBetaClient:
             - include_total: typing.Optional[bool]. Whether to include the total number of items in the response
 
             - offset: typing.Optional[int].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1035,7 +1344,7 @@ class AsyncBetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:search"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -1063,6 +1372,7 @@ class AsyncBetaClient:
         count: typing.Optional[bool] = OMIT,
         first: typing.Optional[bool] = OMIT,
         offset: typing.Optional[int] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> PaginatedResponseAggregateGroup:
         """
         Aggregate agent data with grouping and optional counting/first item retrieval.
@@ -1091,6 +1401,8 @@ class AsyncBetaClient:
             - first: typing.Optional[bool].
 
             - offset: typing.Optional[int].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1125,7 +1437,7 @@ class AsyncBetaClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/agent-data/:aggregate"),
             params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
-            headers=self._client_wrapper.get_headers(),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
@@ -1186,3 +1498,247 @@ 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 create_file(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: FileCreate,
+        project_id: typing.Optional[str] = None,
+    ) -> File:
+        """
+        Create a new file in the project.
+
+        Args:
+        file_create: File creation data
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        The created file
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - request: FileCreate.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud import FileCreate
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.create_file(
+            request=FileCreate(
+                name="string",
+            ),
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/files"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 upsert_file(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: FileCreate,
+        project_id: typing.Optional[str] = None,
+    ) -> File:
+        """
+        Upsert a file (create or update if exists) in the project.
+
+        Args:
+        file_create: File creation/update data
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        The upserted file
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - request: FileCreate.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud import FileCreate
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.upsert_file(
+            request=FileCreate(
+                name="string",
+            ),
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "PUT",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/files"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(File, _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 query_files(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        page_size: typing.Optional[int] = OMIT,
+        page_token: typing.Optional[str] = OMIT,
+        filter: typing.Optional[FileFilter] = OMIT,
+        order_by: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = None,
+    ) -> FileQueryResponse:
+        """
+        Query files with flexible filtering and pagination.
+
+        Args:
+        request: The query request with filters and pagination
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        Paginated response with files
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - filter: typing.Optional[FileFilter].
+
+            - order_by: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud import FileFilter
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.query_files(
+            filter=FileFilter(),
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        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
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/beta/files/query"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(_request),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(FileQueryResponse, _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_file(
+        self,
+        file_id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+    ) -> None:
+        """
+        Delete a single file from the project.
+
+        Args:
+        file_id: The ID of the file to delete
+        project: Validated project from dependency
+        db: Database session
+
+        Returns:
+        None (204 No Content on success)
+
+        Parameters:
+            - file_id: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.beta.delete_file(
+            file_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/files/{file_id}"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=remove_none_from_dict({**self._client_wrapper.get_headers(), "Project-Id": project_id}),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        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)