llama-cloud 0.1.20__py3-none-any.whl → 0.1.22__py3-none-any.whl

llama_cloud/resources/data_sinks/types/data_sink_update_component.py

@@ -1,11 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from __future__ import annotations
-
 import typing
 
-import typing_extensions
-
 from ....types.cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from ....types.cloud_milvus_vector_store import CloudMilvusVectorStore
 from ....types.cloud_mongo_db_atlas_vector_search import CloudMongoDbAtlasVectorSearch
@@ -13,66 +9,12 @@ from ....types.cloud_pinecone_vector_store import CloudPineconeVectorStore
 from ....types.cloud_postgres_vector_store import CloudPostgresVectorStore
 from ....types.cloud_qdrant_vector_store import CloudQdrantVectorStore
 
-
-class DataSinkUpdateComponent_AzureAiSearch(CloudAzureAiSearchVectorStore):
-    type: typing_extensions.Literal["azure_ai_search"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class DataSinkUpdateComponent_Milvus(CloudMilvusVectorStore):
-    type: typing_extensions.Literal["milvus"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class DataSinkUpdateComponent_MongodbAtlas(CloudMongoDbAtlasVectorSearch):
-    type: typing_extensions.Literal["mongodb_atlas"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class DataSinkUpdateComponent_Pinecone(CloudPineconeVectorStore):
-    type: typing_extensions.Literal["pinecone"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class DataSinkUpdateComponent_Postgres(CloudPostgresVectorStore):
-    type: typing_extensions.Literal["postgres"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class DataSinkUpdateComponent_Qdrant(CloudQdrantVectorStore):
-    type: typing_extensions.Literal["qdrant"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
 DataSinkUpdateComponent = typing.Union[
-    DataSinkUpdateComponent_AzureAiSearch,
-    DataSinkUpdateComponent_Milvus,
-    DataSinkUpdateComponent_MongodbAtlas,
-    DataSinkUpdateComponent_Pinecone,
-    DataSinkUpdateComponent_Postgres,
-    DataSinkUpdateComponent_Qdrant,
+    typing.Dict[str, typing.Any],
+    CloudPineconeVectorStore,
+    CloudPostgresVectorStore,
+    CloudQdrantVectorStore,
+    CloudAzureAiSearchVectorStore,
+    CloudMongoDbAtlasVectorSearch,
+    CloudMilvusVectorStore,
 ]

llama_cloud/resources/data_sources/__init__.py

@@ -1,33 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import (
-    DataSourceUpdateComponent,
-    DataSourceUpdateComponentOne,
-    DataSourceUpdateComponentOne_AzureStorageBlob,
-    DataSourceUpdateComponentOne_Box,
-    DataSourceUpdateComponentOne_Confluence,
-    DataSourceUpdateComponentOne_GoogleDrive,
-    DataSourceUpdateComponentOne_Jira,
-    DataSourceUpdateComponentOne_MicrosoftOnedrive,
-    DataSourceUpdateComponentOne_MicrosoftSharepoint,
-    DataSourceUpdateComponentOne_NotionPage,
-    DataSourceUpdateComponentOne_S3,
-    DataSourceUpdateComponentOne_Slack,
-    DataSourceUpdateCustomMetadataValue,
-)
+from .types import DataSourceUpdateComponent, DataSourceUpdateCustomMetadataValue
 
-__all__ = [
-    "DataSourceUpdateComponent",
-    "DataSourceUpdateComponentOne",
-    "DataSourceUpdateComponentOne_AzureStorageBlob",
-    "DataSourceUpdateComponentOne_Box",
-    "DataSourceUpdateComponentOne_Confluence",
-    "DataSourceUpdateComponentOne_GoogleDrive",
-    "DataSourceUpdateComponentOne_Jira",
-    "DataSourceUpdateComponentOne_MicrosoftOnedrive",
-    "DataSourceUpdateComponentOne_MicrosoftSharepoint",
-    "DataSourceUpdateComponentOne_NotionPage",
-    "DataSourceUpdateComponentOne_S3",
-    "DataSourceUpdateComponentOne_Slack",
-    "DataSourceUpdateCustomMetadataValue",
-]
+__all__ = ["DataSourceUpdateComponent", "DataSourceUpdateCustomMetadataValue"]

llama_cloud/resources/data_sources/types/__init__.py

@@ -1,33 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .data_source_update_component import DataSourceUpdateComponent
-from .data_source_update_component_one import (
-    DataSourceUpdateComponentOne,
-    DataSourceUpdateComponentOne_AzureStorageBlob,
-    DataSourceUpdateComponentOne_Box,
-    DataSourceUpdateComponentOne_Confluence,
-    DataSourceUpdateComponentOne_GoogleDrive,
-    DataSourceUpdateComponentOne_Jira,
-    DataSourceUpdateComponentOne_MicrosoftOnedrive,
-    DataSourceUpdateComponentOne_MicrosoftSharepoint,
-    DataSourceUpdateComponentOne_NotionPage,
-    DataSourceUpdateComponentOne_S3,
-    DataSourceUpdateComponentOne_Slack,
-)
 from .data_source_update_custom_metadata_value import DataSourceUpdateCustomMetadataValue
 
-__all__ = [
-    "DataSourceUpdateComponent",
-    "DataSourceUpdateComponentOne",
-    "DataSourceUpdateComponentOne_AzureStorageBlob",
-    "DataSourceUpdateComponentOne_Box",
-    "DataSourceUpdateComponentOne_Confluence",
-    "DataSourceUpdateComponentOne_GoogleDrive",
-    "DataSourceUpdateComponentOne_Jira",
-    "DataSourceUpdateComponentOne_MicrosoftOnedrive",
-    "DataSourceUpdateComponentOne_MicrosoftSharepoint",
-    "DataSourceUpdateComponentOne_NotionPage",
-    "DataSourceUpdateComponentOne_S3",
-    "DataSourceUpdateComponentOne_Slack",
-    "DataSourceUpdateCustomMetadataValue",
-]
+__all__ = ["DataSourceUpdateComponent", "DataSourceUpdateCustomMetadataValue"]

llama_cloud/resources/data_sources/types/data_source_update_component.py

@@ -2,6 +2,27 @@
 
 import typing
 
-from .data_source_update_component_one import DataSourceUpdateComponentOne
+from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+from ....types.cloud_box_data_source import CloudBoxDataSource
+from ....types.cloud_confluence_data_source import CloudConfluenceDataSource
+from ....types.cloud_google_drive_data_source import CloudGoogleDriveDataSource
+from ....types.cloud_jira_data_source import CloudJiraDataSource
+from ....types.cloud_notion_page_data_source import CloudNotionPageDataSource
+from ....types.cloud_one_drive_data_source import CloudOneDriveDataSource
+from ....types.cloud_s_3_data_source import CloudS3DataSource
+from ....types.cloud_sharepoint_data_source import CloudSharepointDataSource
+from ....types.cloud_slack_data_source import CloudSlackDataSource
 
-DataSourceUpdateComponent = typing.Union[typing.Dict[str, typing.Any], DataSourceUpdateComponentOne]
+DataSourceUpdateComponent = typing.Union[
+    typing.Dict[str, typing.Any],
+    CloudS3DataSource,
+    CloudAzStorageBlobDataSource,
+    CloudGoogleDriveDataSource,
+    CloudOneDriveDataSource,
+    CloudSharepointDataSource,
+    CloudSlackDataSource,
+    CloudNotionPageDataSource,
+    CloudConfluenceDataSource,
+    CloudJiraDataSource,
+    CloudBoxDataSource,
+]

llama_cloud/resources/jobs/client.py

@@ -37,6 +37,10 @@ class JobsClient:
         """
         Get jobs for a project.
 
+        Note:
+        The include_usage_metrics parameter is deprecated and will be removed in a future version.
+        We've moved to usage v2 and this parameter will no longer return meaningful data.
+
         Parameters:
             - job_name: typing.Optional[str].
 
@@ -44,7 +48,7 @@ class JobsClient:
 
             - offset: typing.Optional[int].
 
-            - include_usage_metrics: typing.Optional[bool].
+            - include_usage_metrics: typing.Optional[bool]. Deprecated: This parameter is no longer supported as we've moved to usage v2. It will be removed in a future version.
 
             - project_id: typing.Optional[str].
 
@@ -101,6 +105,10 @@ class AsyncJobsClient:
         """
         Get jobs for a project.
 
+        Note:
+        The include_usage_metrics parameter is deprecated and will be removed in a future version.
+        We've moved to usage v2 and this parameter will no longer return meaningful data.
+
         Parameters:
             - job_name: typing.Optional[str].
 
@@ -108,7 +116,7 @@ class AsyncJobsClient:
 
             - offset: typing.Optional[int].
 
-            - include_usage_metrics: typing.Optional[bool].
+            - include_usage_metrics: typing.Optional[bool]. Deprecated: This parameter is no longer supported as we've moved to usage v2. It will be removed in a future version.
 
             - project_id: typing.Optional[str].
 

llama_cloud/resources/llama_extract/client.py

@@ -92,7 +92,12 @@ class LlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
+        from llama_cloud import (
+            DocumentChunkMode,
+            ExtractConfig,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -103,6 +108,7 @@ class LlamaExtractClient:
             config=ExtractConfig(
                 extraction_target=ExtractTarget.PER_DOC,
                 extraction_mode=ExtractMode.FAST,
+                chunk_mode=DocumentChunkMode.PAGE,
             ),
         )
         """
@@ -238,7 +244,12 @@ class LlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
+        from llama_cloud import (
+            DocumentChunkMode,
+            ExtractConfig,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -249,6 +260,7 @@ class LlamaExtractClient:
             config=ExtractConfig(
                 extraction_target=ExtractTarget.PER_DOC,
                 extraction_mode=ExtractMode.FAST,
+                chunk_mode=DocumentChunkMode.PAGE,
             ),
         )
         """
@@ -342,6 +354,7 @@ class LlamaExtractClient:
             - request: ExtractJobCreate.
         ---
         from llama_cloud import (
+            DocumentChunkMode,
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
@@ -359,6 +372,7 @@ class LlamaExtractClient:
                 config_override=ExtractConfig(
                     extraction_target=ExtractTarget.PER_DOC,
                     extraction_mode=ExtractMode.FAST,
+                    chunk_mode=DocumentChunkMode.PAGE,
                 ),
             ),
         )
@@ -428,6 +442,7 @@ class LlamaExtractClient:
         ---
         from llama_cloud import (
             ChunkMode,
+            DocumentChunkMode,
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
@@ -449,6 +464,7 @@ class LlamaExtractClient:
                 config_override=ExtractConfig(
                     extraction_target=ExtractTarget.PER_DOC,
                     extraction_mode=ExtractMode.FAST,
+                    chunk_mode=DocumentChunkMode.PAGE,
                 ),
             ),
             extract_settings=LlamaExtractSettings(
@@ -548,7 +564,12 @@ class LlamaExtractClient:
 
             - config_override: typing.Optional[ExtractConfig].
         ---
-        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
+        from llama_cloud import (
+            DocumentChunkMode,
+            ExtractConfig,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -560,6 +581,7 @@ class LlamaExtractClient:
             config_override=ExtractConfig(
                 extraction_target=ExtractTarget.PER_DOC,
                 extraction_mode=ExtractMode.FAST,
+                chunk_mode=DocumentChunkMode.PAGE,
             ),
         )
         """
@@ -831,7 +853,12 @@ class AsyncLlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
+        from llama_cloud import (
+            DocumentChunkMode,
+            ExtractConfig,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -842,6 +869,7 @@ class AsyncLlamaExtractClient:
             config=ExtractConfig(
                 extraction_target=ExtractTarget.PER_DOC,
                 extraction_mode=ExtractMode.FAST,
+                chunk_mode=DocumentChunkMode.PAGE,
             ),
         )
         """
@@ -977,7 +1005,12 @@ class AsyncLlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
+        from llama_cloud import (
+            DocumentChunkMode,
+            ExtractConfig,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -988,6 +1021,7 @@ class AsyncLlamaExtractClient:
             config=ExtractConfig(
                 extraction_target=ExtractTarget.PER_DOC,
                 extraction_mode=ExtractMode.FAST,
+                chunk_mode=DocumentChunkMode.PAGE,
             ),
         )
         """
@@ -1081,6 +1115,7 @@ class AsyncLlamaExtractClient:
             - request: ExtractJobCreate.
         ---
         from llama_cloud import (
+            DocumentChunkMode,
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
@@ -1098,6 +1133,7 @@ class AsyncLlamaExtractClient:
                 config_override=ExtractConfig(
                     extraction_target=ExtractTarget.PER_DOC,
                     extraction_mode=ExtractMode.FAST,
+                    chunk_mode=DocumentChunkMode.PAGE,
                 ),
             ),
         )
@@ -1167,6 +1203,7 @@ class AsyncLlamaExtractClient:
         ---
         from llama_cloud import (
             ChunkMode,
+            DocumentChunkMode,
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
@@ -1188,6 +1225,7 @@ class AsyncLlamaExtractClient:
                 config_override=ExtractConfig(
                     extraction_target=ExtractTarget.PER_DOC,
                     extraction_mode=ExtractMode.FAST,
+                    chunk_mode=DocumentChunkMode.PAGE,
                 ),
             ),
             extract_settings=LlamaExtractSettings(
@@ -1287,7 +1325,12 @@ class AsyncLlamaExtractClient:
 
             - config_override: typing.Optional[ExtractConfig].
         ---
-        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
+        from llama_cloud import (
+            DocumentChunkMode,
+            ExtractConfig,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -1299,6 +1342,7 @@ class AsyncLlamaExtractClient:
             config_override=ExtractConfig(
                 extraction_target=ExtractTarget.PER_DOC,
                 extraction_mode=ExtractMode.FAST,
+                chunk_mode=DocumentChunkMode.PAGE,
             ),
         )
         """

llama_cloud/resources/organizations/client.py

@@ -520,12 +520,16 @@ class OrganizationsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_user_role(self, organization_id: str) -> typing.Optional[UserOrganizationRole]:
+    def get_user_role(
+        self, organization_id: str, *, project_id: typing.Optional[str] = None
+    ) -> typing.Optional[UserOrganizationRole]:
         """
         Get the role of a user in an organization.
 
         Parameters:
             - organization_id: str.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -541,6 +545,7 @@ class OrganizationsClient:
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles"
             ),
+            params=remove_none_from_dict({"project_id": project_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1216,12 +1221,16 @@ class AsyncOrganizationsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_user_role(self, organization_id: str) -> typing.Optional[UserOrganizationRole]:
+    async def get_user_role(
+        self, organization_id: str, *, project_id: typing.Optional[str] = None
+    ) -> typing.Optional[UserOrganizationRole]:
         """
         Get the role of a user in an organization.
 
         Parameters:
             - organization_id: str.
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1237,6 +1246,7 @@ class AsyncOrganizationsClient:
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/organizations/{organization_id}/users/roles"
             ),
+            params=remove_none_from_dict({"project_id": project_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )

llama_cloud/resources/parsing/client.py

@@ -667,6 +667,9 @@ class ParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -738,6 +741,9 @@ class ParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -843,6 +849,9 @@ class ParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -914,6 +923,9 @@ class ParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -983,6 +995,9 @@ class ParsingClient:
         """
         Get parsing history for user
 
+        This endpoint is deprecated.
+        Use /api/v1/jobs/?job_name=parsing&project_id=YOUR_PROJECT_ID instead.
+
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -1680,6 +1695,9 @@ class AsyncParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -1751,6 +1769,9 @@ class AsyncParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -1858,6 +1879,9 @@ class AsyncParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -1931,6 +1955,9 @@ class AsyncParsingClient:
         """
         Get a job by id
 
+        Note: The 'credits_used' and 'job_credits_usage' fields in the response metadata are deprecated
+        and will be removed in a future release.
+
         Parameters:
             - job_id: str.
 
@@ -2000,6 +2027,9 @@ class AsyncParsingClient:
         """
         Get parsing history for user
 
+        This endpoint is deprecated.
+        Use /api/v1/jobs/?job_name=parsing&project_id=YOUR_PROJECT_ID instead.
+
         ---
         from llama_cloud.client import AsyncLlamaCloud
 

llama_cloud/resources/pipelines/client.py

@@ -517,6 +517,7 @@ class PipelinesClient:
         only_manually_uploaded: typing.Optional[bool] = None,
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
+        order_by: typing.Optional[str] = None,
     ) -> PaginatedListPipelineFilesResponse:
         """
         Get files for a pipeline.
@@ -531,6 +532,8 @@ class PipelinesClient:
             - limit: typing.Optional[int].
 
             - offset: typing.Optional[int].
+
+            - order_by: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -550,6 +553,7 @@ class PipelinesClient:
                     "only_manually_uploaded": only_manually_uploaded,
                     "limit": limit,
                     "offset": offset,
+                    "order_by": order_by,
                 }
             ),
             headers=self._client_wrapper.get_headers(),
@@ -1024,6 +1028,8 @@ class PipelinesClient:
         self,
         pipeline_id: str,
         *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
         dense_similarity_top_k: typing.Optional[int] = OMIT,
         dense_similarity_cutoff: typing.Optional[float] = OMIT,
         sparse_similarity_top_k: typing.Optional[int] = OMIT,
@@ -1043,6 +1049,10 @@ class PipelinesClient:
         Parameters:
             - pipeline_id: str.
 
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
             - dense_similarity_top_k: typing.Optional[int].
 
             - dense_similarity_cutoff: typing.Optional[float].
@@ -1109,6 +1119,7 @@ class PipelinesClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/retrieve"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -2124,6 +2135,7 @@ class AsyncPipelinesClient:
         only_manually_uploaded: typing.Optional[bool] = None,
         limit: typing.Optional[int] = None,
         offset: typing.Optional[int] = None,
+        order_by: typing.Optional[str] = None,
     ) -> PaginatedListPipelineFilesResponse:
         """
         Get files for a pipeline.
@@ -2138,6 +2150,8 @@ class AsyncPipelinesClient:
             - limit: typing.Optional[int].
 
             - offset: typing.Optional[int].
+
+            - order_by: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -2157,6 +2171,7 @@ class AsyncPipelinesClient:
                     "only_manually_uploaded": only_manually_uploaded,
                     "limit": limit,
                     "offset": offset,
+                    "order_by": order_by,
                 }
             ),
             headers=self._client_wrapper.get_headers(),
@@ -2633,6 +2648,8 @@ class AsyncPipelinesClient:
         self,
         pipeline_id: str,
         *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
         dense_similarity_top_k: typing.Optional[int] = OMIT,
         dense_similarity_cutoff: typing.Optional[float] = OMIT,
         sparse_similarity_top_k: typing.Optional[int] = OMIT,
@@ -2652,6 +2669,10 @@ class AsyncPipelinesClient:
         Parameters:
             - pipeline_id: str.
 
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
             - dense_similarity_top_k: typing.Optional[int].
 
             - dense_similarity_cutoff: typing.Optional[float].
@@ -2718,6 +2739,7 @@ class AsyncPipelinesClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/retrieve"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,