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

llama_cloud/resources/chat_apps/client.py

@@ -33,13 +33,19 @@ class ChatAppsClient:
         self._client_wrapper = client_wrapper
 
     def get_chat_apps_api_v_1_apps_get(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[ChatAppResponse]:
         """
         Parameters:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -52,7 +58,7 @@ class ChatAppsClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/apps"),
             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:
@@ -74,6 +80,7 @@ class ChatAppsClient:
         retriever_id: str,
         llm_config: LlmParameters,
         retrieval_config: PresetCompositeRetrievalParams,
+        project_id: typing.Optional[str] = None,
     ) -> ChatApp:
         """
         Create a new chat app.
@@ -90,6 +97,8 @@ class ChatAppsClient:
             - llm_config: LlmParameters. Configuration for the LLM model to use for the chat app
 
             - retrieval_config: PresetCompositeRetrievalParams. Configuration for the retrieval model to use for the chat app
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import (
             CompositeRetrievalMode,
@@ -130,7 +139,7 @@ class ChatAppsClient:
                     "retrieval_config": retrieval_config,
                 }
             ),
-            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:
@@ -184,6 +193,7 @@ class ChatAppsClient:
         name: typing.Optional[str] = OMIT,
         llm_config: typing.Optional[LlmParameters] = OMIT,
         retrieval_config: typing.Optional[PresetCompositeRetrievalParams] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> ChatApp:
         """
         Update a chat app.
@@ -200,6 +210,8 @@ class ChatAppsClient:
             - llm_config: typing.Optional[LlmParameters].
 
             - retrieval_config: typing.Optional[PresetCompositeRetrievalParams].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import (
             CompositeRetrievalMode,
@@ -239,7 +251,7 @@ class ChatAppsClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
             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:
@@ -326,13 +338,19 @@ class AsyncChatAppsClient:
         self._client_wrapper = client_wrapper
 
     async def get_chat_apps_api_v_1_apps_get(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[ChatAppResponse]:
         """
         Parameters:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -345,7 +363,7 @@ class AsyncChatAppsClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/apps"),
             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:
@@ -367,6 +385,7 @@ class AsyncChatAppsClient:
         retriever_id: str,
         llm_config: LlmParameters,
         retrieval_config: PresetCompositeRetrievalParams,
+        project_id: typing.Optional[str] = None,
     ) -> ChatApp:
         """
         Create a new chat app.
@@ -383,6 +402,8 @@ class AsyncChatAppsClient:
             - llm_config: LlmParameters. Configuration for the LLM model to use for the chat app
 
             - retrieval_config: PresetCompositeRetrievalParams. Configuration for the retrieval model to use for the chat app
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import (
             CompositeRetrievalMode,
@@ -423,7 +444,7 @@ class AsyncChatAppsClient:
                     "retrieval_config": retrieval_config,
                 }
             ),
-            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:
@@ -477,6 +498,7 @@ class AsyncChatAppsClient:
         name: typing.Optional[str] = OMIT,
         llm_config: typing.Optional[LlmParameters] = OMIT,
         retrieval_config: typing.Optional[PresetCompositeRetrievalParams] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> ChatApp:
         """
         Update a chat app.
@@ -493,6 +515,8 @@ class AsyncChatAppsClient:
             - llm_config: typing.Optional[LlmParameters].
 
             - retrieval_config: typing.Optional[PresetCompositeRetrievalParams].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import (
             CompositeRetrievalMode,
@@ -532,7 +556,7 @@ class AsyncChatAppsClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
             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:

llama_cloud/resources/classifier/client.py

@@ -12,9 +12,9 @@ from ...errors.unprocessable_entity_error import UnprocessableEntityError
 from ...types.classifier_rule import ClassifierRule
 from ...types.classify_job import ClassifyJob
 from ...types.classify_job_results import ClassifyJobResults
-from ...types.classify_job_with_status import ClassifyJobWithStatus
 from ...types.classify_parsing_configuration import ClassifyParsingConfiguration
 from ...types.http_validation_error import HttpValidationError
+from ...types.paginated_response_classify_job import PaginatedResponseClassifyJob
 
 try:
     import pydantic
@@ -32,6 +32,61 @@ class ClassifierClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    def list_classify_jobs(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        page_size: typing.Optional[int] = None,
+        page_token: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+    ) -> PaginatedResponseClassifyJob:
+        """
+        List classify jobs.
+        Experimental: This endpoint is not yet ready for production use and is subject to change at any time.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.classifier.list_classify_jobs()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/classifier/jobs"),
+            params=remove_none_from_dict(
+                {
+                    "project_id": project_id,
+                    "organization_id": organization_id,
+                    "page_size": page_size,
+                    "page_token": page_token,
+                }
+            ),
+            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(PaginatedResponseClassifyJob, _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_classify_job(
         self,
         *,
@@ -40,6 +95,7 @@ class ClassifierClient:
         rules: typing.List[ClassifierRule],
         file_ids: typing.List[str],
         parsing_configuration: typing.Optional[ClassifyParsingConfiguration] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> ClassifyJob:
         """
         Create a classify job.
@@ -55,6 +111,8 @@ class ClassifierClient:
             - file_ids: typing.List[str]. The IDs of the files to classify
 
             - parsing_configuration: typing.Optional[ClassifyParsingConfiguration]. The configuration for the parsing job
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ClassifyParsingConfiguration, ParserLanguages
         from llama_cloud.client import LlamaCloud
@@ -78,7 +136,7 @@ class ClassifierClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/classifier/jobs"),
             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:
@@ -97,7 +155,8 @@ class ClassifierClient:
         *,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
-    ) -> ClassifyJobWithStatus:
+        project_id: typing.Optional[str] = None,
+    ) -> ClassifyJob:
         """
         Get a classify job.
         Experimental: This endpoint is not yet ready for production use and is subject to change at any time.
@@ -108,6 +167,8 @@ class ClassifierClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -124,11 +185,11 @@ class ClassifierClient:
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/classifier/jobs/{classify_job_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:
-            return pydantic.parse_obj_as(ClassifyJobWithStatus, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ClassifyJob, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -143,6 +204,7 @@ class ClassifierClient:
         *,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> ClassifyJobResults:
         """
         Get the results of a classify job.
@@ -154,6 +216,8 @@ class ClassifierClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -170,7 +234,7 @@ class ClassifierClient:
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/classifier/jobs/{classify_job_id}/results"
             ),
             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:
@@ -188,6 +252,61 @@ class AsyncClassifierClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
+    async def list_classify_jobs(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        page_size: typing.Optional[int] = None,
+        page_token: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+    ) -> PaginatedResponseClassifyJob:
+        """
+        List classify jobs.
+        Experimental: This endpoint is not yet ready for production use and is subject to change at any time.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - page_size: typing.Optional[int].
+
+            - page_token: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.classifier.list_classify_jobs()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/classifier/jobs"),
+            params=remove_none_from_dict(
+                {
+                    "project_id": project_id,
+                    "organization_id": organization_id,
+                    "page_size": page_size,
+                    "page_token": page_token,
+                }
+            ),
+            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(PaginatedResponseClassifyJob, _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_classify_job(
         self,
         *,
@@ -196,6 +315,7 @@ class AsyncClassifierClient:
         rules: typing.List[ClassifierRule],
         file_ids: typing.List[str],
         parsing_configuration: typing.Optional[ClassifyParsingConfiguration] = OMIT,
+        project_id: typing.Optional[str] = None,
     ) -> ClassifyJob:
         """
         Create a classify job.
@@ -211,6 +331,8 @@ class AsyncClassifierClient:
             - file_ids: typing.List[str]. The IDs of the files to classify
 
             - parsing_configuration: typing.Optional[ClassifyParsingConfiguration]. The configuration for the parsing job
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ClassifyParsingConfiguration, ParserLanguages
         from llama_cloud.client import AsyncLlamaCloud
@@ -234,7 +356,7 @@ class AsyncClassifierClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/classifier/jobs"),
             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:
@@ -253,7 +375,8 @@ class AsyncClassifierClient:
         *,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
-    ) -> ClassifyJobWithStatus:
+        project_id: typing.Optional[str] = None,
+    ) -> ClassifyJob:
         """
         Get a classify job.
         Experimental: This endpoint is not yet ready for production use and is subject to change at any time.
@@ -264,6 +387,8 @@ class AsyncClassifierClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -280,11 +405,11 @@ class AsyncClassifierClient:
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/classifier/jobs/{classify_job_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:
-            return pydantic.parse_obj_as(ClassifyJobWithStatus, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ClassifyJob, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -299,6 +424,7 @@ class AsyncClassifierClient:
         *,
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> ClassifyJobResults:
         """
         Get the results of a classify job.
@@ -310,6 +436,8 @@ class AsyncClassifierClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -326,7 +454,7 @@ class AsyncClassifierClient:
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/classifier/jobs/{classify_job_id}/results"
             ),
             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:

llama_cloud/resources/data_sinks/client.py

@@ -32,7 +32,11 @@ class DataSinksClient:
         self._client_wrapper = client_wrapper
 
     def list_data_sinks(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[DataSink]:
         """
         List data sinks for a given project.
@@ -41,6 +45,8 @@ class DataSinksClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -53,7 +59,7 @@ class DataSinksClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
             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:
@@ -72,6 +78,7 @@ class DataSinksClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSinkCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSink:
         """
         Create a new data sink.
@@ -82,6 +89,8 @@ class DataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
         from llama_cloud.client import LlamaCloud
@@ -101,7 +110,7 @@ class DataSinksClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
             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:
@@ -120,6 +129,7 @@ class DataSinksClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSinkCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSink:
         """
         Upserts a data sink.
@@ -131,6 +141,8 @@ class DataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
         from llama_cloud.client import LlamaCloud
@@ -150,7 +162,7 @@ class DataSinksClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
             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:
@@ -286,7 +298,11 @@ class AsyncDataSinksClient:
         self._client_wrapper = client_wrapper
 
     async def list_data_sinks(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[DataSink]:
         """
         List data sinks for a given project.
@@ -295,6 +311,8 @@ class AsyncDataSinksClient:
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -307,7 +325,7 @@ class AsyncDataSinksClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
             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:
@@ -326,6 +344,7 @@ class AsyncDataSinksClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSinkCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSink:
         """
         Create a new data sink.
@@ -336,6 +355,8 @@ class AsyncDataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
         from llama_cloud.client import AsyncLlamaCloud
@@ -355,7 +376,7 @@ class AsyncDataSinksClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
             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:
@@ -374,6 +395,7 @@ class AsyncDataSinksClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         request: DataSinkCreate,
+        project_id: typing.Optional[str] = None,
     ) -> DataSink:
         """
         Upserts a data sink.
@@ -385,6 +407,8 @@ class AsyncDataSinksClient:
             - organization_id: typing.Optional[str].
 
             - request: DataSinkCreate.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
         from llama_cloud.client import AsyncLlamaCloud
@@ -404,7 +428,7 @@ class AsyncDataSinksClient:
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
             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: