"""Experimental txtai/litellm RAG interface built on the OCR pipeline.

This package is a prototype. The supported interface is the `ocr-pipeline` CLI
(src/ocr_pipeline). Extra dependencies are required: `uv sync --extra kg`.
"""

try:
    from .ocr import get_screenshots, extract_text
    from .embeddings import create_and_index
    from .rag import retrieve, ask_wllm
except ImportError as exc:
    raise ImportError(
        "kg_ocr is experimental and needs extra dependencies: "
        "uv sync --extra kg  (or: uv pip install txtai litellm python-dotenv)"
    ) from exc

__all__ = ["get_screenshots", "extract_text", "create_and_index", "retrieve", "ask_wllm"]
