feat(api): add GenerateRequest/GenerateResponse schemas
This commit is contained in:
+36
-6
@@ -6,8 +6,8 @@ from pydantic import BaseModel
|
|||||||
class JokeBase(BaseModel):
|
class JokeBase(BaseModel):
|
||||||
title: str
|
title: str
|
||||||
content: str
|
content: str
|
||||||
type_id: int | None = None
|
type_ids: list[int] | None = None
|
||||||
crowd_id: int | None = None
|
crowd_ids: list[int] | None = None
|
||||||
|
|
||||||
|
|
||||||
class JokeCreate(JokeBase):
|
class JokeCreate(JokeBase):
|
||||||
@@ -17,20 +17,22 @@ class JokeCreate(JokeBase):
|
|||||||
class JokeUpdate(BaseModel):
|
class JokeUpdate(BaseModel):
|
||||||
title: str | None = None
|
title: str | None = None
|
||||||
content: str | None = None
|
content: str | None = None
|
||||||
type_id: int | None = None
|
polished_content: str | None = None
|
||||||
crowd_id: int | None = None
|
type_ids: list[int] | None = None
|
||||||
|
crowd_ids: list[int] | None = None
|
||||||
status: str | None = None
|
status: str | None = None
|
||||||
|
|
||||||
|
|
||||||
class JokeResponse(JokeBase):
|
class JokeResponse(JokeBase):
|
||||||
id: int
|
id: int
|
||||||
status: str
|
status: str
|
||||||
|
polished_content: str | None = None
|
||||||
view_count: int
|
view_count: int
|
||||||
like_count: int
|
like_count: int
|
||||||
created_at: datetime
|
created_at: datetime
|
||||||
updated_at: datetime
|
updated_at: datetime
|
||||||
type_name: str | None = None
|
type_names: list[str] | None = None
|
||||||
crowd_name: str | None = None
|
crowd_names: list[str] | None = None
|
||||||
|
|
||||||
class Config:
|
class Config:
|
||||||
from_attributes = True
|
from_attributes = True
|
||||||
@@ -41,3 +43,31 @@ class PaginatedJokeResponse(BaseModel):
|
|||||||
total: int
|
total: int
|
||||||
page: int
|
page: int
|
||||||
page_size: 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
|
||||||
Reference in New Issue
Block a user