hgzero/ai-python/app/models/response.py

"""응답 모델"""
from pydantic import BaseModel, Field
from typing import List


class SimpleSuggestion(BaseModel):
    """간소화된 AI 제안사항"""

    id: str = Field(..., description="제안 ID")
    content: str = Field(..., description="제안 내용 (1-2문장)")
    timestamp: str = Field(..., description="타임스탬프 (HH:MM:SS)")
    confidence: float = Field(..., ge=0.0, le=1.0, description="신뢰도 (0-1)")

    class Config:
        json_schema_extra = {
            "example": {
                "id": "sugg-001",
                "content": "신제품의 타겟 고객층을 20-30대로 설정하고, 모바일 우선 전략을 취하기로 논의 중입니다.",
                "timestamp": "00:05:23",
                "confidence": 0.92
            }
        }


class RealtimeSuggestionsResponse(BaseModel):
    """실시간 AI 제안사항 응답"""

    suggestions: List[SimpleSuggestion] = Field(
        default_factory=list,
        description="AI 제안사항 목록"
    )

    class Config:
        json_schema_extra = {
            "example": {
                "suggestions": [
                    {
                        "id": "sugg-001",
                        "content": "신제품의 타겟 고객층을 20-30대로 설정하고...",
                        "timestamp": "00:05:23",
                        "confidence": 0.92
                    }
                ]
            }
        }