"""
Goalskill 분류 알고리즘 API 스키마
"""

from pydantic import BaseModel
from typing import Optional, Literal


class ClassifyAndSaveRequest(BaseModel):
    """분류 + 저장 요청"""
    session_id: str
    sender: Literal["I", "M"]       # I=AI, M=User
    part: Literal["A", "B", "C", "D"]
    text: str


class ClassifyOnlyRequest(BaseModel):
    """분류만 (저장 안 함) — 테스트용"""
    sender: Literal["I", "M"]
    part: Literal["A", "B", "C", "D"]
    text: str


class TableQueryRequest(BaseModel):
    """특정 테이블 데이터 조회"""
    session_id: str
    table_name: str