Compare commits
3 Commits
feature/ll
...
f3792915a3
| Author | SHA1 | Date | |
|---|---|---|---|
| f3792915a3 | |||
| fa6907fb8d | |||
| bc130d98f4 |
33
README.md
33
README.md
@@ -122,20 +122,35 @@ npm run dev
|
||||
|
||||
前端将在 `http://localhost:5173` 启动,自动代理 API 请求到后端。
|
||||
|
||||
### Docker 部署
|
||||
### Docker 开发(Windows / Docker Desktop)
|
||||
|
||||
```bash
|
||||
# 一键启动
|
||||
docker-compose up -d
|
||||
日常改代码**不需要重启 Docker Desktop**——后端 uvicorn `--reload`、前端 Vite HMR 会自动生效。
|
||||
|
||||
```powershell
|
||||
# 启动(项目根目录)
|
||||
.\scripts\docker-up.ps1
|
||||
|
||||
# 仅重启容器(HMR/reload 异常时)
|
||||
.\scripts\docker-restart.ps1 -Service frontend # 或 backend / all
|
||||
|
||||
# 依赖或 Dockerfile 变更后重建
|
||||
.\scripts\docker-rebuild.ps1 -Service backend
|
||||
|
||||
# 查看日志
|
||||
docker-compose logs -f
|
||||
|
||||
# 停止服务
|
||||
docker-compose down
|
||||
.\scripts\docker-logs.ps1
|
||||
```
|
||||
|
||||
访问 `http://localhost:80` 即可使用。
|
||||
| 服务 | 地址 |
|
||||
|------|------|
|
||||
| 后端 API | http://localhost:23337 |
|
||||
| 前端 | http://localhost:23338 |
|
||||
|
||||
详细说明(何时 rebuild、何时才需要重启 Docker Desktop、本地开发替代方案)见 **[docs/DOCKER_DEV.md](./docs/DOCKER_DEV.md)**。
|
||||
|
||||
```powershell
|
||||
# 停止服务
|
||||
docker compose down
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
from fastapi import APIRouter
|
||||
from .routes import presetsRoute, chatsRoute, worldbooksRoute, apiConfigRoute, charactersRoute, chatWsRoute, tokenUsageRoute, imageGalleryRoute, regexRoute, chatSummaryRoute
|
||||
from .routes import presetsRoute, chatsRoute, worldbooksRoute, apiConfigRoute, charactersRoute, chatWsRoute, tokenUsageRoute, imageGalleryRoute, regexRoute, chatSummaryRoute, studioRoute
|
||||
from utils.file_utils import get_all_roles_and_chats
|
||||
from core.config import settings
|
||||
from pathlib import Path
|
||||
@@ -18,6 +18,7 @@ router.include_router(tokenUsageRoute.router)
|
||||
router.include_router(imageGalleryRoute.router)
|
||||
router.include_router(regexRoute.router)
|
||||
router.include_router(chatSummaryRoute.router)
|
||||
router.include_router(studioRoute.router)
|
||||
|
||||
# ✅ 注册 WebSocket 路由(必须在 HTTP 路由之后,避免路径冲突)
|
||||
router.include_router(chatWsRoute.router)
|
||||
|
||||
@@ -221,119 +221,47 @@ async def _handle_stream_chat(
|
||||
workflow_service
|
||||
):
|
||||
"""
|
||||
处理流式聊天请求
|
||||
|
||||
Args:
|
||||
websocket: WebSocket 连接
|
||||
role_name: 角色名
|
||||
chat_name: 聊天名
|
||||
request_data: 请求数据
|
||||
workflow_service: 工作流服务实例
|
||||
处理流式聊天请求 – engine callbacks emit worldbook_active / tasks_created / chunk.
|
||||
"""
|
||||
try:
|
||||
print(f"[StreamChat] 🚀 开始流式处理")
|
||||
|
||||
# ✅ 第1步:加载角色卡
|
||||
current_role = request_data.get("currentRole")
|
||||
character_data = request_data.get("characterData")
|
||||
|
||||
if character_data:
|
||||
try:
|
||||
from backend.models.internal import CharacterCard
|
||||
except ImportError:
|
||||
from models.internal import CharacterCard
|
||||
character = CharacterCard(**character_data)
|
||||
else:
|
||||
from backend.services.character_service import CharacterService
|
||||
character_service = CharacterService()
|
||||
character = character_service.get_character_by_name(current_role)
|
||||
|
||||
if not character:
|
||||
print(f"[StreamChat] ❌ 错误: 无法加载角色 '{current_role}'")
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"message": f"角色 '{current_role}' 不存在"
|
||||
})
|
||||
return
|
||||
|
||||
print(f"[StreamChat] ✅ 已加载角色卡: {character.name}")
|
||||
|
||||
# ✅ 第2步:激活世界书条目(在LLM调用之前)
|
||||
print(f"[StreamChat] 📚 正在激活世界书条目...")
|
||||
active_entries = await workflow_service._collect_and_activate_worldbooks(
|
||||
request_data,
|
||||
character
|
||||
)
|
||||
|
||||
# ✅ 发送激活的世界书条目信息(在LLM调用前)
|
||||
if active_entries:
|
||||
print(f"[StreamChat] 📤 发送世界书激活信息: {len(active_entries)} 个条目")
|
||||
# 将 Pydantic 模型转换为字典
|
||||
entries_dict = [entry.model_dump() for entry in active_entries]
|
||||
await websocket.send_json({
|
||||
"type": "worldbook_active",
|
||||
"entries": entries_dict
|
||||
})
|
||||
|
||||
# ✅ TODO: RAG检索(暂时为空,待实现)
|
||||
rag_results = []
|
||||
if rag_results:
|
||||
print(f"[StreamChat] 🔍 发送 RAG 检索结果: {len(rag_results)} 条")
|
||||
await websocket.send_json({
|
||||
"type": "rag_results",
|
||||
"results": rag_results
|
||||
})
|
||||
|
||||
# ✅ 第2步:启动并行任务(在LLM调用前创建任务ID)
|
||||
options = request_data.get("options", {})
|
||||
task_ids = {
|
||||
"imageWorkflow": None,
|
||||
"dynamicTable": None
|
||||
}
|
||||
|
||||
if options.get("imageWorkflow", False):
|
||||
import uuid
|
||||
chat_id = f"{role_name}/{chat_name}"
|
||||
task_ids["imageWorkflow"] = f"img_{uuid.uuid4().hex[:8]}"
|
||||
from backend.services.task_queue_manager import task_queue_manager, TaskType
|
||||
await task_queue_manager.add_task(task_ids["imageWorkflow"], TaskType.IMAGE_WORKFLOW, chat_id)
|
||||
|
||||
if options.get("dynamicTable", False):
|
||||
import uuid
|
||||
chat_id = f"{role_name}/{chat_name}"
|
||||
task_ids["dynamicTable"] = f"tbl_{uuid.uuid4().hex[:8]}"
|
||||
from backend.services.task_queue_manager import task_queue_manager, TaskType
|
||||
await task_queue_manager.add_task(task_ids["dynamicTable"], TaskType.DYNAMIC_TABLE, chat_id)
|
||||
|
||||
# ✅ 发送任务ID信息(在LLM调用前)
|
||||
if task_ids.get("imageWorkflow") or task_ids.get("dynamicTable"):
|
||||
print(f"[StreamChat] 📤 发送任务ID信息: {task_ids}")
|
||||
await websocket.send_json({
|
||||
"type": "tasks_created",
|
||||
"tasks": task_ids
|
||||
})
|
||||
|
||||
# ✅ 第3步:调用LLM流式生成
|
||||
chunk_count = [0] # 使用列表以便在闭包中修改
|
||||
|
||||
chunk_count = [0]
|
||||
|
||||
async def on_worldbook_active(entries):
|
||||
if entries:
|
||||
print(f"[StreamChat] 📤 发送世界书激活信息: {len(entries)} 个条目")
|
||||
await websocket.send_json({
|
||||
"type": "worldbook_active",
|
||||
"entries": entries,
|
||||
})
|
||||
|
||||
async def on_tasks_created(task_ids):
|
||||
if task_ids.get("imageWorkflow") or task_ids.get("dynamicTable"):
|
||||
print(f"[StreamChat] 📤 发送任务ID信息: {task_ids}")
|
||||
await websocket.send_json({
|
||||
"type": "tasks_created",
|
||||
"tasks": task_ids,
|
||||
})
|
||||
|
||||
async def on_chunk(chunk):
|
||||
chunk_count[0] += 1
|
||||
if chunk_count[0] % 10 == 0:
|
||||
print(f"[StreamChat] 📤 已发送 {chunk_count[0]} 个 chunks")
|
||||
await websocket.send_json({"type": "chunk", "content": chunk})
|
||||
|
||||
result = await workflow_service.process_chat_request_stream(
|
||||
request_data,
|
||||
on_chunk=lambda chunk: asyncio.create_task(
|
||||
_send_chunk_with_log(websocket, chunk, chunk_count)
|
||||
)
|
||||
on_chunk=on_chunk,
|
||||
on_worldbook_active=on_worldbook_active,
|
||||
on_tasks_created=on_tasks_created,
|
||||
)
|
||||
|
||||
|
||||
if result["success"]:
|
||||
content = result["content"]
|
||||
|
||||
print(f"\n[StreamChat] ✨ 流式生成成功,总长度: {len(content)}")
|
||||
|
||||
# 发送完成信号
|
||||
print(f"[StreamChat] ✅ 发送完成信号")
|
||||
await websocket.send_json({
|
||||
"type": "complete"
|
||||
})
|
||||
|
||||
# 保存消息
|
||||
await websocket.send_json({"type": "complete"})
|
||||
print(f"[StreamChat] 💾 保存消息到文件...")
|
||||
await _save_messages(role_name, chat_name, request_data, content)
|
||||
print(f"[StreamChat] ✅ 消息保存完成\n")
|
||||
@@ -342,35 +270,19 @@ async def _handle_stream_chat(
|
||||
print(f"[StreamChat] ❌ 流式处理失败: {error_msg}")
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"message": error_msg
|
||||
"message": error_msg,
|
||||
})
|
||||
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n[StreamChat] ⚠️ 错误: {str(e)}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
await websocket.send_json({
|
||||
"type": "error",
|
||||
"message": f"流式处理失败: {str(e)}"
|
||||
"message": f"流式处理失败: {str(e)}",
|
||||
})
|
||||
|
||||
|
||||
async def _send_chunk_with_log(websocket: WebSocket, chunk: str, chunk_count: list):
|
||||
"""
|
||||
发送 chunk 并记录日志
|
||||
|
||||
Args:
|
||||
websocket: WebSocket 连接
|
||||
chunk: 文本片段
|
||||
chunk_count: 计数器(使用列表以便在闭包中修改)
|
||||
"""
|
||||
chunk_count[0] += 1
|
||||
if chunk_count[0] % 10 == 0: # 每10个chunk记录一次
|
||||
print(f"[StreamChat] 📤 已发送 {chunk_count[0]} 个 chunks")
|
||||
|
||||
await websocket.send_json({"type": "chunk", "content": chunk})
|
||||
|
||||
|
||||
async def _save_messages(
|
||||
role_name: str,
|
||||
chat_name: str,
|
||||
|
||||
280
backend/api/routes/studioRoute.py
Normal file
280
backend/api/routes/studioRoute.py
Normal file
@@ -0,0 +1,280 @@
|
||||
import json
|
||||
import logging
|
||||
from typing import Any, Dict, List
|
||||
|
||||
from fastapi import APIRouter, HTTPException
|
||||
from fastapi.responses import StreamingResponse
|
||||
|
||||
from models.studio_models import (
|
||||
AdvanceRunRequest,
|
||||
CreateStudioProjectRequest,
|
||||
PipelineDefinition,
|
||||
RenameRunRequest,
|
||||
RunMessageRequest,
|
||||
StudioProject,
|
||||
StudioProjectSummary,
|
||||
StudioRun,
|
||||
StudioRunSummary,
|
||||
UpdateStudioProjectRequest,
|
||||
WorkflowTemplateSummary,
|
||||
WorkflowVariablesResponse,
|
||||
)
|
||||
from services.studio_project_service import studio_project_service
|
||||
from services.studio_run_service import studio_run_service
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
router = APIRouter(prefix="/studio", tags=["studio"])
|
||||
|
||||
|
||||
@router.get("/projects", response_model=List[StudioProjectSummary])
|
||||
async def list_studio_projects():
|
||||
try:
|
||||
return studio_project_service.list_projects()
|
||||
except Exception as e:
|
||||
logger.error("Failed to list studio projects: %s", e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/projects/{project_id}", response_model=StudioProject)
|
||||
async def get_studio_project(project_id: str):
|
||||
try:
|
||||
return studio_project_service.get_project(project_id)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to get studio project %s: %s", project_id, e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.patch("/projects/{project_id}", response_model=StudioProject)
|
||||
async def update_studio_project(project_id: str, req: UpdateStudioProjectRequest):
|
||||
try:
|
||||
return studio_project_service.update_project_meta(
|
||||
project_id,
|
||||
name=req.name,
|
||||
description=req.description,
|
||||
)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to update studio project %s: %s", project_id, e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.put("/projects/{project_id}/pipeline", response_model=StudioProject)
|
||||
async def save_studio_pipeline(project_id: str, pipeline: PipelineDefinition):
|
||||
try:
|
||||
return studio_project_service.save_pipeline(project_id, pipeline)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to save pipeline for %s: %s", project_id, e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/templates", response_model=List[WorkflowTemplateSummary])
|
||||
async def list_workflow_templates():
|
||||
try:
|
||||
return studio_project_service.list_workflow_templates()
|
||||
except Exception as e:
|
||||
logger.error("Failed to list workflow templates: %s", e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/variables", response_model=WorkflowVariablesResponse)
|
||||
async def get_workflow_variables(projectId: str | None = None):
|
||||
try:
|
||||
return studio_project_service.get_workflow_variables(projectId)
|
||||
except Exception as e:
|
||||
logger.error("Failed to load workflow variables: %s", e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/skill-templates")
|
||||
async def get_skill_templates() -> Dict[str, Any]:
|
||||
try:
|
||||
return studio_project_service.get_skill_templates()
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to load skill templates: %s", e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/niches")
|
||||
async def get_niches() -> Dict[str, Any]:
|
||||
try:
|
||||
return studio_project_service.get_niches()
|
||||
except Exception as e:
|
||||
logger.error("Failed to load niches: %s", e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.delete("/projects/{project_id}")
|
||||
async def delete_studio_project(project_id: str):
|
||||
try:
|
||||
studio_project_service.delete_project(project_id)
|
||||
return {"ok": True, "id": project_id}
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to delete studio project %s: %s", project_id, e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.post("/projects", response_model=StudioProject)
|
||||
async def create_studio_project(req: CreateStudioProjectRequest):
|
||||
try:
|
||||
return studio_project_service.create_project(req)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to create studio project: %s", e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.post("/projects/{project_id}/runs", response_model=StudioRun)
|
||||
async def create_studio_run(project_id: str):
|
||||
try:
|
||||
return studio_run_service.create_run(project_id)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to create studio run for %s: %s", project_id, e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/projects/{project_id}/runs", response_model=List[StudioRunSummary])
|
||||
async def list_studio_runs(project_id: str):
|
||||
try:
|
||||
studio_project_service.get_project(project_id)
|
||||
return studio_run_service.list_runs(project_id)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to list studio runs for %s: %s", project_id, e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/projects/{project_id}/runs/{run_id}", response_model=StudioRun)
|
||||
async def get_studio_run(project_id: str, run_id: str):
|
||||
try:
|
||||
return studio_run_service.get_run(project_id, run_id)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error("Failed to get studio run %s/%s: %s", project_id, run_id, e)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.post("/projects/{project_id}/runs/{run_id}/advance", response_model=StudioRun)
|
||||
async def advance_studio_run(
|
||||
project_id: str, run_id: str, req: AdvanceRunRequest
|
||||
):
|
||||
try:
|
||||
return studio_run_service.advance_run(
|
||||
project_id, run_id, display_params=req.displayParams
|
||||
)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except NotImplementedError as e:
|
||||
raise HTTPException(status_code=501, detail=str(e))
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
"Failed to advance studio run %s/%s: %s", project_id, run_id, e
|
||||
)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.post("/projects/{project_id}/runs/{run_id}/message")
|
||||
async def send_studio_run_message(
|
||||
project_id: str, run_id: str, req: RunMessageRequest
|
||||
):
|
||||
if req.stream:
|
||||
async def ndjson_stream():
|
||||
try:
|
||||
async for event in studio_run_service.send_run_message_stream(
|
||||
project_id,
|
||||
run_id,
|
||||
req.content,
|
||||
profile_id=req.profileId,
|
||||
api_config=req.apiConfig,
|
||||
):
|
||||
yield json.dumps(event, ensure_ascii=False) + "\n"
|
||||
except FileNotFoundError as e:
|
||||
yield json.dumps(
|
||||
{"type": "error", "detail": str(e)}, ensure_ascii=False
|
||||
) + "\n"
|
||||
except ValueError as e:
|
||||
yield json.dumps(
|
||||
{"type": "error", "detail": str(e)}, ensure_ascii=False
|
||||
) + "\n"
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
"Failed to stream studio run message %s/%s: %s",
|
||||
project_id,
|
||||
run_id,
|
||||
e,
|
||||
)
|
||||
yield json.dumps(
|
||||
{"type": "error", "detail": f"消息处理失败:{e}"},
|
||||
ensure_ascii=False,
|
||||
) + "\n"
|
||||
|
||||
return StreamingResponse(
|
||||
ndjson_stream(),
|
||||
media_type="application/x-ndjson",
|
||||
)
|
||||
|
||||
try:
|
||||
return await studio_run_service.send_run_message(
|
||||
project_id,
|
||||
run_id,
|
||||
req.content,
|
||||
stream=req.stream,
|
||||
profile_id=req.profileId,
|
||||
api_config=req.apiConfig,
|
||||
)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
"Failed to send studio run message %s/%s: %s", project_id, run_id, e
|
||||
)
|
||||
raise HTTPException(status_code=500, detail=f"消息处理失败:{e}")
|
||||
|
||||
|
||||
@router.delete("/projects/{project_id}/runs/{run_id}")
|
||||
async def delete_studio_run(project_id: str, run_id: str):
|
||||
try:
|
||||
studio_run_service.delete_run(project_id, run_id)
|
||||
return {"ok": True, "id": run_id}
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
"Failed to delete studio run %s/%s: %s", project_id, run_id, e
|
||||
)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.patch("/projects/{project_id}/runs/{run_id}", response_model=StudioRun)
|
||||
async def rename_studio_run(
|
||||
project_id: str, run_id: str, req: RenameRunRequest
|
||||
):
|
||||
try:
|
||||
return studio_run_service.rename_run(project_id, run_id, req.title)
|
||||
except FileNotFoundError as e:
|
||||
raise HTTPException(status_code=404, detail=str(e))
|
||||
except ValueError as e:
|
||||
raise HTTPException(status_code=400, detail=str(e))
|
||||
except Exception as e:
|
||||
logger.error(
|
||||
"Failed to rename studio run %s/%s: %s", project_id, run_id, e
|
||||
)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
@@ -61,6 +61,15 @@ class Settings:
|
||||
# 图片资源目录
|
||||
IMAGES_PATH = DATA_PATH / "images"
|
||||
|
||||
# Agent 工作流模板与运行记录
|
||||
AGENT_TEMPLATES_PATH = DATA_PATH / "agent" / "templates"
|
||||
AGENT_RUNS_PATH = DATA_PATH / "agent" / "runs"
|
||||
AGENT_STUDIO_PROJECTS_PATH = DATA_PATH / "agent" / "studio_projects"
|
||||
AGENT_STUDIO_RUNS_PATH = DATA_PATH / "agent" / "studio_runs"
|
||||
AGENT_SKILL_TEMPLATES_FILE = DATA_PATH / "agent" / "skill_templates.json"
|
||||
AGENT_NICHES_FILE = DATA_PATH / "agent" / "niches.json"
|
||||
AGENT_WORKFLOW_VARIABLES_FILE = DATA_PATH / "agent" / "workflow_variables.json"
|
||||
|
||||
def ensure_directories(self):
|
||||
"""确保所有配置的目录存在,如果不存在则创建"""
|
||||
directories = [
|
||||
@@ -72,6 +81,10 @@ class Settings:
|
||||
self.COMFYUI_WORKFLOWS_PATH,
|
||||
self.CHARACTERS_PATH,
|
||||
self.IMAGES_PATH,
|
||||
self.AGENT_TEMPLATES_PATH,
|
||||
self.AGENT_RUNS_PATH,
|
||||
self.AGENT_STUDIO_PROJECTS_PATH,
|
||||
self.AGENT_STUDIO_RUNS_PATH,
|
||||
]
|
||||
for directory in directories:
|
||||
directory.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
128
backend/models/agent.py
Normal file
128
backend/models/agent.py
Normal file
@@ -0,0 +1,128 @@
|
||||
"""
|
||||
Agent workflow engine data models.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from datetime import datetime
|
||||
from enum import Enum
|
||||
from typing import Any, Awaitable, Callable, Dict, List, Optional
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
|
||||
class WorkflowTemplateKind(str, Enum):
|
||||
BUILTIN_CHAT = "builtin.chat"
|
||||
|
||||
|
||||
class RunStatus(str, Enum):
|
||||
PENDING = "pending"
|
||||
RUNNING = "running"
|
||||
COMPLETED = "completed"
|
||||
FAILED = "failed"
|
||||
|
||||
|
||||
class RunEventType(str, Enum):
|
||||
STATE_ENTER = "state_enter"
|
||||
TOOL_START = "tool_start"
|
||||
TOOL_END = "tool_end"
|
||||
WORLD_BOOK_ACTIVE = "worldbook_active"
|
||||
TASKS_CREATED = "tasks_created"
|
||||
CHUNK = "chunk"
|
||||
ERROR = "error"
|
||||
COMPLETE = "complete"
|
||||
|
||||
|
||||
class ToolSpec(BaseModel):
|
||||
name: str
|
||||
description: str = ""
|
||||
parameters: Dict[str, Any] = Field(default_factory=dict)
|
||||
|
||||
|
||||
class SkillManifest(BaseModel):
|
||||
id: str
|
||||
name: str = ""
|
||||
description: str = ""
|
||||
path: str = ""
|
||||
|
||||
|
||||
class WorkflowTemplate(BaseModel):
|
||||
id: str
|
||||
kind: WorkflowTemplateKind
|
||||
name: str = ""
|
||||
description: str = ""
|
||||
version: str = "1.0.0"
|
||||
state_machine_path: str = "state_machine.json"
|
||||
skills: List[SkillManifest] = Field(default_factory=list)
|
||||
|
||||
|
||||
class ChatRunBinding(BaseModel):
|
||||
role_name: str
|
||||
chat_name: str
|
||||
template_id: str = WorkflowTemplateKind.BUILTIN_CHAT.value
|
||||
|
||||
|
||||
class TurnCallbacks(BaseModel):
|
||||
"""Optional async callbacks for streaming / WS events."""
|
||||
|
||||
model_config = {"arbitrary_types_allowed": True}
|
||||
|
||||
on_worldbook_active: Optional[Callable[[List[Any]], Awaitable[None]]] = None
|
||||
on_tasks_created: Optional[Callable[[Dict[str, Any]], Awaitable[None]]] = None
|
||||
on_chunk: Optional[Callable[[str], Awaitable[None]]] = None
|
||||
|
||||
|
||||
class TurnContext(BaseModel):
|
||||
"""Mutable per-turn execution context passed between tools."""
|
||||
|
||||
model_config = {"arbitrary_types_allowed": True}
|
||||
|
||||
request_data: Dict[str, Any] = Field(default_factory=dict)
|
||||
template_id: str = WorkflowTemplateKind.BUILTIN_CHAT.value
|
||||
run_id: str = ""
|
||||
stream: bool = False
|
||||
callbacks: Optional[TurnCallbacks] = None
|
||||
|
||||
current_role: str = ""
|
||||
current_chat: str = ""
|
||||
user_message: str = ""
|
||||
preset_name: Optional[str] = None
|
||||
character: Any = None
|
||||
active_entries: List[Any] = Field(default_factory=list)
|
||||
chat_history: List[Any] = Field(default_factory=list)
|
||||
prompt_messages: List[Any] = Field(default_factory=list)
|
||||
generated_content: str = ""
|
||||
token_usage: Dict[str, Any] = Field(default_factory=dict)
|
||||
duration: float = 0.0
|
||||
task_ids: Dict[str, Optional[str]] = Field(default_factory=dict)
|
||||
error: Optional[str] = None
|
||||
|
||||
|
||||
class WorkflowRun(BaseModel):
|
||||
id: str
|
||||
template_id: str
|
||||
binding: ChatRunBinding
|
||||
status: RunStatus = RunStatus.PENDING
|
||||
started_at: str = Field(default_factory=lambda: datetime.now().isoformat())
|
||||
finished_at: Optional[str] = None
|
||||
current_state: Optional[str] = None
|
||||
result_content: str = ""
|
||||
error: Optional[str] = None
|
||||
|
||||
|
||||
class RunEvent(BaseModel):
|
||||
run_id: str
|
||||
type: RunEventType
|
||||
timestamp: str = Field(default_factory=lambda: datetime.now().isoformat())
|
||||
state: Optional[str] = None
|
||||
tool: Optional[str] = None
|
||||
payload: Dict[str, Any] = Field(default_factory=dict)
|
||||
|
||||
|
||||
class ChatTurnResult(BaseModel):
|
||||
success: bool
|
||||
content: str = ""
|
||||
error: Optional[str] = None
|
||||
active_entries: List[Any] = Field(default_factory=list)
|
||||
task_ids: Dict[str, Optional[str]] = Field(default_factory=dict)
|
||||
run_id: str = ""
|
||||
workflow_template_id: str = WorkflowTemplateKind.BUILTIN_CHAT.value
|
||||
@@ -215,6 +215,10 @@ class ChatHeader(BaseModel):
|
||||
messageCount: int = Field(0, description="消息数量")
|
||||
ragLibraryId: Optional[str] = Field(None, description="关联的 RAG 历史消息库 ID")
|
||||
|
||||
# Agent workflow engine (optional, backward compatible)
|
||||
workflowTemplateId: Optional[str] = Field(None, description="工作流模板 ID")
|
||||
engineRunId: Optional[str] = Field(None, description="最近一次引擎运行 ID")
|
||||
|
||||
|
||||
class ChatMessage(BaseModel):
|
||||
"""
|
||||
|
||||
234
backend/models/studio_models.py
Normal file
234
backend/models/studio_models.py
Normal file
@@ -0,0 +1,234 @@
|
||||
"""
|
||||
Studio workflow editor data models.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from enum import Enum
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
|
||||
|
||||
class DisplayParam(BaseModel):
|
||||
key: str
|
||||
label: str
|
||||
type: str = "text"
|
||||
required: bool = True
|
||||
placeholder: str = ""
|
||||
|
||||
|
||||
class InputRef(BaseModel):
|
||||
ref: str
|
||||
label: Optional[str] = None
|
||||
optional: bool = False
|
||||
|
||||
|
||||
class ScoringDimension(BaseModel):
|
||||
id: str
|
||||
name: str
|
||||
criteria: str = ""
|
||||
|
||||
|
||||
class InsertionRagConfig(BaseModel):
|
||||
libraryId: str = ""
|
||||
threshold: float = 0.5
|
||||
maxEntries: int = 3
|
||||
|
||||
|
||||
class InsertionConfig(BaseModel):
|
||||
position: int = 1
|
||||
activationType: str = "permanent"
|
||||
key: str = ""
|
||||
keysecondary: str = ""
|
||||
comment: str = ""
|
||||
ragConfig: Optional[InsertionRagConfig] = None
|
||||
|
||||
|
||||
class ScoringConfig(BaseModel):
|
||||
enabled: bool = True
|
||||
dimensions: List[ScoringDimension] = Field(default_factory=list)
|
||||
rubric: Optional[str] = None
|
||||
|
||||
|
||||
class StudioNode(BaseModel):
|
||||
id: str
|
||||
skillId: str
|
||||
displayName: str
|
||||
enabled: bool = True
|
||||
niche: Optional[str] = None
|
||||
loopUntilSatisfied: bool = False
|
||||
config: Dict[str, Any] = Field(default_factory=dict)
|
||||
displayParams: List[DisplayParam] = Field(default_factory=list)
|
||||
inputs: List[InputRef] = Field(default_factory=list)
|
||||
|
||||
|
||||
class PipelineDefinition(BaseModel):
|
||||
workflowGoal: str = ""
|
||||
nodes: List[StudioNode] = Field(default_factory=list)
|
||||
|
||||
|
||||
class StudioProjectMeta(BaseModel):
|
||||
id: str
|
||||
name: str
|
||||
description: str = ""
|
||||
templateId: Optional[str] = None
|
||||
characterId: Optional[str] = None
|
||||
worldbookId: Optional[str] = None
|
||||
createdAt: str = ""
|
||||
updatedAt: str = ""
|
||||
|
||||
|
||||
class StudioProject(BaseModel):
|
||||
meta: StudioProjectMeta
|
||||
pipeline: PipelineDefinition
|
||||
|
||||
|
||||
class ArtifactDef(BaseModel):
|
||||
type: str
|
||||
displayName: str = ""
|
||||
|
||||
|
||||
class SkillTemplateDef(BaseModel):
|
||||
skillId: str
|
||||
displayName: str
|
||||
description: str = ""
|
||||
displayParams: List[DisplayParam] = Field(default_factory=list)
|
||||
configWhitelist: List[str] = Field(default_factory=list)
|
||||
artifacts: List[ArtifactDef] = Field(default_factory=list)
|
||||
supportsLoopUntilSatisfied: bool = False
|
||||
supportsInputs: bool = False
|
||||
supportsInsertion: bool = False
|
||||
supportsScoring: bool = False
|
||||
|
||||
|
||||
class WorkflowTemplateSummary(BaseModel):
|
||||
id: str
|
||||
name: str
|
||||
description: str = ""
|
||||
|
||||
|
||||
class WorkflowVariableDef(BaseModel):
|
||||
ref: str
|
||||
label: str
|
||||
description: str = ""
|
||||
|
||||
|
||||
class DynamicVariableSuffix(BaseModel):
|
||||
suffix: str
|
||||
labelPattern: str
|
||||
|
||||
|
||||
class WorkflowVariablesResponse(BaseModel):
|
||||
builtIn: List[WorkflowVariableDef] = Field(default_factory=list)
|
||||
dynamic: List[WorkflowVariableDef] = Field(default_factory=list)
|
||||
|
||||
|
||||
class SkillTemplatesCatalog(BaseModel):
|
||||
templates: List[SkillTemplateDef] = Field(default_factory=list)
|
||||
|
||||
|
||||
class StudioProjectSummary(BaseModel):
|
||||
id: str
|
||||
name: str
|
||||
description: str = ""
|
||||
updatedAt: str = ""
|
||||
|
||||
|
||||
class CreateStudioProjectRequest(BaseModel):
|
||||
name: str = "新项目"
|
||||
template_id: str = "builtin.studio.example"
|
||||
project_id: Optional[str] = None
|
||||
|
||||
|
||||
class UpdateStudioProjectRequest(BaseModel):
|
||||
name: Optional[str] = Field(None, min_length=1, max_length=120)
|
||||
description: Optional[str] = Field(None, max_length=500)
|
||||
|
||||
|
||||
class StudioRunStatus(str, Enum):
|
||||
PENDING = "pending"
|
||||
RUNNING = "running"
|
||||
PAUSED = "paused"
|
||||
COMPLETED = "completed"
|
||||
FAILED = "failed"
|
||||
CANCELLED = "cancelled"
|
||||
|
||||
|
||||
class ToolQuestionOption(BaseModel):
|
||||
question: str
|
||||
options: List[str] = Field(default_factory=list)
|
||||
|
||||
|
||||
class StepMessage(BaseModel):
|
||||
"""Short step-scoped dialogue (not full chat history)."""
|
||||
id: str
|
||||
role: str # user | assistant
|
||||
content: str
|
||||
createdAt: Optional[str] = None
|
||||
|
||||
|
||||
class LastToolResponse(BaseModel):
|
||||
"""LLM tool-call payload surfaced to the run UI (R2+)."""
|
||||
thinking: Optional[str] = None
|
||||
evaluation: Optional[str] = None
|
||||
questions: List[ToolQuestionOption] = Field(default_factory=list)
|
||||
generatedAt: Optional[str] = None
|
||||
|
||||
|
||||
class PromptBlock(BaseModel):
|
||||
"""Single assembled context section for LLM prompt (R2 debug / execution)."""
|
||||
id: str
|
||||
label: str
|
||||
content: str
|
||||
source: str = "auto" # auto | manual | workflow
|
||||
|
||||
|
||||
class StudioNodeRunState(BaseModel):
|
||||
nodeId: str
|
||||
displayName: str
|
||||
skillId: str
|
||||
status: str # pending | active | completed | skipped
|
||||
loopUntilSatisfied: bool = False
|
||||
lastDraft: Optional[Dict[str, Any]] = None
|
||||
lastToolResponse: Optional[LastToolResponse] = None
|
||||
stepMessages: List[StepMessage] = Field(default_factory=list)
|
||||
|
||||
|
||||
class StudioRun(BaseModel):
|
||||
id: str
|
||||
projectId: str
|
||||
status: StudioRunStatus
|
||||
pipelineSnapshot: PipelineDefinition
|
||||
pipelineVersionNote: str
|
||||
currentNodeId: Optional[str] = None
|
||||
nodeStates: List[StudioNodeRunState] = Field(default_factory=list)
|
||||
workflowVariables: Dict[str, Any] = Field(default_factory=dict)
|
||||
lastPromptBlocks: List[PromptBlock] = Field(default_factory=list)
|
||||
title: str = ""
|
||||
createdAt: str = ""
|
||||
updatedAt: str = ""
|
||||
|
||||
|
||||
class AdvanceRunRequest(BaseModel):
|
||||
displayParams: Dict[str, str] = Field(default_factory=dict)
|
||||
|
||||
|
||||
class RunMessageRequest(BaseModel):
|
||||
content: str = Field(..., min_length=1, max_length=32000)
|
||||
stream: bool = False
|
||||
profileId: Optional[str] = None
|
||||
apiConfig: Optional[Dict[str, str]] = None
|
||||
|
||||
|
||||
class RenameRunRequest(BaseModel):
|
||||
title: str = Field(..., min_length=1, max_length=120)
|
||||
|
||||
|
||||
class StudioRunSummary(BaseModel):
|
||||
id: str
|
||||
projectId: str
|
||||
status: StudioRunStatus
|
||||
currentNodeId: Optional[str] = None
|
||||
title: str = ""
|
||||
createdAt: str = ""
|
||||
updatedAt: str = ""
|
||||
@@ -17,6 +17,7 @@ try:
|
||||
from backend.utils.llm_client import LLMClient
|
||||
from backend.services.task_queue_manager import task_queue_manager, TaskType
|
||||
from backend.core.config import settings
|
||||
from backend.services.workflow_engine import workflow_engine
|
||||
except ImportError:
|
||||
from services.character_service import CharacterService
|
||||
from services.worldbook_service import WorldBookService
|
||||
@@ -24,6 +25,7 @@ except ImportError:
|
||||
from utils.llm_client import LLMClient
|
||||
from services.task_queue_manager import task_queue_manager, TaskType
|
||||
from core.config import settings
|
||||
from services.workflow_engine import workflow_engine
|
||||
|
||||
|
||||
class ChatWorkflowService:
|
||||
@@ -128,201 +130,17 @@ class ChatWorkflowService:
|
||||
self,
|
||||
request_data: Dict[str, Any]
|
||||
) -> Dict[str, Any]:
|
||||
"""
|
||||
处理聊天请求的核心工作流
|
||||
|
||||
Args:
|
||||
request_data: 前端发送的完整数据
|
||||
|
||||
Returns:
|
||||
{
|
||||
"success": bool,
|
||||
"content": str, # 生成的回复内容
|
||||
"error": str | None
|
||||
}
|
||||
"""
|
||||
try:
|
||||
# === 第1步:解析请求数据 ===
|
||||
current_role = request_data.get("currentRole")
|
||||
current_chat = request_data.get("currentChat")
|
||||
user_message = request_data.get("mes", "")
|
||||
|
||||
if not current_role or not user_message:
|
||||
return {
|
||||
"success": False,
|
||||
"content": "",
|
||||
"error": "缺少必要的参数:currentRole 或 mes"
|
||||
}
|
||||
|
||||
print(f"[ChatWorkflow] 开始处理请求: role={current_role}, chat={current_chat}")
|
||||
|
||||
# ✅ 获取预设名称(用于加载预设绑定的正则规则)
|
||||
preset_config = request_data.get("presetConfig", {})
|
||||
preset_name = preset_config.get("selectedPreset")
|
||||
|
||||
# ✅ 第1.5步:应用用户输入的正则规则
|
||||
from services.regex_service import regex_service
|
||||
from models.regex_rules import RegexPlacement
|
||||
|
||||
processed_user_message = regex_service.apply_rules_by_placement(
|
||||
text=user_message,
|
||||
placement=RegexPlacement.USER_INPUT.value,
|
||||
character_name=current_role,
|
||||
preset_name=preset_name,
|
||||
message_depth=0,
|
||||
is_for_llm=True, # ✅ 用户输入会发送给LLM
|
||||
is_markdown_rendered=False
|
||||
)
|
||||
|
||||
if processed_user_message != user_message:
|
||||
print(f"[Regex] ✅ 已应用用户输入正则规则")
|
||||
|
||||
user_message = processed_user_message
|
||||
|
||||
# === 第2步:加载角色卡 ===
|
||||
character_data = request_data.get("characterData")
|
||||
if not character_data:
|
||||
# 如果没有提供角色卡数据,从后端加载
|
||||
character = self.character_service.get_character_by_name(current_role)
|
||||
if not character:
|
||||
return {
|
||||
"success": False,
|
||||
"content": "",
|
||||
"error": f"角色 '{current_role}' 不存在"
|
||||
}
|
||||
else:
|
||||
# 使用前端提供的角色卡数据(可能包含用户修改)
|
||||
try:
|
||||
from backend.models.internal import CharacterCard
|
||||
except ImportError:
|
||||
from models.internal import CharacterCard
|
||||
character = CharacterCard(**character_data)
|
||||
|
||||
print(f"[ChatWorkflow] 已加载角色卡: {character.name}")
|
||||
|
||||
# === 第3步:收集并激活世界书条目 ===
|
||||
active_entries = await self._collect_and_activate_worldbooks(
|
||||
request_data,
|
||||
character
|
||||
)
|
||||
print(f"[ChatWorkflow] 激活了 {len(active_entries)} 个世界书条目")
|
||||
|
||||
# === 第4步:加载聊天历史 ===
|
||||
chat_history = await self._load_chat_history(current_role, current_chat)
|
||||
print(f"[ChatWorkflow] 加载了 {len(chat_history)} 条历史消息")
|
||||
|
||||
# === 第5步:组装提示词 ===
|
||||
prompt_messages = self._assemble_prompt(
|
||||
character,
|
||||
chat_history,
|
||||
user_message,
|
||||
active_entries,
|
||||
request_data
|
||||
)
|
||||
print(f"[ChatWorkflow] 组装了 {len(prompt_messages)} 条提示消息")
|
||||
|
||||
# === 第6步:调用LLM生成回复 ===
|
||||
api_config = request_data.get("apiConfig", {})
|
||||
preset_config = request_data.get("presetConfig", {})
|
||||
stream_output = request_data.get("stream", False)
|
||||
|
||||
result = await self._generate_response(
|
||||
prompt_messages,
|
||||
api_config,
|
||||
preset_config,
|
||||
stream_output
|
||||
)
|
||||
|
||||
generated_content = result["content"]
|
||||
token_usage = result.get("usage", {})
|
||||
duration = result.get("duration")
|
||||
|
||||
print(f"[ChatWorkflow] 生成完成,内容长度: {len(generated_content)}")
|
||||
print(f"[ChatWorkflow] Token 使用: {token_usage}")
|
||||
|
||||
# ✅ 第6.5步:应用 AI 输出的正则规则
|
||||
processed_ai_output = regex_service.apply_rules_by_placement(
|
||||
text=generated_content,
|
||||
placement=RegexPlacement.AI_OUTPUT.value,
|
||||
character_name=current_role,
|
||||
preset_name=preset_name,
|
||||
message_depth=0,
|
||||
is_for_llm=False, # ✅ AI输出是显示给用户的
|
||||
is_markdown_rendered=False
|
||||
)
|
||||
|
||||
if processed_ai_output != generated_content:
|
||||
print(f"[Regex] ✅ 已应用 AI 输出正则规则")
|
||||
generated_content = processed_ai_output
|
||||
|
||||
# === 第7步:记录 Token 使用 ===
|
||||
chat_id = f"{current_role}/{request_data.get('currentChat', '')}"
|
||||
floor = request_data.get("floor", 0)
|
||||
|
||||
try:
|
||||
try:
|
||||
from backend.services.token_usage_service import token_usage_service
|
||||
from backend.models.internal import TokenUsageStatus
|
||||
except ImportError:
|
||||
from services.token_usage_service import token_usage_service
|
||||
from models.internal import TokenUsageStatus
|
||||
|
||||
await token_usage_service.record_usage(
|
||||
chat_id=chat_id,
|
||||
role_name=current_role,
|
||||
chat_name=request_data.get('currentChat', ''),
|
||||
prompt_tokens=token_usage.get("prompt_tokens", 0),
|
||||
completion_tokens=token_usage.get("completion_tokens", 0),
|
||||
total_tokens=token_usage.get("total_tokens", 0),
|
||||
status=TokenUsageStatus.COMPLETED,
|
||||
floor=floor + 1, # AI 回复的楼层
|
||||
duration=duration,
|
||||
model=api_config.get("model"),
|
||||
api_provider="openai", # TODO: 从 API URL 检测提供商
|
||||
api_url=api_config.get("api_url") # ✅ 记录 API URL
|
||||
)
|
||||
except Exception as e:
|
||||
print(f"[ChatWorkflow] 记录 Token 使用失败: {e}")
|
||||
|
||||
# === 第8步:启动异步并行任务 ===
|
||||
|
||||
# 创建任务ID
|
||||
image_task_id = None
|
||||
table_task_id = None
|
||||
|
||||
options = request_data.get("options", {})
|
||||
if options.get("imageWorkflow", False):
|
||||
import uuid
|
||||
image_task_id = f"img_{uuid.uuid4().hex[:8]}"
|
||||
await task_queue_manager.add_task(image_task_id, TaskType.IMAGE_WORKFLOW, chat_id)
|
||||
|
||||
if options.get("dynamicTable", False):
|
||||
import uuid
|
||||
table_task_id = f"tbl_{uuid.uuid4().hex[:8]}"
|
||||
await task_queue_manager.add_task(table_task_id, TaskType.DYNAMIC_TABLE, chat_id)
|
||||
|
||||
await self._start_parallel_tasks(request_data, generated_content, image_task_id, table_task_id)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"content": generated_content,
|
||||
"error": None,
|
||||
"activeEntries": active_entries,
|
||||
"taskIds": {
|
||||
"imageWorkflow": image_task_id,
|
||||
"dynamicTable": table_task_id
|
||||
}
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
print(f"[ChatWorkflow] 错误: {str(e)}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return {
|
||||
"success": False,
|
||||
"content": "",
|
||||
"error": f"工作流执行失败: {str(e)}"
|
||||
}
|
||||
"""处理聊天请求 – delegates to WorkflowEngine.run_turn."""
|
||||
result = await workflow_engine.run_turn(request_data, stream=False)
|
||||
return {
|
||||
"success": result.success,
|
||||
"content": result.content,
|
||||
"error": result.error,
|
||||
"activeEntries": result.active_entries,
|
||||
"taskIds": result.task_ids,
|
||||
"engineRunId": result.run_id,
|
||||
"workflowTemplateId": result.workflow_template_id,
|
||||
}
|
||||
|
||||
def _normalize_worldbook_entry(self, entry_data: Dict[str, Any]) -> Dict[str, Any]:
|
||||
"""
|
||||
@@ -1211,280 +1029,24 @@ class ChatWorkflowService:
|
||||
async def process_chat_request_stream(
|
||||
self,
|
||||
request_data: Dict[str, Any],
|
||||
on_chunk
|
||||
on_chunk,
|
||||
on_worldbook_active=None,
|
||||
on_tasks_created=None,
|
||||
) -> Dict[str, Any]:
|
||||
"""
|
||||
处理流式聊天请求
|
||||
|
||||
Args:
|
||||
request_data: 前端发送的完整数据
|
||||
on_chunk: 回调函数,每次收到 chunk 时调用
|
||||
|
||||
Returns:
|
||||
{
|
||||
"success": bool,
|
||||
"content": str, # 完整的生成内容
|
||||
"error": str | None,
|
||||
"activeEntries": List,
|
||||
"taskIds": Dict
|
||||
}
|
||||
"""
|
||||
try:
|
||||
# === 第1步:解析请求数据 ===
|
||||
current_role = request_data.get("currentRole")
|
||||
current_chat = request_data.get("currentChat")
|
||||
user_message = request_data.get("mes", "")
|
||||
|
||||
if not current_role or not user_message:
|
||||
return {
|
||||
"success": False,
|
||||
"content": "",
|
||||
"error": "缺少必要的参数:currentRole 或 mes"
|
||||
}
|
||||
|
||||
print(f"\n{'#'*80}")
|
||||
print(f"[ChatWorkflow-Stream] 🚀 开始处理请求")
|
||||
print(f" - Role: {current_role}")
|
||||
print(f" - Chat: {current_chat}")
|
||||
print(f" - Message Length: {len(user_message)}")
|
||||
print(f"{'#'*80}\n")
|
||||
|
||||
# ✅ 获取预设名称(用于加载预设绑定的正则规则)
|
||||
preset_config = request_data.get("presetConfig", {})
|
||||
preset_name = preset_config.get("selectedPreset")
|
||||
|
||||
# ✅ 第1.5步:应用用户输入的正则规则
|
||||
from services.regex_service import regex_service
|
||||
from models.regex_rules import RegexPlacement
|
||||
|
||||
processed_user_message = regex_service.apply_rules_by_placement(
|
||||
text=user_message,
|
||||
placement=RegexPlacement.USER_INPUT.value,
|
||||
character_name=current_role,
|
||||
preset_name=preset_name,
|
||||
message_depth=0,
|
||||
is_for_llm=True, # ✅ 用户输入会发送给LLM
|
||||
is_markdown_rendered=False
|
||||
)
|
||||
|
||||
if processed_user_message != user_message:
|
||||
print(f"[Regex] ✅ 已应用用户输入正则规则")
|
||||
|
||||
user_message = processed_user_message
|
||||
|
||||
# === 第2步:加载角色卡 ===
|
||||
character_data = request_data.get("characterData")
|
||||
if not character_data:
|
||||
character = self.character_service.get_character_by_name(current_role)
|
||||
if not character:
|
||||
return {
|
||||
"success": False,
|
||||
"content": "",
|
||||
"error": f"角色 '{current_role}' 不存在"
|
||||
}
|
||||
else:
|
||||
try:
|
||||
from backend.models.internal import CharacterCard
|
||||
except ImportError:
|
||||
from models.internal import CharacterCard
|
||||
character = CharacterCard(**character_data)
|
||||
|
||||
print(f"[ChatWorkflow-Stream] ✅ 已加载角色卡: {character.name}")
|
||||
|
||||
# === 第3步:收集并激活世界书条目 ===
|
||||
active_entries = await self._collect_and_activate_worldbooks(
|
||||
request_data,
|
||||
character
|
||||
)
|
||||
print(f"\n[ChatWorkflow-Stream] 📚 世界书激活结果: {len(active_entries)} 个条目")
|
||||
for i, entry in enumerate(active_entries, 1):
|
||||
print(f" {i}. {getattr(entry, 'name', 'N/A')} (UID: {getattr(entry, 'uid', 'N/A')})")
|
||||
|
||||
# === 第4步:加载聊天历史 ===
|
||||
chat_history = await self._load_chat_history(current_role, current_chat)
|
||||
print(f"[ChatWorkflow-Stream] 💬 聊天历史加载结果: {len(chat_history)} 条消息")
|
||||
|
||||
# === 第5步:组装提示词 ===
|
||||
prompt_messages = self._assemble_prompt(
|
||||
character,
|
||||
chat_history,
|
||||
user_message,
|
||||
active_entries,
|
||||
request_data
|
||||
)
|
||||
print(f"[ChatWorkflow-Stream] 📝 提示词组装结果: {len(prompt_messages)} 条消息")
|
||||
for i, msg in enumerate(prompt_messages, 1):
|
||||
role = getattr(msg, 'role', 'unknown')
|
||||
content_preview = str(getattr(msg, 'content', ''))[:50]
|
||||
print(f" {i}. [{role}] {content_preview}...")
|
||||
|
||||
# === 第6步:流式调用LLM生成回复 ===
|
||||
api_config = request_data.get("apiConfig", {})
|
||||
preset_config = request_data.get("presetConfig", {})
|
||||
|
||||
print(f"\n[ChatWorkflow-Stream] 🤖 开始流式调用 LLM")
|
||||
print(f" - Model: {api_config.get('model', 'N/A')}")
|
||||
print(f" - API URL: {api_config.get('api_url', 'N/A')[:50]}...")
|
||||
print(f" - API Key: {'已设置' if api_config.get('api_key') else '⚠️ 未设置'}")
|
||||
print(f" - Temperature: {preset_config.get('parameters', {}).get('temperature', 1.0)}")
|
||||
print(f" - Max Tokens: {preset_config.get('parameters', {}).get('max_tokens', 30000)}")
|
||||
print(f" - Request Timeout: {preset_config.get('parameters', {}).get('request_timeout', 60)}s")
|
||||
print(f"{'~'*80}")
|
||||
|
||||
# ✅ 验证 API Key
|
||||
if not api_config.get('api_key'):
|
||||
print(f"[ChatWorkflow-Stream] ❌ 错误: API Key 为空!")
|
||||
print(f" - 请检查配置文件中是否保存了 API Key")
|
||||
print(f" - Profile ID: {request_data.get('currentProfile', {}).get('id', 'N/A')}")
|
||||
raise Exception("API Key 未配置,请先在 API 配置页面保存密钥")
|
||||
|
||||
generated_content = ""
|
||||
token_usage = {}
|
||||
duration = 0
|
||||
chunk_count = 0
|
||||
start_time = time.time()
|
||||
|
||||
print(f"[ChatWorkflow-Stream] 📡 正在调用 llm_client.stream_chat()...")
|
||||
|
||||
try:
|
||||
async for chunk_dict in self.llm_client.stream_chat(
|
||||
messages=prompt_messages,
|
||||
api_url=api_config.get("api_url", ""),
|
||||
api_key=api_config.get("api_key", ""),
|
||||
model=api_config.get("model", ""),
|
||||
temperature=preset_config.get("parameters", {}).get("temperature", 1.0),
|
||||
max_tokens=preset_config.get("parameters", {}).get("max_tokens", 30000),
|
||||
request_timeout=preset_config.get("parameters", {}).get("request_timeout", 60) # ✅ 传递超时时间
|
||||
):
|
||||
# ✅ 处理 LLM 返回的字典格式数据
|
||||
if isinstance(chunk_dict, dict):
|
||||
if chunk_dict.get("type") == "chunk":
|
||||
chunk_content = chunk_dict.get("content", "")
|
||||
elif chunk_dict.get("type") == "usage":
|
||||
# 跳过 usage 信息,不拼接到内容中
|
||||
continue
|
||||
else:
|
||||
# 兼容旧格式或未知类型,尝试直接获取 content
|
||||
chunk_content = chunk_dict.get("content", str(chunk_dict))
|
||||
else:
|
||||
# 如果直接返回字符串(兼容情况)
|
||||
chunk_content = str(chunk_dict)
|
||||
|
||||
# ✅ 拼接纯文本内容
|
||||
generated_content += chunk_content
|
||||
chunk_count += 1
|
||||
|
||||
# 第一个 chunk 到达时记录
|
||||
if chunk_count == 1:
|
||||
first_chunk_time = time.time()
|
||||
print(f"[ChatWorkflow-Stream] ✨ 收到第一个 chunk (耗时: {first_chunk_time - start_time:.2f}s)")
|
||||
|
||||
# 每20个chunk记录一次进度
|
||||
if chunk_count % 20 == 0:
|
||||
elapsed = time.time() - start_time
|
||||
print(f"[ChatWorkflow-Stream] 📊 已接收 {chunk_count} 个 chunks, 当前长度: {len(generated_content)}, 耗时: {elapsed:.2f}s")
|
||||
|
||||
# ✅ 调用回调函数发送纯文本 chunk
|
||||
await on_chunk(chunk_content)
|
||||
|
||||
except Exception as stream_error:
|
||||
elapsed = time.time() - start_time
|
||||
print(f"[ChatWorkflow-Stream] ❌ 流式调用失败 (耗时: {elapsed:.2f}s)")
|
||||
print(f" - 错误类型: {type(stream_error).__name__}")
|
||||
print(f" - 错误信息: {str(stream_error)}")
|
||||
raise
|
||||
|
||||
elapsed = time.time() - start_time
|
||||
print(f"{'~'*80}")
|
||||
print(f"[ChatWorkflow-Stream] ✅ LLM 流式调用完成")
|
||||
print(f" - 总 Chunks: {chunk_count}")
|
||||
print(f" - 内容长度: {len(generated_content)}")
|
||||
print(f" - 总耗时: {elapsed:.2f}s")
|
||||
print(f" - 平均速度: {len(generated_content)/elapsed if elapsed > 0 else 0:.0f} chars/s")
|
||||
print(f"{'#'*80}\n")
|
||||
|
||||
# ✅ 第6.5步:应用 AI 输出的正则规则
|
||||
processed_ai_output = regex_service.apply_rules_by_placement(
|
||||
text=generated_content,
|
||||
placement=RegexPlacement.AI_OUTPUT.value,
|
||||
character_name=current_role,
|
||||
preset_name=preset_name,
|
||||
message_depth=0,
|
||||
is_for_llm=False, # ✅ AI输出是显示给用户的
|
||||
is_markdown_rendered=False
|
||||
)
|
||||
|
||||
if processed_ai_output != generated_content:
|
||||
print(f"[Regex] ✅ 已应用 AI 输出正则规则")
|
||||
generated_content = processed_ai_output
|
||||
|
||||
# === 第7步:记录 Token 使用(估算) ===
|
||||
chat_id = f"{current_role}/{request_data.get('currentChat', '')}"
|
||||
floor = request_data.get("floor", 0)
|
||||
|
||||
try:
|
||||
try:
|
||||
from backend.services.token_usage_service import token_usage_service
|
||||
from backend.models.internal import TokenUsageStatus
|
||||
except ImportError:
|
||||
from services.token_usage_service import token_usage_service
|
||||
from models.internal import TokenUsageStatus
|
||||
|
||||
# 估算 token 数量
|
||||
prompt_tokens = len(str(prompt_messages)) // 4
|
||||
completion_tokens = len(generated_content) // 4
|
||||
|
||||
await token_usage_service.record_usage(
|
||||
chat_id=chat_id,
|
||||
role_name=current_role,
|
||||
chat_name=request_data.get('currentChat', ''),
|
||||
prompt_tokens=prompt_tokens,
|
||||
completion_tokens=completion_tokens,
|
||||
total_tokens=prompt_tokens + completion_tokens,
|
||||
status=TokenUsageStatus.COMPLETED,
|
||||
floor=floor + 1,
|
||||
duration=duration,
|
||||
model=api_config.get("model"),
|
||||
api_provider="openai",
|
||||
api_url=api_config.get("api_url") # ✅ 记录 API URL
|
||||
)
|
||||
except Exception as e:
|
||||
print(f"[ChatWorkflow-Stream] 记录 Token 使用失败: {e}")
|
||||
|
||||
# === 第8步:启动异步并行任务 ===
|
||||
image_task_id = None
|
||||
table_task_id = None
|
||||
|
||||
options = request_data.get("options", {})
|
||||
if options.get("imageWorkflow", False):
|
||||
import uuid
|
||||
image_task_id = f"img_{uuid.uuid4().hex[:8]}"
|
||||
await task_queue_manager.add_task(image_task_id, TaskType.IMAGE_WORKFLOW, chat_id)
|
||||
|
||||
if options.get("dynamicTable", False):
|
||||
import uuid
|
||||
table_task_id = f"tbl_{uuid.uuid4().hex[:8]}"
|
||||
await task_queue_manager.add_task(table_task_id, TaskType.DYNAMIC_TABLE, chat_id)
|
||||
|
||||
await self._start_parallel_tasks(request_data, generated_content, image_task_id, table_task_id)
|
||||
|
||||
return {
|
||||
"success": True,
|
||||
"content": generated_content,
|
||||
"error": None,
|
||||
"activeEntries": active_entries,
|
||||
"taskIds": {
|
||||
"imageWorkflow": image_task_id,
|
||||
"dynamicTable": table_task_id
|
||||
}
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
print(f"[ChatWorkflow-Stream] 错误: {str(e)}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return {
|
||||
"success": False,
|
||||
"content": "",
|
||||
"error": f"工作流执行失败: {str(e)}"
|
||||
}
|
||||
"""处理流式聊天请求 – delegates to WorkflowEngine.run_turn with callbacks."""
|
||||
result = await workflow_engine.run_turn(
|
||||
request_data,
|
||||
stream=True,
|
||||
on_chunk=on_chunk,
|
||||
on_worldbook_active=on_worldbook_active,
|
||||
on_tasks_created=on_tasks_created,
|
||||
)
|
||||
return {
|
||||
"success": result.success,
|
||||
"content": result.content,
|
||||
"error": result.error,
|
||||
"activeEntries": result.active_entries,
|
||||
"taskIds": result.task_ids,
|
||||
"engineRunId": result.run_id,
|
||||
"workflowTemplateId": result.workflow_template_id,
|
||||
}
|
||||
|
||||
105
backend/services/state_machine_runner.py
Normal file
105
backend/services/state_machine_runner.py
Normal file
@@ -0,0 +1,105 @@
|
||||
"""
|
||||
JSON state machine runner for workflow templates.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from pathlib import Path
|
||||
from typing import Any, Callable, Dict, List, Optional
|
||||
|
||||
try:
|
||||
from backend.models.agent import RunEvent, RunEventType, TurnContext, WorkflowRun, RunStatus
|
||||
from backend.services.tool_registry import ToolRegistry
|
||||
except ImportError:
|
||||
from models.agent import RunEvent, RunEventType, TurnContext, WorkflowRun, RunStatus
|
||||
from services.tool_registry import ToolRegistry
|
||||
|
||||
|
||||
class StateMachineRunner:
|
||||
def __init__(
|
||||
self,
|
||||
definition: Dict[str, Any],
|
||||
registry: ToolRegistry,
|
||||
*,
|
||||
on_event: Optional[Callable[[RunEvent], None]] = None,
|
||||
) -> None:
|
||||
self.definition = definition
|
||||
self.registry = registry
|
||||
self.on_event = on_event
|
||||
self.states: Dict[str, Dict[str, Any]] = definition.get("states", {})
|
||||
|
||||
@classmethod
|
||||
def from_file(cls, path: Path, registry: ToolRegistry, **kwargs) -> "StateMachineRunner":
|
||||
with open(path, "r", encoding="utf-8") as f:
|
||||
definition = json.load(f)
|
||||
return cls(definition, registry, **kwargs)
|
||||
|
||||
def _emit(self, run: WorkflowRun, event_type: RunEventType, **payload: Any) -> RunEvent:
|
||||
event = RunEvent(
|
||||
run_id=run.id,
|
||||
type=event_type,
|
||||
state=run.current_state,
|
||||
tool=payload.pop("tool", None),
|
||||
payload=payload,
|
||||
)
|
||||
if self.on_event:
|
||||
self.on_event(event)
|
||||
return event
|
||||
|
||||
async def run(self, run: WorkflowRun, ctx: TurnContext) -> List[RunEvent]:
|
||||
events: List[RunEvent] = []
|
||||
original_on_event = self.on_event
|
||||
|
||||
def collect(event: RunEvent) -> None:
|
||||
events.append(event)
|
||||
if original_on_event:
|
||||
original_on_event(event)
|
||||
|
||||
self.on_event = collect
|
||||
|
||||
initial = self.definition.get("initial")
|
||||
if not initial:
|
||||
raise ValueError("State machine missing 'initial' state")
|
||||
|
||||
current = initial
|
||||
run.status = RunStatus.RUNNING
|
||||
|
||||
try:
|
||||
while current:
|
||||
state_def = self.states.get(current)
|
||||
if not state_def:
|
||||
raise ValueError(f"Unknown state: {current}")
|
||||
|
||||
run.current_state = current
|
||||
events.append(self._emit(run, RunEventType.STATE_ENTER, state=current))
|
||||
|
||||
tool_name = state_def.get("tool")
|
||||
if tool_name:
|
||||
events.append(self._emit(run, RunEventType.TOOL_START, tool=tool_name))
|
||||
await self.registry.execute(tool_name, ctx)
|
||||
events.append(
|
||||
self._emit(
|
||||
run,
|
||||
RunEventType.TOOL_END,
|
||||
tool=tool_name,
|
||||
success=True,
|
||||
)
|
||||
)
|
||||
|
||||
current = state_def.get("next")
|
||||
if current == "end" or current is None:
|
||||
break
|
||||
|
||||
run.status = RunStatus.COMPLETED
|
||||
run.result_content = ctx.generated_content
|
||||
events.append(self._emit(run, RunEventType.COMPLETE))
|
||||
except Exception as exc:
|
||||
run.status = RunStatus.FAILED
|
||||
run.error = str(exc)
|
||||
ctx.error = str(exc)
|
||||
events.append(self._emit(run, RunEventType.ERROR, message=str(exc)))
|
||||
raise
|
||||
finally:
|
||||
self.on_event = original_on_event
|
||||
|
||||
return events
|
||||
188
backend/services/studio_context_service.py
Normal file
188
backend/services/studio_context_service.py
Normal file
@@ -0,0 +1,188 @@
|
||||
"""
|
||||
Assemble Studio run prompt context from pipeline snapshot, workflow variables,
|
||||
and node outputs (R2). Does not include full chat history.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import re
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from models.studio_models import (
|
||||
PipelineDefinition,
|
||||
PromptBlock,
|
||||
StudioNode,
|
||||
StudioNodeRunState,
|
||||
StudioRun,
|
||||
)
|
||||
|
||||
_NODE_OUTPUT_REF = re.compile(r"^([^.]+)\.output$")
|
||||
|
||||
AUTO_BLOCK_SPECS = (
|
||||
("currentProduct", "目前产物", "auto"),
|
||||
("thinkingFlow", "思考流程", "auto"),
|
||||
("coreGoal", "核心目的", "auto"),
|
||||
("scoringCriteria", "评价标准与优化建议", "auto"),
|
||||
)
|
||||
|
||||
|
||||
def _find_node(pipeline: PipelineDefinition, node_id: str) -> Optional[StudioNode]:
|
||||
for node in pipeline.nodes:
|
||||
if node.id == node_id:
|
||||
return node
|
||||
return None
|
||||
|
||||
|
||||
def _state_map(run: StudioRun) -> Dict[str, StudioNodeRunState]:
|
||||
return {s.nodeId: s for s in run.nodeStates}
|
||||
|
||||
|
||||
def _format_draft(draft: Optional[Dict[str, Any]]) -> str:
|
||||
if not draft:
|
||||
return "(暂无内容)"
|
||||
for key in ("entryContent", "content", "text", "body"):
|
||||
value = draft.get(key)
|
||||
if isinstance(value, str) and value.strip():
|
||||
return value.strip()
|
||||
return json.dumps(draft, ensure_ascii=False, indent=2)
|
||||
|
||||
|
||||
def _format_scoring(config: Dict[str, Any]) -> str:
|
||||
scoring = config.get("scoring") or {}
|
||||
if not scoring.get("enabled", True):
|
||||
return "(本步骤未启用评价)"
|
||||
dimensions = scoring.get("dimensions") or []
|
||||
if not dimensions:
|
||||
rubric = scoring.get("rubric")
|
||||
if rubric:
|
||||
return str(rubric).strip()
|
||||
return "(未配置评价维度)"
|
||||
lines: List[str] = []
|
||||
for dim in dimensions:
|
||||
name = dim.get("name") or dim.get("id") or "维度"
|
||||
criteria = (dim.get("criteria") or "").strip()
|
||||
lines.append(f"- {name}:{criteria}" if criteria else f"- {name}")
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
def _resolve_workflow_ref(ref: str, workflow_variables: Dict[str, Any]) -> str:
|
||||
value = workflow_variables.get(ref)
|
||||
if value is None:
|
||||
return "(尚未可用)"
|
||||
if isinstance(value, str):
|
||||
return value.strip() or "(空)"
|
||||
return json.dumps(value, ensure_ascii=False, indent=2)
|
||||
|
||||
|
||||
def _resolve_node_output_ref(
|
||||
ref: str,
|
||||
pipeline: PipelineDefinition,
|
||||
state_by_id: Dict[str, StudioNodeRunState],
|
||||
) -> str:
|
||||
match = _NODE_OUTPUT_REF.match(ref)
|
||||
if not match:
|
||||
return f"(无法解析引用:{ref})"
|
||||
node_id = match.group(1)
|
||||
source_node = _find_node(pipeline, node_id)
|
||||
source_state = state_by_id.get(node_id)
|
||||
label = source_node.displayName if source_node else node_id
|
||||
if not source_state or source_state.status != "completed":
|
||||
return f"(前序步骤「{label}」尚未完成)"
|
||||
return _format_draft(source_state.lastDraft)
|
||||
|
||||
|
||||
def _auto_block_content(
|
||||
block_id: str,
|
||||
node: StudioNode,
|
||||
node_state: Optional[StudioNodeRunState],
|
||||
) -> str:
|
||||
config = node.config or {}
|
||||
if block_id == "currentProduct":
|
||||
return _format_draft(node_state.lastDraft if node_state else None)
|
||||
if block_id == "thinkingFlow":
|
||||
return (config.get("thinkingPrompt") or "").strip() or "(未配置思考流程)"
|
||||
if block_id == "coreGoal":
|
||||
return (config.get("stepGoal") or "").strip() or "(未配置步骤目标)"
|
||||
if block_id == "scoringCriteria":
|
||||
return _format_scoring(config)
|
||||
return ""
|
||||
|
||||
|
||||
def assemble_prompt_blocks(run: StudioRun, node_id: str) -> List[PromptBlock]:
|
||||
"""
|
||||
Build ordered prompt blocks for a worldbook step from inputs[].ref,
|
||||
workflow variables, node outputs, and auto-injected context items.
|
||||
"""
|
||||
pipeline = run.pipelineSnapshot
|
||||
node = _find_node(pipeline, node_id)
|
||||
if not node:
|
||||
return []
|
||||
|
||||
if node.skillId != "studio.worldbook_entry":
|
||||
return []
|
||||
|
||||
state_by_id = _state_map(run)
|
||||
node_state = state_by_id.get(node_id)
|
||||
workflow_variables = dict(run.workflowVariables or {})
|
||||
blocks: List[PromptBlock] = []
|
||||
seen_ids: set[str] = set()
|
||||
|
||||
def append_block(
|
||||
block_id: str,
|
||||
label: str,
|
||||
content: str,
|
||||
source: str,
|
||||
) -> None:
|
||||
if block_id in seen_ids:
|
||||
return
|
||||
seen_ids.add(block_id)
|
||||
blocks.append(
|
||||
PromptBlock(
|
||||
id=block_id,
|
||||
label=label,
|
||||
content=content,
|
||||
source=source,
|
||||
)
|
||||
)
|
||||
|
||||
for inp in node.inputs or []:
|
||||
ref = (inp.ref or "").strip()
|
||||
if not ref:
|
||||
continue
|
||||
label = (inp.label or ref).strip()
|
||||
block_id = f"ref:{ref}"
|
||||
|
||||
if ref.startswith("workflow."):
|
||||
content = _resolve_workflow_ref(ref, workflow_variables)
|
||||
if inp.optional and content in ("(尚未可用)", "(空)"):
|
||||
continue
|
||||
append_block(block_id, label, content, "workflow")
|
||||
continue
|
||||
|
||||
if _NODE_OUTPUT_REF.match(ref):
|
||||
content = _resolve_node_output_ref(ref, pipeline, state_by_id)
|
||||
if inp.optional and content.startswith("(前序步骤"):
|
||||
continue
|
||||
append_block(block_id, label, content, "manual")
|
||||
continue
|
||||
|
||||
append_block(block_id, label, f"(未知引用类型:{ref})", "manual")
|
||||
|
||||
for block_id, label, source in AUTO_BLOCK_SPECS:
|
||||
content = _auto_block_content(block_id, node, node_state)
|
||||
append_block(block_id, label, content, source)
|
||||
|
||||
return blocks
|
||||
|
||||
|
||||
def store_context_on_run(run: StudioRun, node_id: Optional[str]) -> StudioRun:
|
||||
"""Attach assembled prompt blocks to run for debug / frontend display."""
|
||||
if not node_id:
|
||||
return run.model_copy(update={"lastPromptBlocks": []})
|
||||
|
||||
node = _find_node(run.pipelineSnapshot, node_id)
|
||||
if not node or node.skillId != "studio.worldbook_entry":
|
||||
return run.model_copy(update={"lastPromptBlocks": []})
|
||||
|
||||
blocks = assemble_prompt_blocks(run, node_id)
|
||||
return run.model_copy(update={"lastPromptBlocks": blocks})
|
||||
456
backend/services/studio_project_service.py
Normal file
456
backend/services/studio_project_service.py
Normal file
@@ -0,0 +1,456 @@
|
||||
"""
|
||||
Load/save Studio projects and skill templates from data/agent/.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import logging
|
||||
import re
|
||||
import shutil
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from core.config import settings
|
||||
from models.studio_models import (
|
||||
CreateStudioProjectRequest,
|
||||
PipelineDefinition,
|
||||
StudioProject,
|
||||
StudioProjectMeta,
|
||||
StudioProjectSummary,
|
||||
WorkflowTemplateSummary,
|
||||
WorkflowVariablesResponse,
|
||||
WorkflowVariableDef,
|
||||
)
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
DEFAULT_TEMPLATE_ID = "builtin.studio.example"
|
||||
|
||||
POSITION_STRING_MAP = {
|
||||
"after_char": 0,
|
||||
"before_char": 1,
|
||||
"before_example": 2,
|
||||
"after_example": 3,
|
||||
"system": 4,
|
||||
"as_system": 5,
|
||||
"depth": 6,
|
||||
"macro": 7,
|
||||
}
|
||||
|
||||
ACTIVATION_LEGACY_MAP = {
|
||||
"normal": "permanent",
|
||||
"constant": "permanent",
|
||||
"selective": "keyword",
|
||||
}
|
||||
|
||||
|
||||
def _normalize_position(value: Any) -> int:
|
||||
if value is None:
|
||||
return 1
|
||||
if isinstance(value, int):
|
||||
return value
|
||||
if isinstance(value, str):
|
||||
if value.isdigit():
|
||||
return int(value)
|
||||
return POSITION_STRING_MAP.get(value, 1)
|
||||
try:
|
||||
return int(value)
|
||||
except (TypeError, ValueError):
|
||||
return 1
|
||||
|
||||
|
||||
def _normalize_activation(value: Any) -> str:
|
||||
if not value:
|
||||
return "permanent"
|
||||
text = str(value)
|
||||
return ACTIVATION_LEGACY_MAP.get(text, text)
|
||||
|
||||
|
||||
def _migrate_scoring(scoring: Dict[str, Any]) -> Dict[str, Any]:
|
||||
if not scoring:
|
||||
return {"enabled": True, "dimensions": []}
|
||||
if scoring.get("dimensions"):
|
||||
return scoring
|
||||
rubric = scoring.get("rubric")
|
||||
if rubric:
|
||||
scoring = {**scoring}
|
||||
scoring["dimensions"] = [
|
||||
{
|
||||
"id": "default",
|
||||
"name": "综合质量",
|
||||
"criteria": rubric,
|
||||
}
|
||||
]
|
||||
scoring.pop("rubric", None)
|
||||
return scoring
|
||||
|
||||
|
||||
def _normalize_node(node: Dict[str, Any]) -> Dict[str, Any]:
|
||||
node = dict(node)
|
||||
config = dict(node.get("config") or {})
|
||||
insertion = dict(config.get("insertion") or {})
|
||||
if insertion:
|
||||
insertion["position"] = _normalize_position(insertion.get("position"))
|
||||
insertion["activationType"] = _normalize_activation(
|
||||
insertion.get("activationType")
|
||||
)
|
||||
config["insertion"] = insertion
|
||||
if "scoring" in config:
|
||||
config["scoring"] = _migrate_scoring(dict(config.get("scoring") or {}))
|
||||
node["config"] = config
|
||||
return node
|
||||
|
||||
|
||||
def _parse_node_ref(ref: str, node_ids: set[str]) -> Optional[str]:
|
||||
if not ref or not ref.endswith(".output"):
|
||||
return None
|
||||
node_id = ref[: -len(".output")]
|
||||
return node_id if node_id in node_ids else None
|
||||
|
||||
|
||||
def _build_node_dependency_edges(pipeline: Dict[str, Any]) -> List[tuple[str, str]]:
|
||||
nodes = pipeline.get("nodes") or []
|
||||
node_ids = {n["id"] for n in nodes if n.get("id")}
|
||||
edges: List[tuple[str, str]] = []
|
||||
seen: set[tuple[str, str]] = set()
|
||||
for node in nodes:
|
||||
to_id = node.get("id")
|
||||
if not to_id:
|
||||
continue
|
||||
for inp in node.get("inputs") or []:
|
||||
src = _parse_node_ref(inp.get("ref", ""), node_ids)
|
||||
if not src or src == to_id:
|
||||
continue
|
||||
pair = (src, to_id)
|
||||
if pair in seen:
|
||||
continue
|
||||
seen.add(pair)
|
||||
edges.append(pair)
|
||||
return edges
|
||||
|
||||
|
||||
def _detect_reference_cycles(pipeline: Dict[str, Any]) -> List[List[str]]:
|
||||
nodes = pipeline.get("nodes") or []
|
||||
node_ids = [n["id"] for n in nodes if n.get("id")]
|
||||
adj: Dict[str, List[str]] = {nid: [] for nid in node_ids}
|
||||
for src, dst in _build_node_dependency_edges(pipeline):
|
||||
adj[src].append(dst)
|
||||
|
||||
cycles: List[List[str]] = []
|
||||
visited: set[str] = set()
|
||||
stack: set[str] = set()
|
||||
path: List[str] = []
|
||||
|
||||
def dfs(node_id: str) -> None:
|
||||
visited.add(node_id)
|
||||
stack.add(node_id)
|
||||
path.append(node_id)
|
||||
for nxt in adj.get(node_id, []):
|
||||
if nxt not in visited:
|
||||
dfs(nxt)
|
||||
elif nxt in stack:
|
||||
start = path.index(nxt)
|
||||
if start >= 0:
|
||||
cycles.append(path[start:] + [nxt])
|
||||
path.pop()
|
||||
stack.discard(node_id)
|
||||
|
||||
for nid in node_ids:
|
||||
if nid not in visited:
|
||||
dfs(nid)
|
||||
return cycles
|
||||
|
||||
|
||||
def _validate_pipeline_refs(pipeline: Dict[str, Any]) -> None:
|
||||
cycles = _detect_reference_cycles(pipeline)
|
||||
if not cycles:
|
||||
return
|
||||
nodes = {n["id"]: n.get("displayName", n["id"]) for n in pipeline.get("nodes") or []}
|
||||
first = cycles[0]
|
||||
chain = " → ".join(nodes.get(nid, nid) for nid in first)
|
||||
raise ValueError(f"流水线存在循环引用:{chain}")
|
||||
|
||||
|
||||
def _normalize_pipeline_dict(pipeline: Dict[str, Any]) -> Dict[str, Any]:
|
||||
pipeline = dict(pipeline)
|
||||
nodes = pipeline.get("nodes") or []
|
||||
pipeline["nodes"] = [_normalize_node(n) for n in nodes]
|
||||
return pipeline
|
||||
|
||||
|
||||
def _read_json(path: Path) -> Any:
|
||||
with path.open("r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
|
||||
|
||||
def _write_json(path: Path, data: Any) -> None:
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
with path.open("w", encoding="utf-8") as f:
|
||||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||||
|
||||
|
||||
def _slugify(name: str) -> str:
|
||||
slug = re.sub(r"[^\w\u4e00-\u9fff-]+", "-", name.strip(), flags=re.UNICODE)
|
||||
slug = re.sub(r"-+", "-", slug).strip("-").lower()
|
||||
return slug or "project"
|
||||
|
||||
|
||||
class StudioProjectService:
|
||||
@property
|
||||
def projects_root(self) -> Path:
|
||||
return settings.AGENT_STUDIO_PROJECTS_PATH
|
||||
|
||||
@property
|
||||
def templates_root(self) -> Path:
|
||||
return settings.AGENT_TEMPLATES_PATH
|
||||
|
||||
def _project_dir(self, project_id: str) -> Path:
|
||||
return self.projects_root / project_id
|
||||
|
||||
def _meta_path(self, project_id: str) -> Path:
|
||||
return self._project_dir(project_id) / "meta.json"
|
||||
|
||||
def _pipeline_path(self, project_id: str) -> Path:
|
||||
return self._project_dir(project_id) / "pipeline.json"
|
||||
|
||||
def list_projects(self) -> List[StudioProjectSummary]:
|
||||
root = self.projects_root
|
||||
if not root.exists():
|
||||
return []
|
||||
summaries: List[StudioProjectSummary] = []
|
||||
for child in sorted(root.iterdir()):
|
||||
if not child.is_dir():
|
||||
continue
|
||||
meta_path = child / "meta.json"
|
||||
if not meta_path.exists():
|
||||
continue
|
||||
meta = _read_json(meta_path)
|
||||
summaries.append(
|
||||
StudioProjectSummary(
|
||||
id=meta.get("id", child.name),
|
||||
name=meta.get("name", child.name),
|
||||
description=meta.get("description", ""),
|
||||
updatedAt=meta.get("updatedAt", ""),
|
||||
)
|
||||
)
|
||||
return summaries
|
||||
|
||||
def get_project(self, project_id: str) -> StudioProject:
|
||||
meta_path = self._meta_path(project_id)
|
||||
pipeline_path = self._pipeline_path(project_id)
|
||||
if not meta_path.exists() or not pipeline_path.exists():
|
||||
raise FileNotFoundError(f"Studio project not found: {project_id}")
|
||||
meta = _read_json(meta_path)
|
||||
pipeline = _normalize_pipeline_dict(_read_json(pipeline_path))
|
||||
return StudioProject(
|
||||
meta=StudioProjectMeta(**meta),
|
||||
pipeline=PipelineDefinition(**pipeline),
|
||||
)
|
||||
|
||||
def update_project_bindings(
|
||||
self,
|
||||
project_id: str,
|
||||
character_id: str,
|
||||
worldbook_id: str,
|
||||
) -> StudioProject:
|
||||
meta_path = self._meta_path(project_id)
|
||||
if not meta_path.exists():
|
||||
raise FileNotFoundError(f"Studio project not found: {project_id}")
|
||||
meta = _read_json(meta_path)
|
||||
meta["characterId"] = character_id
|
||||
meta["worldbookId"] = worldbook_id
|
||||
meta["updatedAt"] = datetime.now().isoformat()
|
||||
_write_json(meta_path, meta)
|
||||
return self.get_project(project_id)
|
||||
|
||||
def update_project_meta(
|
||||
self,
|
||||
project_id: str,
|
||||
*,
|
||||
name: Optional[str] = None,
|
||||
description: Optional[str] = None,
|
||||
) -> StudioProject:
|
||||
meta_path = self._meta_path(project_id)
|
||||
if not meta_path.exists():
|
||||
raise FileNotFoundError(f"Studio project not found: {project_id}")
|
||||
meta = _read_json(meta_path)
|
||||
if name is not None:
|
||||
meta["name"] = name.strip()
|
||||
if description is not None:
|
||||
meta["description"] = description
|
||||
meta["updatedAt"] = datetime.now().isoformat()
|
||||
_write_json(meta_path, meta)
|
||||
return self.get_project(project_id)
|
||||
|
||||
def save_pipeline(self, project_id: str, pipeline: PipelineDefinition) -> StudioProject:
|
||||
meta_path = self._meta_path(project_id)
|
||||
if not meta_path.exists():
|
||||
raise FileNotFoundError(f"Studio project not found: {project_id}")
|
||||
normalized = _normalize_pipeline_dict(pipeline.model_dump(exclude_none=True))
|
||||
_validate_pipeline_refs(normalized)
|
||||
meta = _read_json(meta_path)
|
||||
now = datetime.now().isoformat()
|
||||
meta["updatedAt"] = now
|
||||
_write_json(meta_path, meta)
|
||||
_write_json(self._pipeline_path(project_id), normalized)
|
||||
return self.get_project(project_id)
|
||||
|
||||
def list_workflow_templates(self) -> List[WorkflowTemplateSummary]:
|
||||
root = self.templates_root
|
||||
if not root.exists():
|
||||
return []
|
||||
summaries: List[WorkflowTemplateSummary] = []
|
||||
for child in sorted(root.iterdir()):
|
||||
if not child.is_dir():
|
||||
continue
|
||||
meta_path = child / "meta.json"
|
||||
if not meta_path.exists():
|
||||
continue
|
||||
meta = _read_json(meta_path)
|
||||
summaries.append(
|
||||
WorkflowTemplateSummary(
|
||||
id=meta.get("id", child.name),
|
||||
name=meta.get("name", child.name),
|
||||
description=meta.get("description", ""),
|
||||
)
|
||||
)
|
||||
return summaries
|
||||
|
||||
def get_workflow_variables(self, project_id: Optional[str] = None) -> WorkflowVariablesResponse:
|
||||
path = settings.AGENT_WORKFLOW_VARIABLES_FILE
|
||||
if path.exists():
|
||||
raw = _read_json(path)
|
||||
else:
|
||||
raw = {
|
||||
"builtIn": [
|
||||
{"ref": "workflow.goal", "label": "工作流目标文本", "description": ""},
|
||||
{"ref": "workflow.boundWorldbook", "label": "绑定世界书摘要", "description": ""},
|
||||
{"ref": "workflow.boundCharacter", "label": "绑定角色卡摘要", "description": ""},
|
||||
],
|
||||
"dynamicSuffixes": [
|
||||
{"suffix": ".output", "labelPattern": "{displayName} · 上轮产物"},
|
||||
{"suffix": ".entryDraft", "labelPattern": "{displayName} · 条目草稿"},
|
||||
],
|
||||
}
|
||||
built_in = [
|
||||
WorkflowVariableDef(**item) for item in raw.get("builtIn", [])
|
||||
]
|
||||
dynamic: List[WorkflowVariableDef] = []
|
||||
suffixes = raw.get("dynamicSuffixes") or [
|
||||
{"suffix": ".output", "labelPattern": "{displayName} · 世界书条目"},
|
||||
]
|
||||
if project_id:
|
||||
try:
|
||||
project = self.get_project(project_id)
|
||||
for node in project.pipeline.nodes:
|
||||
if not node.enabled:
|
||||
continue
|
||||
if node.skillId != "studio.worldbook_entry":
|
||||
continue
|
||||
for suffix_def in suffixes:
|
||||
suffix = suffix_def.get("suffix", ".output")
|
||||
if suffix != ".output":
|
||||
continue
|
||||
pattern = suffix_def.get(
|
||||
"labelPattern", "{displayName} · 世界书条目"
|
||||
)
|
||||
ref = f"{node.id}{suffix}"
|
||||
label = pattern.replace("{displayName}", node.displayName)
|
||||
dynamic.append(
|
||||
WorkflowVariableDef(ref=ref, label=label, description="")
|
||||
)
|
||||
except FileNotFoundError:
|
||||
pass
|
||||
return WorkflowVariablesResponse(builtIn=built_in, dynamic=dynamic)
|
||||
|
||||
def get_skill_templates(self) -> Dict[str, Any]:
|
||||
path = settings.AGENT_SKILL_TEMPLATES_FILE
|
||||
if not path.exists():
|
||||
raise FileNotFoundError("skill_templates.json not found")
|
||||
return _read_json(path)
|
||||
|
||||
def get_niches(self) -> Dict[str, Any]:
|
||||
path = settings.AGENT_NICHES_FILE
|
||||
if not path.exists():
|
||||
return {"niches": []}
|
||||
return _read_json(path)
|
||||
|
||||
def _unique_project_id(self, base_id: str) -> str:
|
||||
candidate = base_id
|
||||
n = 1
|
||||
while self._project_dir(candidate).exists():
|
||||
candidate = f"{base_id}-{n}"
|
||||
n += 1
|
||||
return candidate
|
||||
|
||||
def create_project(self, req: CreateStudioProjectRequest) -> StudioProject:
|
||||
template_id = req.template_id or DEFAULT_TEMPLATE_ID
|
||||
template_dir = self.templates_root / template_id
|
||||
if not template_dir.exists():
|
||||
raise FileNotFoundError(f"Studio template not found: {template_id}")
|
||||
|
||||
base_id = req.project_id or _slugify(req.name)
|
||||
project_id = self._unique_project_id(base_id)
|
||||
dest = self._project_dir(project_id)
|
||||
dest.mkdir(parents=True, exist_ok=False)
|
||||
|
||||
template_meta = _read_json(template_dir / "meta.json")
|
||||
template_pipeline = _read_json(template_dir / "pipeline.json")
|
||||
now = datetime.now().isoformat()
|
||||
|
||||
meta = {
|
||||
"id": project_id,
|
||||
"name": req.name,
|
||||
"description": template_meta.get("description", ""),
|
||||
"templateId": template_id,
|
||||
"characterId": None,
|
||||
"worldbookId": None,
|
||||
"createdAt": now,
|
||||
"updatedAt": now,
|
||||
}
|
||||
_write_json(dest / "meta.json", meta)
|
||||
_write_json(dest / "pipeline.json", _normalize_pipeline_dict(template_pipeline))
|
||||
return self.get_project(project_id)
|
||||
|
||||
def delete_project(self, project_id: str) -> None:
|
||||
project_dir = self._project_dir(project_id)
|
||||
if not project_dir.exists():
|
||||
raise FileNotFoundError(f"Studio project not found: {project_id}")
|
||||
shutil.rmtree(project_dir)
|
||||
runs_dir = settings.AGENT_STUDIO_RUNS_PATH / project_id
|
||||
if runs_dir.exists():
|
||||
shutil.rmtree(runs_dir)
|
||||
|
||||
def ensure_default_project(self) -> None:
|
||||
"""Copy example template into default project if missing."""
|
||||
default_dir = self._project_dir("default")
|
||||
if default_dir.exists():
|
||||
return
|
||||
template_dir = self.templates_root / DEFAULT_TEMPLATE_ID
|
||||
if not template_dir.exists():
|
||||
logger.warning("builtin.studio.example template missing; skip default project seed")
|
||||
return
|
||||
default_dir.mkdir(parents=True, exist_ok=True)
|
||||
template_meta = _read_json(template_dir / "meta.json")
|
||||
now = datetime.now().isoformat()
|
||||
meta = {
|
||||
"id": "default",
|
||||
"name": "示例角色项目",
|
||||
"description": template_meta.get("description", ""),
|
||||
"templateId": DEFAULT_TEMPLATE_ID,
|
||||
"characterId": None,
|
||||
"worldbookId": None,
|
||||
"createdAt": now,
|
||||
"updatedAt": now,
|
||||
}
|
||||
_write_json(default_dir / "meta.json", meta)
|
||||
shutil.copy2(template_dir / "pipeline.json", default_dir / "pipeline.json")
|
||||
|
||||
|
||||
studio_project_service = StudioProjectService()
|
||||
|
||||
try:
|
||||
studio_project_service.ensure_default_project()
|
||||
except Exception as _seed_err:
|
||||
logger.warning("Studio default project seed skipped: %s", _seed_err)
|
||||
502
backend/services/studio_run_service.py
Normal file
502
backend/services/studio_run_service.py
Normal file
@@ -0,0 +1,502 @@
|
||||
"""
|
||||
Create and load Studio pipeline runs with frozen pipeline snapshots.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import copy
|
||||
import json
|
||||
import logging
|
||||
import shutil
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Any, AsyncGenerator, Dict, List, Optional
|
||||
|
||||
from core.config import settings
|
||||
from models.studio_models import (
|
||||
PipelineDefinition,
|
||||
StudioNode,
|
||||
StudioNodeRunState,
|
||||
StudioRun,
|
||||
StudioRunStatus,
|
||||
StudioRunSummary,
|
||||
)
|
||||
from services.character_service import CharacterService
|
||||
from services.studio_context_service import assemble_prompt_blocks, store_context_on_run
|
||||
from services.studio_project_service import studio_project_service
|
||||
from services.studio_step_respond import (
|
||||
resolve_api_config,
|
||||
studio_step_respond,
|
||||
studio_step_respond_stream,
|
||||
)
|
||||
from services.worldbook_service import worldbook_service
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def _read_json(path: Path) -> dict:
|
||||
with path.open("r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
|
||||
|
||||
def _write_json(path: Path, data: dict) -> None:
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
with path.open("w", encoding="utf-8") as f:
|
||||
json.dump(data, f, ensure_ascii=False, indent=2)
|
||||
|
||||
|
||||
def _build_node_states(pipeline: PipelineDefinition) -> tuple[list[StudioNodeRunState], Optional[str]]:
|
||||
"""First enabled node is active; other enabled nodes pending; disabled skipped."""
|
||||
states: list[StudioNodeRunState] = []
|
||||
first_active_id: Optional[str] = None
|
||||
seen_active = False
|
||||
|
||||
for node in pipeline.nodes:
|
||||
if not node.enabled:
|
||||
status = "skipped"
|
||||
elif not seen_active:
|
||||
status = "active"
|
||||
first_active_id = node.id
|
||||
seen_active = True
|
||||
else:
|
||||
status = "pending"
|
||||
|
||||
states.append(
|
||||
StudioNodeRunState(
|
||||
nodeId=node.id,
|
||||
displayName=node.displayName,
|
||||
skillId=node.skillId,
|
||||
status=status,
|
||||
loopUntilSatisfied=node.loopUntilSatisfied,
|
||||
)
|
||||
)
|
||||
|
||||
return states, first_active_id
|
||||
|
||||
|
||||
def _find_node(pipeline: PipelineDefinition, node_id: str) -> Optional[StudioNode]:
|
||||
for node in pipeline.nodes:
|
||||
if node.id == node_id:
|
||||
return node
|
||||
return None
|
||||
|
||||
|
||||
def _next_enabled_node_id(pipeline: PipelineDefinition, after_node_id: str) -> Optional[str]:
|
||||
seen = False
|
||||
for node in pipeline.nodes:
|
||||
if node.id == after_node_id:
|
||||
seen = True
|
||||
continue
|
||||
if seen and node.enabled:
|
||||
return node.id
|
||||
return None
|
||||
|
||||
|
||||
class StudioRunService:
|
||||
def __init__(self) -> None:
|
||||
self._character_service = CharacterService()
|
||||
|
||||
@property
|
||||
def runs_root(self) -> Path:
|
||||
return settings.AGENT_STUDIO_RUNS_PATH
|
||||
|
||||
def _project_runs_dir(self, project_id: str) -> Path:
|
||||
return self.runs_root / project_id
|
||||
|
||||
def _run_dir(self, project_id: str, run_id: str) -> Path:
|
||||
return self._project_runs_dir(project_id) / run_id
|
||||
|
||||
def _run_path(self, project_id: str, run_id: str) -> Path:
|
||||
return self._run_dir(project_id, run_id) / "run.json"
|
||||
|
||||
def _save_run(self, project_id: str, run_id: str, run: StudioRun) -> None:
|
||||
_write_json(
|
||||
self._run_path(project_id, run_id),
|
||||
run.model_dump(mode="json"),
|
||||
)
|
||||
|
||||
def create_run(self, project_id: str) -> StudioRun:
|
||||
project = studio_project_service.get_project(project_id)
|
||||
snapshot = PipelineDefinition(**copy.deepcopy(project.pipeline.model_dump()))
|
||||
now = datetime.now().isoformat()
|
||||
run_id = str(uuid.uuid4())
|
||||
node_states, current_node_id = _build_node_states(snapshot)
|
||||
|
||||
has_active = current_node_id is not None
|
||||
run = StudioRun(
|
||||
id=run_id,
|
||||
projectId=project_id,
|
||||
status=StudioRunStatus.RUNNING if has_active else StudioRunStatus.PENDING,
|
||||
pipelineSnapshot=snapshot,
|
||||
pipelineVersionNote=now,
|
||||
currentNodeId=current_node_id,
|
||||
nodeStates=node_states,
|
||||
workflowVariables={"workflow.goal": snapshot.workflowGoal},
|
||||
createdAt=now,
|
||||
updatedAt=now,
|
||||
)
|
||||
self._save_run(project_id, run_id, run)
|
||||
return run
|
||||
|
||||
def list_runs(self, project_id: str) -> List[StudioRunSummary]:
|
||||
root = self._project_runs_dir(project_id)
|
||||
if not root.exists():
|
||||
return []
|
||||
|
||||
summaries: List[StudioRunSummary] = []
|
||||
for child in sorted(root.iterdir(), key=lambda p: p.stat().st_mtime, reverse=True):
|
||||
if not child.is_dir():
|
||||
continue
|
||||
run_path = child / "run.json"
|
||||
if not run_path.exists():
|
||||
continue
|
||||
raw = _read_json(run_path)
|
||||
summaries.append(
|
||||
StudioRunSummary(
|
||||
id=raw.get("id", child.name),
|
||||
projectId=raw.get("projectId", project_id),
|
||||
status=StudioRunStatus(raw.get("status", StudioRunStatus.PENDING.value)),
|
||||
currentNodeId=raw.get("currentNodeId"),
|
||||
title=raw.get("title", ""),
|
||||
createdAt=raw.get("createdAt", ""),
|
||||
updatedAt=raw.get("updatedAt", ""),
|
||||
)
|
||||
)
|
||||
return summaries
|
||||
|
||||
def get_run(self, project_id: str, run_id: str) -> StudioRun:
|
||||
run_path = self._run_path(project_id, run_id)
|
||||
if not run_path.exists():
|
||||
raise FileNotFoundError(f"Studio run not found: {project_id}/{run_id}")
|
||||
run = StudioRun(**_read_json(run_path))
|
||||
return run
|
||||
|
||||
def _validate_display_params(
|
||||
self, node: StudioNode, display_params: Dict[str, str]
|
||||
) -> Dict[str, str]:
|
||||
normalized: Dict[str, str] = {}
|
||||
for dp in node.displayParams:
|
||||
raw = display_params.get(dp.key, "")
|
||||
value = (raw or "").strip()
|
||||
if dp.required and not value:
|
||||
raise ValueError(f"缺少必填项:{dp.label}")
|
||||
normalized[dp.key] = value
|
||||
return normalized
|
||||
|
||||
def _execute_init_bind(
|
||||
self,
|
||||
project_id: str,
|
||||
run: StudioRun,
|
||||
node: StudioNode,
|
||||
display_params: Dict[str, str],
|
||||
) -> tuple[Dict[str, Any], Dict[str, Any]]:
|
||||
params = self._validate_display_params(node, display_params)
|
||||
character_name = params.get("characterName", "")
|
||||
worldbook_name = params.get("worldbookName", "")
|
||||
|
||||
if self._character_service.get_character_by_name(character_name):
|
||||
raise ValueError(f"角色「{character_name}」已存在")
|
||||
if worldbook_service._get_worldbook_path(worldbook_name).exists():
|
||||
raise ValueError(f"世界书「{worldbook_name}」已存在")
|
||||
|
||||
worldbook = worldbook_service.create_worldbook(worldbook_name)
|
||||
worldbook_id = worldbook["id"]
|
||||
|
||||
character = self._character_service.create_character(
|
||||
{
|
||||
"name": character_name,
|
||||
"description": "",
|
||||
"personality": "",
|
||||
"scenario": "",
|
||||
"first_mes": "",
|
||||
"mes_example": "",
|
||||
"categories": [],
|
||||
"tags": [],
|
||||
"worldInfoId": worldbook_id,
|
||||
}
|
||||
)
|
||||
character_id = character.id
|
||||
|
||||
studio_project_service.update_project_bindings(
|
||||
project_id, character_id, worldbook_id
|
||||
)
|
||||
|
||||
workflow_variables = dict(run.workflowVariables or {})
|
||||
workflow_variables["workflow.goal"] = run.pipelineSnapshot.workflowGoal
|
||||
workflow_variables["workflow.boundCharacter"] = (
|
||||
f"名称:{character_name}\nID:{character_id}"
|
||||
)
|
||||
workflow_variables["workflow.boundWorldbook"] = (
|
||||
f"名称:{worldbook_name}\nID:{worldbook_id}"
|
||||
)
|
||||
|
||||
last_draft = {
|
||||
"displayParams": params,
|
||||
"characterId": character_id,
|
||||
"worldbookId": worldbook_id,
|
||||
"characterName": character_name,
|
||||
"worldbookName": worldbook_name,
|
||||
}
|
||||
return workflow_variables, last_draft
|
||||
|
||||
def advance_run(
|
||||
self,
|
||||
project_id: str,
|
||||
run_id: str,
|
||||
display_params: Optional[Dict[str, str]] = None,
|
||||
) -> StudioRun:
|
||||
run = self.get_run(project_id, run_id)
|
||||
if run.status not in (StudioRunStatus.RUNNING, StudioRunStatus.PENDING):
|
||||
raise ValueError("运行已结束,无法继续推进")
|
||||
|
||||
current_node_id = run.currentNodeId
|
||||
if not current_node_id:
|
||||
raise ValueError("当前运行无活动节点")
|
||||
|
||||
current_node = _find_node(run.pipelineSnapshot, current_node_id)
|
||||
if not current_node:
|
||||
raise ValueError(f"节点不存在:{current_node_id}")
|
||||
|
||||
current_state = next(
|
||||
(s for s in run.nodeStates if s.nodeId == current_node_id), None
|
||||
)
|
||||
if not current_state or current_state.status != "active":
|
||||
raise ValueError("当前节点不可执行")
|
||||
|
||||
workflow_variables = dict(run.workflowVariables or {})
|
||||
last_draft: Optional[Dict[str, Any]] = None
|
||||
|
||||
if current_node.skillId == "studio.init_bind":
|
||||
if not display_params:
|
||||
raise ValueError("请填写引导表单后再提交")
|
||||
workflow_variables, last_draft = self._execute_init_bind(
|
||||
project_id, run, current_node, display_params
|
||||
)
|
||||
else:
|
||||
raise NotImplementedError(
|
||||
f"技能「{current_node.skillId}」的执行尚未实现(R2+)"
|
||||
)
|
||||
|
||||
next_node_id = _next_enabled_node_id(run.pipelineSnapshot, current_node_id)
|
||||
now = datetime.now().isoformat()
|
||||
|
||||
new_node_states: list[StudioNodeRunState] = []
|
||||
for state in run.nodeStates:
|
||||
updated = state.model_copy()
|
||||
if state.nodeId == current_node_id:
|
||||
updated.status = "completed"
|
||||
if last_draft is not None:
|
||||
updated.lastDraft = last_draft
|
||||
elif next_node_id and state.nodeId == next_node_id:
|
||||
updated.status = "active"
|
||||
new_node_states.append(updated)
|
||||
|
||||
new_status = (
|
||||
StudioRunStatus.COMPLETED if not next_node_id else StudioRunStatus.RUNNING
|
||||
)
|
||||
updated_run = run.model_copy(
|
||||
update={
|
||||
"currentNodeId": next_node_id,
|
||||
"nodeStates": new_node_states,
|
||||
"workflowVariables": workflow_variables,
|
||||
"status": new_status,
|
||||
"updatedAt": now,
|
||||
}
|
||||
)
|
||||
updated_run = store_context_on_run(updated_run, next_node_id)
|
||||
self._save_run(project_id, run_id, updated_run)
|
||||
return updated_run
|
||||
|
||||
async def send_run_message(
|
||||
self,
|
||||
project_id: str,
|
||||
run_id: str,
|
||||
content: str,
|
||||
*,
|
||||
stream: bool = False,
|
||||
profile_id: Optional[str] = None,
|
||||
api_config: Optional[Dict[str, str]] = None,
|
||||
) -> StudioRun:
|
||||
"""Send a user message to the active worldbook step and invoke LLM (R3)."""
|
||||
trimmed = (content or "").strip()
|
||||
if not trimmed:
|
||||
raise ValueError("消息内容不能为空")
|
||||
|
||||
run, current_node, current_state, current_node_id = self._prepare_run_message(
|
||||
project_id, run_id, trimmed
|
||||
)
|
||||
|
||||
resolved_api = resolve_api_config(profile_id, api_config)
|
||||
prompt_blocks = assemble_prompt_blocks(run, current_node_id)
|
||||
step_messages = list(current_state.stepMessages or [])
|
||||
|
||||
last_draft, last_tool_response, user_msg, assistant_msg = (
|
||||
await studio_step_respond(
|
||||
node=current_node,
|
||||
prompt_blocks=prompt_blocks,
|
||||
step_messages=step_messages,
|
||||
user_message=trimmed,
|
||||
existing_draft=current_state.lastDraft,
|
||||
api_config=resolved_api,
|
||||
stream=stream,
|
||||
)
|
||||
)
|
||||
|
||||
return self._persist_run_message_turn(
|
||||
project_id,
|
||||
run_id,
|
||||
run,
|
||||
current_node_id,
|
||||
prompt_blocks,
|
||||
step_messages,
|
||||
last_draft,
|
||||
last_tool_response,
|
||||
user_msg,
|
||||
assistant_msg,
|
||||
)
|
||||
|
||||
async def send_run_message_stream(
|
||||
self,
|
||||
project_id: str,
|
||||
run_id: str,
|
||||
content: str,
|
||||
*,
|
||||
profile_id: Optional[str] = None,
|
||||
api_config: Optional[Dict[str, str]] = None,
|
||||
) -> AsyncGenerator[Dict[str, Any], None]:
|
||||
"""Stream thinking deltas, then persist and emit complete run (R4)."""
|
||||
trimmed = (content or "").strip()
|
||||
if not trimmed:
|
||||
raise ValueError("消息内容不能为空")
|
||||
|
||||
run, current_node, current_state, current_node_id = self._prepare_run_message(
|
||||
project_id, run_id, trimmed
|
||||
)
|
||||
|
||||
resolved_api = resolve_api_config(profile_id, api_config)
|
||||
prompt_blocks = assemble_prompt_blocks(run, current_node_id)
|
||||
step_messages = list(current_state.stepMessages or [])
|
||||
|
||||
async for event in studio_step_respond_stream(
|
||||
node=current_node,
|
||||
prompt_blocks=prompt_blocks,
|
||||
step_messages=step_messages,
|
||||
user_message=trimmed,
|
||||
existing_draft=current_state.lastDraft,
|
||||
api_config=resolved_api,
|
||||
):
|
||||
if event.get("type") == "thinking_delta":
|
||||
yield event
|
||||
continue
|
||||
|
||||
if event.get("type") == "complete":
|
||||
from models.studio_models import LastToolResponse, StepMessage
|
||||
|
||||
last_draft = event["last_draft"]
|
||||
last_tool_response = LastToolResponse(**event["last_tool_response"])
|
||||
user_msg = StepMessage(**event["user_msg"])
|
||||
assistant_msg = StepMessage(**event["assistant_msg"])
|
||||
|
||||
updated_run = self._persist_run_message_turn(
|
||||
project_id,
|
||||
run_id,
|
||||
run,
|
||||
current_node_id,
|
||||
prompt_blocks,
|
||||
step_messages,
|
||||
last_draft,
|
||||
last_tool_response,
|
||||
user_msg,
|
||||
assistant_msg,
|
||||
)
|
||||
yield {
|
||||
"type": "complete",
|
||||
"run": updated_run.model_dump(mode="json"),
|
||||
}
|
||||
|
||||
def _prepare_run_message(
|
||||
self,
|
||||
project_id: str,
|
||||
run_id: str,
|
||||
trimmed: str,
|
||||
) -> tuple[StudioRun, StudioNode, StudioNodeRunState, str]:
|
||||
run = self.get_run(project_id, run_id)
|
||||
if run.status != StudioRunStatus.RUNNING:
|
||||
raise ValueError("运行未处于进行中,无法发送消息")
|
||||
|
||||
current_node_id = run.currentNodeId
|
||||
if not current_node_id:
|
||||
raise ValueError("当前运行无活动节点")
|
||||
|
||||
current_node = _find_node(run.pipelineSnapshot, current_node_id)
|
||||
if not current_node:
|
||||
raise ValueError(f"节点不存在:{current_node_id}")
|
||||
|
||||
if current_node.skillId != "studio.worldbook_entry":
|
||||
raise ValueError(
|
||||
f"当前步骤「{current_node.displayName}」不支持对话消息"
|
||||
)
|
||||
|
||||
current_state = next(
|
||||
(s for s in run.nodeStates if s.nodeId == current_node_id), None
|
||||
)
|
||||
if not current_state or current_state.status != "active":
|
||||
raise ValueError("当前节点不可执行")
|
||||
|
||||
return run, current_node, current_state, current_node_id
|
||||
|
||||
def _persist_run_message_turn(
|
||||
self,
|
||||
project_id: str,
|
||||
run_id: str,
|
||||
run: StudioRun,
|
||||
current_node_id: str,
|
||||
prompt_blocks: list,
|
||||
step_messages: list,
|
||||
last_draft: Dict[str, Any],
|
||||
last_tool_response,
|
||||
user_msg,
|
||||
assistant_msg,
|
||||
) -> StudioRun:
|
||||
step_messages = list(step_messages)
|
||||
step_messages.extend([user_msg, assistant_msg])
|
||||
now = datetime.now().isoformat()
|
||||
|
||||
new_node_states: list[StudioNodeRunState] = []
|
||||
for state in run.nodeStates:
|
||||
updated = state.model_copy()
|
||||
if state.nodeId == current_node_id:
|
||||
updated.lastDraft = last_draft
|
||||
updated.lastToolResponse = last_tool_response
|
||||
updated.stepMessages = step_messages
|
||||
new_node_states.append(updated)
|
||||
|
||||
updated_run = run.model_copy(
|
||||
update={
|
||||
"nodeStates": new_node_states,
|
||||
"lastPromptBlocks": prompt_blocks,
|
||||
"updatedAt": now,
|
||||
}
|
||||
)
|
||||
updated_run = store_context_on_run(updated_run, current_node_id)
|
||||
self._save_run(project_id, run_id, updated_run)
|
||||
return updated_run
|
||||
|
||||
def delete_run(self, project_id: str, run_id: str) -> None:
|
||||
run_dir = self._run_dir(project_id, run_id)
|
||||
if not run_dir.exists():
|
||||
raise FileNotFoundError(f"Studio run not found: {project_id}/{run_id}")
|
||||
shutil.rmtree(run_dir)
|
||||
|
||||
def rename_run(self, project_id: str, run_id: str, title: str) -> StudioRun:
|
||||
run = self.get_run(project_id, run_id)
|
||||
trimmed = (title or "").strip()
|
||||
if not trimmed:
|
||||
raise ValueError("运行名称不能为空")
|
||||
now = datetime.now().isoformat()
|
||||
updated = run.model_copy(update={"title": trimmed, "updatedAt": now})
|
||||
self._save_run(project_id, run_id, updated)
|
||||
return updated
|
||||
|
||||
|
||||
studio_run_service = StudioRunService()
|
||||
427
backend/services/studio_step_respond.py
Normal file
427
backend/services/studio_step_respond.py
Normal file
@@ -0,0 +1,427 @@
|
||||
"""
|
||||
Studio worldbook step LLM responder (R3/R4).
|
||||
|
||||
Assembles R2 context blocks + short step dialogue, calls LLM for structured JSON,
|
||||
returns thinking, draft, questions, and evaluation.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import re
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
from typing import Any, AsyncGenerator, Dict, List, Optional, Tuple
|
||||
|
||||
from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
|
||||
|
||||
from models.studio_models import (
|
||||
LastToolResponse,
|
||||
PromptBlock,
|
||||
StudioNode,
|
||||
StepMessage,
|
||||
ToolQuestionOption,
|
||||
)
|
||||
from services.studio_context_service import assemble_prompt_blocks
|
||||
from utils.llm_client import LLMClient
|
||||
|
||||
_llm_client = LLMClient()
|
||||
|
||||
_JSON_FENCE = re.compile(r"```(?:json)?\s*([\s\S]*?)```", re.IGNORECASE)
|
||||
|
||||
|
||||
def resolve_api_config(
|
||||
profile_id: Optional[str],
|
||||
api_config: Optional[Dict[str, str]],
|
||||
) -> Dict[str, str]:
|
||||
"""Merge frontend apiConfig with stored profile mainLLM key (same as chat WS)."""
|
||||
resolved = dict(api_config or {})
|
||||
if profile_id:
|
||||
try:
|
||||
try:
|
||||
from api.routes.apiConfigRoute import load_profile
|
||||
except ImportError:
|
||||
from backend.api.routes.apiConfigRoute import load_profile
|
||||
|
||||
profile = load_profile(profile_id)
|
||||
if profile:
|
||||
main_llm = profile.get("apis", {}).get("mainLLM", {})
|
||||
if main_llm.get("apiUrl") and not resolved.get("api_url"):
|
||||
resolved["api_url"] = main_llm.get("apiUrl", "")
|
||||
if main_llm.get("model") and not resolved.get("model"):
|
||||
resolved["model"] = main_llm.get("model", "")
|
||||
api_key = main_llm.get("apiKey", "")
|
||||
if api_key:
|
||||
resolved["api_key"] = api_key
|
||||
except Exception as exc:
|
||||
print(f"[StudioStepRespond] 加载 API 配置失败: {exc}")
|
||||
return resolved
|
||||
|
||||
|
||||
def _blocks_to_context_text(blocks: List[PromptBlock]) -> str:
|
||||
sections: List[str] = []
|
||||
for block in blocks:
|
||||
sections.append(f"## {block.label}\n{block.content}")
|
||||
return "\n\n".join(sections)
|
||||
|
||||
|
||||
def _build_system_prompt(node: StudioNode) -> str:
|
||||
insertion = (node.config or {}).get("insertion") or {}
|
||||
key = insertion.get("key") or "(未配置关键词)"
|
||||
comment = insertion.get("comment") or ""
|
||||
|
||||
return f"""你是 Studio 创作助手,负责为当前流水线步骤生成或修订世界书条目草稿。
|
||||
|
||||
当前步骤:{node.displayName}
|
||||
目标关键词:{key}
|
||||
备注:{comment or "(无)"}
|
||||
|
||||
你必须只输出一个 JSON 对象(不要 markdown 代码块外的其他文字),字段如下:
|
||||
{{
|
||||
"thinking": "你的内部思考过程(逐步推理,中文)",
|
||||
"currentProduct": "世界书条目正文(纯文本或 Markdown,可直接写入条目 content)",
|
||||
"questions": [
|
||||
{{
|
||||
"question": "需要用户澄清的问题",
|
||||
"options": ["选项A", "选项B", "选项C"]
|
||||
}}
|
||||
],
|
||||
"evaluation": "对照评价维度的自检与优化建议(中文,面向用户)"
|
||||
}}
|
||||
|
||||
规则:
|
||||
1. currentProduct 必须是完整、可注入世界书的条目正文。
|
||||
2. questions 为 0–3 条;每条至少 2 个 options;若无需澄清则 questions 为空数组。
|
||||
3. evaluation 需引用上下文中的评价标准,给出具体、可操作的反馈。
|
||||
4. 若用户要求修改,在 currentProduct 中输出修订后的完整条目,而非仅说明改了什么。
|
||||
5. 全部字段使用中文(专有名词除外)。"""
|
||||
|
||||
|
||||
def _dialogue_to_langchain(
|
||||
step_messages: List[StepMessage],
|
||||
) -> List[Any]:
|
||||
messages: List[Any] = []
|
||||
for msg in step_messages:
|
||||
if msg.role == "user":
|
||||
messages.append(HumanMessage(content=msg.content))
|
||||
elif msg.role == "assistant":
|
||||
messages.append(AIMessage(content=msg.content))
|
||||
return messages
|
||||
|
||||
|
||||
def _build_llm_messages(
|
||||
node: StudioNode,
|
||||
prompt_blocks: List[PromptBlock],
|
||||
step_messages: List[StepMessage],
|
||||
user_message: str,
|
||||
) -> List[Any]:
|
||||
context_text = _blocks_to_context_text(prompt_blocks)
|
||||
system_prompt = _build_system_prompt(node)
|
||||
|
||||
messages: List[Any] = [SystemMessage(content=system_prompt)]
|
||||
messages.append(
|
||||
HumanMessage(
|
||||
content=f"以下为当前步骤上下文(不含完整聊天历史):\n\n{context_text}"
|
||||
)
|
||||
)
|
||||
messages.extend(_dialogue_to_langchain(step_messages))
|
||||
messages.append(HumanMessage(content=user_message))
|
||||
return messages
|
||||
|
||||
|
||||
def _validate_api_config(api_config: Dict[str, str]) -> None:
|
||||
if not api_config.get("api_key"):
|
||||
raise ValueError("API Key 未配置,请先在 API 配置页面保存密钥")
|
||||
if not api_config.get("api_url"):
|
||||
raise ValueError("API 地址未配置,请先在 API 配置页面保存 mainLLM")
|
||||
|
||||
|
||||
def _extract_json(raw: str) -> Dict[str, Any]:
|
||||
text = (raw or "").strip()
|
||||
if not text:
|
||||
raise ValueError("模型返回为空")
|
||||
|
||||
fence = _JSON_FENCE.search(text)
|
||||
if fence:
|
||||
text = fence.group(1).strip()
|
||||
|
||||
try:
|
||||
return json.loads(text)
|
||||
except json.JSONDecodeError:
|
||||
start = text.find("{")
|
||||
end = text.rfind("}")
|
||||
if start >= 0 and end > start:
|
||||
return json.loads(text[start : end + 1])
|
||||
raise ValueError("无法解析模型返回的 JSON")
|
||||
|
||||
|
||||
def _decode_json_string_partial(raw: str) -> str:
|
||||
"""Decode a possibly incomplete JSON string body (no surrounding quotes)."""
|
||||
out: List[str] = []
|
||||
i = 0
|
||||
while i < len(raw):
|
||||
if raw[i] == "\\" and i + 1 < len(raw):
|
||||
nxt = raw[i + 1]
|
||||
if nxt == "n":
|
||||
out.append("\n")
|
||||
elif nxt == "t":
|
||||
out.append("\t")
|
||||
elif nxt == "r":
|
||||
out.append("\r")
|
||||
elif nxt == '"':
|
||||
out.append('"')
|
||||
elif nxt == "\\":
|
||||
out.append("\\")
|
||||
elif nxt == "/":
|
||||
out.append("/")
|
||||
elif nxt == "u" and i + 5 < len(raw):
|
||||
try:
|
||||
out.append(chr(int(raw[i + 2 : i + 6], 16)))
|
||||
i += 6
|
||||
continue
|
||||
except ValueError:
|
||||
out.append(nxt)
|
||||
else:
|
||||
out.append(nxt)
|
||||
i += 2
|
||||
else:
|
||||
out.append(raw[i])
|
||||
i += 1
|
||||
return "".join(out)
|
||||
|
||||
|
||||
def _extract_partial_thinking(raw: str) -> Optional[str]:
|
||||
"""Best-effort extraction of thinking field from incomplete JSON stream."""
|
||||
marker = '"thinking"'
|
||||
idx = raw.find(marker)
|
||||
if idx < 0:
|
||||
return None
|
||||
|
||||
colon = raw.find(":", idx + len(marker))
|
||||
if colon < 0:
|
||||
return None
|
||||
|
||||
rest = raw[colon + 1 :].lstrip()
|
||||
if not rest.startswith('"'):
|
||||
return None
|
||||
|
||||
body_start = 1
|
||||
i = body_start
|
||||
while i < len(rest):
|
||||
ch = rest[i]
|
||||
if ch == '"':
|
||||
break
|
||||
if ch == "\\":
|
||||
i += 2
|
||||
continue
|
||||
i += 1
|
||||
|
||||
partial = rest[body_start:i]
|
||||
if not partial:
|
||||
return None
|
||||
return _decode_json_string_partial(partial)
|
||||
|
||||
|
||||
def _normalize_draft(
|
||||
current_product: Any,
|
||||
node: StudioNode,
|
||||
existing_draft: Optional[Dict[str, Any]],
|
||||
) -> Dict[str, Any]:
|
||||
draft: Dict[str, Any] = dict(existing_draft or {})
|
||||
insertion = (node.config or {}).get("insertion") or {}
|
||||
|
||||
if isinstance(current_product, str):
|
||||
draft["entryContent"] = current_product.strip()
|
||||
elif isinstance(current_product, dict):
|
||||
draft.update(current_product)
|
||||
if "entryContent" not in draft and "content" in draft:
|
||||
draft["entryContent"] = draft["content"]
|
||||
else:
|
||||
draft["entryContent"] = str(current_product)
|
||||
|
||||
if insertion.get("key"):
|
||||
draft["insertionKey"] = insertion["key"]
|
||||
if insertion.get("comment"):
|
||||
draft["insertionComment"] = insertion["comment"]
|
||||
draft["nodeId"] = node.id
|
||||
draft["displayName"] = node.displayName
|
||||
return draft
|
||||
|
||||
|
||||
def _normalize_questions(raw: Any) -> List[ToolQuestionOption]:
|
||||
if not isinstance(raw, list):
|
||||
return []
|
||||
result: List[ToolQuestionOption] = []
|
||||
for item in raw:
|
||||
if not isinstance(item, dict):
|
||||
continue
|
||||
question = (item.get("question") or "").strip()
|
||||
if not question:
|
||||
continue
|
||||
options = [
|
||||
str(o).strip()
|
||||
for o in (item.get("options") or [])
|
||||
if str(o).strip()
|
||||
]
|
||||
if len(options) < 2:
|
||||
continue
|
||||
result.append(ToolQuestionOption(question=question, options=options))
|
||||
return result[:3]
|
||||
|
||||
|
||||
def _assistant_message_text(parsed: Dict[str, Any]) -> str:
|
||||
evaluation = (parsed.get("evaluation") or "").strip()
|
||||
if evaluation:
|
||||
return evaluation
|
||||
product = parsed.get("currentProduct")
|
||||
if isinstance(product, str) and product.strip():
|
||||
preview = product.strip()
|
||||
if len(preview) > 400:
|
||||
preview = preview[:400] + "…"
|
||||
return f"已更新条目草稿:\n\n{preview}"
|
||||
return "已处理您的消息,请查看左侧目前产物。"
|
||||
|
||||
|
||||
def _build_turn_result(
|
||||
parsed: Dict[str, Any],
|
||||
*,
|
||||
node: StudioNode,
|
||||
user_message: str,
|
||||
existing_draft: Optional[Dict[str, Any]],
|
||||
) -> Tuple[Dict[str, Any], LastToolResponse, StepMessage, StepMessage]:
|
||||
now = datetime.now().isoformat()
|
||||
|
||||
last_draft = _normalize_draft(
|
||||
parsed.get("currentProduct"),
|
||||
node,
|
||||
existing_draft,
|
||||
)
|
||||
last_tool_response = LastToolResponse(
|
||||
thinking=(parsed.get("thinking") or "").strip() or None,
|
||||
evaluation=(parsed.get("evaluation") or "").strip() or None,
|
||||
questions=_normalize_questions(parsed.get("questions")),
|
||||
generatedAt=now,
|
||||
)
|
||||
|
||||
user_step_msg = StepMessage(
|
||||
id=f"msg-{uuid.uuid4().hex[:12]}",
|
||||
role="user",
|
||||
content=user_message,
|
||||
createdAt=now,
|
||||
)
|
||||
assistant_step_msg = StepMessage(
|
||||
id=f"msg-{uuid.uuid4().hex[:12]}",
|
||||
role="assistant",
|
||||
content=_assistant_message_text(parsed),
|
||||
createdAt=now,
|
||||
)
|
||||
|
||||
return last_draft, last_tool_response, user_step_msg, assistant_step_msg
|
||||
|
||||
|
||||
async def studio_step_respond(
|
||||
*,
|
||||
node: StudioNode,
|
||||
prompt_blocks: List[PromptBlock],
|
||||
step_messages: List[StepMessage],
|
||||
user_message: str,
|
||||
existing_draft: Optional[Dict[str, Any]],
|
||||
api_config: Dict[str, str],
|
||||
stream: bool = False,
|
||||
) -> Tuple[Dict[str, Any], LastToolResponse, StepMessage, StepMessage]:
|
||||
"""
|
||||
Execute one worldbook step turn (non-streaming).
|
||||
|
||||
Returns (last_draft, last_tool_response, user_step_msg, assistant_step_msg).
|
||||
"""
|
||||
_validate_api_config(api_config)
|
||||
messages = _build_llm_messages(node, prompt_blocks, step_messages, user_message)
|
||||
model = api_config.get("model") or "gpt-4o-mini"
|
||||
|
||||
if stream:
|
||||
print("[StudioStepRespond] stream=True 应使用 studio_step_respond_stream")
|
||||
|
||||
response = await _llm_client.chat_completion(
|
||||
messages=messages,
|
||||
api_url=api_config.get("api_url", ""),
|
||||
api_key=api_config.get("api_key", ""),
|
||||
model=model,
|
||||
temperature=0.7,
|
||||
max_tokens=8000,
|
||||
request_timeout=120,
|
||||
stream=False,
|
||||
)
|
||||
|
||||
raw_content = ""
|
||||
if isinstance(response, dict):
|
||||
raw_content = (
|
||||
response.get("choices", [{}])[0]
|
||||
.get("message", {})
|
||||
.get("content", "")
|
||||
)
|
||||
else:
|
||||
raw_content = str(response)
|
||||
|
||||
parsed = _extract_json(raw_content)
|
||||
return _build_turn_result(
|
||||
parsed,
|
||||
node=node,
|
||||
user_message=user_message,
|
||||
existing_draft=existing_draft,
|
||||
)
|
||||
|
||||
|
||||
async def studio_step_respond_stream(
|
||||
*,
|
||||
node: StudioNode,
|
||||
prompt_blocks: List[PromptBlock],
|
||||
step_messages: List[StepMessage],
|
||||
user_message: str,
|
||||
existing_draft: Optional[Dict[str, Any]],
|
||||
api_config: Dict[str, str],
|
||||
) -> AsyncGenerator[Dict[str, Any], None]:
|
||||
"""
|
||||
Stream thinking field while LLM generates structured JSON (R4).
|
||||
|
||||
Yields:
|
||||
- {"type": "thinking_delta", "content": "..."}
|
||||
- {"type": "complete", "last_draft", "last_tool_response", "user_msg", "assistant_msg"}
|
||||
"""
|
||||
_validate_api_config(api_config)
|
||||
messages = _build_llm_messages(node, prompt_blocks, step_messages, user_message)
|
||||
model = api_config.get("model") or "gpt-4o-mini"
|
||||
|
||||
accumulated = ""
|
||||
last_thinking = ""
|
||||
|
||||
async for chunk in _llm_client.stream_chat(
|
||||
messages=messages,
|
||||
api_url=api_config.get("api_url", ""),
|
||||
api_key=api_config.get("api_key", ""),
|
||||
model=model,
|
||||
temperature=0.7,
|
||||
max_tokens=8000,
|
||||
request_timeout=120,
|
||||
):
|
||||
if chunk.get("type") != "chunk":
|
||||
continue
|
||||
accumulated += chunk.get("content") or ""
|
||||
partial = _extract_partial_thinking(accumulated)
|
||||
if partial and partial != last_thinking:
|
||||
last_thinking = partial
|
||||
yield {"type": "thinking_delta", "content": partial}
|
||||
|
||||
parsed = _extract_json(accumulated)
|
||||
last_draft, last_tool_response, user_msg, assistant_msg = _build_turn_result(
|
||||
parsed,
|
||||
node=node,
|
||||
user_message=user_message,
|
||||
existing_draft=existing_draft,
|
||||
)
|
||||
|
||||
yield {
|
||||
"type": "complete",
|
||||
"last_draft": last_draft,
|
||||
"last_tool_response": last_tool_response.model_dump(mode="json"),
|
||||
"user_msg": user_msg.model_dump(mode="json"),
|
||||
"assistant_msg": assistant_msg.model_dump(mode="json"),
|
||||
}
|
||||
49
backend/services/tool_registry.py
Normal file
49
backend/services/tool_registry.py
Normal file
@@ -0,0 +1,49 @@
|
||||
"""
|
||||
Tool registry for workflow engine steps.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
from typing import Any, Awaitable, Callable, Dict, Optional
|
||||
|
||||
try:
|
||||
from backend.models.agent import TurnContext, ToolSpec
|
||||
except ImportError:
|
||||
from models.agent import TurnContext, ToolSpec
|
||||
|
||||
ToolHandler = Callable[[TurnContext], Awaitable[None]]
|
||||
|
||||
|
||||
class ToolRegistry:
|
||||
def __init__(self) -> None:
|
||||
self._tools: Dict[str, ToolHandler] = {}
|
||||
self._specs: Dict[str, ToolSpec] = {}
|
||||
|
||||
def register(
|
||||
self,
|
||||
name: str,
|
||||
handler: ToolHandler,
|
||||
*,
|
||||
description: str = "",
|
||||
parameters: Optional[Dict[str, Any]] = None,
|
||||
) -> None:
|
||||
self._tools[name] = handler
|
||||
self._specs[name] = ToolSpec(
|
||||
name=name,
|
||||
description=description,
|
||||
parameters=parameters or {},
|
||||
)
|
||||
|
||||
def get(self, name: str) -> ToolHandler:
|
||||
if name not in self._tools:
|
||||
raise KeyError(f"Unknown tool: {name}")
|
||||
return self._tools[name]
|
||||
|
||||
def list_specs(self) -> list[ToolSpec]:
|
||||
return list(self._specs.values())
|
||||
|
||||
async def execute(self, name: str, ctx: TurnContext) -> None:
|
||||
handler = self.get(name)
|
||||
await handler(ctx)
|
||||
|
||||
|
||||
default_tool_registry = ToolRegistry()
|
||||
1
backend/services/tools/__init__.py
Normal file
1
backend/services/tools/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Workflow chat tools package."""
|
||||
261
backend/services/tools/chat_tools.py
Normal file
261
backend/services/tools/chat_tools.py
Normal file
@@ -0,0 +1,261 @@
|
||||
"""
|
||||
Chat workflow tools extracted from ChatWorkflowService.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
import time
|
||||
import uuid
|
||||
from typing import Any, Dict, List
|
||||
|
||||
try:
|
||||
from backend.models.agent import TurnContext
|
||||
from backend.models.internal import CharacterCard, TokenUsageStatus
|
||||
from backend.models.regex_rules import RegexPlacement
|
||||
from backend.services.character_service import CharacterService
|
||||
from backend.services.regex_service import regex_service
|
||||
from backend.services.task_queue_manager import TaskType, task_queue_manager
|
||||
from backend.services.token_usage_service import token_usage_service
|
||||
from backend.core.config import settings
|
||||
except ImportError:
|
||||
from models.agent import TurnContext
|
||||
from models.internal import CharacterCard, TokenUsageStatus
|
||||
from models.regex_rules import RegexPlacement
|
||||
from services.character_service import CharacterService
|
||||
from services.regex_service import regex_service
|
||||
from services.task_queue_manager import TaskType, task_queue_manager
|
||||
from services.token_usage_service import token_usage_service
|
||||
from core.config import settings
|
||||
|
||||
|
||||
_character_service = CharacterService()
|
||||
_workflow_service = None
|
||||
|
||||
|
||||
def _get_workflow_service():
|
||||
"""Lazy init to avoid circular import with chat_workflow_service."""
|
||||
global _workflow_service
|
||||
if _workflow_service is None:
|
||||
try:
|
||||
from backend.services.chat_workflow_service import ChatWorkflowService
|
||||
except ImportError:
|
||||
from services.chat_workflow_service import ChatWorkflowService
|
||||
_workflow_service = ChatWorkflowService()
|
||||
return _workflow_service
|
||||
|
||||
|
||||
async def regex_apply_user_input(ctx: TurnContext) -> None:
|
||||
processed = regex_service.apply_rules_by_placement(
|
||||
text=ctx.user_message,
|
||||
placement=RegexPlacement.USER_INPUT.value,
|
||||
character_name=ctx.current_role,
|
||||
preset_name=ctx.preset_name,
|
||||
message_depth=0,
|
||||
is_for_llm=True,
|
||||
is_markdown_rendered=False,
|
||||
)
|
||||
if processed != ctx.user_message:
|
||||
print("[WorkflowTool] Applied user-input regex rules")
|
||||
ctx.user_message = processed
|
||||
|
||||
|
||||
async def load_character(ctx: TurnContext) -> None:
|
||||
character_data = ctx.request_data.get("characterData")
|
||||
if not character_data:
|
||||
character = _character_service.get_character_by_name(ctx.current_role)
|
||||
if not character:
|
||||
raise ValueError(f"角色 '{ctx.current_role}' 不存在")
|
||||
else:
|
||||
character = CharacterCard(**character_data)
|
||||
ctx.character = character
|
||||
print(f"[WorkflowTool] Loaded character: {character.name}")
|
||||
|
||||
|
||||
async def activate_worldbook(ctx: TurnContext) -> None:
|
||||
svc = _get_workflow_service()
|
||||
active_entries = await svc._collect_and_activate_worldbooks(
|
||||
ctx.request_data,
|
||||
ctx.character,
|
||||
)
|
||||
ctx.active_entries = active_entries
|
||||
print(f"[WorkflowTool] Activated {len(active_entries)} worldbook entries")
|
||||
|
||||
if ctx.callbacks and ctx.callbacks.on_worldbook_active:
|
||||
entries_payload = [
|
||||
entry.model_dump() if hasattr(entry, "model_dump") else entry
|
||||
for entry in active_entries
|
||||
]
|
||||
await ctx.callbacks.on_worldbook_active(entries_payload)
|
||||
|
||||
|
||||
async def load_chat_history(ctx: TurnContext) -> None:
|
||||
svc = _get_workflow_service()
|
||||
chat_history = await svc._load_chat_history(
|
||||
ctx.current_role,
|
||||
ctx.current_chat,
|
||||
)
|
||||
ctx.chat_history = chat_history
|
||||
print(f"[WorkflowTool] Loaded {len(chat_history)} history messages")
|
||||
|
||||
|
||||
async def build_prompt_messages(ctx: TurnContext) -> None:
|
||||
svc = _get_workflow_service()
|
||||
prompt_messages = svc._assemble_prompt(
|
||||
ctx.character,
|
||||
ctx.chat_history,
|
||||
ctx.user_message,
|
||||
ctx.active_entries,
|
||||
ctx.request_data,
|
||||
)
|
||||
ctx.prompt_messages = prompt_messages
|
||||
print(f"[WorkflowTool] Built {len(prompt_messages)} prompt messages")
|
||||
|
||||
|
||||
async def llm_main_reply(ctx: TurnContext) -> None:
|
||||
svc = _get_workflow_service()
|
||||
api_config = ctx.request_data.get("apiConfig", {})
|
||||
preset_config = ctx.request_data.get("presetConfig", {})
|
||||
|
||||
if ctx.stream:
|
||||
if not api_config.get("api_key"):
|
||||
raise ValueError("API Key 未配置,请先在 API 配置页面保存密钥")
|
||||
|
||||
generated_content = ""
|
||||
chunk_count = 0
|
||||
start_time = time.time()
|
||||
|
||||
async for chunk_dict in svc.llm_client.stream_chat(
|
||||
messages=ctx.prompt_messages,
|
||||
api_url=api_config.get("api_url", ""),
|
||||
api_key=api_config.get("api_key", ""),
|
||||
model=api_config.get("model", ""),
|
||||
temperature=preset_config.get("parameters", {}).get("temperature", 1.0),
|
||||
max_tokens=preset_config.get("parameters", {}).get("max_tokens", 30000),
|
||||
request_timeout=preset_config.get("parameters", {}).get("request_timeout", 60),
|
||||
):
|
||||
if isinstance(chunk_dict, dict):
|
||||
if chunk_dict.get("type") == "chunk":
|
||||
chunk_content = chunk_dict.get("content", "")
|
||||
elif chunk_dict.get("type") == "usage":
|
||||
continue
|
||||
else:
|
||||
chunk_content = chunk_dict.get("content", str(chunk_dict))
|
||||
else:
|
||||
chunk_content = str(chunk_dict)
|
||||
|
||||
generated_content += chunk_content
|
||||
chunk_count += 1
|
||||
|
||||
if ctx.callbacks and ctx.callbacks.on_chunk:
|
||||
await ctx.callbacks.on_chunk(chunk_content)
|
||||
|
||||
ctx.duration = time.time() - start_time
|
||||
ctx.generated_content = generated_content
|
||||
ctx.token_usage = {
|
||||
"prompt_tokens": len(str(ctx.prompt_messages)) // 4,
|
||||
"completion_tokens": len(generated_content) // 4,
|
||||
"total_tokens": (len(str(ctx.prompt_messages)) // 4)
|
||||
+ (len(generated_content) // 4),
|
||||
}
|
||||
print(
|
||||
f"[WorkflowTool] Stream LLM complete: {chunk_count} chunks, "
|
||||
f"{len(generated_content)} chars"
|
||||
)
|
||||
else:
|
||||
result = await svc._generate_response(
|
||||
ctx.prompt_messages,
|
||||
api_config,
|
||||
preset_config,
|
||||
stream=False,
|
||||
)
|
||||
ctx.generated_content = result["content"]
|
||||
ctx.token_usage = result.get("usage", {})
|
||||
ctx.duration = result.get("duration", 0.0)
|
||||
print(f"[WorkflowTool] LLM complete: {len(ctx.generated_content)} chars")
|
||||
|
||||
|
||||
async def regex_apply_ai_output(ctx: TurnContext) -> None:
|
||||
processed = regex_service.apply_rules_by_placement(
|
||||
text=ctx.generated_content,
|
||||
placement=RegexPlacement.AI_OUTPUT.value,
|
||||
character_name=ctx.current_role,
|
||||
preset_name=ctx.preset_name,
|
||||
message_depth=0,
|
||||
is_for_llm=False,
|
||||
is_markdown_rendered=False,
|
||||
)
|
||||
if processed != ctx.generated_content:
|
||||
print("[WorkflowTool] Applied AI-output regex rules")
|
||||
ctx.generated_content = processed
|
||||
|
||||
|
||||
async def record_token_usage(ctx: TurnContext) -> None:
|
||||
chat_id = f"{ctx.current_role}/{ctx.current_chat}"
|
||||
floor = ctx.request_data.get("floor", 0)
|
||||
api_config = ctx.request_data.get("apiConfig", {})
|
||||
|
||||
try:
|
||||
await token_usage_service.record_usage(
|
||||
chat_id=chat_id,
|
||||
role_name=ctx.current_role,
|
||||
chat_name=ctx.current_chat,
|
||||
prompt_tokens=ctx.token_usage.get("prompt_tokens", 0),
|
||||
completion_tokens=ctx.token_usage.get("completion_tokens", 0),
|
||||
total_tokens=ctx.token_usage.get("total_tokens", 0),
|
||||
status=TokenUsageStatus.COMPLETED,
|
||||
floor=floor + 1,
|
||||
duration=ctx.duration,
|
||||
model=api_config.get("model"),
|
||||
api_provider="openai",
|
||||
api_url=api_config.get("api_url"),
|
||||
)
|
||||
except Exception as exc:
|
||||
print(f"[WorkflowTool] Token usage recording failed: {exc}")
|
||||
|
||||
|
||||
async def enqueue_parallel_tasks(ctx: TurnContext) -> None:
|
||||
chat_id = f"{ctx.current_role}/{ctx.current_chat}"
|
||||
options = ctx.request_data.get("options", {})
|
||||
image_task_id = None
|
||||
table_task_id = None
|
||||
|
||||
if options.get("imageWorkflow", False):
|
||||
image_task_id = f"img_{uuid.uuid4().hex[:8]}"
|
||||
await task_queue_manager.add_task(image_task_id, TaskType.IMAGE_WORKFLOW, chat_id)
|
||||
|
||||
if options.get("dynamicTable", False):
|
||||
table_task_id = f"tbl_{uuid.uuid4().hex[:8]}"
|
||||
await task_queue_manager.add_task(table_task_id, TaskType.DYNAMIC_TABLE, chat_id)
|
||||
|
||||
ctx.task_ids = {
|
||||
"imageWorkflow": image_task_id,
|
||||
"dynamicTable": table_task_id,
|
||||
}
|
||||
|
||||
if ctx.callbacks and ctx.callbacks.on_tasks_created:
|
||||
if image_task_id or table_task_id:
|
||||
await ctx.callbacks.on_tasks_created(ctx.task_ids)
|
||||
|
||||
# Fire-and-forget parallel workers (same as legacy service)
|
||||
svc = _get_workflow_service()
|
||||
asyncio.create_task(
|
||||
svc._start_parallel_tasks(
|
||||
ctx.request_data,
|
||||
ctx.generated_content,
|
||||
image_task_id,
|
||||
table_task_id,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def register_chat_tools(registry) -> None:
|
||||
"""Register all chat workflow tools on the given registry."""
|
||||
registry.register("regex_apply_user_input", regex_apply_user_input, description="Apply user-input regex")
|
||||
registry.register("load_character", load_character, description="Load character card")
|
||||
registry.register("activate_worldbook", activate_worldbook, description="Activate worldbook entries")
|
||||
registry.register("load_chat_history", load_chat_history, description="Load chat history")
|
||||
registry.register("build_prompt_messages", build_prompt_messages, description="Assemble LLM prompt")
|
||||
registry.register("llm_main_reply", llm_main_reply, description="Call main LLM (supports stream)")
|
||||
registry.register("regex_apply_ai_output", regex_apply_ai_output, description="Apply AI-output regex")
|
||||
registry.register("record_token_usage", record_token_usage, description="Persist token usage")
|
||||
registry.register("enqueue_parallel_tasks", enqueue_parallel_tasks, description="Enqueue parallel tasks")
|
||||
170
backend/services/workflow_engine.py
Normal file
170
backend/services/workflow_engine.py
Normal file
@@ -0,0 +1,170 @@
|
||||
"""
|
||||
Workflow engine – orchestrates template loading, state machine execution, and run persistence.
|
||||
"""
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
from typing import Any, Awaitable, Callable, Dict, List, Optional
|
||||
|
||||
try:
|
||||
from backend.core.config import settings
|
||||
from backend.models.agent import (
|
||||
ChatRunBinding,
|
||||
ChatTurnResult,
|
||||
RunEvent,
|
||||
RunStatus,
|
||||
TurnCallbacks,
|
||||
TurnContext,
|
||||
WorkflowRun,
|
||||
WorkflowTemplate,
|
||||
WorkflowTemplateKind,
|
||||
)
|
||||
from backend.services.state_machine_runner import StateMachineRunner
|
||||
from backend.services.tool_registry import ToolRegistry, default_tool_registry
|
||||
from backend.services.tools.chat_tools import register_chat_tools
|
||||
except ImportError:
|
||||
from core.config import settings
|
||||
from models.agent import (
|
||||
ChatRunBinding,
|
||||
ChatTurnResult,
|
||||
RunEvent,
|
||||
RunStatus,
|
||||
TurnCallbacks,
|
||||
TurnContext,
|
||||
WorkflowRun,
|
||||
WorkflowTemplate,
|
||||
WorkflowTemplateKind,
|
||||
)
|
||||
from services.state_machine_runner import StateMachineRunner
|
||||
from services.tool_registry import ToolRegistry, default_tool_registry
|
||||
from services.tools.chat_tools import register_chat_tools
|
||||
|
||||
|
||||
class WorkflowEngine:
|
||||
def __init__(self, registry: Optional[ToolRegistry] = None) -> None:
|
||||
self.registry = registry or default_tool_registry
|
||||
if not self.registry.list_specs():
|
||||
register_chat_tools(self.registry)
|
||||
|
||||
def _template_dir(self, template_id: str) -> Path:
|
||||
return settings.AGENT_TEMPLATES_PATH / template_id
|
||||
|
||||
def load_template(self, template_id: str = WorkflowTemplateKind.BUILTIN_CHAT.value) -> WorkflowTemplate:
|
||||
template_path = self._template_dir(template_id) / "template.json"
|
||||
with open(template_path, "r", encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
return WorkflowTemplate(**data)
|
||||
|
||||
def _run_dir(self, role_name: str, chat_name: str) -> Path:
|
||||
return settings.AGENT_RUNS_PATH / "chat" / role_name / chat_name
|
||||
|
||||
def _persist_run(self, run: WorkflowRun, events: List[RunEvent]) -> None:
|
||||
run_dir = self._run_dir(run.binding.role_name, run.binding.chat_name)
|
||||
run_dir.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
run_file = run_dir / "run.json"
|
||||
run.finished_at = datetime.now().isoformat()
|
||||
with open(run_file, "w", encoding="utf-8") as f:
|
||||
json.dump(run.model_dump(), f, ensure_ascii=False, indent=2)
|
||||
|
||||
events_file = run_dir / "events.jsonl"
|
||||
with open(events_file, "a", encoding="utf-8") as f:
|
||||
for event in events:
|
||||
f.write(json.dumps(event.model_dump(), ensure_ascii=False) + "\n")
|
||||
|
||||
async def run_turn(
|
||||
self,
|
||||
request_data: Dict[str, Any],
|
||||
*,
|
||||
stream: bool = False,
|
||||
on_chunk: Optional[Callable[[str], Awaitable[None]]] = None,
|
||||
on_worldbook_active: Optional[Callable[[List[Any]], Awaitable[None]]] = None,
|
||||
on_tasks_created: Optional[Callable[[Dict[str, Any]], Awaitable[None]]] = None,
|
||||
template_id: str = WorkflowTemplateKind.BUILTIN_CHAT.value,
|
||||
) -> ChatTurnResult:
|
||||
current_role = request_data.get("currentRole", "")
|
||||
current_chat = request_data.get("currentChat", "")
|
||||
user_message = request_data.get("mes", "")
|
||||
|
||||
if not current_role or not user_message:
|
||||
return ChatTurnResult(
|
||||
success=False,
|
||||
error="缺少必要的参数:currentRole 或 mes",
|
||||
workflow_template_id=template_id,
|
||||
)
|
||||
|
||||
preset_config = request_data.get("presetConfig", {})
|
||||
preset_name = preset_config.get("selectedPreset")
|
||||
|
||||
run_id = uuid.uuid4().hex
|
||||
binding = ChatRunBinding(
|
||||
role_name=current_role,
|
||||
chat_name=current_chat or "",
|
||||
template_id=template_id,
|
||||
)
|
||||
run = WorkflowRun(
|
||||
id=run_id,
|
||||
template_id=template_id,
|
||||
binding=binding,
|
||||
status=RunStatus.PENDING,
|
||||
)
|
||||
|
||||
callbacks = TurnCallbacks(
|
||||
on_chunk=on_chunk,
|
||||
on_worldbook_active=on_worldbook_active,
|
||||
on_tasks_created=on_tasks_created,
|
||||
)
|
||||
|
||||
ctx = TurnContext(
|
||||
request_data=request_data,
|
||||
template_id=template_id,
|
||||
run_id=run_id,
|
||||
stream=stream,
|
||||
callbacks=callbacks,
|
||||
current_role=current_role,
|
||||
current_chat=current_chat or "",
|
||||
user_message=user_message,
|
||||
preset_name=preset_name,
|
||||
)
|
||||
|
||||
template = self.load_template(template_id)
|
||||
sm_path = self._template_dir(template_id) / template.state_machine_path
|
||||
runner = StateMachineRunner.from_file(sm_path, self.registry)
|
||||
|
||||
try:
|
||||
events = await runner.run(run, ctx)
|
||||
self._persist_run(run, events)
|
||||
|
||||
active_entries = [
|
||||
entry.model_dump() if hasattr(entry, "model_dump") else entry
|
||||
for entry in ctx.active_entries
|
||||
]
|
||||
|
||||
return ChatTurnResult(
|
||||
success=True,
|
||||
content=ctx.generated_content,
|
||||
active_entries=active_entries,
|
||||
task_ids=ctx.task_ids,
|
||||
run_id=run_id,
|
||||
workflow_template_id=template_id,
|
||||
)
|
||||
except Exception as exc:
|
||||
run.status = RunStatus.FAILED
|
||||
run.error = str(exc)
|
||||
try:
|
||||
self._persist_run(run, [])
|
||||
except Exception:
|
||||
pass
|
||||
return ChatTurnResult(
|
||||
success=False,
|
||||
error=f"工作流执行失败: {exc}",
|
||||
run_id=run_id,
|
||||
workflow_template_id=template_id,
|
||||
)
|
||||
|
||||
|
||||
# Module-level singleton
|
||||
workflow_engine = WorkflowEngine()
|
||||
16
data/agent/niches.json
Normal file
16
data/agent/niches.json
Normal file
@@ -0,0 +1,16 @@
|
||||
{
|
||||
"niches": [
|
||||
{
|
||||
"id": "aesthetic_tone",
|
||||
"label": "整体美学",
|
||||
"description": "视觉、氛围、叙事基调等宏观美学设定",
|
||||
"suggestedStepGoal": "描述角色的整体美学:色调、材质、氛围、叙事基调,供后续人设与世界书条目引用。"
|
||||
},
|
||||
{
|
||||
"id": "persona_detail",
|
||||
"label": "具体人设",
|
||||
"description": "性格、口癖、关系、行为模式等可扮演细节",
|
||||
"suggestedStepGoal": "在整体美学基础上,细化可扮演的人设:性格、动机、口癖、与他人关系。"
|
||||
}
|
||||
]
|
||||
}
|
||||
73
data/agent/skill_templates.json
Normal file
73
data/agent/skill_templates.json
Normal file
@@ -0,0 +1,73 @@
|
||||
{
|
||||
"templates": [
|
||||
{
|
||||
"skillId": "studio.init_bind",
|
||||
"displayName": "创建并绑定",
|
||||
"description": "创建角色卡与世界书,并绑定到当前 Studio 项目",
|
||||
"displayParams": [
|
||||
{
|
||||
"key": "characterName",
|
||||
"label": "角色卡名称",
|
||||
"type": "text",
|
||||
"required": true,
|
||||
"placeholder": "例如:帝国骑士维尔"
|
||||
},
|
||||
{
|
||||
"key": "worldbookName",
|
||||
"label": "世界书名称",
|
||||
"type": "text",
|
||||
"required": true,
|
||||
"placeholder": "例如:维尔的世界观"
|
||||
}
|
||||
],
|
||||
"configWhitelist": [],
|
||||
"artifacts": [],
|
||||
"supportsLoopUntilSatisfied": false,
|
||||
"supportsInputs": false,
|
||||
"supportsInsertion": false,
|
||||
"supportsScoring": false
|
||||
},
|
||||
{
|
||||
"skillId": "studio.worldbook_entry",
|
||||
"displayName": "创作世界书条目",
|
||||
"description": "根据 stepGoal 与上文引用,生成并写入世界书条目",
|
||||
"displayParams": [],
|
||||
"configWhitelist": [
|
||||
"stepGoal",
|
||||
"thinkingPrompt",
|
||||
"insertion.position",
|
||||
"insertion.activationType",
|
||||
"insertion.key",
|
||||
"insertion.keysecondary",
|
||||
"insertion.ragConfig",
|
||||
"insertion.comment",
|
||||
"scoring"
|
||||
],
|
||||
"artifacts": [
|
||||
{
|
||||
"type": "worldbook.entries",
|
||||
"displayName": "世界书条目"
|
||||
}
|
||||
],
|
||||
"supportsLoopUntilSatisfied": true,
|
||||
"supportsInputs": true,
|
||||
"supportsInsertion": true,
|
||||
"supportsScoring": true,
|
||||
"configDefaults": {
|
||||
"stepGoal": "",
|
||||
"thinkingPrompt": "====== 思考流程 ======\nStep1: 简短确认任务性质(新设计/修改)\nStep2: 阅读绑定角色/世界书与上文引用\nStep3: 按步骤目标起草世界书条目\nStep4: 对照评价维度自检并优化表述\n\n(核心目的、评价标准与优化建议由系统在运行时自动注入,无需在此填写)",
|
||||
"insertion": {
|
||||
"position": 1,
|
||||
"activationType": "permanent",
|
||||
"key": "",
|
||||
"keysecondary": "",
|
||||
"comment": ""
|
||||
},
|
||||
"scoring": {
|
||||
"enabled": true,
|
||||
"dimensions": []
|
||||
}
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
10
data/agent/studio_projects/default/meta.json
Normal file
10
data/agent/studio_projects/default/meta.json
Normal file
@@ -0,0 +1,10 @@
|
||||
{
|
||||
"id": "default",
|
||||
"name": "单人类角色卡",
|
||||
"description": "从创建绑定到世界书条目的默认三步流水线,可在工作流编辑页复制并修改。",
|
||||
"templateId": "builtin.studio.example",
|
||||
"characterId": "f04ba2d6-1ffd-4c33-9cfe-cf6fd026c175",
|
||||
"worldbookId": "8682f790-1b9d-4842-826b-c07764be6b9b",
|
||||
"createdAt": "2026-05-31T00:00:00",
|
||||
"updatedAt": "2026-05-31T13:11:55.525055"
|
||||
}
|
||||
122
data/agent/studio_projects/default/pipeline.json
Normal file
122
data/agent/studio_projects/default/pipeline.json
Normal file
@@ -0,0 +1,122 @@
|
||||
{
|
||||
"workflowGoal": "设计一个单人角色:先绑定角色卡与世界书,再迭代整体美学与具体人设,写入世界书条目。",
|
||||
"nodes": [
|
||||
{
|
||||
"id": "init",
|
||||
"skillId": "studio.init_bind",
|
||||
"displayName": "创建并绑定",
|
||||
"enabled": true,
|
||||
"loopUntilSatisfied": false,
|
||||
"config": {},
|
||||
"displayParams": [
|
||||
{
|
||||
"key": "characterName",
|
||||
"label": "角色卡名称",
|
||||
"type": "text",
|
||||
"required": true,
|
||||
"placeholder": ""
|
||||
},
|
||||
{
|
||||
"key": "worldbookName",
|
||||
"label": "世界书名称",
|
||||
"type": "text",
|
||||
"required": true,
|
||||
"placeholder": ""
|
||||
}
|
||||
],
|
||||
"inputs": []
|
||||
},
|
||||
{
|
||||
"id": "aesthetic",
|
||||
"skillId": "studio.worldbook_entry",
|
||||
"displayName": "整体美学",
|
||||
"enabled": true,
|
||||
"niche": "aesthetic_tone",
|
||||
"loopUntilSatisfied": true,
|
||||
"config": {
|
||||
"stepGoal": "产出角色的整体美学设定:视觉风格、氛围、叙事基调,供后续人设步骤引用。",
|
||||
"thinkingPrompt": "step1:首先思考整个故事是怎么样的\nstep2:然后思考如何展示\nstep3:选中核心爽点",
|
||||
"insertion": {
|
||||
"position": 1,
|
||||
"activationType": "permanent",
|
||||
"key": "整体美学",
|
||||
"comment": "Studio · 整体美学"
|
||||
},
|
||||
"scoring": {
|
||||
"enabled": true,
|
||||
"dimensions": [
|
||||
{
|
||||
"id": "authenticity",
|
||||
"name": "真实性",
|
||||
"criteria": "美学设定是否自洽、可感知,而非空泛形容词堆砌。"
|
||||
},
|
||||
{
|
||||
"id": "fit",
|
||||
"name": "贴合度",
|
||||
"criteria": "是否覆盖色调/材质/氛围/叙事基调;是否可与后续人设衔接;表述是否简洁可注入世界书。"
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"displayParams": [],
|
||||
"inputs": [
|
||||
{
|
||||
"ref": "workflow.goal",
|
||||
"label": "工作流目标文本",
|
||||
"optional": false
|
||||
}
|
||||
]
|
||||
},
|
||||
{
|
||||
"id": "persona",
|
||||
"skillId": "studio.worldbook_entry",
|
||||
"displayName": "具体人设",
|
||||
"enabled": true,
|
||||
"niche": "persona_detail",
|
||||
"loopUntilSatisfied": true,
|
||||
"config": {
|
||||
"stepGoal": "在整体美学基础上,写出可扮演的人设:性格、动机、口癖、关系与行为模式。",
|
||||
"thinkingPrompt": "",
|
||||
"insertion": {
|
||||
"position": 1,
|
||||
"activationType": "keyword",
|
||||
"key": "具体人设",
|
||||
"comment": "Studio · 具体人设"
|
||||
},
|
||||
"scoring": {
|
||||
"enabled": true,
|
||||
"dimensions": [
|
||||
{
|
||||
"id": "authenticity",
|
||||
"name": "真实性",
|
||||
"criteria": "人设细节是否具体、可扮演,动机与行为是否一致。"
|
||||
},
|
||||
{
|
||||
"id": "fit",
|
||||
"name": "贴合度",
|
||||
"criteria": "是否与人设目标一致;是否与整体美学一致;是否避免与已有条目冲突。"
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"displayParams": [],
|
||||
"inputs": [
|
||||
{
|
||||
"ref": "workflow.goal",
|
||||
"label": "工作流目标文本",
|
||||
"optional": false
|
||||
},
|
||||
{
|
||||
"ref": "aesthetic.output",
|
||||
"label": "整体美学 · 上轮产物",
|
||||
"optional": false
|
||||
},
|
||||
{
|
||||
"ref": "persona.output",
|
||||
"label": "具体人设 · 上轮产物",
|
||||
"optional": true
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
@@ -0,0 +1,5 @@
|
||||
# Chat Reply Skill
|
||||
|
||||
Minimal skill placeholder for the builtin.chat workflow template.
|
||||
|
||||
This skill orchestrates a single chat turn: load character, activate worldbooks, assemble prompt, call LLM, apply regex, record usage, and enqueue parallel tasks.
|
||||
@@ -0,0 +1,3 @@
|
||||
id: chat_reply
|
||||
name: Chat Reply
|
||||
description: Minimal chat reply skill for builtin.chat template
|
||||
41
data/agent/templates/builtin.chat/state_machine.json
Normal file
41
data/agent/templates/builtin.chat/state_machine.json
Normal file
@@ -0,0 +1,41 @@
|
||||
{
|
||||
"initial": "regex_apply_user_input",
|
||||
"states": {
|
||||
"regex_apply_user_input": {
|
||||
"tool": "regex_apply_user_input",
|
||||
"next": "load_character"
|
||||
},
|
||||
"load_character": {
|
||||
"tool": "load_character",
|
||||
"next": "activate_worldbook"
|
||||
},
|
||||
"activate_worldbook": {
|
||||
"tool": "activate_worldbook",
|
||||
"next": "load_chat_history"
|
||||
},
|
||||
"load_chat_history": {
|
||||
"tool": "load_chat_history",
|
||||
"next": "build_prompt_messages"
|
||||
},
|
||||
"build_prompt_messages": {
|
||||
"tool": "build_prompt_messages",
|
||||
"next": "llm_main_reply"
|
||||
},
|
||||
"llm_main_reply": {
|
||||
"tool": "llm_main_reply",
|
||||
"next": "regex_apply_ai_output"
|
||||
},
|
||||
"regex_apply_ai_output": {
|
||||
"tool": "regex_apply_ai_output",
|
||||
"next": "record_token_usage"
|
||||
},
|
||||
"record_token_usage": {
|
||||
"tool": "record_token_usage",
|
||||
"next": "enqueue_parallel_tasks"
|
||||
},
|
||||
"enqueue_parallel_tasks": {
|
||||
"tool": "enqueue_parallel_tasks",
|
||||
"next": "end"
|
||||
}
|
||||
}
|
||||
}
|
||||
16
data/agent/templates/builtin.chat/template.json
Normal file
16
data/agent/templates/builtin.chat/template.json
Normal file
@@ -0,0 +1,16 @@
|
||||
{
|
||||
"id": "builtin.chat",
|
||||
"kind": "builtin.chat",
|
||||
"name": "Builtin Chat Reply",
|
||||
"description": "Default single-turn chat workflow migrated from ChatWorkflowService",
|
||||
"version": "1.0.0",
|
||||
"state_machine_path": "state_machine.json",
|
||||
"skills": [
|
||||
{
|
||||
"id": "chat_reply",
|
||||
"name": "Chat Reply",
|
||||
"description": "Main chat reply skill",
|
||||
"path": "skill/chat_reply"
|
||||
}
|
||||
]
|
||||
}
|
||||
6
data/agent/templates/builtin.studio.example/meta.json
Normal file
6
data/agent/templates/builtin.studio.example/meta.json
Normal file
@@ -0,0 +1,6 @@
|
||||
{
|
||||
"id": "builtin.studio.example",
|
||||
"name": "世界书条目创建",
|
||||
"description": "从创建绑定到世界书条目的默认三步流水线,可在工作流编辑页复制并修改。",
|
||||
"version": "1.0.0"
|
||||
}
|
||||
102
data/agent/templates/builtin.studio.example/pipeline.json
Normal file
102
data/agent/templates/builtin.studio.example/pipeline.json
Normal file
@@ -0,0 +1,102 @@
|
||||
{
|
||||
"workflowGoal": "设计一个单人角色:先绑定角色卡与世界书,再迭代整体美学与具体人设,写入世界书条目。",
|
||||
"nodes": [
|
||||
{
|
||||
"id": "init",
|
||||
"skillId": "studio.init_bind",
|
||||
"displayName": "创建并绑定",
|
||||
"enabled": true,
|
||||
"config": {},
|
||||
"displayParams": [
|
||||
{
|
||||
"key": "characterName",
|
||||
"label": "角色卡名称",
|
||||
"type": "text",
|
||||
"required": true,
|
||||
"placeholder": ""
|
||||
},
|
||||
{
|
||||
"key": "worldbookName",
|
||||
"label": "世界书名称",
|
||||
"type": "text",
|
||||
"required": true,
|
||||
"placeholder": ""
|
||||
}
|
||||
],
|
||||
"inputs": []
|
||||
},
|
||||
{
|
||||
"id": "aesthetic",
|
||||
"skillId": "studio.worldbook_entry",
|
||||
"displayName": "整体美学",
|
||||
"enabled": true,
|
||||
"loopUntilSatisfied": true,
|
||||
"config": {
|
||||
"stepGoal": "产出角色的整体美学设定:视觉风格、氛围、叙事基调,供后续人设步骤引用。",
|
||||
"thinkingPrompt": "====== 思考流程 ======\nStep1: 简短确认任务性质(新设计/修改)\nStep2: 阅读绑定角色/世界书与上文引用\nStep3: 按步骤目标起草世界书条目\nStep4: 对照评价维度自检并优化表述\n\n(核心目的、评价标准与优化建议由系统在运行时自动注入,无需在此填写)",
|
||||
"insertion": {
|
||||
"position": 1,
|
||||
"activationType": "permanent",
|
||||
"key": "整体美学",
|
||||
"comment": "Studio · 整体美学"
|
||||
},
|
||||
"scoring": {
|
||||
"enabled": true,
|
||||
"dimensions": [
|
||||
{
|
||||
"id": "authenticity",
|
||||
"name": "真实性",
|
||||
"criteria": "美学设定是否自洽、可感知,而非空泛形容词堆砌。"
|
||||
},
|
||||
{
|
||||
"id": "fit",
|
||||
"name": "贴合度",
|
||||
"criteria": "是否覆盖色调/材质/氛围/叙事基调;是否可与后续人设衔接;表述是否简洁可注入世界书。"
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"displayParams": [],
|
||||
"inputs": []
|
||||
},
|
||||
{
|
||||
"id": "persona",
|
||||
"skillId": "studio.worldbook_entry",
|
||||
"displayName": "具体人设",
|
||||
"enabled": true,
|
||||
"loopUntilSatisfied": true,
|
||||
"config": {
|
||||
"stepGoal": "在整体美学基础上,写出可扮演的人设:性格、动机、口癖、关系与行为模式。",
|
||||
"thinkingPrompt": "====== 思考流程 ======\nStep1: 简短确认任务性质(新设计/修改)\nStep2: 阅读绑定角色/世界书与上文引用\nStep3: 按步骤目标起草世界书条目\nStep4: 对照评价维度自检并优化表述\n\n(核心目的、评价标准与优化建议由系统在运行时自动注入,无需在此填写)",
|
||||
"insertion": {
|
||||
"position": 1,
|
||||
"activationType": "keyword",
|
||||
"key": "具体人设",
|
||||
"comment": "Studio · 具体人设"
|
||||
},
|
||||
"scoring": {
|
||||
"enabled": true,
|
||||
"dimensions": [
|
||||
{
|
||||
"id": "authenticity",
|
||||
"name": "真实性",
|
||||
"criteria": "人设细节是否具体、可扮演,动机与行为是否一致。"
|
||||
},
|
||||
{
|
||||
"id": "fit",
|
||||
"name": "贴合度",
|
||||
"criteria": "是否与人设目标一致;是否与整体美学一致;是否避免与已有条目冲突。"
|
||||
}
|
||||
]
|
||||
}
|
||||
},
|
||||
"displayParams": [],
|
||||
"inputs": [
|
||||
{
|
||||
"ref": "aesthetic.output",
|
||||
"label": "整体美学 · 世界书条目"
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
56
data/agent/workflow_variables.json
Normal file
56
data/agent/workflow_variables.json
Normal file
@@ -0,0 +1,56 @@
|
||||
{
|
||||
"builtIn": [
|
||||
{
|
||||
"ref": "workflow.goal",
|
||||
"label": "工作流目标文本",
|
||||
"description": "当前 Studio 项目的 workflowGoal 全文(系统自动注入,不可手动选择)",
|
||||
"autoInjected": true
|
||||
},
|
||||
{
|
||||
"ref": "workflow.boundWorldbook",
|
||||
"label": "绑定世界书摘要",
|
||||
"description": "项目绑定的世界书 meta / 摘要(系统自动注入,不可手动选择)",
|
||||
"autoInjected": true
|
||||
},
|
||||
{
|
||||
"ref": "workflow.boundCharacter",
|
||||
"label": "绑定角色卡摘要",
|
||||
"description": "项目绑定的角色卡摘要(系统自动注入,不可手动选择)",
|
||||
"autoInjected": true
|
||||
}
|
||||
],
|
||||
"dynamicSuffixes": [
|
||||
{
|
||||
"suffix": ".output",
|
||||
"labelPattern": "{displayName} · 世界书条目"
|
||||
}
|
||||
],
|
||||
"autoInjectedContext": [
|
||||
"目前产物",
|
||||
"思考流程",
|
||||
"核心目的",
|
||||
"评价标准与优化建议"
|
||||
],
|
||||
"autoInjectedContextDefs": [
|
||||
{
|
||||
"id": "currentProduct",
|
||||
"label": "目前产物",
|
||||
"description": "当前步骤已生成的世界书条目草稿或最新版本,供模型在迭代修改时对照与延续。"
|
||||
},
|
||||
{
|
||||
"id": "thinkingFlow",
|
||||
"label": "思考流程",
|
||||
"description": "本步骤配置的 thinkingPrompt,引导模型按既定步骤推理与自检。"
|
||||
},
|
||||
{
|
||||
"id": "coreGoal",
|
||||
"label": "核心目的",
|
||||
"description": "本步骤的 stepGoal(步骤目标),明确本步要产出的内容与边界。"
|
||||
},
|
||||
{
|
||||
"id": "scoringCriteria",
|
||||
"label": "评价标准与优化建议",
|
||||
"description": "本步骤启用的 scoring 评价维度及准则,用于模型自检与优化表述。"
|
||||
}
|
||||
]
|
||||
}
|
||||
10
docker-compose.dev.yml
Normal file
10
docker-compose.dev.yml
Normal file
@@ -0,0 +1,10 @@
|
||||
# 开发环境 override(可选)
|
||||
# 用法: docker compose -f docker-compose.yml -f docker-compose.dev.yml up -d
|
||||
#
|
||||
# 与主 compose 策略一致:源码 volume 挂载 + HMR/reload,启动时不跑 npm install。
|
||||
# 依赖变更: scripts/docker-rebuild.ps1 -Service frontend
|
||||
# 或 docker compose exec frontend npm install && docker compose restart frontend
|
||||
|
||||
services:
|
||||
frontend:
|
||||
command: npm run dev -- --host 0.0.0.0
|
||||
@@ -36,10 +36,11 @@ services:
|
||||
- "23338:5173"
|
||||
volumes:
|
||||
- ./frontend:/app
|
||||
- /app/node_modules
|
||||
- node_modules:/app/node_modules
|
||||
environment:
|
||||
- NODE_ENV=development
|
||||
command: sh -c "npm install && npm run dev -- --host 0.0.0.0"
|
||||
# 依赖在镜像构建时写入 node_modules volume;package.json 变更见 docs/DOCKER_DEV.md
|
||||
command: npm run dev -- --host 0.0.0.0
|
||||
depends_on:
|
||||
backend:
|
||||
condition: service_healthy
|
||||
|
||||
227
docs/DOCKER_DEV.md
Normal file
227
docs/DOCKER_DEV.md
Normal file
@@ -0,0 +1,227 @@
|
||||
# Docker 开发指南(Windows / Docker Desktop)
|
||||
|
||||
本文说明如何在 **不重启 Docker Desktop** 的前提下进行日常开发。绝大多数代码改动无需任何容器操作;需要时只需重启**单个容器**。
|
||||
|
||||
---
|
||||
|
||||
## 核心原则
|
||||
|
||||
| 场景 | 需要做什么 | 是否需要重启 Docker Desktop |
|
||||
|------|-----------|---------------------------|
|
||||
| 修改 Python 后端代码 | **什么都不做**(uvicorn `--reload` 自动重载) | ❌ 不需要 |
|
||||
| 修改 React 前端代码 | **什么都不做**(Vite HMR 热更新) | ❌ 不需要 |
|
||||
| 容器异常 / 需要刷新进程 | `docker compose restart backend` 或 `frontend` | ❌ 不需要 |
|
||||
| 修改 `Dockerfile` 或依赖文件 | `docker compose up -d --build <service>` | ❌ 不需要 |
|
||||
| Docker 引擎崩溃 / 端口被占用且无法释放 | 见下文「极少需要重启 Docker Desktop」 | ⚠️ 极少需要 |
|
||||
|
||||
**日常开发不要重启 Docker Desktop。** 那是 Windows + WSL2 下最慢、最打断节奏的操作。
|
||||
|
||||
---
|
||||
|
||||
## 端口对照表
|
||||
|
||||
与 `docker-compose.yml` 一致:
|
||||
|
||||
| 服务 | 容器内端口 | 宿主机端口 | 访问地址 |
|
||||
|------|-----------|-----------|---------|
|
||||
| backend | 8000 | **23337** | http://localhost:23337 |
|
||||
| frontend | 5173 | **23338** | http://localhost:23338 |
|
||||
|
||||
健康检查:`http://localhost:23337/health`
|
||||
|
||||
---
|
||||
|
||||
## 代码改动:自动生效
|
||||
|
||||
### 后端(FastAPI + uvicorn)
|
||||
|
||||
- 启动命令:`uvicorn main:app --host 0.0.0.0 --port 8000 --reload`
|
||||
- 源码通过 volume 挂载:`./backend` → `/app`
|
||||
- 保存 `.py` 文件后,uvicorn 自动检测并重载,**无需重启容器**
|
||||
|
||||
### 前端(Vite + React)
|
||||
|
||||
- 启动命令:`npm run dev -- --host 0.0.0.0`
|
||||
- 源码通过 volume 挂载:`./frontend` → `/app`
|
||||
- `node_modules` 保存在独立 volume 中,不随宿主机目录覆盖
|
||||
- 保存 `.jsx` / `.css` 等文件后,Vite HMR 自动更新浏览器,**无需重启容器**
|
||||
|
||||
---
|
||||
|
||||
## 何时只需重启容器(不是 Docker Desktop)
|
||||
|
||||
以下情况用 `scripts/docker-restart.ps1` 或 `docker compose restart` 即可:
|
||||
|
||||
- 修改了环境变量(`docker-compose.yml` 中的 `environment`)并已 `docker compose up -d`
|
||||
- 容器内进程卡死、内存泄漏
|
||||
- 前端 HMR 断开、WebSocket 连接异常
|
||||
- 后端 reload 失败(极少数语法错误导致 worker 无法恢复)
|
||||
|
||||
```powershell
|
||||
# 重启单个服务
|
||||
.\scripts\docker-restart.ps1 -Service backend
|
||||
.\scripts\docker-restart.ps1 -Service frontend
|
||||
|
||||
# 重启全部
|
||||
.\scripts\docker-restart.ps1 -Service all
|
||||
|
||||
# 或直接
|
||||
docker compose restart backend
|
||||
docker compose restart frontend
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 何时需要重新构建镜像
|
||||
|
||||
以下变更需要 **rebuild**,仍然 **不需要重启 Docker Desktop**:
|
||||
|
||||
| 变更内容 | 命令 |
|
||||
|---------|------|
|
||||
| `backend/requirements.txt` | `.\scripts\docker-rebuild.ps1 -Service backend` |
|
||||
| `backend/Dockerfile` | 同上 |
|
||||
| `frontend/package.json` / `package-lock.json` | `.\scripts\docker-rebuild.ps1 -Service frontend` |
|
||||
| `frontend/Dockerfile` | 同上 |
|
||||
|
||||
```powershell
|
||||
# 等价命令
|
||||
docker compose up -d --build backend
|
||||
docker compose up -d --build frontend
|
||||
```
|
||||
|
||||
### 前端依赖变更(package.json 改了但不想 rebuild)
|
||||
|
||||
若只新增了 npm 包、尚未 rebuild 镜像,可在运行中的容器内安装一次:
|
||||
|
||||
```powershell
|
||||
docker compose exec frontend npm install
|
||||
docker compose restart frontend
|
||||
```
|
||||
|
||||
首次 `docker compose up` 时,镜像构建阶段会执行 `npm install`,依赖写入 `node_modules` volume,之后日常启动**不再**每次 `npm install`。
|
||||
|
||||
---
|
||||
|
||||
## 极少需要重启 Docker Desktop 的情况
|
||||
|
||||
仅在以下情况才考虑重启 Docker Desktop 或 WSL:
|
||||
|
||||
1. **Docker 引擎无响应** — `docker ps` 一直挂起或报错 `Cannot connect to the Docker daemon`
|
||||
2. **端口被占用且 compose down 无法释放** — 例如 23337/23338 被僵尸进程占用
|
||||
3. **WSL2 后端异常** — 内存耗尽、磁盘满、网络栈故障
|
||||
|
||||
**优先尝试的替代方案(由轻到重):**
|
||||
|
||||
```powershell
|
||||
# 1. 停止并重新启动 compose 栈(不碰 Docker Desktop)
|
||||
docker compose down
|
||||
docker compose up -d
|
||||
|
||||
# 2. 查看日志定位问题
|
||||
.\scripts\docker-logs.ps1
|
||||
.\scripts\docker-logs.ps1 -Service backend
|
||||
|
||||
# 3. 仅当 Docker 完全无响应时,关闭 WSL(会连带重启 Docker 引擎)
|
||||
wsl --shutdown
|
||||
# 然后重新打开 Docker Desktop
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 更快的日常开发方式(推荐)
|
||||
|
||||
Docker 适合「全栈联调 / 验收环境」。纯改代码时,**本地直接跑**通常更快:
|
||||
|
||||
### 后端(本地)
|
||||
|
||||
```powershell
|
||||
python -m venv venv
|
||||
.\venv\Scripts\Activate.ps1
|
||||
pip install -r backend\requirements.txt
|
||||
cd backend
|
||||
python main.py
|
||||
```
|
||||
|
||||
### 前端(本地)
|
||||
|
||||
```powershell
|
||||
cd frontend
|
||||
npm install
|
||||
npm run dev
|
||||
```
|
||||
|
||||
本地开发时前端默认代理到本地后端;Docker 栈仅在需要容器化联调时使用。
|
||||
|
||||
---
|
||||
|
||||
## 辅助脚本速查
|
||||
|
||||
所有脚本位于 `scripts/`,在项目根目录执行:
|
||||
|
||||
| 脚本 | 用途 |
|
||||
|------|------|
|
||||
| `docker-up.ps1` | 后台启动全部服务 |
|
||||
| `docker-restart.ps1` | 重启 backend / frontend / all(**不**重启 Docker Desktop) |
|
||||
| `docker-logs.ps1` | 跟踪日志(可选 `-Service backend`) |
|
||||
| `docker-rebuild.ps1` | 重新构建并启动指定服务 |
|
||||
|
||||
### 典型一日工作流
|
||||
|
||||
```powershell
|
||||
# 早上第一次
|
||||
.\scripts\docker-up.ps1
|
||||
|
||||
# 白天改代码 — 保存即可,backend/frontend 自动更新
|
||||
|
||||
# 偶尔 HMR 或 reload 异常
|
||||
.\scripts\docker-restart.ps1 -Service frontend
|
||||
|
||||
# 改了 requirements.txt
|
||||
.\scripts\docker-rebuild.ps1 -Service backend
|
||||
|
||||
# 下班
|
||||
docker compose down
|
||||
```
|
||||
|
||||
---
|
||||
|
||||
## 使用 docker-compose.dev.yml(可选)
|
||||
|
||||
开发环境可使用 override 文件,与主 compose 合并:
|
||||
|
||||
```powershell
|
||||
docker compose -f docker-compose.yml -f docker-compose.dev.yml up -d
|
||||
```
|
||||
|
||||
内容与主文件优化策略一致;便于将来追加仅开发用的配置而不改动默认 compose。
|
||||
|
||||
---
|
||||
|
||||
## 常见问题
|
||||
|
||||
### Q: 改了前端代码但页面没更新?
|
||||
|
||||
1. 确认保存了文件
|
||||
2. 浏览器硬刷新(Ctrl+Shift+R)
|
||||
3. `.\scripts\docker-restart.ps1 -Service frontend`
|
||||
4. 查看日志:`.\scripts\docker-logs.ps1 -Service frontend`
|
||||
|
||||
### Q: 改了后端代码但 API 行为没变?
|
||||
|
||||
1. 查看 backend 日志是否有 reload 报错
|
||||
2. `.\scripts\docker-restart.ps1 -Service backend`
|
||||
3. 若改了依赖,需 rebuild
|
||||
|
||||
### Q: 每次启动 frontend 都很慢?
|
||||
|
||||
旧版 compose 在每次容器启动时执行 `npm install`。当前配置已改为直接 `npm run dev`;依赖在**镜像构建时**或**手动 exec npm install** 时装入 volume。若仍慢,检查是否误删了 `node_modules` volume:
|
||||
|
||||
```powershell
|
||||
docker volume ls | Select-String node_modules
|
||||
```
|
||||
|
||||
### Q: 必须重启 Docker Desktop 吗?
|
||||
|
||||
**正常代码编辑:不需要。**
|
||||
**依赖 / Dockerfile 变更:rebuild 容器即可。**
|
||||
**只有 Docker 引擎本身故障时才考虑重启 Docker Desktop 或 `wsl --shutdown`。**
|
||||
@@ -4,7 +4,11 @@ import TopBar from './components/TopBar';
|
||||
import { ChatBox } from './components/Mid';
|
||||
import SideBarLeft from './components/SideBarLeft';
|
||||
import SideBarRight from './components/SideBarRight';
|
||||
import PlaceholderPage from './components/PlaceholderPage';
|
||||
import StudioEditPage from './components/Studio/StudioEditPage';
|
||||
import StudioRunPage from './components/Studio/StudioRunPage';
|
||||
import useAppLayoutStore from './Store/AppLayoutSlice'; // ✅ 新增
|
||||
import useStudioStore from './Store/Studio/StudioSlice';
|
||||
import useApiConfigStore from './Store/SideBarLeft/ApiConfigSlice'; // ✅ 引入 API 配置 Store
|
||||
import usePresetStore from './Store/SideBarLeft/PresetSlice'; // ✅ 引入预设 Store
|
||||
import useCharacterStore from './Store/SideBarLeft/CharacterSlice'; // ✅ 引入角色卡 Store
|
||||
@@ -18,6 +22,7 @@ function App() {
|
||||
sidebarMode,
|
||||
isSidebarHovered,
|
||||
colorTheme,
|
||||
activePage,
|
||||
setLayoutMode,
|
||||
setSidebarMode,
|
||||
setSidebarHovered,
|
||||
@@ -187,34 +192,62 @@ function App() {
|
||||
});
|
||||
}, []); // 仅在应用启动时执行一次
|
||||
|
||||
const initStudio = useStudioStore((s) => s.initStudio);
|
||||
const initStudioRun = useStudioStore((s) => s.initStudioRun);
|
||||
const isStudioEditPage = activePage === 'studio_edit';
|
||||
const isStudioRunPage = activePage === 'studio_run';
|
||||
const isStudioPage = isStudioEditPage || isStudioRunPage;
|
||||
|
||||
useEffect(() => {
|
||||
if (isStudioEditPage) {
|
||||
initStudio();
|
||||
} else if (isStudioRunPage) {
|
||||
initStudioRun();
|
||||
}
|
||||
}, [isStudioEditPage, isStudioRunPage, initStudio, initStudioRun]);
|
||||
|
||||
return (
|
||||
<div className={`app ${layoutMode}-mode`}>
|
||||
{/* ✅ TopBar 不再需要 props,直接从 Store 读取状态 */}
|
||||
<TopBar />
|
||||
|
||||
{/* 主内容容器 */}
|
||||
<div className="main-container">
|
||||
{/* 左侧栏 - 智能模式下悬停展开 */}
|
||||
<div
|
||||
className={`sidebar-left-wrapper sidebar-mode-${sidebarMode} ${sidebarMode === 'smart' && isSidebarHovered ? 'sidebar-expanded' : ''}`}
|
||||
onMouseEnter={handleMouseEnter}
|
||||
onMouseLeave={handleMouseLeave}
|
||||
>
|
||||
<SideBarLeft />
|
||||
</div>
|
||||
{activePage === 'chat' ? (
|
||||
<div className="main-container">
|
||||
{/* 左侧栏 - 智能模式下悬停展开 */}
|
||||
<div
|
||||
className={`sidebar-left-wrapper sidebar-mode-${sidebarMode} ${sidebarMode === 'smart' && isSidebarHovered ? 'sidebar-expanded' : ''}`}
|
||||
onMouseEnter={handleMouseEnter}
|
||||
onMouseLeave={handleMouseLeave}
|
||||
>
|
||||
<SideBarLeft />
|
||||
</div>
|
||||
|
||||
{/* 中间栏:聊天框 */}
|
||||
<div className="chat-area-wrapper">
|
||||
<div className="chat-area">
|
||||
<ChatBox />
|
||||
{/* 中间栏:聊天框 */}
|
||||
<div className="chat-area-wrapper">
|
||||
<div className="chat-area">
|
||||
<ChatBox />
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* 右侧栏 */}
|
||||
<div className="sidebar-right-wrapper">
|
||||
<SideBarRight />
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{/* 右侧栏 */}
|
||||
<div className="sidebar-right-wrapper">
|
||||
<SideBarRight />
|
||||
) : activePage === 'studio_edit' ? (
|
||||
<div className="main-container studio-container">
|
||||
<StudioEditPage />
|
||||
</div>
|
||||
</div>
|
||||
) : activePage === 'studio_run' ? (
|
||||
<div className="main-container studio-container">
|
||||
<StudioRunPage />
|
||||
</div>
|
||||
) : (
|
||||
<div className="main-container placeholder-container">
|
||||
<PlaceholderPage page={activePage} />
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
@@ -22,6 +22,9 @@ const useAppLayoutStore = create(
|
||||
// 颜色主题:'light' | 'dark'
|
||||
colorTheme: 'dark',
|
||||
|
||||
// 当前页面:'chat' | 'studio_edit' | 'studio_run' | 'novel' | 'room'
|
||||
activePage: 'chat',
|
||||
|
||||
// ==================== Actions ====================
|
||||
|
||||
/**
|
||||
@@ -48,6 +51,14 @@ const useAppLayoutStore = create(
|
||||
set({ isSidebarHovered: hovered });
|
||||
},
|
||||
|
||||
/**
|
||||
* 设置当前页面
|
||||
* @param {string} page - 'chat' | 'studio_edit' | 'studio_run' | 'novel' | 'room'
|
||||
*/
|
||||
setActivePage: (page) => {
|
||||
set({ activePage: page });
|
||||
},
|
||||
|
||||
/**
|
||||
* 设置颜色主题
|
||||
* @param {string} theme - 主题名
|
||||
@@ -78,16 +89,26 @@ const useAppLayoutStore = create(
|
||||
layoutMode: 'chat',
|
||||
sidebarMode: 'both',
|
||||
isSidebarHovered: false,
|
||||
colorTheme: 'dark'
|
||||
colorTheme: 'dark',
|
||||
activePage: 'chat',
|
||||
});
|
||||
}
|
||||
}),
|
||||
{
|
||||
name: 'app-layout-storage', // localStorage key
|
||||
name: 'app-layout-storage',
|
||||
version: 1,
|
||||
migrate: (persistedState) => {
|
||||
if (!persistedState) return persistedState;
|
||||
if (persistedState.activePage === 'studio') {
|
||||
return { ...persistedState, activePage: 'studio_edit' };
|
||||
}
|
||||
return persistedState;
|
||||
},
|
||||
partialize: (state) => ({
|
||||
layoutMode: state.layoutMode,
|
||||
sidebarMode: state.sidebarMode,
|
||||
colorTheme: state.colorTheme
|
||||
colorTheme: state.colorTheme,
|
||||
activePage: state.activePage,
|
||||
})
|
||||
}
|
||||
)
|
||||
|
||||
@@ -88,11 +88,14 @@ const useCharacterStore = create(
|
||||
}
|
||||
},
|
||||
|
||||
// 选择角色
|
||||
// 选择角色(传 null 清除选中)
|
||||
selectCharacter: (character) => {
|
||||
// 使用函数式更新避免不必要的重渲染
|
||||
set((state) => {
|
||||
// 如果选中的是同一个角色,不更新状态
|
||||
if (!character) {
|
||||
return state.selectedCharacter === null
|
||||
? state
|
||||
: { selectedCharacter: null };
|
||||
}
|
||||
if (state.selectedCharacter?.id === character.id) {
|
||||
return state;
|
||||
}
|
||||
@@ -100,6 +103,8 @@ const useCharacterStore = create(
|
||||
});
|
||||
},
|
||||
|
||||
clearSelectedCharacter: () => set({ selectedCharacter: null }),
|
||||
|
||||
// 创建角色
|
||||
createCharacter: async (characterData) => {
|
||||
try {
|
||||
@@ -150,7 +155,19 @@ const useCharacterStore = create(
|
||||
});
|
||||
|
||||
if (!response.ok) throw new Error('Failed to delete character');
|
||||
|
||||
|
||||
set((state) => {
|
||||
const next = {};
|
||||
if (state.selectedCharacter?.name === name) {
|
||||
next.selectedCharacter = null;
|
||||
}
|
||||
if (state.characterChats[name]) {
|
||||
const { [name]: _removed, ...restChats } = state.characterChats;
|
||||
next.characterChats = restChats;
|
||||
}
|
||||
return Object.keys(next).length ? next : state;
|
||||
});
|
||||
|
||||
// 刷新列表
|
||||
await get().fetchCharacters();
|
||||
} catch (error) {
|
||||
|
||||
694
frontend/src/Store/Studio/StudioSlice.jsx
Normal file
694
frontend/src/Store/Studio/StudioSlice.jsx
Normal file
@@ -0,0 +1,694 @@
|
||||
import { create } from 'zustand';
|
||||
|
||||
const DEFAULT_TEMPLATE_ID = 'builtin.studio.example';
|
||||
|
||||
function migrateScoringInNode(node) {
|
||||
const scoring = node.config?.scoring;
|
||||
if (!scoring || scoring.dimensions?.length) return node;
|
||||
if (!scoring.rubric) return node;
|
||||
return {
|
||||
...node,
|
||||
config: {
|
||||
...node.config,
|
||||
scoring: {
|
||||
...scoring,
|
||||
dimensions: [
|
||||
{
|
||||
id: 'default',
|
||||
name: '综合质量',
|
||||
criteria: scoring.rubric,
|
||||
},
|
||||
],
|
||||
rubric: undefined,
|
||||
},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
function migratePipeline(pipeline) {
|
||||
if (!pipeline) return pipeline;
|
||||
return {
|
||||
...pipeline,
|
||||
nodes: (pipeline.nodes || []).map(migrateScoringInNode),
|
||||
};
|
||||
}
|
||||
|
||||
const useStudioStore = create((set, get) => ({
|
||||
projects: [],
|
||||
workflowTemplates: [],
|
||||
workflowVariables: { builtIn: [], dynamic: [] },
|
||||
currentProjectId: null,
|
||||
meta: null,
|
||||
pipeline: null,
|
||||
skillTemplates: [],
|
||||
selectedNodeId: null,
|
||||
loading: false,
|
||||
saving: false,
|
||||
error: null,
|
||||
saveMessage: null,
|
||||
|
||||
// Studio run (R0)
|
||||
runProjectId: null,
|
||||
runs: [],
|
||||
currentRunId: null,
|
||||
currentRun: null,
|
||||
runLoading: false,
|
||||
runCreating: false,
|
||||
runAdvancing: false,
|
||||
runMessaging: false,
|
||||
runStreamingThinking: null,
|
||||
runError: null,
|
||||
runSessionEntered: false,
|
||||
|
||||
setSelectedNodeId: (id) => set({ selectedNodeId: id }),
|
||||
clearSaveMessage: () => set({ saveMessage: null, error: null }),
|
||||
|
||||
setMetaLocal: (patch) => {
|
||||
const { meta } = get();
|
||||
if (!meta) return;
|
||||
set({ meta: { ...meta, ...patch } });
|
||||
},
|
||||
|
||||
setPipelineLocal: (pipeline) => set({ pipeline: migratePipeline(pipeline) }),
|
||||
|
||||
updateNode: (nodeId, patch) => {
|
||||
const { pipeline } = get();
|
||||
if (!pipeline) return;
|
||||
const nodes = pipeline.nodes.map((n) =>
|
||||
n.id === nodeId ? { ...n, ...patch } : n
|
||||
);
|
||||
set({ pipeline: { ...pipeline, nodes } });
|
||||
},
|
||||
|
||||
updateNodeConfig: (nodeId, configPatch) => {
|
||||
const { pipeline } = get();
|
||||
if (!pipeline) return;
|
||||
const nodes = pipeline.nodes.map((n) =>
|
||||
n.id === nodeId
|
||||
? { ...n, config: { ...(n.config || {}), ...configPatch } }
|
||||
: n
|
||||
);
|
||||
set({ pipeline: { ...pipeline, nodes } });
|
||||
},
|
||||
|
||||
reorderNodes: (fromIndex, toIndex) => {
|
||||
const { pipeline } = get();
|
||||
if (!pipeline || fromIndex === toIndex) return;
|
||||
const nodes = [...pipeline.nodes];
|
||||
const [moved] = nodes.splice(fromIndex, 1);
|
||||
nodes.splice(toIndex, 0, moved);
|
||||
set({ pipeline: { ...pipeline, nodes } });
|
||||
},
|
||||
|
||||
addNode: (skillId, displayName) => {
|
||||
const { pipeline, skillTemplates } = get();
|
||||
if (!pipeline) return;
|
||||
const tpl = skillTemplates.find((t) => t.skillId === skillId);
|
||||
const id = `node-${Date.now()}`;
|
||||
const base = {
|
||||
id,
|
||||
skillId,
|
||||
displayName: displayName || tpl?.displayName || skillId,
|
||||
enabled: true,
|
||||
config: {},
|
||||
displayParams: tpl?.displayParams ? [...tpl.displayParams] : [],
|
||||
inputs: [],
|
||||
};
|
||||
if (tpl?.configDefaults) {
|
||||
base.config = JSON.parse(JSON.stringify(tpl.configDefaults));
|
||||
}
|
||||
if (skillId === 'studio.worldbook_entry') {
|
||||
base.loopUntilSatisfied = false;
|
||||
}
|
||||
const nodes = [...pipeline.nodes, base];
|
||||
set({
|
||||
pipeline: { ...pipeline, nodes },
|
||||
selectedNodeId: id,
|
||||
});
|
||||
},
|
||||
|
||||
removeNode: (nodeId) => {
|
||||
const { pipeline, selectedNodeId } = get();
|
||||
if (!pipeline) return;
|
||||
const nodes = pipeline.nodes.filter((n) => n.id !== nodeId);
|
||||
set({
|
||||
pipeline: { ...pipeline, nodes },
|
||||
selectedNodeId:
|
||||
selectedNodeId === nodeId
|
||||
? nodes[0]?.id ?? null
|
||||
: selectedNodeId,
|
||||
});
|
||||
},
|
||||
|
||||
fetchProjects: async () => {
|
||||
set({ loading: true, error: null });
|
||||
try {
|
||||
const res = await fetch('/api/studio/projects');
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const projects = await res.json();
|
||||
set({ projects, loading: false });
|
||||
return projects;
|
||||
} catch (e) {
|
||||
set({ loading: false, error: e.message || '加载项目列表失败' });
|
||||
return [];
|
||||
}
|
||||
},
|
||||
|
||||
fetchWorkflowTemplates: async () => {
|
||||
try {
|
||||
const res = await fetch('/api/studio/templates');
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const templates = await res.json();
|
||||
set({ workflowTemplates: templates });
|
||||
return templates;
|
||||
} catch (e) {
|
||||
set({ error: e.message || '加载工作流模板失败' });
|
||||
return [];
|
||||
}
|
||||
},
|
||||
|
||||
fetchWorkflowVariables: async (projectId) => {
|
||||
try {
|
||||
const qs = projectId
|
||||
? `?projectId=${encodeURIComponent(projectId)}`
|
||||
: '';
|
||||
const res = await fetch(`/api/studio/variables${qs}`);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
set({ workflowVariables: data });
|
||||
return data;
|
||||
} catch (e) {
|
||||
set({ error: e.message || '加载工作流变量失败' });
|
||||
return { builtIn: [], dynamic: [] };
|
||||
}
|
||||
},
|
||||
|
||||
fetchSkillTemplates: async () => {
|
||||
try {
|
||||
const res = await fetch('/api/studio/skill-templates');
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const tplData = await res.json();
|
||||
set({ skillTemplates: tplData.templates || [] });
|
||||
} catch (e) {
|
||||
set({ error: e.message || '加载技能模板失败' });
|
||||
}
|
||||
},
|
||||
|
||||
fetchProject: async (projectId) => {
|
||||
set({ loading: true, error: null });
|
||||
try {
|
||||
const res = await fetch(`/api/studio/projects/${encodeURIComponent(projectId)}`);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
const pipeline = migratePipeline(data.pipeline);
|
||||
set({
|
||||
currentProjectId: data.meta.id,
|
||||
meta: data.meta,
|
||||
pipeline,
|
||||
selectedNodeId: pipeline?.nodes?.[0]?.id ?? null,
|
||||
loading: false,
|
||||
});
|
||||
await get().fetchWorkflowVariables(projectId);
|
||||
return { ...data, pipeline };
|
||||
} catch (e) {
|
||||
set({ loading: false, error: e.message || '加载项目失败' });
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
savePipeline: async () => {
|
||||
const { currentProjectId, pipeline } = get();
|
||||
if (!currentProjectId || !pipeline) return false;
|
||||
set({ saving: true, error: null, saveMessage: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(currentProjectId)}/pipeline`,
|
||||
{
|
||||
method: 'PUT',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify(pipeline),
|
||||
}
|
||||
);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
set({
|
||||
meta: data.meta,
|
||||
pipeline: migratePipeline(data.pipeline),
|
||||
saving: false,
|
||||
saveMessage: '已保存到本地',
|
||||
});
|
||||
return true;
|
||||
} catch (e) {
|
||||
set({
|
||||
saving: false,
|
||||
error: e.message || '保存失败',
|
||||
});
|
||||
return false;
|
||||
}
|
||||
},
|
||||
|
||||
createProject: async (name, templateId = DEFAULT_TEMPLATE_ID) => {
|
||||
set({ loading: true, error: null });
|
||||
try {
|
||||
const res = await fetch('/api/studio/projects', {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ name, template_id: templateId }),
|
||||
});
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
await get().fetchProjects();
|
||||
const pipeline = migratePipeline(data.pipeline);
|
||||
set({
|
||||
currentProjectId: data.meta.id,
|
||||
meta: data.meta,
|
||||
pipeline,
|
||||
selectedNodeId: pipeline?.nodes?.[0]?.id ?? null,
|
||||
loading: false,
|
||||
});
|
||||
await get().fetchWorkflowVariables(data.meta.id);
|
||||
return data;
|
||||
} catch (e) {
|
||||
set({ loading: false, error: e.message || '创建项目失败' });
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
updateProjectMeta: async (projectId, { name, description } = {}) => {
|
||||
if (!projectId) return null;
|
||||
set({ saving: true, error: null, saveMessage: null });
|
||||
try {
|
||||
const body = {};
|
||||
if (name !== undefined) body.name = name;
|
||||
if (description !== undefined) body.description = description;
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(projectId)}`,
|
||||
{
|
||||
method: 'PATCH',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify(body),
|
||||
}
|
||||
);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const data = await res.json();
|
||||
const pipeline = migratePipeline(data.pipeline);
|
||||
set({
|
||||
meta: data.meta,
|
||||
pipeline,
|
||||
saving: false,
|
||||
saveMessage: '项目信息已保存',
|
||||
});
|
||||
await get().fetchProjects();
|
||||
return data;
|
||||
} catch (e) {
|
||||
set({
|
||||
saving: false,
|
||||
error: e.message || '保存项目信息失败',
|
||||
});
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
renameProject: async (projectId, newName) => {
|
||||
const trimmed = (newName || '').trim();
|
||||
if (!projectId || !trimmed) return null;
|
||||
return get().updateProjectMeta(projectId, { name: trimmed });
|
||||
},
|
||||
|
||||
deleteProject: async (projectId) => {
|
||||
if (!projectId) return false;
|
||||
set({ loading: true, error: null, saveMessage: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(projectId)}`,
|
||||
{ method: 'DELETE' }
|
||||
);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const projects = await get().fetchProjects();
|
||||
const nextId = projects[0]?.id ?? null;
|
||||
if (nextId) {
|
||||
await get().fetchProject(nextId);
|
||||
} else {
|
||||
set({
|
||||
currentProjectId: null,
|
||||
meta: null,
|
||||
pipeline: null,
|
||||
selectedNodeId: null,
|
||||
});
|
||||
}
|
||||
if (get().runProjectId === projectId) {
|
||||
set({ runProjectId: nextId, runs: [], currentRunId: null, currentRun: null });
|
||||
}
|
||||
set({ loading: false, saveMessage: '项目已删除' });
|
||||
return true;
|
||||
} catch (e) {
|
||||
set({ loading: false, error: e.message || '删除项目失败' });
|
||||
return false;
|
||||
}
|
||||
},
|
||||
|
||||
initStudio: async () => {
|
||||
await Promise.all([
|
||||
get().fetchSkillTemplates(),
|
||||
get().fetchWorkflowTemplates(),
|
||||
get().fetchProjects(),
|
||||
]);
|
||||
const projects = get().projects;
|
||||
const id =
|
||||
get().currentProjectId ||
|
||||
projects[0]?.id ||
|
||||
'default';
|
||||
if (id) {
|
||||
await get().fetchProject(id);
|
||||
}
|
||||
},
|
||||
|
||||
setRunProjectId: (projectId) => {
|
||||
set({
|
||||
runProjectId: projectId,
|
||||
currentRunId: null,
|
||||
currentRun: null,
|
||||
runs: [],
|
||||
runSessionEntered: false,
|
||||
});
|
||||
},
|
||||
|
||||
setRunSessionEntered: (entered) => set({ runSessionEntered: !!entered }),
|
||||
|
||||
fetchRuns: async (projectId) => {
|
||||
if (!projectId) return [];
|
||||
set({ runLoading: true, runError: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(projectId)}/runs`
|
||||
);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const runs = await res.json();
|
||||
set({ runs, runLoading: false });
|
||||
return runs;
|
||||
} catch (e) {
|
||||
set({
|
||||
runLoading: false,
|
||||
runError: e.message || '加载运行列表失败',
|
||||
});
|
||||
return [];
|
||||
}
|
||||
},
|
||||
|
||||
fetchRun: async (projectId, runId) => {
|
||||
if (!projectId || !runId) return null;
|
||||
set({ runLoading: true, runError: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(projectId)}/runs/${encodeURIComponent(runId)}`
|
||||
);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const run = await res.json();
|
||||
set({
|
||||
currentRunId: runId,
|
||||
currentRun: run,
|
||||
runLoading: false,
|
||||
runSessionEntered: false,
|
||||
});
|
||||
return run;
|
||||
} catch (e) {
|
||||
set({
|
||||
runLoading: false,
|
||||
runError: e.message || '加载运行详情失败',
|
||||
});
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
createRun: async (projectId) => {
|
||||
if (!projectId) return null;
|
||||
set({ runCreating: true, runError: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(projectId)}/runs`,
|
||||
{ method: 'POST' }
|
||||
);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const run = await res.json();
|
||||
await get().fetchRuns(projectId);
|
||||
set({
|
||||
currentRunId: run.id,
|
||||
currentRun: run,
|
||||
runCreating: false,
|
||||
runSessionEntered: false,
|
||||
});
|
||||
return run;
|
||||
} catch (e) {
|
||||
set({
|
||||
runCreating: false,
|
||||
runError: e.message || '创建运行失败',
|
||||
});
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
selectRun: async (runId) => {
|
||||
const projectId = get().runProjectId;
|
||||
if (!projectId || !runId) return null;
|
||||
return get().fetchRun(projectId, runId);
|
||||
},
|
||||
|
||||
advanceRun: async (displayParams) => {
|
||||
const { runProjectId, currentRunId } = get();
|
||||
if (!runProjectId || !currentRunId) return null;
|
||||
set({ runAdvancing: true, runError: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(runProjectId)}/runs/${encodeURIComponent(currentRunId)}/advance`,
|
||||
{
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ displayParams }),
|
||||
}
|
||||
);
|
||||
if (!res.ok) {
|
||||
let detail = await res.text();
|
||||
try {
|
||||
const parsed = JSON.parse(detail);
|
||||
detail = parsed.detail || detail;
|
||||
} catch {
|
||||
/* keep raw text */
|
||||
}
|
||||
throw new Error(detail || '推进运行失败');
|
||||
}
|
||||
const run = await res.json();
|
||||
set({
|
||||
currentRun: run,
|
||||
runAdvancing: false,
|
||||
});
|
||||
await get().fetchRuns(runProjectId);
|
||||
return run;
|
||||
} catch (e) {
|
||||
set({
|
||||
runAdvancing: false,
|
||||
runError: e.message || '推进运行失败',
|
||||
});
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
sendRunMessage: async (content, { stream = false } = {}) => {
|
||||
const { runProjectId, currentRunId } = get();
|
||||
if (!runProjectId || !currentRunId) return null;
|
||||
set({ runMessaging: true, runStreamingThinking: null, runError: null });
|
||||
try {
|
||||
let profileId = null;
|
||||
let apiConfig = {};
|
||||
try {
|
||||
const { default: useApiConfigStore } = await import(
|
||||
'../SideBarLeft/ApiConfigSlice'
|
||||
);
|
||||
const apiState = useApiConfigStore.getState();
|
||||
profileId = apiState.currentProfile?.id || null;
|
||||
const mainLLM = apiState.currentProfile?.apis?.mainLLM;
|
||||
if (mainLLM) {
|
||||
apiConfig = {
|
||||
api_url: mainLLM.apiUrl || '',
|
||||
model: mainLLM.model || '',
|
||||
};
|
||||
}
|
||||
} catch {
|
||||
/* optional store */
|
||||
}
|
||||
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(runProjectId)}/runs/${encodeURIComponent(currentRunId)}/message`,
|
||||
{
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({
|
||||
content,
|
||||
stream,
|
||||
profileId,
|
||||
apiConfig,
|
||||
}),
|
||||
}
|
||||
);
|
||||
|
||||
if (!res.ok) {
|
||||
let detail = await res.text();
|
||||
try {
|
||||
const parsed = JSON.parse(detail);
|
||||
detail = parsed.detail || detail;
|
||||
} catch {
|
||||
/* keep raw text */
|
||||
}
|
||||
throw new Error(detail || '发送消息失败');
|
||||
}
|
||||
|
||||
if (stream && res.body) {
|
||||
const reader = res.body.getReader();
|
||||
const decoder = new TextDecoder();
|
||||
let buffer = '';
|
||||
let finalRun = null;
|
||||
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
buffer += decoder.decode(value, { stream: true });
|
||||
const lines = buffer.split('\n');
|
||||
buffer = lines.pop() || '';
|
||||
for (const line of lines) {
|
||||
if (!line.trim()) continue;
|
||||
const event = JSON.parse(line);
|
||||
if (event.type === 'thinking_delta') {
|
||||
set({ runStreamingThinking: event.content || '' });
|
||||
} else if (event.type === 'complete') {
|
||||
finalRun = event.run;
|
||||
} else if (event.type === 'error') {
|
||||
throw new Error(event.detail || '流式消息处理失败');
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (buffer.trim()) {
|
||||
const event = JSON.parse(buffer);
|
||||
if (event.type === 'complete') {
|
||||
finalRun = event.run;
|
||||
} else if (event.type === 'error') {
|
||||
throw new Error(event.detail || '流式消息处理失败');
|
||||
} else if (event.type === 'thinking_delta') {
|
||||
set({ runStreamingThinking: event.content || '' });
|
||||
}
|
||||
}
|
||||
|
||||
if (!finalRun) {
|
||||
throw new Error('流式响应未完成');
|
||||
}
|
||||
|
||||
set({
|
||||
currentRun: finalRun,
|
||||
runMessaging: false,
|
||||
runStreamingThinking: null,
|
||||
});
|
||||
return finalRun;
|
||||
}
|
||||
|
||||
const run = await res.json();
|
||||
set({
|
||||
currentRun: run,
|
||||
runMessaging: false,
|
||||
runStreamingThinking: null,
|
||||
});
|
||||
return run;
|
||||
} catch (e) {
|
||||
set({
|
||||
runMessaging: false,
|
||||
runStreamingThinking: null,
|
||||
runError: e.message || '发送消息失败',
|
||||
});
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
deleteRun: async (runId) => {
|
||||
const projectId = get().runProjectId;
|
||||
if (!projectId || !runId) return false;
|
||||
set({ runLoading: true, runError: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(projectId)}/runs/${encodeURIComponent(runId)}`,
|
||||
{ method: 'DELETE' }
|
||||
);
|
||||
if (!res.ok) throw new Error(await res.text());
|
||||
const runs = await get().fetchRuns(projectId);
|
||||
const { currentRunId } = get();
|
||||
if (currentRunId === runId) {
|
||||
const nextId = runs[0]?.id ?? null;
|
||||
if (nextId) {
|
||||
await get().fetchRun(projectId, nextId);
|
||||
} else {
|
||||
set({ currentRunId: null, currentRun: null, runLoading: false });
|
||||
}
|
||||
} else {
|
||||
set({ runLoading: false });
|
||||
}
|
||||
return true;
|
||||
} catch (e) {
|
||||
set({
|
||||
runLoading: false,
|
||||
runError: e.message || '删除运行失败',
|
||||
});
|
||||
return false;
|
||||
}
|
||||
},
|
||||
|
||||
renameRun: async (runId, title) => {
|
||||
const projectId = get().runProjectId;
|
||||
if (!projectId || !runId) return null;
|
||||
set({ runError: null });
|
||||
try {
|
||||
const res = await fetch(
|
||||
`/api/studio/projects/${encodeURIComponent(projectId)}/runs/${encodeURIComponent(runId)}`,
|
||||
{
|
||||
method: 'PATCH',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ title }),
|
||||
}
|
||||
);
|
||||
if (!res.ok) {
|
||||
let detail = await res.text();
|
||||
try {
|
||||
const parsed = JSON.parse(detail);
|
||||
detail = parsed.detail || detail;
|
||||
} catch {
|
||||
/* keep raw text */
|
||||
}
|
||||
throw new Error(detail || '重命名失败');
|
||||
}
|
||||
const run = await res.json();
|
||||
if (get().currentRunId === runId) {
|
||||
set({ currentRun: run });
|
||||
}
|
||||
await get().fetchRuns(projectId);
|
||||
return run;
|
||||
} catch (e) {
|
||||
set({ runError: e.message || '重命名失败' });
|
||||
return null;
|
||||
}
|
||||
},
|
||||
|
||||
initStudioRun: async () => {
|
||||
await get().fetchProjects();
|
||||
const projects = get().projects;
|
||||
const projectId =
|
||||
get().runProjectId ||
|
||||
get().currentProjectId ||
|
||||
projects[0]?.id ||
|
||||
'default';
|
||||
set({ runProjectId: projectId });
|
||||
await get().fetchWorkflowVariables(projectId);
|
||||
const runs = await get().fetchRuns(projectId);
|
||||
const runId = get().currentRunId || runs[0]?.id || null;
|
||||
if (runId) {
|
||||
await get().fetchRun(projectId, runId);
|
||||
}
|
||||
},
|
||||
}));
|
||||
|
||||
export default useStudioStore;
|
||||
46
frontend/src/components/PlaceholderPage/PlaceholderPage.css
Normal file
46
frontend/src/components/PlaceholderPage/PlaceholderPage.css
Normal file
@@ -0,0 +1,46 @@
|
||||
.placeholder-page {
|
||||
flex: 1;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
min-height: 0;
|
||||
padding: var(--spacing-xl);
|
||||
background: var(--color-bg-primary);
|
||||
}
|
||||
|
||||
.placeholder-card {
|
||||
text-align: center;
|
||||
max-width: 420px;
|
||||
padding: var(--spacing-xl) var(--spacing-lg);
|
||||
border-radius: var(--radius-lg);
|
||||
border: 1px solid var(--color-border-light);
|
||||
background: var(--color-bg-secondary);
|
||||
box-shadow: var(--shadow-sm);
|
||||
}
|
||||
|
||||
.placeholder-icon {
|
||||
font-size: 3rem;
|
||||
display: block;
|
||||
margin-bottom: var(--spacing-md);
|
||||
opacity: 0.85;
|
||||
}
|
||||
|
||||
.placeholder-title {
|
||||
margin: 0 0 var(--spacing-sm);
|
||||
font-size: 1.5rem;
|
||||
font-weight: 600;
|
||||
color: var(--color-text-primary);
|
||||
}
|
||||
|
||||
.placeholder-subtitle {
|
||||
margin: 0 0 var(--spacing-md);
|
||||
font-size: 0.9rem;
|
||||
color: var(--color-text-secondary);
|
||||
}
|
||||
|
||||
.placeholder-message {
|
||||
margin: 0;
|
||||
font-size: 0.95rem;
|
||||
color: var(--color-text-tertiary, var(--color-text-secondary));
|
||||
letter-spacing: 0.02em;
|
||||
}
|
||||
43
frontend/src/components/PlaceholderPage/PlaceholderPage.jsx
Normal file
43
frontend/src/components/PlaceholderPage/PlaceholderPage.jsx
Normal file
@@ -0,0 +1,43 @@
|
||||
import React from 'react';
|
||||
import './PlaceholderPage.css';
|
||||
|
||||
const PLACEHOLDER_META = {
|
||||
studio: {
|
||||
title: '角色 / 世界书工作室',
|
||||
subtitle: '用途1 · Character & Worldbook Studio',
|
||||
icon: '🎭',
|
||||
},
|
||||
novel: {
|
||||
title: '爽文模式',
|
||||
subtitle: '用途3 · Novel Mode',
|
||||
icon: '📖',
|
||||
},
|
||||
room: {
|
||||
title: '多角色房间',
|
||||
subtitle: '用途4 · Multi-Character Room',
|
||||
icon: '🏠',
|
||||
},
|
||||
};
|
||||
|
||||
function PlaceholderPage({ page }) {
|
||||
const meta = PLACEHOLDER_META[page] || {
|
||||
title: 'Coming Soon',
|
||||
subtitle: '',
|
||||
icon: '🚧',
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="placeholder-page">
|
||||
<div className="placeholder-card">
|
||||
<span className="placeholder-icon" aria-hidden="true">{meta.icon}</span>
|
||||
<h1 className="placeholder-title">{meta.title}</h1>
|
||||
{meta.subtitle && (
|
||||
<p className="placeholder-subtitle">{meta.subtitle}</p>
|
||||
)}
|
||||
<p className="placeholder-message">Coming soon</p>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default PlaceholderPage;
|
||||
1
frontend/src/components/PlaceholderPage/index.js
Normal file
1
frontend/src/components/PlaceholderPage/index.js
Normal file
@@ -0,0 +1 @@
|
||||
export { default } from './PlaceholderPage';
|
||||
@@ -132,6 +132,8 @@
|
||||
|
||||
/* 角色卡片列表 - 网格布局 */
|
||||
.character-list {
|
||||
position: relative;
|
||||
z-index: var(--z-base-content);
|
||||
flex: 1;
|
||||
overflow-y: auto;
|
||||
display: grid;
|
||||
@@ -318,14 +320,16 @@
|
||||
flex-direction: column;
|
||||
gap: 8px;
|
||||
padding: 8px;
|
||||
position: relative; /* 为悬浮工具栏提供定位上下文 */
|
||||
position: relative;
|
||||
z-index: var(--z-base-content);
|
||||
isolation: isolate;
|
||||
}
|
||||
|
||||
/* 悬浮工具栏 */
|
||||
.floating-toolbar {
|
||||
position: sticky;
|
||||
top: 0;
|
||||
z-index: var(--z-top-bar); /* ✅ 组件层 - TopBar 同级 */
|
||||
z-index: 2;
|
||||
margin-bottom: 8px;
|
||||
}
|
||||
|
||||
|
||||
@@ -237,12 +237,21 @@ const CharacterCard = () => {
|
||||
|
||||
try {
|
||||
await deleteCharacter(name);
|
||||
|
||||
// ✅ 删除成功后自动切换到角色分页
|
||||
|
||||
cancelEditing();
|
||||
selectCharacter(null);
|
||||
|
||||
const chatBox = useChatBoxStore.getState();
|
||||
if (chatBox.currentRole === name) {
|
||||
chatBox.setCurrentRole(null);
|
||||
chatBox.setCurrentChat(null);
|
||||
chatBox.setCharacterName('');
|
||||
}
|
||||
|
||||
const { setActiveTab } = useSideBarLeftStore.getState();
|
||||
setActiveTab('character');
|
||||
|
||||
console.log('[CharacterCard] 角色已删除,已切换到角色分页');
|
||||
|
||||
console.log('[CharacterCard] 角色已删除,已返回角色选择列表');
|
||||
} catch (err) {
|
||||
console.error('删除失败:', err);
|
||||
}
|
||||
|
||||
18
frontend/src/components/Studio/StudioContextBlockPopup.css
Normal file
18
frontend/src/components/Studio/StudioContextBlockPopup.css
Normal file
@@ -0,0 +1,18 @@
|
||||
.studio-context-block-popup__content {
|
||||
margin: 0;
|
||||
padding: var(--spacing-sm);
|
||||
border-radius: var(--radius-sm);
|
||||
border: 1px solid var(--color-border-light);
|
||||
background: var(--color-bg-tertiary);
|
||||
white-space: pre-wrap;
|
||||
word-break: break-word;
|
||||
font-size: 0.72rem;
|
||||
line-height: 1.45;
|
||||
max-height: none;
|
||||
}
|
||||
|
||||
.studio-context-block-popup__preview {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 4px;
|
||||
}
|
||||
33
frontend/src/components/Studio/StudioContextBlockPopup.jsx
Normal file
33
frontend/src/components/Studio/StudioContextBlockPopup.jsx
Normal file
@@ -0,0 +1,33 @@
|
||||
import React from 'react';
|
||||
|
||||
import StudioInsertionPopup from './StudioInsertionPopup';
|
||||
|
||||
import './StudioContextBlockPopup.css';
|
||||
|
||||
function StudioContextBlockPopup({ open, block, onClose }) {
|
||||
if (!block) return null;
|
||||
|
||||
return (
|
||||
<StudioInsertionPopup
|
||||
open={open}
|
||||
title={block.label}
|
||||
defaultExpanded
|
||||
onClose={onClose}
|
||||
>
|
||||
{({ expanded }) =>
|
||||
expanded ? (
|
||||
<pre className="studio-context-block-popup__content">
|
||||
{block.content}
|
||||
</pre>
|
||||
) : (
|
||||
<div className="studio-context-block-popup__preview">
|
||||
<span className="studio-insertion-popup__field-label">来源</span>
|
||||
<span className="studio-insertion-popup__field-value">{block.source}</span>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
</StudioInsertionPopup>
|
||||
);
|
||||
}
|
||||
|
||||
export default StudioContextBlockPopup;
|
||||
1184
frontend/src/components/Studio/StudioEditPage.css
Normal file
1184
frontend/src/components/Studio/StudioEditPage.css
Normal file
File diff suppressed because it is too large
Load Diff
764
frontend/src/components/Studio/StudioEditPage.jsx
Normal file
764
frontend/src/components/Studio/StudioEditPage.jsx
Normal file
@@ -0,0 +1,764 @@
|
||||
import React, { useEffect, useMemo, useState } from 'react';
|
||||
|
||||
import useStudioStore from '../../Store/Studio/StudioSlice';
|
||||
import FieldLabel from './edit/FieldLabel';
|
||||
import VariableChips from './edit/VariableChips';
|
||||
import {
|
||||
DEFAULT_THINKING_PROMPT,
|
||||
orderNodesByIds,
|
||||
sortNodesByLogic,
|
||||
} from './edit/variableUtils';
|
||||
import WorldbookInsertion from './edit/WorldbookInsertion';
|
||||
import ScoringDimensions from './edit/ScoringDimensions';
|
||||
|
||||
import './StudioEditPage.css';
|
||||
|
||||
/*
|
||||
* Node detail layout: primary config center + secondary fixed edge rail.
|
||||
* Industry refs: n8n (Parameters center + Settings sidebar), Retool inspector,
|
||||
* Figma right properties rail — main editing in fluid center, refs in ~300px rail.
|
||||
*/
|
||||
|
||||
const DEFAULT_TEMPLATE_ID = 'builtin.studio.example';
|
||||
|
||||
const WORKFLOW_GOAL_TIP =
|
||||
'此内容会在运行时被注入到 LLM 系统提示中,作为工作流整体目标,供各步骤参考。';
|
||||
|
||||
const WORKFLOW_DESC_TIP =
|
||||
'仅供人类阅读的简短说明,不会写入 LLM 提示。';
|
||||
|
||||
const NODE_SORT_STORAGE_KEY = 'studio-node-sort-mode';
|
||||
|
||||
function getStoredSortMode(projectId) {
|
||||
if (!projectId || typeof sessionStorage === 'undefined') return 'manual';
|
||||
return sessionStorage.getItem(`${NODE_SORT_STORAGE_KEY}:${projectId}`) || 'manual';
|
||||
}
|
||||
|
||||
function storeSortMode(projectId, mode) {
|
||||
if (!projectId || typeof sessionStorage === 'undefined') return;
|
||||
sessionStorage.setItem(`${NODE_SORT_STORAGE_KEY}:${projectId}`, mode);
|
||||
}
|
||||
function getSkillTemplate(skillTemplates, skillId) {
|
||||
return skillTemplates.find((t) => t.skillId === skillId);
|
||||
}
|
||||
|
||||
function StudioEditPage() {
|
||||
const {
|
||||
projects,
|
||||
workflowTemplates,
|
||||
currentProjectId,
|
||||
meta,
|
||||
pipeline,
|
||||
skillTemplates,
|
||||
selectedNodeId,
|
||||
loading,
|
||||
saving,
|
||||
error,
|
||||
saveMessage,
|
||||
setPipelineLocal,
|
||||
setMetaLocal,
|
||||
setSelectedNodeId,
|
||||
updateNode,
|
||||
updateNodeConfig,
|
||||
addNode,
|
||||
removeNode,
|
||||
fetchProject,
|
||||
savePipeline,
|
||||
createProject,
|
||||
deleteProject,
|
||||
renameProject,
|
||||
updateProjectMeta,
|
||||
clearSaveMessage,
|
||||
} = useStudioStore();
|
||||
|
||||
const [dragIndex, setDragIndex] = useState(null);
|
||||
const [showAddNode, setShowAddNode] = useState(false);
|
||||
const [showNewProjectModal, setShowNewProjectModal] = useState(false);
|
||||
const [showRenameProjectModal, setShowRenameProjectModal] = useState(false);
|
||||
const [newNodeSkillId, setNewNodeSkillId] = useState('studio.worldbook_entry');
|
||||
const [newNodeName, setNewNodeName] = useState('');
|
||||
const [newProjectName, setNewProjectName] = useState('');
|
||||
const [newProjectTemplateId, setNewProjectTemplateId] = useState(DEFAULT_TEMPLATE_ID);
|
||||
const [renameProjectName, setRenameProjectName] = useState('');
|
||||
const [nodeSortMode, setNodeSortMode] = useState('manual');
|
||||
const [manualOrder, setManualOrder] = useState([]);
|
||||
|
||||
useEffect(() => {
|
||||
if (!currentProjectId) return;
|
||||
setNodeSortMode(getStoredSortMode(currentProjectId));
|
||||
}, [currentProjectId]);
|
||||
|
||||
useEffect(() => {
|
||||
if (!pipeline?.nodes) {
|
||||
setManualOrder([]);
|
||||
return;
|
||||
}
|
||||
setManualOrder(pipeline.nodes.map((n) => n.id));
|
||||
}, [currentProjectId, pipeline?.nodes?.length]);
|
||||
|
||||
const displayNodes = useMemo(() => {
|
||||
const nodes = pipeline?.nodes || [];
|
||||
if (nodeSortMode === 'logic') {
|
||||
return sortNodesByLogic(pipeline);
|
||||
}
|
||||
return orderNodesByIds(nodes, manualOrder);
|
||||
}, [pipeline, nodeSortMode, manualOrder]);
|
||||
|
||||
const handleSortModeChange = (mode) => {
|
||||
setNodeSortMode(mode);
|
||||
if (currentProjectId) storeSortMode(currentProjectId, mode);
|
||||
if (mode === 'manual' && pipeline?.nodes) {
|
||||
setManualOrder(pipeline.nodes.map((n) => n.id));
|
||||
}
|
||||
};
|
||||
const selectedNode = pipeline?.nodes?.find((n) => n.id === selectedNodeId) ?? null;
|
||||
const selectedTpl = selectedNode
|
||||
? getSkillTemplate(skillTemplates, selectedNode.skillId)
|
||||
: null;
|
||||
const isWorldbook = selectedNode?.skillId === 'studio.worldbook_entry';
|
||||
const insertion = selectedNode?.config?.insertion || {};
|
||||
const scoring = selectedNode?.config?.scoring || { enabled: true, dimensions: [] };
|
||||
const dimensions = scoring.dimensions || [];
|
||||
|
||||
const templateOptions =
|
||||
workflowTemplates.length > 0
|
||||
? workflowTemplates
|
||||
: [{ id: DEFAULT_TEMPLATE_ID, name: '世界书条目创建' }];
|
||||
|
||||
const handleProjectChange = async (e) => {
|
||||
const id = e.target.value;
|
||||
if (id) await fetchProject(id);
|
||||
};
|
||||
|
||||
const openNewProjectModal = () => {
|
||||
setNewProjectName('');
|
||||
setNewProjectTemplateId(templateOptions[0]?.id || DEFAULT_TEMPLATE_ID);
|
||||
setShowNewProjectModal(true);
|
||||
};
|
||||
|
||||
const openRenameProjectModal = () => {
|
||||
setRenameProjectName(meta?.name || '');
|
||||
setShowRenameProjectModal(true);
|
||||
};
|
||||
|
||||
const handleCreateProject = async (e) => {
|
||||
e.preventDefault();
|
||||
const name = newProjectName.trim() || '新项目';
|
||||
const result = await createProject(name, newProjectTemplateId);
|
||||
if (result) {
|
||||
setShowNewProjectModal(false);
|
||||
setNewProjectName('');
|
||||
}
|
||||
};
|
||||
|
||||
const handleRenameProject = async (e) => {
|
||||
e.preventDefault();
|
||||
if (!currentProjectId) return;
|
||||
const name = renameProjectName.trim();
|
||||
if (!name) return;
|
||||
const result = await renameProject(currentProjectId, name);
|
||||
if (result) {
|
||||
setShowRenameProjectModal(false);
|
||||
}
|
||||
};
|
||||
|
||||
const handleDeleteProject = async () => {
|
||||
if (!currentProjectId || !meta) return;
|
||||
const confirmed = window.confirm(
|
||||
`确定删除项目「${meta.name}」?\n\n将同时删除该项目下的所有运行记录,此操作不可撤销。`
|
||||
);
|
||||
if (!confirmed) return;
|
||||
await deleteProject(currentProjectId);
|
||||
};
|
||||
|
||||
const handleDescriptionBlur = () => {
|
||||
if (!currentProjectId || !meta) return;
|
||||
updateProjectMeta(currentProjectId, { description: meta.description ?? '' });
|
||||
};
|
||||
|
||||
const handleDragStart = (index) => {
|
||||
if (nodeSortMode !== 'manual') return;
|
||||
setDragIndex(index);
|
||||
};
|
||||
|
||||
const handleDragOver = (e, index) => {
|
||||
if (nodeSortMode !== 'manual') return;
|
||||
e.preventDefault();
|
||||
if (dragIndex === null || dragIndex === index) return;
|
||||
|
||||
const ids = displayNodes.map((n) => n.id);
|
||||
const nextIds = [...ids];
|
||||
const [moved] = nextIds.splice(dragIndex, 1);
|
||||
nextIds.splice(index, 0, moved);
|
||||
|
||||
const nodeMap = Object.fromEntries((pipeline.nodes || []).map((n) => [n.id, n]));
|
||||
const reordered = nextIds.map((id) => nodeMap[id]).filter(Boolean);
|
||||
setPipelineLocal({ ...pipeline, nodes: reordered });
|
||||
setManualOrder(nextIds);
|
||||
setDragIndex(index);
|
||||
};
|
||||
const handleDragEnd = () => setDragIndex(null);
|
||||
|
||||
const toggleInputRef = (nodeId, ref, label) => {
|
||||
const node = pipeline.nodes.find((n) => n.id === nodeId);
|
||||
if (!node) return;
|
||||
const exists = (node.inputs || []).some((i) => i.ref === ref);
|
||||
let inputs;
|
||||
if (exists) {
|
||||
inputs = (node.inputs || []).filter((i) => i.ref !== ref);
|
||||
} else {
|
||||
inputs = [...(node.inputs || []), { ref, label }];
|
||||
}
|
||||
updateNode(nodeId, { inputs });
|
||||
};
|
||||
|
||||
const updateInsertion = (field, value) => {
|
||||
if (!selectedNode) return;
|
||||
const next = { ...(selectedNode.config?.insertion || {}), [field]: value };
|
||||
updateNodeConfig(selectedNode.id, { insertion: next });
|
||||
};
|
||||
|
||||
const updateRagConfig = (field, value) => {
|
||||
if (!selectedNode) return;
|
||||
const ragConfig = {
|
||||
...(insertion.ragConfig || {}),
|
||||
[field]: value,
|
||||
};
|
||||
updateInsertion('ragConfig', ragConfig);
|
||||
};
|
||||
|
||||
const updateScoring = (field, value) => {
|
||||
if (!selectedNode) return;
|
||||
const next = { ...(selectedNode.config?.scoring || {}), [field]: value };
|
||||
updateNodeConfig(selectedNode.id, { scoring: next });
|
||||
};
|
||||
|
||||
const addDimension = () => {
|
||||
const id = `dim-${Date.now()}`;
|
||||
updateScoring('dimensions', [
|
||||
...dimensions,
|
||||
{ id, name: '', criteria: '' },
|
||||
]);
|
||||
};
|
||||
|
||||
const updateDimension = (index, field, value) => {
|
||||
const next = dimensions.map((d, i) =>
|
||||
i === index ? { ...d, [field]: value } : d
|
||||
);
|
||||
updateScoring('dimensions', next);
|
||||
};
|
||||
|
||||
const removeDimension = (index) => {
|
||||
updateScoring(
|
||||
'dimensions',
|
||||
dimensions.filter((_, i) => i !== index)
|
||||
);
|
||||
};
|
||||
|
||||
if (loading && !pipeline) {
|
||||
return <div className="studio-edit-loading">加载中…</div>;
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="studio-edit-page">
|
||||
<header className="studio-edit-top">
|
||||
<div className="studio-edit-top__label studio-edit-top__label--project">
|
||||
<span className="studio-edit-top__label-text">项目</span>
|
||||
</div>
|
||||
<div className="studio-edit-top__label studio-edit-top__label--goal">
|
||||
<FieldLabel tip={WORKFLOW_GOAL_TIP}>工作流目标</FieldLabel>
|
||||
</div>
|
||||
<div className="studio-edit-top__label studio-edit-top__label--desc">
|
||||
<FieldLabel tip={WORKFLOW_DESC_TIP}>工作流简介</FieldLabel>
|
||||
</div>
|
||||
|
||||
<div className="studio-edit-top__control studio-edit-top__control--project">
|
||||
<select
|
||||
className="studio-edit-select studio-edit-select--project"
|
||||
value={currentProjectId || ''}
|
||||
onChange={handleProjectChange}
|
||||
aria-label="选择项目"
|
||||
>
|
||||
{projects.length === 0 && <option value="">无项目</option>}
|
||||
{projects.map((p) => (
|
||||
<option key={p.id} value={p.id}>
|
||||
{p.name}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
</div>
|
||||
<div className="studio-edit-top__control studio-edit-top__control--goal">
|
||||
<textarea
|
||||
className="studio-edit-textarea studio-edit-textarea--goal"
|
||||
rows={2}
|
||||
value={pipeline?.workflowGoal ?? ''}
|
||||
onChange={(e) =>
|
||||
setPipelineLocal({ ...pipeline, workflowGoal: e.target.value })
|
||||
}
|
||||
placeholder="描述本 Studio 项目要完成的整体设计目标…"
|
||||
aria-label="工作流目标"
|
||||
/>
|
||||
</div>
|
||||
<div className="studio-edit-top__control studio-edit-top__control--desc">
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input studio-edit-input--desc"
|
||||
value={meta?.description ?? ''}
|
||||
onChange={(e) => setMetaLocal({ description: e.target.value })}
|
||||
onBlur={handleDescriptionBlur}
|
||||
placeholder="简短备注"
|
||||
maxLength={120}
|
||||
disabled={!currentProjectId || saving}
|
||||
aria-label="工作流简介"
|
||||
/>
|
||||
</div>
|
||||
|
||||
<div className="studio-edit-top__actions">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-danger studio-edit-btn-sm"
|
||||
onClick={handleDeleteProject}
|
||||
disabled={!currentProjectId || loading || saving}
|
||||
title="删除当前项目及其运行记录"
|
||||
>
|
||||
删除项目
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-sm"
|
||||
onClick={openRenameProjectModal}
|
||||
disabled={!currentProjectId || loading || saving}
|
||||
>
|
||||
改名
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-sm"
|
||||
onClick={openNewProjectModal}
|
||||
disabled={loading}
|
||||
>
|
||||
新建项目
|
||||
</button>
|
||||
</div>
|
||||
</header>
|
||||
{(error || saveMessage) && (
|
||||
<div
|
||||
className={`studio-edit-banner ${error ? 'error' : 'success'}`}
|
||||
role="status"
|
||||
>
|
||||
{error || saveMessage}
|
||||
<button type="button" className="studio-edit-banner-close" onClick={clearSaveMessage}>
|
||||
×
|
||||
</button>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div className="studio-edit-main">
|
||||
<aside className="studio-edit-nodes">
|
||||
<div className="studio-edit-nodes-header">
|
||||
<h2 className="studio-edit-subtitle">节点列表</h2>
|
||||
<div className="studio-edit-nodes-header__actions">
|
||||
<div className="studio-edit-sort-toggle" role="group" aria-label="节点排序">
|
||||
<button
|
||||
type="button"
|
||||
className={`studio-edit-sort-btn${nodeSortMode === 'manual' ? ' active' : ''}`}
|
||||
onClick={() => handleSortModeChange('manual')}
|
||||
title="按拖拽/保存顺序排列"
|
||||
>
|
||||
手动
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
className={`studio-edit-sort-btn${nodeSortMode === 'logic' ? ' active' : ''}`}
|
||||
onClick={() => handleSortModeChange('logic')}
|
||||
title="按引用依赖拓扑排序"
|
||||
>
|
||||
逻辑
|
||||
</button>
|
||||
</div>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-sm"
|
||||
onClick={() => setShowAddNode(!showAddNode)}
|
||||
>
|
||||
添加节点
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
{showAddNode && (
|
||||
<div className="studio-edit-add-node">
|
||||
<select
|
||||
className="studio-edit-select"
|
||||
value={newNodeSkillId}
|
||||
onChange={(e) => setNewNodeSkillId(e.target.value)}
|
||||
>
|
||||
{skillTemplates.map((t) => (
|
||||
<option key={t.skillId} value={t.skillId}>
|
||||
{t.displayName}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
placeholder="展示名"
|
||||
value={newNodeName}
|
||||
onChange={(e) => setNewNodeName(e.target.value)}
|
||||
/>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-sm"
|
||||
onClick={() => {
|
||||
addNode(
|
||||
newNodeSkillId,
|
||||
newNodeName ||
|
||||
getSkillTemplate(skillTemplates, newNodeSkillId)?.displayName
|
||||
);
|
||||
setNewNodeName('');
|
||||
setShowAddNode(false);
|
||||
}}
|
||||
>
|
||||
确认
|
||||
</button>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<ul className="studio-edit-node-list">
|
||||
{displayNodes.map((node, index) => {
|
||||
const tpl = getSkillTemplate(skillTemplates, node.skillId);
|
||||
const isManualSort = nodeSortMode === 'manual';
|
||||
return (
|
||||
<li
|
||||
key={node.id}
|
||||
className={`studio-edit-node-item ${
|
||||
selectedNodeId === node.id ? 'selected' : ''
|
||||
} ${!node.enabled ? 'disabled' : ''}${!isManualSort ? ' no-drag' : ''}`}
|
||||
draggable={isManualSort}
|
||||
onDragStart={() => handleDragStart(index)}
|
||||
onDragOver={(e) => handleDragOver(e, index)}
|
||||
onDragEnd={handleDragEnd}
|
||||
onClick={() => setSelectedNodeId(node.id)}
|
||||
>
|
||||
{isManualSort ? (
|
||||
<span className="studio-edit-drag-handle" title="拖拽排序">
|
||||
⋮⋮
|
||||
</span>
|
||||
) : (
|
||||
<span className="studio-edit-drag-handle studio-edit-drag-handle--muted" title="逻辑排序模式下不可拖拽">
|
||||
↕
|
||||
</span>
|
||||
)} <div className="studio-edit-node-text">
|
||||
<span className="studio-edit-node-title">{node.displayName}</span>
|
||||
<span className="studio-edit-node-skill">
|
||||
{tpl?.displayName || node.skillId}
|
||||
</span>
|
||||
</div>
|
||||
<label
|
||||
className="studio-edit-switch"
|
||||
onClick={(e) => e.stopPropagation()}
|
||||
>
|
||||
<input
|
||||
type="checkbox"
|
||||
checked={node.enabled}
|
||||
onChange={(e) =>
|
||||
updateNode(node.id, { enabled: e.target.checked })
|
||||
}
|
||||
/>
|
||||
<span>启用</span>
|
||||
</label>
|
||||
</li>
|
||||
);
|
||||
})}
|
||||
</ul>
|
||||
</aside>
|
||||
|
||||
<section className="studio-edit-detail">
|
||||
{!selectedNode ? (
|
||||
<p className="studio-edit-empty">选择左侧节点进行编辑</p>
|
||||
) : (
|
||||
<div className="studio-edit-detail-inner">
|
||||
<div className="studio-edit-meta-strip">
|
||||
<label className="studio-edit-meta-strip__field">
|
||||
<span className="studio-edit-field-label">展示名</span>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={selectedNode.displayName}
|
||||
onChange={(e) =>
|
||||
updateNode(selectedNode.id, { displayName: e.target.value })
|
||||
}
|
||||
/>
|
||||
</label>
|
||||
<label className="studio-edit-meta-strip__field">
|
||||
<span className="studio-edit-field-label">模板类型</span>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
readOnly
|
||||
value={selectedTpl?.displayName || selectedNode.skillId}
|
||||
/>
|
||||
</label>
|
||||
{selectedTpl?.artifacts?.length > 0 && (
|
||||
<span className="studio-edit-meta-strip__hint">
|
||||
产物:
|
||||
{selectedTpl.artifacts.map((a) => a.displayName).join('、')}
|
||||
</span>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<div className="studio-edit-detail-body">
|
||||
<div className="studio-edit-primary">
|
||||
<div className="studio-edit-primary-card">
|
||||
<span className="studio-edit-primary-card__badge">主要设置</span>
|
||||
|
||||
{isWorldbook ? (
|
||||
<>
|
||||
<label className="studio-edit-field studio-edit-field--primary">
|
||||
<span className="studio-edit-field-label studio-edit-field-label--primary">
|
||||
<FieldLabel tip="本步骤要产出的内容目标,会注入模型系统提示">
|
||||
步骤目标
|
||||
</FieldLabel>
|
||||
</span>
|
||||
<textarea
|
||||
className="studio-edit-textarea studio-edit-textarea--primary"
|
||||
rows={5}
|
||||
value={selectedNode.config?.stepGoal ?? ''}
|
||||
onChange={(e) =>
|
||||
updateNodeConfig(selectedNode.id, {
|
||||
stepGoal: e.target.value,
|
||||
})
|
||||
}
|
||||
placeholder="描述本步骤要产出的内容…"
|
||||
/>
|
||||
</label>
|
||||
|
||||
<label className="studio-edit-field studio-edit-field--primary">
|
||||
<span className="studio-edit-field-label studio-edit-field-label--primary">
|
||||
<FieldLabel tip="引导模型在本步骤中的思考方式、关注点与推理路径">
|
||||
思考提示词
|
||||
</FieldLabel>
|
||||
</span>
|
||||
<textarea
|
||||
className="studio-edit-textarea studio-edit-textarea--primary"
|
||||
rows={5}
|
||||
value={selectedNode.config?.thinkingPrompt ?? ''}
|
||||
onChange={(e) =>
|
||||
updateNodeConfig(selectedNode.id, {
|
||||
thinkingPrompt: e.target.value,
|
||||
})
|
||||
}
|
||||
placeholder="引导模型如何思考本步骤…"
|
||||
/>
|
||||
{!selectedNode.config?.thinkingPrompt && (
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-sm studio-edit-think-default"
|
||||
onClick={() =>
|
||||
updateNodeConfig(selectedNode.id, {
|
||||
thinkingPrompt: DEFAULT_THINKING_PROMPT,
|
||||
})
|
||||
}
|
||||
>
|
||||
填入默认思考流程模板
|
||||
</button>
|
||||
)}
|
||||
</label>
|
||||
|
||||
<ScoringDimensions
|
||||
scoring={scoring}
|
||||
dimensions={dimensions}
|
||||
onUpdateScoring={updateScoring}
|
||||
onUpdateDimension={updateDimension}
|
||||
onRemoveDimension={removeDimension}
|
||||
onAddDimension={addDimension}
|
||||
/>
|
||||
</>
|
||||
) : selectedNode.skillId === 'studio.init_bind' ? (
|
||||
<>
|
||||
<p className="studio-edit-hint">
|
||||
运行时在引导区填写;编辑页可预览字段定义。
|
||||
</p>
|
||||
<ul className="studio-edit-dp-list studio-edit-dp-list--compact">
|
||||
{(selectedNode.displayParams || []).map((dp) => (
|
||||
<li key={dp.key}>
|
||||
{dp.label} ({dp.key})
|
||||
{dp.required ? ' · 必填' : ''}
|
||||
</li>
|
||||
))}
|
||||
</ul>
|
||||
</>
|
||||
) : null}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<aside className="studio-edit-edge-rail">
|
||||
{isWorldbook ? (
|
||||
<>
|
||||
<div className="studio-edit-edge-card">
|
||||
<VariableChips
|
||||
variant="compact"
|
||||
pipeline={pipeline}
|
||||
selectedNode={selectedNode}
|
||||
onToggleRef={(ref, label) =>
|
||||
toggleInputRef(selectedNode.id, ref, label)
|
||||
}
|
||||
/>
|
||||
</div>
|
||||
<div className="studio-edit-edge-card">
|
||||
<WorldbookInsertion
|
||||
insertion={insertion}
|
||||
onUpdateInsertion={updateInsertion}
|
||||
onUpdateRagConfig={updateRagConfig}
|
||||
/>
|
||||
</div>
|
||||
</>
|
||||
) : (
|
||||
<div className="studio-edit-edge-card studio-edit-edge-card--empty">
|
||||
<p className="studio-edit-edge-empty-tip">创建步骤无需引用变量</p>
|
||||
</div>
|
||||
)}
|
||||
</aside>
|
||||
</div>
|
||||
|
||||
<div className="studio-edit-detail-footer">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-danger"
|
||||
onClick={() => {
|
||||
if (window.confirm(`删除节点「${selectedNode.displayName}」?`)) {
|
||||
removeNode(selectedNode.id);
|
||||
}
|
||||
}}
|
||||
>
|
||||
删除节点
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn studio-edit-btn-primary"
|
||||
onClick={savePipeline}
|
||||
disabled={saving || !currentProjectId}
|
||||
>
|
||||
{saving ? '保存中…' : '保存'}
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
</section>
|
||||
</div>
|
||||
|
||||
{showNewProjectModal && (
|
||||
<div
|
||||
className="studio-modal-overlay"
|
||||
role="presentation"
|
||||
onClick={() => setShowNewProjectModal(false)}
|
||||
>
|
||||
<div
|
||||
className="studio-modal"
|
||||
role="dialog"
|
||||
aria-labelledby="new-project-title"
|
||||
onClick={(e) => e.stopPropagation()}
|
||||
>
|
||||
<h2 id="new-project-title" className="studio-modal-title">
|
||||
新建项目
|
||||
</h2>
|
||||
<form onSubmit={handleCreateProject}>
|
||||
<label className="studio-edit-label-block">
|
||||
项目名称
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={newProjectName}
|
||||
onChange={(e) => setNewProjectName(e.target.value)}
|
||||
placeholder="例如:帝国骑士维尔"
|
||||
autoFocus
|
||||
required
|
||||
/>
|
||||
</label>
|
||||
<label className="studio-edit-label-block">
|
||||
选择工作流模板
|
||||
<select
|
||||
className="studio-edit-select"
|
||||
value={newProjectTemplateId}
|
||||
onChange={(e) => setNewProjectTemplateId(e.target.value)}
|
||||
>
|
||||
{templateOptions.map((t) => (
|
||||
<option key={t.id} value={t.id}>
|
||||
{t.name}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
</label>
|
||||
<div className="studio-modal-actions">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn"
|
||||
onClick={() => setShowNewProjectModal(false)}
|
||||
>
|
||||
取消
|
||||
</button>
|
||||
<button
|
||||
type="submit"
|
||||
className="studio-edit-btn studio-edit-btn-primary"
|
||||
disabled={loading}
|
||||
>
|
||||
{loading ? '创建中…' : '创建'}
|
||||
</button>
|
||||
</div>
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{showRenameProjectModal && (
|
||||
<div
|
||||
className="studio-modal-overlay"
|
||||
role="presentation"
|
||||
onClick={() => setShowRenameProjectModal(false)}
|
||||
>
|
||||
<div
|
||||
className="studio-modal"
|
||||
role="dialog"
|
||||
aria-labelledby="rename-project-title"
|
||||
onClick={(e) => e.stopPropagation()}
|
||||
>
|
||||
<h2 id="rename-project-title" className="studio-modal-title">
|
||||
改名
|
||||
</h2>
|
||||
<form onSubmit={handleRenameProject}>
|
||||
<label className="studio-edit-label-block">
|
||||
项目名称
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={renameProjectName}
|
||||
onChange={(e) => setRenameProjectName(e.target.value)}
|
||||
autoFocus
|
||||
required
|
||||
maxLength={120}
|
||||
/>
|
||||
</label>
|
||||
<div className="studio-modal-actions">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn"
|
||||
onClick={() => setShowRenameProjectModal(false)}
|
||||
>
|
||||
取消
|
||||
</button>
|
||||
<button
|
||||
type="submit"
|
||||
className="studio-edit-btn studio-edit-btn-primary"
|
||||
disabled={saving}
|
||||
>
|
||||
{saving ? '保存中…' : '确定'}
|
||||
</button>
|
||||
</div>
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default StudioEditPage;
|
||||
107
frontend/src/components/Studio/StudioInsertionPopup.css
Normal file
107
frontend/src/components/Studio/StudioInsertionPopup.css
Normal file
@@ -0,0 +1,107 @@
|
||||
.studio-insertion-popup {
|
||||
position: fixed;
|
||||
z-index: 1200;
|
||||
width: 280px;
|
||||
max-width: calc(100vw - 32px);
|
||||
border-radius: var(--radius-md);
|
||||
border: 1px solid var(--color-border);
|
||||
background: var(--color-bg-secondary);
|
||||
box-shadow: 0 8px 32px rgba(0, 0, 0, 0.18);
|
||||
pointer-events: auto;
|
||||
}
|
||||
|
||||
.studio-insertion-popup.is-expanded {
|
||||
width: min(420px, calc(100vw - 32px));
|
||||
max-height: min(70vh, 520px);
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__header {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: space-between;
|
||||
gap: var(--spacing-sm);
|
||||
padding: var(--spacing-xs) var(--spacing-sm);
|
||||
border-bottom: 1px solid var(--color-border-light);
|
||||
background: var(--color-bg-tertiary);
|
||||
border-radius: var(--radius-md) var(--radius-md) 0 0;
|
||||
cursor: grab;
|
||||
user-select: none;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__header:active {
|
||||
cursor: grabbing;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__title {
|
||||
font-size: 0.75rem;
|
||||
font-weight: 600;
|
||||
color: var(--color-text-primary);
|
||||
}
|
||||
|
||||
.studio-insertion-popup__actions {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 4px;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__btn {
|
||||
padding: 2px 8px;
|
||||
border: 1px solid var(--color-border-light);
|
||||
border-radius: var(--radius-sm);
|
||||
background: var(--color-bg-secondary);
|
||||
font-size: 0.65rem;
|
||||
color: var(--color-text-muted);
|
||||
cursor: pointer;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__btn:hover {
|
||||
border-color: var(--color-accent);
|
||||
color: var(--color-text-primary);
|
||||
}
|
||||
|
||||
.studio-insertion-popup__btn--close {
|
||||
font-size: 1rem;
|
||||
line-height: 1;
|
||||
padding: 0 6px;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__body {
|
||||
padding: var(--spacing-sm);
|
||||
overflow: auto;
|
||||
font-size: 0.75rem;
|
||||
}
|
||||
|
||||
.studio-insertion-popup.is-expanded .studio-insertion-popup__body {
|
||||
flex: 1;
|
||||
min-height: 0;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__preview {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 4px;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__field-label {
|
||||
font-size: 0.65rem;
|
||||
color: var(--color-text-muted);
|
||||
}
|
||||
|
||||
.studio-insertion-popup__field-value {
|
||||
font-size: 0.8rem;
|
||||
color: var(--color-text-primary);
|
||||
line-height: 1.4;
|
||||
}
|
||||
|
||||
.studio-insertion-popup__full {
|
||||
margin: 0;
|
||||
padding: var(--spacing-sm);
|
||||
border-radius: var(--radius-sm);
|
||||
border: 1px solid var(--color-border-light);
|
||||
background: var(--color-bg-tertiary);
|
||||
white-space: pre-wrap;
|
||||
font-size: 0.72rem;
|
||||
max-height: none;
|
||||
}
|
||||
95
frontend/src/components/Studio/StudioInsertionPopup.jsx
Normal file
95
frontend/src/components/Studio/StudioInsertionPopup.jsx
Normal file
@@ -0,0 +1,95 @@
|
||||
import React, { useCallback, useEffect, useRef, useState } from 'react';
|
||||
|
||||
import './StudioInsertionPopup.css';
|
||||
|
||||
const DEFAULT_POS = { x: 72, y: 96 };
|
||||
|
||||
function StudioInsertionPopup({ open, title, onClose, defaultExpanded = false, children }) {
|
||||
const popupRef = useRef(null);
|
||||
const dragRef = useRef(null);
|
||||
const [pos, setPos] = useState(DEFAULT_POS);
|
||||
const [expanded, setExpanded] = useState(defaultExpanded);
|
||||
|
||||
useEffect(() => {
|
||||
if (open) {
|
||||
setExpanded(defaultExpanded);
|
||||
setPos(DEFAULT_POS);
|
||||
}
|
||||
}, [open, defaultExpanded]);
|
||||
|
||||
const handleDragStart = useCallback((e) => {
|
||||
if (e.button !== 0) return;
|
||||
e.preventDefault();
|
||||
const startX = e.clientX;
|
||||
const startY = e.clientY;
|
||||
const origin = { ...pos };
|
||||
|
||||
const onMove = (ev) => {
|
||||
setPos({
|
||||
x: origin.x + (ev.clientX - startX),
|
||||
y: origin.y + (ev.clientY - startY),
|
||||
});
|
||||
};
|
||||
|
||||
const onUp = () => {
|
||||
window.removeEventListener('mousemove', onMove);
|
||||
window.removeEventListener('mouseup', onUp);
|
||||
dragRef.current = null;
|
||||
};
|
||||
|
||||
dragRef.current = { startX, startY, origin };
|
||||
window.addEventListener('mousemove', onMove);
|
||||
window.addEventListener('mouseup', onUp);
|
||||
}, [pos]);
|
||||
|
||||
useEffect(() => {
|
||||
return () => {
|
||||
if (dragRef.current) {
|
||||
window.removeEventListener('mousemove', () => {});
|
||||
window.removeEventListener('mouseup', () => {});
|
||||
}
|
||||
};
|
||||
}, []);
|
||||
|
||||
if (!open) return null;
|
||||
|
||||
return (
|
||||
<div
|
||||
ref={popupRef}
|
||||
className={`studio-insertion-popup${expanded ? ' is-expanded' : ''}`}
|
||||
style={{ left: pos.x, top: pos.y }}
|
||||
role="dialog"
|
||||
aria-label={title}
|
||||
>
|
||||
<div
|
||||
className="studio-insertion-popup__header"
|
||||
onMouseDown={handleDragStart}
|
||||
>
|
||||
<span className="studio-insertion-popup__title">{title}</span>
|
||||
<div className="studio-insertion-popup__actions">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-insertion-popup__btn"
|
||||
onClick={() => setExpanded((v) => !v)}
|
||||
title={expanded ? '收起详情' : '展开详情'}
|
||||
>
|
||||
{expanded ? '收起' : '展开'}
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-insertion-popup__btn studio-insertion-popup__btn--close"
|
||||
onClick={onClose}
|
||||
aria-label="关闭"
|
||||
>
|
||||
×
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
<div className="studio-insertion-popup__body">
|
||||
{children({ expanded })}
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default StudioInsertionPopup;
|
||||
70
frontend/src/components/Studio/StudioPage.css
Normal file
70
frontend/src/components/Studio/StudioPage.css
Normal file
@@ -0,0 +1,70 @@
|
||||
.studio-page {
|
||||
flex: 1;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
min-height: 0;
|
||||
background: var(--color-bg-primary);
|
||||
color: var(--color-text-primary);
|
||||
}
|
||||
|
||||
.studio-header {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: space-between;
|
||||
gap: var(--spacing-lg);
|
||||
padding: var(--spacing-md) var(--spacing-lg);
|
||||
border-bottom: 1px solid var(--color-border-light);
|
||||
background: var(--color-bg-secondary);
|
||||
flex-shrink: 0;
|
||||
}
|
||||
|
||||
.studio-header-title {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: var(--spacing-sm);
|
||||
}
|
||||
|
||||
.studio-header-icon {
|
||||
font-size: 1.25rem;
|
||||
}
|
||||
|
||||
.studio-title {
|
||||
margin: 0;
|
||||
font-size: 1.1rem;
|
||||
font-weight: 600;
|
||||
}
|
||||
|
||||
.studio-tabs {
|
||||
display: flex;
|
||||
gap: var(--spacing-xs);
|
||||
}
|
||||
|
||||
.studio-tab {
|
||||
padding: var(--spacing-sm) var(--spacing-md);
|
||||
border: 1px solid var(--color-border);
|
||||
border-radius: var(--radius-md);
|
||||
background: var(--color-bg-tertiary);
|
||||
color: var(--color-text-secondary);
|
||||
cursor: pointer;
|
||||
font-size: 0.875rem;
|
||||
transition: background var(--transition-fast), color var(--transition-fast);
|
||||
}
|
||||
|
||||
.studio-tab:hover {
|
||||
color: var(--color-text-primary);
|
||||
border-color: var(--color-border-focus);
|
||||
}
|
||||
|
||||
.studio-tab.active {
|
||||
background: var(--color-accent-light);
|
||||
border-color: var(--color-accent);
|
||||
color: var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-body {
|
||||
flex: 1;
|
||||
min-height: 0;
|
||||
overflow: hidden;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
}
|
||||
34
frontend/src/components/Studio/StudioPage.jsx
Normal file
34
frontend/src/components/Studio/StudioPage.jsx
Normal file
@@ -0,0 +1,34 @@
|
||||
import React, { useEffect } from 'react';
|
||||
|
||||
import useAppLayoutStore from '../../Store/AppLayoutSlice';
|
||||
|
||||
|
||||
|
||||
/**
|
||||
|
||||
* 兼容旧路由:studio → studio_edit
|
||||
|
||||
*/
|
||||
|
||||
function StudioPage() {
|
||||
|
||||
const setActivePage = useAppLayoutStore((s) => s.setActivePage);
|
||||
|
||||
|
||||
|
||||
useEffect(() => {
|
||||
|
||||
setActivePage('studio_edit');
|
||||
|
||||
}, [setActivePage]);
|
||||
|
||||
|
||||
|
||||
return null;
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
export default StudioPage;
|
||||
|
||||
434
frontend/src/components/Studio/StudioRunChat.css
Normal file
434
frontend/src/components/Studio/StudioRunChat.css
Normal file
@@ -0,0 +1,434 @@
|
||||
.studio-run-chat {
|
||||
flex: 1;
|
||||
min-height: 0;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
overflow: hidden;
|
||||
}
|
||||
|
||||
.studio-run-chat-messages {
|
||||
flex: 1;
|
||||
overflow-y: auto;
|
||||
padding: var(--spacing-lg);
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: var(--spacing-sm);
|
||||
min-height: 0;
|
||||
}
|
||||
|
||||
.studio-run-chat-empty {
|
||||
flex: 1;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
color: var(--color-text-muted);
|
||||
font-size: 0.9rem;
|
||||
}
|
||||
|
||||
.studio-run-chat-message {
|
||||
padding: var(--spacing-md) var(--spacing-lg);
|
||||
border-radius: var(--radius-md);
|
||||
line-height: 1.6;
|
||||
}
|
||||
|
||||
.studio-run-chat-message.role-user {
|
||||
align-self: flex-end;
|
||||
max-width: 85%;
|
||||
background: linear-gradient(to right, rgba(102, 126, 234, 0.08), rgba(102, 126, 234, 0.15));
|
||||
border-left: 3px solid var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-run-chat-message.role-assistant,
|
||||
.studio-run-chat-message.role-system {
|
||||
align-self: stretch;
|
||||
background: rgba(255, 255, 255, 0.6);
|
||||
}
|
||||
|
||||
[data-color-theme='dark'] .studio-run-chat-message.role-assistant,
|
||||
[data-color-theme='dark'] .studio-run-chat-message.role-system {
|
||||
background-color: rgba(45, 45, 48, 0.5);
|
||||
color: var(--color-text-primary);
|
||||
}
|
||||
|
||||
.studio-run-chat-history {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 2px;
|
||||
margin-bottom: var(--spacing-md);
|
||||
opacity: 0.35;
|
||||
max-height: 120px;
|
||||
overflow: hidden;
|
||||
pointer-events: none;
|
||||
user-select: none;
|
||||
}
|
||||
|
||||
.studio-run-chat-history-item {
|
||||
display: flex;
|
||||
gap: var(--spacing-xs);
|
||||
font-size: 0.72rem;
|
||||
line-height: 1.35;
|
||||
color: var(--color-text-muted);
|
||||
white-space: nowrap;
|
||||
overflow: hidden;
|
||||
text-overflow: ellipsis;
|
||||
}
|
||||
|
||||
.studio-run-chat-history-role {
|
||||
flex-shrink: 0;
|
||||
font-weight: 600;
|
||||
}
|
||||
|
||||
.studio-run-chat-history-text {
|
||||
overflow: hidden;
|
||||
text-overflow: ellipsis;
|
||||
}
|
||||
|
||||
.studio-run-chat-focus {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: var(--spacing-md);
|
||||
min-height: min(40vh, 320px);
|
||||
}
|
||||
|
||||
.studio-run-chat-last-user {
|
||||
display: flex;
|
||||
gap: var(--spacing-sm);
|
||||
align-items: flex-start;
|
||||
padding: var(--spacing-sm) var(--spacing-md);
|
||||
border-radius: var(--radius-md);
|
||||
background: rgba(102, 126, 234, 0.08);
|
||||
border-left: 3px solid var(--color-accent);
|
||||
}
|
||||
|
||||
[data-color-theme='dark'] .studio-run-chat-last-user {
|
||||
background: rgba(212, 167, 106, 0.1);
|
||||
}
|
||||
|
||||
.studio-run-chat-last-user--pending {
|
||||
opacity: 0.85;
|
||||
}
|
||||
|
||||
.studio-run-chat-last-user-label {
|
||||
flex-shrink: 0;
|
||||
font-size: 0.7rem;
|
||||
font-weight: 600;
|
||||
color: var(--color-text-muted);
|
||||
}
|
||||
|
||||
.studio-run-chat-last-user-text {
|
||||
font-size: 0.85rem;
|
||||
color: var(--color-text-primary);
|
||||
line-height: 1.5;
|
||||
white-space: pre-wrap;
|
||||
}
|
||||
|
||||
.studio-run-chat-thinking,
|
||||
.studio-run-chat-summary {
|
||||
padding: var(--spacing-md) var(--spacing-lg);
|
||||
border-radius: var(--radius-md);
|
||||
border: 1px solid var(--color-border-light);
|
||||
background: var(--color-bg-primary);
|
||||
}
|
||||
|
||||
[data-color-theme='dark'] .studio-run-chat-thinking,
|
||||
[data-color-theme='dark'] .studio-run-chat-summary {
|
||||
background-color: rgba(45, 45, 48, 0.5);
|
||||
border-color: var(--color-border);
|
||||
}
|
||||
|
||||
.studio-run-chat-thinking--pending {
|
||||
border-style: dashed;
|
||||
}
|
||||
|
||||
.studio-run-chat-block-label {
|
||||
display: block;
|
||||
font-size: 0.72rem;
|
||||
font-weight: 600;
|
||||
color: var(--color-text-muted);
|
||||
text-transform: uppercase;
|
||||
letter-spacing: 0.04em;
|
||||
margin-bottom: var(--spacing-sm);
|
||||
}
|
||||
|
||||
.studio-run-chat-block-body {
|
||||
font-size: 0.95rem;
|
||||
color: var(--color-text-primary);
|
||||
line-height: 1.6;
|
||||
}
|
||||
|
||||
.studio-run-chat-message-role {
|
||||
display: block;
|
||||
font-size: 0.7rem;
|
||||
color: var(--color-text-muted);
|
||||
margin-bottom: var(--spacing-xs);
|
||||
}
|
||||
|
||||
.studio-run-chat-message-body {
|
||||
font-size: 0.95rem;
|
||||
}
|
||||
|
||||
.studio-run-chat-plain {
|
||||
white-space: pre-wrap;
|
||||
}
|
||||
|
||||
/* Input area — aligned with ChatBox */
|
||||
.studio-run-chat-input-wrapper {
|
||||
position: sticky;
|
||||
bottom: 0;
|
||||
left: 0;
|
||||
right: 0;
|
||||
flex-shrink: 0;
|
||||
padding: var(--spacing-md) var(--spacing-lg);
|
||||
border-top: 1px solid var(--color-border-light);
|
||||
background-color: var(--color-bg-primary);
|
||||
z-index: var(--z-divider);
|
||||
}
|
||||
|
||||
.studio-run-chat-input-container {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: var(--spacing-sm);
|
||||
padding: var(--spacing-xs) var(--spacing-sm);
|
||||
background-color: var(--color-bg-elevated);
|
||||
border: 1px solid var(--color-border);
|
||||
border-radius: var(--radius-md);
|
||||
box-shadow: 0 2px 8px rgba(0, 0, 0, 0.06);
|
||||
transition: all 0.2s ease;
|
||||
}
|
||||
|
||||
.studio-run-chat-input-container:focus-within {
|
||||
border-color: var(--color-accent);
|
||||
box-shadow: 0 2px 12px rgba(0, 0, 0, 0.1);
|
||||
}
|
||||
|
||||
[data-color-theme='dark'] .studio-run-chat-input-container {
|
||||
background-color: var(--color-bg-tertiary);
|
||||
box-shadow: none;
|
||||
}
|
||||
|
||||
[data-color-theme='dark'] .studio-run-chat-input-container:focus-within {
|
||||
box-shadow: 0 0 0 1px var(--color-accent);
|
||||
}
|
||||
|
||||
[data-color-theme='dark'] .studio-run-chat-options-toggle:hover,
|
||||
[data-color-theme='dark'] .studio-run-chat-options-toggle.active {
|
||||
background-color: var(--color-accent-ultra-light);
|
||||
}
|
||||
|
||||
[data-color-theme='dark'] .studio-run-chat-textarea:focus {
|
||||
background-color: transparent;
|
||||
}
|
||||
|
||||
.studio-run-chat-options-wrapper {
|
||||
position: relative;
|
||||
flex-shrink: 0;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
}
|
||||
|
||||
.studio-run-chat-options-toggle {
|
||||
width: 36px;
|
||||
height: 36px;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
background: transparent;
|
||||
border: none;
|
||||
border-radius: var(--radius-md);
|
||||
color: var(--color-text-muted);
|
||||
font-size: 1.3rem;
|
||||
line-height: 1;
|
||||
cursor: pointer;
|
||||
transition: all 0.2s ease;
|
||||
}
|
||||
|
||||
.studio-run-chat-options-toggle:hover {
|
||||
background-color: rgba(102, 126, 234, 0.08);
|
||||
color: var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-run-chat-options-toggle.active {
|
||||
background-color: rgba(102, 126, 234, 0.12);
|
||||
color: var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-run-chat-options {
|
||||
position: absolute;
|
||||
bottom: calc(100% + var(--spacing-sm));
|
||||
left: 0;
|
||||
min-width: 220px;
|
||||
padding: var(--spacing-sm);
|
||||
border: 1px solid var(--color-border);
|
||||
border-radius: var(--radius-lg);
|
||||
background: var(--color-bg-elevated);
|
||||
box-shadow: 0 8px 24px rgba(0, 0, 0, 0.15);
|
||||
z-index: var(--z-dropdown-menu);
|
||||
backdrop-filter: blur(10px);
|
||||
}
|
||||
|
||||
.studio-run-chat-options-title {
|
||||
font-size: 0.7rem;
|
||||
font-weight: 600;
|
||||
color: var(--color-text-muted);
|
||||
text-transform: uppercase;
|
||||
letter-spacing: 0.5px;
|
||||
padding: var(--spacing-xs) var(--spacing-sm);
|
||||
margin-bottom: var(--spacing-xs);
|
||||
}
|
||||
|
||||
.studio-run-chat-render-btn {
|
||||
width: 100%;
|
||||
padding: var(--spacing-sm) var(--spacing-md);
|
||||
background: transparent;
|
||||
color: var(--color-text-primary);
|
||||
border: 1px solid var(--color-border);
|
||||
border-radius: var(--radius-md);
|
||||
font-size: 0.85rem;
|
||||
font-weight: 500;
|
||||
text-align: left;
|
||||
cursor: pointer;
|
||||
transition: all 0.15s ease;
|
||||
}
|
||||
|
||||
.studio-run-chat-render-btn:hover {
|
||||
background-color: rgba(102, 126, 234, 0.08);
|
||||
border-color: var(--color-accent);
|
||||
color: var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-run-chat-option-checkbox {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: var(--spacing-sm);
|
||||
padding: var(--spacing-sm) var(--spacing-md);
|
||||
cursor: pointer;
|
||||
border-radius: var(--radius-md);
|
||||
transition: all 0.15s ease;
|
||||
margin-bottom: var(--spacing-xs);
|
||||
}
|
||||
|
||||
.studio-run-chat-option-checkbox:hover {
|
||||
background-color: rgba(102, 126, 234, 0.06);
|
||||
}
|
||||
|
||||
.studio-run-chat-option-checkbox input[type="checkbox"] {
|
||||
display: none;
|
||||
}
|
||||
|
||||
.studio-run-chat-checkmark {
|
||||
width: 16px;
|
||||
height: 16px;
|
||||
border: 2px solid var(--color-border);
|
||||
border-radius: 3px;
|
||||
position: relative;
|
||||
transition: all 0.15s ease;
|
||||
flex-shrink: 0;
|
||||
}
|
||||
|
||||
.studio-run-chat-option-checkbox:hover .studio-run-chat-checkmark {
|
||||
border-color: var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-run-chat-option-checkbox input:checked + .studio-run-chat-checkmark {
|
||||
background-color: var(--color-accent);
|
||||
border-color: var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-run-chat-option-checkbox input:checked + .studio-run-chat-checkmark::after {
|
||||
content: '';
|
||||
position: absolute;
|
||||
left: 4px;
|
||||
top: 1px;
|
||||
width: 4px;
|
||||
height: 8px;
|
||||
border: solid white;
|
||||
border-width: 0 2px 2px 0;
|
||||
transform: rotate(45deg);
|
||||
}
|
||||
|
||||
.studio-run-chat-option-label {
|
||||
font-size: 0.85rem;
|
||||
color: var(--color-text-primary);
|
||||
}
|
||||
|
||||
.studio-run-chat-pending {
|
||||
color: var(--color-text-muted);
|
||||
font-style: italic;
|
||||
}
|
||||
|
||||
.studio-run-chat-message.is-pending {
|
||||
opacity: 0.85;
|
||||
}
|
||||
|
||||
.studio-run-chat-input-area {
|
||||
flex: 1;
|
||||
min-width: 0;
|
||||
position: relative;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
}
|
||||
|
||||
.studio-run-chat-textarea {
|
||||
width: 100%;
|
||||
padding: var(--spacing-sm) var(--spacing-md);
|
||||
border: none;
|
||||
border-radius: var(--radius-md);
|
||||
background-color: transparent;
|
||||
color: var(--color-text-primary);
|
||||
font-family: inherit;
|
||||
font-size: 0.95rem;
|
||||
resize: none;
|
||||
overflow-y: auto;
|
||||
min-height: 36px;
|
||||
max-height: 300px;
|
||||
line-height: 1.6;
|
||||
display: block;
|
||||
height: auto;
|
||||
field-sizing: content;
|
||||
}
|
||||
|
||||
.studio-run-chat-textarea:focus {
|
||||
outline: none;
|
||||
background-color: rgba(102, 126, 234, 0.03);
|
||||
}
|
||||
|
||||
.studio-run-chat-textarea::placeholder {
|
||||
color: var(--color-text-muted);
|
||||
opacity: 0.5;
|
||||
}
|
||||
|
||||
.studio-run-chat-send {
|
||||
width: 40px;
|
||||
height: 40px;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
justify-content: center;
|
||||
flex-shrink: 0;
|
||||
border: none;
|
||||
border-radius: var(--radius-md);
|
||||
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
||||
color: #fff;
|
||||
font-size: 1.3rem;
|
||||
line-height: 1;
|
||||
cursor: pointer;
|
||||
transition: all 0.2s ease;
|
||||
box-shadow: 0 2px 4px rgba(102, 126, 234, 0.2);
|
||||
align-self: center;
|
||||
}
|
||||
|
||||
.studio-run-chat-send:hover:not(:disabled) {
|
||||
background: linear-gradient(135deg, #5568d3 0%, #6a3f8f 100%);
|
||||
transform: translateY(-1px);
|
||||
box-shadow: 0 4px 8px rgba(102, 126, 234, 0.3);
|
||||
}
|
||||
|
||||
.studio-run-chat-send:active:not(:disabled) {
|
||||
transform: translateY(0);
|
||||
}
|
||||
|
||||
.studio-run-chat-send:disabled {
|
||||
opacity: 0.4;
|
||||
cursor: not-allowed;
|
||||
transform: none;
|
||||
box-shadow: none;
|
||||
}
|
||||
257
frontend/src/components/Studio/StudioRunChat.jsx
Normal file
257
frontend/src/components/Studio/StudioRunChat.jsx
Normal file
@@ -0,0 +1,257 @@
|
||||
import React, { useEffect, useRef, useState } from 'react';
|
||||
|
||||
import MarkdownRenderer from '../shared/MarkdownRenderer';
|
||||
import { findLastUserMessageIndex } from './studioRunUtils';
|
||||
|
||||
import './StudioRunChat.css';
|
||||
|
||||
const RENDER_MODES = ['none', 'markdown', 'html'];
|
||||
const RENDER_MODE_LABELS = {
|
||||
none: '📄 纯文本',
|
||||
html: '🌐 HTML',
|
||||
markdown: '📝 Markdown',
|
||||
};
|
||||
|
||||
function renderBody(text, renderMode, isUser) {
|
||||
if (!text) return null;
|
||||
if (renderMode === 'html' && !isUser) {
|
||||
const hasHtmlTags = /<[a-z][\s\S]*>/i.test(text);
|
||||
if (hasHtmlTags) {
|
||||
return <div dangerouslySetInnerHTML={{ __html: text }} />;
|
||||
}
|
||||
return <div className="studio-run-chat-plain">{text}</div>;
|
||||
}
|
||||
if (renderMode === 'markdown') {
|
||||
return <MarkdownRenderer content={text} />;
|
||||
}
|
||||
return <div className="studio-run-chat-plain">{text}</div>;
|
||||
}
|
||||
|
||||
function StudioRunChat({
|
||||
stepMessages = [],
|
||||
thinking,
|
||||
evaluation,
|
||||
pendingUserText,
|
||||
inputValue,
|
||||
onInputChange,
|
||||
onSend,
|
||||
disabled = false,
|
||||
sending = false,
|
||||
streamOutput = false,
|
||||
onStreamOutputChange,
|
||||
placeholder = '输入消息…',
|
||||
}) {
|
||||
const containerRef = useRef(null);
|
||||
const focusAnchorRef = useRef(null);
|
||||
const optionsRef = useRef(null);
|
||||
const [showOptions, setShowOptions] = useState(false);
|
||||
const [renderMode, setRenderMode] = useState('markdown');
|
||||
|
||||
const lastUserIdx = findLastUserMessageIndex(stepMessages);
|
||||
const historyMessages =
|
||||
lastUserIdx > 0 ? stepMessages.slice(0, lastUserIdx) : [];
|
||||
const lastUserMessage =
|
||||
lastUserIdx >= 0 ? stepMessages[lastUserIdx] : pendingUserText
|
||||
? { role: 'user', content: pendingUserText }
|
||||
: null;
|
||||
|
||||
const summaryText =
|
||||
evaluation ||
|
||||
(lastUserIdx >= 0 && stepMessages[lastUserIdx + 1]?.role === 'assistant'
|
||||
? stepMessages[lastUserIdx + 1].content
|
||||
: null);
|
||||
|
||||
const hasFocusContent =
|
||||
thinking || summaryText || pendingUserText || sending || lastUserMessage;
|
||||
|
||||
useEffect(() => {
|
||||
const container = containerRef.current;
|
||||
const anchor = focusAnchorRef.current;
|
||||
if (!container || !anchor) return;
|
||||
container.scrollTop = Math.max(0, anchor.offsetTop - container.offsetTop);
|
||||
}, [stepMessages, thinking, evaluation, pendingUserText, sending]);
|
||||
|
||||
useEffect(() => {
|
||||
const handleClickOutside = (event) => {
|
||||
if (
|
||||
optionsRef.current &&
|
||||
!optionsRef.current.contains(event.target) &&
|
||||
!event.target.closest('.studio-run-chat-options-toggle')
|
||||
) {
|
||||
setShowOptions(false);
|
||||
}
|
||||
};
|
||||
if (showOptions) {
|
||||
document.addEventListener('mousedown', handleClickOutside);
|
||||
}
|
||||
return () => document.removeEventListener('mousedown', handleClickOutside);
|
||||
}, [showOptions]);
|
||||
|
||||
const handleInputHeight = (e) => {
|
||||
const textarea = e.target;
|
||||
textarea.style.height = 'auto';
|
||||
textarea.style.height = `${Math.min(textarea.scrollHeight, 300)}px`;
|
||||
};
|
||||
|
||||
const handleSubmit = (e) => {
|
||||
e.preventDefault();
|
||||
if (disabled || sending || !inputValue.trim()) return;
|
||||
onSend(inputValue.trim());
|
||||
const textarea = e.target.querySelector('.studio-run-chat-textarea');
|
||||
if (textarea) textarea.style.height = 'auto';
|
||||
};
|
||||
|
||||
const handleKeyDown = (e) => {
|
||||
if (e.key === 'Enter' && !e.shiftKey) {
|
||||
e.preventDefault();
|
||||
if (!disabled && !sending && inputValue.trim()) {
|
||||
onSend(inputValue.trim());
|
||||
e.target.style.height = 'auto';
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
const cycleRenderMode = () => {
|
||||
const idx = RENDER_MODES.indexOf(renderMode);
|
||||
setRenderMode(RENDER_MODES[(idx + 1) % RENDER_MODES.length]);
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="studio-run-chat">
|
||||
<div className="studio-run-chat-messages" ref={containerRef}>
|
||||
{historyMessages.length > 0 && (
|
||||
<div className="studio-run-chat-history" aria-hidden="true">
|
||||
{historyMessages.map((msg) => (
|
||||
<div
|
||||
key={msg.id}
|
||||
className={`studio-run-chat-history-item role-${msg.role}`}
|
||||
>
|
||||
<span className="studio-run-chat-history-role">
|
||||
{msg.role === 'user' ? '你' : '助手'}
|
||||
</span>
|
||||
<span className="studio-run-chat-history-text">{msg.content}</span>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div ref={focusAnchorRef} className="studio-run-chat-focus">
|
||||
{!hasFocusContent ? (
|
||||
<div className="studio-run-chat-empty">
|
||||
暂无本轮对话,在下方输入开始交流
|
||||
</div>
|
||||
) : (
|
||||
<>
|
||||
{lastUserMessage && !pendingUserText && (
|
||||
<div className="studio-run-chat-last-user">
|
||||
<span className="studio-run-chat-last-user-label">你</span>
|
||||
<span className="studio-run-chat-last-user-text">
|
||||
{lastUserMessage.content}
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{pendingUserText && (
|
||||
<div className="studio-run-chat-last-user studio-run-chat-last-user--pending">
|
||||
<span className="studio-run-chat-last-user-label">你</span>
|
||||
<span className="studio-run-chat-last-user-text">
|
||||
{pendingUserText}
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{sending && !thinking && (
|
||||
<div className="studio-run-chat-thinking studio-run-chat-thinking--pending">
|
||||
<span className="studio-run-chat-block-label">思考</span>
|
||||
<span className="studio-run-chat-pending">正在等待模型回复…</span>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{thinking && (
|
||||
<div className="studio-run-chat-thinking">
|
||||
<span className="studio-run-chat-block-label">思考</span>
|
||||
<div className="studio-run-chat-block-body">
|
||||
{renderBody(thinking, renderMode, false)}
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{summaryText && (
|
||||
<div className="studio-run-chat-summary">
|
||||
<span className="studio-run-chat-block-label">评价与修改建议</span>
|
||||
<div className="studio-run-chat-block-body">
|
||||
{renderBody(summaryText, renderMode, false)}
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
</>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<form className="studio-run-chat-input-wrapper" onSubmit={handleSubmit}>
|
||||
<div className="studio-run-chat-input-container">
|
||||
<div className="studio-run-chat-options-wrapper" ref={optionsRef}>
|
||||
<button
|
||||
type="button"
|
||||
className={`studio-run-chat-options-toggle${showOptions ? ' active' : ''}`}
|
||||
title="展开选项"
|
||||
onClick={() => setShowOptions((v) => !v)}
|
||||
>
|
||||
{showOptions ? '×' : '≡'}
|
||||
</button>
|
||||
{showOptions && (
|
||||
<div className="studio-run-chat-options">
|
||||
<div className="studio-run-chat-options-title">显示</div>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-chat-render-btn"
|
||||
onClick={cycleRenderMode}
|
||||
title={`当前:${RENDER_MODE_LABELS[renderMode]}`}
|
||||
>
|
||||
{RENDER_MODE_LABELS[renderMode]}
|
||||
</button>
|
||||
<div className="studio-run-chat-options-title">功能</div>
|
||||
<label className="studio-run-chat-option-checkbox">
|
||||
<input
|
||||
type="checkbox"
|
||||
checked={streamOutput}
|
||||
onChange={(e) => onStreamOutputChange?.(e.target.checked)}
|
||||
/>
|
||||
<span className="studio-run-chat-checkmark" />
|
||||
<span className="studio-run-chat-option-label">流式输出</span>
|
||||
</label>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
<div className="studio-run-chat-input-area">
|
||||
<textarea
|
||||
className="studio-run-chat-textarea"
|
||||
rows={1}
|
||||
value={inputValue}
|
||||
onChange={(e) => {
|
||||
onInputChange(e.target.value);
|
||||
handleInputHeight(e);
|
||||
}}
|
||||
onKeyDown={handleKeyDown}
|
||||
placeholder={placeholder}
|
||||
disabled={disabled || sending}
|
||||
aria-label="输入消息"
|
||||
/>
|
||||
</div>
|
||||
<button
|
||||
type="submit"
|
||||
className="studio-run-chat-send"
|
||||
disabled={disabled || sending || !inputValue.trim()}
|
||||
aria-label="发送"
|
||||
title="发送"
|
||||
>
|
||||
{sending ? '…' : '>'}
|
||||
</button>
|
||||
</div>
|
||||
</form>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default StudioRunChat;
|
||||
89
frontend/src/components/Studio/StudioRunNodeGraph.css
Normal file
89
frontend/src/components/Studio/StudioRunNodeGraph.css
Normal file
@@ -0,0 +1,89 @@
|
||||
.studio-run-graph {
|
||||
position: relative;
|
||||
height: 240px;
|
||||
min-height: 200px;
|
||||
border-radius: var(--radius-md);
|
||||
border: 1px solid var(--color-border-light);
|
||||
background: var(--color-bg-secondary);
|
||||
overflow: hidden;
|
||||
cursor: grab;
|
||||
}
|
||||
|
||||
.studio-run-graph--center {
|
||||
flex: 1;
|
||||
min-height: 280px;
|
||||
height: auto;
|
||||
}
|
||||
|
||||
.studio-run-graph:active {
|
||||
cursor: grabbing;
|
||||
}
|
||||
|
||||
.studio-run-graph__svg {
|
||||
display: block;
|
||||
}
|
||||
|
||||
.studio-run-graph__edge {
|
||||
fill: none;
|
||||
stroke: var(--color-border);
|
||||
stroke-width: 1.5;
|
||||
opacity: 0.85;
|
||||
}
|
||||
|
||||
.studio-run-graph__node {
|
||||
cursor: pointer;
|
||||
}
|
||||
|
||||
.studio-run-graph__node-bg {
|
||||
fill: var(--color-bg-tertiary);
|
||||
stroke: var(--color-border-light);
|
||||
stroke-width: 1.5;
|
||||
}
|
||||
|
||||
.studio-run-graph__node.status-active .studio-run-graph__node-bg {
|
||||
stroke: var(--color-accent);
|
||||
fill: rgba(var(--color-accent-rgb, 59, 130, 246), 0.1);
|
||||
}
|
||||
|
||||
.studio-run-graph__node.status-completed .studio-run-graph__node-bg {
|
||||
stroke: #22c55e;
|
||||
}
|
||||
|
||||
.studio-run-graph__node.is-selected .studio-run-graph__node-bg {
|
||||
stroke-width: 2.5;
|
||||
filter: drop-shadow(0 2px 6px rgba(0, 0, 0, 0.12));
|
||||
}
|
||||
|
||||
.studio-run-graph__node.is-current .studio-run-graph__node-bg {
|
||||
stroke: var(--color-accent);
|
||||
}
|
||||
|
||||
.studio-run-graph__node-label {
|
||||
font-size: 12px;
|
||||
font-weight: 600;
|
||||
fill: var(--color-text-primary);
|
||||
pointer-events: none;
|
||||
}
|
||||
|
||||
.studio-run-graph__node-status {
|
||||
font-size: 10px;
|
||||
fill: var(--color-text-muted);
|
||||
pointer-events: none;
|
||||
}
|
||||
|
||||
.studio-run-graph__hint {
|
||||
position: absolute;
|
||||
bottom: 4px;
|
||||
right: 6px;
|
||||
font-size: 0.6rem;
|
||||
color: var(--color-text-muted);
|
||||
opacity: 0.75;
|
||||
pointer-events: none;
|
||||
}
|
||||
|
||||
.studio-run-graph-empty {
|
||||
padding: var(--spacing-md);
|
||||
font-size: 0.75rem;
|
||||
color: var(--color-text-muted);
|
||||
text-align: center;
|
||||
}
|
||||
233
frontend/src/components/Studio/StudioRunNodeGraph.jsx
Normal file
233
frontend/src/components/Studio/StudioRunNodeGraph.jsx
Normal file
@@ -0,0 +1,233 @@
|
||||
import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react';
|
||||
|
||||
|
||||
|
||||
import { buildGraphLayout } from './edit/variableUtils';
|
||||
|
||||
|
||||
|
||||
import './StudioRunNodeGraph.css';
|
||||
|
||||
|
||||
|
||||
const NODE_STATUS_LABEL = {
|
||||
|
||||
pending: '待执行',
|
||||
|
||||
active: '进行中',
|
||||
|
||||
completed: '已完成',
|
||||
|
||||
skipped: '已跳过',
|
||||
|
||||
};
|
||||
|
||||
|
||||
|
||||
function defaultEdgePath(from, to, nodeW, nodeH) {
|
||||
|
||||
const x1 = from.x + nodeW / 2;
|
||||
|
||||
const y1 = from.y + nodeH;
|
||||
|
||||
const x2 = to.x + nodeW / 2;
|
||||
|
||||
const y2 = to.y;
|
||||
|
||||
const dy = Math.max(24, (y2 - y1) * 0.45);
|
||||
|
||||
return `M ${x1} ${y1} C ${x1} ${y1 + dy}, ${x2} ${y2 - dy}, ${x2} ${y2}`;
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
||||
function StudioRunNodeGraph({
|
||||
|
||||
pipeline,
|
||||
|
||||
nodeStates,
|
||||
|
||||
currentNodeId,
|
||||
|
||||
variant = 'sidebar',
|
||||
|
||||
}) {
|
||||
|
||||
const containerRef = useRef(null);
|
||||
|
||||
const [view, setView] = useState({ x: 0, y: 0, scale: 1 });
|
||||
|
||||
const [dragging, setDragging] = useState(null);
|
||||
|
||||
const [selectedId, setSelectedId] = useState(currentNodeId);
|
||||
|
||||
const [nodeOverrides, setNodeOverrides] = useState({});
|
||||
|
||||
|
||||
|
||||
const isCenter = variant === 'center';
|
||||
|
||||
|
||||
|
||||
const layout = useMemo(
|
||||
|
||||
() => buildGraphLayout(pipeline, nodeStates, { vertical: true }),
|
||||
|
||||
[pipeline, nodeStates]
|
||||
|
||||
);
|
||||
|
||||
|
||||
|
||||
const { nodeW, nodeH } = layout;
|
||||
|
||||
|
||||
|
||||
useEffect(() => {
|
||||
|
||||
setSelectedId(currentNodeId);
|
||||
|
||||
}, [currentNodeId]);
|
||||
|
||||
|
||||
|
||||
useEffect(() => {
|
||||
|
||||
const el = containerRef.current;
|
||||
|
||||
if (!el || !layout.width) return;
|
||||
|
||||
const cw = el.clientWidth || 200;
|
||||
|
||||
const ch = el.clientHeight || 160;
|
||||
|
||||
const scale = Math.min(1, (cw - 16) / layout.width, (ch - 16) / layout.height);
|
||||
|
||||
setView({
|
||||
|
||||
x: (cw - layout.width * scale) / 2,
|
||||
|
||||
y: Math.max(8, (ch - layout.height * scale) / 2),
|
||||
|
||||
scale: Math.max(0.45, scale),
|
||||
|
||||
});
|
||||
|
||||
}, [layout.width, layout.height, pipeline, variant]);
|
||||
|
||||
|
||||
|
||||
const nodeMap = useMemo(
|
||||
|
||||
() => Object.fromEntries(layout.nodes.map((n) => [n.id, n])),
|
||||
|
||||
[layout.nodes]
|
||||
|
||||
);
|
||||
|
||||
|
||||
|
||||
const handleWheel = useCallback((e) => {
|
||||
|
||||
e.preventDefault();
|
||||
|
||||
const delta = e.deltaY > 0 ? 0.92 : 1.08;
|
||||
|
||||
setView((v) => ({
|
||||
|
||||
...v,
|
||||
|
||||
scale: Math.min(2.5, Math.max(0.35, v.scale * delta)),
|
||||
|
||||
}));
|
||||
|
||||
}, []);
|
||||
|
||||
|
||||
|
||||
const handleBgMouseDown = useCallback(
|
||||
|
||||
(e) => {
|
||||
|
||||
if (e.button !== 0) return;
|
||||
|
||||
e.preventDefault();
|
||||
|
||||
setDragging({ type: 'pan', startX: e.clientX, startY: e.clientY, origin: { ...view } });
|
||||
|
||||
},
|
||||
|
||||
[view]
|
||||
|
||||
);
|
||||
|
||||
|
||||
|
||||
const handleNodeMouseDown = useCallback(
|
||||
|
||||
(e, nodeId) => {
|
||||
|
||||
if (e.button !== 0) return;
|
||||
|
||||
e.stopPropagation();
|
||||
|
||||
setSelectedId(nodeId);
|
||||
|
||||
const node = nodeMap[nodeId];
|
||||
|
||||
if (!node) return;
|
||||
|
||||
setDragging({
|
||||
|
||||
type: 'node',
|
||||
|
||||
nodeId,
|
||||
|
||||
startX: e.clientX,
|
||||
|
||||
startY: e.clientY,
|
||||
|
||||
origin: { x: node.x, y: node.y },
|
||||
|
||||
});
|
||||
|
||||
},
|
||||
|
||||
[nodeMap]
|
||||
|
||||
);
|
||||
|
||||
|
||||
|
||||
useEffect(() => {
|
||||
|
||||
if (!dragging) return undefined;
|
||||
|
||||
|
||||
|
||||
const onMove = (e) => {
|
||||
|
||||
if (dragging.type === 'pan') {
|
||||
|
||||
setView((v) => ({
|
||||
|
||||
...v,
|
||||
|
||||
x: dragging.origin.x + (e.clientX - dragging.startX),
|
||||
|
||||
y: dragging.origin.y + (e.clientY - dragging.startY),
|
||||
|
||||
}));
|
||||
|
||||
return;
|
||||
|
||||
}
|
||||
|
||||
if (dragging.type === 'node') {
|
||||
|
||||
const dx = (e.clientX - dragging.startX) / view.scale;
|
||||
|
||||
const dy = (e.clientY - dragging.startY) / view.scale;
|
||||
|
||||
setNodeOverrides((prev) => ({
|
||||
1208
frontend/src/components/Studio/StudioRunPage.css
Normal file
1208
frontend/src/components/Studio/StudioRunPage.css
Normal file
File diff suppressed because it is too large
Load Diff
978
frontend/src/components/Studio/StudioRunPage.jsx
Normal file
978
frontend/src/components/Studio/StudioRunPage.jsx
Normal file
@@ -0,0 +1,978 @@
|
||||
import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react';
|
||||
|
||||
import useAppLayoutStore from '../../Store/AppLayoutSlice';
|
||||
import useStudioStore from '../../Store/Studio/StudioSlice';
|
||||
import {
|
||||
buildWorkflowVariableLabelMap,
|
||||
getWorkflowVariableLabel,
|
||||
} from './edit/workflowVariableLabels';
|
||||
import { resolveBoundCharacterName } from './studioRunUtils';
|
||||
import useCharacterStore from '../../Store/SideBarLeft/CharacterSlice';
|
||||
|
||||
import StudioContextBlockPopup from './StudioContextBlockPopup';
|
||||
import StudioInsertionPopup from './StudioInsertionPopup';
|
||||
import StudioRunChat from './StudioRunChat';
|
||||
import StudioRunNodeGraph from './StudioRunNodeGraph';
|
||||
|
||||
import './StudioRunPage.css';
|
||||
|
||||
const RUN_STATUS_LABEL = {
|
||||
pending: '待开始',
|
||||
running: '运行中',
|
||||
paused: '已暂停',
|
||||
completed: '已完成',
|
||||
failed: '失败',
|
||||
cancelled: '已取消',
|
||||
};
|
||||
|
||||
const RUN_LIST_COLLAPSE_THRESHOLD = 4;
|
||||
const RUN_ITEM_HEIGHT = 56;
|
||||
|
||||
function formatRunTime(iso) {
|
||||
if (!iso) return '';
|
||||
try {
|
||||
const d = new Date(iso);
|
||||
return d.toLocaleString('zh-CN', {
|
||||
month: '2-digit',
|
||||
day: '2-digit',
|
||||
hour: '2-digit',
|
||||
minute: '2-digit',
|
||||
});
|
||||
} catch {
|
||||
return iso;
|
||||
}
|
||||
}
|
||||
|
||||
function runDisplayName(run) {
|
||||
if (run.title?.trim()) return run.title.trim();
|
||||
return formatRunTime(run.createdAt) || '未命名运行';
|
||||
}
|
||||
|
||||
function InitBindGuidanceForm({ node, runAdvancing, onSubmit, compact = false }) {
|
||||
const displayParams = node?.displayParams || [];
|
||||
const [formValues, setFormValues] = useState({});
|
||||
const [fieldErrors, setFieldErrors] = useState({});
|
||||
|
||||
useEffect(() => {
|
||||
const initial = {};
|
||||
displayParams.forEach((dp) => {
|
||||
initial[dp.key] = '';
|
||||
});
|
||||
setFormValues(initial);
|
||||
setFieldErrors({});
|
||||
}, [node?.nodeId, displayParams]);
|
||||
|
||||
const handleChange = (key, value) => {
|
||||
setFormValues((prev) => ({ ...prev, [key]: value }));
|
||||
setFieldErrors((prev) => ({ ...prev, [key]: null }));
|
||||
};
|
||||
|
||||
const handleSubmit = (e) => {
|
||||
e.preventDefault();
|
||||
const errors = {};
|
||||
displayParams.forEach((dp) => {
|
||||
if (dp.required && !(formValues[dp.key] || '').trim()) {
|
||||
errors[dp.key] = `请填写${dp.label}`;
|
||||
}
|
||||
});
|
||||
if (Object.keys(errors).length > 0) {
|
||||
setFieldErrors(errors);
|
||||
return;
|
||||
}
|
||||
onSubmit(formValues);
|
||||
};
|
||||
|
||||
if (displayParams.length === 0) {
|
||||
return <div className="studio-run-empty">当前步骤未定义引导字段</div>;
|
||||
}
|
||||
|
||||
return (
|
||||
<form
|
||||
className={`studio-run-guidance-form${compact ? ' studio-run-guidance-form--compact' : ''}`}
|
||||
onSubmit={handleSubmit}
|
||||
>
|
||||
<p className="studio-run-hint">
|
||||
填写以下信息后将创建角色卡与世界书,并绑定到当前 Studio 项目。
|
||||
</p>
|
||||
{displayParams.map((dp) => (
|
||||
<label key={dp.key} className="studio-run-form-field">
|
||||
<span className="studio-run-form-label">
|
||||
{dp.label}
|
||||
{dp.required ? <span className="studio-run-required">*</span> : null}
|
||||
</span>
|
||||
<input
|
||||
type="text"
|
||||
className={`studio-run-form-input${fieldErrors[dp.key] ? ' has-error' : ''}`}
|
||||
value={formValues[dp.key] || ''}
|
||||
onChange={(ev) => handleChange(dp.key, ev.target.value)}
|
||||
placeholder={dp.placeholder || ''}
|
||||
disabled={runAdvancing}
|
||||
/>
|
||||
{fieldErrors[dp.key] ? (
|
||||
<span className="studio-run-field-error">{fieldErrors[dp.key]}</span>
|
||||
) : null}
|
||||
</label>
|
||||
))}
|
||||
<button type="submit" className="studio-run-submit-btn" disabled={runAdvancing}>
|
||||
{runAdvancing ? '创建中…' : '创建并绑定'}
|
||||
</button>
|
||||
</form>
|
||||
);
|
||||
}
|
||||
|
||||
function ToolOptionsCarousel({ options, index, onPrev, onNext, runAdvancing }) {
|
||||
const current = options[index];
|
||||
const [selectedOption, setSelectedOption] = useState(null);
|
||||
const [copyHint, setCopyHint] = useState('');
|
||||
|
||||
useEffect(() => {
|
||||
setSelectedOption(null);
|
||||
setCopyHint('');
|
||||
}, [index, current?.question]);
|
||||
|
||||
if (!options.length || !current) return null;
|
||||
|
||||
const handleSelect = (opt) => {
|
||||
setSelectedOption(opt);
|
||||
setCopyHint('');
|
||||
};
|
||||
|
||||
const handleCopy = async () => {
|
||||
if (!selectedOption) return;
|
||||
try {
|
||||
await navigator.clipboard.writeText(selectedOption);
|
||||
setCopyHint('已复制到剪贴板');
|
||||
setTimeout(() => setCopyHint(''), 2000);
|
||||
} catch {
|
||||
setCopyHint('复制失败,请手动选择文本');
|
||||
}
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="studio-run-tool-carousel">
|
||||
<div className="studio-run-tool-carousel__head">
|
||||
<span className="studio-run-tool-carousel__label">模型提问</span>
|
||||
<span className="studio-run-tool-carousel__pager">
|
||||
{index + 1} / {options.length}
|
||||
</span>
|
||||
</div>
|
||||
<p className="studio-run-tool-carousel__question">{current.question}</p>
|
||||
<div className="studio-run-tool-carousel__nav">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-icon-btn"
|
||||
onClick={onPrev}
|
||||
disabled={options.length <= 1}
|
||||
aria-label="上一题"
|
||||
>
|
||||
‹
|
||||
</button>
|
||||
<div className="studio-run-tool-carousel__options">
|
||||
{(current.options || []).map((opt) => (
|
||||
<button
|
||||
key={opt}
|
||||
type="button"
|
||||
className={`studio-run-tool-option${selectedOption === opt ? ' is-selected' : ''}`}
|
||||
onClick={() => handleSelect(opt)}
|
||||
disabled={runAdvancing}
|
||||
>
|
||||
{opt}
|
||||
</button>
|
||||
))}
|
||||
</div>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-icon-btn"
|
||||
onClick={onNext}
|
||||
disabled={options.length <= 1}
|
||||
aria-label="下一题"
|
||||
>
|
||||
›
|
||||
</button>
|
||||
</div>
|
||||
{selectedOption ? (
|
||||
<div className="studio-run-tool-copy">
|
||||
<div className="studio-run-tool-copy__head">
|
||||
<span className="studio-run-tool-copy__label">复制以下内容到输入框</span>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-tool-copy__btn"
|
||||
onClick={handleCopy}
|
||||
>
|
||||
复制
|
||||
</button>
|
||||
</div>
|
||||
<textarea
|
||||
className="studio-run-tool-copy__textarea"
|
||||
readOnly
|
||||
value={selectedOption}
|
||||
rows={3}
|
||||
aria-label="选项内容"
|
||||
/>
|
||||
{copyHint ? (
|
||||
<span className="studio-run-tool-copy__hint">{copyHint}</span>
|
||||
) : null}
|
||||
</div>
|
||||
) : null}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
function PromptBlocksDebug({ blocks, onBlockClick }) {
|
||||
const [open, setOpen] = useState(false);
|
||||
if (!blocks?.length) return null;
|
||||
|
||||
return (
|
||||
<div className="studio-run-prompt-debug">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-prompt-debug__toggle"
|
||||
onClick={() => setOpen((v) => !v)}
|
||||
>
|
||||
{open ? '▼' : '▶'} 上下文块(调试 · {blocks.length})
|
||||
</button>
|
||||
{open && (
|
||||
<div className="studio-run-prompt-debug__list">
|
||||
{blocks.map((block) => (
|
||||
<button
|
||||
key={block.id}
|
||||
type="button"
|
||||
className="studio-run-prompt-debug__item"
|
||||
onClick={() => onBlockClick?.(block)}
|
||||
>
|
||||
<span className="studio-run-prompt-debug__label">{block.label}</span>
|
||||
<span className="studio-run-prompt-debug__source">{block.source}</span>
|
||||
</button>
|
||||
))}
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
function RunListItem({
|
||||
run,
|
||||
selected,
|
||||
onSelect,
|
||||
onDelete,
|
||||
renamingId,
|
||||
renameValue,
|
||||
onRenameChange,
|
||||
onRenameSubmit,
|
||||
onRenameCancel,
|
||||
onStartRename,
|
||||
}) {
|
||||
const isRenaming = renamingId === run.id;
|
||||
|
||||
return (
|
||||
<li className="studio-run-list-row">
|
||||
{isRenaming ? (
|
||||
<form
|
||||
className="studio-run-rename-form"
|
||||
onSubmit={(e) => {
|
||||
e.preventDefault();
|
||||
onRenameSubmit(run.id);
|
||||
}}
|
||||
>
|
||||
<input
|
||||
className="studio-run-rename-input"
|
||||
value={renameValue}
|
||||
onChange={(e) => onRenameChange(e.target.value)}
|
||||
autoFocus
|
||||
maxLength={120}
|
||||
/>
|
||||
<div className="studio-run-rename-actions">
|
||||
<button type="submit" className="studio-run-list-action" title="保存">
|
||||
✓
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-list-action"
|
||||
onClick={onRenameCancel}
|
||||
title="取消"
|
||||
>
|
||||
✕
|
||||
</button>
|
||||
</div>
|
||||
</form>
|
||||
) : (
|
||||
<>
|
||||
<button
|
||||
type="button"
|
||||
className={`studio-run-list-item${selected ? ' selected' : ''}`}
|
||||
onClick={() => onSelect(run.id)}
|
||||
>
|
||||
<span className="studio-run-list-name">{runDisplayName(run)}</span>
|
||||
<span className={`studio-run-status-badge status-${run.status}`}>
|
||||
{RUN_STATUS_LABEL[run.status] || run.status}
|
||||
</span>
|
||||
</button>
|
||||
<div className="studio-run-list-item-actions">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-list-action"
|
||||
title="重命名"
|
||||
onClick={(e) => {
|
||||
e.stopPropagation();
|
||||
onStartRename(run);
|
||||
}}
|
||||
>
|
||||
✎
|
||||
</button>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-list-action studio-run-list-action--danger"
|
||||
title="删除"
|
||||
onClick={(e) => {
|
||||
e.stopPropagation();
|
||||
onDelete(run.id);
|
||||
}}
|
||||
>
|
||||
🗑
|
||||
</button>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
</li>
|
||||
);
|
||||
}
|
||||
|
||||
function ProductPreview({ activeNodeState, isWorldbookActive, compact = false }) {
|
||||
let productPreview = '暂无产物';
|
||||
if (activeNodeState?.lastDraft) {
|
||||
const draft = activeNodeState.lastDraft;
|
||||
for (const key of ['entryContent', 'content', 'text', 'body']) {
|
||||
if (typeof draft[key] === 'string' && draft[key].trim()) {
|
||||
productPreview = draft[key];
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (productPreview === '暂无产物') {
|
||||
productPreview = JSON.stringify(draft, null, 2);
|
||||
}
|
||||
} else if (isWorldbookActive) {
|
||||
productPreview = '(等待模型生成条目草稿…)';
|
||||
}
|
||||
|
||||
return (
|
||||
<div className={`studio-run-product-main${compact ? ' studio-run-product-main--compact' : ''}`}>
|
||||
<div className="studio-run-product-main__head">
|
||||
<h2 className="studio-run-section-title">目前产物</h2>
|
||||
{activeNodeState?.displayName && (
|
||||
<span className="studio-run-product-main__step">
|
||||
{activeNodeState.displayName}
|
||||
</span>
|
||||
)}
|
||||
</div>
|
||||
<pre className="studio-run-product-main__body">{productPreview}</pre>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
function WorkflowVariablesPanel({ variables, labelMap }) {
|
||||
const entries = Object.entries(variables || {});
|
||||
if (entries.length === 0) return null;
|
||||
|
||||
return (
|
||||
<section className="studio-run-preview-section">
|
||||
<h2 className="studio-run-section-title">工作流变量</h2>
|
||||
<dl className="studio-run-variables">
|
||||
{entries.map(([key, value]) => (
|
||||
<div key={key} className="studio-run-variables__row">
|
||||
<dt title={key}>{getWorkflowVariableLabel(key, labelMap)}</dt>
|
||||
<dd>{typeof value === 'string' ? value : JSON.stringify(value)}</dd>
|
||||
</div>
|
||||
))}
|
||||
</dl>
|
||||
</section>
|
||||
);
|
||||
}
|
||||
|
||||
function CurrentNodePanel({ activeNodeState, graphCurrentNodeId, graphPipeline }) {
|
||||
const nodeName =
|
||||
activeNodeState?.displayName ||
|
||||
graphPipeline?.nodes?.find((n) => n.id === graphCurrentNodeId)?.displayName ||
|
||||
'—';
|
||||
const nodeStatus = activeNodeState?.status;
|
||||
|
||||
return (
|
||||
<section className="studio-run-preview-section">
|
||||
<h2 className="studio-run-section-title">当前节点</h2>
|
||||
<dl className="studio-run-overview studio-run-overview--inline">
|
||||
<div>
|
||||
<dt>步骤</dt>
|
||||
<dd>{nodeName}</dd>
|
||||
</div>
|
||||
{nodeStatus && (
|
||||
<div>
|
||||
<dt>状态</dt>
|
||||
<dd>{RUN_STATUS_LABEL[nodeStatus] || nodeStatus}</dd>
|
||||
</div>
|
||||
)}
|
||||
{activeNodeState?.skillId && (
|
||||
<div>
|
||||
<dt>技能</dt>
|
||||
<dd>{activeNodeState.skillId}</dd>
|
||||
</div>
|
||||
)}
|
||||
</dl>
|
||||
</section>
|
||||
);
|
||||
}
|
||||
|
||||
function InsertionSidebar({ insertionPreview, onOpenEntry }) {
|
||||
if (!insertionPreview) {
|
||||
return (
|
||||
<section className="studio-run-right-section">
|
||||
<h2 className="studio-run-section-title">插入内容</h2>
|
||||
<p className="studio-run-right-placeholder">当前步骤无世界书插入配置</p>
|
||||
</section>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<section className="studio-run-right-section">
|
||||
<h2 className="studio-run-section-title">插入内容</h2>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-insertion-entry"
|
||||
onClick={() => onOpenEntry(insertionPreview)}
|
||||
title={insertionPreview.comment || insertionPreview.key}
|
||||
>
|
||||
<span className="studio-run-insertion-entry__key">{insertionPreview.key}</span>
|
||||
{insertionPreview.comment ? (
|
||||
<span className="studio-run-insertion-entry__meta">{insertionPreview.comment}</span>
|
||||
) : null}
|
||||
</button>
|
||||
</section>
|
||||
);
|
||||
}
|
||||
|
||||
function StudioRunPage() {
|
||||
const {
|
||||
pipeline: projectPipeline,
|
||||
workflowVariables: workflowVariablesCatalog,
|
||||
meta,
|
||||
runProjectId,
|
||||
runs,
|
||||
currentRunId,
|
||||
currentRun,
|
||||
runLoading,
|
||||
runAdvancing,
|
||||
runMessaging,
|
||||
runStreamingThinking,
|
||||
runError,
|
||||
runSessionEntered,
|
||||
setRunSessionEntered,
|
||||
initStudioRun,
|
||||
fetchProject,
|
||||
selectRun,
|
||||
advanceRun,
|
||||
sendRunMessage,
|
||||
deleteRun,
|
||||
renameRun,
|
||||
} = useStudioStore();
|
||||
|
||||
const sidebarMode = useAppLayoutStore((s) => s.sidebarMode);
|
||||
const isSidebarHovered = useAppLayoutStore((s) => s.isSidebarHovered);
|
||||
const setSidebarHovered = useAppLayoutStore((s) => s.setSidebarHovered);
|
||||
|
||||
const hoverTimeoutRef = useRef(null);
|
||||
const leaveTimeoutRef = useRef(null);
|
||||
|
||||
const handleLeftMouseEnter = useCallback(() => {
|
||||
if (sidebarMode !== 'smart') return;
|
||||
if (leaveTimeoutRef.current) clearTimeout(leaveTimeoutRef.current);
|
||||
hoverTimeoutRef.current = setTimeout(() => {
|
||||
setSidebarHovered(true);
|
||||
}, 400);
|
||||
}, [sidebarMode, setSidebarHovered]);
|
||||
|
||||
const handleLeftMouseLeave = useCallback(() => {
|
||||
if (sidebarMode !== 'smart') return;
|
||||
if (hoverTimeoutRef.current) clearTimeout(hoverTimeoutRef.current);
|
||||
leaveTimeoutRef.current = setTimeout(() => {
|
||||
setSidebarHovered(false);
|
||||
}, 250);
|
||||
}, [sidebarMode, setSidebarHovered]);
|
||||
|
||||
useEffect(() => {
|
||||
return () => {
|
||||
if (hoverTimeoutRef.current) clearTimeout(hoverTimeoutRef.current);
|
||||
if (leaveTimeoutRef.current) clearTimeout(leaveTimeoutRef.current);
|
||||
};
|
||||
}, []);
|
||||
|
||||
const variableLabelMap = useMemo(
|
||||
() => buildWorkflowVariableLabelMap(workflowVariablesCatalog),
|
||||
[workflowVariablesCatalog]
|
||||
);
|
||||
const [insertionPopupOpen, setInsertionPopupOpen] = useState(false);
|
||||
const [insertionPopupExpanded, setInsertionPopupExpanded] = useState(true);
|
||||
const [contextBlockPopup, setContextBlockPopup] = useState(null);
|
||||
const [chatInput, setChatInput] = useState('');
|
||||
const [streamOutput, setStreamOutput] = useState(false);
|
||||
const [toolOptionIndex, setToolOptionIndex] = useState(0);
|
||||
const [pendingUserText, setPendingUserText] = useState(null);
|
||||
const [runListExpanded, setRunListExpanded] = useState(false);
|
||||
const [renamingId, setRenamingId] = useState(null);
|
||||
const [renameValue, setRenameValue] = useState('');
|
||||
|
||||
useEffect(() => {
|
||||
initStudioRun();
|
||||
}, [initStudioRun]);
|
||||
|
||||
useEffect(() => {
|
||||
if (runProjectId) {
|
||||
fetchProject(runProjectId);
|
||||
}
|
||||
}, [runProjectId, fetchProject]);
|
||||
|
||||
useEffect(() => {
|
||||
setRunSessionEntered(false);
|
||||
setInsertionPopupOpen(false);
|
||||
setContextBlockPopup(null);
|
||||
setPendingUserText(null);
|
||||
}, [currentRunId, runProjectId, setRunSessionEntered]);
|
||||
|
||||
const activeNodeState = useMemo(() => {
|
||||
if (!currentRun?.currentNodeId) return null;
|
||||
return currentRun.nodeStates?.find((n) => n.nodeId === currentRun.currentNodeId);
|
||||
}, [currentRun]);
|
||||
|
||||
const activePipelineNode = useMemo(() => {
|
||||
if (!currentRun?.currentNodeId) return null;
|
||||
return currentRun.pipelineSnapshot?.nodes?.find(
|
||||
(n) => n.id === currentRun.currentNodeId
|
||||
);
|
||||
}, [currentRun]);
|
||||
|
||||
const isInitBindActive =
|
||||
activeNodeState?.status === 'active' &&
|
||||
activeNodeState?.skillId === 'studio.init_bind';
|
||||
|
||||
const isWorldbookActive =
|
||||
activeNodeState?.status === 'active' &&
|
||||
activeNodeState?.skillId === 'studio.worldbook_entry';
|
||||
|
||||
const toolQuestions = useMemo(
|
||||
() => activeNodeState?.lastToolResponse?.questions || [],
|
||||
[activeNodeState]
|
||||
);
|
||||
|
||||
useEffect(() => {
|
||||
setToolOptionIndex(0);
|
||||
}, [currentRunId, activeNodeState?.nodeId, toolQuestions.length]);
|
||||
|
||||
const graphPipeline = currentRun?.pipelineSnapshot || projectPipeline;
|
||||
const graphNodeStates = currentRun?.nodeStates || [];
|
||||
const graphCurrentNodeId = currentRun?.currentNodeId || null;
|
||||
|
||||
const insertionPreview = useMemo(() => {
|
||||
const ins = activePipelineNode?.config?.insertion;
|
||||
if (!ins) return null;
|
||||
return {
|
||||
key: ins.key || '(未设置关键词)',
|
||||
comment: ins.comment || '',
|
||||
position: ins.position,
|
||||
activationType: ins.activationType,
|
||||
keysecondary: ins.keysecondary || '',
|
||||
};
|
||||
}, [activePipelineNode]);
|
||||
|
||||
const promptBlocks = currentRun?.lastPromptBlocks || [];
|
||||
const runWorkflowVariables = currentRun?.workflowVariables || {};
|
||||
const stepMessages = activeNodeState?.stepMessages || [];
|
||||
const lastToolResponse = activeNodeState?.lastToolResponse;
|
||||
|
||||
const characters = useCharacterStore((s) => s.characters);
|
||||
const boundCharacterName = useMemo(
|
||||
() =>
|
||||
resolveBoundCharacterName({
|
||||
boundCharacterVar: runWorkflowVariables['workflow.boundCharacter'],
|
||||
characterId: meta?.characterId,
|
||||
characters,
|
||||
}),
|
||||
[runWorkflowVariables, meta?.characterId, characters]
|
||||
);
|
||||
|
||||
const handleOpenInsertion = () => {
|
||||
setInsertionPopupExpanded(true);
|
||||
setInsertionPopupOpen(true);
|
||||
};
|
||||
|
||||
const handleInitBindSubmit = async (displayParams) => {
|
||||
await advanceRun(displayParams);
|
||||
};
|
||||
|
||||
const handleSelectRun = async (runId) => {
|
||||
setRunSessionEntered(false);
|
||||
await selectRun(runId);
|
||||
};
|
||||
|
||||
const handleEnterSession = () => {
|
||||
if (currentRun) {
|
||||
setRunSessionEntered(true);
|
||||
}
|
||||
};
|
||||
|
||||
const handleChatSend = async (text) => {
|
||||
if (!text || isInitBindActive || !isWorldbookActive || runMessaging) return;
|
||||
|
||||
setPendingUserText(text);
|
||||
setChatInput('');
|
||||
|
||||
const run = await sendRunMessage(text, { stream: streamOutput });
|
||||
setPendingUserText(null);
|
||||
if (!run) {
|
||||
/* error surfaced via runError */
|
||||
}
|
||||
};
|
||||
|
||||
const handleNextStep = () => {
|
||||
window.alert('(R5 占位:确认当前产物后将进入下一步)');
|
||||
};
|
||||
|
||||
const handleDeleteRun = async (runId) => {
|
||||
const name = runDisplayName(runs.find((r) => r.id === runId) || {});
|
||||
if (!window.confirm(`确定删除运行「${name}」?此操作不可撤销。`)) return;
|
||||
if (currentRunId === runId) setRunSessionEntered(false);
|
||||
await deleteRun(runId);
|
||||
};
|
||||
|
||||
const handleStartRename = (run) => {
|
||||
setRenamingId(run.id);
|
||||
setRenameValue(run.title?.trim() || runDisplayName(run));
|
||||
};
|
||||
|
||||
const handleRenameSubmit = async (runId) => {
|
||||
const title = renameValue.trim();
|
||||
if (!title) return;
|
||||
const ok = await renameRun(runId, title);
|
||||
if (ok) {
|
||||
setRenamingId(null);
|
||||
setRenameValue('');
|
||||
}
|
||||
};
|
||||
|
||||
const runListNeedsCollapse = runs.length > RUN_LIST_COLLAPSE_THRESHOLD;
|
||||
const inSession = runSessionEntered && !!currentRun;
|
||||
const showToolCarousel = inSession && isWorldbookActive && toolQuestions.length > 0;
|
||||
const showNextStep = inSession && isWorldbookActive;
|
||||
|
||||
const renderRunList = () => (
|
||||
<>
|
||||
{runLoading && runs.length === 0 ? (
|
||||
<div className="studio-run-empty">加载中…</div>
|
||||
) : runs.length === 0 ? (
|
||||
<div className="studio-run-empty">暂无会话记录,点击「新开会话」开始</div>
|
||||
) : (
|
||||
<>
|
||||
<ul
|
||||
className={`studio-run-list${runListNeedsCollapse && !runListExpanded ? ' is-collapsed' : ''}`}
|
||||
style={
|
||||
runListNeedsCollapse && !runListExpanded
|
||||
? { maxHeight: `${RUN_ITEM_HEIGHT * 3.5}px` }
|
||||
: undefined
|
||||
}
|
||||
>
|
||||
{runs.map((run) => (
|
||||
<RunListItem
|
||||
key={run.id}
|
||||
run={run}
|
||||
selected={currentRunId === run.id}
|
||||
onSelect={handleSelectRun}
|
||||
onDelete={handleDeleteRun}
|
||||
renamingId={renamingId}
|
||||
renameValue={renameValue}
|
||||
onRenameChange={setRenameValue}
|
||||
onRenameSubmit={handleRenameSubmit}
|
||||
onRenameCancel={() => {
|
||||
setRenamingId(null);
|
||||
setRenameValue('');
|
||||
}}
|
||||
onStartRename={handleStartRename}
|
||||
/>
|
||||
))}
|
||||
</ul>
|
||||
{runListNeedsCollapse && (
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-list-fold"
|
||||
onClick={() => setRunListExpanded((v) => !v)}
|
||||
>
|
||||
{runListExpanded ? '收起列表' : `展开全部(共 ${runs.length} 条)`}
|
||||
</button>
|
||||
)}
|
||||
</>
|
||||
)}
|
||||
</>
|
||||
);
|
||||
|
||||
const renderNodeProgress = () => {
|
||||
if (!graphPipeline) {
|
||||
return <div className="studio-run-empty">暂无流水线数据</div>;
|
||||
}
|
||||
return (
|
||||
<StudioRunNodeGraph
|
||||
pipeline={graphPipeline}
|
||||
nodeStates={graphNodeStates}
|
||||
currentNodeId={graphCurrentNodeId}
|
||||
variant="sidebar"
|
||||
/>
|
||||
);
|
||||
};
|
||||
|
||||
const renderLeftBrowseView = () => (
|
||||
<div className="studio-run-left-stack">
|
||||
<section className="studio-run-left-stack__sessions">
|
||||
<h2 className="studio-run-section-title">会话列表</h2>
|
||||
{renderRunList()}
|
||||
</section>
|
||||
<section className="studio-run-left-stack__progress">
|
||||
<h2 className="studio-run-section-title">节点进度</h2>
|
||||
<p className="studio-run-hint studio-run-hint--inline">
|
||||
基于各步骤上文引用(inputs.ref)生成的依赖顺序。
|
||||
</p>
|
||||
{renderNodeProgress()}
|
||||
</section>
|
||||
</div>
|
||||
);
|
||||
|
||||
const renderLeftSessionView = () => (
|
||||
<div className="studio-run-left-session">
|
||||
<ProductPreview
|
||||
activeNodeState={activeNodeState}
|
||||
isWorldbookActive={isWorldbookActive}
|
||||
/>
|
||||
<PromptBlocksDebug
|
||||
blocks={promptBlocks}
|
||||
onBlockClick={setContextBlockPopup}
|
||||
/>
|
||||
</div>
|
||||
);
|
||||
|
||||
const renderRunPreviewCenter = () => {
|
||||
if (!runProjectId) {
|
||||
return <div className="studio-run-chat-empty">请先选择项目</div>;
|
||||
}
|
||||
|
||||
if (!currentRun) {
|
||||
return (
|
||||
<div className="studio-run-preview-empty">
|
||||
<p className="studio-run-hint">在左侧选择或新建会话,此处将预览运行内容与节点状态。</p>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="studio-run-preview">
|
||||
<header className="studio-run-preview__header">
|
||||
<div className="studio-run-preview__title-row">
|
||||
<h2 className="studio-run-preview__title">{runDisplayName(currentRun)}</h2>
|
||||
<span className={`studio-run-status-badge status-${currentRun.status}`}>
|
||||
{RUN_STATUS_LABEL[currentRun.status] || currentRun.status}
|
||||
</span>
|
||||
</div>
|
||||
<div className="studio-run-preview__meta">
|
||||
<span>创建 {formatRunTime(currentRun.createdAt)}</span>
|
||||
{boundCharacterName ? (
|
||||
<span className="studio-run-preview__character">
|
||||
角色卡:{boundCharacterName}
|
||||
</span>
|
||||
) : null}
|
||||
</div>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-enter-btn studio-run-enter-btn--header"
|
||||
onClick={handleEnterSession}
|
||||
disabled={runLoading || runAdvancing}
|
||||
>
|
||||
进入会话
|
||||
</button>
|
||||
</header>
|
||||
|
||||
<div className="studio-run-preview__body">
|
||||
<CurrentNodePanel
|
||||
activeNodeState={activeNodeState}
|
||||
graphCurrentNodeId={graphCurrentNodeId}
|
||||
graphPipeline={graphPipeline}
|
||||
/>
|
||||
<ProductPreview
|
||||
activeNodeState={activeNodeState}
|
||||
isWorldbookActive={isWorldbookActive}
|
||||
compact
|
||||
/>
|
||||
<WorkflowVariablesPanel
|
||||
variables={runWorkflowVariables}
|
||||
labelMap={variableLabelMap}
|
||||
/>
|
||||
{promptBlocks.length > 0 && (
|
||||
<section className="studio-run-preview-section">
|
||||
<PromptBlocksDebug
|
||||
blocks={promptBlocks}
|
||||
onBlockClick={setContextBlockPopup}
|
||||
/>
|
||||
</section>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
};
|
||||
|
||||
const renderSessionCenter = () => {
|
||||
if (isInitBindActive && activePipelineNode) {
|
||||
return (
|
||||
<div className="studio-run-chat-card">
|
||||
<InitBindGuidanceForm
|
||||
node={{ ...activePipelineNode, nodeId: activePipelineNode.id }}
|
||||
runAdvancing={runAdvancing}
|
||||
onSubmit={handleInitBindSubmit}
|
||||
compact
|
||||
/>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<div className="studio-run-chat-panel">
|
||||
<StudioRunChat
|
||||
stepMessages={stepMessages}
|
||||
thinking={runStreamingThinking ?? lastToolResponse?.thinking}
|
||||
evaluation={runStreamingThinking ? null : lastToolResponse?.evaluation}
|
||||
pendingUserText={pendingUserText}
|
||||
inputValue={chatInput}
|
||||
onInputChange={setChatInput}
|
||||
onSend={handleChatSend}
|
||||
disabled={!isWorldbookActive || runAdvancing || runMessaging || isInitBindActive}
|
||||
sending={runMessaging}
|
||||
streamOutput={streamOutput}
|
||||
onStreamOutputChange={setStreamOutput}
|
||||
placeholder={
|
||||
isWorldbookActive
|
||||
? '输入修改意见或与模型对话…'
|
||||
: currentRun?.status === 'completed'
|
||||
? '运行已完成'
|
||||
: '等待当前步骤…'
|
||||
}
|
||||
/>
|
||||
</div>
|
||||
);
|
||||
};
|
||||
|
||||
const renderRightSidebar = () => (
|
||||
<>
|
||||
<InsertionSidebar
|
||||
insertionPreview={insertionPreview}
|
||||
onOpenEntry={handleOpenInsertion}
|
||||
/>
|
||||
|
||||
{showToolCarousel && (
|
||||
<ToolOptionsCarousel
|
||||
options={toolQuestions}
|
||||
index={toolOptionIndex}
|
||||
onPrev={() =>
|
||||
setToolOptionIndex((i) => (i - 1 + toolQuestions.length) % toolQuestions.length)
|
||||
}
|
||||
onNext={() =>
|
||||
setToolOptionIndex((i) => (i + 1) % toolQuestions.length)
|
||||
}
|
||||
runAdvancing={runAdvancing || runMessaging}
|
||||
/>
|
||||
)}
|
||||
|
||||
{showNextStep && (
|
||||
<div className="studio-run-right-section studio-run-right-actions">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-next-btn"
|
||||
onClick={handleNextStep}
|
||||
disabled={runAdvancing || runMessaging}
|
||||
>
|
||||
下一步
|
||||
</button>
|
||||
<p className="studio-run-stub-note">确认左侧目前产物无误后进入下一节点</p>
|
||||
</div>
|
||||
)}
|
||||
</>
|
||||
);
|
||||
|
||||
return (
|
||||
<div className={`studio-run-page${inSession ? ' studio-run-page--session' : ''}`}>
|
||||
{runError && (
|
||||
<div className="studio-run-banner error" role="alert">{runError}</div>
|
||||
)}
|
||||
|
||||
<div
|
||||
className={`studio-run-body studio-run-sidebar-${sidebarMode}${
|
||||
sidebarMode === 'smart' && isSidebarHovered ? ' studio-run-sidebar-expanded' : ''
|
||||
}${inSession ? ' in-session' : ''}`}
|
||||
>
|
||||
<aside
|
||||
className={`studio-run-left sidebar-mode-${sidebarMode}${
|
||||
sidebarMode === 'smart' && isSidebarHovered ? ' sidebar-expanded' : ''
|
||||
}`}
|
||||
aria-label="左栏"
|
||||
onMouseEnter={handleLeftMouseEnter}
|
||||
onMouseLeave={handleLeftMouseLeave}
|
||||
>
|
||||
<div className="studio-run-left-content">
|
||||
{inSession ? renderLeftSessionView() : renderLeftBrowseView()}
|
||||
</div>
|
||||
</aside>
|
||||
|
||||
<main className="studio-run-center" aria-label={inSession ? '对话' : '运行预览'}>
|
||||
{inSession ? renderSessionCenter() : renderRunPreviewCenter()}
|
||||
</main>
|
||||
|
||||
<aside className="studio-run-right" aria-label="控制与插入">
|
||||
{currentRun ? renderRightSidebar() : (
|
||||
<p className="studio-run-right-placeholder">选择会话后显示插入内容与控制项</p>
|
||||
)}
|
||||
</aside>
|
||||
</div>
|
||||
|
||||
<StudioContextBlockPopup
|
||||
open={!!contextBlockPopup}
|
||||
block={contextBlockPopup}
|
||||
onClose={() => setContextBlockPopup(null)}
|
||||
/>
|
||||
|
||||
<StudioInsertionPopup
|
||||
open={insertionPopupOpen && !!insertionPreview}
|
||||
title="插入内容"
|
||||
defaultExpanded={insertionPopupExpanded}
|
||||
onClose={() => setInsertionPopupOpen(false)}
|
||||
>
|
||||
{({ expanded }) =>
|
||||
expanded ? (
|
||||
<pre className="studio-insertion-popup__full">
|
||||
{JSON.stringify(insertionPreview, null, 2)}
|
||||
</pre>
|
||||
) : (
|
||||
<div className="studio-insertion-popup__preview">
|
||||
<span className="studio-insertion-popup__field-label">关键词</span>
|
||||
<span className="studio-insertion-popup__field-value">
|
||||
{insertionPreview?.key}
|
||||
</span>
|
||||
{insertionPreview?.comment ? (
|
||||
<>
|
||||
<span className="studio-insertion-popup__field-label">备注</span>
|
||||
<span className="studio-insertion-popup__field-value">
|
||||
{insertionPreview.comment}
|
||||
</span>
|
||||
</>
|
||||
) : null}
|
||||
<span className="studio-insertion-popup__field-label">位置 / 激活</span>
|
||||
<span className="studio-insertion-popup__field-value">
|
||||
{insertionPreview?.position} · {insertionPreview?.activationType}
|
||||
</span>
|
||||
</div>
|
||||
)
|
||||
}
|
||||
</StudioInsertionPopup>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default StudioRunPage;
|
||||
15
frontend/src/components/Studio/edit/FieldLabel.jsx
Normal file
15
frontend/src/components/Studio/edit/FieldLabel.jsx
Normal file
@@ -0,0 +1,15 @@
|
||||
import React from 'react';
|
||||
|
||||
/** Label with optional dotted underline + native title tooltip for field hints. */
|
||||
export default function FieldLabel({ children, tip }) {
|
||||
if (tip) {
|
||||
return (
|
||||
<span className="studio-field-label" title={tip}>
|
||||
<span className="studio-field-label__text studio-field-label__text--tip">
|
||||
{children}
|
||||
</span>
|
||||
</span>
|
||||
);
|
||||
}
|
||||
return <span className="studio-field-label">{children}</span>;
|
||||
}
|
||||
83
frontend/src/components/Studio/edit/ScoringDimensions.jsx
Normal file
83
frontend/src/components/Studio/edit/ScoringDimensions.jsx
Normal file
@@ -0,0 +1,83 @@
|
||||
import React from 'react';
|
||||
import FieldLabel from './FieldLabel';
|
||||
|
||||
export default function ScoringDimensions({
|
||||
scoring,
|
||||
dimensions,
|
||||
onUpdateScoring,
|
||||
onUpdateDimension,
|
||||
onRemoveDimension,
|
||||
onAddDimension,
|
||||
}) {
|
||||
return (
|
||||
<div className="studio-scoring-block">
|
||||
<h3 className="studio-scoring-block__title">
|
||||
<FieldLabel tip="定义评分维度及修改时的参考标准,用于循环迭代直至满意">
|
||||
评判维度与如何完善
|
||||
</FieldLabel>
|
||||
</h3>
|
||||
|
||||
<label className="studio-edit-switch-row studio-edit-switch-row--compact">
|
||||
<input
|
||||
type="checkbox"
|
||||
checked={scoring.enabled ?? true}
|
||||
onChange={(e) => onUpdateScoring('enabled', e.target.checked)}
|
||||
/>
|
||||
启用评分
|
||||
</label>
|
||||
|
||||
<div className="studio-dim-list">
|
||||
{dimensions.map((dim, index) => (
|
||||
<div key={dim.id || index} className="studio-dim-card">
|
||||
<div className="studio-dim-card__fields">
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">维度名称</span>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={dim.name}
|
||||
onChange={(e) =>
|
||||
onUpdateDimension(index, 'name', e.target.value)
|
||||
}
|
||||
placeholder="例如:真实性"
|
||||
/>
|
||||
</label>
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">
|
||||
<FieldLabel tip="说明如何改进,并作为评判参考">
|
||||
如何完善(含评判参考)
|
||||
</FieldLabel>
|
||||
</span>
|
||||
<textarea
|
||||
className="studio-edit-textarea studio-edit-textarea--compact"
|
||||
rows={2}
|
||||
value={dim.criteria}
|
||||
onChange={(e) =>
|
||||
onUpdateDimension(index, 'criteria', e.target.value)
|
||||
}
|
||||
placeholder="说明如何改进,并作为评判参考…"
|
||||
/>
|
||||
</label>
|
||||
</div>
|
||||
<button
|
||||
type="button"
|
||||
className="studio-dim-card__remove"
|
||||
title="删除此维度"
|
||||
onClick={() => onRemoveDimension(index)}
|
||||
>
|
||||
删除
|
||||
</button>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
|
||||
<button
|
||||
type="button"
|
||||
className="studio-edit-btn-ghost"
|
||||
onClick={onAddDimension}
|
||||
>
|
||||
+ 新增维度
|
||||
</button>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
146
frontend/src/components/Studio/edit/VariableChips.jsx
Normal file
146
frontend/src/components/Studio/edit/VariableChips.jsx
Normal file
@@ -0,0 +1,146 @@
|
||||
import React, { useState } from 'react';
|
||||
import FieldLabel from './FieldLabel';
|
||||
import {
|
||||
AUTO_INJECTED_CONTEXT_ITEMS,
|
||||
buildStepOutputRef,
|
||||
detectReferenceCycles,
|
||||
dynamicChipLabel,
|
||||
dynamicTooltip,
|
||||
formatCycleWarning,
|
||||
getSelectableStepOutputRefs,
|
||||
stepOutputLabel,
|
||||
} from './variableUtils';
|
||||
|
||||
function AutoInjectedItem({ item }) {
|
||||
const [showPopover, setShowPopover] = useState(false);
|
||||
|
||||
return (
|
||||
<span
|
||||
className="studio-var-auto-chip"
|
||||
onMouseEnter={() => setShowPopover(true)}
|
||||
onMouseLeave={() => setShowPopover(false)}
|
||||
onFocus={() => setShowPopover(true)}
|
||||
onBlur={() => setShowPopover(false)}
|
||||
tabIndex={0}
|
||||
>
|
||||
<span className="studio-var-auto-chip__label">{item.label}</span>
|
||||
{showPopover && (
|
||||
<span className="studio-var-auto-chip__popover" role="tooltip">
|
||||
{item.description}
|
||||
</span>
|
||||
)}
|
||||
</span>
|
||||
);
|
||||
}
|
||||
|
||||
function VariableSwitch({ label, selected, onToggle, tooltip }) {
|
||||
return (
|
||||
<label
|
||||
className={`studio-var-switch ${selected ? 'studio-var-switch--on' : ''}`}
|
||||
title={tooltip}
|
||||
>
|
||||
<span className="studio-var-switch__label">{label}</span>
|
||||
<span className="studio-var-switch__track" aria-hidden="true">
|
||||
<input
|
||||
type="checkbox"
|
||||
className="studio-var-switch__input"
|
||||
checked={selected}
|
||||
onChange={onToggle}
|
||||
/>
|
||||
<span className="studio-var-switch__thumb" />
|
||||
</span>
|
||||
</label>
|
||||
);
|
||||
}
|
||||
|
||||
export default function VariableChips({
|
||||
pipeline,
|
||||
selectedNode,
|
||||
onToggleRef,
|
||||
variant = 'default',
|
||||
}) {
|
||||
const isCompact = variant === 'compact';
|
||||
const [autoExpanded, setAutoExpanded] = useState(false);
|
||||
|
||||
const previousSteps = getSelectableStepOutputRefs(pipeline, selectedNode);
|
||||
const cycleWarning = formatCycleWarning(detectReferenceCycles(pipeline), pipeline);
|
||||
|
||||
const isChecked = (ref) =>
|
||||
(selectedNode.inputs || []).some((i) => i.ref === ref);
|
||||
|
||||
return (
|
||||
<div className={`studio-var-panel ${isCompact ? 'studio-var-panel--compact' : ''}`}>
|
||||
<h3 className="studio-var-panel__title">
|
||||
<FieldLabel tip="引用前序步骤的世界书条目;核心目的、思考流程等由系统自动注入">
|
||||
上文引用
|
||||
</FieldLabel>
|
||||
</h3>
|
||||
|
||||
{cycleWarning && (
|
||||
<p className="studio-var-cycle-warn" role="alert">
|
||||
{cycleWarning}
|
||||
</p>
|
||||
)}
|
||||
|
||||
<div className="studio-var-group studio-var-group--auto">
|
||||
<button
|
||||
type="button"
|
||||
className="studio-var-auto-toggle"
|
||||
onClick={() => setAutoExpanded((v) => !v)}
|
||||
aria-expanded={autoExpanded}
|
||||
>
|
||||
<span className="studio-var-group__title">系统自动注入</span>
|
||||
<span className="studio-var-auto-toggle__meta">
|
||||
{autoExpanded
|
||||
? '收起'
|
||||
: `${AUTO_INJECTED_CONTEXT_ITEMS.length} 项已默认注入`}
|
||||
</span>
|
||||
<span className="studio-var-auto-toggle__chevron" aria-hidden="true">
|
||||
{autoExpanded ? '▾' : '▸'}
|
||||
</span>
|
||||
</button>
|
||||
|
||||
{autoExpanded ? (
|
||||
<div className="studio-var-auto-list">
|
||||
{AUTO_INJECTED_CONTEXT_ITEMS.map((item) => (
|
||||
<AutoInjectedItem key={item.id} item={item} />
|
||||
))}
|
||||
</div>
|
||||
) : (
|
||||
<p className="studio-edit-hint studio-var-auto-hint">
|
||||
运行时自动附带,无需手动开启。展开可查看各项含义。
|
||||
</p>
|
||||
)}
|
||||
</div>
|
||||
|
||||
<div className="studio-var-group">
|
||||
<h4 className="studio-var-group__title">前序步骤产物</h4>
|
||||
<p className="studio-edit-hint studio-var-manual-hint">
|
||||
开关开启后,将把对应步骤的最终世界书条目注入本步上下文。
|
||||
</p>
|
||||
<div className={`studio-var-grid ${isCompact ? 'studio-var-grid--compact' : ''}`}>
|
||||
{previousSteps.length === 0 ? (
|
||||
<p className="studio-edit-hint studio-var-empty">暂无可引用的前序世界书步骤</p>
|
||||
) : (
|
||||
previousSteps.map((node) => {
|
||||
const ref = buildStepOutputRef(node.id);
|
||||
const checked = isChecked(ref);
|
||||
const label = dynamicChipLabel(ref, pipeline, stepOutputLabel(node.displayName));
|
||||
return (
|
||||
<VariableSwitch
|
||||
key={ref}
|
||||
label={label}
|
||||
selected={checked}
|
||||
tooltip={dynamicTooltip(ref, pipeline)}
|
||||
onToggle={() =>
|
||||
onToggleRef(ref, stepOutputLabel(node.displayName))
|
||||
}
|
||||
/>
|
||||
);
|
||||
})
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
163
frontend/src/components/Studio/edit/WorldbookInsertion.jsx
Normal file
163
frontend/src/components/Studio/edit/WorldbookInsertion.jsx
Normal file
@@ -0,0 +1,163 @@
|
||||
import React from 'react';
|
||||
import FieldLabel from './FieldLabel';
|
||||
|
||||
const POSITION_OPTIONS = [
|
||||
{ value: 0, label: '角色定义之后' },
|
||||
{ value: 1, label: '角色定义之前' },
|
||||
{ value: 2, label: '示例对话之前' },
|
||||
{ value: 3, label: '示例对话之后' },
|
||||
{ value: 4, label: '系统提示/作者注释' },
|
||||
{ value: 5, label: '作为系统消息' },
|
||||
{ value: 6, label: '深度插入' },
|
||||
{ value: 7, label: '宏替换' },
|
||||
];
|
||||
|
||||
const ACTIVATION_OPTIONS = [
|
||||
{ value: 'permanent', label: '永久激活' },
|
||||
{ value: 'keyword', label: '关键词触发' },
|
||||
{ value: 'rag', label: 'RAG 检索' },
|
||||
];
|
||||
|
||||
export default function WorldbookInsertion({
|
||||
insertion,
|
||||
onUpdateInsertion,
|
||||
onUpdateRagConfig,
|
||||
}) {
|
||||
const activation = insertion.activationType ?? 'permanent';
|
||||
|
||||
return (
|
||||
<div className="studio-wb-insert">
|
||||
<h3 className="studio-wb-insert__title">世界书插入</h3>
|
||||
|
||||
<div className="studio-wb-insert__fields">
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">
|
||||
<FieldLabel tip="条目写入角色卡时的深度位置,影响模型读取上下文的顺序">
|
||||
插入位置
|
||||
</FieldLabel>
|
||||
</span>
|
||||
<select
|
||||
className="studio-edit-select"
|
||||
value={insertion.position ?? 1}
|
||||
onChange={(e) =>
|
||||
onUpdateInsertion('position', Number(e.target.value))
|
||||
}
|
||||
>
|
||||
{POSITION_OPTIONS.map((opt) => (
|
||||
<option key={opt.value} value={opt.value}>
|
||||
{opt.label}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
</label>
|
||||
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">
|
||||
<FieldLabel tip="永久激活始终注入;关键词/RAG 仅在匹配时注入">
|
||||
激活方式
|
||||
</FieldLabel>
|
||||
</span>
|
||||
<select
|
||||
className="studio-edit-select"
|
||||
value={activation}
|
||||
onChange={(e) =>
|
||||
onUpdateInsertion('activationType', e.target.value)
|
||||
}
|
||||
>
|
||||
{ACTIVATION_OPTIONS.map((opt) => (
|
||||
<option key={opt.value} value={opt.value}>
|
||||
{opt.label}
|
||||
</option>
|
||||
))}
|
||||
</select>
|
||||
</label>
|
||||
|
||||
{activation === 'keyword' && (
|
||||
<>
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">
|
||||
<FieldLabel tip="主关键词用于触发本条目进入上下文">主关键词</FieldLabel>
|
||||
</span>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={insertion.key ?? ''}
|
||||
onChange={(e) => onUpdateInsertion('key', e.target.value)}
|
||||
placeholder="触发本条目所需主词"
|
||||
/>
|
||||
</label>
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">
|
||||
<FieldLabel tip="可选,辅助匹配多个相关词">次要关键词</FieldLabel>
|
||||
</span>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={insertion.keysecondary ?? ''}
|
||||
onChange={(e) =>
|
||||
onUpdateInsertion('keysecondary', e.target.value)
|
||||
}
|
||||
placeholder="可选,辅助匹配"
|
||||
/>
|
||||
</label>
|
||||
</>
|
||||
)}
|
||||
|
||||
{activation === 'rag' && (
|
||||
<>
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">RAG 库 ID</span>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={insertion.ragConfig?.libraryId ?? ''}
|
||||
onChange={(e) =>
|
||||
onUpdateRagConfig('libraryId', e.target.value)
|
||||
}
|
||||
/>
|
||||
</label>
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">相似度阈值</span>
|
||||
<input
|
||||
type="number"
|
||||
step="0.01"
|
||||
min="0"
|
||||
max="1"
|
||||
className="studio-edit-input"
|
||||
value={insertion.ragConfig?.threshold ?? 0.5}
|
||||
onChange={(e) =>
|
||||
onUpdateRagConfig('threshold', Number(e.target.value))
|
||||
}
|
||||
/>
|
||||
</label>
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">最大条数</span>
|
||||
<input
|
||||
type="number"
|
||||
min="1"
|
||||
className="studio-edit-input"
|
||||
value={insertion.ragConfig?.maxEntries ?? 3}
|
||||
onChange={(e) =>
|
||||
onUpdateRagConfig('maxEntries', Number(e.target.value))
|
||||
}
|
||||
/>
|
||||
</label>
|
||||
</>
|
||||
)}
|
||||
|
||||
<label className="studio-edit-field">
|
||||
<span className="studio-edit-field-label">
|
||||
<FieldLabel tip="写入世界书条目的说明注释,便于在编辑器中识别">备注</FieldLabel>
|
||||
</span>
|
||||
<input
|
||||
type="text"
|
||||
className="studio-edit-input"
|
||||
value={insertion.comment ?? ''}
|
||||
onChange={(e) => onUpdateInsertion('comment', e.target.value)}
|
||||
placeholder="写入世界书条目的说明注释"
|
||||
/>
|
||||
</label>
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
347
frontend/src/components/Studio/edit/variableUtils.js
Normal file
347
frontend/src/components/Studio/edit/variableUtils.js
Normal file
@@ -0,0 +1,347 @@
|
||||
/** Resolve node displayName from dynamic ref like `aesthetic.output`. */
|
||||
export function nodeDisplayNameFromRef(ref, pipeline) {
|
||||
const nodes = pipeline?.nodes || [];
|
||||
const node = nodes.find((n) => ref.startsWith(`${n.id}.`));
|
||||
return node?.displayName || ref.split('.')[0];
|
||||
}
|
||||
|
||||
/** Extract pipeline node id from a step output ref (`nodeId.output`). */
|
||||
export function parseNodeRef(ref) {
|
||||
if (!ref || typeof ref !== 'string') return null;
|
||||
const m = ref.match(/^([^.]+)\.output$/);
|
||||
return m ? m[1] : null;
|
||||
}
|
||||
|
||||
/** Locale-aware display name compare (Chinese + numeric tie-break). */
|
||||
export function compareNodeDisplayNames(a, b) {
|
||||
const nameA = a?.displayName || '';
|
||||
const nameB = b?.displayName || '';
|
||||
return nameA.localeCompare(nameB, 'zh-CN', {
|
||||
numeric: true,
|
||||
sensitivity: 'base',
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* Topological layer sort by `inputs[].ref` dependencies.
|
||||
* Same layer → display name (zh-CN localeCompare).
|
||||
*/
|
||||
export function sortNodesByLogic(pipeline) {
|
||||
const nodes = pipeline?.nodes || [];
|
||||
if (nodes.length <= 1) return [...nodes];
|
||||
|
||||
const edges = buildNodeDependencyEdges(pipeline);
|
||||
const depth = Object.fromEntries(nodes.map((n) => [n.id, 0]));
|
||||
const pipelineIndex = Object.fromEntries(nodes.map((n, i) => [n.id, i]));
|
||||
|
||||
let changed = true;
|
||||
while (changed) {
|
||||
changed = false;
|
||||
edges.forEach(({ from, to }) => {
|
||||
const next = (depth[from] || 0) + 1;
|
||||
if (next > (depth[to] || 0)) {
|
||||
depth[to] = next;
|
||||
changed = true;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
const layers = {};
|
||||
nodes.forEach((n) => {
|
||||
const d = depth[n.id] || 0;
|
||||
if (!layers[d]) layers[d] = [];
|
||||
layers[d].push(n);
|
||||
});
|
||||
|
||||
const sorted = [];
|
||||
Object.keys(layers)
|
||||
.map(Number)
|
||||
.sort((a, b) => a - b)
|
||||
.forEach((d) => {
|
||||
layers[d].sort((a, b) => {
|
||||
const byName = compareNodeDisplayNames(a, b);
|
||||
if (byName !== 0) return byName;
|
||||
return (pipelineIndex[a.id] ?? 0) - (pipelineIndex[b.id] ?? 0);
|
||||
});
|
||||
sorted.push(...layers[d]);
|
||||
});
|
||||
|
||||
return sorted;
|
||||
}
|
||||
|
||||
/** Order node ids by manual pipeline order snapshot. */
|
||||
export function orderNodesByIds(nodes, orderIds) {
|
||||
const orderMap = Object.fromEntries((orderIds || []).map((id, i) => [id, i]));
|
||||
return [...(nodes || [])].sort((a, b) => {
|
||||
const ia = orderMap[a.id] ?? Number.MAX_SAFE_INTEGER;
|
||||
const ib = orderMap[b.id] ?? Number.MAX_SAFE_INTEGER;
|
||||
if (ia !== ib) return ia - ib;
|
||||
return compareNodeDisplayNames(a, b);
|
||||
});
|
||||
}
|
||||
|
||||
/** Dependency edges derived from `inputs[].ref` between pipeline nodes. */
|
||||
export function buildNodeDependencyEdges(pipeline) {
|
||||
const nodes = pipeline?.nodes || [];
|
||||
const nodeIds = new Set(nodes.map((n) => n.id));
|
||||
const edges = [];
|
||||
const edgeKeys = new Set();
|
||||
|
||||
nodes.forEach((node) => {
|
||||
(node.inputs || []).forEach((inp) => {
|
||||
const src = parseNodeRef(inp.ref);
|
||||
if (!src || src === node.id || !nodeIds.has(src)) return;
|
||||
const key = `${src}->${node.id}`;
|
||||
if (edgeKeys.has(key)) return;
|
||||
edgeKeys.add(key);
|
||||
edges.push({ from: src, to: node.id });
|
||||
});
|
||||
});
|
||||
|
||||
return edges;
|
||||
}
|
||||
|
||||
/** True if adding `fromNodeId → toNodeId` would close a dependency cycle. */
|
||||
export function wouldCreateDependencyCycle(pipeline, toNodeId, fromNodeId) {
|
||||
if (!toNodeId || !fromNodeId || toNodeId === fromNodeId) return true;
|
||||
|
||||
const edges = buildNodeDependencyEdges(pipeline);
|
||||
const visited = new Set();
|
||||
const stack = [toNodeId];
|
||||
|
||||
while (stack.length) {
|
||||
const cur = stack.pop();
|
||||
if (cur === fromNodeId) return true;
|
||||
if (visited.has(cur)) continue;
|
||||
visited.add(cur);
|
||||
edges.filter((e) => e.from === cur).forEach((e) => stack.push(e.to));
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
/** Find dependency cycles in the pipeline (node id paths). */
|
||||
export function detectReferenceCycles(pipeline) {
|
||||
const nodes = pipeline?.nodes || [];
|
||||
const nodeIds = nodes.map((n) => n.id);
|
||||
const adj = Object.fromEntries(nodeIds.map((id) => [id, []]));
|
||||
|
||||
buildNodeDependencyEdges(pipeline).forEach(({ from, to }) => {
|
||||
adj[from].push(to);
|
||||
});
|
||||
|
||||
const cycles = [];
|
||||
const visited = new Set();
|
||||
const stack = new Set();
|
||||
const path = [];
|
||||
|
||||
const dfs = (nodeId) => {
|
||||
visited.add(nodeId);
|
||||
stack.add(nodeId);
|
||||
path.push(nodeId);
|
||||
|
||||
(adj[nodeId] || []).forEach((next) => {
|
||||
if (!visited.has(next)) {
|
||||
dfs(next);
|
||||
} else if (stack.has(next)) {
|
||||
const start = path.indexOf(next);
|
||||
if (start >= 0) {
|
||||
cycles.push([...path.slice(start), next]);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
path.pop();
|
||||
stack.delete(nodeId);
|
||||
};
|
||||
|
||||
nodeIds.forEach((id) => {
|
||||
if (!visited.has(id)) dfs(id);
|
||||
});
|
||||
|
||||
return cycles;
|
||||
}
|
||||
|
||||
export function formatCycleWarning(cycles, pipeline) {
|
||||
if (!cycles.length) return null;
|
||||
const nameOf = (id) =>
|
||||
pipeline?.nodes?.find((n) => n.id === id)?.displayName || id;
|
||||
const first = cycles[0];
|
||||
const chain = first.map(nameOf).join(' → ');
|
||||
return `检测到循环引用:${chain}`;
|
||||
}
|
||||
|
||||
/** Worldbook steps that may be referenced without creating a cycle. */
|
||||
export function getSelectableStepOutputRefs(pipeline, selectedNode) {
|
||||
if (!selectedNode?.id) return [];
|
||||
|
||||
return (pipeline?.nodes || []).filter(
|
||||
(n) =>
|
||||
n.enabled !== false &&
|
||||
n.skillId === 'studio.worldbook_entry' &&
|
||||
n.id !== selectedNode.id &&
|
||||
!wouldCreateDependencyCycle(pipeline, selectedNode.id, n.id)
|
||||
);
|
||||
}
|
||||
|
||||
export function buildStepOutputRef(nodeId) {
|
||||
return `${nodeId}.output`;
|
||||
}
|
||||
|
||||
export function stepOutputLabel(displayName) {
|
||||
return `${displayName} · 世界书条目`;
|
||||
}
|
||||
|
||||
export function dynamicChipLabel(ref, pipeline, fallbackLabel) {
|
||||
const name = nodeDisplayNameFromRef(ref, pipeline);
|
||||
if (name && name !== ref.split('.')[0]) return stepOutputLabel(name);
|
||||
return fallbackLabel || stepOutputLabel(name);
|
||||
}
|
||||
|
||||
export function dynamicTooltip(ref, pipeline) {
|
||||
const name = nodeDisplayNameFromRef(ref, pipeline);
|
||||
return `引用前序步骤「${name}」的最终世界书条目 · ${ref}`;
|
||||
}
|
||||
|
||||
export const DEFAULT_THINKING_PROMPT = `====== 思考流程 ======
|
||||
Step1: 简短确认任务性质(新设计/修改)
|
||||
Step2: 阅读绑定角色/世界书与上文引用
|
||||
Step3: 按步骤目标起草世界书条目
|
||||
Step4: 对照评价维度自检并优化表述
|
||||
|
||||
(核心目的、评价标准与优化建议由系统在运行时自动注入,无需在此填写)`;
|
||||
|
||||
export const AUTO_INJECTED_CONTEXT_ITEMS = [
|
||||
{
|
||||
id: 'currentProduct',
|
||||
label: '目前产物',
|
||||
description:
|
||||
'当前步骤已生成的世界书条目草稿或最新版本,供模型在迭代修改时对照与延续。',
|
||||
},
|
||||
{
|
||||
id: 'thinkingFlow',
|
||||
label: '思考流程',
|
||||
description:
|
||||
'本步骤配置的 thinkingPrompt,引导模型按既定步骤推理与自检。',
|
||||
},
|
||||
{
|
||||
id: 'coreGoal',
|
||||
label: '核心目的',
|
||||
description:
|
||||
'本步骤的 stepGoal(步骤目标),明确本步要产出的内容与边界。',
|
||||
},
|
||||
{
|
||||
id: 'scoringCriteria',
|
||||
label: '评价标准与优化建议',
|
||||
description:
|
||||
'本步骤启用的 scoring 评价维度及准则,用于模型自检与优化表述。',
|
||||
},
|
||||
];
|
||||
|
||||
export const AUTO_INJECTED_CONTEXT_LABELS = AUTO_INJECTED_CONTEXT_ITEMS.map(
|
||||
(item) => item.label
|
||||
);
|
||||
|
||||
const NODE_W = 128;
|
||||
const NODE_H = 52;
|
||||
const GAP_X = 48;
|
||||
const GAP_Y = 64;
|
||||
const PAD = 24;
|
||||
|
||||
function verticalEdgePath(from, to) {
|
||||
const x1 = from.x + NODE_W / 2;
|
||||
const y1 = from.y + NODE_H;
|
||||
const x2 = to.x + NODE_W / 2;
|
||||
const y2 = to.y;
|
||||
const dy = Math.max(24, (y2 - y1) * 0.45);
|
||||
return `M ${x1} ${y1} C ${x1} ${y1 + dy}, ${x2} ${y2 - dy}, ${x2} ${y2}`;
|
||||
}
|
||||
|
||||
/** Layout nodes as a vertical dependency tree for SVG progress graphs. */
|
||||
export function buildGraphLayout(pipeline, nodeStates, options = {}) {
|
||||
const vertical = options.vertical !== false;
|
||||
const nodes = (pipeline?.nodes || []).filter((n) => n.enabled !== false);
|
||||
const stateMap = Object.fromEntries(
|
||||
(nodeStates || []).map((s) => [s.nodeId, s])
|
||||
);
|
||||
const pipelineIndex = Object.fromEntries(
|
||||
nodes.map((n, i) => [n.id, i])
|
||||
);
|
||||
|
||||
const edges = buildNodeDependencyEdges(pipeline);
|
||||
|
||||
const depth = {};
|
||||
nodes.forEach((n) => {
|
||||
depth[n.id] = 0;
|
||||
});
|
||||
|
||||
let changed = true;
|
||||
while (changed) {
|
||||
changed = false;
|
||||
edges.forEach(({ from, to }) => {
|
||||
const next = (depth[from] || 0) + 1;
|
||||
if (next > (depth[to] || 0)) {
|
||||
depth[to] = next;
|
||||
changed = true;
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
const layers = {};
|
||||
nodes.forEach((n) => {
|
||||
const d = depth[n.id] || 0;
|
||||
if (!layers[d]) layers[d] = [];
|
||||
layers[d].push(n);
|
||||
});
|
||||
|
||||
Object.values(layers).forEach((layer) => {
|
||||
layer.sort((a, b) => (pipelineIndex[a.id] ?? 0) - (pipelineIndex[b.id] ?? 0));
|
||||
});
|
||||
|
||||
const positions = {};
|
||||
Object.keys(layers)
|
||||
.map(Number)
|
||||
.sort((a, b) => a - b)
|
||||
.forEach((d) => {
|
||||
const layer = layers[d];
|
||||
layer.forEach((node, col) => {
|
||||
if (vertical) {
|
||||
positions[node.id] = {
|
||||
x: PAD + col * (NODE_W + GAP_X),
|
||||
y: PAD + d * (NODE_H + GAP_Y),
|
||||
};
|
||||
} else {
|
||||
positions[node.id] = {
|
||||
x: PAD + d * (NODE_W + GAP_X),
|
||||
y: PAD + col * (NODE_H + GAP_Y),
|
||||
};
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
const maxLayer = Math.max(0, ...Object.keys(layers).map(Number));
|
||||
const maxCols = Math.max(1, ...Object.values(layers).map((l) => l.length));
|
||||
|
||||
const width = vertical
|
||||
? PAD * 2 + maxCols * NODE_W + (maxCols - 1) * GAP_X
|
||||
: PAD * 2 + (maxLayer + 1) * NODE_W + maxLayer * GAP_X;
|
||||
const height = vertical
|
||||
? PAD * 2 + (maxLayer + 1) * NODE_H + maxLayer * GAP_Y
|
||||
: PAD * 2 + maxCols * NODE_H + (maxCols - 1) * GAP_Y;
|
||||
|
||||
return {
|
||||
nodes: nodes.map((n) => ({
|
||||
id: n.id,
|
||||
label: n.displayName,
|
||||
status: stateMap[n.id]?.status || 'pending',
|
||||
x: positions[n.id]?.x ?? PAD,
|
||||
y: positions[n.id]?.y ?? PAD,
|
||||
})),
|
||||
edges,
|
||||
width,
|
||||
height,
|
||||
nodeW: NODE_W,
|
||||
nodeH: NODE_H,
|
||||
edgePath: vertical ? verticalEdgePath : null,
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,37 @@
|
||||
/**
|
||||
* Backend workflow variable ref → Chinese display label.
|
||||
* Mirrors data/agent/workflow_variables.json builtIn entries.
|
||||
*/
|
||||
export const BUILTIN_WORKFLOW_VARIABLE_LABELS = {
|
||||
'workflow.goal': '工作流目标文本',
|
||||
'workflow.boundWorldbook': '绑定世界书摘要',
|
||||
'workflow.boundCharacter': '绑定角色卡摘要',
|
||||
};
|
||||
|
||||
/** Build ref → label map from /api/studio/variables response. */
|
||||
export function buildWorkflowVariableLabelMap(workflowVariables) {
|
||||
const map = { ...BUILTIN_WORKFLOW_VARIABLE_LABELS };
|
||||
const builtIn = workflowVariables?.builtIn || [];
|
||||
const dynamic = workflowVariables?.dynamic || [];
|
||||
[...builtIn, ...dynamic].forEach((item) => {
|
||||
if (item?.ref && item?.label) {
|
||||
map[item.ref] = item.label;
|
||||
}
|
||||
});
|
||||
return map;
|
||||
}
|
||||
|
||||
/**
|
||||
* Resolve Chinese label for a workflow variable key shown in run preview.
|
||||
* @param {string} ref - e.g. workflow.goal or nodeId.output
|
||||
* @param {Record<string, string>} [labelMap] - optional map from buildWorkflowVariableLabelMap
|
||||
*/
|
||||
export function getWorkflowVariableLabel(ref, labelMap = BUILTIN_WORKFLOW_VARIABLE_LABELS) {
|
||||
if (!ref) return '';
|
||||
if (labelMap[ref]) return labelMap[ref];
|
||||
const outputMatch = ref.match(/^([^.]+)\.output$/);
|
||||
if (outputMatch) {
|
||||
return `${outputMatch[1]} · 世界书条目`;
|
||||
}
|
||||
return ref;
|
||||
}
|
||||
1
frontend/src/components/Studio/index.js
Normal file
1
frontend/src/components/Studio/index.js
Normal file
@@ -0,0 +1 @@
|
||||
export { default } from './StudioPage';
|
||||
28
frontend/src/components/Studio/studioRunUtils.js
Normal file
28
frontend/src/components/Studio/studioRunUtils.js
Normal file
@@ -0,0 +1,28 @@
|
||||
/**
|
||||
* Resolve bound character display name from workflow variable and/or project meta.
|
||||
*/
|
||||
export function resolveBoundCharacterName({ boundCharacterVar, characterId, characters = [] }) {
|
||||
if (boundCharacterVar) {
|
||||
const nameMatch = String(boundCharacterVar).match(/名称[::]\s*(.+?)(?:\n|$)/);
|
||||
if (nameMatch?.[1]?.trim()) {
|
||||
return nameMatch[1].trim();
|
||||
}
|
||||
}
|
||||
|
||||
if (characterId && characters.length > 0) {
|
||||
const byId = characters.find((c) => c.id === characterId);
|
||||
if (byId?.name) return byId.name;
|
||||
const byName = characters.find((c) => c.name === characterId);
|
||||
if (byName?.name) return byName.name;
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
/** Index of the last user message in stepMessages, or -1. */
|
||||
export function findLastUserMessageIndex(stepMessages = []) {
|
||||
for (let i = stepMessages.length - 1; i >= 0; i -= 1) {
|
||||
if (stepMessages[i]?.role === 'user') return i;
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
@@ -4,6 +4,8 @@ import useAppLayoutStore from '../../Store/AppLayoutSlice'; // ✅ 新增
|
||||
import useUserStore from '../../Store/UserSlice'; // ✅ 新增 - 用户角色
|
||||
import useWorldBookStore from '../../Store/SideBarLeft/WorldBookSlice';
|
||||
import useApiConfigStore from '../../Store/SideBarLeft/ApiConfigSlice';
|
||||
import PageModeToggle from './items/PageModeToggle';
|
||||
import StudioRunControls from './items/StudioRunControls';
|
||||
import ThemeToggle from './items/ThemeToggle';
|
||||
import RegexPanel from '../SideBarLeft/tabs/Regex/RegexPanel'; // ✅ 新增:导入正则管理组件
|
||||
import './TopBar.css';
|
||||
@@ -16,9 +18,12 @@ const Toolbar = () => {
|
||||
const {
|
||||
sidebarMode,
|
||||
colorTheme,
|
||||
activePage,
|
||||
setSidebarMode,
|
||||
setColorTheme
|
||||
setColorTheme,
|
||||
} = useAppLayoutStore();
|
||||
|
||||
const isStudioRunPage = activePage === 'studio_run';
|
||||
|
||||
// ✅ 从 UserStore 获取用户角色和方法
|
||||
const { currentUserRole, userRoles, updateRoleName, updateRoleDescription, selectUserRole, addUserRole } = useUserStore();
|
||||
@@ -148,6 +153,8 @@ const Toolbar = () => {
|
||||
<div className="top-bar-content">
|
||||
{/* 左侧:状态徽章区域 */}
|
||||
<div className="status-section">
|
||||
{isStudioRunPage ? <StudioRunControls /> : null}
|
||||
|
||||
{/* 当前玩家角色 */}
|
||||
<div
|
||||
className="status-badge"
|
||||
@@ -207,6 +214,9 @@ const Toolbar = () => {
|
||||
⊞
|
||||
</button>
|
||||
|
||||
{/* 页面模式 */}
|
||||
<PageModeToggle />
|
||||
|
||||
{/* 主题切换 */}
|
||||
<ThemeToggle />
|
||||
</div>
|
||||
|
||||
@@ -0,0 +1,88 @@
|
||||
.page-mode-toggle-wrapper {
|
||||
position: relative;
|
||||
}
|
||||
|
||||
.page-mode-toggle {
|
||||
/* inherits from .action-btn */
|
||||
}
|
||||
|
||||
.page-mode-selector-panel {
|
||||
position: absolute;
|
||||
top: calc(100% + 8px);
|
||||
right: 0;
|
||||
background-color: var(--color-bg-secondary);
|
||||
border: 1px solid var(--color-border);
|
||||
border-radius: var(--radius-lg);
|
||||
box-shadow: var(--shadow-xl);
|
||||
min-width: 160px;
|
||||
z-index: var(--z-popover);
|
||||
animation: pageModeFadeInScale var(--transition-fast);
|
||||
}
|
||||
|
||||
@keyframes pageModeFadeInScale {
|
||||
from {
|
||||
opacity: 0;
|
||||
transform: scale(0.95) translateY(-10px);
|
||||
}
|
||||
to {
|
||||
opacity: 1;
|
||||
transform: scale(1) translateY(0);
|
||||
}
|
||||
}
|
||||
|
||||
.page-mode-selector-header {
|
||||
padding: var(--spacing-sm) var(--spacing-md);
|
||||
border-bottom: 1px solid var(--color-border-light);
|
||||
}
|
||||
|
||||
.page-mode-selector-title {
|
||||
font-size: 0.85rem;
|
||||
font-weight: 600;
|
||||
color: var(--color-text-primary);
|
||||
font-family: var(--font-ui);
|
||||
}
|
||||
|
||||
.page-mode-list {
|
||||
padding: var(--spacing-xs);
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
gap: 2px;
|
||||
}
|
||||
|
||||
.page-mode-item {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: var(--spacing-sm);
|
||||
padding: var(--spacing-sm) var(--spacing-md);
|
||||
background: transparent;
|
||||
border: none;
|
||||
border-radius: var(--radius-md);
|
||||
cursor: pointer;
|
||||
transition: all var(--transition-fast);
|
||||
text-align: left;
|
||||
width: 100%;
|
||||
}
|
||||
|
||||
.page-mode-item:hover {
|
||||
background-color: var(--color-bg-tertiary);
|
||||
}
|
||||
|
||||
.page-mode-item.active {
|
||||
background-color: var(--color-accent-light);
|
||||
border: 1px solid var(--color-accent);
|
||||
}
|
||||
|
||||
.page-mode-item-icon {
|
||||
font-size: 1.3rem;
|
||||
line-height: 1;
|
||||
flex-shrink: 0;
|
||||
width: 24px;
|
||||
text-align: center;
|
||||
}
|
||||
|
||||
.page-mode-item-label {
|
||||
font-size: 0.9rem;
|
||||
font-weight: 600;
|
||||
color: var(--color-text-primary);
|
||||
font-family: var(--font-ui);
|
||||
}
|
||||
@@ -0,0 +1,83 @@
|
||||
import React, { useState, useEffect, useRef } from 'react';
|
||||
import useAppLayoutStore from '../../../../Store/AppLayoutSlice';
|
||||
import './PageModeToggle.css';
|
||||
|
||||
const PAGE_MODES = [
|
||||
{ id: 'chat', label: '聊天', icon: '💬' },
|
||||
{ id: 'studio_edit', label: '工作流编辑', icon: '✏️' },
|
||||
{ id: 'studio_run', label: '创作运行', icon: '🚀' },
|
||||
{ id: 'novel', label: '爽文', icon: '📖' },
|
||||
{ id: 'room', label: '房间', icon: '🏠' },
|
||||
];
|
||||
|
||||
const PageModeToggle = () => {
|
||||
const [isOpen, setIsOpen] = useState(false);
|
||||
const panelRef = useRef(null);
|
||||
|
||||
const { activePage, setActivePage } = useAppLayoutStore();
|
||||
|
||||
useEffect(() => {
|
||||
const handleClickOutside = (event) => {
|
||||
if (panelRef.current && !panelRef.current.contains(event.target)) {
|
||||
setIsOpen(false);
|
||||
}
|
||||
};
|
||||
|
||||
if (isOpen) {
|
||||
document.addEventListener('mousedown', handleClickOutside);
|
||||
}
|
||||
|
||||
return () => {
|
||||
document.removeEventListener('mousedown', handleClickOutside);
|
||||
};
|
||||
}, [isOpen]);
|
||||
|
||||
const handleModeSelect = (modeId) => {
|
||||
setActivePage(modeId);
|
||||
setIsOpen(false);
|
||||
};
|
||||
|
||||
const resolvedPage = activePage === 'studio' ? 'studio_edit' : activePage;
|
||||
|
||||
const currentMode =
|
||||
PAGE_MODES.find((m) => m.id === resolvedPage) || PAGE_MODES[0];
|
||||
|
||||
return (
|
||||
<div className="page-mode-toggle-wrapper" ref={panelRef}>
|
||||
<button
|
||||
className="action-btn page-mode-toggle"
|
||||
onClick={() => setIsOpen(!isOpen)}
|
||||
title={`当前模式:${currentMode.label}`}
|
||||
aria-haspopup="listbox"
|
||||
aria-expanded={isOpen}
|
||||
>
|
||||
{currentMode.icon}
|
||||
</button>
|
||||
|
||||
{isOpen && (
|
||||
<div className="page-mode-selector-panel" role="listbox" aria-label="应用模式">
|
||||
<div className="page-mode-selector-header">
|
||||
<span className="page-mode-selector-title">应用模式</span>
|
||||
</div>
|
||||
<div className="page-mode-list">
|
||||
{PAGE_MODES.map((mode) => (
|
||||
<button
|
||||
key={mode.id}
|
||||
type="button"
|
||||
role="option"
|
||||
aria-selected={resolvedPage === mode.id}
|
||||
className={`page-mode-item ${resolvedPage === mode.id ? 'active' : ''}`}
|
||||
onClick={() => handleModeSelect(mode.id)}
|
||||
>
|
||||
<span className="page-mode-item-icon">{mode.icon}</span>
|
||||
<span className="page-mode-item-label">{mode.label}</span>
|
||||
</button>
|
||||
))}
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
);
|
||||
};
|
||||
|
||||
export default PageModeToggle;
|
||||
@@ -0,0 +1 @@
|
||||
export { default } from './PageModeToggle';
|
||||
@@ -0,0 +1,91 @@
|
||||
.studio-run-topbar-controls {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: var(--spacing-sm);
|
||||
flex-shrink: 0;
|
||||
margin-right: var(--spacing-md);
|
||||
}
|
||||
|
||||
.studio-run-topbar-label {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: var(--spacing-xs);
|
||||
font-size: 0.8rem;
|
||||
color: var(--color-text-muted);
|
||||
}
|
||||
|
||||
.studio-run-topbar-label-text {
|
||||
white-space: nowrap;
|
||||
}
|
||||
|
||||
.studio-run-topbar-select {
|
||||
padding: 4px var(--spacing-sm);
|
||||
border-radius: var(--radius-sm);
|
||||
border: 1px solid var(--color-border);
|
||||
background: var(--color-bg-tertiary);
|
||||
color: var(--color-text-primary);
|
||||
font-size: 0.8rem;
|
||||
min-width: 140px;
|
||||
max-width: 200px;
|
||||
}
|
||||
|
||||
.studio-run-topbar-btn {
|
||||
padding: 4px var(--spacing-md);
|
||||
border-radius: var(--radius-sm);
|
||||
font-size: 0.8rem;
|
||||
cursor: pointer;
|
||||
white-space: nowrap;
|
||||
font-family: inherit;
|
||||
transition: opacity 0.15s ease, background-color 0.15s ease;
|
||||
}
|
||||
|
||||
.studio-run-topbar-btn--primary {
|
||||
border: none;
|
||||
background: var(--color-accent);
|
||||
color: var(--color-text-inverse, #fff);
|
||||
}
|
||||
|
||||
.studio-run-topbar-btn--primary:hover:not(:disabled) {
|
||||
opacity: 0.9;
|
||||
}
|
||||
|
||||
.studio-run-topbar-btn--ghost {
|
||||
border: 1px solid var(--color-border);
|
||||
background: transparent;
|
||||
color: var(--color-text-secondary);
|
||||
}
|
||||
|
||||
.studio-run-topbar-btn--ghost:hover:not(:disabled) {
|
||||
background: var(--color-bg-tertiary);
|
||||
color: var(--color-text-primary);
|
||||
}
|
||||
|
||||
.studio-run-topbar-btn:disabled {
|
||||
opacity: 0.5;
|
||||
cursor: not-allowed;
|
||||
}
|
||||
|
||||
.studio-run-topbar-character {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: var(--spacing-xs);
|
||||
padding: 4px var(--spacing-sm);
|
||||
border-radius: var(--radius-sm);
|
||||
background: var(--color-bg-tertiary);
|
||||
border: 1px solid var(--color-border-light);
|
||||
max-width: 180px;
|
||||
}
|
||||
|
||||
.studio-run-topbar-character-icon {
|
||||
font-size: 0.85rem;
|
||||
flex-shrink: 0;
|
||||
opacity: 0.8;
|
||||
}
|
||||
|
||||
.studio-run-topbar-character-name {
|
||||
font-size: 0.8rem;
|
||||
color: var(--color-text-primary);
|
||||
overflow: hidden;
|
||||
text-overflow: ellipsis;
|
||||
white-space: nowrap;
|
||||
}
|
||||
@@ -0,0 +1,119 @@
|
||||
import React, { useCallback, useMemo } from 'react';
|
||||
|
||||
import useStudioStore from '../../../../Store/Studio/StudioSlice';
|
||||
import useCharacterStore from '../../../../Store/SideBarLeft/CharacterSlice';
|
||||
import { resolveBoundCharacterName } from '../../../Studio/studioRunUtils';
|
||||
|
||||
import './StudioRunControls.css';
|
||||
|
||||
function StudioRunControls() {
|
||||
const {
|
||||
projects,
|
||||
meta,
|
||||
runProjectId,
|
||||
currentRun,
|
||||
runLoading,
|
||||
runCreating,
|
||||
runAdvancing,
|
||||
runSessionEntered,
|
||||
setRunProjectId,
|
||||
setRunSessionEntered,
|
||||
fetchWorkflowVariables,
|
||||
fetchRuns,
|
||||
selectRun,
|
||||
createRun,
|
||||
} = useStudioStore();
|
||||
|
||||
const characters = useCharacterStore((s) => s.characters);
|
||||
|
||||
const boundCharacterName = useMemo(() => {
|
||||
return resolveBoundCharacterName({
|
||||
boundCharacterVar: currentRun?.workflowVariables?.['workflow.boundCharacter'],
|
||||
characterId: meta?.characterId,
|
||||
characters,
|
||||
});
|
||||
}, [currentRun?.workflowVariables, meta?.characterId, characters]);
|
||||
|
||||
const handleProjectChange = useCallback(
|
||||
async (e) => {
|
||||
const projectId = e.target.value;
|
||||
setRunProjectId(projectId);
|
||||
setRunSessionEntered(false);
|
||||
await fetchWorkflowVariables(projectId);
|
||||
const list = await fetchRuns(projectId);
|
||||
if (list[0]?.id) {
|
||||
await selectRun(list[0].id);
|
||||
}
|
||||
},
|
||||
[
|
||||
setRunProjectId,
|
||||
setRunSessionEntered,
|
||||
fetchWorkflowVariables,
|
||||
fetchRuns,
|
||||
selectRun,
|
||||
]
|
||||
);
|
||||
|
||||
const handleNewRun = useCallback(async () => {
|
||||
if (!runProjectId) return;
|
||||
setRunSessionEntered(false);
|
||||
await createRun(runProjectId);
|
||||
}, [runProjectId, setRunSessionEntered, createRun]);
|
||||
|
||||
const inSession = runSessionEntered && !!currentRun;
|
||||
|
||||
return (
|
||||
<div className="studio-run-topbar-controls">
|
||||
<label className="studio-run-topbar-label">
|
||||
<span className="studio-run-topbar-label-text">项目</span>
|
||||
<select
|
||||
className="studio-run-topbar-select"
|
||||
value={runProjectId || ''}
|
||||
onChange={handleProjectChange}
|
||||
disabled={runLoading || runCreating || runAdvancing}
|
||||
aria-label="选择 Studio 项目"
|
||||
>
|
||||
{projects.length === 0 ? (
|
||||
<option value="">暂无项目</option>
|
||||
) : (
|
||||
projects.map((p) => (
|
||||
<option key={p.id} value={p.id}>
|
||||
{p.name}
|
||||
</option>
|
||||
))
|
||||
)}
|
||||
</select>
|
||||
</label>
|
||||
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-topbar-btn studio-run-topbar-btn--primary"
|
||||
onClick={handleNewRun}
|
||||
disabled={!runProjectId || runCreating || runAdvancing}
|
||||
>
|
||||
{runCreating ? '创建中…' : '新开会话'}
|
||||
</button>
|
||||
|
||||
{boundCharacterName ? (
|
||||
<div className="studio-run-topbar-character" title="绑定角色卡">
|
||||
<span className="studio-run-topbar-character-icon" aria-hidden="true">
|
||||
🎭
|
||||
</span>
|
||||
<span className="studio-run-topbar-character-name">{boundCharacterName}</span>
|
||||
</div>
|
||||
) : null}
|
||||
|
||||
{inSession ? (
|
||||
<button
|
||||
type="button"
|
||||
className="studio-run-topbar-btn studio-run-topbar-btn--ghost"
|
||||
onClick={() => setRunSessionEntered(false)}
|
||||
>
|
||||
退出会话
|
||||
</button>
|
||||
) : null}
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
export default StudioRunControls;
|
||||
@@ -0,0 +1 @@
|
||||
export { default } from './StudioRunControls';
|
||||
@@ -27,6 +27,16 @@
|
||||
margin-top: 0; /* Ensure panels start from top */
|
||||
}
|
||||
|
||||
.main-container.placeholder-container,
|
||||
.main-container.studio-container {
|
||||
flex-direction: column;
|
||||
}
|
||||
|
||||
.main-container.studio-container {
|
||||
display: flex;
|
||||
min-height: 0;
|
||||
}
|
||||
|
||||
/* Smooth transitions for theme changes - only apply to specific properties */
|
||||
.app {
|
||||
transition: background-color var(--transition-normal),
|
||||
|
||||
15
scripts/docker-logs.ps1
Normal file
15
scripts/docker-logs.ps1
Normal file
@@ -0,0 +1,15 @@
|
||||
# 跟踪 Docker Compose 日志
|
||||
param(
|
||||
[ValidateSet("backend", "frontend", "")]
|
||||
[string]$Service = ""
|
||||
)
|
||||
|
||||
$ErrorActionPreference = "Stop"
|
||||
$Root = Split-Path -Parent (Split-Path -Parent $MyInvocation.MyCommand.Path)
|
||||
Set-Location $Root
|
||||
|
||||
if ($Service) {
|
||||
docker compose logs -f $Service
|
||||
} else {
|
||||
docker compose logs -f
|
||||
}
|
||||
17
scripts/docker-rebuild.ps1
Normal file
17
scripts/docker-rebuild.ps1
Normal file
@@ -0,0 +1,17 @@
|
||||
# 重新构建并启动指定服务(依赖或 Dockerfile 变更时使用)
|
||||
param(
|
||||
[Parameter(Mandatory = $true)]
|
||||
[ValidateSet("backend", "frontend")]
|
||||
[string]$Service
|
||||
)
|
||||
|
||||
$ErrorActionPreference = "Stop"
|
||||
$Root = Split-Path -Parent (Split-Path -Parent $MyInvocation.MyCommand.Path)
|
||||
Set-Location $Root
|
||||
|
||||
Write-Host "Rebuilding and starting $Service..." -ForegroundColor Cyan
|
||||
docker compose up -d --build $Service
|
||||
|
||||
if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
|
||||
|
||||
Write-Host "Done. $Service rebuilt and running." -ForegroundColor Green
|
||||
21
scripts/docker-restart.ps1
Normal file
21
scripts/docker-restart.ps1
Normal file
@@ -0,0 +1,21 @@
|
||||
# 重启容器(backend / frontend / all),无需重启 Docker Desktop
|
||||
param(
|
||||
[ValidateSet("backend", "frontend", "all")]
|
||||
[string]$Service = "all"
|
||||
)
|
||||
|
||||
$ErrorActionPreference = "Stop"
|
||||
$Root = Split-Path -Parent (Split-Path -Parent $MyInvocation.MyCommand.Path)
|
||||
Set-Location $Root
|
||||
|
||||
if ($Service -eq "all") {
|
||||
Write-Host "Restarting backend and frontend..." -ForegroundColor Cyan
|
||||
docker compose restart backend frontend
|
||||
} else {
|
||||
Write-Host "Restarting $Service..." -ForegroundColor Cyan
|
||||
docker compose restart $Service
|
||||
}
|
||||
|
||||
if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
|
||||
|
||||
Write-Host "Done. Docker Desktop was NOT restarted." -ForegroundColor Green
|
||||
16
scripts/docker-up.ps1
Normal file
16
scripts/docker-up.ps1
Normal file
@@ -0,0 +1,16 @@
|
||||
# 后台启动 Docker Compose 栈(不重启 Docker Desktop)
|
||||
$ErrorActionPreference = "Stop"
|
||||
$Root = Split-Path -Parent (Split-Path -Parent $MyInvocation.MyCommand.Path)
|
||||
Set-Location $Root
|
||||
|
||||
Write-Host "Starting services..." -ForegroundColor Cyan
|
||||
docker compose up -d
|
||||
|
||||
if ($LASTEXITCODE -ne 0) { exit $LASTEXITCODE }
|
||||
|
||||
Write-Host ""
|
||||
Write-Host "Services running:" -ForegroundColor Green
|
||||
Write-Host " Backend: http://localhost:23337"
|
||||
Write-Host " Frontend: http://localhost:23338"
|
||||
Write-Host ""
|
||||
Write-Host "Logs: .\scripts\docker-logs.ps1"
|
||||
Reference in New Issue
Block a user