TravelContentCreator/api/models/tweet.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
文字内容API模型定义
"""

from typing import List, Dict, Any, Optional
from pydantic import BaseModel, Field


class TopicRequest(BaseModel):
    """选题生成请求模型"""
    dates: Optional[str] = Field(None, description="日期字符串，可能为单个日期、多个日期用逗号分隔或范围如'2023-01-01 to 2023-01-31'")
    numTopics: int = Field(5, description="要生成的选题数量", ge=1, le=10)
    styleIds: Optional[List[int]] = Field(None, description="风格ID列表")
    audienceIds: Optional[List[int]] = Field(None, description="受众ID列表")
    scenicSpotIds: Optional[List[int]] = Field(None, description="景区ID列表")
    productIds: Optional[List[int]] = Field(None, description="产品ID列表")
    
    class Config:
        schema_extra = {
            "example": {
                "dates": "2023-07-01 to 2023-07-31",
                "numTopics": 5,
                "styleIds": [1, 2],
                "audienceIds": [1, 2],
                "scenicSpotIds": [1, 2],
                "productIds": [1, 2]
            }
        }


class TopicResponse(BaseModel):
    """选题生成响应模型"""
    requestId: str = Field(..., description="请求ID")
    topics: List[Dict[str, Any]] = Field(..., description="生成的选题列表")
    
    class Config:
        schema_extra = {
            "example": {
                "requestId": "topic-20240715-123456-a1b2c3d4",
                "topics": [
                    {
                        "index": "1",
                        "date": "2023-07-15",
                        "object": "北京故宫",
                        "product": "故宫门票",
                        "productLogic": "套票包含水上碰碰船等消暑项目，契合夏季游玩需求",
                        "style": "旅游攻略",
                        "styleLogic": "重点解析如何避开高温时段并高效游玩各园区",
                        "targetAudience": "年轻人",
                        "targetAudienceLogic": "解决家长担心孩子中暑的问题，提供科学游玩方案",
                        "logic": "暑期旅游热门景点推荐"
                    }
                ]
            }
        }


class ContentRequest(BaseModel):
    """内容生成请求模型"""
    topic: Optional[Dict[str, Any]] = Field(None, description="选题信息")
    styleIds: Optional[List[int]] = Field(None, description="风格ID列表")
    audienceIds: Optional[List[int]] = Field(None, description="受众ID列表")
    scenicSpotIds: Optional[List[int]] = Field(None, description="景区ID列表")
    productIds: Optional[List[int]] = Field(None, description="产品ID列表")
    autoJudge: bool = Field(False, description="是否自动进行内容审核")
    
    class Config:
        schema_extra = {
            "example": {
                "topic": {
                    "index": "1",
                    "date": "2024-07-01",
                    "style": "攻略风",
                    "styleLogic": "重点解析如何避开高温时段并高效游玩各园区",
                    "targetAudience": "亲子向",
                    "targetAudienceLogic": "解决家长担心孩子中暑的问题，提供科学游玩方案",
                    "object": "天津冒险湾",
                    "product": "冒险湾-2大2小套票",
                    "productLogic": "套票包含水上碰碰船等消暑项目，契合夏季游玩需求",
                    "logic": "暑期旅游热门景点推荐"
                },
                "styleIds": [1, 2],
                "audienceIds": [1, 2],
                "scenicSpotIds": [1, 2],
                "productIds": [1, 2],
                "autoJudge": True
            }
        }


class ContentResponse(BaseModel):
    """内容生成响应模型"""
    requestId: str = Field(..., description="请求ID")
    topicIndex: str = Field(..., description="选题索引")
    content: Dict[str, Any] = Field(..., description="生成的内容")
    judgeSuccess: Optional[bool] = Field(None, description="审核是否成功（仅在启用自动审核时返回）")
    
    class Config:
        schema_extra = {
            "example": {
                "requestId": "content-20240715-123456-a1b2c3d4",
                "topicIndex": "1",
                "content": {
                    "title": "天津冒险湾亲子游攻略",
                    "content": "详细的游玩攻略内容...",
                    "tags": ["亲子游", "水上乐园", "天津"]
                },
                "judgeSuccess": True
            }
        }


class JudgeRequest(BaseModel):
    """内容审核请求模型"""
    topic: Optional[Dict[str, Any]] = Field(None, description="选题信息")
    content: Dict[str, Any] = Field(..., description="要审核的内容")
    styleIds: Optional[List[int]] = Field(None, description="风格ID列表")
    audienceIds: Optional[List[int]] = Field(None, description="受众ID列表")
    scenicSpotIds: Optional[List[int]] = Field(None, description="景区ID列表")
    productIds: Optional[List[int]] = Field(None, description="产品ID列表")
    
    class Config:
        schema_extra = {
            "example": {
                "topic": {
                    "index": "1",
                    "date": "2024-07-01",
                    "style": "攻略风",
                    "styleLogic": "重点解析如何避开高温时段并高效游玩各园区",
                    "targetAudience": "亲子向",
                    "targetAudienceLogic": "解决家长担心孩子中暑的问题，提供科学游玩方案",
                    "object": "天津冒险湾",
                    "product": "冒险湾-2大2小套票",
                    "productLogic": "套票包含水上碰碰船等消暑项目，契合夏季游玩需求",
                    "logic": "暑期旅游热门景点推荐"
                },
                "content": {
                    "title": "天津冒险湾亲子游攻略",
                    "content": "详细的游玩攻略内容...",
                    "tags": ["亲子游", "水上乐园", "天津"]
                },
                "styleIds": [1],
                "audienceIds": [1],
                "scenicSpotIds": [1],
                "productIds": [1]
            }
        }


class JudgeResponse(BaseModel):
    """内容审核响应模型"""
    requestId: str = Field(..., description="请求ID")
    topicIndex: str = Field(..., description="选题索引")
    content: Dict[str, Any] = Field(..., description="审核后的内容")
    judgeSuccess: bool = Field(..., description="审核是否成功")
    
    class Config:
        schema_extra = {
            "example": {
                "requestId": "judge-20240715-123456-a1b2c3d4",
                "topicIndex": "1",
                "content": {
                    "title": "天津冒险湾亲子游攻略",
                    "content": "经过审核的详细游玩攻略内容...",
                    "tags": ["亲子游", "水上乐园", "天津"],
                    "judgeSuccess": True
                },
                "judgeSuccess": True
            }
        }


class PipelineRequest(BaseModel):
    """流水线请求模型"""
    dates: Optional[str] = Field(None, description="日期范围，如：'2024-07-01 to 2024-07-31'")
    numTopics: int = Field(5, description="要生成的选题数量")
    styleIds: Optional[List[int]] = Field(None, description="风格ID列表")
    audienceIds: Optional[List[int]] = Field(None, description="受众ID列表")
    scenicSpotIds: Optional[List[int]] = Field(None, description="景区ID列表")
    productIds: Optional[List[int]] = Field(None, description="产品ID列表")
    skipJudge: bool = Field(False, description="是否跳过内容审核步骤")
    autoJudge: bool = Field(False, description="是否在内容生成时进行内嵌审核")
    
    class Config:
        schema_extra = {
            "example": {
                "dates": "2024-07-01 to 2024-07-31",
                "numTopics": 5,
                "styleIds": [1, 2],
                "audienceIds": [1, 2],
                "scenicSpotIds": [1, 2],
                "productIds": [1, 2],
                "skipJudge": False,
                "autoJudge": True
            }
        }


class PipelineResponse(BaseModel):
    """完整流程响应模型"""
    requestId: str = Field(..., description="请求ID")
    topics: List[Dict[str, Any]] = Field(..., description="生成的选题列表")
    contents: Dict[str, Dict[str, Any]] = Field(..., description="生成的内容，键为选题索引")
    judgedContents: Dict[str, Dict[str, Any]] = Field(..., description="审核后的内容，键为选题索引")
    
    class Config:
        schema_extra = {
            "example": {
                "requestId": "pipeline-20240715-123456-a1b2c3d4",
                "topics": [
                    {
                        "index": "1",
                        "date": "2024-07-01",
                        "style": "攻略风",
                        "targetAudience": "亲子向",
                        "object": "天津冒险湾",
                        "product": "冒险湾-2大2小套票"
                    }
                ],
                "contents": {
                    "1": {
                        "title": "天津冒险湾亲子游攻略",
                        "content": "详细的游玩攻略内容...",
                        "tags": ["亲子游", "水上乐园", "天津"]
                    }
                },
                "judgedContents": {
                    "1": {
                        "title": "天津冒险湾亲子游攻略",
                        "content": "经过审核的详细游玩攻略内容...",
                        "tags": ["亲子游", "水上乐园", "天津"],
                        "judgeSuccess": True
                    }
                }
            }
        }