"""Document summary index."""

from llama_index.core.indices.document_summary.base import (
    DocumentSummaryIndex,
    GPTDocumentSummaryIndex,
)
from llama_index.core.indices.document_summary.retrievers import (
    DocumentSummaryIndexEmbeddingRetriever,
    DocumentSummaryIndexLLMRetriever,
    DocumentSummaryIndexRetriever,
)

__all__ = [
    "DocumentSummaryIndex",
    "DocumentSummaryIndexLLMRetriever",
    "DocumentSummaryIndexEmbeddingRetriever",
    # legacy
    "GPTDocumentSummaryIndex",
    "DocumentSummaryIndexRetriever",
]

"""Document summary index."""


from llama_index.core.indices.document_summary.base import (
    DocumentSummaryIndex,
    GPTDocumentSummaryIndex,
)
from llama_index.core.indices.document_summary.retrievers import (
    DocumentSummaryIndexEmbeddingRetriever,
    DocumentSummaryIndexLLMRetriever,
    DocumentSummaryIndexRetriever,
)

__all__ = [
    "DocumentSummaryIndex",
    "DocumentSummaryIndexLLMRetriever",
    "DocumentSummaryIndexEmbeddingRetriever",
    # legacy
    "GPTDocumentSummaryIndex",
    "DocumentSummaryIndexRetriever",
]
