from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
from google.protobuf import descriptor as _descriptor
from google.protobuf import message as _message
from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union

DESCRIPTOR: _descriptor.FileDescriptor

class ChunkType(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
    __slots__ = ()
    CHUNK_TYPE_UNKNOWN: _ClassVar[ChunkType]
    CHUNK_TYPE_SENTENCE: _ClassVar[ChunkType]
    CHUNK_TYPE_PARAGRAPH: _ClassVar[ChunkType]
    CHUNK_TYPE_PAGE: _ClassVar[ChunkType]
CHUNK_TYPE_UNKNOWN: ChunkType
CHUNK_TYPE_SENTENCE: ChunkType
CHUNK_TYPE_PARAGRAPH: ChunkType
CHUNK_TYPE_PAGE: ChunkType

class Document(_message.Message):
    __slots__ = ("file_path", "authors", "publish_date")
    FILE_PATH_FIELD_NUMBER: _ClassVar[int]
    AUTHORS_FIELD_NUMBER: _ClassVar[int]
    PUBLISH_DATE_FIELD_NUMBER: _ClassVar[int]
    file_path: str
    authors: str
    publish_date: str
    def __init__(self, file_path: _Optional[str] = ..., authors: _Optional[str] = ..., publish_date: _Optional[str] = ...) -> None: ...

class Chunk(_message.Message):
    __slots__ = ("chunk_id", "chunk_text", "chunk_type", "chunk_index", "parent_chunk_id", "document")
    CHUNK_ID_FIELD_NUMBER: _ClassVar[int]
    CHUNK_TEXT_FIELD_NUMBER: _ClassVar[int]
    CHUNK_TYPE_FIELD_NUMBER: _ClassVar[int]
    CHUNK_INDEX_FIELD_NUMBER: _ClassVar[int]
    PARENT_CHUNK_ID_FIELD_NUMBER: _ClassVar[int]
    DOCUMENT_FIELD_NUMBER: _ClassVar[int]
    chunk_id: str
    chunk_text: str
    chunk_type: ChunkType
    chunk_index: int
    parent_chunk_id: str
    document: Document
    def __init__(self, chunk_id: _Optional[str] = ..., chunk_text: _Optional[str] = ..., chunk_type: _Optional[_Union[ChunkType, str]] = ..., chunk_index: _Optional[int] = ..., parent_chunk_id: _Optional[str] = ..., document: _Optional[_Union[Document, _Mapping]] = ...) -> None: ...