bwstudio
6d015bb56d
- 后端 prompt 支持风格选择（6种）和字数范围 - 后端 AI 输出改为 JSON 格式，含评分和推荐理由 - GenerateRequest/Response schema 新增 style/length/score/reason 字段 - 前端新增风格选择、字数范围控件 - 前端新增随机选题按钮 - 前端新增复制内容功能 - 前端新增生成历史记录（会话级 20 条） - 结果卡片展示 AI 评分和推荐理由
102 lines
2.4 KiB
Python
102 lines
2.4 KiB
Python
from datetime import datetime
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class JokeBase(BaseModel):
|
|
title: str
|
|
content: str
|
|
type_ids: list[int] | None = None
|
|
crowd_ids: list[int] | None = None
|
|
|
|
|
|
class JokeCreate(JokeBase):
|
|
pass
|
|
|
|
|
|
class JokeUpdate(BaseModel):
|
|
title: str | None = None
|
|
content: str | None = None
|
|
polished_content: str | None = None
|
|
ai_score: float | None = None
|
|
ai_level: str | None = None
|
|
type_ids: list[int] | None = None
|
|
crowd_ids: list[int] | None = None
|
|
status: str | None = None
|
|
|
|
|
|
class JokeResponse(JokeBase):
|
|
id: int
|
|
status: str
|
|
polished_content: str | None = None
|
|
# AI 评价字段
|
|
ai_score: float | None = None
|
|
ai_level: str | None = None
|
|
view_count: int
|
|
like_count: int
|
|
dislike_count: int
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
type_names: list[str] | None = None
|
|
crowd_names: list[str] | None = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
@property
|
|
def display_content(self) -> str:
|
|
"""优先显示润色后的内容"""
|
|
return self.polished_content if self.polished_content else self.content
|
|
|
|
def get_level_display(self) -> str:
|
|
"""获取 AI 评级的中文显示"""
|
|
level_map = {
|
|
'excellent': '⭐⭐⭐ 精品',
|
|
'good': '⭐⭐ 良好',
|
|
'ordinary': '⭐ 普通',
|
|
'poor': '⚠️ 待优化'
|
|
}
|
|
return level_map.get(self.ai_level, '') if self.ai_level else ''
|
|
|
|
|
|
class PaginatedJokeResponse(BaseModel):
|
|
items: list[JokeResponse]
|
|
total: int
|
|
page: int
|
|
page_size: int
|
|
|
|
|
|
# ============================================================
|
|
# 笑话生成器 Schema
|
|
# ============================================================
|
|
|
|
class GenerateRequest(BaseModel):
|
|
"""笑话生成请求"""
|
|
keywords: list[str] = []
|
|
scenarios: list[str] = []
|
|
style: str = "twist"
|
|
length: str = "medium"
|
|
temperature: float = 0.8
|
|
|
|
class Config:
|
|
json_schema_extra = {
|
|
"example": {
|
|
"keywords": ["加班"],
|
|
"scenarios": ["职场"],
|
|
"style": "twist",
|
|
"length": "medium",
|
|
"temperature": 0.8,
|
|
}
|
|
}
|
|
|
|
|
|
class GenerateResponse(BaseModel):
|
|
"""笑话生成响应"""
|
|
title: str
|
|
content: str
|
|
score: int = 0
|
|
reason: str = ""
|
|
created_at: datetime | None = None
|
|
|
|
class Config:
|
|
from_attributes = True |