from pydantic import BaseModel, Field


class QueryRequest(BaseModel):
    text: str = Field(..., description="Вопрос от лица пациента")
    top_k: int = Field(5, ge=1, le=20, description="Количество чанков для retrieval")
    document_ids: list[str] | None = Field(None, description="Ограничить поиск конкретными документами")
    temperature: float | None = Field(None, ge=0.0, le=2.0)
    max_tokens: int | None = Field(None, ge=100, le=8000)


class ChatRequest(BaseModel):
    text: str = Field(..., description="Реплика пациента")
    thread_id: int | None = Field(None, description="ID треда; если не передан — создаётся новый")
    top_k: int = Field(5, ge=1, le=20)
    temperature: float | None = Field(None, ge=0.0, le=2.0)
    max_tokens: int | None = Field(None, ge=100, le=8000)


class ThreadRenameRequest(BaseModel):
    name: str = Field(..., min_length=1, max_length=200)


class AgentConfigCreateRequest(BaseModel):
    intent_id: int = Field(..., description="ID ветки (intent), к которой относится конфиг")
    system_prompt: str = Field(..., min_length=1)
    rules_text: str = Field("", description="Правила в свободной markdown-форме")
    name: str | None = Field(None, max_length=200)
    activate: bool = Field(False, description="Сразу сделать новую версию активной в рамках ветки")


class IntentToggleRequest(BaseModel):
    is_enabled: bool


class IntentStepUpdateRequest(BaseModel):
    name: str | None = Field(None, max_length=200)
    system_prompt: str | None = None
    allowed_next: list[str] | None = None
    guards: dict | None = None