from typing import Any

from pydantic import BaseModel, Field

from app.modules.ai_memory.constants import AIMemoryScope


class AIMemoryRecallRequest(BaseModel):
    query: str = Field(..., min_length=1)
    scope: str = AIMemoryScope.GLOBAL
    subject: str | None = None
    limit: int = Field(default=5, ge=1, le=50)


class AIMemoryRead(BaseModel):
    code: str
    scope: str
    subject: str
    content: str
    summary: str | None
    tags: list[Any] | None
    source: str
    importance: int
    status: str
    actor: str
    last_used_at: str | None
    expires_at: str | None
    created_at: str
    updated_at: str