models.py

from dataclasses import dataclass
from pathlib import Path
from typing import Any
import numpy as np

@dataclass(slots=True)
class Document:
    title: str
    category: str
    topic: str
    content: str
    metadata: dict[str, Any]
    source_path: Path

@dataclass(slots=True)
class Chunk:
    id: str
    text: str
    metadata: dict[str, Any]

@dataclass(slots=True)
class SearchResult:
    chunk: Chunk
    score: float

@dataclass(slots=True)
class EmbeddedChunk:
    chunk: Chunk
    embedding: np.ndarray

@dataclass(slots=True)
class Context:
    text: str
    sources: list[str]
    chunk_count: int

@dataclass(slots=True)
class Prompt:
    system: str
    context: str
    history: str
    user: str

    def build(self) -> str:
        return (
            f"{self.system}\n\n"
            f"{self.context}\n\n"
            f"{self.history}\n\n"
            f"User:\n{self.user}"
        )

@dataclass(slots=True)
class SearchRequest:
    query: str
    k: int = 5
    filters: dict[str, Any] | None = None

@dataclass(slots=True)
class Intent:
    categories: list[str]

@dataclass(slots=True)
class SearchResponse:
    context: Context
    results: list[SearchResult]