"""
Study-Buddy RAG package with integrated Corrective RAG (CRAG).
"""

import os
os.environ.setdefault("USE_TORCH", "1")
os.environ.setdefault("USE_TF", "0")
os.environ.setdefault("TRANSFORMERS_NO_TF", "1")

from src.rag.config import Settings, settings
from src.rag.constants import (
    DEFAULT_MAX_RETRIES,
    DEFAULT_HIGH_THRESHOLD,
    DEFAULT_LOW_THRESHOLD,
    DEFAULT_KEEP_THRESHOLD,
)
from src.rag.schemas import (
    Chunk,
    RouterOutput,
    EvaluatorOutput,
    GraphState,
    FinalAnswer,
    Source,
    Verdict,
    Cause,
    TutorQuery,
    TutorResponse,
)
from src.rag.reranker import Reranker, JinaReranker, MockReranker, get_reranker, rerank
from src.rag.crag_llm import LLM, LangChainLLM, MockLLM
from src.rag.crag_graph import build_graph, graph
from src.rag.crag import run_crag

__all__ = [
    "Settings",
    "settings",
    "DEFAULT_MAX_RETRIES",
    "DEFAULT_HIGH_THRESHOLD",
    "DEFAULT_LOW_THRESHOLD",
    "DEFAULT_KEEP_THRESHOLD",
    "Chunk",
    "RouterOutput",
    "EvaluatorOutput",
    "GraphState",
    "FinalAnswer",
    "Source",
    "Verdict",
    "Cause",
    "TutorQuery",
    "TutorResponse",
    "Reranker",
    "JinaReranker",
    "MockReranker",
    "get_reranker",
    "rerank",
    "LLM",
    "LangChainLLM",
    "MockLLM",
    "build_graph",
    "graph",
    "run_crag",
]