llama-cloud 0.1.8__py3-none-any.whl → 0.1.9__py3-none-any.whl

llama_cloud/resources/llama_extract/client.py

@@ -36,81 +36,26 @@ class LlamaExtractClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def create_extraction_agent(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        name: str,
-        data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]],
-        config: ExtractConfig,
-    ) -> ExtractAgent:
-        """
-        Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
-            - name: str. The name of the extraction schema
-
-            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]]. The schema of the data.
-
-            - config: ExtractConfig. The configuration parameters for the extraction agent.
-        ---
-        from llama_cloud import ExtractConfig, ExtractMode
-        from llama_cloud.client import LlamaCloud
-
-        client = LlamaCloud(
-            token="YOUR_TOKEN",
-        )
-        client.llama_extract.create_extraction_agent(
-            name="string",
-            data_schema={},
-            config=ExtractConfig(
-                extraction_mode=ExtractMode.PER_DOC,
-            ),
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction_agent"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            json=jsonable_encoder({"name": name, "data_schema": data_schema, "config": config}),
-            headers=self._client_wrapper.get_headers(),
-            timeout=60,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractAgent, _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 list_extraction_agents(
-        self, *, project_id: str, organization_id: typing.Optional[str] = None
+        self, *, project_id: typing.Optional[str] = None, name: typing.Optional[str] = None
     ) -> typing.List[ExtractAgent]:
         """
         Parameters:
-            - project_id: str.
+            - project_id: typing.Optional[str].
 
-            - organization_id: typing.Optional[str].
+            - name: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.llama_extract.list_extraction_agents(
-            project_id="string",
-        )
+        client.llama_extract.list_extraction_agents()
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction_agents"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction-agents"),
+            params=remove_none_from_dict({"project_id": project_id, "name": name}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -124,32 +69,46 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_extraction_agent_by_name(
-        self, *, name: str, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    def create_extraction_agent(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        name: str,
+        data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]],
+        config: ExtractConfig,
     ) -> ExtractAgent:
         """
         Parameters:
-            - name: str.
-
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - name: str. The name of the extraction schema
+
+            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]]. The schema of the data.
+
+            - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
+        from llama_cloud import ExtractConfig, ExtractMode
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.llama_extract.get_extraction_agent_by_name(
+        client.llama_extract.create_extraction_agent(
             name="string",
+            data_schema={},
+            config=ExtractConfig(
+                extraction_mode=ExtractMode.PER_DOC,
+            ),
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction_agents/by_name"
-            ),
-            params=remove_none_from_dict({"name": name, "project_id": project_id, "organization_id": organization_id}),
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction-agents"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder({"name": name, "data_schema": data_schema, "config": config}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -163,20 +122,10 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_extraction_agent(
-        self,
-        extraction_agent_id: str,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-    ) -> ExtractAgent:
+    def get_extraction_agent(self, extraction_agent_id: str) -> ExtractAgent:
         """
         Parameters:
             - extraction_agent_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -191,9 +140,8 @@ class LlamaExtractClient:
             "GET",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/",
-                f"api/v1/extractionv2/extraction_agents/{extraction_agent_id}",
+                f"api/v1/extractionv2/extraction-agents/{extraction_agent_id}",
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -211,8 +159,6 @@ class LlamaExtractClient:
         self,
         extraction_agent_id: str,
         *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
         data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]],
         config: ExtractConfig,
     ) -> ExtractAgent:
@@ -220,10 +166,6 @@ class LlamaExtractClient:
         Parameters:
             - extraction_agent_id: str.
 
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]]. The schema of the data
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
@@ -246,9 +188,8 @@ class LlamaExtractClient:
             "PUT",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/",
-                f"api/v1/extractionv2/extraction_agents/{extraction_agent_id}",
+                f"api/v1/extractionv2/extraction-agents/{extraction_agent_id}",
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"data_schema": data_schema, "config": config}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -263,20 +204,10 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def delete_extraction_agent(
-        self,
-        extraction_agent_id: str,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-    ) -> typing.Any:
+    def delete_extraction_agent(self, extraction_agent_id: str) -> typing.Any:
         """
         Parameters:
             - extraction_agent_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -291,9 +222,8 @@ class LlamaExtractClient:
             "DELETE",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/",
-                f"api/v1/extractionv2/extraction_agents/{extraction_agent_id}",
+                f"api/v1/extractionv2/extraction-agents/{extraction_agent_id}",
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -307,20 +237,10 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def list_jobs(
-        self,
-        *,
-        extraction_agent_id: str,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-    ) -> typing.List[ExtractJob]:
+    def list_jobs(self, *, extraction_agent_id: str) -> typing.List[ExtractJob]:
         """
         Parameters:
             - extraction_agent_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -334,13 +254,7 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs"),
-            params=remove_none_from_dict(
-                {
-                    "extraction_agent_id": extraction_agent_id,
-                    "project_id": project_id,
-                    "organization_id": organization_id,
-                }
-            ),
+            params=remove_none_from_dict({"extraction_agent_id": extraction_agent_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -354,19 +268,9 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def run_job(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        request: ExtractJobCreate,
-    ) -> ExtractJob:
+    def run_job(self, *, request: ExtractJobCreate) -> ExtractJob:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - request: ExtractJobCreate.
         ---
         from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
@@ -388,7 +292,6 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs"),
-            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,
@@ -403,16 +306,10 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_job(
-        self, job_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
-    ) -> ExtractJob:
+    def get_job(self, job_id: str) -> ExtractJob:
         """
         Parameters:
             - job_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -426,7 +323,6 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/jobs/{job_id}"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -441,19 +337,10 @@ class LlamaExtractClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def run_job_with_parsed_file_test(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        job_create: ExtractJobCreate,
-        extract_settings: typing.Optional[LlamaExtractSettings] = OMIT,
-    ) -> typing.Optional[ExtractResultset]:
+        self, *, job_create: ExtractJobCreate, extract_settings: typing.Optional[LlamaExtractSettings] = OMIT
+    ) -> ExtractResultset:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - job_create: ExtractJobCreate.
 
             - extract_settings: typing.Optional[LlamaExtractSettings].
@@ -486,13 +373,12 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/parsed/test"),
-            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,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Optional[ExtractResultset], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractResultset, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -501,19 +387,9 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def run_job_with_parsed_file(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        request: ExtractJobCreate,
-    ) -> typing.Optional[ExtractResultset]:
+    def run_job_with_parsed_file(self, *, request: ExtractJobCreate) -> ExtractResultset:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - request: ExtractJobCreate.
         ---
         from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
@@ -535,13 +411,12 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/parsed"),
-            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,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Optional[ExtractResultset], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractResultset, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -551,19 +426,10 @@ class LlamaExtractClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def run_job_test_user(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        job_create: ExtractJobCreate,
-        extract_settings: typing.Optional[LlamaExtractSettings] = OMIT,
+        self, *, job_create: ExtractJobCreate, extract_settings: typing.Optional[LlamaExtractSettings] = OMIT
     ) -> ExtractJob:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - job_create: ExtractJobCreate.
 
             - extract_settings: typing.Optional[LlamaExtractSettings].
@@ -596,7 +462,6 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/test"),
-            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,
@@ -611,20 +476,9 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def run_jobs_in_batch(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        extraction_agent_id: str,
-        file_ids: typing.List[str],
-    ) -> typing.List[ExtractJob]:
+    def run_jobs_in_batch(self, *, extraction_agent_id: str, file_ids: typing.List[str]) -> typing.List[ExtractJob]:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - extraction_agent_id: str. The id of the extraction agent
 
             - file_ids: typing.List[str]. The ids of the files
@@ -642,7 +496,6 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/batch"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"extraction_agent_id": extraction_agent_id, "file_ids": file_ids}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -657,16 +510,10 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_job_result(
-        self, job_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
-    ) -> ExtractResultset:
+    def get_job_result(self, job_id: str) -> ExtractResultset:
         """
         Parameters:
             - job_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -682,7 +529,6 @@ class LlamaExtractClient:
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/jobs/{job_id}/result"
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -696,35 +542,39 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_extract_run_api_v_1_extractionv_2_runs_run_id_get(
-        self, run_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
-    ) -> ExtractRun:
+    def list_extract_runs(
+        self,
+        *,
+        extraction_agent_id: typing.Optional[str] = None,
+        run_id: typing.Optional[str] = None,
+        job_id: typing.Optional[str] = None,
+    ) -> typing.List[ExtractRun]:
         """
         Parameters:
-            - run_id: str.
+            - extraction_agent_id: typing.Optional[str].
 
-            - project_id: typing.Optional[str].
+            - run_id: typing.Optional[str].
 
-            - organization_id: typing.Optional[str].
+            - job_id: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.llama_extract.get_extract_run_api_v_1_extractionv_2_runs_run_id_get(
-            run_id="string",
-        )
+        client.llama_extract.list_extract_runs()
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/runs/{run_id}"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/runs"),
+            params=remove_none_from_dict(
+                {"extraction_agent_id": extraction_agent_id, "run_id": run_id, "job_id": job_id}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractRun, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[ExtractRun], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -738,81 +588,26 @@ class AsyncLlamaExtractClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def create_extraction_agent(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        name: str,
-        data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]],
-        config: ExtractConfig,
-    ) -> ExtractAgent:
-        """
-        Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
-            - name: str. The name of the extraction schema
-
-            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]]. The schema of the data.
-
-            - config: ExtractConfig. The configuration parameters for the extraction agent.
-        ---
-        from llama_cloud import ExtractConfig, ExtractMode
-        from llama_cloud.client import AsyncLlamaCloud
-
-        client = AsyncLlamaCloud(
-            token="YOUR_TOKEN",
-        )
-        await client.llama_extract.create_extraction_agent(
-            name="string",
-            data_schema={},
-            config=ExtractConfig(
-                extraction_mode=ExtractMode.PER_DOC,
-            ),
-        )
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction_agent"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
-            json=jsonable_encoder({"name": name, "data_schema": data_schema, "config": config}),
-            headers=self._client_wrapper.get_headers(),
-            timeout=60,
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractAgent, _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 list_extraction_agents(
-        self, *, project_id: str, organization_id: typing.Optional[str] = None
+        self, *, project_id: typing.Optional[str] = None, name: typing.Optional[str] = None
     ) -> typing.List[ExtractAgent]:
         """
         Parameters:
-            - project_id: str.
+            - project_id: typing.Optional[str].
 
-            - organization_id: typing.Optional[str].
+            - name: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.llama_extract.list_extraction_agents(
-            project_id="string",
-        )
+        await client.llama_extract.list_extraction_agents()
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction_agents"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction-agents"),
+            params=remove_none_from_dict({"project_id": project_id, "name": name}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -826,32 +621,46 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_extraction_agent_by_name(
-        self, *, name: str, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    async def create_extraction_agent(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        name: str,
+        data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]],
+        config: ExtractConfig,
     ) -> ExtractAgent:
         """
         Parameters:
-            - name: str.
-
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
+
+            - name: str. The name of the extraction schema
+
+            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]]. The schema of the data.
+
+            - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
+        from llama_cloud import ExtractConfig, ExtractMode
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.llama_extract.get_extraction_agent_by_name(
+        await client.llama_extract.create_extraction_agent(
             name="string",
+            data_schema={},
+            config=ExtractConfig(
+                extraction_mode=ExtractMode.PER_DOC,
+            ),
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction_agents/by_name"
-            ),
-            params=remove_none_from_dict({"name": name, "project_id": project_id, "organization_id": organization_id}),
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction-agents"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder({"name": name, "data_schema": data_schema, "config": config}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -865,20 +674,10 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_extraction_agent(
-        self,
-        extraction_agent_id: str,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-    ) -> ExtractAgent:
+    async def get_extraction_agent(self, extraction_agent_id: str) -> ExtractAgent:
         """
         Parameters:
             - extraction_agent_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -893,9 +692,8 @@ class AsyncLlamaExtractClient:
             "GET",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/",
-                f"api/v1/extractionv2/extraction_agents/{extraction_agent_id}",
+                f"api/v1/extractionv2/extraction-agents/{extraction_agent_id}",
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -913,8 +711,6 @@ class AsyncLlamaExtractClient:
         self,
         extraction_agent_id: str,
         *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
         data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]],
         config: ExtractConfig,
     ) -> ExtractAgent:
@@ -922,10 +718,6 @@ class AsyncLlamaExtractClient:
         Parameters:
             - extraction_agent_id: str.
 
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]]. The schema of the data
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
@@ -948,9 +740,8 @@ class AsyncLlamaExtractClient:
             "PUT",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/",
-                f"api/v1/extractionv2/extraction_agents/{extraction_agent_id}",
+                f"api/v1/extractionv2/extraction-agents/{extraction_agent_id}",
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"data_schema": data_schema, "config": config}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -965,20 +756,10 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def delete_extraction_agent(
-        self,
-        extraction_agent_id: str,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-    ) -> typing.Any:
+    async def delete_extraction_agent(self, extraction_agent_id: str) -> typing.Any:
         """
         Parameters:
             - extraction_agent_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -993,9 +774,8 @@ class AsyncLlamaExtractClient:
             "DELETE",
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/",
-                f"api/v1/extractionv2/extraction_agents/{extraction_agent_id}",
+                f"api/v1/extractionv2/extraction-agents/{extraction_agent_id}",
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1009,20 +789,10 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def list_jobs(
-        self,
-        *,
-        extraction_agent_id: str,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-    ) -> typing.List[ExtractJob]:
+    async def list_jobs(self, *, extraction_agent_id: str) -> typing.List[ExtractJob]:
         """
         Parameters:
             - extraction_agent_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1036,13 +806,7 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs"),
-            params=remove_none_from_dict(
-                {
-                    "extraction_agent_id": extraction_agent_id,
-                    "project_id": project_id,
-                    "organization_id": organization_id,
-                }
-            ),
+            params=remove_none_from_dict({"extraction_agent_id": extraction_agent_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1056,19 +820,9 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def run_job(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        request: ExtractJobCreate,
-    ) -> ExtractJob:
+    async def run_job(self, *, request: ExtractJobCreate) -> ExtractJob:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - request: ExtractJobCreate.
         ---
         from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
@@ -1090,7 +844,6 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs"),
-            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,
@@ -1105,16 +858,10 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_job(
-        self, job_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
-    ) -> ExtractJob:
+    async def get_job(self, job_id: str) -> ExtractJob:
         """
         Parameters:
             - job_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1128,7 +875,6 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/jobs/{job_id}"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1143,19 +889,10 @@ class AsyncLlamaExtractClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def run_job_with_parsed_file_test(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        job_create: ExtractJobCreate,
-        extract_settings: typing.Optional[LlamaExtractSettings] = OMIT,
-    ) -> typing.Optional[ExtractResultset]:
+        self, *, job_create: ExtractJobCreate, extract_settings: typing.Optional[LlamaExtractSettings] = OMIT
+    ) -> ExtractResultset:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - job_create: ExtractJobCreate.
 
             - extract_settings: typing.Optional[LlamaExtractSettings].
@@ -1188,13 +925,12 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/parsed/test"),
-            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,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Optional[ExtractResultset], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractResultset, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -1203,19 +939,9 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def run_job_with_parsed_file(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        request: ExtractJobCreate,
-    ) -> typing.Optional[ExtractResultset]:
+    async def run_job_with_parsed_file(self, *, request: ExtractJobCreate) -> ExtractResultset:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - request: ExtractJobCreate.
         ---
         from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
@@ -1237,13 +963,12 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/parsed"),
-            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,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Optional[ExtractResultset], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractResultset, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -1253,19 +978,10 @@ class AsyncLlamaExtractClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def run_job_test_user(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        job_create: ExtractJobCreate,
-        extract_settings: typing.Optional[LlamaExtractSettings] = OMIT,
+        self, *, job_create: ExtractJobCreate, extract_settings: typing.Optional[LlamaExtractSettings] = OMIT
     ) -> ExtractJob:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - job_create: ExtractJobCreate.
 
             - extract_settings: typing.Optional[LlamaExtractSettings].
@@ -1298,7 +1014,6 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/test"),
-            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,
@@ -1314,19 +1029,10 @@ class AsyncLlamaExtractClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def run_jobs_in_batch(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        organization_id: typing.Optional[str] = None,
-        extraction_agent_id: str,
-        file_ids: typing.List[str],
+        self, *, extraction_agent_id: str, file_ids: typing.List[str]
     ) -> typing.List[ExtractJob]:
         """
         Parameters:
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
-
             - extraction_agent_id: str. The id of the extraction agent
 
             - file_ids: typing.List[str]. The ids of the files
@@ -1344,7 +1050,6 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/batch"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             json=jsonable_encoder({"extraction_agent_id": extraction_agent_id, "file_ids": file_ids}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1359,16 +1064,10 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_job_result(
-        self, job_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
-    ) -> ExtractResultset:
+    async def get_job_result(self, job_id: str) -> ExtractResultset:
         """
         Parameters:
             - job_id: str.
-
-            - project_id: typing.Optional[str].
-
-            - organization_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1384,7 +1083,6 @@ class AsyncLlamaExtractClient:
             urllib.parse.urljoin(
                 f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/jobs/{job_id}/result"
             ),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1398,35 +1096,39 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_extract_run_api_v_1_extractionv_2_runs_run_id_get(
-        self, run_id: str, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
-    ) -> ExtractRun:
+    async def list_extract_runs(
+        self,
+        *,
+        extraction_agent_id: typing.Optional[str] = None,
+        run_id: typing.Optional[str] = None,
+        job_id: typing.Optional[str] = None,
+    ) -> typing.List[ExtractRun]:
         """
         Parameters:
-            - run_id: str.
+            - extraction_agent_id: typing.Optional[str].
 
-            - project_id: typing.Optional[str].
+            - run_id: typing.Optional[str].
 
-            - organization_id: typing.Optional[str].
+            - job_id: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.llama_extract.get_extract_run_api_v_1_extractionv_2_runs_run_id_get(
-            run_id="string",
-        )
+        await client.llama_extract.list_extract_runs()
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/runs/{run_id}"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/runs"),
+            params=remove_none_from_dict(
+                {"extraction_agent_id": extraction_agent_id, "run_id": run_id, "job_id": job_id}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractRun, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[ExtractRun], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try: