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

llama_cloud/resources/llama_extract/client.py

@@ -15,10 +15,12 @@ from ...types.extract_job import ExtractJob
 from ...types.extract_job_create import ExtractJobCreate
 from ...types.extract_resultset import ExtractResultset
 from ...types.extract_run import ExtractRun
+from ...types.extract_schema_validate_response import ExtractSchemaValidateResponse
 from ...types.http_validation_error import HttpValidationError
 from ...types.llama_extract_settings import LlamaExtractSettings
-from .types.extract_agent_create_data_schema_value import ExtractAgentCreateDataSchemaValue
-from .types.extract_agent_update_data_schema_value import ExtractAgentUpdateDataSchemaValue
+from .types.extract_agent_create_data_schema import ExtractAgentCreateDataSchema
+from .types.extract_agent_update_data_schema import ExtractAgentUpdateDataSchema
+from .types.extract_schema_validate_request_data_schema import ExtractSchemaValidateRequestDataSchema
 
 try:
     import pydantic
@@ -36,14 +38,10 @@ class LlamaExtractClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list_extraction_agents(
-        self, *, project_id: typing.Optional[str] = None, name: typing.Optional[str] = None
-    ) -> typing.List[ExtractAgent]:
+    def list_extraction_agents(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractAgent]:
         """
         Parameters:
             - project_id: typing.Optional[str].
-
-            - name: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
 
@@ -55,7 +53,7 @@ class LlamaExtractClient:
         _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, "name": name}),
+            params=remove_none_from_dict({"project_id": project_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -75,7 +73,7 @@ class LlamaExtractClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         name: str,
-        data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]],
+        data_schema: ExtractAgentCreateDataSchema,
         config: ExtractConfig,
     ) -> ExtractAgent:
         """
@@ -86,7 +84,7 @@ class LlamaExtractClient:
 
             - name: str. The name of the extraction schema
 
-            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]]. The schema of the data.
+            - data_schema: ExtractAgentCreateDataSchema. The schema of the data.
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
@@ -98,7 +96,6 @@ class LlamaExtractClient:
         )
         client.llama_extract.create_extraction_agent(
             name="string",
-            data_schema={},
             config=ExtractConfig(
                 extraction_mode=ExtractMode.PER_DOC,
             ),
@@ -122,6 +119,77 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def validate_extraction_schema(
+        self, *, data_schema: ExtractSchemaValidateRequestDataSchema
+    ) -> ExtractSchemaValidateResponse:
+        """
+        Validates an extraction agent's schema definition.
+        Returns the normalized and validated schema if valid, otherwise raises an HTTP 400.
+
+        Parameters:
+            - data_schema: ExtractSchemaValidateRequestDataSchema.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.llama_extract.validate_extraction_schema()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction-agents/schema/validation"
+            ),
+            json=jsonable_encoder({"data_schema": data_schema}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ExtractSchemaValidateResponse, _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 get_extraction_agent_by_name(self, name: str, *, project_id: typing.Optional[str] = None) -> ExtractAgent:
+        """
+        Parameters:
+            - name: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.llama_extract.get_extraction_agent_by_name(
+            name="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/extraction-agents/by-name/{name}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id}),
+            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 get_extraction_agent(self, extraction_agent_id: str) -> ExtractAgent:
         """
         Parameters:
@@ -156,17 +224,13 @@ class LlamaExtractClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def update_extraction_agent(
-        self,
-        extraction_agent_id: str,
-        *,
-        data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]],
-        config: ExtractConfig,
+        self, extraction_agent_id: str, *, data_schema: ExtractAgentUpdateDataSchema, config: ExtractConfig
     ) -> ExtractAgent:
         """
         Parameters:
             - extraction_agent_id: str.
 
-            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]]. The schema of the data
+            - data_schema: ExtractAgentUpdateDataSchema. The schema of the data
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
@@ -178,7 +242,6 @@ class LlamaExtractClient:
         )
         client.llama_extract.update_extraction_agent(
             extraction_agent_id="string",
-            data_schema={},
             config=ExtractConfig(
                 extraction_mode=ExtractMode.PER_DOC,
             ),
@@ -350,6 +413,7 @@ class LlamaExtractClient:
             ExtractJobCreate,
             ExtractMode,
             LlamaExtractSettings,
+            LlamaParseParameters,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -364,7 +428,9 @@ class LlamaExtractClient:
                     extraction_mode=ExtractMode.PER_DOC,
                 ),
             ),
-            extract_settings=LlamaExtractSettings(),
+            extract_settings=LlamaExtractSettings(
+                llama_parse_params=LlamaParseParameters(),
+            ),
         )
         """
         _request: typing.Dict[str, typing.Any] = {"job_create": job_create}
@@ -439,6 +505,7 @@ class LlamaExtractClient:
             ExtractJobCreate,
             ExtractMode,
             LlamaExtractSettings,
+            LlamaParseParameters,
         )
         from llama_cloud.client import LlamaCloud
 
@@ -453,7 +520,9 @@ class LlamaExtractClient:
                     extraction_mode=ExtractMode.PER_DOC,
                 ),
             ),
-            extract_settings=LlamaExtractSettings(),
+            extract_settings=LlamaExtractSettings(
+                llama_parse_params=LlamaParseParameters(),
+            ),
         )
         """
         _request: typing.Dict[str, typing.Any] = {"job_create": job_create}
@@ -476,32 +545,61 @@ 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, *, extraction_agent_id: str, file_ids: typing.List[str]) -> typing.List[ExtractJob]:
+    def get_job_result(self, job_id: str) -> ExtractResultset:
         """
         Parameters:
-            - extraction_agent_id: str. The id of the extraction agent
+            - job_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.llama_extract.get_job_result(
+            job_id="string",
+        )
+        """
+        _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}/result"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            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:
+            _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)
 
-            - file_ids: typing.List[str]. The ids of the files
+    def list_extract_runs(self, *, extraction_agent_id: str) -> typing.List[ExtractRun]:
+        """
+        Parameters:
+            - extraction_agent_id: str.
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.llama_extract.run_jobs_in_batch(
+        client.llama_extract.list_extract_runs(
             extraction_agent_id="string",
-            file_ids=[],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/batch"),
-            json=jsonable_encoder({"extraction_agent_id": extraction_agent_id, "file_ids": file_ids}),
+            "GET",
+            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}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractJob], _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:
@@ -510,7 +608,7 @@ 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) -> ExtractResultset:
+    def get_run_by_job_id(self, job_id: str) -> ExtractRun:
         """
         Parameters:
             - job_id: str.
@@ -520,20 +618,20 @@ class LlamaExtractClient:
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.llama_extract.get_job_result(
+        client.llama_extract.get_run_by_job_id(
             job_id="string",
         )
         """
         _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}/result"
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/runs/by-job/{job_id}"
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractResultset, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractRun, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -542,39 +640,28 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    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]:
+    def get_run(self, run_id: str) -> ExtractRun:
         """
         Parameters:
-            - extraction_agent_id: typing.Optional[str].
-
-            - run_id: typing.Optional[str].
-
-            - job_id: typing.Optional[str].
+            - run_id: str.
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.llama_extract.list_extract_runs()
+        client.llama_extract.get_run(
+            run_id="string",
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            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}
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/runs/{run_id}"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractRun], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractRun, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -588,14 +675,10 @@ class AsyncLlamaExtractClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list_extraction_agents(
-        self, *, project_id: typing.Optional[str] = None, name: typing.Optional[str] = None
-    ) -> typing.List[ExtractAgent]:
+    async def list_extraction_agents(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractAgent]:
         """
         Parameters:
             - project_id: typing.Optional[str].
-
-            - name: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -607,7 +690,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/extraction-agents"),
-            params=remove_none_from_dict({"project_id": project_id, "name": name}),
+            params=remove_none_from_dict({"project_id": project_id}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -627,7 +710,7 @@ class AsyncLlamaExtractClient:
         project_id: typing.Optional[str] = None,
         organization_id: typing.Optional[str] = None,
         name: str,
-        data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]],
+        data_schema: ExtractAgentCreateDataSchema,
         config: ExtractConfig,
     ) -> ExtractAgent:
         """
@@ -638,7 +721,7 @@ class AsyncLlamaExtractClient:
 
             - name: str. The name of the extraction schema
 
-            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentCreateDataSchemaValue]]. The schema of the data.
+            - data_schema: ExtractAgentCreateDataSchema. The schema of the data.
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
@@ -650,7 +733,6 @@ class AsyncLlamaExtractClient:
         )
         await client.llama_extract.create_extraction_agent(
             name="string",
-            data_schema={},
             config=ExtractConfig(
                 extraction_mode=ExtractMode.PER_DOC,
             ),
@@ -674,6 +756,77 @@ class AsyncLlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def validate_extraction_schema(
+        self, *, data_schema: ExtractSchemaValidateRequestDataSchema
+    ) -> ExtractSchemaValidateResponse:
+        """
+        Validates an extraction agent's schema definition.
+        Returns the normalized and validated schema if valid, otherwise raises an HTTP 400.
+
+        Parameters:
+            - data_schema: ExtractSchemaValidateRequestDataSchema.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.llama_extract.validate_extraction_schema()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/extraction-agents/schema/validation"
+            ),
+            json=jsonable_encoder({"data_schema": data_schema}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ExtractSchemaValidateResponse, _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 get_extraction_agent_by_name(self, name: str, *, project_id: typing.Optional[str] = None) -> ExtractAgent:
+        """
+        Parameters:
+            - name: str.
+
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.llama_extract.get_extraction_agent_by_name(
+            name="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/extraction-agents/by-name/{name}"
+            ),
+            params=remove_none_from_dict({"project_id": project_id}),
+            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 get_extraction_agent(self, extraction_agent_id: str) -> ExtractAgent:
         """
         Parameters:
@@ -708,17 +861,13 @@ class AsyncLlamaExtractClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def update_extraction_agent(
-        self,
-        extraction_agent_id: str,
-        *,
-        data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]],
-        config: ExtractConfig,
+        self, extraction_agent_id: str, *, data_schema: ExtractAgentUpdateDataSchema, config: ExtractConfig
     ) -> ExtractAgent:
         """
         Parameters:
             - extraction_agent_id: str.
 
-            - data_schema: typing.Dict[str, typing.Optional[ExtractAgentUpdateDataSchemaValue]]. The schema of the data
+            - data_schema: ExtractAgentUpdateDataSchema. The schema of the data
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
@@ -730,7 +879,6 @@ class AsyncLlamaExtractClient:
         )
         await client.llama_extract.update_extraction_agent(
             extraction_agent_id="string",
-            data_schema={},
             config=ExtractConfig(
                 extraction_mode=ExtractMode.PER_DOC,
             ),
@@ -902,6 +1050,7 @@ class AsyncLlamaExtractClient:
             ExtractJobCreate,
             ExtractMode,
             LlamaExtractSettings,
+            LlamaParseParameters,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -916,7 +1065,9 @@ class AsyncLlamaExtractClient:
                     extraction_mode=ExtractMode.PER_DOC,
                 ),
             ),
-            extract_settings=LlamaExtractSettings(),
+            extract_settings=LlamaExtractSettings(
+                llama_parse_params=LlamaParseParameters(),
+            ),
         )
         """
         _request: typing.Dict[str, typing.Any] = {"job_create": job_create}
@@ -991,6 +1142,7 @@ class AsyncLlamaExtractClient:
             ExtractJobCreate,
             ExtractMode,
             LlamaExtractSettings,
+            LlamaParseParameters,
         )
         from llama_cloud.client import AsyncLlamaCloud
 
@@ -1005,7 +1157,9 @@ class AsyncLlamaExtractClient:
                     extraction_mode=ExtractMode.PER_DOC,
                 ),
             ),
-            extract_settings=LlamaExtractSettings(),
+            extract_settings=LlamaExtractSettings(
+                llama_parse_params=LlamaParseParameters(),
+            ),
         )
         """
         _request: typing.Dict[str, typing.Any] = {"job_create": job_create}
@@ -1028,34 +1182,61 @@ 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_jobs_in_batch(
-        self, *, extraction_agent_id: str, file_ids: typing.List[str]
-    ) -> typing.List[ExtractJob]:
+    async def get_job_result(self, job_id: str) -> ExtractResultset:
         """
         Parameters:
-            - extraction_agent_id: str. The id of the extraction agent
+            - job_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.llama_extract.get_job_result(
+            job_id="string",
+        )
+        """
+        _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}/result"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            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:
+            _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)
 
-            - file_ids: typing.List[str]. The ids of the files
+    async def list_extract_runs(self, *, extraction_agent_id: str) -> typing.List[ExtractRun]:
+        """
+        Parameters:
+            - extraction_agent_id: str.
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.llama_extract.run_jobs_in_batch(
+        await client.llama_extract.list_extract_runs(
             extraction_agent_id="string",
-            file_ids=[],
         )
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extractionv2/jobs/batch"),
-            json=jsonable_encoder({"extraction_agent_id": extraction_agent_id, "file_ids": file_ids}),
+            "GET",
+            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}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractJob], _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:
@@ -1064,7 +1245,7 @@ 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) -> ExtractResultset:
+    async def get_run_by_job_id(self, job_id: str) -> ExtractRun:
         """
         Parameters:
             - job_id: str.
@@ -1074,20 +1255,20 @@ class AsyncLlamaExtractClient:
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.llama_extract.get_job_result(
+        await client.llama_extract.get_run_by_job_id(
             job_id="string",
         )
         """
         _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}/result"
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/runs/by-job/{job_id}"
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractResultset, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractRun, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -1096,39 +1277,28 @@ 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_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]:
+    async def get_run(self, run_id: str) -> ExtractRun:
         """
         Parameters:
-            - extraction_agent_id: typing.Optional[str].
-
-            - run_id: typing.Optional[str].
-
-            - job_id: typing.Optional[str].
+            - run_id: str.
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.llama_extract.list_extract_runs()
+        await client.llama_extract.get_run(
+            run_id="string",
+        )
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            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}
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/extractionv2/runs/{run_id}"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractRun], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractRun, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try: