llama-cloud 0.1.35__py3-none-any.whl → 0.1.36__py3-none-any.whl

llama_cloud/resources/llama_extract/client.py

@@ -45,10 +45,16 @@ class LlamaExtractClient:
         self._client_wrapper = client_wrapper
 
     def list_extraction_agents(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        include_default: typing.Optional[bool] = None,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
     ) -> typing.List[ExtractAgent]:
         """
         Parameters:
+            - include_default: typing.Optional[bool]. Whether to include default agents in the results
+
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
@@ -63,7 +69,9 @@ class LlamaExtractClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/extraction-agents"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            params=remove_none_from_dict(
+                {"include_default": include_default, "project_id": project_id, "organization_id": organization_id}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -265,6 +273,44 @@ class LlamaExtractClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def get_or_create_default_extraction_agent(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> ExtractAgent:
+        """
+        Get or create a default extraction agent for the current project.
+        The default agent has an empty schema and default configuration.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.llama_extract.get_or_create_default_extraction_agent()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/extraction-agents/default"
+            ),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_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:
@@ -992,10 +1038,16 @@ class AsyncLlamaExtractClient:
         self._client_wrapper = client_wrapper
 
     async def list_extraction_agents(
-        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+        self,
+        *,
+        include_default: typing.Optional[bool] = None,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
     ) -> typing.List[ExtractAgent]:
         """
         Parameters:
+            - include_default: typing.Optional[bool]. Whether to include default agents in the results
+
             - project_id: typing.Optional[str].
 
             - organization_id: typing.Optional[str].
@@ -1010,7 +1062,9 @@ class AsyncLlamaExtractClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/extraction-agents"),
-            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            params=remove_none_from_dict(
+                {"include_default": include_default, "project_id": project_id, "organization_id": organization_id}
+            ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1212,6 +1266,44 @@ 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_or_create_default_extraction_agent(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> ExtractAgent:
+        """
+        Get or create a default extraction agent for the current project.
+        The default agent has an empty schema and default configuration.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.llama_extract.get_or_create_default_extraction_agent()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", "api/v1/extraction/extraction-agents/default"
+            ),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_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:

llama_cloud/types/__init__.py

@@ -41,7 +41,12 @@ from .chat_app_response import ChatAppResponse
 from .chat_data import ChatData
 from .chunk_mode import ChunkMode
 from .classification_result import ClassificationResult
-from .classify_response import ClassifyResponse
+from .classifier_rule import ClassifierRule
+from .classify_job import ClassifyJob
+from .classify_job_results import ClassifyJobResults
+from .classify_job_with_status import ClassifyJobWithStatus
+from .classify_parsing_configuration import ClassifyParsingConfiguration
+from .cloud_astra_db_vector_store import CloudAstraDbVectorStore
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
 from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from .cloud_box_data_source import CloudBoxDataSource
@@ -139,13 +144,17 @@ from .extract_schema_validate_response_data_schema_value import ExtractSchemaVal
 from .extract_state import ExtractState
 from .extract_target import ExtractTarget
 from .fail_page_mode import FailPageMode
+from .failure_handling_config import FailureHandlingConfig
 from .file import File
+from .file_classification import FileClassification
 from .file_count_by_status_response import FileCountByStatusResponse
 from .file_data import FileData
 from .file_id_presigned_url import FileIdPresignedUrl
 from .file_parse_public import FileParsePublic
 from .file_permission_info_value import FilePermissionInfoValue
 from .file_resource_info_value import FileResourceInfoValue
+from .file_store_info_response import FileStoreInfoResponse
+from .file_store_info_response_status import FileStoreInfoResponseStatus
 from .filter_condition import FilterCondition
 from .filter_operation import FilterOperation
 from .filter_operation_eq import FilterOperationEq
@@ -418,7 +427,12 @@ __all__ = [
     "ChatData",
     "ChunkMode",
     "ClassificationResult",
-    "ClassifyResponse",
+    "ClassifierRule",
+    "ClassifyJob",
+    "ClassifyJobResults",
+    "ClassifyJobWithStatus",
+    "ClassifyParsingConfiguration",
+    "CloudAstraDbVectorStore",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
     "CloudBoxDataSource",
@@ -512,13 +526,17 @@ __all__ = [
     "ExtractState",
     "ExtractTarget",
     "FailPageMode",
+    "FailureHandlingConfig",
     "File",
+    "FileClassification",
     "FileCountByStatusResponse",
     "FileData",
     "FileIdPresignedUrl",
     "FileParsePublic",
     "FilePermissionInfoValue",
     "FileResourceInfoValue",
+    "FileStoreInfoResponse",
+    "FileStoreInfoResponseStatus",
     "FilterCondition",
     "FilterOperation",
     "FilterOperationEq",

llama_cloud/types/classification_result.py

@@ -17,14 +17,13 @@ except ImportError:
 class ClassificationResult(pydantic.BaseModel):
     """
     Result of classifying a single file.
-
-    Contains the classification outcome with confidence score and matched rule info.
     """
 
-    file_id: str = pydantic.Field(description="The ID of the classified file")
-    type: str = pydantic.Field(description="The assigned document type ('unknown' if no rules matched)")
+    reasoning: str = pydantic.Field(
+        description="Step-by-step explanation of why this classification was chosen and the confidence score assigned"
+    )
     confidence: float = pydantic.Field(description="Confidence score of the classification (0.0-1.0)")
-    matched_rule: typing.Optional[str]
+    type: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/classifier_rule.py

@@ -0,0 +1,43 @@
+# 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 ClassifierRule(pydantic.BaseModel):
+    """
+    A rule for classifying documents - v0 simplified version.
+
+    This represents a single classification rule that will be applied to documents.
+    All rules are content-based and use natural language descriptions.
+    """
+
+    type: str = pydantic.Field(
+        description="The document type to assign when this rule matches (e.g., 'invoice', 'receipt', 'contract')"
+    )
+    description: str = pydantic.Field(
+        description="Natural language description of what to classify. Be specific about the content characteristics that identify this document type."
+    )
+
+    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/classify_job.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 .classifier_rule import ClassifierRule
+from .classify_parsing_configuration import ClassifyParsingConfiguration
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ClassifyJob(pydantic.BaseModel):
+    """
+    A classify job.
+    """
+
+    id: str = pydantic.Field(description="Unique identifier")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
+    rules: typing.List[ClassifierRule] = pydantic.Field(description="The rules to classify the files")
+    user_id: str = pydantic.Field(description="The ID of the user")
+    project_id: str = pydantic.Field(description="The ID of the project")
+    parsing_configuration: typing.Optional[ClassifyParsingConfiguration] = pydantic.Field(
+        description="The configuration for the parsing job"
+    )
+
+    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/{classify_response.py → classify_job_results.py}

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .classification_result import ClassificationResult
+from .file_classification import FileClassification
 
 try:
     import pydantic
@@ -15,17 +15,14 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class ClassifyResponse(pydantic.BaseModel):
+class ClassifyJobResults(pydantic.BaseModel):
     """
     Response model for the classify endpoint following AIP-132 pagination standard.
-
-    Contains classification results with pagination support and summary statistics.
     """
 
-    items: typing.List[ClassificationResult] = pydantic.Field(description="The list of items.")
+    items: typing.List[FileClassification] = pydantic.Field(description="The list of items.")
     next_page_token: typing.Optional[str]
     total_size: typing.Optional[int]
-    unknown_count: int = pydantic.Field(description="Number of files that couldn't be classified")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/classify_job_with_status.py

@@ -0,0 +1,47 @@
+# 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 .classifier_rule import ClassifierRule
+from .classify_parsing_configuration import ClassifyParsingConfiguration
+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 ClassifyJobWithStatus(pydantic.BaseModel):
+    """
+    A classify job with status.
+    """
+
+    id: str = pydantic.Field(description="Unique identifier")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
+    rules: typing.List[ClassifierRule] = pydantic.Field(description="The rules to classify the files")
+    user_id: str = pydantic.Field(description="The ID of the user")
+    project_id: str = pydantic.Field(description="The ID of the project")
+    parsing_configuration: typing.Optional[ClassifyParsingConfiguration] = pydantic.Field(
+        description="The configuration for the parsing job"
+    )
+    status: StatusEnum = pydantic.Field(description="The status of the classify job")
+
+    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/classify_parsing_configuration.py

@@ -0,0 +1,38 @@
+# 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 .parser_languages import ParserLanguages
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ClassifyParsingConfiguration(pydantic.BaseModel):
+    """
+    Parsing configuration for a classify job.
+    """
+
+    lang: typing.Optional[ParserLanguages] = pydantic.Field(description="The language to parse the files in")
+    max_pages: typing.Optional[int]
+    target_pages: typing.Optional[typing.List[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/cloud_astra_db_vector_store.py

@@ -0,0 +1,51 @@
+# 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 CloudAstraDbVectorStore(pydantic.BaseModel):
+    """
+    Cloud AstraDB Vector Store.
+
+    This class is used to store the configuration for an AstraDB vector store, so that it can be
+    created and used in LlamaCloud.
+
+    Args:
+    token (str): The Astra DB Application Token to use.
+    api_endpoint (str): The Astra DB JSON API endpoint for your database.
+    collection_name (str): Collection name to use. If not existing, it will be created.
+    embedding_dimension (int): Length of the embedding vectors in use.
+    keyspace (optional[str]): The keyspace to use. If not provided, 'default_keyspace'
+    """
+
+    supports_nested_metadata_filters: typing.Optional[bool]
+    token: str = pydantic.Field(description="The Astra DB Application Token to use")
+    api_endpoint: str = pydantic.Field(description="The Astra DB JSON API endpoint for your database")
+    collection_name: str = pydantic.Field(description="Collection name to use. If not existing, it will be created")
+    embedding_dimension: int = pydantic.Field(description="Length of the embedding vectors in use")
+    keyspace: typing.Optional[str]
+    class_name: 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/cloud_confluence_data_source.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .failure_handling_config import FailureHandlingConfig
 
 try:
     import pydantic
@@ -28,6 +29,20 @@ class CloudConfluenceDataSource(pydantic.BaseModel):
     label: typing.Optional[str]
     index_restricted_pages: typing.Optional[bool] = pydantic.Field(description="Whether to index restricted pages.")
     keep_markdown_format: typing.Optional[bool] = pydantic.Field(description="Whether to keep the markdown format.")
+    failure_handling: typing.Optional[FailureHandlingConfig] = pydantic.Field(
+        description=(
+            "Configuration for handling failures during processing. Key-value object controlling failure handling behaviors.\n"
+            "\n"
+            "Example:\n"
+            "{\n"
+            '"skip_list_failures": true\n'
+            "}\n"
+            "\n"
+            "Currently supports:\n"
+            "\n"
+            "- skip_list_failures: Skip failed batches/lists and continue processing\n"
+        )
+    )
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/configurable_data_sink_names.py

@@ -13,6 +13,7 @@ class ConfigurableDataSinkNames(str, enum.Enum):
     AZUREAI_SEARCH = "AZUREAI_SEARCH"
     MONGODB_ATLAS = "MONGODB_ATLAS"
     MILVUS = "MILVUS"
+    ASTRA_DB = "ASTRA_DB"
 
     def visit(
         self,
@@ -22,6 +23,7 @@ class ConfigurableDataSinkNames(str, enum.Enum):
         azureai_search: typing.Callable[[], T_Result],
         mongodb_atlas: typing.Callable[[], T_Result],
         milvus: typing.Callable[[], T_Result],
+        astra_db: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is ConfigurableDataSinkNames.PINECONE:
             return pinecone()
@@ -35,3 +37,5 @@ class ConfigurableDataSinkNames(str, enum.Enum):
             return mongodb_atlas()
         if self is ConfigurableDataSinkNames.MILVUS:
             return milvus()
+        if self is ConfigurableDataSinkNames.ASTRA_DB:
+            return astra_db()

llama_cloud/types/data_sink_component.py

@@ -2,6 +2,7 @@
 
 import typing
 
+from .cloud_astra_db_vector_store import CloudAstraDbVectorStore
 from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from .cloud_milvus_vector_store import CloudMilvusVectorStore
 from .cloud_mongo_db_atlas_vector_search import CloudMongoDbAtlasVectorSearch
@@ -17,4 +18,5 @@ DataSinkComponent = typing.Union[
     CloudAzureAiSearchVectorStore,
     CloudMongoDbAtlasVectorSearch,
     CloudMilvusVectorStore,
+    CloudAstraDbVectorStore,
 ]

llama_cloud/types/data_sink_create_component.py

@@ -2,6 +2,7 @@
 
 import typing
 
+from .cloud_astra_db_vector_store import CloudAstraDbVectorStore
 from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from .cloud_milvus_vector_store import CloudMilvusVectorStore
 from .cloud_mongo_db_atlas_vector_search import CloudMongoDbAtlasVectorSearch
@@ -17,4 +18,5 @@ DataSinkCreateComponent = typing.Union[
     CloudAzureAiSearchVectorStore,
     CloudMongoDbAtlasVectorSearch,
     CloudMilvusVectorStore,
+    CloudAstraDbVectorStore,
 ]

llama_cloud/types/failure_handling_config.py

@@ -0,0 +1,37 @@
+# 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 FailureHandlingConfig(pydantic.BaseModel):
+    """
+    Configuration for handling different types of failures during data source processing.
+    """
+
+    skip_list_failures: typing.Optional[bool] = pydantic.Field(
+        description="Whether to skip failed batches/lists and continue processing"
+    )
+
+    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/file_classification.py

@@ -0,0 +1,41 @@
+# 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 .classification_result import ClassificationResult
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FileClassification(pydantic.BaseModel):
+    """
+    A file classification.
+    """
+
+    id: str = pydantic.Field(description="Unique identifier")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
+    classify_job_id: str = pydantic.Field(description="The ID of the classify job")
+    file_id: str = pydantic.Field(description="The ID of the classified file")
+    result: typing.Optional[ClassificationResult]
+
+    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}