302 lines
12 KiB
Python
302 lines
12 KiB
Python
import json
|
||
import logging
|
||
from typing import Dict, List
|
||
|
||
from fastapi import APIRouter, Body, Depends, HTTPException, status
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from ...core.dependencies import get_current_user
|
||
from ...db.session import get_session
|
||
from ...schemas.novel import (
|
||
Blueprint,
|
||
BlueprintGenerationResponse,
|
||
BlueprintPatch,
|
||
Chapter as ChapterSchema,
|
||
ConverseRequest,
|
||
ConverseResponse,
|
||
NovelProject as NovelProjectSchema,
|
||
NovelProjectSummary,
|
||
NovelSectionResponse,
|
||
NovelSectionType,
|
||
)
|
||
from ...schemas.user import UserInDB
|
||
from ...services.llm_service import LLMService
|
||
from ...services.novel_service import NovelService
|
||
from ...services.prompt_service import PromptService
|
||
from ...utils.json_utils import remove_think_tags, unwrap_markdown_json
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
router = APIRouter(prefix="/api/novels", tags=["Novels"])
|
||
|
||
JSON_RESPONSE_INSTRUCTION = """
|
||
IMPORTANT: 你的回复必须是合法的 JSON 对象，并严格包含以下字段：
|
||
{
|
||
"ai_message": "string",
|
||
"ui_control": {
|
||
"type": "single_choice | text_input | info_display",
|
||
"options": [
|
||
{"id": "option_1", "label": "string"}
|
||
],
|
||
"placeholder": "string"
|
||
},
|
||
"conversation_state": {},
|
||
"is_complete": false
|
||
}
|
||
不要输出额外的文本或解释。
|
||
"""
|
||
|
||
|
||
def _ensure_prompt(prompt: str | None, name: str) -> str:
|
||
if not prompt:
|
||
raise HTTPException(status_code=500, detail=f"未配置名为 {name} 的提示词，请联系管理员")
|
||
return prompt
|
||
|
||
|
||
@router.post("", response_model=NovelProjectSchema, status_code=status.HTTP_201_CREATED)
|
||
async def create_novel(
|
||
title: str = Body(...),
|
||
initial_prompt: str = Body(...),
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> NovelProjectSchema:
|
||
"""为当前用户创建一个新的小说项目。"""
|
||
novel_service = NovelService(session)
|
||
project = await novel_service.create_project(current_user.id, title, initial_prompt)
|
||
logger.info("用户 %s 创建项目 %s", current_user.id, project.id)
|
||
return await novel_service.get_project_schema(project.id, current_user.id)
|
||
|
||
|
||
@router.get("", response_model=List[NovelProjectSummary])
|
||
async def list_novels(
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> List[NovelProjectSummary]:
|
||
"""列出用户的全部小说项目摘要信息。"""
|
||
novel_service = NovelService(session)
|
||
projects = await novel_service.list_projects_for_user(current_user.id)
|
||
logger.info("用户 %s 获取项目列表，共 %s 个", current_user.id, len(projects))
|
||
return projects
|
||
|
||
|
||
@router.get("/{project_id}", response_model=NovelProjectSchema)
|
||
async def get_novel(
|
||
project_id: str,
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> NovelProjectSchema:
|
||
novel_service = NovelService(session)
|
||
logger.info("用户 %s 查询项目 %s", current_user.id, project_id)
|
||
return await novel_service.get_project_schema(project_id, current_user.id)
|
||
|
||
|
||
@router.get("/{project_id}/sections/{section}", response_model=NovelSectionResponse)
|
||
async def get_novel_section(
|
||
project_id: str,
|
||
section: NovelSectionType,
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> NovelSectionResponse:
|
||
novel_service = NovelService(session)
|
||
logger.info("用户 %s 获取项目 %s 的 %s 区段", current_user.id, project_id, section)
|
||
return await novel_service.get_section_data(project_id, current_user.id, section)
|
||
|
||
|
||
@router.get("/{project_id}/chapters/{chapter_number}", response_model=ChapterSchema)
|
||
async def get_chapter(
|
||
project_id: str,
|
||
chapter_number: int,
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> ChapterSchema:
|
||
novel_service = NovelService(session)
|
||
logger.info("用户 %s 获取项目 %s 第 %s 章", current_user.id, project_id, chapter_number)
|
||
return await novel_service.get_chapter_schema(project_id, current_user.id, chapter_number)
|
||
|
||
|
||
@router.delete("", status_code=status.HTTP_200_OK)
|
||
async def delete_novels(
|
||
project_ids: List[str] = Body(...),
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> Dict[str, str]:
|
||
novel_service = NovelService(session)
|
||
await novel_service.delete_projects(project_ids, current_user.id)
|
||
logger.info("用户 %s 删除项目 %s", current_user.id, project_ids)
|
||
return {"status": "success", "message": f"成功删除 {len(project_ids)} 个项目"}
|
||
|
||
|
||
@router.post("/{project_id}/concept/converse", response_model=ConverseResponse)
|
||
async def converse_with_concept(
|
||
project_id: str,
|
||
request: ConverseRequest,
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> ConverseResponse:
|
||
"""与概念设计师（LLM）进行对话，引导蓝图筹备。"""
|
||
novel_service = NovelService(session)
|
||
prompt_service = PromptService(session)
|
||
llm_service = LLMService(session)
|
||
|
||
project = await novel_service.ensure_project_owner(project_id, current_user.id)
|
||
|
||
history_records = await novel_service.list_conversations(project_id)
|
||
logger.info(
|
||
"项目 %s 概念对话请求，用户 %s，历史记录 %s 条",
|
||
project_id,
|
||
current_user.id,
|
||
len(history_records),
|
||
)
|
||
conversation_history = [
|
||
{"role": record.role, "content": record.content}
|
||
for record in history_records
|
||
]
|
||
user_content = json.dumps(request.user_input, ensure_ascii=False)
|
||
conversation_history.append({"role": "user", "content": user_content})
|
||
|
||
system_prompt = _ensure_prompt(await prompt_service.get_prompt("concept"), "concept")
|
||
system_prompt = f"{system_prompt}\n{JSON_RESPONSE_INSTRUCTION}"
|
||
|
||
llm_response = await llm_service.get_llm_response(
|
||
system_prompt=system_prompt,
|
||
conversation_history=conversation_history,
|
||
temperature=0.8,
|
||
user_id=current_user.id,
|
||
timeout=240.0,
|
||
)
|
||
llm_response = remove_think_tags(llm_response)
|
||
|
||
try:
|
||
normalized = unwrap_markdown_json(llm_response)
|
||
parsed = json.loads(normalized)
|
||
except json.JSONDecodeError as exc:
|
||
logger.exception(
|
||
"Failed to parse concept converse response: project_id=%s user_id=%s normalized=%s",
|
||
project_id,
|
||
current_user.id,
|
||
normalized,
|
||
)
|
||
raise HTTPException(status_code=500, detail="AI 返回内容不是有效的 JSON") from exc
|
||
|
||
await novel_service.append_conversation(project_id, "user", user_content)
|
||
await novel_service.append_conversation(project_id, "assistant", normalized)
|
||
|
||
logger.info("项目 %s 概念对话完成，is_complete=%s", project_id, parsed.get("is_complete"))
|
||
|
||
if parsed.get("is_complete"):
|
||
parsed["ready_for_blueprint"] = True
|
||
|
||
parsed.setdefault("conversation_state", parsed.get("conversation_state", {}))
|
||
return ConverseResponse(**parsed)
|
||
|
||
|
||
@router.post("/{project_id}/blueprint/generate", response_model=BlueprintGenerationResponse)
|
||
async def generate_blueprint(
|
||
project_id: str,
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> BlueprintGenerationResponse:
|
||
"""根据完整对话生成可执行的小说蓝图。"""
|
||
novel_service = NovelService(session)
|
||
prompt_service = PromptService(session)
|
||
llm_service = LLMService(session)
|
||
|
||
project = await novel_service.ensure_project_owner(project_id, current_user.id)
|
||
logger.info("项目 %s 开始生成蓝图", project_id)
|
||
|
||
history_records = await novel_service.list_conversations(project_id)
|
||
if not history_records:
|
||
raise HTTPException(status_code=400, detail="缺少对话历史，无法生成蓝图")
|
||
|
||
formatted_history: List[Dict[str, str]] = []
|
||
for record in history_records:
|
||
role = record.role
|
||
content = record.content
|
||
if not role or not content:
|
||
continue
|
||
try:
|
||
normalized = unwrap_markdown_json(content)
|
||
data = json.loads(normalized)
|
||
if role == "user":
|
||
user_value = data.get("value", data)
|
||
if isinstance(user_value, str):
|
||
formatted_history.append({"role": "user", "content": user_value})
|
||
elif role == "assistant":
|
||
ai_message = data.get("ai_message") if isinstance(data, dict) else None
|
||
if ai_message:
|
||
formatted_history.append({"role": "assistant", "content": ai_message})
|
||
except (json.JSONDecodeError, AttributeError):
|
||
continue
|
||
|
||
if not formatted_history:
|
||
raise HTTPException(status_code=400, detail="无法从历史对话中提取内容")
|
||
|
||
system_prompt = _ensure_prompt(await prompt_service.get_prompt("screenwriting"), "screenwriting")
|
||
blueprint_raw = await llm_service.get_llm_response(
|
||
system_prompt=system_prompt,
|
||
conversation_history=formatted_history,
|
||
temperature=0.3,
|
||
user_id=current_user.id,
|
||
timeout=480.0,
|
||
)
|
||
blueprint_raw = remove_think_tags(blueprint_raw)
|
||
|
||
blueprint_normalized = unwrap_markdown_json(blueprint_raw)
|
||
try:
|
||
blueprint_data = json.loads(blueprint_normalized)
|
||
except json.JSONDecodeError as exc:
|
||
raise HTTPException(status_code=500, detail="蓝图生成失败，请稍后重试") from exc
|
||
|
||
blueprint = Blueprint(**blueprint_data)
|
||
await novel_service.replace_blueprint(project_id, blueprint)
|
||
if blueprint.title:
|
||
project.title = blueprint.title
|
||
project.status = "blueprint_ready"
|
||
await session.commit()
|
||
logger.info("项目 %s 更新标题为 %s，并标记为 blueprint_ready", project_id, blueprint.title)
|
||
|
||
ai_message = (
|
||
"太棒了！我已经根据我们的对话整理出完整的小说蓝图。请确认是否进入写作阶段，或提出修改意见。"
|
||
)
|
||
return BlueprintGenerationResponse(blueprint=blueprint, ai_message=ai_message)
|
||
|
||
|
||
@router.post("/{project_id}/blueprint/save", response_model=NovelProjectSchema)
|
||
async def save_blueprint(
|
||
project_id: str,
|
||
blueprint_data: Blueprint | None = Body(None),
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> NovelProjectSchema:
|
||
"""保存蓝图信息，可用于手动覆盖自动生成结果。"""
|
||
novel_service = NovelService(session)
|
||
project = await novel_service.ensure_project_owner(project_id, current_user.id)
|
||
|
||
if blueprint_data:
|
||
await novel_service.replace_blueprint(project_id, blueprint_data)
|
||
if blueprint_data.title:
|
||
project.title = blueprint_data.title
|
||
await session.commit()
|
||
logger.info("项目 %s 手动保存蓝图", project_id)
|
||
else:
|
||
raise HTTPException(status_code=400, detail="缺少蓝图数据")
|
||
|
||
return await novel_service.get_project_schema(project_id, current_user.id)
|
||
|
||
|
||
@router.patch("/{project_id}/blueprint", response_model=NovelProjectSchema)
|
||
async def patch_blueprint(
|
||
project_id: str,
|
||
payload: BlueprintPatch,
|
||
session: AsyncSession = Depends(get_session),
|
||
current_user: UserInDB = Depends(get_current_user),
|
||
) -> NovelProjectSchema:
|
||
"""局部更新蓝图字段，对世界观或角色做微调。"""
|
||
novel_service = NovelService(session)
|
||
project = await novel_service.ensure_project_owner(project_id, current_user.id)
|
||
|
||
update_data = payload.model_dump(exclude_unset=True)
|
||
await novel_service.patch_blueprint(project_id, update_data)
|
||
logger.info("项目 %s 局部更新蓝图字段：%s", project_id, list(update_data.keys()))
|
||
return await novel_service.get_project_schema(project_id, current_user.id)
|