gossip_semantic_search/models.py

from dataclasses import dataclass

from datetime import datetime
from numpy.typing import NDArray

@dataclass
class Article:
    author: str
    title: str
    link: str
    description: str
    published_date: datetime
    content: str
    embeded_content: NDArray = None
    questions: list[str] = None

@dataclass
class Answer:
    link: str
    content: str
    answer: str

@dataclass
class ProcessedDataset:
    y_true: NDArray[int]
    embedded_queries: NDArray[float]
    embedded_context: NDArray[float]