204 lines
4.3 KiB
Python
204 lines
4.3 KiB
Python
from __future__ import annotations
|
||
|
||
from datetime import datetime
|
||
|
||
from pydantic import BaseModel, Field
|
||
|
||
|
||
class NovelCreate(BaseModel):
|
||
title: str = Field(min_length=1, max_length=200)
|
||
description: str = Field(default="", max_length=2000)
|
||
|
||
|
||
class NovelUpdate(BaseModel):
|
||
title: str | None = Field(default=None, min_length=1, max_length=200)
|
||
description: str | None = Field(default=None, max_length=2000)
|
||
|
||
|
||
class NovelRead(BaseModel):
|
||
id: int
|
||
title: str
|
||
description: str
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
|
||
class NovelList(BaseModel):
|
||
items: list[NovelRead]
|
||
total: int
|
||
|
||
|
||
# ── 大纲 ──
|
||
|
||
|
||
class OutlineCreate(BaseModel):
|
||
summary: str = Field(min_length=1, max_length=200)
|
||
detail: str = Field(default="", max_length=1000)
|
||
parent_id: int | None = None
|
||
|
||
|
||
class OutlineUpdate(BaseModel):
|
||
summary: str | None = Field(default=None, min_length=1, max_length=200)
|
||
detail: str | None = Field(default=None, max_length=1000)
|
||
sort_order: int | None = None
|
||
|
||
|
||
class OutlineRead(BaseModel):
|
||
id: int
|
||
novel_id: int
|
||
parent_id: int | None = None
|
||
sort_order: int
|
||
summary: str
|
||
detail: str
|
||
children: list[OutlineRead] = []
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
|
||
class OutlineList(BaseModel):
|
||
items: list[OutlineRead]
|
||
total: int
|
||
|
||
|
||
class OutlineReorder(BaseModel):
|
||
"""排序请求：按顺序传入大纲id列表"""
|
||
ordered_ids: list[int]
|
||
|
||
|
||
# ── 角色 ──
|
||
|
||
|
||
class CharacterCreate(BaseModel):
|
||
name: str = Field(min_length=1, max_length=100)
|
||
description: str = Field(default="", max_length=1000)
|
||
|
||
|
||
class CharacterUpdate(BaseModel):
|
||
name: str | None = Field(default=None, min_length=1, max_length=100)
|
||
description: str | None = Field(default=None, max_length=1000)
|
||
|
||
|
||
class CharacterRead(BaseModel):
|
||
id: int
|
||
novel_id: int
|
||
name: str
|
||
description: str
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
|
||
class CharacterList(BaseModel):
|
||
items: list[CharacterRead]
|
||
total: int
|
||
|
||
|
||
# ── 章节 ──
|
||
|
||
|
||
class ChapterCreate(BaseModel):
|
||
title: str = Field(min_length=1, max_length=200)
|
||
content: str = Field(default="", max_length=5000)
|
||
|
||
|
||
class ChapterUpdate(BaseModel):
|
||
title: str | None = Field(default=None, min_length=1, max_length=200)
|
||
content: str | None = Field(default=None, max_length=5000)
|
||
sort_order: int | None = None
|
||
|
||
|
||
class ChapterRead(BaseModel):
|
||
id: int
|
||
novel_id: int
|
||
sort_order: int
|
||
title: str
|
||
content: str
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
|
||
class ChapterList(BaseModel):
|
||
items: list[ChapterRead]
|
||
total: int
|
||
|
||
|
||
class ChapterReorder(BaseModel):
|
||
"""排序请求：按顺序传入章节id列表"""
|
||
ordered_ids: list[int]
|
||
|
||
|
||
# ── 世界观 ──
|
||
|
||
|
||
class WorldSettingUpdate(BaseModel):
|
||
content: str = Field(default="", max_length=5000)
|
||
|
||
|
||
class WorldSettingRead(BaseModel):
|
||
id: int
|
||
novel_id: int
|
||
content: str
|
||
created_at: datetime
|
||
updated_at: datetime
|
||
|
||
|
||
# ── AI配置 ──
|
||
|
||
|
||
class AIConfigSave(BaseModel):
|
||
provider: str = Field(default="openai")
|
||
base_url: str = Field(default="https://api.openai.com/v1")
|
||
api_key: str = Field(min_length=1)
|
||
model: str = Field(default="gpt-4o")
|
||
|
||
|
||
class AIConfigRead(BaseModel):
|
||
id: int
|
||
provider: str
|
||
base_url: str
|
||
api_key_masked: str # 脱敏后的key
|
||
model: str
|
||
updated_at: datetime
|
||
|
||
|
||
# ── 聊天 ──
|
||
|
||
|
||
class ChatMessageInput(BaseModel):
|
||
role: str
|
||
content: str
|
||
|
||
|
||
class PendingToolCall(BaseModel):
|
||
"""待审批的工具调用"""
|
||
id: str
|
||
name: str
|
||
arguments: dict
|
||
|
||
|
||
class ChatRequest(BaseModel):
|
||
messages: list[ChatMessageInput]
|
||
novel_id: int | None = None
|
||
page_context: str | None = None # 用户当前所在页面，如 "大纲页" "世界观页"
|
||
tools_enabled: bool = True # 是否启用工具调用
|
||
auto_approve_tools: bool = False # 是否自动执行工具（无需人工审批）
|
||
# 工具审批续传
|
||
pending_tool_calls: list[PendingToolCall] | None = None
|
||
assistant_text: str | None = None # LLM 在工具调用前输出的文本
|
||
|
||
|
||
class ChatMessageRead(BaseModel):
|
||
id: int
|
||
novel_id: int | None
|
||
role: str
|
||
content: str
|
||
created_at: datetime
|
||
|
||
|
||
class ChatMessageList(BaseModel):
|
||
items: list[ChatMessageRead]
|
||
total: int
|
||
|
||
|
||
# 解决 OutlineRead 自引用
|
||
OutlineRead.model_rebuild()
|