feat(api): add GenerateRequest/GenerateResponse schemas
This commit is contained in:
+36
-6
@@ -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
|
||||
Reference in New Issue
Block a user