from __future__ import annotations

from typing import Any

from ketju.rag.llama_index.contracts import LlamaIndexIngestionHandler
from ketju.rag.llama_index.ingestion.simple_docling import SimpleDoclingIngestionHandler
from ketju.rag.llama_index.variants.simple import LlamaIndexRag


class SimpleDoclingRag(LlamaIndexRag):
    def __init__(
        self,
        ingestion_handler: LlamaIndexIngestionHandler | None = None,
        **kwargs: Any,
    ) -> None:
        super().__init__(**kwargs)
        self.ingestion_handler = ingestion_handler or SimpleDoclingIngestionHandler()
