src/models/rag.py

# src/models/rag.py
from dataclasses import dataclass
from typing import List, Optional, Dict

@dataclass
class RAGResponse:
    """Response model for RAG-based generation"""
    response: str
    context_docs: Optional[List[str]] = None
    sources: Optional[List[Dict]] = None
    scores: Optional[List[float]] = None