261 lines
9.8 KiB
Python
261 lines
9.8 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
API Request Models
|
|||
|
|
API 请求数据模型
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
from typing import List, Dict, Any, Optional, Union
|
|||
|
|
from pydantic import BaseModel, Field, validator
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TopicGenerationRequest(BaseModel):
|
|||
|
|
"""主题生成请求"""
|
|||
|
|
creative_materials: str = Field(..., description="创意素材")
|
|||
|
|
num_topics: int = Field(default=5, ge=1, le=20, description="生成主题数量")
|
|||
|
|
month: Optional[str] = Field(default=None, description="月份限制")
|
|||
|
|
style: str = Field(default="活泼有趣", description="内容风格")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"creative_materials": "春节期间的三亚旅游，包含海滩、美食、住宿等内容",
|
|||
|
|
"num_topics": 5,
|
|||
|
|
"month": "2024年2月",
|
|||
|
|
"style": "活泼有趣"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ContentGenerationRequest(BaseModel):
|
|||
|
|
"""内容生成请求"""
|
|||
|
|
scenic_info: str = Field(..., description="景区信息")
|
|||
|
|
product_info: str = Field(..., description="产品信息")
|
|||
|
|
additional_requirements: Optional[str] = Field(default="", description="额外要求")
|
|||
|
|
style: str = Field(default="活泼有趣", description="内容风格")
|
|||
|
|
target_audience: str = Field(default="年轻旅行者", description="目标受众")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"scenic_info": "三亚亚龙湾，拥有7公里长的银白色海滩，被誉为天下第一湾",
|
|||
|
|
"product_info": "亚龙湾度假套餐，包含豪华酒店住宿、海滩活动、特色餐饮",
|
|||
|
|
"additional_requirements": "突出性价比和家庭友好性",
|
|||
|
|
"style": "活泼有趣",
|
|||
|
|
"target_audience": "年轻家庭"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ContentJudgingRequest(BaseModel):
|
|||
|
|
"""内容评判请求"""
|
|||
|
|
product_info: str = Field(..., description="产品信息")
|
|||
|
|
content_to_judge: str = Field(..., description="待评判内容")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"product_info": "三亚亚龙湾度假套餐，价格2999元，包含3天2晚住宿",
|
|||
|
|
"content_to_judge": "三亚超值游，只要1999元！包含所有费用，豪华体验等你来！"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PosterGenerationRequest(BaseModel):
|
|||
|
|
"""海报生成请求"""
|
|||
|
|
template_name: str = Field(default="vibrant", description="模板名称")
|
|||
|
|
content: Optional[Dict[str, Any]] = Field(default=None, description="海报内容")
|
|||
|
|
images: Optional[List[str]] = Field(default=None, description="图片路径列表")
|
|||
|
|
style_options: Optional[Dict[str, Any]] = Field(default=None, description="样式选项")
|
|||
|
|
|
|||
|
|
# 生成内容的参数（当content为空时使用）
|
|||
|
|
scenic_info: Optional[str] = Field(default=None, description="景区信息")
|
|||
|
|
product_info: Optional[str] = Field(default=None, description="产品信息")
|
|||
|
|
tweet_info: Optional[str] = Field(default="", description="推文信息")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"template_name": "vibrant",
|
|||
|
|
"scenic_info": "正佳极地海洋世界",
|
|||
|
|
"product_info": "夜场票，价格199元",
|
|||
|
|
"tweet_info": "520特惠活动",
|
|||
|
|
"style_options": {
|
|||
|
|
"color_theme": "vibrant",
|
|||
|
|
"transparent_background": True
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DocumentProcessingRequest(BaseModel):
|
|||
|
|
"""文档处理请求"""
|
|||
|
|
action: str = Field(..., description="处理动作", regex="^(extract|integrate|transform)$")
|
|||
|
|
document_paths: List[str] = Field(..., description="文档路径列表")
|
|||
|
|
format_type: Optional[str] = Field(default=None, description="转换格式类型")
|
|||
|
|
additional_requirements: Optional[str] = Field(default="", description="额外要求")
|
|||
|
|
|
|||
|
|
@validator('document_paths')
|
|||
|
|
def validate_document_paths(cls, v):
|
|||
|
|
for path in v:
|
|||
|
|
if not Path(path).exists():
|
|||
|
|
raise ValueError(f"文档路径不存在: {path}")
|
|||
|
|
return v
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"action": "transform",
|
|||
|
|
"document_paths": [
|
|||
|
|
"documents/travel_guide.pdf",
|
|||
|
|
"documents/hotel_info.docx"
|
|||
|
|
],
|
|||
|
|
"format_type": "travel_guide",
|
|||
|
|
"additional_requirements": "突出实用信息和联系方式"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class XHSSearchRequest(BaseModel):
|
|||
|
|
"""小红书搜索请求"""
|
|||
|
|
keyword: str = Field(..., description="搜索关键词")
|
|||
|
|
max_notes: int = Field(default=20, ge=1, le=100, description="最大笔记数量")
|
|||
|
|
sort_type: int = Field(default=0, ge=0, le=4, description="排序类型")
|
|||
|
|
note_type: int = Field(default=0, ge=0, le=2, description="笔记类型")
|
|||
|
|
cookies: Optional[str] = Field(default=None, description="Cookie字符串")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"keyword": "三亚旅游攻略",
|
|||
|
|
"max_notes": 20,
|
|||
|
|
"sort_type": 2,
|
|||
|
|
"note_type": 0,
|
|||
|
|
"cookies": "your_cookie_string_here"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class KeywordAnalysisRequest(BaseModel):
|
|||
|
|
"""关键词分析请求"""
|
|||
|
|
content: str = Field(..., description="待分析内容")
|
|||
|
|
source_type: str = Field(default="general", description="内容源类型")
|
|||
|
|
enable_ai_analysis: bool = Field(default=True, description="是否启用AI分析")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"content": "三亚亚龙湾海滩度假攻略，包含住宿、美食、交通指南",
|
|||
|
|
"source_type": "travel_guide",
|
|||
|
|
"enable_ai_analysis": True
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class ContentAnalysisRequest(BaseModel):
|
|||
|
|
"""内容分析请求"""
|
|||
|
|
search_result: Optional[Dict[str, Any]] = Field(default=None, description="搜索结果")
|
|||
|
|
notes: Optional[List[Dict[str, Any]]] = Field(default=None, description="笔记列表")
|
|||
|
|
|
|||
|
|
@validator('notes', 'search_result')
|
|||
|
|
def validate_input(cls, v, values):
|
|||
|
|
# 至少需要提供一个
|
|||
|
|
if not v and not values.get('search_result') and not values.get('notes'):
|
|||
|
|
raise ValueError("必须提供search_result或notes中的至少一个")
|
|||
|
|
return v
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"notes": [
|
|||
|
|
{
|
|||
|
|
"note_id": "123",
|
|||
|
|
"title": "三亚旅游攻略",
|
|||
|
|
"content": "详细的旅游内容...",
|
|||
|
|
"author": "旅游达人",
|
|||
|
|
"likes": 1000,
|
|||
|
|
"comments": 50
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class FileUploadRequest(BaseModel):
|
|||
|
|
"""文件上传请求"""
|
|||
|
|
description: Optional[str] = Field(default="", description="文件描述")
|
|||
|
|
process_immediately: bool = Field(default=True, description="是否立即处理")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"description": "旅游攻略文档",
|
|||
|
|
"process_immediately": True
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PipelineRequest(BaseModel):
|
|||
|
|
"""完整流水线请求"""
|
|||
|
|
creative_materials: str = Field(..., description="创意素材")
|
|||
|
|
num_topics: int = Field(default=5, ge=1, le=20, description="生成主题数量")
|
|||
|
|
scenic_info: str = Field(..., description="景区信息")
|
|||
|
|
product_info: str = Field(..., description="产品信息")
|
|||
|
|
style: str = Field(default="活泼有趣", description="内容风格")
|
|||
|
|
target_audience: str = Field(default="年轻旅行者", description="目标受众")
|
|||
|
|
generate_poster: bool = Field(default=False, description="是否生成海报")
|
|||
|
|
poster_template: str = Field(default="vibrant", description="海报模板")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"creative_materials": "春节期间的三亚旅游",
|
|||
|
|
"num_topics": 3,
|
|||
|
|
"scenic_info": "三亚亚龙湾",
|
|||
|
|
"product_info": "度假套餐，价格2999元",
|
|||
|
|
"style": "活泼有趣",
|
|||
|
|
"target_audience": "年轻家庭",
|
|||
|
|
"generate_poster": True,
|
|||
|
|
"poster_template": "vibrant"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class CookieManagementRequest(BaseModel):
|
|||
|
|
"""Cookie管理请求"""
|
|||
|
|
name: str = Field(..., description="Cookie名称")
|
|||
|
|
cookie_string: str = Field(..., description="Cookie字符串")
|
|||
|
|
description: Optional[str] = Field(default="", description="Cookie描述")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"name": "xhs_main",
|
|||
|
|
"cookie_string": "a1=xxx; web_session=xxx;",
|
|||
|
|
"description": "主要的小红书Cookie"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
class PromptBuildRequest(BaseModel):
|
|||
|
|
"""提示词构建请求"""
|
|||
|
|
task_type: str = Field(..., description="任务类型")
|
|||
|
|
scenic_info: Optional[str] = Field(default="", description="景区信息")
|
|||
|
|
product_info: Optional[str] = Field(default="", description="产品信息")
|
|||
|
|
style: Optional[str] = Field(default="", description="风格要求")
|
|||
|
|
target_audience: Optional[str] = Field(default="", description="目标受众")
|
|||
|
|
custom_requirements: Optional[str] = Field(default="", description="自定义要求")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"task_type": "content_generation",
|
|||
|
|
"scenic_info": "三亚亚龙湾",
|
|||
|
|
"product_info": "度假套餐",
|
|||
|
|
"style": "活泼有趣",
|
|||
|
|
"target_audience": "年轻家庭",
|
|||
|
|
"custom_requirements": "突出性价比"
|
|||
|
|
}
|
|||
|
|
}
|