"""Routes de lecture et mise à jour de la configuration de l'agent."""
from __future__ import annotations

from fastapi import APIRouter
from pydantic import BaseModel

from .. import agent_config

router = APIRouter(prefix="/api/config", tags=["config"])


class ConfigPatch(BaseModel):
    system_prompt: str | None = None
    temperature: float | None = None
    top_p: float | None = None
    top_k: int | None = None
    max_tokens: int | None = None
    num_ctx: int | None = None
    num_gpu: int | None = None
    num_batch: int | None = None
    tools: dict[str, bool] | None = None
    confirm_shell: bool | None = None
    think: bool | None = None
    code_model: str | None = None
    plan_mode: bool | None = None
    self_review: bool | None = None
    rag_enabled: bool | None = None
    embed_model: str | None = None


@router.get("")
async def get_config(model: str | None = None) -> dict:
    return {
        "config": agent_config.get_config(model),
        "available_tools": agent_config.AVAILABLE_TOOLS,
    }


@router.put("")
async def put_config(patch: ConfigPatch, model: str | None = None) -> dict:
    cfg = agent_config.save_config(patch.model_dump(exclude_none=True), model)
    return {"config": cfg}