llama-cloud 0.1.41__py3-none-any.whl → 1.0.0b4__py3-none-any.whl

llama_cloud/types/data_source_update_dispatcher_config.py

@@ -1,38 +0,0 @@
-# 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 .delete_params import DeleteParams
-
-try:
-    import pydantic
-    if pydantic.__version__.startswith("1."):
-        raise ImportError
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class DataSourceUpdateDispatcherConfig(pydantic.BaseModel):
-    """
-    Schema for the parameters of a data source dispatcher job.
-    """
-
-    should_delete: typing.Optional[bool]
-    custom_metadata: typing.Optional[typing.Dict[str, typing.Any]]
-    delete_info: typing.Optional[DeleteParams]
-
-    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/delete_params.py

@@ -1,39 +0,0 @@
-# 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 DeleteParams(pydantic.BaseModel):
-    """
-    Schema for the parameters of a delete job.
-    """
-
-    document_ids_to_delete: typing.Optional[typing.List[str]]
-    files_ids_to_delete: typing.Optional[typing.List[str]]
-    data_sources_ids_to_delete: typing.Optional[typing.List[str]]
-    embed_collection_name: typing.Optional[str]
-    data_sink_id: 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/document_chunk_mode.py

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

llama_cloud/types/document_ingestion_job_params.py

@@ -1,43 +0,0 @@
-# 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 .delete_params import DeleteParams
-
-try:
-    import pydantic
-    if pydantic.__version__.startswith("1."):
-        raise ImportError
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class DocumentIngestionJobParams(pydantic.BaseModel):
-    """
-    Schema for the parameters of a document ingestion job.
-    """
-
-    custom_metadata: typing.Optional[typing.Dict[str, typing.Any]]
-    resource_info: typing.Optional[typing.Dict[str, typing.Any]]
-    should_delete: typing.Optional[bool]
-    document_ids: typing.Optional[typing.List[str]]
-    pipeline_file_id: typing.Optional[str]
-    delete_info: typing.Optional[DeleteParams]
-    is_new_file: typing.Optional[bool] = pydantic.Field(description="Whether the file is new")
-    page_count: typing.Optional[int]
-
-    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/element_segmentation_config.py

@@ -1,29 +0,0 @@
-# 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 ElementSegmentationConfig(pydantic.BaseModel):
-    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/embedding_model_config.py

@@ -1,43 +0,0 @@
-# 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 .embedding_model_config_embedding_config import EmbeddingModelConfigEmbeddingConfig
-
-try:
-    import pydantic
-    if pydantic.__version__.startswith("1."):
-        raise ImportError
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class EmbeddingModelConfig(pydantic.BaseModel):
-    """
-    Schema for an embedding model config.
-    """
-
-    id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
-    name: str = pydantic.Field(description="The name of the embedding model config.")
-    embedding_config: EmbeddingModelConfigEmbeddingConfig = pydantic.Field(
-        description="The embedding configuration for the embedding model config."
-    )
-    project_id: 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/embedding_model_config_embedding_config.py

@@ -1,89 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import typing
-
-import typing_extensions
-
-from .azure_open_ai_embedding_config import AzureOpenAiEmbeddingConfig
-from .bedrock_embedding_config import BedrockEmbeddingConfig
-from .cohere_embedding_config import CohereEmbeddingConfig
-from .gemini_embedding_config import GeminiEmbeddingConfig
-from .hugging_face_inference_api_embedding_config import HuggingFaceInferenceApiEmbeddingConfig
-from .open_ai_embedding_config import OpenAiEmbeddingConfig
-from .vertex_ai_embedding_config import VertexAiEmbeddingConfig
-
-
-class EmbeddingModelConfigEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["AZURE_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
-    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
-    type: typing_extensions.Literal["COHERE_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
-    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
-    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding(VertexAiEmbeddingConfig):
-    type: typing_extensions.Literal["VERTEXAI_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-EmbeddingModelConfigEmbeddingConfig = typing.Union[
-    EmbeddingModelConfigEmbeddingConfig_AzureEmbedding,
-    EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding,
-    EmbeddingModelConfigEmbeddingConfig_CohereEmbedding,
-    EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding,
-    EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding,
-    EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding,
-    EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding,
-]

llama_cloud/types/embedding_model_config_update.py

@@ -1,33 +0,0 @@
-# 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 .embedding_model_config_update_embedding_config import EmbeddingModelConfigUpdateEmbeddingConfig
-
-try:
-    import pydantic
-    if pydantic.__version__.startswith("1."):
-        raise ImportError
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class EmbeddingModelConfigUpdate(pydantic.BaseModel):
-    name: typing.Optional[str]
-    embedding_config: typing.Optional[EmbeddingModelConfigUpdateEmbeddingConfig]
-
-    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/embedding_model_config_update_embedding_config.py

@@ -1,89 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import typing
-
-import typing_extensions
-
-from .azure_open_ai_embedding_config import AzureOpenAiEmbeddingConfig
-from .bedrock_embedding_config import BedrockEmbeddingConfig
-from .cohere_embedding_config import CohereEmbeddingConfig
-from .gemini_embedding_config import GeminiEmbeddingConfig
-from .hugging_face_inference_api_embedding_config import HuggingFaceInferenceApiEmbeddingConfig
-from .open_ai_embedding_config import OpenAiEmbeddingConfig
-from .vertex_ai_embedding_config import VertexAiEmbeddingConfig
-
-
-class EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["AZURE_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
-    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
-    type: typing_extensions.Literal["COHERE_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
-    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
-    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
-    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-class EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding(VertexAiEmbeddingConfig):
-    type: typing_extensions.Literal["VERTEXAI_EMBEDDING"]
-
-    class Config:
-        frozen = True
-        smart_union = True
-        allow_population_by_field_name = True
-
-
-EmbeddingModelConfigUpdateEmbeddingConfig = typing.Union[
-    EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding,
-    EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding,
-    EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding,
-    EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding,
-    EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
-    EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding,
-    EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding,
-]

llama_cloud/types/eval_execution_params.py

@@ -1,41 +0,0 @@
-# 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 .supported_llm_model_names import SupportedLlmModelNames
-
-try:
-    import pydantic
-    if pydantic.__version__.startswith("1."):
-        raise ImportError
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class EvalExecutionParams(pydantic.BaseModel):
-    """
-    Schema for the params for an eval execution.
-    """
-
-    llm_model: typing.Optional[SupportedLlmModelNames] = pydantic.Field(
-        description="The LLM model to use within eval execution."
-    )
-    qa_prompt_tmpl: typing.Optional[str] = pydantic.Field(
-        description="The template to use for the question answering prompt."
-    )
-
-    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.py

@@ -1,48 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-import typing_extensions
-
-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.")
-    custom_configuration: typing.Optional[typing_extensions.Literal["default"]]
-    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

@@ -1,5 +0,0 @@
-# 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

@@ -1,66 +0,0 @@
-# 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 .document_chunk_mode import DocumentChunkMode
-from .extract_config_priority import ExtractConfigPriority
-from .extract_mode import ExtractMode
-from .extract_models import ExtractModels
-from .extract_target import ExtractTarget
-from .public_model_name import PublicModelName
-
-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.
-    """
-
-    priority: typing.Optional[ExtractConfigPriority]
-    extraction_target: typing.Optional[ExtractTarget] = pydantic.Field(description="The extraction target specified.")
-    extraction_mode: typing.Optional[ExtractMode] = pydantic.Field(
-        description="The extraction mode specified (FAST, BALANCED, MULTIMODAL, PREMIUM)."
-    )
-    parse_model: typing.Optional[PublicModelName]
-    extract_model: typing.Optional[ExtractModels]
-    multimodal_fast_mode: typing.Optional[bool] = pydantic.Field(
-        description="DEPRECATED: Whether to use fast mode for multimodal extraction."
-    )
-    system_prompt: typing.Optional[str]
-    use_reasoning: typing.Optional[bool] = pydantic.Field(description="Whether to use reasoning for the extraction.")
-    cite_sources: typing.Optional[bool] = pydantic.Field(description="Whether to cite sources for the extraction.")
-    confidence_scores: typing.Optional[bool] = pydantic.Field(
-        description="Whether to fetch confidence scores for the extraction."
-    )
-    chunk_mode: typing.Optional[DocumentChunkMode] = pydantic.Field(
-        description="The mode to use for chunking the document."
-    )
-    high_resolution_mode: typing.Optional[bool] = pydantic.Field(
-        description="Whether to use high resolution mode for the extraction."
-    )
-    invalidate_cache: typing.Optional[bool] = pydantic.Field(
-        description="Whether to invalidate the cache for the extraction."
-    )
-    page_range: 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_config_priority.py

@@ -1,29 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class ExtractConfigPriority(str, enum.Enum):
-    LOW = "low"
-    MEDIUM = "medium"
-    HIGH = "high"
-    CRITICAL = "critical"
-
-    def visit(
-        self,
-        low: typing.Callable[[], T_Result],
-        medium: typing.Callable[[], T_Result],
-        high: typing.Callable[[], T_Result],
-        critical: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is ExtractConfigPriority.LOW:
-            return low()
-        if self is ExtractConfigPriority.MEDIUM:
-            return medium()
-        if self is ExtractConfigPriority.HIGH:
-            return high()
-        if self is ExtractConfigPriority.CRITICAL:
-            return critical()

llama_cloud/types/extract_job.py

@@ -1,38 +0,0 @@
-# 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 import ExtractAgent
-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")
-    extraction_agent: ExtractAgent = pydantic.Field(description="The agent that the job was run on.")
-    status: StatusEnum = pydantic.Field(description="The status of the extraction job")
-    error: typing.Optional[str]
-    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}