models.py

"""Pydantic request and response models for the settings/onboarding endpoints.

Lifted verbatim from the original `src/api/settings.py` (lines 49–223). No
shape changes — every external caller sees the same fields and validators
they did before. See `src/api/settings/__init__.py` for re-exports.
"""

from typing import Any

from pydantic import BaseModel, Field

from services.docling_service import DoclingConfig


class SettingsUpdateBody(BaseModel):
    llm_model: str | None = Field(None, min_length=1)
    llm_provider: str | None = Field(None, pattern="^(openai|anthropic|watsonx|ollama)$")
    system_prompt: str | None = None
    chunk_size: int | None = Field(None, gt=0)
    chunk_overlap: int | None = Field(None, ge=0)
    table_structure: bool | None = None
    ocr: bool | None = None
    picture_descriptions: bool | None = None
    disable_ingest_with_langflow: bool | None = None
    embedding_model: str | None = Field(None, min_length=1)
    embedding_provider: str | None = Field(None, pattern="^(openai|watsonx|ollama)$")
    index_name: str | None = Field(None, min_length=1)
    openai_api_key: str | None = Field(None, min_length=1)
    anthropic_api_key: str | None = Field(None, min_length=1)
    watsonx_api_key: str | None = Field(None, min_length=1)
    watsonx_endpoint: str | None = Field(None, min_length=1)
    watsonx_project_id: str | None = Field(None, min_length=1)
    ollama_endpoint: str | None = Field(None, min_length=1)
    remove_ollama_config: bool | None = None
    remove_openai_config: bool | None = None
    remove_anthropic_config: bool | None = None
    remove_watsonx_config: bool | None = None
    # Explicit confirmation that the caller accepts removing a provider whose
    # embedding models are still in use by indexed documents. Without this,
    # the backend returns 409 and the frontend prompts the user.
    force_remove: bool | None = False


class OnboardingBody(BaseModel):
    llm_provider: str | None = Field(None, pattern="^(openai|anthropic|watsonx|ollama)$")
    llm_model: str | None = Field(None, min_length=1)
    embedding_provider: str | None = Field(None, pattern="^(openai|watsonx|ollama)$")
    embedding_model: str | None = Field(None, min_length=1)
    openai_api_key: str | None = Field(None, min_length=1)
    anthropic_api_key: str | None = Field(None, min_length=1)
    watsonx_api_key: str | None = Field(None, min_length=1)
    watsonx_endpoint: str | None = Field(None, min_length=1)
    watsonx_project_id: str | None = Field(None, min_length=1)
    ollama_endpoint: str | None = Field(None, min_length=1)


class AssistantMessage(BaseModel):
    role: str
    content: str
    timestamp: str


class OnboardingStateBody(BaseModel):
    current_step: int | None = None
    assistant_message: AssistantMessage | None = None
    selected_nudge: str | None = None
    card_steps: dict[str, Any] | None = None
    upload_steps: dict[str, Any] | None = None
    openrag_docs_filter_id: str | None = None
    user_doc_filter_id: str | None = None
    openrag_docs_ingested_version: str | None = None
    openrag_docs_remote_signature: str | None = None


class DoclingPresetBody(BaseModel):
    preset: str | None = None
    table_structure: bool | None = None
    ocr: bool | None = None
    picture_descriptions: bool | None = None


class OnboardingStateConfig(BaseModel):
    current_step: int | None
    assistant_message: AssistantMessage | None
    selected_nudge: str | None
    card_steps: dict[str, Any] | None
    upload_steps: dict[str, Any] | None
    openrag_docs_filter_id: str | None
    user_doc_filter_id: str | None
    openrag_docs_ingested_version: str | None
    openrag_docs_remote_signature: str | None


class OpenAIProviderConfig(BaseModel):
    has_api_key: bool
    configured: bool


class AnthropicProviderConfig(BaseModel):
    has_api_key: bool
    configured: bool


class WatsonXProviderConfig(BaseModel):
    has_api_key: bool
    endpoint: str | None
    project_id: str | None
    configured: bool


class OllamaProviderConfig(BaseModel):
    endpoint: str | None
    configured: bool


class ProvidersConfig(BaseModel):
    openai: OpenAIProviderConfig
    anthropic: AnthropicProviderConfig
    watsonx: WatsonXProviderConfig
    ollama: OllamaProviderConfig


class KnowledgeConfig(BaseModel):
    embedding_model: str | None
    embedding_provider: str | None
    chunk_size: int | None
    chunk_overlap: int | None
    table_structure: bool | None
    ocr: bool | None
    picture_descriptions: bool | None
    index_name: str | None
    disable_ingest_with_langflow: bool | None


class AgentConfig(BaseModel):
    llm_model: str | None
    llm_provider: str | None
    system_prompt: str | None


class IngestionDefaultsConfig(BaseModel):
    chunkSize: int | None
    chunkOverlap: int | None
    separator: str | None
    embeddingModel: str | None


class SettingsResponse(BaseModel):
    langflow_url: str
    flow_id: str | None
    ingest_flow_id: str | None
    langflow_public_url: str | None
    edited: bool
    onboarding: OnboardingStateConfig
    providers: ProvidersConfig
    knowledge: KnowledgeConfig
    agent: AgentConfig
    localhost_url: str
    langflow_edit_url: str | None = None
    langflow_ingest_edit_url: str | None = None
    ingestion_defaults: IngestionDefaultsConfig | None = None
    ingest_via_chat: bool = False
    segment_write_key: str | None = None
    environment: str | None = None


class OnboardingResponse(BaseModel):
    message: str
    edited: bool
    sample_data_ingested: bool
    openrag_docs_filter_id: str | None = None
    task_id: str | None = None


class RefreshOpenRAGDocsResponse(BaseModel):
    message: str
    refreshed: bool


class DoclingPresetResponse(BaseModel):
    message: str
    settings: dict
    preset_config: DoclingConfig


class OnboardingStateResponse(BaseModel):
    message: str
    updated_fields: list[str]


class SettingsUpdateResponse(BaseModel):
    message: str


class RollbackResponse(BaseModel):
    message: str
    cancelled_tasks: int
    deleted_files: int
    reset_flows: int
    deleted_conversations: int


class RollbackBody(BaseModel):
    embedding_only: bool = False