from __future__ import annotations

from dataclasses import dataclass, field, asdict
from typing import Any


@dataclass
class Message:
    message_id: str
    role: str
    content: str
    created_at: str | None = None
    metadata: dict[str, Any] = field(default_factory=dict)


@dataclass
class Conversation:
    source: str
    conversation_id: str
    title: str
    created_at: str | None = None
    updated_at: str | None = None
    raw_path: str | None = None
    messages: list[Message] = field(default_factory=list)
    tags: list[str] = field(default_factory=list)
    metadata: dict[str, Any] = field(default_factory=dict)

    def to_dict(self) -> dict[str, Any]:
        data = asdict(self)
        data["messages"] = [asdict(message) for message in self.messages]
        return data


@dataclass
class Chunk:
    chunk_id: str
    conversation_id: str
    source: str
    title: str
    message_start: int
    message_end: int
    text: str
    created_at: str | None = None
    updated_at: str | None = None
    metadata: dict[str, Any] = field(default_factory=dict)