from core_db import AuditMixin, Base, EntityMixin
from core_shared import JSONValue
from sqlalchemy import String, Text
from sqlalchemy import JSON
from sqlalchemy.orm import Mapped, mapped_column


class AgentConfig(EntityMixin, AuditMixin, Base):
    __tablename__ = "agent_config"

    agent_id: Mapped[str] = mapped_column(String(36), index=True)
    role: Mapped[str] = mapped_column(String(64), default="assistant")
    goal: Mapped[str | None] = mapped_column(Text, nullable=True)
    system_prompt: Mapped[str] = mapped_column(Text)
    model_config_json: Mapped[dict[str, JSONValue]] = mapped_column(JSON, default=dict)
    memory_policy_json: Mapped[dict[str, JSONValue]] = mapped_column(JSON, default=dict)
    tool_refs_json: Mapped[list[dict[str, JSONValue]]] = mapped_column(JSON, default=list)
    skill_refs_json: Mapped[list[dict[str, JSONValue]]] = mapped_column(JSON, default=list)