diff --git a/cohere/compass/models/documents.py b/cohere/compass/models/documents.py index 585e589..daf1933 100644 --- a/cohere/compass/models/documents.py +++ b/cohere/compass/models/documents.py @@ -22,20 +22,7 @@ class CompassDocumentMetadata(ValidatedModel): parent_document_id: str = "" -class AssetType(str, Enum): - def __str__(self) -> str: - return self.value - - # A page that has been rendered as an image - PAGE_IMAGE = "page_image" - # A Markdown representation of a page's content - PAGE_MARKDOWN = "page_markdown" - # A dump of the text extracted from a document - DOCUMENT_TEXT = "document_text" - - class CompassDocumentChunkAsset(BaseModel): - asset_type: AssetType content_type: str asset_data: str diff --git a/cohere/compass/models/search.py b/cohere/compass/models/search.py index 8fe847a..9cbd8b2 100644 --- a/cohere/compass/models/search.py +++ b/cohere/compass/models/search.py @@ -5,11 +5,8 @@ # 3rd party imports from pydantic import BaseModel -from cohere.compass.models.documents import AssetType - class AssetInfo(BaseModel): - asset_type: AssetType content_type: str presigned_url: str