from __future__ import annotations

from evalkit.metrics import ContextualRelevancyMetric, FaithfulnessMetric
from evalkit.models import Suite
from evalkit.suites.base import EvaluationSuite


def DocumentSearchSuite() -> EvaluationSuite:
    """Document search/retrieval RAG evaluation suite.

    Evaluates context quality and answer faithfulness for document search/retrieval RAG.
    """
    return EvaluationSuite(
        name=Suite.DOCUMENT_SEARCH,
        description="Evaluates context quality and answer faithfulness for document search/retrieval RAG",
        metrics=[
            ContextualRelevancyMetric(),
            FaithfulnessMetric(),
        ],
    )
