from __future__ import annotations

from typing import Any, Literal

from pydantic import BaseModel, Field


class AgentChatRequest(BaseModel):
    message: str
    thread_id: str | None = None
    mine_id: str | None = None
    workflow: str = "chat"
    context: dict[str, Any] = Field(default_factory=dict)


class AgentRunResponse(BaseModel):
    run_id: str
    session_id: str
    workflow: str
    status: str
    output: Any = None
    error: str = ""


class WorkflowRunRequest(BaseModel):
    mine_id: str
    workflow: Literal["diagnosis", "evaluation", "knowledge_qa", "document_analysis"]
    payload: dict[str, Any] = Field(default_factory=dict)
    requested_by: str = ""


class AgentCapability(BaseModel):
    name: str
    label: str
    enabled: bool
    description: str
    model: str = ""