206 lines
4.4 KiB
Python
206 lines
4.4 KiB
Python
from typing import Any
|
||
|
||
from pydantic import BaseModel, Field
|
||
|
||
|
||
class RewriteRequest(BaseModel):
|
||
source_text: str = Field(..., min_length=20)
|
||
title_hint: str = ""
|
||
writing_style: str = "科普解读"
|
||
tone: str = "专业、可信、可读性强"
|
||
audience: str = "公众号读者"
|
||
keep_points: str = ""
|
||
avoid_words: str = ""
|
||
target_body_chars: int = Field(
|
||
default=500,
|
||
ge=180,
|
||
le=2200,
|
||
description="目标改写正文字数，前端可调（中文字符数近似）",
|
||
)
|
||
|
||
|
||
class RewriteResponse(BaseModel):
|
||
title: str
|
||
summary: str
|
||
body_markdown: str
|
||
mode: str = "ai"
|
||
quality_notes: list[str] = []
|
||
trace: dict[str, Any] | None = Field(
|
||
default=None,
|
||
description="改写链路追踪：请求 ID、耗时、模型、质检与降级原因，便于监测与回溯",
|
||
)
|
||
|
||
|
||
class WechatPublishRequest(BaseModel):
|
||
title: str
|
||
summary: str = ""
|
||
body_markdown: str
|
||
author: str = ""
|
||
thumb_media_id: str = ""
|
||
|
||
|
||
class IMPublishRequest(BaseModel):
|
||
title: str
|
||
body_markdown: str
|
||
|
||
|
||
class PublishResponse(BaseModel):
|
||
ok: bool
|
||
detail: str
|
||
data: dict | None = None
|
||
|
||
|
||
class AuthCredentialRequest(BaseModel):
|
||
username: str
|
||
password: str
|
||
remember_me: bool = False
|
||
challenge_id: str = ""
|
||
challenge_answer: str = ""
|
||
honeypot: str = ""
|
||
|
||
|
||
class ChangePasswordRequest(BaseModel):
|
||
old_password: str
|
||
new_password: str
|
||
|
||
|
||
class ForgotPasswordResetRequest(BaseModel):
|
||
username: str
|
||
reset_key: str
|
||
new_password: str
|
||
|
||
|
||
class DeleteAccountRequest(BaseModel):
|
||
password: str
|
||
reset_key: str
|
||
|
||
|
||
class WechatBindingRequest(BaseModel):
|
||
account_name: str = ""
|
||
appid: str
|
||
secret: str
|
||
author: str = ""
|
||
thumb_media_id: str = ""
|
||
thumb_image_path: str = ""
|
||
|
||
|
||
class WechatSwitchRequest(BaseModel):
|
||
account_id: int
|
||
|
||
|
||
class WechatDeleteRequest(BaseModel):
|
||
account_id: int
|
||
|
||
|
||
class WechatCoverUploadByUrlRequest(BaseModel):
|
||
image_url: str
|
||
|
||
|
||
class WechatCoverGenerateRequest(BaseModel):
|
||
title: str = ""
|
||
summary: str = ""
|
||
style_hint: str = ""
|
||
image_model: str = ""
|
||
upload_to_wechat: bool = True
|
||
|
||
|
||
class AIModelCreateRequest(BaseModel):
|
||
model_name: str
|
||
api_key: str
|
||
base_url: str = ""
|
||
model: str
|
||
image_model: str = ""
|
||
timeout_sec: float = 120.0
|
||
max_output_tokens: int = 8192
|
||
max_retries: int = 0
|
||
|
||
|
||
class AIModelSwitchRequest(BaseModel):
|
||
model_id: int
|
||
|
||
|
||
class AIModelDeleteRequest(BaseModel):
|
||
model_id: int
|
||
|
||
|
||
class AIImageModelUpdateRequest(BaseModel):
|
||
image_model: str
|
||
|
||
|
||
class VipToggleRequest(BaseModel):
|
||
enabled: bool = True
|
||
|
||
|
||
class VipRechargeRequest(BaseModel):
|
||
tokens: int = Field(default=1500, ge=1, le=10_000_000)
|
||
|
||
|
||
class BillingRechargeCreateRequest(BaseModel):
|
||
tokens: int = Field(default=1500, ge=1, le=10_000_000)
|
||
amount_cny: float = Field(default=19.9, ge=0.01, le=999999)
|
||
channel: str = "wechat"
|
||
subscriber_name: str = ""
|
||
subscriber_phone: str = ""
|
||
shipping_address: str = ""
|
||
|
||
|
||
class BillingRechargeNotifyRequest(BaseModel):
|
||
order_no: str
|
||
paid_amount_cny: float = Field(default=0.0, ge=0.0)
|
||
external_txn_id: str = ""
|
||
status: str = "paid"
|
||
|
||
|
||
class BillingPayNowRequest(BaseModel):
|
||
order_no: str
|
||
|
||
|
||
class UserProfileUpdateRequest(BaseModel):
|
||
subscriber_name: str = ""
|
||
subscriber_phone: str = ""
|
||
shipping_address: str = ""
|
||
|
||
|
||
class ResetCodeRegenerateRequest(BaseModel):
|
||
password: str
|
||
|
||
|
||
class PosterGenerateRequest(BaseModel):
|
||
title: str = ""
|
||
summary: str = ""
|
||
body_markdown: str = Field(..., min_length=20)
|
||
style_hint: str = ""
|
||
image_model: str = ""
|
||
upload_to_wechat: bool = True
|
||
max_images: int = Field(default=6, ge=1, le=12)
|
||
|
||
|
||
class PosterPreviewItem(BaseModel):
|
||
paragraph_index: int
|
||
paragraph_excerpt: str = ""
|
||
prompt: str = ""
|
||
preview_data_url: str
|
||
wechat_url: str = ""
|
||
uploaded: bool = False
|
||
note: str = ""
|
||
|
||
|
||
class PosterGenerateResponse(BaseModel):
|
||
ok: bool
|
||
detail: str
|
||
skipped_first_paragraph: bool = True
|
||
posters: list[PosterPreviewItem] = Field(default_factory=list)
|
||
body_markdown_with_posters: str = ""
|
||
warnings: list[str] = Field(default_factory=list)
|
||
|
||
|
||
class CoverGenerateResponse(BaseModel):
|
||
ok: bool
|
||
detail: str
|
||
preview_data_url: str = ""
|
||
thumb_media_id: str = ""
|
||
width: int = 900
|
||
height: int = 383
|
||
note: str = ""
|
||
warnings: list[str] = Field(default_factory=list)
|