llama-cloud 0.0.8__tar.gz → 0.0.9__tar.gz

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-cloud
-Version: 0.0.8
+Version: 0.0.9
 Summary: 
 Author: Logan Markewich
 Author-email: logan@runllama.ai
@@ -10,6 +10,7 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: httpx (>=0.20.0)
 Requires-Dist: pydantic (>=1.10)
 Description-Content-Type: text/markdown

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/llama_cloud/__init__.py

@@ -6,11 +6,13 @@ from .types import (
     BasePromptTemplate,
     BedrockEmbedding,
     ChatMessage,
+    ChatParams,
     CloudAzStorageBlobDataSource,
     CloudAzureAiSearchVectorStore,
     CloudChromaVectorStore,
     CloudDocument,
     CloudDocumentCreate,
+    CloudJiraDataSource,
     CloudNotionPageDataSource,
     CloudOneDriveDataSource,
     CloudPineconeVectorStore,
@@ -87,6 +89,8 @@ from .types import (
     NodeParser,
     ObjectType,
     OpenAiEmbedding,
+    Organization,
+    OrganizationCreate,
     ParserLanguages,
     ParsingHistoryItem,
     ParsingJob,
@@ -117,6 +121,7 @@ from .types import (
     PromptSpec,
     PydanticProgramMode,
     RelatedNodeInfo,
+    RetrievalMode,
     RetrieveResults,
     SentenceSplitter,
     SimpleFileNodeParser,
@@ -128,6 +133,8 @@ from .types import (
     TextNodeWithScore,
     TokenTextSplitter,
     TransformationCategoryNames,
+    UserOrganization,
+    UserOrganizationCreate,
     ValidationError,
     ValidationErrorLocItem,
 )
@@ -147,6 +154,7 @@ from .resources import (
     evals,
     extraction,
     files,
+    organizations,
     parsing,
     pipelines,
     projects,
@@ -159,11 +167,13 @@ __all__ = [
     "BasePromptTemplate",
     "BedrockEmbedding",
     "ChatMessage",
+    "ChatParams",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
     "CloudChromaVectorStore",
     "CloudDocument",
     "CloudDocumentCreate",
+    "CloudJiraDataSource",
     "CloudNotionPageDataSource",
     "CloudOneDriveDataSource",
     "CloudPineconeVectorStore",
@@ -248,6 +258,8 @@ __all__ = [
     "NodeParser",
     "ObjectType",
     "OpenAiEmbedding",
+    "Organization",
+    "OrganizationCreate",
     "ParserLanguages",
     "ParsingHistoryItem",
     "ParsingJob",
@@ -279,6 +291,7 @@ __all__ = [
     "PromptSpec",
     "PydanticProgramMode",
     "RelatedNodeInfo",
+    "RetrievalMode",
     "RetrieveResults",
     "SentenceSplitter",
     "SimpleFileNodeParser",
@@ -291,6 +304,8 @@ __all__ = [
     "TokenTextSplitter",
     "TransformationCategoryNames",
     "UnprocessableEntityError",
+    "UserOrganization",
+    "UserOrganizationCreate",
     "ValidationError",
     "ValidationErrorLocItem",
     "component_definitions",
@@ -299,6 +314,7 @@ __all__ = [
     "evals",
     "extraction",
     "files",
+    "organizations",
     "parsing",
     "pipelines",
     "projects",

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/llama_cloud/client.py

@@ -12,6 +12,7 @@ from .resources.data_sources.client import AsyncDataSourcesClient, DataSourcesCl
 from .resources.evals.client import AsyncEvalsClient, EvalsClient
 from .resources.extraction.client import AsyncExtractionClient, ExtractionClient
 from .resources.files.client import AsyncFilesClient, FilesClient
+from .resources.organizations.client import AsyncOrganizationsClient, OrganizationsClient
 from .resources.parsing.client import AsyncParsingClient, ParsingClient
 from .resources.pipelines.client import AsyncPipelinesClient, PipelinesClient
 from .resources.projects.client import AsyncProjectsClient, ProjectsClient
@@ -34,6 +35,7 @@ class LlamaCloud:
         )
         self.data_sinks = DataSinksClient(client_wrapper=self._client_wrapper)
         self.data_sources = DataSourcesClient(client_wrapper=self._client_wrapper)
+        self.organizations = OrganizationsClient(client_wrapper=self._client_wrapper)
         self.projects = ProjectsClient(client_wrapper=self._client_wrapper)
         self.files = FilesClient(client_wrapper=self._client_wrapper)
         self.pipelines = PipelinesClient(client_wrapper=self._client_wrapper)
@@ -60,6 +62,7 @@ class AsyncLlamaCloud:
         )
         self.data_sinks = AsyncDataSinksClient(client_wrapper=self._client_wrapper)
         self.data_sources = AsyncDataSourcesClient(client_wrapper=self._client_wrapper)
+        self.organizations = AsyncOrganizationsClient(client_wrapper=self._client_wrapper)
         self.projects = AsyncProjectsClient(client_wrapper=self._client_wrapper)
         self.files = AsyncFilesClient(client_wrapper=self._client_wrapper)
         self.pipelines = AsyncPipelinesClient(client_wrapper=self._client_wrapper)

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/llama_cloud/resources/__init__.py

@@ -1,6 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from . import component_definitions, data_sinks, data_sources, evals, extraction, files, parsing, pipelines, projects
+from . import (
+    component_definitions,
+    data_sinks,
+    data_sources,
+    evals,
+    extraction,
+    files,
+    organizations,
+    parsing,
+    pipelines,
+    projects,
+)
 from .data_sinks import DataSinkUpdateComponent, DataSinkUpdateComponentOne
 from .data_sources import DataSourceUpdateComponent, DataSourceUpdateComponentOne, DataSourceUpdateCustomMetadataValue
 from .extraction import ExtractionSchemaUpdateDataSchemaValue
@@ -22,6 +33,7 @@ __all__ = [
     "evals",
     "extraction",
     "files",
+    "organizations",
     "parsing",
     "pipelines",
     "projects",

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/llama_cloud/resources/data_sinks/client.py

@@ -63,13 +63,21 @@ class DataSinksClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create_data_sink(self, *, project_id: typing.Optional[str] = None, request: DataSinkCreate) -> DataSink:
+    def create_data_sink(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSinkCreate,
+    ) -> DataSink:
         """
         Create a new data sink.
 
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSinkCreate.
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
@@ -88,7 +96,7 @@ class DataSinksClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,
@@ -103,7 +111,13 @@ class DataSinksClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def upsert_data_sink(self, *, project_id: typing.Optional[str] = None, request: DataSinkCreate) -> DataSink:
+    def upsert_data_sink(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSinkCreate,
+    ) -> DataSink:
         """
         Upserts a data sink.
         Updates if a data sink with the same name and project_id already exists. Otherwise, creates a new data sink.
@@ -111,6 +125,8 @@ class DataSinksClient:
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSinkCreate.
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
@@ -129,7 +145,7 @@ class DataSinksClient:
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,
@@ -298,13 +314,21 @@ class AsyncDataSinksClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def create_data_sink(self, *, project_id: typing.Optional[str] = None, request: DataSinkCreate) -> DataSink:
+    async def create_data_sink(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSinkCreate,
+    ) -> DataSink:
         """
         Create a new data sink.
 
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSinkCreate.
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
@@ -323,7 +347,7 @@ class AsyncDataSinksClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,
@@ -338,7 +362,13 @@ class AsyncDataSinksClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def upsert_data_sink(self, *, project_id: typing.Optional[str] = None, request: DataSinkCreate) -> DataSink:
+    async def upsert_data_sink(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSinkCreate,
+    ) -> DataSink:
         """
         Upserts a data sink.
         Updates if a data sink with the same name and project_id already exists. Otherwise, creates a new data sink.
@@ -346,6 +376,8 @@ class AsyncDataSinksClient:
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSinkCreate.
         ---
         from llama_cloud import ConfigurableDataSinkNames, DataSinkCreate
@@ -364,7 +396,7 @@ class AsyncDataSinksClient:
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sinks"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/llama_cloud/resources/data_sources/client.py

@@ -32,13 +32,17 @@ class DataSourcesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list_data_sources(self, *, project_id: typing.Optional[str] = None) -> typing.List[DataSource]:
+    def list_data_sources(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[DataSource]:
         """
         List data sources for a given project.
         If project_id is not provided, uses the default project.
 
         Parameters:
             - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -50,7 +54,7 @@ class DataSourcesClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -64,13 +68,21 @@ class DataSourcesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create_data_source(self, *, project_id: typing.Optional[str] = None, request: DataSourceCreate) -> DataSource:
+    def create_data_source(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSourceCreate,
+    ) -> DataSource:
         """
         Create a new data source.
 
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSourceCreate.
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
@@ -89,7 +101,7 @@ class DataSourcesClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,
@@ -104,7 +116,13 @@ class DataSourcesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def upsert_data_source(self, *, project_id: typing.Optional[str] = None, request: DataSourceCreate) -> DataSource:
+    def upsert_data_source(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSourceCreate,
+    ) -> DataSource:
         """
         Upserts a data source.
         Updates if a data source with the same name and project_id already exists. Otherwise, creates a new data source.
@@ -112,6 +130,8 @@ class DataSourcesClient:
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSourceCreate.
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
@@ -130,7 +150,7 @@ class DataSourcesClient:
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,
@@ -272,13 +292,17 @@ class AsyncDataSourcesClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list_data_sources(self, *, project_id: typing.Optional[str] = None) -> typing.List[DataSource]:
+    async def list_data_sources(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[DataSource]:
         """
         List data sources for a given project.
         If project_id is not provided, uses the default project.
 
         Parameters:
             - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -290,7 +314,7 @@ class AsyncDataSourcesClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -305,7 +329,11 @@ class AsyncDataSourcesClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def create_data_source(
-        self, *, project_id: typing.Optional[str] = None, request: DataSourceCreate
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSourceCreate,
     ) -> DataSource:
         """
         Create a new data source.
@@ -313,6 +341,8 @@ class AsyncDataSourcesClient:
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSourceCreate.
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
@@ -331,7 +361,7 @@ class AsyncDataSourcesClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,
@@ -347,7 +377,11 @@ class AsyncDataSourcesClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def upsert_data_source(
-        self, *, project_id: typing.Optional[str] = None, request: DataSourceCreate
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        request: DataSourceCreate,
     ) -> DataSource:
         """
         Upserts a data source.
@@ -356,6 +390,8 @@ class AsyncDataSourcesClient:
         Parameters:
             - project_id: typing.Optional[str].
 
+            - organization_id: typing.Optional[str].
+
             - request: DataSourceCreate.
         ---
         from llama_cloud import ConfigurableDataSourceNames, DataSourceCreate
@@ -374,7 +410,7 @@ class AsyncDataSourcesClient:
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/data-sources"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            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,

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/llama_cloud/resources/data_sources/types/data_source_update_component_one.py

@@ -3,6 +3,7 @@
 import typing
 
 from ....types.cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+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
@@ -16,4 +17,5 @@ DataSourceUpdateComponentOne = typing.Union[
     CloudSharepointDataSource,
     CloudSlackDataSource,
     CloudNotionPageDataSource,
+    CloudJiraDataSource,
 ]

{llama_cloud-0.0.8 → llama_cloud-0.0.9}/llama_cloud/resources/extraction/client.py

@@ -31,7 +31,7 @@ class ExtractionClient:
         self._client_wrapper = client_wrapper
 
     def infer_schema(
-        self, *, name: str, project_id: typing.Optional[str] = OMIT, file_ids: typing.List[str], openai_api_key: str
+        self, *, name: str, project_id: typing.Optional[str] = OMIT, file_ids: typing.List[str]
     ) -> ExtractionSchema:
         """
         Parameters:
@@ -40,8 +40,6 @@ class ExtractionClient:
             - project_id: typing.Optional[str]. The ID of the project that the extraction schema belongs to
 
             - file_ids: typing.List[str]. The IDs of the files that the extraction schema contains
-
-            - openai_api_key: str. The API key for the OpenAI API
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -51,10 +49,9 @@ class ExtractionClient:
         client.extraction.infer_schema(
             name="string",
             file_ids=[],
-            openai_api_key="string",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids, "openai_api_key": openai_api_key}
+        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
         if project_id is not OMIT:
             _request["project_id"] = project_id
         _response = self._client_wrapper.httpx_client.request(
@@ -138,15 +135,12 @@ class ExtractionClient:
         schema_id: str,
         *,
         data_schema: typing.Optional[typing.Dict[str, ExtractionSchemaUpdateDataSchemaValue]] = OMIT,
-        openai_api_key: typing.Optional[str] = OMIT,
     ) -> ExtractionSchema:
         """
         Parameters:
             - schema_id: str.
 
             - data_schema: typing.Optional[typing.Dict[str, ExtractionSchemaUpdateDataSchemaValue]]. The schema of the data
-
-            - openai_api_key: typing.Optional[str]. The API key for the OpenAI API
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -160,8 +154,6 @@ class ExtractionClient:
         _request: typing.Dict[str, typing.Any] = {}
         if data_schema is not OMIT:
             _request["data_schema"] = data_schema
-        if openai_api_key is not OMIT:
-            _request["openai_api_key"] = openai_api_key
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extraction/schemas/{schema_id}"),
@@ -342,7 +334,7 @@ class AsyncExtractionClient:
         self._client_wrapper = client_wrapper
 
     async def infer_schema(
-        self, *, name: str, project_id: typing.Optional[str] = OMIT, file_ids: typing.List[str], openai_api_key: str
+        self, *, name: str, project_id: typing.Optional[str] = OMIT, file_ids: typing.List[str]
     ) -> ExtractionSchema:
         """
         Parameters:
@@ -351,8 +343,6 @@ class AsyncExtractionClient:
             - project_id: typing.Optional[str]. The ID of the project that the extraction schema belongs to
 
             - file_ids: typing.List[str]. The IDs of the files that the extraction schema contains
-
-            - openai_api_key: str. The API key for the OpenAI API
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -362,10 +352,9 @@ class AsyncExtractionClient:
         await client.extraction.infer_schema(
             name="string",
             file_ids=[],
-            openai_api_key="string",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids, "openai_api_key": openai_api_key}
+        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
         if project_id is not OMIT:
             _request["project_id"] = project_id
         _response = await self._client_wrapper.httpx_client.request(
@@ -449,15 +438,12 @@ class AsyncExtractionClient:
         schema_id: str,
         *,
         data_schema: typing.Optional[typing.Dict[str, ExtractionSchemaUpdateDataSchemaValue]] = OMIT,
-        openai_api_key: typing.Optional[str] = OMIT,
     ) -> ExtractionSchema:
         """
         Parameters:
             - schema_id: str.
 
             - data_schema: typing.Optional[typing.Dict[str, ExtractionSchemaUpdateDataSchemaValue]]. The schema of the data
-
-            - openai_api_key: typing.Optional[str]. The API key for the OpenAI API
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -471,8 +457,6 @@ class AsyncExtractionClient:
         _request: typing.Dict[str, typing.Any] = {}
         if data_schema is not OMIT:
             _request["data_schema"] = data_schema
-        if openai_api_key is not OMIT:
-            _request["openai_api_key"] = openai_api_key
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extraction/schemas/{schema_id}"),