api/core/rag/entities/context_entities.py

from typing import Optional

from pydantic import BaseModel


class DocumentContext(BaseModel):
    """
    Model class for document context.
    """

    content: str
    score: Optional[float] = None