llama-cloud 0.1.5__py3-none-any.whl → 0.1.7a1__py3-none-any.whl

llama_cloud/types/edit_suggestion.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .report_block import ReportBlock
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class EditSuggestion(pydantic.BaseModel):
+    """
+    A suggestion for an edit to a report.
+    """
+
+    justification: str
+    start_line: int
+    end_line: int
+    blocks: typing.List[ReportBlock]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/eval_dataset_job_record.py

@@ -34,6 +34,7 @@ class EvalDatasetJobRecord(pydantic.BaseModel):
     parent_job_execution_id: typing.Optional[str]
     user_id: typing.Optional[str]
     created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
+    project_id: typing.Optional[str]
     id: typing.Optional[str] = pydantic.Field(description="Unique identifier")
     status: StatusEnum
     error_code: typing.Optional[str]

llama_cloud/types/extract_agent.py

@@ -0,0 +1,45 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .extract_agent_data_schema_value import ExtractAgentDataSchemaValue
+from .extract_config import ExtractConfig
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractAgent(pydantic.BaseModel):
+    """
+    Schema and configuration for creating an extraction agent.
+    """
+
+    id: str = pydantic.Field(description="The id of the extraction agent.")
+    name: str = pydantic.Field(description="The name of the extraction agent.")
+    project_id: str = pydantic.Field(description="The ID of the project that the extraction agent belongs to.")
+    data_schema: typing.Dict[str, typing.Optional[ExtractAgentDataSchemaValue]] = pydantic.Field(
+        description="The schema of the data."
+    )
+    config: ExtractConfig = pydantic.Field(description="The configuration parameters for the extraction agent.")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/extract_agent_data_schema_value.py

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

llama_cloud/types/extract_config.py

@@ -0,0 +1,40 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .extract_mode import ExtractMode
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractConfig(pydantic.BaseModel):
+    """
+    Additional parameters for the extraction agent.
+    """
+
+    extraction_mode: typing.Optional[ExtractMode] = pydantic.Field(description="The extraction mode specified.")
+    handle_missing: typing.Optional[bool] = pydantic.Field(
+        description="Whether to handle missing fields in the schema."
+    )
+    system_prompt: typing.Optional[str]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/extract_job.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .file import File
+from .status_enum import StatusEnum
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractJob(pydantic.BaseModel):
+    id: str = pydantic.Field(description="The id of the extraction job")
+    status: StatusEnum = pydantic.Field(description="The status of the extraction job")
+    file: File = pydantic.Field(description="The file that the extract was extracted from")
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/extract_job_create.py

@@ -0,0 +1,40 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .extract_config import ExtractConfig
+from .extract_job_create_data_schema_override_value import ExtractJobCreateDataSchemaOverrideValue
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractJobCreate(pydantic.BaseModel):
+    """
+    Schema for creating an extraction job.
+    """
+
+    extraction_agent_id: str = pydantic.Field(description="The id of the extraction agent")
+    file_id: str = pydantic.Field(description="The id of the file")
+    data_schema_override: typing.Optional[typing.Dict[str, typing.Optional[ExtractJobCreateDataSchemaOverrideValue]]]
+    config_override: typing.Optional[ExtractConfig]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/extract_job_create_data_schema_override_value.py

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

llama_cloud/types/extract_mode.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class ExtractMode(str, enum.Enum):
+    PER_DOC = "PER_DOC"
+    PER_PAGE = "PER_PAGE"
+
+    def visit(self, per_doc: typing.Callable[[], T_Result], per_page: typing.Callable[[], T_Result]) -> T_Result:
+        if self is ExtractMode.PER_DOC:
+            return per_doc()
+        if self is ExtractMode.PER_PAGE:
+            return per_page()

llama_cloud/types/extract_resultset.py

@@ -0,0 +1,46 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .extract_resultset_data import ExtractResultsetData
+from .extract_resultset_extraction_metadata_value import ExtractResultsetExtractionMetadataValue
+from .file import File
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractResultset(pydantic.BaseModel):
+    """
+    Schema for an extraction result.
+    """
+
+    id: str = pydantic.Field(description="Unique identifier")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
+    extraction_agent_id: str = pydantic.Field(description="The id of the extraction agent")
+    data: typing.Optional[ExtractResultsetData] = pydantic.Field(description="The data extracted from the file")
+    extraction_metadata: typing.Dict[str, typing.Optional[ExtractResultsetExtractionMetadataValue]] = pydantic.Field(
+        description="The metadata extracted from the file"
+    )
+    file: File = pydantic.Field(description="The file that the extract was extracted from")
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/extract_resultset_data.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .extract_resultset_data_item_value import ExtractResultsetDataItemValue
+from .extract_resultset_data_zero_value import ExtractResultsetDataZeroValue
+
+ExtractResultsetData = typing.Union[
+    typing.Dict[str, typing.Optional[ExtractResultsetDataZeroValue]],
+    typing.List[typing.Dict[str, typing.Optional[ExtractResultsetDataItemValue]]],
+]

llama_cloud/types/extract_resultset_data_item_value.py

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

llama_cloud/types/extract_resultset_data_zero_value.py

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

llama_cloud/types/extract_resultset_extraction_metadata_value.py

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

llama_cloud/types/file.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .file_permission_info_value import FilePermissionInfoValue
 from .file_resource_info_value import FileResourceInfoValue
 
 try:
@@ -24,11 +25,13 @@ class File(pydantic.BaseModel):
     created_at: typing.Optional[dt.datetime]
     updated_at: typing.Optional[dt.datetime]
     name: str
+    external_file_id: str = pydantic.Field(description="The ID of the file in the external system")
     file_size: typing.Optional[int]
     file_type: typing.Optional[str]
     project_id: str = pydantic.Field(description="The ID of the project that the file belongs to")
     last_modified_at: typing.Optional[dt.datetime]
     resource_info: typing.Optional[typing.Dict[str, typing.Optional[FileResourceInfoValue]]]
+    permission_info: typing.Optional[typing.Dict[str, typing.Optional[FilePermissionInfoValue]]]
     data_source_id: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/file_permission_info_value.py

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

llama_cloud/types/filter_condition.py

@@ -13,9 +13,17 @@ class FilterCondition(str, enum.Enum):
 
     AND = "and"
     OR = "or"
+    NOT = "not"
 
-    def visit(self, and_: typing.Callable[[], T_Result], or_: typing.Callable[[], T_Result]) -> T_Result:
+    def visit(
+        self,
+        and_: typing.Callable[[], T_Result],
+        or_: typing.Callable[[], T_Result],
+        not_: typing.Callable[[], T_Result],
+    ) -> T_Result:
         if self is FilterCondition.AND:
             return and_()
         if self is FilterCondition.OR:
             return or_()
+        if self is FilterCondition.NOT:
+            return not_()

llama_cloud/types/filter_operator.py

@@ -22,6 +22,7 @@ class FilterOperator(str, enum.Enum):
     ANY = "any"
     ALL = "all"
     TEXT_MATCH = "text_match"
+    TEXT_MATCH_INSENSITIVE = "text_match_insensitive"
     CONTAINS = "contains"
     IS_EMPTY = "is_empty"
 
@@ -38,6 +39,7 @@ class FilterOperator(str, enum.Enum):
         any: typing.Callable[[], T_Result],
         all: typing.Callable[[], T_Result],
         text_match: typing.Callable[[], T_Result],
+        text_match_insensitive: typing.Callable[[], T_Result],
         contains: typing.Callable[[], T_Result],
         is_empty: typing.Callable[[], T_Result],
     ) -> T_Result:
@@ -63,6 +65,8 @@ class FilterOperator(str, enum.Enum):
             return all()
         if self is FilterOperator.TEXT_MATCH:
             return text_match()
+        if self is FilterOperator.TEXT_MATCH_INSENSITIVE:
+            return text_match_insensitive()
         if self is FilterOperator.CONTAINS:
             return contains()
         if self is FilterOperator.IS_EMPTY:

llama_cloud/types/image_block.py

@@ -0,0 +1,35 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ImageBlock(pydantic.BaseModel):
+    image: typing.Optional[str]
+    path: typing.Optional[str]
+    url: typing.Optional[str]
+    image_mimetype: typing.Optional[str]
+    detail: typing.Optional[str]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/input_message.py

@@ -20,7 +20,7 @@ class InputMessage(pydantic.BaseModel):
     This is distinct from a ChatMessage because this schema is enforced by the AI Chat library used in the frontend
     """
 
-    id: str = pydantic.Field(description="ID of the message, if any. a UUID.")
+    id: typing.Optional[str] = pydantic.Field(description="ID of the message, if any. a UUID.")
     role: MessageRole
     content: str
     data: typing.Optional[typing.Dict[str, typing.Any]]

llama_cloud/types/job_name_mapping.py

@@ -18,6 +18,7 @@ class JobNameMapping(str, enum.Enum):
     PARSE = "PARSE"
     TRANSFORM = "TRANSFORM"
     INGESTION = "INGESTION"
+    METADATA_UPDATE = "METADATA_UPDATE"
 
     def visit(
         self,
@@ -28,6 +29,7 @@ class JobNameMapping(str, enum.Enum):
         parse: typing.Callable[[], T_Result],
         transform: typing.Callable[[], T_Result],
         ingestion: typing.Callable[[], T_Result],
+        metadata_update: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is JobNameMapping.MANAGED_INGESTION:
             return managed_ingestion()
@@ -43,3 +45,5 @@ class JobNameMapping(str, enum.Enum):
             return transform()
         if self is JobNameMapping.INGESTION:
             return ingestion()
+        if self is JobNameMapping.METADATA_UPDATE:
+            return metadata_update()

llama_cloud/types/job_names.py

@@ -0,0 +1,89 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class JobNames(str, enum.Enum):
+    """
+    Enum for executable pipeline job names.
+    """
+
+    LOAD_DOCUMENTS_JOB = "load_documents_job"
+    LOAD_FILES_JOB = "load_files_job"
+    PLAYGROUND_JOB = "playground_job"
+    EVAL_DATASET_JOB = "eval_dataset_job"
+    PIPELINE_MANAGED_INGESTION_JOB = "pipeline_managed_ingestion_job"
+    DATA_SOURCE_MANAGED_INGESTION_JOB = "data_source_managed_ingestion_job"
+    DATA_SOURCE_UPDATE_DISPATCHER_JOB = "data_source_update_dispatcher_job"
+    PIPELINE_FILE_UPDATE_DISPATCHER_JOB = "pipeline_file_update_dispatcher_job"
+    PIPELINE_FILE_UPDATER_JOB = "pipeline_file_updater_job"
+    FILE_MANAGED_INGESTION_JOB = "file_managed_ingestion_job"
+    DOCUMENT_INGESTION_JOB = "document_ingestion_job"
+    PARSE_RAW_FILE_JOB = "parse_raw_file_job"
+    LLAMA_PARSE_TRANSFORM_JOB = "llama_parse_transform_job"
+    METADATA_UPDATE_JOB = "metadata_update_job"
+    PARSE_RAW_FILE_JOB_CACHED = "parse_raw_file_job_cached"
+    EXTRACTION_JOB = "extraction_job"
+    EXTRACT_JOB = "extract_job"
+    ASYNCIO_TEST_JOB = "asyncio_test_job"
+
+    def visit(
+        self,
+        load_documents_job: typing.Callable[[], T_Result],
+        load_files_job: typing.Callable[[], T_Result],
+        playground_job: typing.Callable[[], T_Result],
+        eval_dataset_job: typing.Callable[[], T_Result],
+        pipeline_managed_ingestion_job: typing.Callable[[], T_Result],
+        data_source_managed_ingestion_job: typing.Callable[[], T_Result],
+        data_source_update_dispatcher_job: typing.Callable[[], T_Result],
+        pipeline_file_update_dispatcher_job: typing.Callable[[], T_Result],
+        pipeline_file_updater_job: typing.Callable[[], T_Result],
+        file_managed_ingestion_job: typing.Callable[[], T_Result],
+        document_ingestion_job: typing.Callable[[], T_Result],
+        parse_raw_file_job: typing.Callable[[], T_Result],
+        llama_parse_transform_job: typing.Callable[[], T_Result],
+        metadata_update_job: typing.Callable[[], T_Result],
+        parse_raw_file_job_cached: typing.Callable[[], T_Result],
+        extraction_job: typing.Callable[[], T_Result],
+        extract_job: typing.Callable[[], T_Result],
+        asyncio_test_job: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is JobNames.LOAD_DOCUMENTS_JOB:
+            return load_documents_job()
+        if self is JobNames.LOAD_FILES_JOB:
+            return load_files_job()
+        if self is JobNames.PLAYGROUND_JOB:
+            return playground_job()
+        if self is JobNames.EVAL_DATASET_JOB:
+            return eval_dataset_job()
+        if self is JobNames.PIPELINE_MANAGED_INGESTION_JOB:
+            return pipeline_managed_ingestion_job()
+        if self is JobNames.DATA_SOURCE_MANAGED_INGESTION_JOB:
+            return data_source_managed_ingestion_job()
+        if self is JobNames.DATA_SOURCE_UPDATE_DISPATCHER_JOB:
+            return data_source_update_dispatcher_job()
+        if self is JobNames.PIPELINE_FILE_UPDATE_DISPATCHER_JOB:
+            return pipeline_file_update_dispatcher_job()
+        if self is JobNames.PIPELINE_FILE_UPDATER_JOB:
+            return pipeline_file_updater_job()
+        if self is JobNames.FILE_MANAGED_INGESTION_JOB:
+            return file_managed_ingestion_job()
+        if self is JobNames.DOCUMENT_INGESTION_JOB:
+            return document_ingestion_job()
+        if self is JobNames.PARSE_RAW_FILE_JOB:
+            return parse_raw_file_job()
+        if self is JobNames.LLAMA_PARSE_TRANSFORM_JOB:
+            return llama_parse_transform_job()
+        if self is JobNames.METADATA_UPDATE_JOB:
+            return metadata_update_job()
+        if self is JobNames.PARSE_RAW_FILE_JOB_CACHED:
+            return parse_raw_file_job_cached()
+        if self is JobNames.EXTRACTION_JOB:
+            return extraction_job()
+        if self is JobNames.EXTRACT_JOB:
+            return extract_job()
+        if self is JobNames.ASYNCIO_TEST_JOB:
+            return asyncio_test_job()

llama_cloud/types/job_record.py

@@ -0,0 +1,57 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .base import Base
+from .job_names import JobNames
+from .status_enum import StatusEnum
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class JobRecord(pydantic.BaseModel):
+    """
+    Schema for a job's metadata.
+    """
+
+    job_name: JobNames = pydantic.Field(description="The name of the job.")
+    partitions: typing.Dict[str, str] = pydantic.Field(
+        description="The partitions for this execution. Used for determining where to save job output."
+    )
+    parameters: typing.Optional[Base]
+    session_id: typing.Optional[str]
+    correlation_id: typing.Optional[str]
+    parent_job_execution_id: typing.Optional[str]
+    user_id: typing.Optional[str]
+    created_at: dt.datetime = pydantic.Field(description="Creation datetime")
+    project_id: typing.Optional[str]
+    id: typing.Optional[str] = pydantic.Field(description="Unique identifier")
+    status: StatusEnum
+    error_code: typing.Optional[str]
+    error_message: typing.Optional[str]
+    attempts: typing.Optional[int]
+    started_at: typing.Optional[dt.datetime]
+    ended_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
+    data: typing.Optional[Base]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/job_record_with_usage_metrics.py

@@ -0,0 +1,36 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .job_record import JobRecord
+from .usage_metric_response import UsageMetricResponse
+from .user_job_record import UserJobRecord
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class JobRecordWithUsageMetrics(pydantic.BaseModel):
+    job_record: JobRecord
+    usage_metrics: typing.Optional[UsageMetricResponse]
+    user: UserJobRecord
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}