context_entities.py 202 B

123456789101112
  1. from typing import Optional
  2. from pydantic import BaseModel
  3. class DocumentContext(BaseModel):
  4. """
  5. Model class for document context.
  6. """
  7. content: str
  8. score: Optional[float] = None