345 lines
12 KiB
Python
345 lines
12 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
API Response Models
|
|||
|
|
API 响应数据模型
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from typing import List, Dict, Any, Optional, Union
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ApiResponse(BaseModel):
|
|||
|
|
"""通用API响应"""
|
|||
|
|
success: bool = Field(..., description="是否成功")
|
|||
|
|
message: str = Field(default="", description="响应消息")
|
|||
|
|
data: Optional[Any] = Field(default=None, description="响应数据")
|
|||
|
|
error: Optional[str] = Field(default=None, description="错误信息")
|
|||
|
|
request_id: Optional[str] = Field(default=None, description="请求ID")
|
|||
|
|
timestamp: datetime = Field(default_factory=datetime.now, description="响应时间")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "操作成功",
|
|||
|
|
"data": {"result": "example_data"},
|
|||
|
|
"error": None,
|
|||
|
|
"request_id": "req_123456",
|
|||
|
|
"timestamp": "2024-01-01T12:00:00"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TopicGenerationResponse(ApiResponse):
|
|||
|
|
"""主题生成响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="生成的主题数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "主题生成成功",
|
|||
|
|
"data": {
|
|||
|
|
"topics": [
|
|||
|
|
{
|
|||
|
|
"index": 1,
|
|||
|
|
"date": "2024-02-01",
|
|||
|
|
"logic": "春节期间家庭出游需求旺盛",
|
|||
|
|
"object": "三亚亚龙湾",
|
|||
|
|
"product": "家庭度假套餐",
|
|||
|
|
"style": "温馨家庭",
|
|||
|
|
"targetAudience": "年轻家庭"
|
|||
|
|
}
|
|||
|
|
],
|
|||
|
|
"generation_stats": {
|
|||
|
|
"total_generated": 5,
|
|||
|
|
"processing_time": 2.5
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ContentGenerationResponse(ApiResponse):
|
|||
|
|
"""内容生成响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="生成的内容数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "内容生成成功",
|
|||
|
|
"data": {
|
|||
|
|
"content": {
|
|||
|
|
"title": "三亚亚龙湾| 天下第一湾的绝美体验",
|
|||
|
|
"content": "详细的旅游内容描述...",
|
|||
|
|
"tag": ["三亚", "亚龙湾", "海滩度假"]
|
|||
|
|
},
|
|||
|
|
"generation_stats": {
|
|||
|
|
"word_count": 500,
|
|||
|
|
"processing_time": 3.2
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ContentJudgingResponse(ApiResponse):
|
|||
|
|
"""内容评判响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="评判结果数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "内容评判完成",
|
|||
|
|
"data": {
|
|||
|
|
"is_passed": False,
|
|||
|
|
"confidence": 0.85,
|
|||
|
|
"analysis": "价格信息与产品资料不符，存在虚假宣传风险",
|
|||
|
|
"suggested_revisions": "建议修改价格为2999元，并添加具体包含项目"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PosterGenerationResponse(ApiResponse):
|
|||
|
|
"""海报生成响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="海报生成数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "海报生成成功",
|
|||
|
|
"data": {
|
|||
|
|
"poster_path": "output/posters/vibrant_poster_20240101.png",
|
|||
|
|
"fabric_json": "output/json/fabric_data_20240101.json",
|
|||
|
|
"content": {
|
|||
|
|
"title": "正佳极地海洋世界",
|
|||
|
|
"slogan": "夜游海洋，梦幻之旅",
|
|||
|
|
"price": "199",
|
|||
|
|
"ticket_type": "夜场票"
|
|||
|
|
},
|
|||
|
|
"template_info": {
|
|||
|
|
"name": "vibrant",
|
|||
|
|
"size": [900, 1200],
|
|||
|
|
"transparent_background": True
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DocumentProcessingResponse(ApiResponse):
|
|||
|
|
"""文档处理响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="文档处理数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "文档处理完成",
|
|||
|
|
"data": {
|
|||
|
|
"processed_documents": 2,
|
|||
|
|
"output_files": [
|
|||
|
|
"output/documents/integrated_content.txt",
|
|||
|
|
"output/documents/travel_guide_transformed.txt"
|
|||
|
|
],
|
|||
|
|
"statistics": {
|
|||
|
|
"total_characters": 50000,
|
|||
|
|
"key_topics": ["住宿", "交通", "美食", "景点"],
|
|||
|
|
"processing_time": 15.3
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class XHSSearchResponse(ApiResponse):
|
|||
|
|
"""小红书搜索响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="搜索结果数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "搜索完成",
|
|||
|
|
"data": {
|
|||
|
|
"keyword": "三亚旅游攻略",
|
|||
|
|
"notes": [
|
|||
|
|
{
|
|||
|
|
"note_id": "123abc",
|
|||
|
|
"title": "三亚7天6晚详细攻略",
|
|||
|
|
"content": "超详细的三亚旅游攻略...",
|
|||
|
|
"author": "旅游达人小明",
|
|||
|
|
"likes": 1500,
|
|||
|
|
"comments": 89,
|
|||
|
|
"tags": ["三亚", "攻略", "海滩"]
|
|||
|
|
}
|
|||
|
|
],
|
|||
|
|
"total_count": 20,
|
|||
|
|
"search_time": "2024-01-01T12:00:00"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class KeywordAnalysisResponse(ApiResponse):
|
|||
|
|
"""关键词分析响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="关键词分析数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "关键词分析完成",
|
|||
|
|
"data": {
|
|||
|
|
"keywords": ["三亚", "亚龙湾", "海滩", "度假", "攻略"],
|
|||
|
|
"keyword_scores": {
|
|||
|
|
"三亚": 85.5,
|
|||
|
|
"亚龙湾": 78.2,
|
|||
|
|
"海滩": 65.8
|
|||
|
|
},
|
|||
|
|
"categories": {
|
|||
|
|
"地理位置": ["三亚", "亚龙湾"],
|
|||
|
|
"景点类型": ["海滩"],
|
|||
|
|
"活动体验": ["度假"]
|
|||
|
|
},
|
|||
|
|
"search_suggestions": [
|
|||
|
|
"三亚攻略",
|
|||
|
|
"亚龙湾住宿",
|
|||
|
|
"三亚美食"
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ContentAnalysisResponse(ApiResponse):
|
|||
|
|
"""内容分析响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="内容分析数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "内容分析完成",
|
|||
|
|
"data": {
|
|||
|
|
"content_quality_score": 78.5,
|
|||
|
|
"engagement_score": 65.2,
|
|||
|
|
"content_themes": ["旅游攻略", "美食体验", "住宿推荐"],
|
|||
|
|
"sentiment_analysis": {
|
|||
|
|
"positive": 0.7,
|
|||
|
|
"negative": 0.1,
|
|||
|
|
"neutral": 0.2,
|
|||
|
|
"overall": "positive"
|
|||
|
|
},
|
|||
|
|
"recommendations": [
|
|||
|
|
"内容质量较好，建议增加更多实用信息",
|
|||
|
|
"互动数据中等，可关注更热门的内容"
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class FileUploadResponse(ApiResponse):
|
|||
|
|
"""文件上传响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="上传结果数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "文件上传成功",
|
|||
|
|
"data": {
|
|||
|
|
"file_id": "upload_123456",
|
|||
|
|
"file_path": "uploads/document_20240101.pdf",
|
|||
|
|
"file_size": 1024000,
|
|||
|
|
"processing_status": "completed"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PipelineResponse(ApiResponse):
|
|||
|
|
"""完整流水线响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="流水线执行结果")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "流水线执行完成",
|
|||
|
|
"data": {
|
|||
|
|
"topics": [{"index": 1, "title": "三亚亲子游攻略"}],
|
|||
|
|
"content": {"title": "三亚亲子游", "content": "详细内容..."},
|
|||
|
|
"judge_result": {"is_passed": True, "confidence": 0.9},
|
|||
|
|
"poster": {"poster_path": "output/poster.png"},
|
|||
|
|
"execution_time": 30.5
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class CookieManagementResponse(ApiResponse):
|
|||
|
|
"""Cookie管理响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="Cookie管理结果")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "Cookie管理操作成功",
|
|||
|
|
"data": {
|
|||
|
|
"total_cookies": 3,
|
|||
|
|
"valid_cookies": 2,
|
|||
|
|
"current_cookie": "xhs_main",
|
|||
|
|
"operation": "add"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PromptBuildResponse(ApiResponse):
|
|||
|
|
"""提示词构建响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="构建的提示词")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "提示词构建成功",
|
|||
|
|
"data": {
|
|||
|
|
"system_prompt": "你是一位专业的旅游内容创作者...",
|
|||
|
|
"user_prompt": "请为三亚亚龙湾创作内容...",
|
|||
|
|
"prompt_template": "content_generation",
|
|||
|
|
"variables": {"scenic_info": "三亚亚龙湾"}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TaskManagementResponse(ApiResponse):
|
|||
|
|
"""任务管理响应"""
|
|||
|
|
data: Optional[Dict[str, Any]] = Field(default=None, description="任务管理数据")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "任务查询成功",
|
|||
|
|
"data": {
|
|||
|
|
"task_id": "task_123456",
|
|||
|
|
"status": "completed",
|
|||
|
|
"created_at": "2024-01-01T12:00:00",
|
|||
|
|
"completed_at": "2024-01-01T12:05:00",
|
|||
|
|
"result_files": ["output/result.json"],
|
|||
|
|
"statistics": {"processing_time": 300}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|