llama-cloud 0.0.9__py3-none-any.whl → 0.0.11__py3-none-any.whl

llama_cloud/__init__.py

@@ -1,15 +1,17 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AutoTransformConfig,
     AzureOpenAiEmbedding,
     Base,
     BasePromptTemplate,
     BedrockEmbedding,
+    ChatData,
     ChatMessage,
-    ChatParams,
     CloudAzStorageBlobDataSource,
     CloudAzureAiSearchVectorStore,
     CloudChromaVectorStore,
+    CloudConfluenceDataSource,
     CloudDocument,
     CloudDocumentCreate,
     CloudJiraDataSource,
@@ -47,6 +49,9 @@ from .types import (
     DataSourceCreateCustomMetadataValue,
     DataSourceCustomMetadataValue,
     DataSourceDefinition,
+    EmbeddingConfig,
+    EmbeddingConfigComponent,
+    EmbeddingConfigType,
     EvalDataset,
     EvalDatasetJobParams,
     EvalDatasetJobRecord,
@@ -56,6 +61,7 @@ from .types import (
     EvalQuestion,
     EvalQuestionCreate,
     EvalQuestionResult,
+    ExtractionJob,
     ExtractionResult,
     ExtractionResultDataValue,
     ExtractionSchema,
@@ -132,9 +138,12 @@ from .types import (
     TextNodeRelationshipsValue,
     TextNodeWithScore,
     TokenTextSplitter,
+    TransformConfig,
+    TransformConfigMode,
     TransformationCategoryNames,
     UserOrganization,
     UserOrganizationCreate,
+    UserOrganizationDelete,
     ValidationError,
     ValidationErrorLocItem,
 )
@@ -145,6 +154,7 @@ from .resources import (
     DataSourceUpdateComponent,
     DataSourceUpdateComponentOne,
     DataSourceUpdateCustomMetadataValue,
+    ExtractionSchemaCreateDataSchemaValue,
     ExtractionSchemaUpdateDataSchemaValue,
     FileCreateResourceInfoValue,
     PipelineFileUpdateCustomMetadataValue,
@@ -162,15 +172,17 @@ from .resources import (
 from .environment import LlamaCloudEnvironment
 
 __all__ = [
+    "AutoTransformConfig",
     "AzureOpenAiEmbedding",
     "Base",
     "BasePromptTemplate",
     "BedrockEmbedding",
+    "ChatData",
     "ChatMessage",
-    "ChatParams",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
     "CloudChromaVectorStore",
+    "CloudConfluenceDataSource",
     "CloudDocument",
     "CloudDocumentCreate",
     "CloudJiraDataSource",
@@ -213,6 +225,9 @@ __all__ = [
     "DataSourceUpdateComponent",
     "DataSourceUpdateComponentOne",
     "DataSourceUpdateCustomMetadataValue",
+    "EmbeddingConfig",
+    "EmbeddingConfigComponent",
+    "EmbeddingConfigType",
     "EvalDataset",
     "EvalDatasetJobParams",
     "EvalDatasetJobRecord",
@@ -222,9 +237,11 @@ __all__ = [
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
+    "ExtractionJob",
     "ExtractionResult",
     "ExtractionResultDataValue",
     "ExtractionSchema",
+    "ExtractionSchemaCreateDataSchemaValue",
     "ExtractionSchemaDataSchemaValue",
     "ExtractionSchemaUpdateDataSchemaValue",
     "File",
@@ -302,10 +319,13 @@ __all__ = [
     "TextNodeRelationshipsValue",
     "TextNodeWithScore",
     "TokenTextSplitter",
+    "TransformConfig",
+    "TransformConfigMode",
     "TransformationCategoryNames",
     "UnprocessableEntityError",
     "UserOrganization",
     "UserOrganizationCreate",
+    "UserOrganizationDelete",
     "ValidationError",
     "ValidationErrorLocItem",
     "component_definitions",

llama_cloud/resources/__init__.py

@@ -14,7 +14,7 @@ from . import (
 )
 from .data_sinks import DataSinkUpdateComponent, DataSinkUpdateComponentOne
 from .data_sources import DataSourceUpdateComponent, DataSourceUpdateComponentOne, DataSourceUpdateCustomMetadataValue
-from .extraction import ExtractionSchemaUpdateDataSchemaValue
+from .extraction import ExtractionSchemaCreateDataSchemaValue, ExtractionSchemaUpdateDataSchemaValue
 from .files import FileCreateResourceInfoValue
 from .pipelines import PipelineFileUpdateCustomMetadataValue
 
@@ -24,6 +24,7 @@ __all__ = [
     "DataSourceUpdateComponent",
     "DataSourceUpdateComponentOne",
     "DataSourceUpdateCustomMetadataValue",
+    "ExtractionSchemaCreateDataSchemaValue",
     "ExtractionSchemaUpdateDataSchemaValue",
     "FileCreateResourceInfoValue",
     "PipelineFileUpdateCustomMetadataValue",

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_confluence_data_source import CloudConfluenceDataSource
 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
@@ -17,5 +18,6 @@ DataSourceUpdateComponentOne = typing.Union[
     CloudSharepointDataSource,
     CloudSlackDataSource,
     CloudNotionPageDataSource,
+    CloudConfluenceDataSource,
     CloudJiraDataSource,
 ]

llama_cloud/resources/extraction/__init__.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import ExtractionSchemaUpdateDataSchemaValue
+from .types import ExtractionSchemaCreateDataSchemaValue, ExtractionSchemaUpdateDataSchemaValue
 
-__all__ = ["ExtractionSchemaUpdateDataSchemaValue"]
+__all__ = ["ExtractionSchemaCreateDataSchemaValue", "ExtractionSchemaUpdateDataSchemaValue"]

llama_cloud/resources/extraction/client.py

@@ -9,9 +9,11 @@ from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ...core.jsonable_encoder import jsonable_encoder
 from ...core.remove_none_from_dict import remove_none_from_dict
 from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.extraction_job import ExtractionJob
 from ...types.extraction_result import ExtractionResult
 from ...types.extraction_schema import ExtractionSchema
 from ...types.http_validation_error import HttpValidationError
+from .types.extraction_schema_create_data_schema_value import ExtractionSchemaCreateDataSchemaValue
 from .types.extraction_schema_update_data_schema_value import ExtractionSchemaUpdateDataSchemaValue
 
 try:
@@ -30,8 +32,41 @@ class ExtractionClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def infer_schema(
-        self, *, name: str, project_id: typing.Optional[str] = OMIT, file_ids: typing.List[str]
+    def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.extraction.list_schemas()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
+            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(typing.List[ExtractionSchema], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def create_schema(
+        self,
+        *,
+        name: str,
+        project_id: typing.Optional[str] = OMIT,
+        data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue],
     ) -> ExtractionSchema:
         """
         Parameters:
@@ -39,24 +74,24 @@ 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
+            - data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue]. The schema of the data
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.extraction.infer_schema(
+        client.extraction.create_schema(
             name="string",
-            file_ids=[],
+            data_schema={},
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
+        _request: typing.Dict[str, typing.Any] = {"name": name, "data_schema": data_schema}
         if project_id is not OMIT:
             _request["project_id"] = project_id
         _response = self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -71,27 +106,53 @@ class ExtractionClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+    def infer_schema(
+        self,
+        *,
+        schema_id: typing.Optional[str] = OMIT,
+        name: str,
+        project_id: typing.Optional[str] = OMIT,
+        file_ids: typing.List[str],
+        stream: typing.Optional[bool] = OMIT,
+    ) -> ExtractionSchema:
         """
         Parameters:
-            - project_id: typing.Optional[str].
+            - schema_id: typing.Optional[str]. The ID of a schema to update with the new schema
+
+            - name: str. The name of the extraction schema
+
+            - 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
+
+            - stream: typing.Optional[bool]. Whether to stream the results of the extraction schema
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.extraction.list_schemas()
+        client.extraction.infer_schema(
+            name="string",
+            file_ids=[],
+        )
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
+        if schema_id is not OMIT:
+            _request["schema_id"] = schema_id
+        if project_id is not OMIT:
+            _request["project_id"] = project_id
+        if stream is not OMIT:
+            _request["stream"] = stream
         _response = self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionSchema], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionSchema, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -171,17 +232,19 @@ class ExtractionClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def list_jobs(self, *, schema_id: typing.Optional[str] = None) -> typing.List[ExtractionResult]:
+    def list_jobs(self, *, schema_id: str) -> typing.List[ExtractionJob]:
         """
         Parameters:
-            - schema_id: typing.Optional[str].
+            - schema_id: str.
         ---
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
             token="YOUR_TOKEN",
         )
-        client.extraction.list_jobs()
+        client.extraction.list_jobs(
+            schema_id="string",
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
@@ -191,7 +254,7 @@ class ExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionResult], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[ExtractionJob], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -200,7 +263,7 @@ class ExtractionClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def run_job(self, *, schema_id: str, file_id: str) -> ExtractionResult:
+    def run_job(self, *, schema_id: str, file_id: str) -> ExtractionJob:
         """
         Parameters:
             - schema_id: str. The id of the schema
@@ -225,7 +288,7 @@ class ExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractionResult, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionJob, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -234,7 +297,7 @@ class ExtractionClient:
             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) -> ExtractionResult:
+    def get_job(self, job_id: str) -> ExtractionJob:
         """
         Parameters:
             - job_id: str.
@@ -255,7 +318,7 @@ class ExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractionResult, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionJob, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -264,7 +327,7 @@ class ExtractionClient:
             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, *, schema_id: str, file_ids: typing.List[str]) -> typing.List[ExtractionResult]:
+    def run_jobs_in_batch(self, *, schema_id: str, file_ids: typing.List[str]) -> typing.List[ExtractionJob]:
         """
         Parameters:
             - schema_id: str. The id of the schema
@@ -289,7 +352,7 @@ class ExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionResult], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[ExtractionJob], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -333,8 +396,41 @@ class AsyncExtractionClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def infer_schema(
-        self, *, name: str, project_id: typing.Optional[str] = OMIT, file_ids: typing.List[str]
+    async def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.extraction.list_schemas()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
+            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(typing.List[ExtractionSchema], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create_schema(
+        self,
+        *,
+        name: str,
+        project_id: typing.Optional[str] = OMIT,
+        data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue],
     ) -> ExtractionSchema:
         """
         Parameters:
@@ -342,24 +438,24 @@ 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
+            - data_schema: typing.Dict[str, ExtractionSchemaCreateDataSchemaValue]. The schema of the data
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.extraction.infer_schema(
+        await client.extraction.create_schema(
             name="string",
-            file_ids=[],
+            data_schema={},
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
+        _request: typing.Dict[str, typing.Any] = {"name": name, "data_schema": data_schema}
         if project_id is not OMIT:
             _request["project_id"] = project_id
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -374,27 +470,53 @@ class AsyncExtractionClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def list_schemas(self, *, project_id: typing.Optional[str] = None) -> typing.List[ExtractionSchema]:
+    async def infer_schema(
+        self,
+        *,
+        schema_id: typing.Optional[str] = OMIT,
+        name: str,
+        project_id: typing.Optional[str] = OMIT,
+        file_ids: typing.List[str],
+        stream: typing.Optional[bool] = OMIT,
+    ) -> ExtractionSchema:
         """
         Parameters:
-            - project_id: typing.Optional[str].
+            - schema_id: typing.Optional[str]. The ID of a schema to update with the new schema
+
+            - name: str. The name of the extraction schema
+
+            - 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
+
+            - stream: typing.Optional[bool]. Whether to stream the results of the extraction schema
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.extraction.list_schemas()
+        await client.extraction.infer_schema(
+            name="string",
+            file_ids=[],
+        )
         """
+        _request: typing.Dict[str, typing.Any] = {"name": name, "file_ids": file_ids}
+        if schema_id is not OMIT:
+            _request["schema_id"] = schema_id
+        if project_id is not OMIT:
+            _request["project_id"] = project_id
+        if stream is not OMIT:
+            _request["stream"] = stream
         _response = await self._client_wrapper.httpx_client.request(
-            "GET",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas"),
-            params=remove_none_from_dict({"project_id": project_id}),
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/schemas/infer"),
+            json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionSchema], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionSchema, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -474,17 +596,19 @@ class AsyncExtractionClient:
             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, *, schema_id: typing.Optional[str] = None) -> typing.List[ExtractionResult]:
+    async def list_jobs(self, *, schema_id: str) -> typing.List[ExtractionJob]:
         """
         Parameters:
-            - schema_id: typing.Optional[str].
+            - schema_id: str.
         ---
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
             token="YOUR_TOKEN",
         )
-        await client.extraction.list_jobs()
+        await client.extraction.list_jobs(
+            schema_id="string",
+        )
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
@@ -494,7 +618,7 @@ class AsyncExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionResult], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[ExtractionJob], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -503,7 +627,7 @@ class AsyncExtractionClient:
             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, *, schema_id: str, file_id: str) -> ExtractionResult:
+    async def run_job(self, *, schema_id: str, file_id: str) -> ExtractionJob:
         """
         Parameters:
             - schema_id: str. The id of the schema
@@ -528,7 +652,7 @@ class AsyncExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractionResult, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionJob, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -537,7 +661,7 @@ class AsyncExtractionClient:
             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) -> ExtractionResult:
+    async def get_job(self, job_id: str) -> ExtractionJob:
         """
         Parameters:
             - job_id: str.
@@ -558,7 +682,7 @@ class AsyncExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(ExtractionResult, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ExtractionJob, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -567,7 +691,7 @@ class AsyncExtractionClient:
             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, *, schema_id: str, file_ids: typing.List[str]) -> typing.List[ExtractionResult]:
+    async def run_jobs_in_batch(self, *, schema_id: str, file_ids: typing.List[str]) -> typing.List[ExtractionJob]:
         """
         Parameters:
             - schema_id: str. The id of the schema
@@ -592,7 +716,7 @@ class AsyncExtractionClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.List[ExtractionResult], _response.json())  # type: ignore
+            return pydantic.parse_obj_as(typing.List[ExtractionJob], _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:

llama_cloud/resources/extraction/types/__init__.py

@@ -1,5 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from .extraction_schema_create_data_schema_value import ExtractionSchemaCreateDataSchemaValue
 from .extraction_schema_update_data_schema_value import ExtractionSchemaUpdateDataSchemaValue
 
-__all__ = ["ExtractionSchemaUpdateDataSchemaValue"]
+__all__ = ["ExtractionSchemaCreateDataSchemaValue", "ExtractionSchemaUpdateDataSchemaValue"]

llama_cloud/resources/extraction/types/extraction_schema_create_data_schema_value.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ExtractionSchemaCreateDataSchemaValue = typing.Union[
+    typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
+]