feat(api): add GenerateRequest/GenerateResponse schemas

This commit is contained in:
bwstudio
2026-06-02 19:56:35 +08:00
parent 4651781bc3
commit 0892e6aa04
+36 -6
View File
@@ -6,8 +6,8 @@ from pydantic import BaseModel
class JokeBase(BaseModel):
title: str
content: str
type_id: int | None = None
crowd_id: int | None = None
type_ids: list[int] | None = None
crowd_ids: list[int] | None = None
class JokeCreate(JokeBase):
@@ -17,20 +17,22 @@ class JokeCreate(JokeBase):
class JokeUpdate(BaseModel):
title: str | None = None
content: str | None = None
type_id: int | None = None
crowd_id: int | None = None
polished_content: 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
view_count: int
like_count: int
created_at: datetime
updated_at: datetime
type_name: str | None = None
crowd_name: str | None = None
type_names: list[str] | None = None
crowd_names: list[str] | None = None
class Config:
from_attributes = True
@@ -41,3 +43,31 @@ class PaginatedJokeResponse(BaseModel):
total: int
page: int
page_size: int
# ============================================================
# 笑话生成器 Schema
# ============================================================
class GenerateRequest(BaseModel):
"""笑话生成请求"""
keywords: list[str] = []
scenarios: list[str] = []
class Config:
json_schema_extra = {
"example": {
"keywords": ["加班"],
"scenarios": ["职场"]
}
}
class GenerateResponse(BaseModel):
"""笑话生成响应"""
title: str
content: str
created_at: datetime | None = None
class Config:
from_attributes = True