models.py

from __future__ import annotations

from dataclasses import dataclass
from typing import Any, Dict, List, Optional

from pydantic import BaseModel


class ChatRequest(BaseModel):
    message: Optional[str] = None
    prompt: Optional[str] = None
    query: Optional[str] = None
    text: Optional[str] = None
    user_message: Optional[str] = None

    tone: Optional[float] = 0.5
    verbosity: Optional[float] = 0.5
    transparency: Optional[float] = 0.5

    help_mode: Optional[str] = "answer"
    history: Optional[List[Dict[str, Any]]] = None


@dataclass
class SolverResult:
    reply: str
    domain: str
    solved: bool
    help_mode: str
    answer_letter: Optional[str] = None
    answer_value: Optional[str] = None


@dataclass
class ParsedContext:
    raw_message: str
    visible_user_text: str
    full_context_text: str
    question_text: str
    options_text: str
    combined_question_block: str
    recent_conversation: str