mirror of
https://github.com/basketikun/infinite-canvas.git
synced 2026-07-24 15:24:06 +08:00
Compare commits
12 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 49851f2b5f | |||
| c81bb8b651 | |||
| b7d0fe10bc | |||
| d91e67a5ad | |||
| d566d5cc46 | |||
| 8fb8d0b224 | |||
| 5e1fd7a825 | |||
| 062e4569aa | |||
| b5dc9c8a87 | |||
| 75c22c2f9f | |||
| bdca6b0a5c | |||
| d4130bbb79 |
+15
-3
@@ -2,15 +2,27 @@
|
|||||||
|
|
||||||
## Unreleased
|
## Unreleased
|
||||||
|
|
||||||
|
+ [新增] 提示词来源新增 Banana Prompt Quicker,并支持添加自定义标准 JSON 来源。
|
||||||
|
+ [新增] 提示词中心新增「我的提示词」,支持本地管理、从公共库保存,并可在画布提示词库中直接使用。
|
||||||
|
+ [新增] Agent 新增统一生成任务状态查询,支持查看画布、生图工作台和视频工作台任务进度。
|
||||||
|
+ [调整] 内置提示词来源改为读取 Image Prompts 统一 JSON 数据,不再由画布端分别解析。
|
||||||
|
+ [调整] 提示词来源配置保留卡片式交互,并展示来源数量、同步状态和上次成功时间。
|
||||||
|
+ [优化] 提示词来源支持独立缓存和更新,更新失败时保留上一次成功内容。
|
||||||
|
+ [优化] Agent 对话区分用户与 AI 消息,并优化会话切换和回到最新消息的交互。
|
||||||
|
+ [优化] 画布生成完成后保留提示词输入,并改善提示词库和资产面板的交互性能。
|
||||||
|
+ [修复] 本地 Agent 完善多标签页请求隔离、结果归属、焦点回退和 Codex 会话状态同步。
|
||||||
|
+ [修复] 本地 Agent 上传的图片附件可正确创建画布图片节点并连接生成流程。
|
||||||
|
+ [修复] 画布提示词库支持跨来源搜索,插入节点时保留提示词标题。
|
||||||
|
|
||||||
## v0.9.0 - 2026-07-17
|
## v0.9.0 - 2026-07-17
|
||||||
|
|
||||||
+ [调整] 画布节点名称默认不再显示,仅在选中/悬停/编辑时出现,画布更简洁。
|
|
||||||
+ [新增] 左侧面板「资产」Tab 支持上传添加图片/视频资产、卡片悬停移除资产。
|
+ [新增] 左侧面板「资产」Tab 支持上传添加图片/视频资产、卡片悬停移除资产。
|
||||||
+ [新增] 左侧画布面板支持拖拽调整宽度、展开/收起(带动画),顶栏菜单左侧新增面板开关按钮。
|
+ [新增] 左侧画布面板支持拖拽调整宽度、展开/收起(带动画),顶栏菜单左侧新增面板开关按钮。
|
||||||
+ [新增] 顶栏菜单新增「导出当前画布」,导出为包含全部资源的压缩包。
|
+ [新增] 顶栏菜单新增「导出当前画布」,导出为包含全部资源的压缩包。
|
||||||
+ [新增] 左侧画布元素列表支持多选并批量导出选中元素为压缩包。
|
+ [新增] 左侧画布元素列表支持多选并批量导出选中元素为压缩包。
|
||||||
+ [优化] 左侧面板「画布/资产」切换改为带滑动下划线的动画,移除非图片元素图标的灰色底色。
|
|
||||||
+ [新增] 可选的网站统计分析:支持 Google Analytics 4 与百度统计。
|
+ [新增] 可选的网站统计分析:支持 Google Analytics 4 与百度统计。
|
||||||
|
+ [调整] 画布节点名称默认不再显示,仅在选中/悬停/编辑时出现,画布更简洁。
|
||||||
|
+ [优化] 左侧面板「画布/资产」切换改为带滑动下划线的动画,移除非图片元素图标的灰色底色。
|
||||||
+ [优化] 画布节点提示词面板 `@` 引用图片时,输入框内直接显示真实缩略图。
|
+ [优化] 画布节点提示词面板 `@` 引用图片时,输入框内直接显示真实缩略图。
|
||||||
+ [优化] 移除画布节点右上角的「图片1/文本1」资源角标,引用改在对话面板 `@` 直接选取。
|
+ [优化] 移除画布节点右上角的「图片1/文本1」资源角标,引用改在对话面板 `@` 直接选取。
|
||||||
+ [修复] 连接本地 Codex Agent 后,拖拽画布节点边框缩放等高频编辑导致页面崩溃。
|
+ [修复] 连接本地 Codex Agent 后,拖拽画布节点边框缩放等高频编辑导致页面崩溃。
|
||||||
@@ -18,8 +30,8 @@
|
|||||||
## v0.8.2 - 2026-07-16
|
## v0.8.2 - 2026-07-16
|
||||||
|
|
||||||
+ [新增] 图像设置新增「透明背景」开关,开启后生成无背景的透明图像。
|
+ [新增] 图像设置新增「透明背景」开关,开启后生成无背景的透明图像。
|
||||||
+ [修复] 画布节点提示词输入框补上悬停文本光标。
|
|
||||||
+ [优化] 画布节点输入区域移除灰色底色与边框、美化样式。
|
+ [优化] 画布节点输入区域移除灰色底色与边框、美化样式。
|
||||||
|
+ [修复] 画布节点提示词输入框补上悬停文本光标。
|
||||||
|
|
||||||
## v0.8.1 - 2026-07-16
|
## v0.8.1 - 2026-07-16
|
||||||
|
|
||||||
|
|||||||
@@ -13,6 +13,7 @@
|
|||||||
],
|
],
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"dev": "tsx src/index.ts",
|
"dev": "tsx src/index.ts",
|
||||||
|
"test": "tsx --test src/canvas-session.test.ts",
|
||||||
"build": "tsc -p tsconfig.json",
|
"build": "tsc -p tsconfig.json",
|
||||||
"start": "node dist/index.js",
|
"start": "node dist/index.js",
|
||||||
"prepack": "npm run build"
|
"prepack": "npm run build"
|
||||||
|
|||||||
+54
-29
@@ -11,11 +11,12 @@ import type { AgentAttachment, AgentEmit } from "./types.js";
|
|||||||
type Json = Record<string, unknown>;
|
type Json = Record<string, unknown>;
|
||||||
type AgentEvent = Json & { type: string; usage?: unknown };
|
type AgentEvent = Json & { type: string; usage?: unknown };
|
||||||
type PendingRequest = { resolve: (value: unknown) => void; reject: (error: Error) => void };
|
type PendingRequest = { resolve: (value: unknown) => void; reject: (error: Error) => void };
|
||||||
type CodexRunOptions = { threadId?: string; cwd?: string };
|
type CodexRunOptions = { threadId?: string; cwd?: string; appEmit?: AgentEmit; onStart?: () => void; onThread?: (threadId: string) => void; onTurn?: (turnId: string) => void; onFinish?: () => void };
|
||||||
type AgentHistoryMessage = { id: string; role: "user" | "assistant" | "tool" | "error"; title?: string; text: string; detail?: unknown; streamId?: string };
|
type AgentHistoryMessage = { id: string; role: "user" | "assistant" | "tool" | "error"; title?: string; text: string; detail?: unknown; streamId?: string };
|
||||||
|
|
||||||
let codexQueue: Promise<unknown> = Promise.resolve();
|
let codexQueue: Promise<unknown> = Promise.resolve();
|
||||||
let codexApp: CodexAppClient | null = null;
|
let codexApp: CodexAppClient | null = null;
|
||||||
|
let codexAppStart: Promise<CodexAppClient> | null = null;
|
||||||
let codexThreadId = "";
|
let codexThreadId = "";
|
||||||
const canvasAgentMcp = canvasAgentMcpCommand();
|
const canvasAgentMcp = canvasAgentMcpCommand();
|
||||||
const require = createRequire(import.meta.url);
|
const require = createRequire(import.meta.url);
|
||||||
@@ -30,52 +31,56 @@ export async function runCodexTurn(prompt: string, emit: AgentEmit, attachments:
|
|||||||
await codexQueue;
|
await codexQueue;
|
||||||
}
|
}
|
||||||
|
|
||||||
export function interruptCodexTurn() {
|
export function interruptCodexTurn(threadId?: string) {
|
||||||
if (!codexApp) return false;
|
if (!codexApp || (threadId && threadId !== codexThreadId)) return false;
|
||||||
return codexApp.interruptCurrentTurn();
|
return codexApp.interruptCurrentTurn();
|
||||||
}
|
}
|
||||||
|
|
||||||
async function runCodexTurnNow(prompt: string, emit: AgentEmit, attachments: AgentAttachment[], options: CodexRunOptions) {
|
async function runCodexTurnNow(prompt: string, emit: AgentEmit, attachments: AgentAttachment[], options: CodexRunOptions) {
|
||||||
let files: string[] = [];
|
let files: string[] = [];
|
||||||
try {
|
try {
|
||||||
|
options.onStart?.();
|
||||||
files = await writeAttachmentFiles(attachments);
|
files = await writeAttachmentFiles(attachments);
|
||||||
codexApp ||= await CodexAppClient.start(emit);
|
const app = await getCodexApp(options.appEmit || emit);
|
||||||
let threadId = await ensureCodexThread(codexApp, options, emit);
|
let threadId = await ensureCodexThread(app, options, emit);
|
||||||
|
options.onThread?.(threadId);
|
||||||
try {
|
try {
|
||||||
await codexApp.startTurn(threadId, prompt, files);
|
await app.startTurn(threadId, prompt, files, options.onTurn);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
if (!isRecoverableThreadError(error)) throw error;
|
if (!isRecoverableThreadError(error)) throw error;
|
||||||
emit("agent_log", { text: `Codex thread unavailable, starting a new thread: ${errorMessage(error)}` });
|
emit("agent_log", { text: `Codex thread unavailable, starting a new thread: ${errorMessage(error)}` });
|
||||||
codexThreadId = "";
|
codexThreadId = "";
|
||||||
threadId = await ensureCodexThread(codexApp, { cwd: options.cwd }, emit);
|
threadId = await ensureCodexThread(app, { cwd: options.cwd }, emit);
|
||||||
await codexApp.startTurn(threadId, prompt, files);
|
options.onThread?.(threadId);
|
||||||
|
await app.startTurn(threadId, prompt, files, options.onTurn);
|
||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
emit("agent_error", { message: errorMessage(error) });
|
emit("agent_error", { message: errorMessage(error) });
|
||||||
} finally {
|
} finally {
|
||||||
|
options.onFinish?.();
|
||||||
await Promise.all(files.map((file) => fs.unlink(file).catch(() => undefined)));
|
await Promise.all(files.map((file) => fs.unlink(file).catch(() => undefined)));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function startCodexThread(emit: AgentEmit, cwd?: string) {
|
export async function startCodexThread(emit: AgentEmit, cwd?: string) {
|
||||||
codexApp ||= await CodexAppClient.start(emit);
|
const app = await getCodexApp(emit);
|
||||||
const thread = await codexApp.startThread(cwd);
|
const thread = await app.startThread(cwd);
|
||||||
codexThreadId = String(field(thread, "id") || "");
|
codexThreadId = String(field(thread, "id") || "");
|
||||||
return thread;
|
return thread;
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function resumeCodexThread(emit: AgentEmit, threadId: string, cwd?: string) {
|
export async function resumeCodexThread(emit: AgentEmit, threadId: string, cwd?: string) {
|
||||||
codexApp ||= await CodexAppClient.start(emit);
|
const app = await getCodexApp(emit);
|
||||||
await loadCodexThread(emit, threadId, cwd, false);
|
await loadCodexThread(emit, threadId, cwd, false);
|
||||||
const thread = await codexApp.resumeThread(threadId, cwd);
|
const thread = await app.resumeThread(threadId, cwd);
|
||||||
assertThreadWorkspace(thread, cwd);
|
assertThreadWorkspace(thread, cwd);
|
||||||
codexThreadId = String(field(thread, "id") || threadId);
|
codexThreadId = String(field(thread, "id") || threadId);
|
||||||
return { thread, messages: threadMessages(thread) };
|
return { thread, messages: threadMessages(thread) };
|
||||||
}
|
}
|
||||||
|
|
||||||
export async function listCodexThreads(emit: AgentEmit, options: { cwd: string; searchTerm?: string; limit?: number }) {
|
export async function listCodexThreads(emit: AgentEmit, options: { cwd: string; searchTerm?: string; limit?: number }) {
|
||||||
codexApp ||= await CodexAppClient.start(emit);
|
const app = await getCodexApp(emit);
|
||||||
const result = await codexApp.listThreads({
|
const result = await app.listThreads({
|
||||||
limit: options.limit || 40,
|
limit: options.limit || 40,
|
||||||
sortKey: "updated_at",
|
sortKey: "updated_at",
|
||||||
sortDirection: "desc",
|
sortDirection: "desc",
|
||||||
@@ -97,9 +102,9 @@ export async function verifyCodexThreadWorkspace(emit: AgentEmit, threadId: stri
|
|||||||
}
|
}
|
||||||
|
|
||||||
export async function archiveCodexThread(emit: AgentEmit, threadId: string, cwd?: string) {
|
export async function archiveCodexThread(emit: AgentEmit, threadId: string, cwd?: string) {
|
||||||
codexApp ||= await CodexAppClient.start(emit);
|
const app = await getCodexApp(emit);
|
||||||
await loadCodexThread(emit, threadId, cwd, false);
|
await loadCodexThread(emit, threadId, cwd, false);
|
||||||
await codexApp.archiveThread(threadId);
|
await app.archiveThread(threadId);
|
||||||
}
|
}
|
||||||
|
|
||||||
export function runClaudeTurn(prompt: string, emit: AgentEmit) {
|
export function runClaudeTurn(prompt: string, emit: AgentEmit) {
|
||||||
@@ -131,7 +136,7 @@ async function ensureCodexThread(app: CodexAppClient, options: CodexRunOptions,
|
|||||||
return codexThreadId;
|
return codexThreadId;
|
||||||
}
|
}
|
||||||
|
|
||||||
function isRecoverableThreadError(error: unknown) {
|
export function isRecoverableThreadError(error: unknown) {
|
||||||
return /thread not loaded|no rollout found/i.test(errorMessage(error));
|
return /thread not loaded|no rollout found/i.test(errorMessage(error));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -192,10 +197,11 @@ class CodexAppClient {
|
|||||||
return this.request("thread/archive", { threadId });
|
return this.request("thread/archive", { threadId });
|
||||||
}
|
}
|
||||||
|
|
||||||
async startTurn(threadId: string, prompt: string, images: string[]) {
|
async startTurn(threadId: string, prompt: string, images: string[], onTurn?: (turnId: string) => void) {
|
||||||
const result = await this.request("turn/start", { threadId, input: codexInput(prompt, images), approvalPolicy: "never" });
|
const result = await this.request("turn/start", { threadId, input: codexInput(prompt, images), approvalPolicy: "never" });
|
||||||
const turnId = String(field(field(result, "turn"), "id") || "");
|
const turnId = String(field(field(result, "turn"), "id") || "");
|
||||||
if (!turnId) throw new Error("Codex app-server 没有返回 turn id");
|
if (!turnId) throw new Error("Codex app-server 没有返回 turn id");
|
||||||
|
onTurn?.(turnId);
|
||||||
const completed = this.completedTurns.get(turnId);
|
const completed = this.completedTurns.get(turnId);
|
||||||
if (this.completedTurns.has(turnId)) {
|
if (this.completedTurns.has(turnId)) {
|
||||||
this.completedTurns.delete(turnId);
|
this.completedTurns.delete(turnId);
|
||||||
@@ -267,9 +273,9 @@ class CodexAppClient {
|
|||||||
} else if (turnId) {
|
} else if (turnId) {
|
||||||
this.completedTurns.set(turnId, error ? new Error(String(field(error, "message") || "Codex turn failed")) : null);
|
this.completedTurns.set(turnId, error ? new Error(String(field(error, "message") || "Codex turn failed")) : null);
|
||||||
}
|
}
|
||||||
this.emit("agent_event", { agent: "codex", type: "stream.summary", delta_count: this.deltaCount });
|
this.emit("agent_event", { agent: "codex", type: "stream.summary", delta_count: this.deltaCount, ...codexEventScope(params) });
|
||||||
this.deltaCount = 0;
|
this.deltaCount = 0;
|
||||||
this.emit("agent_done", { agent: "codex", usage: event.usage });
|
this.emit("agent_done", { agent: "codex", usage: event.usage, ...codexEventScope(params) });
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -278,7 +284,7 @@ class CodexAppClient {
|
|||||||
const text = `${this.textByItem.get(id) || ""}${String(field(params, "delta") || "")}`;
|
const text = `${this.textByItem.get(id) || ""}${String(field(params, "delta") || "")}`;
|
||||||
this.deltaCount += 1;
|
this.deltaCount += 1;
|
||||||
this.textByItem.set(id, text);
|
this.textByItem.set(id, text);
|
||||||
this.emit("agent_event", { agent: "codex", type: "item.updated", item: { id, type: "agent_message", text } });
|
this.emit("agent_event", { agent: "codex", type: "item.updated", item: { id, type: "agent_message", text }, ...codexEventScope(params) });
|
||||||
}
|
}
|
||||||
|
|
||||||
private answerServerRequest(message: Json) {
|
private answerServerRequest(message: Json) {
|
||||||
@@ -321,23 +327,41 @@ function codexInput(prompt: string, images: string[]) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
function normalizeCodexNotification(method: string, params: Json): AgentEvent | null {
|
function normalizeCodexNotification(method: string, params: Json): AgentEvent | null {
|
||||||
if (method === "thread/started") return { type: "thread.started", thread_id: field(field(params, "thread"), "id") };
|
const scope = codexEventScope(params);
|
||||||
if (method === "turn/started") return { type: "turn.started" };
|
if (method === "thread/started") return { type: "thread.started", ...scope };
|
||||||
if (method === "turn/completed") return { type: "turn.completed", usage: null };
|
if (method === "turn/started") return { type: "turn.started", ...scope };
|
||||||
if (method === "item/started") return { type: "item.started", item: normalizeItem(field(params, "item")) };
|
if (method === "turn/completed") return { type: "turn.completed", usage: null, ...scope };
|
||||||
if (method === "item/completed") return { type: "item.completed", item: normalizeItem(field(params, "item")) };
|
if (method === "item/started") return { type: "item.started", item: normalizeItem(field(params, "item")), ...scope };
|
||||||
if (method === "error") return { type: "error", message: field(params, "message") };
|
if (method === "item/completed") return { type: "item.completed", item: normalizeItem(field(params, "item")), ...scope };
|
||||||
|
if (method === "error") return { type: "error", message: field(params, "message"), ...scope };
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function codexEventScope(params: Json) {
|
||||||
|
const threadId = String(field(params, "threadId") || field(field(params, "thread"), "id") || "");
|
||||||
|
const turnId = String(field(params, "turnId") || field(field(params, "turn"), "id") || "");
|
||||||
|
return { ...(threadId ? { thread_id: threadId } : {}), ...(turnId ? { turn_id: turnId } : {}) };
|
||||||
|
}
|
||||||
|
|
||||||
async function loadCodexThread(emit: AgentEmit, threadId: string, cwd: string | undefined, includeTurns: boolean) {
|
async function loadCodexThread(emit: AgentEmit, threadId: string, cwd: string | undefined, includeTurns: boolean) {
|
||||||
codexApp ||= await CodexAppClient.start(emit);
|
const app = await getCodexApp(emit);
|
||||||
const result = await codexApp.readThread(threadId, includeTurns);
|
const result = await app.readThread(threadId, includeTurns);
|
||||||
const thread = field(result, "thread") || {};
|
const thread = field(result, "thread") || {};
|
||||||
assertThreadWorkspace(thread, cwd);
|
assertThreadWorkspace(thread, cwd);
|
||||||
return thread;
|
return thread;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function getCodexApp(emit: AgentEmit) {
|
||||||
|
if (codexApp) return codexApp;
|
||||||
|
codexAppStart ||= CodexAppClient.start(emit);
|
||||||
|
try {
|
||||||
|
codexApp = await codexAppStart;
|
||||||
|
return codexApp;
|
||||||
|
} finally {
|
||||||
|
codexAppStart = null;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
function assertThreadWorkspace(thread: unknown, cwd?: string) {
|
function assertThreadWorkspace(thread: unknown, cwd?: string) {
|
||||||
if (!cwd || threadInWorkspace(thread, cwd)) return;
|
if (!cwd || threadInWorkspace(thread, cwd)) return;
|
||||||
throw new Error("该 Codex 会话不属于当前画布工作空间");
|
throw new Error("该 Codex 会话不属于当前画布工作空间");
|
||||||
@@ -458,6 +482,7 @@ function toolName(name: string) {
|
|||||||
if (name === "canvas_get_state") return "读取画布";
|
if (name === "canvas_get_state") return "读取画布";
|
||||||
if (name === "canvas_get_selection") return "读取选区";
|
if (name === "canvas_get_selection") return "读取选区";
|
||||||
if (name === "canvas_export_snapshot") return "导出快照";
|
if (name === "canvas_export_snapshot") return "导出快照";
|
||||||
|
if (name === "canvas_create_attachment_nodes") return "添加附件图片";
|
||||||
if (name === "canvas_create_text_node") return "创建文本";
|
if (name === "canvas_create_text_node") return "创建文本";
|
||||||
if (name === "canvas_create_image_prompt_flow") return "创建生图流程";
|
if (name === "canvas_create_image_prompt_flow") return "创建生图流程";
|
||||||
if (name === "canvas_create_generation_flow") return "创建生成流程";
|
if (name === "canvas_create_generation_flow") return "创建生成流程";
|
||||||
|
|||||||
@@ -0,0 +1,289 @@
|
|||||||
|
import { EventEmitter } from "node:events";
|
||||||
|
import type { ServerResponse } from "node:http";
|
||||||
|
import assert from "node:assert/strict";
|
||||||
|
import test from "node:test";
|
||||||
|
|
||||||
|
import { CanvasSession } from "./canvas-session.js";
|
||||||
|
|
||||||
|
test("MCP 读取当前激活网页的画布", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.updateState(snapshot("canvas-first"), "first");
|
||||||
|
session.updateState(snapshot("canvas-second"), "second");
|
||||||
|
|
||||||
|
session.activateClient("first");
|
||||||
|
assert.equal(field(await session.callTool("canvas_get_state", {}), "projectId"), "canvas-first");
|
||||||
|
|
||||||
|
session.activateClient("second");
|
||||||
|
assert.equal(field(await session.callTool("canvas_get_state", {}), "projectId"), "canvas-second");
|
||||||
|
});
|
||||||
|
|
||||||
|
test("画布写操作只发送给当前激活网页", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.updateState(snapshot("canvas-first"), "first");
|
||||||
|
session.updateState(snapshot("canvas-second"), "second");
|
||||||
|
session.activateClient("second");
|
||||||
|
|
||||||
|
const result = session.callTool("canvas_create_text_node", { text: "只写入第二个画布" });
|
||||||
|
const call = second.event("tool_call");
|
||||||
|
assert.equal(first.event("tool_call"), undefined);
|
||||||
|
assert.equal(field(call, "name"), "canvas_apply_ops");
|
||||||
|
session.resolveResult("second", { requestId: String(field(call, "requestId")), result: { ok: true } });
|
||||||
|
assert.deepEqual(await result, { ok: true });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("当前 turn 的图片附件可在发起标签页画布创建图片节点", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
t.after(() => first.close());
|
||||||
|
const dataUrl = "data:image/png;base64,aW1hZ2U=";
|
||||||
|
session.setTurnAttachments("first", [{ id: "attachment-1", name: "商品.png", type: "image/png", size: 5, width: 1200, height: 600, dataUrl }]);
|
||||||
|
session.bindClient("first");
|
||||||
|
|
||||||
|
const result = session.callTool("canvas_create_attachment_nodes", { attachmentIds: ["attachment-1"], x: 100, y: 200 });
|
||||||
|
const call = first.event("tool_call");
|
||||||
|
const input = field(call, "input") as Record<string, unknown>;
|
||||||
|
const nodes = input.nodes as Array<Record<string, unknown>>;
|
||||||
|
assert.equal(field(call, "name"), "canvas_create_attachment_nodes");
|
||||||
|
assert.equal(nodes.length, 1);
|
||||||
|
assert.equal(nodes[0].attachmentId, "attachment-1");
|
||||||
|
assert.equal(nodes[0].title, "商品.png");
|
||||||
|
assert.deepEqual(nodes[0].position, { x: 100, y: 200 });
|
||||||
|
assert.equal(nodes[0].width, 640);
|
||||||
|
assert.equal(nodes[0].height, 320);
|
||||||
|
assert.equal("dataUrl" in nodes[0], false);
|
||||||
|
assert.equal(session.getTurnAttachment("first", "attachment-1").dataUrl, dataUrl);
|
||||||
|
|
||||||
|
session.resolveResult("first", { requestId: String(field(call, "requestId")), result: { ok: true } });
|
||||||
|
const created = (await result) as { nodes: Array<{ id: string; attachmentId: string; title: string }> };
|
||||||
|
assert.equal(created.nodes[0].id, nodes[0].id);
|
||||||
|
assert.equal(created.nodes[0].attachmentId, "attachment-1");
|
||||||
|
session.clearTurnAttachments("first");
|
||||||
|
assert.throws(() => session.getTurnAttachment("first", "attachment-1"), /找不到/);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("图片附件只允许发起 turn 的标签页读取和落入画布", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.setTurnAttachments("first", [{ id: "attachment-1", name: "商品.png", type: "image/png", dataUrl: "data:image/png;base64,aW1hZ2U=" }]);
|
||||||
|
session.bindClient("second");
|
||||||
|
|
||||||
|
await assert.rejects(session.callTool("canvas_create_attachment_nodes", { attachmentIds: ["attachment-1"] }), /发起标签页/);
|
||||||
|
assert.throws(() => session.getTurnAttachment("second", "attachment-1"), /发起标签页/);
|
||||||
|
assert.equal(first.event("tool_call"), undefined);
|
||||||
|
assert.equal(second.event("tool_call"), undefined);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("tool result is accepted only from the request client", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.activateClient("first");
|
||||||
|
|
||||||
|
const result = session.callTool("canvas_create_text_node", { text: "first only" });
|
||||||
|
const call = first.event("tool_call");
|
||||||
|
const requestId = String(field(call, "requestId"));
|
||||||
|
|
||||||
|
assert.equal(session.resolveResult("second", { requestId, result: { client: "second" } }), false);
|
||||||
|
assert.equal(session.resolveResult("first", { requestId, result: { client: "first" } }), true);
|
||||||
|
assert.deepEqual(await result, { client: "first" });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("生成状态查询由当前激活网页返回", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.activateClient("second");
|
||||||
|
|
||||||
|
const result = session.callTool("generation_get_status", { scope: "all" });
|
||||||
|
const call = second.event("tool_call");
|
||||||
|
assert.equal(first.event("tool_call"), undefined);
|
||||||
|
assert.equal(field(call, "name"), "generation_get_status");
|
||||||
|
session.resolveResult("second", { requestId: String(field(call, "requestId")), result: { total: 1, tasks: [{ id: "image-1", status: "running" }] } });
|
||||||
|
assert.deepEqual(await result, { total: 1, tasks: [{ id: "image-1", status: "running" }] });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("活动网页关闭后回退到仍连接的画布", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.updateState(snapshot("canvas-first"), "first");
|
||||||
|
session.updateState(snapshot("canvas-second"), "second");
|
||||||
|
session.activateClient("second");
|
||||||
|
second.close();
|
||||||
|
|
||||||
|
assert.equal(field(await session.callTool("canvas_get_state", {}), "projectId"), "canvas-first");
|
||||||
|
});
|
||||||
|
|
||||||
|
test("closing the active client falls back to the most recently focused client", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
const third = connect(session, "third");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
third.close();
|
||||||
|
});
|
||||||
|
session.updateState(snapshot("canvas-first"), "first");
|
||||||
|
session.updateState(snapshot("canvas-second"), "second");
|
||||||
|
session.updateState(snapshot("canvas-third"), "third");
|
||||||
|
session.activateClient("third");
|
||||||
|
session.activateClient("second");
|
||||||
|
second.close();
|
||||||
|
|
||||||
|
assert.equal(field(await session.callTool("canvas_get_state", {}), "projectId"), "canvas-third");
|
||||||
|
});
|
||||||
|
|
||||||
|
test("closing a client rejects its pending tool requests", async () => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const result = session.callTool("canvas_create_text_node", { text: "pending" });
|
||||||
|
const call = first.event("tool_call");
|
||||||
|
const requestId = String(field(call, "requestId"));
|
||||||
|
first.close();
|
||||||
|
|
||||||
|
const outcome = await Promise.race([
|
||||||
|
result.then(() => "resolved", (error) => error instanceof Error ? error.message : String(error)),
|
||||||
|
new Promise<string>((resolve) => setTimeout(() => resolve("pending"), 20)),
|
||||||
|
]);
|
||||||
|
if (outcome === "pending") session.resolveResult("first", { requestId, result: null });
|
||||||
|
assert.match(outcome, /断开/);
|
||||||
|
});
|
||||||
|
|
||||||
|
test("shared thread events are broadcast with the active thread id", (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
|
||||||
|
session.emitThread("workspace_changed", "thread-2", { activeThreadId: "thread-2" });
|
||||||
|
|
||||||
|
assert.deepEqual(first.event("workspace_changed"), { activeThreadId: "thread-2", threadId: "thread-2" });
|
||||||
|
assert.deepEqual(second.event("workspace_changed"), { activeThreadId: "thread-2", threadId: "thread-2" });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("new clients receive the current Codex state and later updates", (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
session.setCodexState({ busy: true, threadId: "thread-2", turnId: "turn-1" });
|
||||||
|
const client = connect(session, "first");
|
||||||
|
t.after(() => client.close());
|
||||||
|
|
||||||
|
assert.deepEqual(field(client.event("hello"), "codex"), { busy: true, threadId: "thread-2", turnId: "turn-1" });
|
||||||
|
|
||||||
|
session.setCodexState({ busy: false });
|
||||||
|
assert.deepEqual(client.event("codex_state"), { busy: false, threadId: "thread-2", turnId: "turn-1" });
|
||||||
|
});
|
||||||
|
|
||||||
|
test("a bound client remains the tool target while focus changes", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.updateState(snapshot("canvas-first"), "first");
|
||||||
|
session.updateState(snapshot("canvas-second"), "second");
|
||||||
|
session.bindClient("first");
|
||||||
|
session.activateClient("second");
|
||||||
|
|
||||||
|
assert.equal(field(await session.callTool("canvas_get_state", {}), "projectId"), "canvas-first");
|
||||||
|
const result = session.callTool("canvas_create_text_node", { text: "bound" });
|
||||||
|
const call = first.event("tool_call");
|
||||||
|
assert.equal(second.event("tool_call"), undefined);
|
||||||
|
session.resolveResult("first", { requestId: String(field(call, "requestId")), result: { ok: true } });
|
||||||
|
assert.deepEqual(await result, { ok: true });
|
||||||
|
|
||||||
|
session.releaseClient("first");
|
||||||
|
assert.equal(field(await session.callTool("canvas_get_state", {}), "projectId"), "canvas-second");
|
||||||
|
});
|
||||||
|
|
||||||
|
test("closing the bound client falls back to the active client", async (t) => {
|
||||||
|
const session = new CanvasSession();
|
||||||
|
const first = connect(session, "first");
|
||||||
|
const second = connect(session, "second");
|
||||||
|
t.after(() => {
|
||||||
|
first.close();
|
||||||
|
second.close();
|
||||||
|
});
|
||||||
|
session.updateState(snapshot("canvas-first"), "first");
|
||||||
|
session.updateState(snapshot("canvas-second"), "second");
|
||||||
|
session.bindClient("first");
|
||||||
|
session.activateClient("second");
|
||||||
|
first.close();
|
||||||
|
|
||||||
|
assert.equal(field(await session.callTool("canvas_get_state", {}), "projectId"), "canvas-second");
|
||||||
|
const result = session.callTool("canvas_create_text_node", { text: "fallback" });
|
||||||
|
const call = second.event("tool_call");
|
||||||
|
session.resolveResult("second", { requestId: String(field(call, "requestId")), result: { ok: true } });
|
||||||
|
assert.deepEqual(await result, { ok: true });
|
||||||
|
});
|
||||||
|
|
||||||
|
function connect(session: CanvasSession, clientId: string) {
|
||||||
|
const response = new FakeSseResponse();
|
||||||
|
session.openEvents(new URL(`http://127.0.0.1/events?clientId=${clientId}`), response as unknown as ServerResponse);
|
||||||
|
return response;
|
||||||
|
}
|
||||||
|
|
||||||
|
function snapshot(projectId: string) {
|
||||||
|
return { projectId, title: projectId, nodes: [], connections: [], selectedNodeIds: [], viewport: { x: 0, y: 0, k: 1 } };
|
||||||
|
}
|
||||||
|
|
||||||
|
function field(value: unknown, key: string) {
|
||||||
|
return value && typeof value === "object" ? (value as Record<string, unknown>)[key] : undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
class FakeSseResponse extends EventEmitter {
|
||||||
|
private chunks: string[] = [];
|
||||||
|
|
||||||
|
writeHead() {
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
write(chunk: string) {
|
||||||
|
this.chunks.push(chunk);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
event(type: string) {
|
||||||
|
const chunk = this.chunks.find((item) => item.startsWith(`event: ${type}\n`));
|
||||||
|
const data = chunk?.split("\n").find((line) => line.startsWith("data: "))?.slice(6);
|
||||||
|
return data ? (JSON.parse(data) as unknown) : undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
close() {
|
||||||
|
this.emit("close");
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -3,9 +3,11 @@ import type { ServerResponse } from "node:http";
|
|||||||
|
|
||||||
import { type ToolName } from "./schemas.js";
|
import { type ToolName } from "./schemas.js";
|
||||||
import { compactCanvasState, compactNode, isToolName, nextCanvasX, parseToolInput } from "./tools.js";
|
import { compactCanvasState, compactNode, isToolName, nextCanvasX, parseToolInput } from "./tools.js";
|
||||||
import type { CanvasNode, CanvasNodeType, CanvasSnapshot } from "./types.js";
|
import type { AgentAttachment, CanvasNode, CanvasNodeType, CanvasSnapshot } from "./types.js";
|
||||||
|
|
||||||
type PendingRequest = { resolve: (value: unknown) => void; reject: (error: Error) => void };
|
type PendingRequest = { clientId: string; resolve: (value: unknown) => void; reject: (error: Error) => void };
|
||||||
|
type TurnAttachment = { clientId: string; id: string; name: string; type: string; size: number; width: number; height: number; dataUrl: string };
|
||||||
|
export type CodexState = { busy: boolean; threadId: string; turnId: string };
|
||||||
|
|
||||||
const SITE_TOOLS = new Set<ToolName>([
|
const SITE_TOOLS = new Set<ToolName>([
|
||||||
"site_navigate",
|
"site_navigate",
|
||||||
@@ -17,46 +19,141 @@ const SITE_TOOLS = new Set<ToolName>([
|
|||||||
"prompts_search",
|
"prompts_search",
|
||||||
"assets_list",
|
"assets_list",
|
||||||
"assets_add",
|
"assets_add",
|
||||||
|
"generation_get_status",
|
||||||
]);
|
]);
|
||||||
|
|
||||||
export class CanvasSession {
|
export class CanvasSession {
|
||||||
private clients = new Map<string, ServerResponse>();
|
private clients = new Map<string, ServerResponse>();
|
||||||
|
private clientFocusOrder = new Map<string, number>();
|
||||||
private pending = new Map<string, PendingRequest>();
|
private pending = new Map<string, PendingRequest>();
|
||||||
private canvasState: CanvasSnapshot | null = null;
|
private canvasStates = new Map<string, CanvasSnapshot>();
|
||||||
|
private turnAttachments = new Map<string, TurnAttachment>();
|
||||||
|
private activeClientId = "";
|
||||||
|
private boundClientId = "";
|
||||||
|
private focusSequence = 0;
|
||||||
|
private codexState: CodexState = { busy: false, threadId: "", turnId: "" };
|
||||||
|
|
||||||
|
private get canvasState() {
|
||||||
|
return this.canvasStates.get(this.targetClientId) || null;
|
||||||
|
}
|
||||||
|
|
||||||
|
private get targetClientId() {
|
||||||
|
return this.boundClientId || this.activeClientId;
|
||||||
|
}
|
||||||
|
|
||||||
health() {
|
health() {
|
||||||
return { ok: true, hasCanvas: Boolean(this.canvasState), clients: this.clients.size };
|
return { ok: true, hasCanvas: Boolean(this.canvasState), clients: this.clients.size, codexBusy: this.codexState.busy };
|
||||||
|
}
|
||||||
|
|
||||||
|
get codexBusy() {
|
||||||
|
return this.codexState.busy;
|
||||||
|
}
|
||||||
|
|
||||||
|
setCodexState(patch: Partial<CodexState>) {
|
||||||
|
this.codexState = { ...this.codexState, ...patch };
|
||||||
|
this.emitAll("codex_state", this.codexState);
|
||||||
}
|
}
|
||||||
|
|
||||||
openEvents(url: URL, res: ServerResponse) {
|
openEvents(url: URL, res: ServerResponse) {
|
||||||
const clientId = url.searchParams.get("clientId") || crypto.randomUUID();
|
const clientId = url.searchParams.get("clientId") || crypto.randomUUID();
|
||||||
const statusOnly = url.searchParams.get("role") === "status";
|
const statusOnly = url.searchParams.get("role") === "status";
|
||||||
res.writeHead(200, { "Content-Type": "text/event-stream", "Cache-Control": "no-cache", Connection: "keep-alive" });
|
res.writeHead(200, { "Content-Type": "text/event-stream", "Cache-Control": "no-cache", Connection: "keep-alive" });
|
||||||
if (!statusOnly) this.clients.set(clientId, res);
|
if (!statusOnly) {
|
||||||
sendEvent(res, "hello", { ok: true, clientId });
|
this.clients.set(clientId, res);
|
||||||
|
if (!this.clientFocusOrder.has(clientId)) this.clientFocusOrder.set(clientId, 0);
|
||||||
|
if (!this.activeClientId) {
|
||||||
|
this.activeClientId = clientId;
|
||||||
|
this.clientFocusOrder.set(clientId, ++this.focusSequence);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
sendEvent(res, "hello", { ok: true, clientId, codex: this.codexState });
|
||||||
const timer = setInterval(() => sendEvent(res, "ping", { time: Date.now() }), 15000);
|
const timer = setInterval(() => sendEvent(res, "ping", { time: Date.now() }), 15000);
|
||||||
res.on("close", () => {
|
res.on("close", () => {
|
||||||
clearInterval(timer);
|
clearInterval(timer);
|
||||||
if (!statusOnly) this.clients.delete(clientId);
|
if (statusOnly || this.clients.get(clientId) !== res) return;
|
||||||
if (this.canvasState?.clientId === clientId) this.canvasState = null;
|
this.clients.delete(clientId);
|
||||||
|
this.clientFocusOrder.delete(clientId);
|
||||||
|
this.canvasStates.delete(clientId);
|
||||||
|
if (this.boundClientId === clientId) this.boundClientId = "";
|
||||||
|
this.pending.forEach((item, requestId) => {
|
||||||
|
if (item.clientId !== clientId) return;
|
||||||
|
this.pending.delete(requestId);
|
||||||
|
item.reject(new Error("请求页面已断开"));
|
||||||
|
});
|
||||||
|
if (this.activeClientId === clientId) this.activeClientId = [...this.clients.keys()].sort((a, b) => (this.clientFocusOrder.get(b) || 0) - (this.clientFocusOrder.get(a) || 0))[0] || "";
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
updateState(body: unknown, clientId?: string) {
|
updateState(body: unknown, clientId?: string) {
|
||||||
this.canvasState = { ...((body && typeof body === "object" && !Array.isArray(body) ? body : {}) as Record<string, unknown>), clientId } as CanvasSnapshot;
|
const targetClientId = clientId || this.activeClientId;
|
||||||
|
if (!targetClientId) return;
|
||||||
|
this.canvasStates.set(targetClientId, { ...((body && typeof body === "object" && !Array.isArray(body) ? body : {}) as Record<string, unknown>), clientId: targetClientId } as CanvasSnapshot);
|
||||||
}
|
}
|
||||||
|
|
||||||
resolveResult(body: { requestId?: string; error?: string; result?: unknown }) {
|
activateClient(clientId: string) {
|
||||||
|
if (!this.clients.has(clientId)) throw new Error("当前网页未连接");
|
||||||
|
this.activeClientId = clientId;
|
||||||
|
this.clientFocusOrder.set(clientId, ++this.focusSequence);
|
||||||
|
}
|
||||||
|
|
||||||
|
bindClient(clientId: string) {
|
||||||
|
if (!this.clients.has(clientId)) throw new Error("当前网页未连接");
|
||||||
|
this.boundClientId = clientId;
|
||||||
|
}
|
||||||
|
|
||||||
|
releaseClient(clientId: string) {
|
||||||
|
if (this.boundClientId === clientId) this.boundClientId = "";
|
||||||
|
}
|
||||||
|
|
||||||
|
setTurnAttachments(clientId: string, attachments: AgentAttachment[]) {
|
||||||
|
this.turnAttachments.clear();
|
||||||
|
return attachments.flatMap((item, index) => {
|
||||||
|
if (!item.dataUrl?.startsWith("data:image/")) return [];
|
||||||
|
const id = item.id?.trim() || `attachment-${crypto.randomUUID()}`;
|
||||||
|
const attachment: TurnAttachment = {
|
||||||
|
clientId,
|
||||||
|
id,
|
||||||
|
name: item.name?.trim() || `图片 ${index + 1}`,
|
||||||
|
type: item.type?.startsWith("image/") ? item.type : item.dataUrl.match(/^data:([^;]+)/)?.[1] || "image/png",
|
||||||
|
size: positiveNumber(item.size, 0),
|
||||||
|
width: positiveNumber(item.width, 1024),
|
||||||
|
height: positiveNumber(item.height, 1024),
|
||||||
|
dataUrl: item.dataUrl,
|
||||||
|
};
|
||||||
|
this.turnAttachments.set(id, attachment);
|
||||||
|
return [{ id, name: attachment.name, type: attachment.type, size: attachment.size, width: attachment.width, height: attachment.height }];
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
clearTurnAttachments(clientId?: string) {
|
||||||
|
this.turnAttachments.forEach((item, id) => {
|
||||||
|
if (!clientId || item.clientId === clientId) this.turnAttachments.delete(id);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
getTurnAttachment(clientId: string, attachmentId: string) {
|
||||||
|
const attachment = this.turnAttachments.get(attachmentId);
|
||||||
|
if (!attachment) throw new Error(`找不到本轮图片附件:${attachmentId}`);
|
||||||
|
if (attachment.clientId !== clientId) throw new Error("图片附件不属于当前 turn 的发起标签页");
|
||||||
|
return attachment;
|
||||||
|
}
|
||||||
|
|
||||||
|
resolveResult(clientId: string, body: { requestId?: string; error?: string; result?: unknown }) {
|
||||||
const item = body.requestId ? this.pending.get(body.requestId) : null;
|
const item = body.requestId ? this.pending.get(body.requestId) : null;
|
||||||
if (!item || !body.requestId) return;
|
if (!item || !body.requestId || item.clientId !== clientId) return false;
|
||||||
this.pending.delete(body.requestId);
|
this.pending.delete(body.requestId);
|
||||||
body.error ? item.reject(new Error(body.error)) : item.resolve(body.result);
|
body.error ? item.reject(new Error(body.error)) : item.resolve(body.result);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
emitAll(type: string, payload: unknown) {
|
emitAll(type: string, payload: unknown) {
|
||||||
this.clients.forEach((client) => sendEvent(client, type, payload));
|
this.clients.forEach((client) => sendEvent(client, type, payload));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
emitThread(type: string, threadId: string, payload: Record<string, unknown> = {}) {
|
||||||
|
this.emitAll(type, { ...payload, threadId });
|
||||||
|
}
|
||||||
|
|
||||||
async callTool(name: unknown, rawInput: unknown) {
|
async callTool(name: unknown, rawInput: unknown) {
|
||||||
if (!isToolName(name)) throw new Error(`未知工具:${String(name)}`);
|
if (!isToolName(name)) throw new Error(`未知工具:${String(name)}`);
|
||||||
let tool: ToolName = name;
|
let tool: ToolName = name;
|
||||||
@@ -72,6 +169,7 @@ export class CanvasSession {
|
|||||||
const ids = new Set(this.canvasState?.selectedNodeIds || []);
|
const ids = new Set(this.canvasState?.selectedNodeIds || []);
|
||||||
return { nodes: (this.canvasState?.nodes || []).filter((node) => ids.has(node.id)).map(compactNode) };
|
return { nodes: (this.canvasState?.nodes || []).filter((node) => ids.has(node.id)).map(compactNode) };
|
||||||
}
|
}
|
||||||
|
if (tool === "canvas_create_attachment_nodes") return await this.createAttachmentNodes(input as { attachmentIds: string[]; x?: number; y?: number; gap?: number; direction?: "row" | "column" });
|
||||||
if (tool === "canvas_create_node") {
|
if (tool === "canvas_create_node") {
|
||||||
const data = input as { nodeType: CanvasNodeType; title?: string; x?: number; y?: number; width?: number; height?: number; metadata?: Record<string, unknown> };
|
const data = input as { nodeType: CanvasNodeType; title?: string; x?: number; y?: number; width?: number; height?: number; metadata?: Record<string, unknown> };
|
||||||
input = { ops: [{ type: "add_node", nodeType: data.nodeType, title: data.title, position: { x: data.x ?? nextCanvasX(this.canvasState), y: data.y ?? 0 }, width: data.width, height: data.height, metadata: data.metadata }] };
|
input = { ops: [{ type: "add_node", nodeType: data.nodeType, title: data.title, position: { x: data.x ?? nextCanvasX(this.canvasState), y: data.y ?? 0 }, width: data.width, height: data.height, metadata: data.metadata }] };
|
||||||
@@ -166,9 +264,36 @@ export class CanvasSession {
|
|||||||
return await this.requestCanvasTool(tool, input);
|
return await this.requestCanvasTool(tool, input);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private async createAttachmentNodes(input: { attachmentIds: string[]; x?: number; y?: number; gap?: number; direction?: "row" | "column" }) {
|
||||||
|
const clientId = this.targetClientId;
|
||||||
|
if (!this.clients.has(clientId)) throw new Error("当前没有已连接画布");
|
||||||
|
const attachments = input.attachmentIds.map((id) => this.getTurnAttachment(clientId, id));
|
||||||
|
const x = Number(input.x ?? nextCanvasX(this.canvasState));
|
||||||
|
const y = Number(input.y ?? 0);
|
||||||
|
const gap = Number(input.gap ?? 40);
|
||||||
|
const direction = input.direction || "row";
|
||||||
|
let offset = 0;
|
||||||
|
const nodes = attachments.map((attachment) => {
|
||||||
|
const size = fitAttachmentNodeSize(attachment.width, attachment.height);
|
||||||
|
const node = {
|
||||||
|
id: `image-${crypto.randomUUID()}`,
|
||||||
|
attachmentId: attachment.id,
|
||||||
|
title: attachment.name,
|
||||||
|
position: { x: direction === "row" ? x + offset : x, y: direction === "column" ? y + offset : y },
|
||||||
|
width: size.width,
|
||||||
|
height: size.height,
|
||||||
|
};
|
||||||
|
offset += (direction === "row" ? size.width : size.height) + gap;
|
||||||
|
return node;
|
||||||
|
});
|
||||||
|
await this.requestCanvasTool("canvas_create_attachment_nodes", { nodes });
|
||||||
|
return { nodes: nodes.map(({ id, attachmentId, title }) => ({ id, attachmentId, title })) };
|
||||||
|
}
|
||||||
|
|
||||||
private async requestCanvasTool(name: ToolName, input: Record<string, unknown>) {
|
private async requestCanvasTool(name: ToolName, input: Record<string, unknown>) {
|
||||||
const requestId = crypto.randomUUID();
|
const requestId = crypto.randomUUID();
|
||||||
const client = this.clients.get(this.canvasState?.clientId || "") || this.clients.values().next().value;
|
const clientId = this.targetClientId;
|
||||||
|
const client = this.clients.get(clientId);
|
||||||
if (!client) throw new Error("当前没有已连接画布");
|
if (!client) throw new Error("当前没有已连接画布");
|
||||||
sendEvent(client, "tool_call", { requestId, name, input });
|
sendEvent(client, "tool_call", { requestId, name, input });
|
||||||
return await new Promise((resolve, reject) => {
|
return await new Promise((resolve, reject) => {
|
||||||
@@ -176,7 +301,7 @@ export class CanvasSession {
|
|||||||
this.pending.delete(requestId);
|
this.pending.delete(requestId);
|
||||||
reject(new Error("画布操作超时"));
|
reject(new Error("画布操作超时"));
|
||||||
}, 30000);
|
}, 30000);
|
||||||
this.pending.set(requestId, { resolve: (value) => (clearTimeout(timer), resolve(value)), reject: (error) => (clearTimeout(timer), reject(error)) });
|
this.pending.set(requestId, { clientId, resolve: (value) => (clearTimeout(timer), resolve(value)), reject: (error) => (clearTimeout(timer), reject(error)) });
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -263,3 +388,13 @@ function findNode(state: CanvasSnapshot | null, id: string): CanvasNode | undefi
|
|||||||
function cleanRecord(value: Record<string, unknown>) {
|
function cleanRecord(value: Record<string, unknown>) {
|
||||||
return Object.fromEntries(Object.entries(value).filter(([, item]) => item !== undefined && item !== ""));
|
return Object.fromEntries(Object.entries(value).filter(([, item]) => item !== undefined && item !== ""));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function positiveNumber(value: unknown, fallback: number) {
|
||||||
|
const number = Number(value);
|
||||||
|
return Number.isFinite(number) && number > 0 ? number : fallback;
|
||||||
|
}
|
||||||
|
|
||||||
|
function fitAttachmentNodeSize(width: number, height: number) {
|
||||||
|
const scale = Math.min(1, 640 / width, 640 / height);
|
||||||
|
return { width: width * scale, height: height * scale };
|
||||||
|
}
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ export const DEFAULT_PORT = 17371;
|
|||||||
export const CONFIG_DIR = path.join(os.homedir(), ".infinite-canvas");
|
export const CONFIG_DIR = path.join(os.homedir(), ".infinite-canvas");
|
||||||
export const CONFIG_FILE = path.join(CONFIG_DIR, "canvas-agent.json");
|
export const CONFIG_FILE = path.join(CONFIG_DIR, "canvas-agent.json");
|
||||||
export const VERSION = readPackageVersion();
|
export const VERSION = readPackageVersion();
|
||||||
export const AGENT_PROMPT = "你正在帮助用户操作 Infinite Canvas 网站。切换网站页面用 site_navigate,可跳 / (首页)、/canvas (我的画布)、/canvas/:id (指定画布)、/image、/video、/prompts、/assets、/config。需要改动画布时优先使用已配置的 infinite-canvas MCP 工具:先 canvas_get_state 读取当前画布,再根据任务使用 canvas_create_text_node、canvas_generate_text、canvas_generate_image、canvas_generate_video、canvas_generate_audio、canvas_create_generation_flow、canvas_create_config_node、canvas_run_generation、canvas_update_node、canvas_connect_nodes 等通用工具;复杂批量改动再用 canvas_apply_ops,删除连线可用 delete_connections。若当前不在画布页,画布工具会报错,需先用 site_navigate 打开画布。想了解或打开用户已有画布,用 canvas_list_projects 获取画布清单和 id,再用 site_navigate 跳 /canvas/:id 打开。生图工作台可用 workbench_image_get_config 看可选项、workbench_image_generate 填提示词并生成;视频创作台对应 workbench_video_get_config 与 workbench_video_generate;用 prompts_search 分页搜索提示词库;用 assets_list 查看「我的素材」、assets_add 新增文本或图片素材。需要生成内容时直接调用对应生成工具,不要绑定特定业务场景。不要模拟鼠标点击,不要要求用户手动复制 JSON。";
|
export const AGENT_PROMPT = "你正在帮助用户操作 Infinite Canvas 网站。切换网站页面用 site_navigate,可跳 / (首页)、/canvas (我的画布)、/canvas/:id (指定画布)、/image、/video、/prompts、/assets、/config。需要改动画布时优先使用已配置的 infinite-canvas MCP 工具:先 canvas_get_state 读取当前画布,再根据任务使用 canvas_create_text_node、canvas_generate_text、canvas_generate_image、canvas_generate_video、canvas_generate_audio、canvas_create_generation_flow、canvas_create_config_node、canvas_run_generation、canvas_update_node、canvas_connect_nodes 等通用工具;复杂批量改动再用 canvas_apply_ops,删除连线可用 delete_connections。本轮若有用户上传的图片附件,会同时给出 attachmentId;用户要求把附件放入画布或作为生成参考图时,必须先用 canvas_create_attachment_nodes 创建真实图片节点,再把返回的节点 ID 传给 canvas_create_generation_flow.referenceNodeIds,不要创建空图片占位节点。若当前不在画布页,画布工具会报错,需先用 site_navigate 打开画布。想了解或打开用户已有画布,用 canvas_list_projects 获取画布清单和 id,再用 site_navigate 跳 /canvas/:id 打开。生图工作台可用 workbench_image_get_config 看可选项、workbench_image_generate 填提示词并生成;视频创作台对应 workbench_video_get_config 与 workbench_video_generate;用 prompts_search 分页搜索提示词库;用 assets_list 查看「我的素材」、assets_add 新增文本或图片素材。需要生成内容时直接调用对应生成工具,不要绑定特定业务场景。不要模拟鼠标点击,不要要求用户手动复制 JSON。";
|
||||||
|
|
||||||
export type SiteWorkspaceConfig = { workspacePath: string; activeThreadId?: string; pinnedThreadIds?: string[] };
|
export type SiteWorkspaceConfig = { workspacePath: string; activeThreadId?: string; pinnedThreadIds?: string[] };
|
||||||
export type CanvasAgentConfig = { url: string; token: string; origins?: string[]; workspace?: SiteWorkspaceConfig };
|
export type CanvasAgentConfig = { url: string; token: string; origins?: string[]; workspace?: SiteWorkspaceConfig };
|
||||||
|
|||||||
+102
-22
@@ -2,7 +2,7 @@ import express, { type NextFunction, type Request, type Response } from "express
|
|||||||
|
|
||||||
import { DEFAULT_PORT, ensureSiteWorkspace, loadConfig, saveConfig, updateSiteWorkspace, type CanvasAgentConfig } from "./config.js";
|
import { DEFAULT_PORT, ensureSiteWorkspace, loadConfig, saveConfig, updateSiteWorkspace, type CanvasAgentConfig } from "./config.js";
|
||||||
import { CanvasSession } from "./canvas-session.js";
|
import { CanvasSession } from "./canvas-session.js";
|
||||||
import { archiveCodexThread, interruptCodexTurn, listCodexThreads, readCodexThread, resumeCodexThread, runClaudeTurn, runCodexTurn, startCodexThread, summarizeCodexThread, verifyCodexThreadWorkspace, withAgentPrompt } from "./agents.js";
|
import { archiveCodexThread, interruptCodexTurn, isRecoverableThreadError, listCodexThreads, readCodexThread, resumeCodexThread, runClaudeTurn, runCodexTurn, startCodexThread, summarizeCodexThread, verifyCodexThreadWorkspace, withAgentPrompt } from "./agents.js";
|
||||||
import type { AgentAttachment } from "./types.js";
|
import type { AgentAttachment } from "./types.js";
|
||||||
|
|
||||||
export function startHttpServer() {
|
export function startHttpServer() {
|
||||||
@@ -12,7 +12,16 @@ export function startHttpServer() {
|
|||||||
saveConfig(config);
|
saveConfig(config);
|
||||||
|
|
||||||
const session = new CanvasSession();
|
const session = new CanvasSession();
|
||||||
const emit = (type: string, payload: unknown) => session.emitAll(type, payload);
|
const emit = (type: string, payload: unknown) => {
|
||||||
|
const data = payload && typeof payload === "object" && !Array.isArray(payload) ? payload as Record<string, unknown> : { value: payload };
|
||||||
|
const threadId = String(data.threadId || data.thread_id || ensureSiteWorkspace(config).activeThreadId || "");
|
||||||
|
threadId ? session.emitThread(type, threadId, data) : session.emitAll(type, data);
|
||||||
|
};
|
||||||
|
const setActiveThread = (activeThreadId: string, payload: Record<string, unknown> = {}) => {
|
||||||
|
const workspace = updateSiteWorkspace(config, { activeThreadId: activeThreadId || undefined });
|
||||||
|
session.emitThread("workspace_changed", activeThreadId, { ...payload, activeThreadId });
|
||||||
|
return workspace;
|
||||||
|
};
|
||||||
const app = express();
|
const app = express();
|
||||||
app.disable("x-powered-by");
|
app.disable("x-powered-by");
|
||||||
app.use(express.json({ limit: "30mb" }));
|
app.use(express.json({ limit: "30mb" }));
|
||||||
@@ -33,10 +42,21 @@ export function startHttpServer() {
|
|||||||
session.updateState(req.body, String(req.query.clientId || "") || undefined);
|
session.updateState(req.body, String(req.query.clientId || "") || undefined);
|
||||||
res.json({ ok: true });
|
res.json({ ok: true });
|
||||||
});
|
});
|
||||||
app.post("/canvas/result", (req, res) => {
|
app.post("/canvas/activate", (req, res) => {
|
||||||
session.resolveResult(req.body);
|
session.activateClient(String(req.query.clientId || ""));
|
||||||
res.json({ ok: true });
|
res.json({ ok: true });
|
||||||
});
|
});
|
||||||
|
app.post("/canvas/result", (req, res) => {
|
||||||
|
const ok = session.resolveResult(String(req.query.clientId || ""), req.body);
|
||||||
|
res.status(ok ? 200 : 409).json({ ok });
|
||||||
|
});
|
||||||
|
app.get("/agent/attachments/:attachmentId", route(async (req, res) => {
|
||||||
|
const attachment = session.getTurnAttachment(String(req.query.clientId || ""), routeParam(req.params.attachmentId));
|
||||||
|
const data = attachment.dataUrl.split(",", 2)[1];
|
||||||
|
if (!data) throw new Error("图片附件内容无效");
|
||||||
|
res.setHeader("Cache-Control", "no-store");
|
||||||
|
res.type(attachment.type).send(Buffer.from(data, "base64"));
|
||||||
|
}));
|
||||||
app.post("/api/tools", route(async (req, res) => res.json({ ok: true, result: await session.callTool(req.body?.name, req.body?.input || {}) })));
|
app.post("/api/tools", route(async (req, res) => res.json({ ok: true, result: await session.callTool(req.body?.name, req.body?.input || {}) })));
|
||||||
app.get("/agent/codex/workspace", (_req, res) => {
|
app.get("/agent/codex/workspace", (_req, res) => {
|
||||||
const workspace = ensureSiteWorkspace(config);
|
const workspace = ensureSiteWorkspace(config);
|
||||||
@@ -48,48 +68,102 @@ export function startHttpServer() {
|
|||||||
res.json({ ok: true, workspace, ...result });
|
res.json({ ok: true, workspace, ...result });
|
||||||
}));
|
}));
|
||||||
app.post("/agent/codex/threads/new", route(async (_req, res) => {
|
app.post("/agent/codex/threads/new", route(async (_req, res) => {
|
||||||
|
if (session.codexBusy) return res.status(409).json({ ok: false, error: "Codex 正在运行,请等待当前任务完成" });
|
||||||
const workspace = ensureSiteWorkspace(config);
|
const workspace = ensureSiteWorkspace(config);
|
||||||
const thread = await startCodexThread(emit, workspace.workspacePath);
|
const thread = await startCodexThread(emit, workspace.workspacePath);
|
||||||
const activeThreadId = String((thread as Record<string, unknown>).id || "");
|
const activeThreadId = String((thread as Record<string, unknown>).id || "");
|
||||||
updateSiteWorkspace(config, { activeThreadId });
|
const nextWorkspace = setActiveThread(activeThreadId, { emptyThread: true });
|
||||||
res.json({ ok: true, workspace: { ...workspace, activeThreadId }, thread: summarizeCodexThread(thread), messages: [] });
|
res.json({ ok: true, workspace: nextWorkspace, thread: summarizeCodexThread(thread), messages: [] });
|
||||||
}));
|
}));
|
||||||
app.get("/agent/codex/threads/:threadId", route(async (req, res) => {
|
app.get("/agent/codex/threads/:threadId", route(async (req, res) => {
|
||||||
const workspace = ensureSiteWorkspace(config);
|
const workspace = ensureSiteWorkspace(config);
|
||||||
const threadId = routeParam(req.params.threadId);
|
const threadId = routeParam(req.params.threadId);
|
||||||
res.json({ ok: true, workspace, ...(await readCodexThread(emit, threadId, workspace.workspacePath)) });
|
try {
|
||||||
|
res.json({ ok: true, workspace, ...(await readCodexThread(emit, threadId, workspace.workspacePath)) });
|
||||||
|
} catch (error) {
|
||||||
|
if (workspace.activeThreadId !== threadId || !isRecoverableThreadError(error)) throw error;
|
||||||
|
res.json({ ok: true, workspace, thread: { id: threadId, preview: "", cwd: workspace.workspacePath }, messages: [] });
|
||||||
|
}
|
||||||
}));
|
}));
|
||||||
app.post("/agent/codex/threads/:threadId/resume", route(async (req, res) => {
|
app.post("/agent/codex/threads/:threadId/resume", route(async (req, res) => {
|
||||||
|
if (session.codexBusy) return res.status(409).json({ ok: false, error: "Codex 正在运行,请等待当前任务完成" });
|
||||||
const workspace = ensureSiteWorkspace(config);
|
const workspace = ensureSiteWorkspace(config);
|
||||||
const threadId = routeParam(req.params.threadId);
|
const threadId = routeParam(req.params.threadId);
|
||||||
const result = await resumeCodexThread(emit, threadId, workspace.workspacePath);
|
const result = await resumeCodexThread(emit, threadId, workspace.workspacePath);
|
||||||
updateSiteWorkspace(config, { activeThreadId: threadId });
|
const nextWorkspace = setActiveThread(threadId);
|
||||||
res.json({ ok: true, workspace: { ...workspace, activeThreadId: threadId }, ...result });
|
res.json({ ok: true, workspace: nextWorkspace, ...result });
|
||||||
}));
|
}));
|
||||||
app.post("/agent/codex/threads/:threadId/delete", route(async (req, res) => {
|
app.post("/agent/codex/threads/:threadId/delete", route(async (req, res) => {
|
||||||
|
if (session.codexBusy) return res.status(409).json({ ok: false, error: "Codex 正在运行,请等待当前任务完成" });
|
||||||
const workspace = ensureSiteWorkspace(config);
|
const workspace = ensureSiteWorkspace(config);
|
||||||
const threadId = routeParam(req.params.threadId);
|
const threadId = routeParam(req.params.threadId);
|
||||||
await archiveCodexThread(emit, threadId, workspace.workspacePath);
|
await archiveCodexThread(emit, threadId, workspace.workspacePath);
|
||||||
if (workspace.activeThreadId === threadId) updateSiteWorkspace(config, { activeThreadId: undefined });
|
setActiveThread(workspace.activeThreadId === threadId ? "" : workspace.activeThreadId || "");
|
||||||
res.json({ ok: true });
|
res.json({ ok: true });
|
||||||
}));
|
}));
|
||||||
app.post("/agent/codex/turn", route(async (req, res) => {
|
app.post("/agent/codex/turn", route(async (req, res) => {
|
||||||
|
if (session.codexBusy) return res.status(409).json({ ok: false, error: "Codex 正在运行,请等待当前任务完成" });
|
||||||
const attachments = Array.isArray(req.body?.attachments) ? (req.body.attachments as AgentAttachment[]) : [];
|
const attachments = Array.isArray(req.body?.attachments) ? (req.body.attachments as AgentAttachment[]) : [];
|
||||||
const workspace = ensureSiteWorkspace(config);
|
const workspace = ensureSiteWorkspace(config);
|
||||||
let threadId = String(req.body?.threadId || workspace.activeThreadId || "");
|
const prompt = String(req.body?.prompt || "");
|
||||||
if (!threadId) {
|
if (!prompt.trim()) return res.status(400).json({ ok: false, error: "请输入任务内容" });
|
||||||
const thread = await startCodexThread(emit, workspace.workspacePath);
|
const clientId = String(req.body?.clientId || "");
|
||||||
threadId = String((thread as Record<string, unknown>).id || "");
|
session.setCodexState({ busy: true, threadId: String(req.body?.threadId || workspace.activeThreadId || ""), turnId: "" });
|
||||||
updateSiteWorkspace(config, { activeThreadId: threadId });
|
try {
|
||||||
} else if (threadId !== workspace.activeThreadId) {
|
let threadId = String(req.body?.threadId || workspace.activeThreadId || "");
|
||||||
await verifyCodexThreadWorkspace(emit, threadId, workspace.workspacePath);
|
let turnId = "";
|
||||||
updateSiteWorkspace(config, { activeThreadId: threadId });
|
if (!threadId) {
|
||||||
|
const thread = await startCodexThread(emit, workspace.workspacePath);
|
||||||
|
threadId = String((thread as Record<string, unknown>).id || "");
|
||||||
|
setActiveThread(threadId, { emptyThread: true });
|
||||||
|
} else if (threadId !== workspace.activeThreadId) {
|
||||||
|
await verifyCodexThreadWorkspace(emit, threadId, workspace.workspacePath);
|
||||||
|
setActiveThread(threadId);
|
||||||
|
}
|
||||||
|
const attachmentRefs = session.setTurnAttachments(clientId, attachments);
|
||||||
|
const chatMessage = {
|
||||||
|
sourceClientId: clientId,
|
||||||
|
message: { id: String(req.body?.messageId || Date.now()), role: "user", text: String(req.body?.messageText || prompt || `发送了 ${attachments.length} 张图片`) },
|
||||||
|
};
|
||||||
|
let chatThreadId = "";
|
||||||
|
const turnEmit = (type: string, payload: unknown) => {
|
||||||
|
const data = payload && typeof payload === "object" && !Array.isArray(payload) ? payload as Record<string, unknown> : { value: payload };
|
||||||
|
session.emitThread(type, threadId, data);
|
||||||
|
};
|
||||||
|
void runCodexTurn(withAgentPrompt(withAttachmentContext(prompt, attachmentRefs)), turnEmit, attachments, {
|
||||||
|
threadId,
|
||||||
|
cwd: workspace.workspacePath,
|
||||||
|
appEmit: emit,
|
||||||
|
onStart: clientId ? () => session.bindClient(clientId) : undefined,
|
||||||
|
onThread: (actualThreadId) => {
|
||||||
|
if (actualThreadId !== threadId) {
|
||||||
|
threadId = actualThreadId;
|
||||||
|
setActiveThread(threadId, { emptyThread: true });
|
||||||
|
}
|
||||||
|
session.setCodexState({ busy: true, threadId, turnId: "" });
|
||||||
|
if (chatThreadId !== threadId) {
|
||||||
|
chatThreadId = threadId;
|
||||||
|
session.emitThread("chat_message", threadId, chatMessage);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
onTurn: (actualTurnId) => {
|
||||||
|
turnId = actualTurnId;
|
||||||
|
session.setCodexState({ busy: true, threadId, turnId });
|
||||||
|
},
|
||||||
|
onFinish: () => {
|
||||||
|
session.clearTurnAttachments(clientId);
|
||||||
|
if (clientId) session.releaseClient(clientId);
|
||||||
|
session.setCodexState({ busy: false, threadId, turnId });
|
||||||
|
},
|
||||||
|
});
|
||||||
|
res.json({ ok: true, threadId });
|
||||||
|
} catch (error) {
|
||||||
|
session.setCodexState({ busy: false, threadId: String(req.body?.threadId || workspace.activeThreadId || ""), turnId: "" });
|
||||||
|
throw error;
|
||||||
}
|
}
|
||||||
void runCodexTurn(withAgentPrompt(String(req.body?.prompt || "")), emit, attachments, { threadId, cwd: workspace.workspacePath });
|
|
||||||
res.json({ ok: true, threadId });
|
|
||||||
}));
|
}));
|
||||||
app.post("/agent/codex/interrupt", (_req, res) => {
|
app.post("/agent/codex/interrupt", (req, res) => {
|
||||||
const ok = interruptCodexTurn();
|
const ok = interruptCodexTurn(String(req.body?.threadId || ""));
|
||||||
res.json({ ok });
|
res.json({ ok });
|
||||||
});
|
});
|
||||||
app.post("/agent/claude/turn", (req, res) => {
|
app.post("/agent/claude/turn", (req, res) => {
|
||||||
@@ -141,3 +215,9 @@ function validToken(req: Request, url: URL, token: string) {
|
|||||||
const header = req.headers["x-canvas-agent-token"];
|
const header = req.headers["x-canvas-agent-token"];
|
||||||
return url.searchParams.get("token") === token || header === token || (Array.isArray(header) && header.includes(token));
|
return url.searchParams.get("token") === token || header === token || (Array.isArray(header) && header.includes(token));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function withAttachmentContext(prompt: string, attachments: Array<{ id: string; name: string }>) {
|
||||||
|
if (!attachments.length) return prompt;
|
||||||
|
const list = attachments.map((item, index) => `${index + 1}. attachmentId=${item.id}, name=${JSON.stringify(item.name)}`).join("\n");
|
||||||
|
return `${prompt}\n\n本轮可用图片附件(顺序与图片输入一致):\n${list}\n需要把附件放入画布或作为生成参考图时,先调用 canvas_create_attachment_nodes,再使用返回的画布节点 ID 创建生成流程。`;
|
||||||
|
}
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ export const toolNames = [
|
|||||||
"canvas_export_snapshot",
|
"canvas_export_snapshot",
|
||||||
"canvas_apply_ops",
|
"canvas_apply_ops",
|
||||||
"canvas_create_node",
|
"canvas_create_node",
|
||||||
|
"canvas_create_attachment_nodes",
|
||||||
"canvas_create_text_node",
|
"canvas_create_text_node",
|
||||||
"canvas_create_text_nodes",
|
"canvas_create_text_nodes",
|
||||||
"canvas_create_config_node",
|
"canvas_create_config_node",
|
||||||
@@ -32,6 +33,7 @@ export const toolNames = [
|
|||||||
"canvas_select_nodes",
|
"canvas_select_nodes",
|
||||||
"canvas_set_viewport",
|
"canvas_set_viewport",
|
||||||
"canvas_run_generation",
|
"canvas_run_generation",
|
||||||
|
"generation_get_status",
|
||||||
"workbench_image_get_config",
|
"workbench_image_get_config",
|
||||||
"workbench_image_generate",
|
"workbench_image_generate",
|
||||||
"workbench_video_get_config",
|
"workbench_video_get_config",
|
||||||
@@ -93,6 +95,7 @@ export const toolInputSchemas = {
|
|||||||
canvas_export_snapshot: z.object({}).passthrough(),
|
canvas_export_snapshot: z.object({}).passthrough(),
|
||||||
canvas_apply_ops: z.object({ ops: z.array(canvasOpSchema) }),
|
canvas_apply_ops: z.object({ ops: z.array(canvasOpSchema) }),
|
||||||
canvas_create_node: z.object({ nodeType: nodeTypeSchema, title: z.string().optional(), x: z.number().optional(), y: z.number().optional(), width: z.number().optional(), height: z.number().optional(), metadata: recordSchema.optional() }),
|
canvas_create_node: z.object({ nodeType: nodeTypeSchema, title: z.string().optional(), x: z.number().optional(), y: z.number().optional(), width: z.number().optional(), height: z.number().optional(), metadata: recordSchema.optional() }),
|
||||||
|
canvas_create_attachment_nodes: z.object({ attachmentIds: z.array(z.string()).min(1), x: z.number().optional(), y: z.number().optional(), gap: z.number().optional(), direction: z.enum(["row", "column"]).optional() }),
|
||||||
canvas_create_text_node: z.object({ text: z.string().optional(), x: z.number().optional(), y: z.number().optional(), title: z.string().optional(), width: z.number().optional(), height: z.number().optional() }),
|
canvas_create_text_node: z.object({ text: z.string().optional(), x: z.number().optional(), y: z.number().optional(), title: z.string().optional(), width: z.number().optional(), height: z.number().optional() }),
|
||||||
canvas_create_text_nodes: z.object({ items: z.array(textNodeSchema).min(1), x: z.number().optional(), y: z.number().optional(), gap: z.number().optional(), direction: z.enum(["row", "column"]).optional() }),
|
canvas_create_text_nodes: z.object({ items: z.array(textNodeSchema).min(1), x: z.number().optional(), y: z.number().optional(), gap: z.number().optional(), direction: z.enum(["row", "column"]).optional() }),
|
||||||
canvas_create_config_node: z.object({ prompt: z.string().optional(), mode: generationModeSchema.optional(), title: z.string().optional(), x: z.number().optional(), y: z.number().optional(), width: z.number().optional(), height: z.number().optional(), autoRun: z.boolean().optional() }).merge(generationOptionsSchema),
|
canvas_create_config_node: z.object({ prompt: z.string().optional(), mode: generationModeSchema.optional(), title: z.string().optional(), x: z.number().optional(), y: z.number().optional(), width: z.number().optional(), height: z.number().optional(), autoRun: z.boolean().optional() }).merge(generationOptionsSchema),
|
||||||
@@ -111,6 +114,7 @@ export const toolInputSchemas = {
|
|||||||
canvas_select_nodes: z.object({ ids: z.array(z.string()) }),
|
canvas_select_nodes: z.object({ ids: z.array(z.string()) }),
|
||||||
canvas_set_viewport: z.object({ viewport: viewportSchema }),
|
canvas_set_viewport: z.object({ viewport: viewportSchema }),
|
||||||
canvas_run_generation: z.object({ nodeId: z.string(), mode: generationModeSchema.optional(), prompt: z.string().optional() }),
|
canvas_run_generation: z.object({ nodeId: z.string(), mode: generationModeSchema.optional(), prompt: z.string().optional() }),
|
||||||
|
generation_get_status: z.object({ scope: z.enum(["all", "canvas", "image", "video"]).optional(), taskId: z.string().optional(), nodeIds: z.array(z.string()).optional(), limit: z.number().optional() }),
|
||||||
workbench_image_get_config: z.object({}).passthrough(),
|
workbench_image_get_config: z.object({}).passthrough(),
|
||||||
workbench_image_generate: z.object({ prompt: z.string(), model: z.string().optional(), quality: z.string().optional(), size: z.string().optional(), count: z.number().optional(), run: z.boolean().optional() }),
|
workbench_image_generate: z.object({ prompt: z.string(), model: z.string().optional(), quality: z.string().optional(), size: z.string().optional(), count: z.number().optional(), run: z.boolean().optional() }),
|
||||||
workbench_video_get_config: z.object({}).passthrough(),
|
workbench_video_get_config: z.object({}).passthrough(),
|
||||||
@@ -128,6 +132,7 @@ export const toolDescriptions: Record<ToolName, string> = {
|
|||||||
canvas_export_snapshot: "导出当前画布快照,用于理解布局。",
|
canvas_export_snapshot: "导出当前画布快照,用于理解布局。",
|
||||||
canvas_apply_ops: "批量操作当前网页画布。ops 支持 add_node、update_node、delete_node、delete_connections、connect_nodes、set_viewport、select_nodes、run_generation。",
|
canvas_apply_ops: "批量操作当前网页画布。ops 支持 add_node、update_node、delete_node、delete_connections、connect_nodes、set_viewport、select_nodes、run_generation。",
|
||||||
canvas_create_node: "创建任意类型节点:text、image、config、video、audio。适合创建占位图、媒体占位、配置节点或自定义 metadata 节点。",
|
canvas_create_node: "创建任意类型节点:text、image、config、video、audio。适合创建占位图、媒体占位、配置节点或自定义 metadata 节点。",
|
||||||
|
canvas_create_attachment_nodes: "把当前对话中用户上传的图片附件创建成真实画布图片节点。attachmentIds 使用本轮附件清单中的 ID;返回的节点 ID 可传给 canvas_create_generation_flow.referenceNodeIds 作为生成参考图。",
|
||||||
canvas_create_text_node: "在当前画布创建单个文本节点。",
|
canvas_create_text_node: "在当前画布创建单个文本节点。",
|
||||||
canvas_create_text_nodes: "批量创建文本节点,适合生成标题、段落、脚本、说明等内容块。",
|
canvas_create_text_nodes: "批量创建文本节点,适合生成标题、段落、脚本、说明等内容块。",
|
||||||
canvas_create_config_node: "创建生成配置节点,可指定 text/image/video/audio 模式和生成参数,可选择立即触发生成。",
|
canvas_create_config_node: "创建生成配置节点,可指定 text/image/video/audio 模式和生成参数,可选择立即触发生成。",
|
||||||
@@ -146,10 +151,11 @@ export const toolDescriptions: Record<ToolName, string> = {
|
|||||||
canvas_select_nodes: "设置当前选中节点。",
|
canvas_select_nodes: "设置当前选中节点。",
|
||||||
canvas_set_viewport: "调整画布视口。",
|
canvas_set_viewport: "调整画布视口。",
|
||||||
canvas_run_generation: "触发指定节点生成,通常用于配置节点或文本/图片/视频/音频节点。",
|
canvas_run_generation: "触发指定节点生成,通常用于配置节点或文本/图片/视频/音频节点。",
|
||||||
|
generation_get_status: "查询当前活动网页的生成任务状态。默认返回画布、生图工作台和视频工作台最近任务;可用 scope 过滤来源,用 taskId 查询工作台任务,用 nodeIds 查询画布节点。",
|
||||||
workbench_image_get_config: "读取生图工作台的当前参数和可选项(可用模型、质量、尺寸/宽高比、张数范围),在调用 workbench_image_generate 前先了解可选值。",
|
workbench_image_get_config: "读取生图工作台的当前参数和可选项(可用模型、质量、尺寸/宽高比、张数范围),在调用 workbench_image_generate 前先了解可选值。",
|
||||||
workbench_image_generate: "在生图工作台填入提示词并按需设置 model、quality、size(如 1:1 或 1024x1024)、count,run 默认 true 会自动点击生成按钮。会自动跳转到生图工作台。生成为异步过程,工具返回代表已提交,结果请在工作台查看。",
|
workbench_image_generate: "在生图工作台填入提示词并按需设置 model、quality、size(如 1:1 或 1024x1024)、count,run 默认 true 会自动点击生成按钮。会自动跳转到生图工作台。生成为异步过程,提交后返回 taskId,可用 generation_get_status 查询状态。",
|
||||||
workbench_video_get_config: "读取视频创作台的当前参数和可选项(可用模型、尺寸/比例、时长、清晰度/分辨率、是否生成声音与水印)。",
|
workbench_video_get_config: "读取视频创作台的当前参数和可选项(可用模型、尺寸/比例、时长、清晰度/分辨率、是否生成声音与水印)。",
|
||||||
workbench_video_generate: "在视频创作台填入提示词并按需设置 model、size、seconds、resolution、generateAudio、watermark,run 默认 true 会自动点击生成按钮。会自动跳转到视频创作台。生成为异步过程,工具返回代表已提交。",
|
workbench_video_generate: "在视频创作台填入提示词并按需设置 model、size、seconds、resolution、generateAudio、watermark,run 默认 true 会自动点击生成按钮。会自动跳转到视频创作台。生成为异步过程,提交后返回 taskId,可用 generation_get_status 查询状态。",
|
||||||
prompts_search: "搜索提示词库(第三方提示词合集),支持 keyword、category、tags 过滤和 page/pageSize 分页,返回标题、提示词、分类、标签、封面等。",
|
prompts_search: "搜索提示词库(第三方提示词合集),支持 keyword、category、tags 过滤和 page/pageSize 分页,返回标题、提示词、分类、标签、封面等。",
|
||||||
assets_list: "列出用户「我的素材」,支持 kind(text/image/video)过滤、keyword 搜索和 page/pageSize 分页。为控制体积不返回图片/视频原始 data,仅返回封面与元信息。",
|
assets_list: "列出用户「我的素材」,支持 kind(text/image/video)过滤、keyword 搜索和 page/pageSize 分页。为控制体积不返回图片/视频原始 data,仅返回封面与元信息。",
|
||||||
assets_add: "向「我的素材」新增素材。kind=text 时用 content 传文本内容;kind=image 时用 imageUrl 传图片地址或 dataURL。可附带 title、tags、source、note。",
|
assets_add: "向「我的素材」新增素材。kind=text 时用 content 传文本内容;kind=image 时用 imageUrl 传图片地址或 dataURL。可附带 title、tags、source、note。",
|
||||||
|
|||||||
@@ -5,4 +5,4 @@ export type CanvasNode = { id: string; type: CanvasNodeType; title?: string; pos
|
|||||||
export type CanvasConnection = { id: string; fromNodeId: string; toNodeId: string };
|
export type CanvasConnection = { id: string; fromNodeId: string; toNodeId: string };
|
||||||
export type CanvasSnapshot = { projectId?: string; title?: string; nodes?: CanvasNode[]; connections?: CanvasConnection[]; selectedNodeIds?: string[]; viewport?: Viewport; clientId?: string };
|
export type CanvasSnapshot = { projectId?: string; title?: string; nodes?: CanvasNode[]; connections?: CanvasConnection[]; selectedNodeIds?: string[]; viewport?: Viewport; clientId?: string };
|
||||||
export type AgentEmit = (type: string, payload: unknown) => void;
|
export type AgentEmit = (type: string, payload: unknown) => void;
|
||||||
export type AgentAttachment = { name?: string; type?: string; dataUrl?: string };
|
export type AgentAttachment = { id?: string; name?: string; type?: string; size?: number; width?: number; height?: number; dataUrl?: string };
|
||||||
|
|||||||
@@ -10,5 +10,6 @@
|
|||||||
"rootDir": "src",
|
"rootDir": "src",
|
||||||
"skipLibCheck": true
|
"skipLibCheck": true
|
||||||
},
|
},
|
||||||
"include": ["src/**/*.ts"]
|
"include": ["src/**/*.ts"],
|
||||||
|
"exclude": ["src/**/*.test.ts"]
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -7,7 +7,6 @@ description: 当前已接入同步逻辑的第三方提示词仓库
|
|||||||
|
|
||||||
| 地址 | 状态 |
|
| 地址 | 状态 |
|
||||||
| --- | --- |
|
| --- | --- |
|
||||||
| https://github.com/EvoLinkAI/awesome-gpt-image-2-API-and-Prompts | 已实现同步逻辑 |
|
|
||||||
| https://github.com/ZeroLu/awesome-gpt-image | 已实现同步逻辑 |
|
| https://github.com/ZeroLu/awesome-gpt-image | 已实现同步逻辑 |
|
||||||
| https://github.com/ImgEdify/Awesome-GPT4o-Image-Prompts | 已实现同步逻辑 |
|
| https://github.com/ImgEdify/Awesome-GPT4o-Image-Prompts | 已实现同步逻辑 |
|
||||||
| https://github.com/YouMind-OpenLab/awesome-gpt-image-2 | 已实现同步逻辑 |
|
| https://github.com/YouMind-OpenLab/awesome-gpt-image-2 | 已实现同步逻辑 |
|
||||||
|
|||||||
@@ -5,52 +5,14 @@ description: 当前版本已实现但仍需人工验证的变更项
|
|||||||
|
|
||||||
# 待测试
|
# 待测试
|
||||||
|
|
||||||
- 左侧画布面板「资产」Tab:搜索框旁「添加」按钮可上传图片/视频文件(支持多选)加入资产库;每张资产卡片悬停显示移除按钮,二次确认后删除;点击卡片仍插入到画布;需验证添加图片/视频、非图片视频文件的提示、移除确认与插入行为。
|
- 提示词来源:6 个内置来源应从 Image Prompts 统一仓库读取,更新后数量依次为 323、494、53、76、126、129;提示词仍按 6 个来源分组并可独立启用,来源内可继续按 `tags` 筛选。添加标准 JSON URL 后应能查看内容,填写非数组 JSON 或不可访问地址时应显示失败,并继续保留该来源上一次成功缓存的内容。
|
||||||
- 左侧画布面板:支持拖拽右边缘调整宽度(记忆到本地)、通过顶栏菜单左侧的开关按钮展开/收起,展开收起有动画;需验证宽度拖拽、动画、刷新后保持展开/收起状态与宽度。
|
- 提示词来源界面:来源应以卡片列表展示,启用开关位于左侧,数量、同步状态和上次成功时间作为次级信息显示,查看、拉取及自定义来源编辑/删除操作使用带文字按钮;底部定时拉取区域应保持独立边框布局。
|
||||||
- 左侧画布面板:「画布/资产」切换改为带滑动下划线的动画;点击「画布元素」列表项跳转到对应节点时带缓动动画(视图平滑移动缩放而非瞬移);文本/配置/视频/音频等非图片元素的图标去掉灰色底色;需验证视觉与交互。
|
- 我的提示词:在提示词中心切换到「我的提示词」,验证新增、编辑、删除和搜索;从提示词库保存后应出现在个人列表、画布左侧提示词面板和提示词选择弹窗中,刷新页面后数据仍保留在本地,Agent 搜索提示词时也应能查到。
|
||||||
- 左侧画布面板:画布元素列表点击「选择」进入多选模式,可勾选/全选节点后「导出选中」为压缩包;压缩包为扁平结构——图片/视频/音频直接是对应文件(按节点名称命名、重名自动加序号),文本节点为 `.txt`,其余无法识别的节点各自导出为同名 `.json`;需验证多选、全选、导出内容与命名正确。
|
- 画布提示词库:不展开任何公共来源直接搜索其中的提示词,匹配项应自动显示;点击「插入画布」后应创建正文正确且标题保持为提示词标题的文本节点。
|
||||||
- 顶栏菜单:新增「导出当前画布」,导出当前画布为压缩包(与项目列表页单个导出格式一致);需验证导出文件名与内容。
|
- 全站 Agent:新增 `generation_get_status` 工具,画布生成节点可按 `nodeIds` 查询,生图和视频工作台提交后会返回 `taskId` 并可查询排队、运行、成功或失败状态;需验证查询只由当前活动标签页返回。
|
||||||
- 画布节点插件系统:内置节点(图片/文本/视频/音频/配置/组)改为统一注册表管理,创建菜单由注册表动态生成;需验证六种内置节点的创建、渲染、缩放、连线、小地图颜色、作为生成输入等行为与之前一致。内置节点仅这六种,其余均为插件。
|
- 本地 Agent 多标签页隔离:同时打开两个不同画布并连接同一个 Agent,分别聚焦标签页后通过 MCP 读取和修改画布,操作应只落在当前聚焦页面;网页面板发起的整个 Codex turn 应固定操作发起页面,即使中途聚焦另一标签页也不能切换目标;关闭当前页面后应回退到最近聚焦且仍连接的页面,其他页面回传同一请求结果应被拒绝。
|
||||||
- 画布节点插件系统:左上菜单新增「节点插件」,可通过 URL 安装远程插件、启用/禁用、更新、卸载;安装时有安全警告说明插件代码在页面内执行、可访问本地数据。可用 `/plugins/sticky-note.js` 安装便利贴节点验证:换色、编辑、「衍生文本节点」(演示 applyOps 增节点+连线)。
|
- 本地 Agent 多标签页会话同步:所有标签页共享同一个站点级 Codex 活跃线程;任一页面发送消息、新建、恢复或删除会话后,其他页面应同步活跃线程和聊天记录;Agent 输出仅显示在事件所属线程,运行中不能新建、恢复、删除或再次发送任务。
|
||||||
- 画布节点插件系统:示例插件 Markdown(编辑/渲染)、HTML(沙箱 iframe 渲染,支持 `{{input}}` 注入上游文本)、SVG(渲染/编辑,可取上游文本 SVG 源码)、3D 全景(three.js CDN 动态加载,可从上游图片节点取全景图并拖拽查看);均为远程插件,需分别安装 `/plugins/markdown.js`、`/plugins/html.js`、`/plugins/svg.js`、`/plugins/panorama.js` 后验证创建、渲染、编辑及与上游节点的交互。
|
- 本地 Agent 运行状态同步:在一个标签页运行较长 Codex 任务,等待某张工具卡显示「工具完成」后再打开或刷新第二个标签页;第二个标签页应立即显示 Codex 正在运行并禁用发送,整轮结束后两个标签页同时恢复;工具卡只显示「工具完成」,整轮结束由「本轮完成」表示。
|
||||||
- 画布节点插件系统:卸载或禁用插件后,画布上遗留的该类型节点显示「缺少插件」占位且数据保留,重新安装/启用后恢复渲染;需验证刷新后已启用插件自动加载。
|
- 本地 Agent 图片附件落画布:在右侧 Agent 上传参考图并要求基于商品信息创建生图流程,附件应创建为保持原比例的真实图片节点,分析提示词应创建为文本节点,二者都应连接到生成配置节点;刷新页面后参考图仍可显示并参与生成。任务中途切换到其他标签页时,附件只能写入发起任务的标签页;若发起标签页关闭,附件读取应失败且不能落入其他画布。
|
||||||
- 画布节点插件系统:每个插件为 `plugins/canvas/<name>/` 独立目录(各自 package.json + esbuild + src + README,互不耦合),`npm run build` 产物落到 `dist/<name>.js` 并同步到 `web/public/plugins/`;需验证各插件可独立构建与安装。
|
- Agent 对话滚动:从历史或日志切回对话、恢复其他会话时应自动定位到最新消息;手动向上浏览后应显示向下箭头且新消息不强制打断阅读,点击箭头后平滑回到底部并继续跟随新消息。
|
||||||
- 画布节点插件本地开发:每个插件支持 `npm run dev`(watch 自动构建并同步到 `web/public/plugins/`),配合 `web/.env.local` 的 `VITE_DEV_PLUGINS`(逗号分隔 URL)每次刷新重新拉取、无需反复安装;需验证改源码刷新页面即生效。
|
- Agent 消息区分:用户消息应在右侧以主题浅底气泡显示,AI 回复应在左侧保留 Agent 图标并使用开放式 Markdown 排版;长文本、多张附件、错误消息及浅色/深色主题下均应清晰且不溢出。
|
||||||
- 画布节点创建:鼠标左键双击画布空白区域会在点击位置打开节点选择菜单,可创建文本、图片、视频、音频、生成配置和组节点;双击已有节点或连线不应触发菜单。
|
|
||||||
- 本地 Agent 连接说明:新增提醒说明只有安装 Codex 插件或手动添加 MCP 后才会增加 Codex token 消耗,直接运行 `npx -y @basketikun/canvas-agent` 不会安装 MCP;需验证文案位置清晰。
|
|
||||||
- 本地 Agent 连接说明:画布面板和配置页 Codex Tab 改为区分“Codex 插件启动画布”和“直接运行 Agent 后网页连接”两种方式,需验证文案和布局清晰。
|
|
||||||
- 全站 Agent:Agent 面板从画布页抽离为全站常驻面板,挂在全局布局右侧,开/关只挤压左侧内容不遮挡;顶栏和画布工具条各有一个开关按钮,需验证在首页、生图、视频、素材、提示词库、画布等页面都能打开面板并保持 SSE 连接不断开。
|
|
||||||
- 全站 Agent:新增 `site_navigate` 工具,Agent 可直接跳转 `/`、`/canvas`、`/canvas/:id`、`/image`、`/video`、`/prompts`、`/assets`、`/config`;导航直接执行不走工具确认,需验证跳转正确且面板不消失。
|
|
||||||
- 全站 Agent:画布工具(canvas_apply_ops 等)仅在当前处于画布页时可用,不在画布页调用会返回“当前不在画布页,请先用 site_navigate 打开画布”,需验证 Agent 会先导航到画布再执行画布操作。
|
|
||||||
- 全站 Agent:Codex 会话改为单一站点级连续线程(不再按 canvasId 分),跨页面/跨画布保持同一条对话上下文,历史记录即这条线程;本地 Agent workspace 从按画布分改为固定 `site` workspace(`~/.infinite-canvas/codex-workspaces/site`),已装 `@basketikun/canvas-agent` 的用户需更新到新版本才生效。
|
|
||||||
- 全站 Agent:移除仅前端调用 OpenAI responses 接口的在线画布助手死代码(含 `requestToolResponse`),Agent 统一走 MCP + 本地 Codex 链路。
|
|
||||||
- 画布节点名称:每个节点保留 `title` 名称字段,默认名称显示在节点上方,双击名称可直接编辑,节点信息弹窗和 JSON 可查看该字段。
|
|
||||||
- 画布节点工具条:节点顶部工具条改为点击选中节点后显示,单纯鼠标移入 / 移出节点不应再频繁弹出。
|
|
||||||
- 画布组节点:组内子节点优先响应点击和拖拽;当鼠标位于组内节点上时,应移动 / 编辑该节点,而不是移动整个组。
|
|
||||||
- 画布组节点:工具栏新增“组”节点,普通节点拖入组区域会高亮目标组,松手后自动吸附进组;拖出组区域会解除归属,拖动组节点会一起移动组内子节点。
|
|
||||||
- GitHub Pages:新增版本 tag 触发的前端静态站点自动发布 workflow,需在仓库 Pages 设置中选择 GitHub Actions 并验证路由刷新回退。
|
|
||||||
- 画布图片切图:行数 / 列数会生成可直接拖拽的等分切图线,并支持新增横向 / 纵向线、删除单条线、重置切图线和切片数量预览。
|
|
||||||
- Codex App 插件:精简插件 README 安装步骤,更新插件网站和作者信息,并将 `open-canvas` 调整为默认打开在线画布、仅按需启动本地前端;需重新安装后确认详情页和两种打开流程正确。
|
|
||||||
- 本地 Codex 连接画布文档:新增 Canvas Agent、MCP、SSE 与浏览器画布交互原理说明,需确认文档导航和描述是否清晰。
|
|
||||||
- 顶部导航:缩小桌面顶部导航栏高度,导航项下划线和内容垂直对齐需确认。
|
|
||||||
- 配置与用户偏好:新增独立顶部导航页面,并在页面和弹窗中复用同一套配置面板;新增 Codex Tab,展示插件安装和本地 Agent 启动步骤,可配置本地 Agent 地址、Connect token 和工具执行确认开关。
|
|
||||||
- Codex 全局连接:配置页可直接连接本地 Canvas Agent,连接状态同步到画布和顶部 OpenAI 状态图标,刷新后会按本地保存的地址和 token 自动尝试重连;需验证已连接、连接中、连接失败、断开和刷新自动重连状态。
|
|
||||||
- 画布 Agent:Codex 旧 thread 在重启后不可用时会自动新建会话并重试,需验证进入画布直接发送消息不会再出现 `thread not loaded` 或 `no rollout found`。
|
|
||||||
- 画布 Agent:切换不同画布时会清空并重新加载当前 canvasId 对应的 Codex 会话,需验证不会继续显示上一个画布的对话。
|
|
||||||
- 画布 Agent:右侧 Agent 面板移除网站 / 本机模式切换,固定使用本机 Agent。
|
|
||||||
- 配置与用户偏好:每个模型渠道新增 OpenAI / Gemini 调用格式选择,默认 OpenAI,切换默认格式时同步更新默认 Base URL;渠道 Tab 增加紧凑醒目的提醒,说明模型是否可选需要到“模型”Tab 选择。
|
|
||||||
- Gemini 调用格式:支持拉取 Gemini 模型列表,并用于文本问答、在线 Agent 工具调用、基础生图和图生图请求。
|
|
||||||
- Gemini 生图:生图工作台和画布生图会把尺寸配置转换为 Gemini `responseFormat.image.aspectRatio` 传入,需验证 1:1、9:16、16:9 及自定义像素尺寸的输出比例。
|
|
||||||
- 音频和视频生成:选择 Gemini 调用格式时先给出不支持提示,仍需使用 OpenAI 兼容渠道。
|
|
||||||
- WebDAV 同步:移除 Next.js 转发代理和连接方式选择,改为浏览器前端直接连接 WebDAV 服务。
|
|
||||||
- 提示词库:移除 Next.js route 抓取和服务端内存缓存,改为浏览器前端直连 GitHub 原始文件并缓存到 IndexedDB。
|
|
||||||
- 提示词库:移除已 404 的 `EvoLinkAI/awesome-gpt-image-2-API-and-Prompts` 来源。
|
|
||||||
- 前端构建:移除 Next.js,改为 Vite + React Router,并将前端目录调整为 `pages` / `components` / `stores` / `lib` / `layouts` / `styles`,需验证首页、图片/视频工作台、素材、提示词库和画布路由跳转。
|
|
||||||
- 前端构建:修复 TypeScript 严格检查下的构建报错。
|
|
||||||
- Vercel 部署:新增 `web` 和 `docs` 两个子项目的 Vercel 配置,需分别验证静态前端和文档站部署。
|
|
||||||
- Docker 部署:运行镜像从 Node serve 改为 nginx 静态托管,需验证前端路由刷新回退。
|
|
||||||
- 环境变量:移除过期的后端 `.env.example` 示例文件。
|
|
||||||
- 生图工作台:修复失败图片点击重试后只保存在页面状态、刷新或跳转后丢失的问题,重试成功结果会写入本地图片存储和生成记录。
|
|
||||||
- 视频生成:任务查询接口直接返回 `url` / `result_url` / `video_url` 时会跳过 `/content` 下载,并优先保存到浏览器本地媒体存储。
|
|
||||||
- 视频生成:任务创建或查询失败时会解析接口返回的 `msg` / `message` / 嵌套 JSON 错误,并在生成结果失败卡片里展示真实原因。
|
|
||||||
- 画布生成配置:修复切换文本/视频/音频模式时模型选择仍显示为生图模型(gpt-image-2)的问题,切换模式后应自动匹配对应能力的默认模型。
|
|
||||||
|
|||||||
@@ -32,7 +32,6 @@
|
|||||||
"streamdown": "^2.5.0",
|
"streamdown": "^2.5.0",
|
||||||
"tailwind-merge": "^3.6.0",
|
"tailwind-merge": "^3.6.0",
|
||||||
"tailwindcss": "^4",
|
"tailwindcss": "^4",
|
||||||
"three": "^0.180.0",
|
|
||||||
"tw-animate-css": "^1.4.0",
|
"tw-animate-css": "^1.4.0",
|
||||||
"zustand": "^5.0.12",
|
"zustand": "^5.0.12",
|
||||||
},
|
},
|
||||||
@@ -42,7 +41,6 @@
|
|||||||
"@types/node": "^20",
|
"@types/node": "^20",
|
||||||
"@types/react": "19.1.12",
|
"@types/react": "19.1.12",
|
||||||
"@types/react-dom": "19.1.9",
|
"@types/react-dom": "19.1.9",
|
||||||
"@types/three": "^0.180.0",
|
|
||||||
"@vitejs/plugin-react": "^5.2.0",
|
"@vitejs/plugin-react": "^5.2.0",
|
||||||
"prettier": "^3.8.3",
|
"prettier": "^3.8.3",
|
||||||
"typescript": "^5",
|
"typescript": "^5",
|
||||||
@@ -159,8 +157,6 @@
|
|||||||
|
|
||||||
"@ctrl/tinycolor": ["@ctrl/tinycolor@4.2.0", "https://registry.npmmirror.com/@ctrl/tinycolor/-/tinycolor-4.2.0.tgz", {}, "sha512-kzyuwOAQnXJNLS9PSyrk0CWk35nWJW/zl/6KvnTBMFK65gm7U1/Z5BqjxeapjZCIhQcM/DsrEmcbRwDyXyXK4A=="],
|
"@ctrl/tinycolor": ["@ctrl/tinycolor@4.2.0", "https://registry.npmmirror.com/@ctrl/tinycolor/-/tinycolor-4.2.0.tgz", {}, "sha512-kzyuwOAQnXJNLS9PSyrk0CWk35nWJW/zl/6KvnTBMFK65gm7U1/Z5BqjxeapjZCIhQcM/DsrEmcbRwDyXyXK4A=="],
|
||||||
|
|
||||||
"@dimforge/rapier3d-compat": ["@dimforge/rapier3d-compat@0.12.0", "", {}, "sha512-uekIGetywIgopfD97oDL5PfeezkFpNhwlzlaEYNOA0N6ghdsOvh/HYjSMek5Q2O1PYvRSDFcqFVJl4r4ZBwOow=="],
|
|
||||||
|
|
||||||
"@dnd-kit/accessibility": ["@dnd-kit/accessibility@3.1.1", "https://registry.npmmirror.com/@dnd-kit/accessibility/-/accessibility-3.1.1.tgz", { "dependencies": { "tslib": "^2.0.0" }, "peerDependencies": { "react": ">=16.8.0" } }, "sha512-2P+YgaXF+gRsIihwwY1gCsQSYnu9Zyj2py8kY5fFvUM1qm2WA2u639R6YNVfU4GWr+ZM5mqEsfHZZLoRONbemw=="],
|
"@dnd-kit/accessibility": ["@dnd-kit/accessibility@3.1.1", "https://registry.npmmirror.com/@dnd-kit/accessibility/-/accessibility-3.1.1.tgz", { "dependencies": { "tslib": "^2.0.0" }, "peerDependencies": { "react": ">=16.8.0" } }, "sha512-2P+YgaXF+gRsIihwwY1gCsQSYnu9Zyj2py8kY5fFvUM1qm2WA2u639R6YNVfU4GWr+ZM5mqEsfHZZLoRONbemw=="],
|
||||||
|
|
||||||
"@dnd-kit/core": ["@dnd-kit/core@6.3.1", "https://registry.npmmirror.com/@dnd-kit/core/-/core-6.3.1.tgz", { "dependencies": { "@dnd-kit/accessibility": "^3.1.1", "@dnd-kit/utilities": "^3.2.2", "tslib": "^2.0.0" }, "peerDependencies": { "react": ">=16.8.0", "react-dom": ">=16.8.0" } }, "sha512-xkGBRQQab4RLwgXxoqETICr6S5JlogafbhNsidmrkVv2YRs5MLwpjoF2qpiGjQt8S9AoxtIV603s0GIUpY5eYQ=="],
|
"@dnd-kit/core": ["@dnd-kit/core@6.3.1", "https://registry.npmmirror.com/@dnd-kit/core/-/core-6.3.1.tgz", { "dependencies": { "@dnd-kit/accessibility": "^3.1.1", "@dnd-kit/utilities": "^3.2.2", "tslib": "^2.0.0" }, "peerDependencies": { "react": ">=16.8.0", "react-dom": ">=16.8.0" } }, "sha512-xkGBRQQab4RLwgXxoqETICr6S5JlogafbhNsidmrkVv2YRs5MLwpjoF2qpiGjQt8S9AoxtIV603s0GIUpY5eYQ=="],
|
||||||
@@ -613,8 +609,6 @@
|
|||||||
|
|
||||||
"@ts-morph/common": ["@ts-morph/common@0.27.0", "https://registry.npmmirror.com/@ts-morph/common/-/common-0.27.0.tgz", { "dependencies": { "fast-glob": "^3.3.3", "minimatch": "^10.0.1", "path-browserify": "^1.0.1" } }, "sha512-Wf29UqxWDpc+i61k3oIOzcUfQt79PIT9y/MWfAGlrkjg6lBC1hwDECLXPVJAhWjiGbfBCxZd65F/LIZF3+jeJQ=="],
|
"@ts-morph/common": ["@ts-morph/common@0.27.0", "https://registry.npmmirror.com/@ts-morph/common/-/common-0.27.0.tgz", { "dependencies": { "fast-glob": "^3.3.3", "minimatch": "^10.0.1", "path-browserify": "^1.0.1" } }, "sha512-Wf29UqxWDpc+i61k3oIOzcUfQt79PIT9y/MWfAGlrkjg6lBC1hwDECLXPVJAhWjiGbfBCxZd65F/LIZF3+jeJQ=="],
|
||||||
|
|
||||||
"@tweenjs/tween.js": ["@tweenjs/tween.js@23.1.3", "", {}, "sha512-vJmvvwFxYuGnF2axRtPYocag6Clbb5YS7kLL+SO/TeVFzHqDIWrNKYtcsPMibjDx9O+bu+psAy9NKfWklassUA=="],
|
|
||||||
|
|
||||||
"@types/babel__core": ["@types/babel__core@7.20.5", "", { "dependencies": { "@babel/parser": "^7.20.7", "@babel/types": "^7.20.7", "@types/babel__generator": "*", "@types/babel__template": "*", "@types/babel__traverse": "*" } }, "sha512-qoQprZvz5wQFJwMDqeseRXWv3rqMvhgpbXFfVyWhbx9X47POIA6i/+dXefEmZKoAgOaTdaIgNSMqMIU61yRyzA=="],
|
"@types/babel__core": ["@types/babel__core@7.20.5", "", { "dependencies": { "@babel/parser": "^7.20.7", "@babel/types": "^7.20.7", "@types/babel__generator": "*", "@types/babel__template": "*", "@types/babel__traverse": "*" } }, "sha512-qoQprZvz5wQFJwMDqeseRXWv3rqMvhgpbXFfVyWhbx9X47POIA6i/+dXefEmZKoAgOaTdaIgNSMqMIU61yRyzA=="],
|
||||||
|
|
||||||
"@types/babel__generator": ["@types/babel__generator@7.27.0", "", { "dependencies": { "@babel/types": "^7.0.0" } }, "sha512-ufFd2Xi92OAVPYsy+P4n7/U7e68fex0+Ee8gSG9KX7eo084CWiQ4sdxktvdl0bOPupXtVJPY19zk6EwWqUQ8lg=="],
|
"@types/babel__generator": ["@types/babel__generator@7.27.0", "", { "dependencies": { "@babel/types": "^7.0.0" } }, "sha512-ufFd2Xi92OAVPYsy+P4n7/U7e68fex0+Ee8gSG9KX7eo084CWiQ4sdxktvdl0bOPupXtVJPY19zk6EwWqUQ8lg=="],
|
||||||
@@ -713,20 +707,14 @@
|
|||||||
|
|
||||||
"@types/set-cookie-parser": ["@types/set-cookie-parser@2.4.10", "https://registry.npmmirror.com/@types/set-cookie-parser/-/set-cookie-parser-2.4.10.tgz", { "dependencies": { "@types/node": "*" } }, "sha512-GGmQVGpQWUe5qglJozEjZV/5dyxbOOZ0LHe/lqyWssB88Y4svNfst0uqBVscdDeIKl5Jy5+aPSvy7mI9tYRguw=="],
|
"@types/set-cookie-parser": ["@types/set-cookie-parser@2.4.10", "https://registry.npmmirror.com/@types/set-cookie-parser/-/set-cookie-parser-2.4.10.tgz", { "dependencies": { "@types/node": "*" } }, "sha512-GGmQVGpQWUe5qglJozEjZV/5dyxbOOZ0LHe/lqyWssB88Y4svNfst0uqBVscdDeIKl5Jy5+aPSvy7mI9tYRguw=="],
|
||||||
|
|
||||||
"@types/stats.js": ["@types/stats.js@0.17.4", "", {}, "sha512-jIBvWWShCvlBqBNIZt0KAshWpvSjhkwkEu4ZUcASoAvhmrgAUI2t1dXrjSL4xXVLB4FznPrIsX3nKXFl/Dt4vA=="],
|
|
||||||
|
|
||||||
"@types/statuses": ["@types/statuses@2.0.6", "https://registry.npmmirror.com/@types/statuses/-/statuses-2.0.6.tgz", {}, "sha512-xMAgYwceFhRA2zY+XbEA7mxYbA093wdiW8Vu6gZPGWy9cmOyU9XesH1tNcEWsKFd5Vzrqx5T3D38PWx1FIIXkA=="],
|
"@types/statuses": ["@types/statuses@2.0.6", "https://registry.npmmirror.com/@types/statuses/-/statuses-2.0.6.tgz", {}, "sha512-xMAgYwceFhRA2zY+XbEA7mxYbA093wdiW8Vu6gZPGWy9cmOyU9XesH1tNcEWsKFd5Vzrqx5T3D38PWx1FIIXkA=="],
|
||||||
|
|
||||||
"@types/three": ["@types/three@0.180.0", "", { "dependencies": { "@dimforge/rapier3d-compat": "~0.12.0", "@tweenjs/tween.js": "~23.1.3", "@types/stats.js": "*", "@types/webxr": "*", "@webgpu/types": "*", "fflate": "~0.8.2", "meshoptimizer": "~0.22.0" } }, "sha512-ykFtgCqNnY0IPvDro7h+9ZeLY+qjgUWv+qEvUt84grhenO60Hqd4hScHE7VTB9nOQ/3QM8lkbNE+4vKjEpUxKg=="],
|
|
||||||
|
|
||||||
"@types/trusted-types": ["@types/trusted-types@2.0.7", "", {}, "sha512-ScaPdn1dQczgbl0QFTeTOmVHFULt394XJgOQNoyVhZ6r2vLnMLJfBPd53SB52T/3G36VI1/g2MZaX0cwDuXsfw=="],
|
"@types/trusted-types": ["@types/trusted-types@2.0.7", "", {}, "sha512-ScaPdn1dQczgbl0QFTeTOmVHFULt394XJgOQNoyVhZ6r2vLnMLJfBPd53SB52T/3G36VI1/g2MZaX0cwDuXsfw=="],
|
||||||
|
|
||||||
"@types/unist": ["@types/unist@2.0.11", "https://registry.npmmirror.com/@types/unist/-/unist-2.0.11.tgz", {}, "sha512-CmBKiL6NNo/OqgmMn95Fk9Whlp2mtvIv+KNpQKN2F4SjvrEesubTRWGYSg+BnWZOnlCaSTU1sMpsBOzgbYhnsA=="],
|
"@types/unist": ["@types/unist@2.0.11", "https://registry.npmmirror.com/@types/unist/-/unist-2.0.11.tgz", {}, "sha512-CmBKiL6NNo/OqgmMn95Fk9Whlp2mtvIv+KNpQKN2F4SjvrEesubTRWGYSg+BnWZOnlCaSTU1sMpsBOzgbYhnsA=="],
|
||||||
|
|
||||||
"@types/validate-npm-package-name": ["@types/validate-npm-package-name@4.0.2", "https://registry.npmmirror.com/@types/validate-npm-package-name/-/validate-npm-package-name-4.0.2.tgz", {}, "sha512-lrpDziQipxCEeK5kWxvljWYhUvOiB2A9izZd9B2AFarYAkqZshb4lPbRs7zKEic6eGtH8V/2qJW+dPp9OtF6bw=="],
|
"@types/validate-npm-package-name": ["@types/validate-npm-package-name@4.0.2", "https://registry.npmmirror.com/@types/validate-npm-package-name/-/validate-npm-package-name-4.0.2.tgz", {}, "sha512-lrpDziQipxCEeK5kWxvljWYhUvOiB2A9izZd9B2AFarYAkqZshb4lPbRs7zKEic6eGtH8V/2qJW+dPp9OtF6bw=="],
|
||||||
|
|
||||||
"@types/webxr": ["@types/webxr@0.5.24", "", {}, "sha512-h8fgEd/DpoS9CBrjEQXR+dIDraopAEfu4wYVNY2tEPwk60stPWhvZMf4Foo5FakuQ7HFZoa8WceaWFervK2Ovg=="],
|
|
||||||
|
|
||||||
"@uiw/codemirror-extensions-basic-setup": ["@uiw/codemirror-extensions-basic-setup@4.25.9", "", { "dependencies": { "@codemirror/autocomplete": "^6.0.0", "@codemirror/commands": "^6.0.0", "@codemirror/language": "^6.0.0", "@codemirror/lint": "^6.0.0", "@codemirror/search": "^6.0.0", "@codemirror/state": "^6.0.0", "@codemirror/view": "^6.0.0" } }, "sha512-QFAqr+pu6lDmNpAlecODcF49TlsrZ0bj15zPzfhiqSDl+Um3EsDLFLppixC7kFLn+rdDM2LTvVjn5CPvefpRgw=="],
|
"@uiw/codemirror-extensions-basic-setup": ["@uiw/codemirror-extensions-basic-setup@4.25.9", "", { "dependencies": { "@codemirror/autocomplete": "^6.0.0", "@codemirror/commands": "^6.0.0", "@codemirror/language": "^6.0.0", "@codemirror/lint": "^6.0.0", "@codemirror/search": "^6.0.0", "@codemirror/state": "^6.0.0", "@codemirror/view": "^6.0.0" } }, "sha512-QFAqr+pu6lDmNpAlecODcF49TlsrZ0bj15zPzfhiqSDl+Um3EsDLFLppixC7kFLn+rdDM2LTvVjn5CPvefpRgw=="],
|
||||||
|
|
||||||
"@uiw/react-codemirror": ["@uiw/react-codemirror@4.25.9", "", { "dependencies": { "@babel/runtime": "^7.18.6", "@codemirror/commands": "^6.1.0", "@codemirror/state": "^6.1.1", "@codemirror/theme-one-dark": "^6.0.0", "@uiw/codemirror-extensions-basic-setup": "4.25.9", "codemirror": "^6.0.0" }, "peerDependencies": { "@codemirror/view": ">=6.0.0", "react": ">=17.0.0", "react-dom": ">=17.0.0" } }, "sha512-HftqCBUYShAOH0pGi1CHP8vfm5L8fQ3+0j0VI6lQD6QpK+UBu3J7nxfEN5O/BXMilMNf9ZyFJRvRcuMMOLHMng=="],
|
"@uiw/react-codemirror": ["@uiw/react-codemirror@4.25.9", "", { "dependencies": { "@babel/runtime": "^7.18.6", "@codemirror/commands": "^6.1.0", "@codemirror/state": "^6.1.1", "@codemirror/theme-one-dark": "^6.0.0", "@uiw/codemirror-extensions-basic-setup": "4.25.9", "codemirror": "^6.0.0" }, "peerDependencies": { "@codemirror/view": ">=6.0.0", "react": ">=17.0.0", "react-dom": ">=17.0.0" } }, "sha512-HftqCBUYShAOH0pGi1CHP8vfm5L8fQ3+0j0VI6lQD6QpK+UBu3J7nxfEN5O/BXMilMNf9ZyFJRvRcuMMOLHMng=="],
|
||||||
@@ -741,8 +729,6 @@
|
|||||||
|
|
||||||
"@vitejs/plugin-react": ["@vitejs/plugin-react@5.2.0", "", { "dependencies": { "@babel/core": "^7.29.0", "@babel/plugin-transform-react-jsx-self": "^7.27.1", "@babel/plugin-transform-react-jsx-source": "^7.27.1", "@rolldown/pluginutils": "1.0.0-rc.3", "@types/babel__core": "^7.20.5", "react-refresh": "^0.18.0" }, "peerDependencies": { "vite": "^4.2.0 || ^5.0.0 || ^6.0.0 || ^7.0.0 || ^8.0.0" } }, "sha512-YmKkfhOAi3wsB1PhJq5Scj3GXMn3WvtQ/JC0xoopuHoXSdmtdStOpFrYaT1kie2YgFBcIe64ROzMYRjCrYOdYw=="],
|
"@vitejs/plugin-react": ["@vitejs/plugin-react@5.2.0", "", { "dependencies": { "@babel/core": "^7.29.0", "@babel/plugin-transform-react-jsx-self": "^7.27.1", "@babel/plugin-transform-react-jsx-source": "^7.27.1", "@rolldown/pluginutils": "1.0.0-rc.3", "@types/babel__core": "^7.20.5", "react-refresh": "^0.18.0" }, "peerDependencies": { "vite": "^4.2.0 || ^5.0.0 || ^6.0.0 || ^7.0.0 || ^8.0.0" } }, "sha512-YmKkfhOAi3wsB1PhJq5Scj3GXMn3WvtQ/JC0xoopuHoXSdmtdStOpFrYaT1kie2YgFBcIe64ROzMYRjCrYOdYw=="],
|
||||||
|
|
||||||
"@webgpu/types": ["@webgpu/types@0.1.71", "", {}, "sha512-mMy8/ODcKhab808co15eW+yN+HgXoQxRQHTiBV9Mrvl1r0ufnid7YOcI+gi4eUWSWl9ezD6TW2KXccrL8HCh2A=="],
|
|
||||||
|
|
||||||
"accepts": ["accepts@2.0.0", "https://registry.npmmirror.com/accepts/-/accepts-2.0.0.tgz", { "dependencies": { "mime-types": "^3.0.0", "negotiator": "^1.0.0" } }, "sha512-5cvg6CtKwfgdmVqY1WIiXKc3Q1bkRqGLi+2W/6ao+6Y7gu/RCwRuAhGEzh5B4KlszSuTLgZYuqFqo5bImjNKng=="],
|
"accepts": ["accepts@2.0.0", "https://registry.npmmirror.com/accepts/-/accepts-2.0.0.tgz", { "dependencies": { "mime-types": "^3.0.0", "negotiator": "^1.0.0" } }, "sha512-5cvg6CtKwfgdmVqY1WIiXKc3Q1bkRqGLi+2W/6ao+6Y7gu/RCwRuAhGEzh5B4KlszSuTLgZYuqFqo5bImjNKng=="],
|
||||||
|
|
||||||
"agent-base": ["agent-base@7.1.4", "https://registry.npmmirror.com/agent-base/-/agent-base-7.1.4.tgz", {}, "sha512-MnA+YT8fwfJPgBx3m60MNqakm30XOkyIoH1y6huTQvC0PwZG7ki8NacLBcrPbNoo8vEZy7Jpuk7+jMO+CUovTQ=="],
|
"agent-base": ["agent-base@7.1.4", "https://registry.npmmirror.com/agent-base/-/agent-base-7.1.4.tgz", {}, "sha512-MnA+YT8fwfJPgBx3m60MNqakm30XOkyIoH1y6huTQvC0PwZG7ki8NacLBcrPbNoo8vEZy7Jpuk7+jMO+CUovTQ=="],
|
||||||
@@ -1343,8 +1329,6 @@
|
|||||||
|
|
||||||
"mermaid": ["mermaid@11.16.0", "", { "dependencies": { "@braintree/sanitize-url": "^7.1.2", "@iconify/utils": "^3.0.2", "@mermaid-js/parser": "^1.2.0", "@types/d3": "^7.4.3", "@upsetjs/venn.js": "^2.0.0", "cytoscape": "^3.33.3", "cytoscape-cose-bilkent": "^4.1.0", "cytoscape-fcose": "^2.2.0", "d3": "^7.9.0", "d3-sankey": "^0.12.3", "dagre-d3-es": "7.0.14", "dayjs": "^1.11.20", "dompurify": "^3.3.3", "es-toolkit": "^1.45.1", "katex": "^0.16.45", "khroma": "^2.1.0", "marked": "^16.3.0", "roughjs": "^4.6.6", "stylis": "^4.3.6", "ts-dedent": "^2.2.0", "uuid": "^11.1.0 || ^12 || ^13 || ^14.0.0" } }, "sha512-Zvm3kbstgdpvIJPPItlL7fppIZ3kibvc1oZIGxdvk9t6UFz6flv+Jw7FtRGKwfcI8OckmH04LqG6LlS6X4B1pA=="],
|
"mermaid": ["mermaid@11.16.0", "", { "dependencies": { "@braintree/sanitize-url": "^7.1.2", "@iconify/utils": "^3.0.2", "@mermaid-js/parser": "^1.2.0", "@types/d3": "^7.4.3", "@upsetjs/venn.js": "^2.0.0", "cytoscape": "^3.33.3", "cytoscape-cose-bilkent": "^4.1.0", "cytoscape-fcose": "^2.2.0", "d3": "^7.9.0", "d3-sankey": "^0.12.3", "dagre-d3-es": "7.0.14", "dayjs": "^1.11.20", "dompurify": "^3.3.3", "es-toolkit": "^1.45.1", "katex": "^0.16.45", "khroma": "^2.1.0", "marked": "^16.3.0", "roughjs": "^4.6.6", "stylis": "^4.3.6", "ts-dedent": "^2.2.0", "uuid": "^11.1.0 || ^12 || ^13 || ^14.0.0" } }, "sha512-Zvm3kbstgdpvIJPPItlL7fppIZ3kibvc1oZIGxdvk9t6UFz6flv+Jw7FtRGKwfcI8OckmH04LqG6LlS6X4B1pA=="],
|
||||||
|
|
||||||
"meshoptimizer": ["meshoptimizer@0.22.0", "", {}, "sha512-IebiK79sqIy+E4EgOr+CAw+Ke8hAspXKzBd0JdgEmPHiAwmvEj2S4h1rfvo+o/BnfEYd/jAOg5IeeIjzlzSnDg=="],
|
|
||||||
|
|
||||||
"micromark": ["micromark@4.0.2", "", { "dependencies": { "@types/debug": "^4.0.0", "debug": "^4.0.0", "decode-named-character-reference": "^1.0.0", "devlop": "^1.0.0", "micromark-core-commonmark": "^2.0.0", "micromark-factory-space": "^2.0.0", "micromark-util-character": "^2.0.0", "micromark-util-chunked": "^2.0.0", "micromark-util-combine-extensions": "^2.0.0", "micromark-util-decode-numeric-character-reference": "^2.0.0", "micromark-util-encode": "^2.0.0", "micromark-util-normalize-identifier": "^2.0.0", "micromark-util-resolve-all": "^2.0.0", "micromark-util-sanitize-uri": "^2.0.0", "micromark-util-subtokenize": "^2.0.0", "micromark-util-symbol": "^2.0.0", "micromark-util-types": "^2.0.0" } }, "sha512-zpe98Q6kvavpCr1NPVSCMebCKfD7CA2NqZ+rykeNhONIJBpc1tFKt9hucLGwha3jNTNI8lHpctWJWoimVF4PfA=="],
|
"micromark": ["micromark@4.0.2", "", { "dependencies": { "@types/debug": "^4.0.0", "debug": "^4.0.0", "decode-named-character-reference": "^1.0.0", "devlop": "^1.0.0", "micromark-core-commonmark": "^2.0.0", "micromark-factory-space": "^2.0.0", "micromark-util-character": "^2.0.0", "micromark-util-chunked": "^2.0.0", "micromark-util-combine-extensions": "^2.0.0", "micromark-util-decode-numeric-character-reference": "^2.0.0", "micromark-util-encode": "^2.0.0", "micromark-util-normalize-identifier": "^2.0.0", "micromark-util-resolve-all": "^2.0.0", "micromark-util-sanitize-uri": "^2.0.0", "micromark-util-subtokenize": "^2.0.0", "micromark-util-symbol": "^2.0.0", "micromark-util-types": "^2.0.0" } }, "sha512-zpe98Q6kvavpCr1NPVSCMebCKfD7CA2NqZ+rykeNhONIJBpc1tFKt9hucLGwha3jNTNI8lHpctWJWoimVF4PfA=="],
|
||||||
|
|
||||||
"micromark-core-commonmark": ["micromark-core-commonmark@2.0.3", "", { "dependencies": { "decode-named-character-reference": "^1.0.0", "devlop": "^1.0.0", "micromark-factory-destination": "^2.0.0", "micromark-factory-label": "^2.0.0", "micromark-factory-space": "^2.0.0", "micromark-factory-title": "^2.0.0", "micromark-factory-whitespace": "^2.0.0", "micromark-util-character": "^2.0.0", "micromark-util-chunked": "^2.0.0", "micromark-util-classify-character": "^2.0.0", "micromark-util-html-tag-name": "^2.0.0", "micromark-util-normalize-identifier": "^2.0.0", "micromark-util-resolve-all": "^2.0.0", "micromark-util-subtokenize": "^2.0.0", "micromark-util-symbol": "^2.0.0", "micromark-util-types": "^2.0.0" } }, "sha512-RDBrHEMSxVFLg6xvnXmb1Ayr2WzLAWjeSATAoxwKYJV94TeNavgoIdA0a9ytzDSVzBy2YKFK+emCPOEibLeCrg=="],
|
"micromark-core-commonmark": ["micromark-core-commonmark@2.0.3", "", { "dependencies": { "decode-named-character-reference": "^1.0.0", "devlop": "^1.0.0", "micromark-factory-destination": "^2.0.0", "micromark-factory-label": "^2.0.0", "micromark-factory-space": "^2.0.0", "micromark-factory-title": "^2.0.0", "micromark-factory-whitespace": "^2.0.0", "micromark-util-character": "^2.0.0", "micromark-util-chunked": "^2.0.0", "micromark-util-classify-character": "^2.0.0", "micromark-util-html-tag-name": "^2.0.0", "micromark-util-normalize-identifier": "^2.0.0", "micromark-util-resolve-all": "^2.0.0", "micromark-util-subtokenize": "^2.0.0", "micromark-util-symbol": "^2.0.0", "micromark-util-types": "^2.0.0" } }, "sha512-RDBrHEMSxVFLg6xvnXmb1Ayr2WzLAWjeSATAoxwKYJV94TeNavgoIdA0a9ytzDSVzBy2YKFK+emCPOEibLeCrg=="],
|
||||||
@@ -1707,8 +1691,6 @@
|
|||||||
|
|
||||||
"tapable": ["tapable@2.3.3", "", {}, "sha512-uxc/zpqFg6x7C8vOE7lh6Lbda8eEL9zmVm/PLeTPBRhh1xCgdWaQ+J1CUieGpIfm2HdtsUpRv+HshiasBMcc6A=="],
|
"tapable": ["tapable@2.3.3", "", {}, "sha512-uxc/zpqFg6x7C8vOE7lh6Lbda8eEL9zmVm/PLeTPBRhh1xCgdWaQ+J1CUieGpIfm2HdtsUpRv+HshiasBMcc6A=="],
|
||||||
|
|
||||||
"three": ["three@0.180.0", "", {}, "sha512-o+qycAMZrh+TsE01GqWUxUIKR1AL0S8pq7zDkYOQw8GqfX8b8VoCKYUoHbhiX5j+7hr8XsuHDVU6+gkQJQKg9w=="],
|
|
||||||
|
|
||||||
"throttle-debounce": ["throttle-debounce@5.0.2", "https://registry.npmmirror.com/throttle-debounce/-/throttle-debounce-5.0.2.tgz", {}, "sha512-B71/4oyj61iNH0KeCamLuE2rmKuTO5byTOSVwECM5FA7TiAiAW+UqTKZ9ERueC4qvgSttUhdmq1mXC3kJqGX7A=="],
|
"throttle-debounce": ["throttle-debounce@5.0.2", "https://registry.npmmirror.com/throttle-debounce/-/throttle-debounce-5.0.2.tgz", {}, "sha512-B71/4oyj61iNH0KeCamLuE2rmKuTO5byTOSVwECM5FA7TiAiAW+UqTKZ9ERueC4qvgSttUhdmq1mXC3kJqGX7A=="],
|
||||||
|
|
||||||
"tiny-invariant": ["tiny-invariant@1.3.3", "https://registry.npmmirror.com/tiny-invariant/-/tiny-invariant-1.3.3.tgz", {}, "sha512-+FbBPE1o9QAYvviau/qC5SE3caw21q3xkvWKBtja5vgqOWIHHJ3ioaq1VPfn/Szqctz2bU/oYeKd9/z5BL+PVg=="],
|
"tiny-invariant": ["tiny-invariant@1.3.3", "https://registry.npmmirror.com/tiny-invariant/-/tiny-invariant-1.3.3.tgz", {}, "sha512-+FbBPE1o9QAYvviau/qC5SE3caw21q3xkvWKBtja5vgqOWIHHJ3ioaq1VPfn/Szqctz2bU/oYeKd9/z5BL+PVg=="],
|
||||||
|
|||||||
@@ -48,14 +48,25 @@ export function AgentChatMessage({ item, theme, user, onRejectTool, onApproveToo
|
|||||||
return (
|
return (
|
||||||
<div className={`flex items-start gap-3 ${isUser ? "justify-end" : "justify-start"}`}>
|
<div className={`flex items-start gap-3 ${isUser ? "justify-end" : "justify-start"}`}>
|
||||||
{!isUser ? <AgentAvatar theme={theme} /> : null}
|
{!isUser ? <AgentAvatar theme={theme} /> : null}
|
||||||
<div className={`min-w-0 max-w-[82%] text-sm leading-6 ${isUser ? "text-right" : "text-left"}`} style={{ color }}>
|
<div
|
||||||
|
className={isUser ? "min-w-0 max-w-[82%] rounded-xl rounded-br-sm border px-3.5 py-2.5 text-left text-sm leading-6" : "min-w-0 flex-1 text-left text-sm leading-6"}
|
||||||
|
style={
|
||||||
|
isUser
|
||||||
|
? {
|
||||||
|
color,
|
||||||
|
background: `color-mix(in srgb, ${theme.node.text} 7%, ${theme.toolbar.panel})`,
|
||||||
|
borderColor: `color-mix(in srgb, ${theme.node.text} 14%, transparent)`,
|
||||||
|
}
|
||||||
|
: { color }
|
||||||
|
}
|
||||||
|
>
|
||||||
{isUser ? (
|
{isUser ? (
|
||||||
<div className="whitespace-pre-wrap break-words text-left">{item.text}</div>
|
<div className="whitespace-pre-wrap break-words">{item.text}</div>
|
||||||
) : (
|
) : (
|
||||||
<Streamdown animated isAnimating={!!item.streamId}>{item.text}</Streamdown>
|
<Streamdown animated isAnimating={!!item.streamId}>{item.text}</Streamdown>
|
||||||
)}
|
)}
|
||||||
{item.attachments?.length ? <AgentMessageAttachments attachments={item.attachments} /> : null}
|
{item.attachments?.length ? <AgentMessageAttachments attachments={item.attachments} /> : null}
|
||||||
{item.meta ? <div className="mt-1 text-[11px] opacity-45">{item.meta}</div> : null}
|
{item.meta ? <div className={`mt-1 text-[11px] opacity-45 ${isUser ? "text-right" : ""}`}>{item.meta}</div> : null}
|
||||||
</div>
|
</div>
|
||||||
{isUser ? <AgentUserAvatar user={user} theme={theme} /> : null}
|
{isUser ? <AgentUserAvatar user={user} theme={theme} /> : null}
|
||||||
</div>
|
</div>
|
||||||
@@ -303,7 +314,7 @@ function toolCardState(title: string, text: string, detail?: unknown) {
|
|||||||
if (objectField(detail, "status") === "noop" || /未生效|无需|没有找到|没有.*可|已存在/.test(raw)) return { label: "未生效", color: "#d97706", softBorder: "rgba(217,119,6,.22)", softBg: "rgba(217,119,6,.04)", icon: <CircleAlert className="size-4" />, isError: false };
|
if (objectField(detail, "status") === "noop" || /未生效|无需|没有找到|没有.*可|已存在/.test(raw)) return { label: "未生效", color: "#d97706", softBorder: "rgba(217,119,6,.22)", softBg: "rgba(217,119,6,.04)", icon: <CircleAlert className="size-4" />, isError: false };
|
||||||
if (/拒绝|取消/.test(raw) || lower.includes("rejected")) return { label: "拒绝执行", color: "#dc2626", softBorder: "rgba(220,38,38,.20)", softBg: "rgba(220,38,38,.04)", icon: <XCircle className="size-4" />, isError: true };
|
if (/拒绝|取消/.test(raw) || lower.includes("rejected")) return { label: "拒绝执行", color: "#dc2626", softBorder: "rgba(220,38,38,.20)", softBg: "rgba(220,38,38,.04)", icon: <XCircle className="size-4" />, isError: true };
|
||||||
if (/失败|错误/.test(raw) || lower.includes("failed") || lower.includes("error")) return { label: "执行失败", color: "#dc2626", softBorder: "rgba(220,38,38,.20)", softBg: "rgba(220,38,38,.04)", icon: <XCircle className="size-4" />, isError: true };
|
if (/失败|错误/.test(raw) || lower.includes("failed") || lower.includes("error")) return { label: "执行失败", color: "#dc2626", softBorder: "rgba(220,38,38,.20)", softBg: "rgba(220,38,38,.04)", icon: <XCircle className="size-4" />, isError: true };
|
||||||
if (/完成|成功/.test(raw) || lower.includes("completed") || lower.includes("succeeded")) return { label: tool === "canvas_apply_ops" || /画布操作/.test(title) ? "已批准执行" : "执行完成", color: "#16a34a", softBorder: "rgba(22,163,74,.20)", softBg: "rgba(22,163,74,.04)", icon: <CheckCircle2 className="size-4" />, isError: false };
|
if (/完成|成功/.test(raw) || lower.includes("completed") || lower.includes("succeeded")) return { label: tool === "canvas_apply_ops" || /画布操作/.test(title) ? "已批准执行" : "工具完成", color: "#16a34a", softBorder: "rgba(22,163,74,.20)", softBg: "rgba(22,163,74,.04)", icon: <CheckCircle2 className="size-4" />, isError: false };
|
||||||
return { label: "工具调用", color: "#2563eb", softBorder: "rgba(37,99,235,.20)", softBg: "rgba(37,99,235,.04)", icon: <Wrench className="size-4" />, isError: false };
|
return { label: "工具调用", color: "#2563eb", softBorder: "rgba(37,99,235,.20)", softBg: "rgba(37,99,235,.04)", icon: <Wrench className="size-4" />, isError: false };
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -2,10 +2,14 @@ import { Fragment, useCallback, useEffect, useMemo, useRef, useState } from "rea
|
|||||||
import { useNavigate, useSearchParams } from "react-router-dom";
|
import { useNavigate, useSearchParams } from "react-router-dom";
|
||||||
import { App, Button, Input, Segmented, Tooltip } from "antd";
|
import { App, Button, Input, Segmented, Tooltip } from "antd";
|
||||||
import copyToClipboard from "copy-to-clipboard";
|
import copyToClipboard from "copy-to-clipboard";
|
||||||
import { Copy, FolderOpen, History, KeyRound, Link2, LoaderCircle, PlugZap, Plus, RefreshCw, Square, Terminal, Trash2 } from "lucide-react";
|
import { ChevronDown, Copy, FolderOpen, History, KeyRound, Link2, LoaderCircle, MessageSquare, PlugZap, Plus, RefreshCw, Square, Terminal, Trash2 } from "lucide-react";
|
||||||
|
|
||||||
import { canvasThemes } from "@/lib/canvas-theme";
|
import { canvasThemes } from "@/lib/canvas-theme";
|
||||||
|
import { imageMetadata } from "@/lib/canvas/canvas-node-factory";
|
||||||
|
import { fitNodeSize } from "@/lib/canvas/canvas-node-size";
|
||||||
|
import { readImageMeta } from "@/lib/image-utils";
|
||||||
import { randomId } from "@/lib/utils";
|
import { randomId } from "@/lib/utils";
|
||||||
|
import { uploadImage } from "@/services/image-storage";
|
||||||
import { useThemeStore } from "@/stores/use-theme-store";
|
import { useThemeStore } from "@/stores/use-theme-store";
|
||||||
import { useUserStore } from "@/stores/use-user-store";
|
import { useUserStore } from "@/stores/use-user-store";
|
||||||
import { useShallow } from "zustand/react/shallow";
|
import { useShallow } from "zustand/react/shallow";
|
||||||
@@ -16,6 +20,7 @@ import { AgentChatComposer, AgentChatMessage, AgentPanelTabs, AgentPendingToolCa
|
|||||||
|
|
||||||
const MAX_ATTACHMENTS = 6;
|
const MAX_ATTACHMENTS = 6;
|
||||||
const MAX_ATTACHMENT_PAYLOAD_BYTES = 28 * 1024 * 1024;
|
const MAX_ATTACHMENT_PAYLOAD_BYTES = 28 * 1024 * 1024;
|
||||||
|
const SCROLL_BOTTOM_THRESHOLD = 48;
|
||||||
const DEFAULT_AGENT_URL = "http://127.0.0.1:17371";
|
const DEFAULT_AGENT_URL = "http://127.0.0.1:17371";
|
||||||
const AGENT_CONNECT_STEPS = [
|
const AGENT_CONNECT_STEPS = [
|
||||||
{ title: "方式一:在 Codex 中使用插件", text: "在 Codex app 安装 Infinite Canvas 插件后,通过插件启动画布,插件会自动启动本地 Agent 并带上连接信息。" },
|
{ title: "方式一:在 Codex 中使用插件", text: "在 Codex app 安装 Infinite Canvas 插件后,通过插件启动画布,插件会自动启动本地 Agent 并带上连接信息。" },
|
||||||
@@ -27,7 +32,9 @@ const AGENT_MCP_REMOVE_COMMAND = "codex mcp remove infinite-canvas";
|
|||||||
type AgentEventPayload = {
|
type AgentEventPayload = {
|
||||||
agent?: string;
|
agent?: string;
|
||||||
type?: string;
|
type?: string;
|
||||||
|
threadId?: string;
|
||||||
thread_id?: string;
|
thread_id?: string;
|
||||||
|
turn_id?: string;
|
||||||
item?: AgentEventItem;
|
item?: AgentEventItem;
|
||||||
error?: { message?: string };
|
error?: { message?: string };
|
||||||
message?: string;
|
message?: string;
|
||||||
@@ -40,6 +47,10 @@ type AgentWorkspace = { workspacePath: string; activeThreadId?: string };
|
|||||||
type AgentThreadsResponse = { ok?: boolean; workspace?: AgentWorkspace; data?: AgentThreadSummary[] };
|
type AgentThreadsResponse = { ok?: boolean; workspace?: AgentWorkspace; data?: AgentThreadSummary[] };
|
||||||
type AgentThreadResponse = { ok?: boolean; workspace?: AgentWorkspace; thread?: AgentThreadSummary; messages?: AgentChatItem[] };
|
type AgentThreadResponse = { ok?: boolean; workspace?: AgentWorkspace; thread?: AgentThreadSummary; messages?: AgentChatItem[] };
|
||||||
type AgentConfigResponse = { ok?: boolean; url?: string; token?: string; hasToken?: boolean };
|
type AgentConfigResponse = { ok?: boolean; url?: string; token?: string; hasToken?: boolean };
|
||||||
|
type AgentCodexState = { busy?: boolean; threadId?: string; turnId?: string };
|
||||||
|
type AgentHelloEvent = { ok?: boolean; clientId?: string; codex?: AgentCodexState };
|
||||||
|
type AgentWorkspaceEvent = { activeThreadId?: string; threadId?: string; emptyThread?: boolean };
|
||||||
|
type AgentChatEvent = { threadId?: string; sourceClientId?: string; message?: AgentChatItem };
|
||||||
|
|
||||||
export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { embedded?: boolean; headless?: boolean; autoConnect?: boolean }) {
|
export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { embedded?: boolean; headless?: boolean; autoConnect?: boolean }) {
|
||||||
const theme = canvasThemes[useThemeStore((state) => state.theme)];
|
const theme = canvasThemes[useThemeStore((state) => state.theme)];
|
||||||
@@ -80,6 +91,8 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
const pushEventLog = useAgentStore((state) => state.addEventLog);
|
const pushEventLog = useAgentStore((state) => state.addEventLog);
|
||||||
const clearEventLogs = useAgentStore((state) => state.clearEventLogs);
|
const clearEventLogs = useAgentStore((state) => state.clearEventLogs);
|
||||||
const listRef = useRef<HTMLDivElement>(null);
|
const listRef = useRef<HTMLDivElement>(null);
|
||||||
|
const followMessagesRef = useRef(true);
|
||||||
|
const [showScrollToBottom, setShowScrollToBottom] = useState(false);
|
||||||
const canvasContextRef = useRef<AgentCanvasContext | null>(useAgentStore.getState().canvasContext);
|
const canvasContextRef = useRef<AgentCanvasContext | null>(useAgentStore.getState().canvasContext);
|
||||||
const confirmToolsRef = useRef(confirmTools);
|
const confirmToolsRef = useRef(confirmTools);
|
||||||
const pendingToolRef = useRef<AgentPendingToolCall | null>(null);
|
const pendingToolRef = useRef<AgentPendingToolCall | null>(null);
|
||||||
@@ -88,28 +101,27 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
const errorLoggedRef = useRef(false);
|
const errorLoggedRef = useRef(false);
|
||||||
const attachmentUrlsRef = useRef(new Set<string>());
|
const attachmentUrlsRef = useRef(new Set<string>());
|
||||||
const clientIdRef = useRef(randomId());
|
const clientIdRef = useRef(randomId());
|
||||||
|
const loadThreadsSequenceRef = useRef(0);
|
||||||
const endpoint = useMemo(() => url.trim().replace(/\/$/, ""), [url]);
|
const endpoint = useMemo(() => url.trim().replace(/\/$/, ""), [url]);
|
||||||
const urlAgentAutoConnect = searchParams.has("agentUrl") && searchParams.has("agentToken");
|
const urlAgentAutoConnect = searchParams.has("agentUrl") && searchParams.has("agentToken");
|
||||||
const loadThreads = useCallback(async () => {
|
const loadThreads = useCallback(async (skipHistory = false) => {
|
||||||
if (!connectedRef.current && !useAgentStore.getState().connected) return;
|
if (!connectedRef.current && !useAgentStore.getState().connected) return;
|
||||||
|
const sequence = ++loadThreadsSequenceRef.current;
|
||||||
setAgentState({ loadingThreads: true });
|
setAgentState({ loadingThreads: true });
|
||||||
try {
|
try {
|
||||||
const data = await fetchAgentJson<AgentThreadsResponse>(endpoint, token, `/agent/codex/threads`);
|
const data = await fetchAgentJson<AgentThreadsResponse>(endpoint, token, `/agent/codex/threads`);
|
||||||
const nextThreadId = data.workspace?.activeThreadId || "";
|
const nextThreadId = data.workspace?.activeThreadId || "";
|
||||||
setAgentState({
|
let nextMessages: AgentChatItem[] = [];
|
||||||
threads: data.data || [],
|
if (nextThreadId && !skipHistory) {
|
||||||
workspacePath: data.workspace?.workspacePath || "",
|
|
||||||
activeThreadId: nextThreadId,
|
|
||||||
messages: [],
|
|
||||||
});
|
|
||||||
if (nextThreadId) {
|
|
||||||
const thread = await fetchAgentJson<AgentThreadResponse>(endpoint, token, `/agent/codex/threads/${encodeURIComponent(nextThreadId)}`);
|
const thread = await fetchAgentJson<AgentThreadResponse>(endpoint, token, `/agent/codex/threads/${encodeURIComponent(nextThreadId)}`);
|
||||||
setAgentState({ messages: normalizeHistoryMessages(thread.messages || []) });
|
nextMessages = normalizeHistoryMessages(thread.messages || []);
|
||||||
}
|
}
|
||||||
|
if (sequence !== loadThreadsSequenceRef.current) return;
|
||||||
|
setAgentState({ threads: data.data || [], workspacePath: data.workspace?.workspacePath || "", activeThreadId: nextThreadId, messages: nextMessages });
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
addEventLog("读取历史失败", error);
|
addEventLog("读取历史失败", error);
|
||||||
} finally {
|
} finally {
|
||||||
setAgentState({ loadingThreads: false });
|
if (sequence === loadThreadsSequenceRef.current) setAgentState({ loadingThreads: false });
|
||||||
}
|
}
|
||||||
}, [endpoint, setAgentState, token]);
|
}, [endpoint, setAgentState, token]);
|
||||||
|
|
||||||
@@ -134,9 +146,30 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
pendingToolRef.current = pendingTool;
|
pendingToolRef.current = pendingTool;
|
||||||
}, [pendingTool]);
|
}, [pendingTool]);
|
||||||
|
const updateScrollState = useCallback(() => {
|
||||||
|
const list = listRef.current;
|
||||||
|
if (!list) return;
|
||||||
|
const atBottom = list.scrollHeight - list.scrollTop - list.clientHeight <= SCROLL_BOTTOM_THRESHOLD;
|
||||||
|
followMessagesRef.current = atBottom;
|
||||||
|
setShowScrollToBottom(!atBottom);
|
||||||
|
}, []);
|
||||||
|
const scrollToBottom = useCallback((behavior: ScrollBehavior = "smooth") => {
|
||||||
|
const list = listRef.current;
|
||||||
|
if (!list) return;
|
||||||
|
followMessagesRef.current = true;
|
||||||
|
list.scrollTo({ top: list.scrollHeight, behavior });
|
||||||
|
setShowScrollToBottom(false);
|
||||||
|
}, []);
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
listRef.current?.scrollTo({ top: listRef.current.scrollHeight });
|
if (activeTab !== "chat") return;
|
||||||
}, [messages, pendingTool, waiting]);
|
const frame = requestAnimationFrame(() => scrollToBottom("auto"));
|
||||||
|
return () => cancelAnimationFrame(frame);
|
||||||
|
}, [activeTab, activeThreadId, scrollToBottom]);
|
||||||
|
useEffect(() => {
|
||||||
|
if (activeTab !== "chat") return;
|
||||||
|
const frame = requestAnimationFrame(() => (followMessagesRef.current ? scrollToBottom("auto") : updateScrollState()));
|
||||||
|
return () => cancelAnimationFrame(frame);
|
||||||
|
}, [activeTab, messages, pendingTool, scrollToBottom, updateScrollState, waiting]);
|
||||||
useEffect(() => () => attachmentUrlsRef.current.forEach((url) => URL.revokeObjectURL(url)), []);
|
useEffect(() => () => attachmentUrlsRef.current.forEach((url) => URL.revokeObjectURL(url)), []);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
@@ -144,13 +177,28 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
localStorage.setItem("canvas-agent-url", endpoint);
|
localStorage.setItem("canvas-agent-url", endpoint);
|
||||||
localStorage.setItem("canvas-agent-token", token);
|
localStorage.setItem("canvas-agent-token", token);
|
||||||
const clientId = clientIdRef.current;
|
const clientId = clientIdRef.current;
|
||||||
|
let eventQueue = Promise.resolve();
|
||||||
|
const enqueueEvent = (task: () => void | Promise<void>) => {
|
||||||
|
eventQueue = eventQueue.then(task).catch((error) => addEventLog("同步会话失败", error));
|
||||||
|
};
|
||||||
const source = new EventSource(`${endpoint}/events?token=${encodeURIComponent(token)}&clientId=${encodeURIComponent(clientId)}`);
|
const source = new EventSource(`${endpoint}/events?token=${encodeURIComponent(token)}&clientId=${encodeURIComponent(clientId)}`);
|
||||||
source.addEventListener("hello", () => {
|
source.addEventListener("hello", (event) => {
|
||||||
|
const busy = Boolean(parseEventData<AgentHelloEvent>(event)?.codex?.busy);
|
||||||
errorLoggedRef.current = false;
|
errorLoggedRef.current = false;
|
||||||
connectedRef.current = true;
|
connectedRef.current = true;
|
||||||
setAgentState({ connected: true, activity: "已连接", connectError: "", silentConnect: false, messages: useAgentStore.getState().messages.filter((item) => !isConnectionErrorMessage(item)) });
|
setAgentState({ connected: true, activity: busy ? "Codex 正在运行" : "已连接", waiting: busy, sending: false, connectError: "", silentConnect: false, messages: useAgentStore.getState().messages.filter((item) => !isConnectionErrorMessage(item)) });
|
||||||
if (!headless) message.success("本地 Agent 已连接");
|
if (!headless) message.success("本地 Agent 已连接");
|
||||||
void postState(endpoint, token, clientId, canvasContextRef.current?.snapshot || null);
|
void postState(endpoint, token, clientId, canvasContextRef.current?.snapshot || null);
|
||||||
|
if (document.visibilityState === "visible" && document.hasFocus()) void activateAgentClient(endpoint, token, clientId);
|
||||||
|
});
|
||||||
|
source.addEventListener("codex_state", (event) => {
|
||||||
|
const data = parseEventData<AgentCodexState>(event);
|
||||||
|
if (!data) return;
|
||||||
|
enqueueEvent(async () => {
|
||||||
|
const busy = Boolean(data.busy);
|
||||||
|
setAgentState({ activity: busy ? "Codex 正在运行" : "完成", waiting: busy, ...(busy ? {} : { sending: false }) });
|
||||||
|
if (!busy) await loadThreads();
|
||||||
|
});
|
||||||
});
|
});
|
||||||
source.addEventListener("tool_call", (event) => {
|
source.addEventListener("tool_call", (event) => {
|
||||||
const data = parseEventData<AgentPendingToolCall>(event);
|
const data = parseEventData<AgentPendingToolCall>(event);
|
||||||
@@ -158,21 +206,40 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
});
|
});
|
||||||
source.addEventListener("agent_event", (event) => {
|
source.addEventListener("agent_event", (event) => {
|
||||||
const data = parseEventData<AgentEventPayload>(event);
|
const data = parseEventData<AgentEventPayload>(event);
|
||||||
if (data) handleAgentEvent(data);
|
if (data) enqueueEvent(() => {
|
||||||
|
if (isCurrentThreadEvent(data)) handleAgentEvent(data);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
source.addEventListener("workspace_changed", (event) => {
|
||||||
|
const data = parseEventData<AgentWorkspaceEvent>(event);
|
||||||
|
if (!data) return;
|
||||||
|
enqueueEvent(async () => {
|
||||||
|
const nextThreadId = data.activeThreadId ?? data.threadId ?? "";
|
||||||
|
pendingToolRef.current = null;
|
||||||
|
setAgentState({ activeThreadId: nextThreadId, messages: [], pendingTool: null });
|
||||||
|
await loadThreads(data.emptyThread);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
source.addEventListener("chat_message", (event) => {
|
||||||
|
const data = parseEventData<AgentChatEvent>(event);
|
||||||
|
if (!data?.message) return;
|
||||||
|
enqueueEvent(() => {
|
||||||
|
if (!isCurrentThreadEvent(data)) return;
|
||||||
|
addMessage(data.message!);
|
||||||
|
});
|
||||||
});
|
});
|
||||||
source.addEventListener("agent_log", (event) => {
|
source.addEventListener("agent_log", (event) => {
|
||||||
const text = parseEventData<{ text?: unknown }>(event)?.text;
|
const text = parseEventData<{ text?: unknown }>(event)?.text;
|
||||||
addEventLog("日志", text, text);
|
addEventLog("日志", text, text);
|
||||||
});
|
});
|
||||||
source.addEventListener("agent_error", (event) => {
|
source.addEventListener("agent_error", (event) => {
|
||||||
const message = parseEventData<{ message?: unknown }>(event)?.message;
|
const data = parseEventData<AgentEventPayload>(event);
|
||||||
setAgentState({ activity: "出错", waiting: false });
|
if (!data) return;
|
||||||
addMessage({ role: "error", title: "错误", text: normalizeText(message) });
|
enqueueEvent(() => {
|
||||||
addEventLog("错误", message, message);
|
if (!isCurrentThreadEvent(data)) return;
|
||||||
});
|
addMessage({ role: "error", title: "错误", text: normalizeText(data.message) });
|
||||||
source.addEventListener("agent_done", () => {
|
addEventLog("错误", data.message, data.message);
|
||||||
setAgentState({ activity: "完成", waiting: false, sending: false });
|
});
|
||||||
void loadThreads();
|
|
||||||
});
|
});
|
||||||
source.onerror = () => {
|
source.onerror = () => {
|
||||||
const wasConnected = connectedRef.current;
|
const wasConnected = connectedRef.current;
|
||||||
@@ -193,6 +260,7 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
return () => {
|
return () => {
|
||||||
source.close();
|
source.close();
|
||||||
connectedRef.current = false;
|
connectedRef.current = false;
|
||||||
|
loadThreadsSequenceRef.current += 1;
|
||||||
};
|
};
|
||||||
}, [enabled, endpoint, loadThreads, message, setAgentState, token]);
|
}, [enabled, endpoint, loadThreads, message, setAgentState, token]);
|
||||||
|
|
||||||
@@ -200,6 +268,19 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
if (connected) void loadThreads();
|
if (connected) void loadThreads();
|
||||||
}, [connected, loadThreads]);
|
}, [connected, loadThreads]);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (!connected) return;
|
||||||
|
const activate = () => void activateAgentClient(endpoint, token, clientIdRef.current);
|
||||||
|
const activateVisible = () => {
|
||||||
|
if (document.visibilityState === "visible") activate();
|
||||||
|
};
|
||||||
|
window.addEventListener("focus", activate);
|
||||||
|
document.addEventListener("visibilitychange", activateVisible);
|
||||||
|
return () => {
|
||||||
|
window.removeEventListener("focus", activate);
|
||||||
|
document.removeEventListener("visibilitychange", activateVisible);
|
||||||
|
};
|
||||||
|
}, [connected, endpoint, token]);
|
||||||
const sendPrompt = async () => {
|
const sendPrompt = async () => {
|
||||||
const text = prompt.trim();
|
const text = prompt.trim();
|
||||||
const files = attachments;
|
const files = attachments;
|
||||||
@@ -209,27 +290,35 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
addMessage({ role: "error", title: "图片过大", text: "图片附件超过 30MB,请删减后再发送。" });
|
addMessage({ role: "error", title: "图片过大", text: "图片附件超过 30MB,请删减后再发送。" });
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
setAgentState({ activity: "发送中", sending: true, waiting: true });
|
setAgentState({ activity: "发送中", sending: true });
|
||||||
addMessage({ role: "user", text: text || "发送了图片", attachments: files });
|
const messageId = createId();
|
||||||
|
addMessage({ id: messageId, role: "user", text: text || "发送了图片", attachments: files });
|
||||||
addEventLog("用户发送", { text, attachments: files.map(({ name, type, size }) => ({ name, type, size })) });
|
addEventLog("用户发送", { text, attachments: files.map(({ name, type, size }) => ({ name, type, size })) });
|
||||||
try {
|
try {
|
||||||
const res = await fetch(`${endpoint}/agent/codex/turn?token=${encodeURIComponent(token)}`, {
|
const data = await fetchAgentJson<{ threadId?: string }>(endpoint, token, "/agent/codex/turn", {
|
||||||
method: "POST",
|
method: "POST",
|
||||||
headers: { "content-type": "application/json" },
|
headers: { "content-type": "application/json" },
|
||||||
body: JSON.stringify({ prompt: requestPrompt, threadId: useAgentStore.getState().activeThreadId || undefined, attachments: files.map(({ name, type, dataUrl }) => ({ name, type, dataUrl })) }),
|
body: JSON.stringify({
|
||||||
|
prompt: requestPrompt,
|
||||||
|
messageText: text || `发送了 ${files.length} 张图片`,
|
||||||
|
messageId,
|
||||||
|
clientId: clientIdRef.current,
|
||||||
|
threadId: useAgentStore.getState().activeThreadId || undefined,
|
||||||
|
attachments: files.map(({ id, name, type, size, width, height, dataUrl }) => ({ id, name, type, size, width, height, dataUrl })),
|
||||||
|
}),
|
||||||
});
|
});
|
||||||
if (!res.ok) throw new Error("本地 Agent 拒绝了请求");
|
|
||||||
const data = (await res.json()) as { threadId?: string };
|
|
||||||
if (data.threadId) setAgentState({ activeThreadId: data.threadId });
|
if (data.threadId) setAgentState({ activeThreadId: data.threadId });
|
||||||
addEventLog("本地 Agent 已接收", { status: res.status });
|
addEventLog("本地 Agent 已接收", { threadId: data.threadId });
|
||||||
files.forEach((item) => {
|
files.forEach((item) => {
|
||||||
URL.revokeObjectURL(item.url);
|
URL.revokeObjectURL(item.url);
|
||||||
attachmentUrlsRef.current.delete(item.url);
|
attachmentUrlsRef.current.delete(item.url);
|
||||||
});
|
});
|
||||||
setAgentState({ prompt: "", attachments: [] });
|
setAgentState({ prompt: "", attachments: [] });
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
setAgentState({ activity: "发送失败", waiting: false });
|
const text = error instanceof Error ? error.message : "发送失败";
|
||||||
addMessage({ role: "error", title: "发送失败", text: error instanceof Error ? error.message : "发送失败" });
|
const busy = text.includes("Codex 正在运行");
|
||||||
|
setAgentState({ activity: busy ? "Codex 正在运行" : "发送失败" });
|
||||||
|
addMessage({ role: "error", title: busy ? "任务仍在运行" : "发送失败", text });
|
||||||
addEventLog("发送失败", error);
|
addEventLog("发送失败", error);
|
||||||
} finally {
|
} finally {
|
||||||
setAgentState({ sending: false });
|
setAgentState({ sending: false });
|
||||||
@@ -240,11 +329,10 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
if (!connected || (!sending && !waiting)) return;
|
if (!connected || (!sending && !waiting)) return;
|
||||||
setAgentState({ activity: "停止中" });
|
setAgentState({ activity: "停止中" });
|
||||||
try {
|
try {
|
||||||
await fetch(`${endpoint}/agent/codex/interrupt?token=${encodeURIComponent(token)}`, { method: "POST", headers: { "content-type": "application/json" } });
|
await fetch(`${endpoint}/agent/codex/interrupt?token=${encodeURIComponent(token)}`, { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify({ threadId: useAgentStore.getState().activeThreadId || undefined }) });
|
||||||
setAgentState({ activity: "已停止", sending: false, waiting: false });
|
|
||||||
addEventLog("用户停止", {});
|
addEventLog("用户停止", {});
|
||||||
} catch {
|
} catch {
|
||||||
setAgentState({ activity: "就绪", sending: false, waiting: false });
|
setAgentState({ activity: "停止失败" });
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -256,9 +344,10 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
const next = await Promise.all(
|
const next = await Promise.all(
|
||||||
images.slice(0, Math.max(0, MAX_ATTACHMENTS - prev.length)).map(async (file) => {
|
images.slice(0, Math.max(0, MAX_ATTACHMENTS - prev.length)).map(async (file) => {
|
||||||
const dataUrl = await readDataUrl(file);
|
const dataUrl = await readDataUrl(file);
|
||||||
|
const meta = await readImageMeta(dataUrl);
|
||||||
const url = URL.createObjectURL(file);
|
const url = URL.createObjectURL(file);
|
||||||
attachmentUrlsRef.current.add(url);
|
attachmentUrlsRef.current.add(url);
|
||||||
return { id: createId(), name: file.name, type: file.type, size: file.size, url, dataUrl };
|
return { id: createId(), name: file.name, type: file.type, size: file.size, width: meta.width, height: meta.height, url, dataUrl };
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
const merged = [...prev, ...next];
|
const merged = [...prev, ...next];
|
||||||
@@ -286,13 +375,13 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
};
|
};
|
||||||
|
|
||||||
const handleToolCall = async (endpoint: string, token: string, payload: AgentPendingToolCall) => {
|
const handleToolCall = async (endpoint: string, token: string, payload: AgentPendingToolCall) => {
|
||||||
if (confirmToolsRef.current && payload.name === "canvas_apply_ops") {
|
if (confirmToolsRef.current && isCanvasWriteTool(payload.name)) {
|
||||||
if (pendingToolRef.current) {
|
if (pendingToolRef.current) {
|
||||||
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, error: "仍有待确认的画布工具调用" });
|
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, error: "仍有待确认的画布工具调用" });
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
pendingToolRef.current = payload;
|
pendingToolRef.current = payload;
|
||||||
setAgentState({ pendingTool: payload, activity: "等待确认", waiting: false });
|
setAgentState({ pendingTool: payload });
|
||||||
addEventLog("等待确认", payload, payload);
|
addEventLog("等待确认", payload, payload);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@@ -302,16 +391,13 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
const runToolCall = async (endpoint: string, token: string, payload: AgentPendingToolCall) => {
|
const runToolCall = async (endpoint: string, token: string, payload: AgentPendingToolCall) => {
|
||||||
if (isSiteTool(payload.name)) {
|
if (isSiteTool(payload.name)) {
|
||||||
try {
|
try {
|
||||||
setAgentState({ activity: SITE_TOOL_LABELS[payload.name], waiting: true });
|
|
||||||
addEventLog(toolName(payload.name), payload, payload);
|
addEventLog(toolName(payload.name), payload, payload);
|
||||||
const result = await runSiteTool(payload.name, payload.input || {}, navigate);
|
const result = await runSiteTool(payload.name, payload.input || {}, navigate, { canvasSnapshot: canvasContextRef.current?.snapshot || null });
|
||||||
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, result });
|
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, result });
|
||||||
setAgentState({ activity: "工具完成", waiting: true });
|
|
||||||
addEventLog(`${toolName(payload.name)}完成`, result, result);
|
addEventLog(`${toolName(payload.name)}完成`, result, result);
|
||||||
addMessage({ role: "tool", title: `${toolName(payload.name)}完成`, text: siteToolSummary(payload.name, result), detail: { requestId: payload.requestId, name: payload.name, input: payload.input, result } });
|
addMessage({ role: "tool", title: `${toolName(payload.name)}完成`, text: siteToolSummary(payload.name, result), detail: { requestId: payload.requestId, name: payload.name, input: payload.input, result } });
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
const message = error instanceof Error ? error.message : "工具执行失败";
|
const message = error instanceof Error ? error.message : "工具执行失败";
|
||||||
setAgentState({ activity: "工具失败", waiting: false });
|
|
||||||
addMessage({ role: "tool", title: "工具失败", text: message, detail: payload });
|
addMessage({ role: "tool", title: "工具失败", text: message, detail: payload });
|
||||||
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, error: message });
|
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, error: message });
|
||||||
}
|
}
|
||||||
@@ -319,9 +405,9 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
const input: { ops?: CanvasAgentOp[]; path?: string } = payload.input || {};
|
const input: { ops?: CanvasAgentOp[]; path?: string } = payload.input || {};
|
||||||
setAgentState({ activity: payload.name === "canvas_apply_ops" ? "执行画布操作" : payload.name === "site_navigate" ? "跳转页面" : "读取画布", waiting: true });
|
|
||||||
addEventLog(toolName(payload.name), payload, payload);
|
addEventLog(toolName(payload.name), payload, payload);
|
||||||
let result: unknown;
|
let result: unknown;
|
||||||
|
let appliedOps = input.ops || [];
|
||||||
if (payload.name === "site_navigate") {
|
if (payload.name === "site_navigate") {
|
||||||
const path = input.path || "/";
|
const path = input.path || "/";
|
||||||
navigate(path);
|
navigate(path);
|
||||||
@@ -329,25 +415,29 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
} else if (payload.name === "canvas_apply_ops") {
|
} else if (payload.name === "canvas_apply_ops") {
|
||||||
const context = canvasContextRef.current;
|
const context = canvasContextRef.current;
|
||||||
if (!context) throw new Error("当前不在画布页,请先用 site_navigate 打开画布");
|
if (!context) throw new Error("当前不在画布页,请先用 site_navigate 打开画布");
|
||||||
result = context.applyOps(input.ops || []);
|
result = context.applyOps(appliedOps);
|
||||||
void postState(endpoint, token, clientIdRef.current, result as CanvasAgentSnapshot);
|
void postState(endpoint, token, clientIdRef.current, result as CanvasAgentSnapshot);
|
||||||
|
} else if (payload.name === "canvas_create_attachment_nodes") {
|
||||||
|
const context = canvasContextRef.current;
|
||||||
|
if (!context) throw new Error("当前不在画布页,请先用 site_navigate 打开画布");
|
||||||
|
appliedOps = await attachmentNodeOps(endpoint, token, clientIdRef.current, payload.input?.nodes);
|
||||||
|
result = context.applyOps(appliedOps);
|
||||||
|
await postState(endpoint, token, clientIdRef.current, result as CanvasAgentSnapshot);
|
||||||
} else {
|
} else {
|
||||||
const snapshot = canvasContextRef.current?.snapshot;
|
const snapshot = canvasContextRef.current?.snapshot;
|
||||||
if (!snapshot) throw new Error("当前不在画布页,请先用 site_navigate 打开画布");
|
if (!snapshot) throw new Error("当前不在画布页,请先用 site_navigate 打开画布");
|
||||||
result = snapshot;
|
result = snapshot;
|
||||||
}
|
}
|
||||||
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, result });
|
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, result });
|
||||||
setAgentState({ activity: "工具完成", waiting: true });
|
|
||||||
addEventLog(`${toolName(payload.name)}完成`, result, result);
|
addEventLog(`${toolName(payload.name)}完成`, result, result);
|
||||||
addMessage({
|
addMessage({
|
||||||
role: "tool",
|
role: "tool",
|
||||||
title: `${toolName(payload.name)}完成`,
|
title: `${toolName(payload.name)}完成`,
|
||||||
text: payload.name === "canvas_apply_ops" ? summarizeCanvasAgentOps(input.ops || []) || "画布操作" : payload.name === "site_navigate" ? `已跳转到 ${input.path || "/"}` : "已完成",
|
text: appliedOps.length ? summarizeCanvasAgentOps(appliedOps) || "画布操作" : payload.name === "site_navigate" ? `已跳转到 ${input.path || "/"}` : "已完成",
|
||||||
detail: { requestId: payload.requestId, name: payload.name, input, result },
|
detail: { requestId: payload.requestId, name: payload.name, input, result },
|
||||||
});
|
});
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
const message = error instanceof Error ? error.message : "画布操作失败";
|
const message = error instanceof Error ? error.message : "画布操作失败";
|
||||||
setAgentState({ activity: "工具失败", waiting: false });
|
|
||||||
addMessage({ role: "tool", title: "工具失败", text: message, detail: payload });
|
addMessage({ role: "tool", title: "工具失败", text: message, detail: payload });
|
||||||
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, error: message });
|
await postToolResult(endpoint, token, clientIdRef.current, { requestId: payload.requestId, error: message });
|
||||||
}
|
}
|
||||||
@@ -356,7 +446,6 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
const rejectPendingTool = async () => {
|
const rejectPendingTool = async () => {
|
||||||
if (!pendingTool) return;
|
if (!pendingTool) return;
|
||||||
await postToolResult(endpoint, token, clientIdRef.current, { requestId: pendingTool.requestId, error: "用户取消了画布工具调用" });
|
await postToolResult(endpoint, token, clientIdRef.current, { requestId: pendingTool.requestId, error: "用户取消了画布工具调用" });
|
||||||
setAgentState({ activity: "已取消", waiting: false });
|
|
||||||
addMessage({ role: "tool", title: "拒绝执行", text: toolName(pendingTool.name), detail: { requestId: pendingTool.requestId, name: pendingTool.name, input: pendingTool.input } });
|
addMessage({ role: "tool", title: "拒绝执行", text: toolName(pendingTool.name), detail: { requestId: pendingTool.requestId, name: pendingTool.name, input: pendingTool.input } });
|
||||||
pendingToolRef.current = null;
|
pendingToolRef.current = null;
|
||||||
setAgentState({ pendingTool: null });
|
setAgentState({ pendingTool: null });
|
||||||
@@ -422,6 +511,7 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
}, [autoConnect, connected, enabled]);
|
}, [autoConnect, connected, enabled]);
|
||||||
|
|
||||||
function clearAgentSession(patch: Parameters<typeof setAgentState>[0] = {}) {
|
function clearAgentSession(patch: Parameters<typeof setAgentState>[0] = {}) {
|
||||||
|
loadThreadsSequenceRef.current += 1;
|
||||||
setAgentState({
|
setAgentState({
|
||||||
messages: [],
|
messages: [],
|
||||||
threads: [],
|
threads: [],
|
||||||
@@ -437,12 +527,11 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
}
|
}
|
||||||
|
|
||||||
const startNewThread = async () => {
|
const startNewThread = async () => {
|
||||||
if (!connected) return;
|
if (!connected || sending || waiting) return;
|
||||||
setAgentState({ loadingThreads: true });
|
setAgentState({ loadingThreads: true });
|
||||||
try {
|
try {
|
||||||
const data = await fetchAgentJson<AgentThreadResponse>(endpoint, token, "/agent/codex/threads/new", { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify({}) });
|
const data = await fetchAgentJson<AgentThreadResponse>(endpoint, token, "/agent/codex/threads/new", { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify({}) });
|
||||||
setAgentState({ activeThreadId: data.thread?.id || data.workspace?.activeThreadId || "", messages: [], activeTab: "chat", activity: "新对话" });
|
setAgentState({ activeThreadId: data.thread?.id || data.workspace?.activeThreadId || "", messages: [], activeTab: "chat", activity: "新对话" });
|
||||||
await loadThreads();
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
addEventLog("新建对话失败", error);
|
addEventLog("新建对话失败", error);
|
||||||
message.error(error instanceof Error ? error.message : "新建对话失败");
|
message.error(error instanceof Error ? error.message : "新建对话失败");
|
||||||
@@ -452,12 +541,11 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
};
|
};
|
||||||
|
|
||||||
const resumeThread = async (threadId: string) => {
|
const resumeThread = async (threadId: string) => {
|
||||||
if (!connected || !threadId) return;
|
if (!connected || !threadId || sending || waiting) return;
|
||||||
setAgentState({ loadingThreads: true });
|
setAgentState({ loadingThreads: true });
|
||||||
try {
|
try {
|
||||||
const data = await fetchAgentJson<AgentThreadResponse>(endpoint, token, `/agent/codex/threads/${encodeURIComponent(threadId)}/resume`, { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify({}) });
|
const data = await fetchAgentJson<AgentThreadResponse>(endpoint, token, `/agent/codex/threads/${encodeURIComponent(threadId)}/resume`, { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify({}) });
|
||||||
setAgentState({ activeThreadId: data.thread?.id || threadId, messages: normalizeHistoryMessages(data.messages || []), activeTab: "chat", activity: "已恢复会话" });
|
setAgentState({ activeThreadId: data.thread?.id || threadId, messages: normalizeHistoryMessages(data.messages || []), activeTab: "chat", activity: "已恢复会话" });
|
||||||
await loadThreads();
|
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
addEventLog("恢复对话失败", error);
|
addEventLog("恢复对话失败", error);
|
||||||
message.error(error instanceof Error ? error.message : "恢复对话失败");
|
message.error(error instanceof Error ? error.message : "恢复对话失败");
|
||||||
@@ -467,7 +555,7 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
};
|
};
|
||||||
|
|
||||||
const deleteThread = async (threadId: string) => {
|
const deleteThread = async (threadId: string) => {
|
||||||
if (!connected || !threadId) return;
|
if (!connected || !threadId || sending || waiting) return;
|
||||||
setAgentState({ loadingThreads: true });
|
setAgentState({ loadingThreads: true });
|
||||||
try {
|
try {
|
||||||
await fetchAgentJson(endpoint, token, `/agent/codex/threads/${encodeURIComponent(threadId)}/delete`, { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify({}) });
|
await fetchAgentJson(endpoint, token, `/agent/codex/threads/${encodeURIComponent(threadId)}/delete`, { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify({}) });
|
||||||
@@ -498,11 +586,12 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
});
|
});
|
||||||
};
|
};
|
||||||
|
|
||||||
const addMessage = (item: Omit<AgentChatItem, "id">) => {
|
const addMessage = (item: Omit<AgentChatItem, "id"> & { id?: string }) => {
|
||||||
const text = normalizeText(item.text);
|
const text = normalizeText(item.text);
|
||||||
if (!text && !item.attachments?.length) return;
|
if (!text && !item.attachments?.length) return;
|
||||||
const next = { ...item, id: `${Date.now()}-${Math.random()}`, text };
|
const next = { ...item, id: item.id || `${Date.now()}-${Math.random()}`, text } as AgentChatItem;
|
||||||
const currentMessages = useAgentStore.getState().messages;
|
const currentMessages = useAgentStore.getState().messages;
|
||||||
|
if (currentMessages.some((message) => message.id === next.id)) return;
|
||||||
if (next.streamId) {
|
if (next.streamId) {
|
||||||
const index = currentMessages.findIndex((message) => message.streamId === next.streamId);
|
const index = currentMessages.findIndex((message) => message.streamId === next.streamId);
|
||||||
if (index >= 0) {
|
if (index >= 0) {
|
||||||
@@ -527,15 +616,8 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
const handleAgentEvent = (event: AgentEventPayload) => {
|
const handleAgentEvent = (event: AgentEventPayload) => {
|
||||||
if (shouldLogAgentEvent(event)) addEventLog(eventTitle(event), event, event);
|
if (shouldLogAgentEvent(event)) addEventLog(eventTitle(event), event, event);
|
||||||
if (event.type === "thread.started" && event.thread_id) setAgentState({ activeThreadId: event.thread_id });
|
if (event.type === "thread.started" && event.thread_id) setAgentState({ activeThreadId: event.thread_id });
|
||||||
const nextActivity = activityText(event);
|
|
||||||
if (nextActivity) setAgentState({ activity: nextActivity });
|
|
||||||
if (event.type === "turn.started") setAgentState({ waiting: true });
|
|
||||||
if (event.type === "turn.completed" || event.type === "turn.failed" || event.type === "error") setAgentState({ waiting: false, sending: false });
|
|
||||||
const item = formatAgentEvent(event);
|
const item = formatAgentEvent(event);
|
||||||
if (item) {
|
if (item) addMessage(item);
|
||||||
if (item.role === "error") setAgentState({ waiting: false, sending: false });
|
|
||||||
addMessage(item);
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
const content = (
|
const content = (
|
||||||
@@ -545,7 +627,7 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
theme={theme}
|
theme={theme}
|
||||||
items={[
|
items={[
|
||||||
{ value: "setup", label: "连接", icon: <PlugZap className="size-3.5" /> },
|
{ value: "setup", label: "连接", icon: <PlugZap className="size-3.5" /> },
|
||||||
{ value: "chat", label: "对话" },
|
{ value: "chat", label: "对话", icon: <MessageSquare className="size-3.5" /> },
|
||||||
{ value: "history", label: "历史", icon: <History className="size-3.5" />, count: threads.length },
|
{ value: "history", label: "历史", icon: <History className="size-3.5" />, count: threads.length },
|
||||||
{ value: "log", label: "日志", icon: <Terminal className="size-3.5" />, count: eventLogs.length },
|
{ value: "log", label: "日志", icon: <Terminal className="size-3.5" />, count: eventLogs.length },
|
||||||
]}
|
]}
|
||||||
@@ -555,7 +637,7 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
}}
|
}}
|
||||||
right={
|
right={
|
||||||
<>
|
<>
|
||||||
<Button size="small" type="text" disabled={!connected || loadingThreads} icon={<Plus className="size-3.5" />} onClick={startNewThread}>
|
<Button size="small" type="text" disabled={!connected || loadingThreads || sending || waiting} icon={<Plus className="size-3.5" />} onClick={startNewThread}>
|
||||||
新对话
|
新对话
|
||||||
</Button>
|
</Button>
|
||||||
</>
|
</>
|
||||||
@@ -582,6 +664,7 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
activeThreadId={activeThreadId}
|
activeThreadId={activeThreadId}
|
||||||
workspacePath={workspacePath}
|
workspacePath={workspacePath}
|
||||||
loading={loadingThreads}
|
loading={loadingThreads}
|
||||||
|
busy={sending || waiting}
|
||||||
connected={connected}
|
connected={connected}
|
||||||
onRefresh={() => void loadThreads()}
|
onRefresh={() => void loadThreads()}
|
||||||
onNewThread={() => void startNewThread()}
|
onNewThread={() => void startNewThread()}
|
||||||
@@ -599,20 +682,35 @@ export function CanvasLocalAgentPanel({ embedded, headless, autoConnect }: { emb
|
|||||||
/>
|
/>
|
||||||
) : (
|
) : (
|
||||||
<>
|
<>
|
||||||
<div ref={listRef} className="thin-scrollbar min-h-0 flex-1 space-y-4 overflow-y-auto p-4">
|
<div className="relative min-h-0 flex-1">
|
||||||
{messages.map((item) => (
|
<div ref={listRef} className="thin-scrollbar h-full space-y-4 overflow-y-auto px-4 pb-12 pt-4" onScroll={updateScrollState}>
|
||||||
<AgentChatMessage key={item.id} item={agentMessageToChatMessage(item)} theme={theme} user={user} />
|
{messages.map((item) => (
|
||||||
))}
|
<AgentChatMessage key={item.id} item={agentMessageToChatMessage(item)} theme={theme} user={user} />
|
||||||
{pendingTool ? (
|
))}
|
||||||
<AgentPendingToolCard
|
{pendingTool ? (
|
||||||
summary={summarizeCanvasAgentOps(pendingTool.input?.ops || []) || toolName(pendingTool.name)}
|
<AgentPendingToolCard
|
||||||
detail={{ requestId: pendingTool.requestId, name: pendingTool.name, input: pendingTool.input }}
|
summary={summarizeCanvasAgentOps(pendingTool.input?.ops || []) || toolName(pendingTool.name)}
|
||||||
theme={theme}
|
detail={{ requestId: pendingTool.requestId, name: pendingTool.name, input: pendingTool.input }}
|
||||||
onReject={rejectPendingTool}
|
theme={theme}
|
||||||
onApprove={approvePendingTool}
|
onReject={rejectPendingTool}
|
||||||
/>
|
onApprove={approvePendingTool}
|
||||||
|
/>
|
||||||
|
) : null}
|
||||||
|
{waiting && !pendingTool ? <AgentWorkingMessage theme={theme} /> : null}
|
||||||
|
</div>
|
||||||
|
{showScrollToBottom ? (
|
||||||
|
<Tooltip title="滚动到底部" placement="left">
|
||||||
|
<Button
|
||||||
|
type="text"
|
||||||
|
shape="circle"
|
||||||
|
aria-label="滚动到底部"
|
||||||
|
className="!absolute bottom-3 left-1/2 z-10 !h-8 !w-8 !min-w-8 -translate-x-1/2 backdrop-blur transition hover:-translate-y-0.5"
|
||||||
|
style={{ background: theme.toolbar.panel, border: `1px solid ${theme.node.stroke}`, color: theme.node.text }}
|
||||||
|
icon={<ChevronDown className="size-4" />}
|
||||||
|
onClick={() => scrollToBottom()}
|
||||||
|
/>
|
||||||
|
</Tooltip>
|
||||||
) : null}
|
) : null}
|
||||||
{waiting && !pendingTool ? <AgentWorkingMessage theme={theme} /> : null}
|
|
||||||
</div>
|
</div>
|
||||||
<AgentChatComposer
|
<AgentChatComposer
|
||||||
prompt={prompt}
|
prompt={prompt}
|
||||||
@@ -864,6 +962,7 @@ function AgentHistoryView({
|
|||||||
activeThreadId,
|
activeThreadId,
|
||||||
workspacePath,
|
workspacePath,
|
||||||
loading,
|
loading,
|
||||||
|
busy,
|
||||||
connected,
|
connected,
|
||||||
onRefresh,
|
onRefresh,
|
||||||
onNewThread,
|
onNewThread,
|
||||||
@@ -875,6 +974,7 @@ function AgentHistoryView({
|
|||||||
activeThreadId: string;
|
activeThreadId: string;
|
||||||
workspacePath: string;
|
workspacePath: string;
|
||||||
loading: boolean;
|
loading: boolean;
|
||||||
|
busy: boolean;
|
||||||
connected: boolean;
|
connected: boolean;
|
||||||
onRefresh: () => void;
|
onRefresh: () => void;
|
||||||
onNewThread: () => void;
|
onNewThread: () => void;
|
||||||
@@ -899,7 +999,7 @@ function AgentHistoryView({
|
|||||||
<Button size="small" icon={<RefreshCw className={`size-3.5 ${loading ? "animate-spin" : ""}`} />} disabled={!connected || loading} onClick={onRefresh}>
|
<Button size="small" icon={<RefreshCw className={`size-3.5 ${loading ? "animate-spin" : ""}`} />} disabled={!connected || loading} onClick={onRefresh}>
|
||||||
刷新
|
刷新
|
||||||
</Button>
|
</Button>
|
||||||
<Button size="small" type="primary" icon={<Plus className="size-3.5" />} disabled={!connected || loading} onClick={onNewThread}>
|
<Button size="small" type="primary" icon={<Plus className="size-3.5" />} disabled={!connected || loading || busy} onClick={onNewThread}>
|
||||||
新对话
|
新对话
|
||||||
</Button>
|
</Button>
|
||||||
</div>
|
</div>
|
||||||
@@ -923,11 +1023,11 @@ function AgentHistoryView({
|
|||||||
</div>
|
</div>
|
||||||
<div className="flex shrink-0 items-center gap-1">
|
<div className="flex shrink-0 items-center gap-1">
|
||||||
<span className="text-[10px] opacity-55">{formatThreadTime(thread.updatedAt || thread.createdAt)}</span>
|
<span className="text-[10px] opacity-55">{formatThreadTime(thread.updatedAt || thread.createdAt)}</span>
|
||||||
<Button size="small" className="!h-6 !px-2" disabled={loading} onClick={() => onResumeThread(thread.id)}>
|
<Button size="small" className="!h-6 !px-2" disabled={loading || busy} onClick={() => onResumeThread(thread.id)}>
|
||||||
进入
|
进入
|
||||||
</Button>
|
</Button>
|
||||||
<Tooltip title="删除记录">
|
<Tooltip title="删除记录">
|
||||||
<Button size="small" danger type="text" className="!h-6 !w-6 !min-w-6" disabled={loading} icon={<Trash2 className="size-3.5" />} onClick={() => onDeleteThread(thread)} />
|
<Button size="small" danger type="text" className="!h-6 !w-6 !min-w-6" disabled={loading || busy} icon={<Trash2 className="size-3.5" />} onClick={() => onDeleteThread(thread)} />
|
||||||
</Tooltip>
|
</Tooltip>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
@@ -955,6 +1055,12 @@ async function postState(endpoint: string, token: string, clientId: string, snap
|
|||||||
} catch {}
|
} catch {}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function activateAgentClient(endpoint: string, token: string, clientId: string) {
|
||||||
|
try {
|
||||||
|
await fetch(`${endpoint}/canvas/activate?token=${encodeURIComponent(token)}&clientId=${encodeURIComponent(clientId)}`, { method: "POST" });
|
||||||
|
} catch {}
|
||||||
|
}
|
||||||
|
|
||||||
async function postToolResult(endpoint: string, token: string, clientId: string, body: { requestId: string; result?: unknown; error?: string }) {
|
async function postToolResult(endpoint: string, token: string, clientId: string, body: { requestId: string; result?: unknown; error?: string }) {
|
||||||
await fetch(`${endpoint}/canvas/result?token=${encodeURIComponent(token)}&clientId=${encodeURIComponent(clientId)}`, { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify(body) });
|
await fetch(`${endpoint}/canvas/result?token=${encodeURIComponent(token)}&clientId=${encodeURIComponent(clientId)}`, { method: "POST", headers: { "content-type": "application/json" }, body: JSON.stringify(body) });
|
||||||
}
|
}
|
||||||
@@ -985,6 +1091,11 @@ function parseEventData<T>(event: Event) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function isCurrentThreadEvent(event: { threadId?: string; thread_id?: string }) {
|
||||||
|
const threadId = event.threadId || event.thread_id || "";
|
||||||
|
return Boolean(threadId) && threadId === useAgentStore.getState().activeThreadId;
|
||||||
|
}
|
||||||
|
|
||||||
function formatLogText(logs: AgentEventLog[], context: AgentLogContext) {
|
function formatLogText(logs: AgentEventLog[], context: AgentLogContext) {
|
||||||
const head = [
|
const head = [
|
||||||
"Infinite Canvas Agent 诊断日志",
|
"Infinite Canvas Agent 诊断日志",
|
||||||
@@ -1025,17 +1136,6 @@ function usageText(event: AgentEventPayload) {
|
|||||||
return undefined;
|
return undefined;
|
||||||
}
|
}
|
||||||
|
|
||||||
function activityText(event: AgentEventPayload) {
|
|
||||||
const name = event.type || "";
|
|
||||||
if (name === "thread.started") return "已创建会话";
|
|
||||||
if (name === "turn.started") return "思考中";
|
|
||||||
if (name === "turn.completed") return "完成";
|
|
||||||
if (name === "turn.failed" || name === "error") return "出错";
|
|
||||||
if (name === "item.started") return isMcpToolItem(event.item) ? `调用${toolName(String(event.item?.tool || ""))}` : "执行步骤";
|
|
||||||
if (name === "item.completed") return isMcpToolItem(event.item) ? "工具完成" : "更新消息";
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
|
|
||||||
function eventTitle(event: AgentEventPayload) {
|
function eventTitle(event: AgentEventPayload) {
|
||||||
const item = event.item;
|
const item = event.item;
|
||||||
if (event.type === "thread.started") return "已创建 Codex 会话";
|
if (event.type === "thread.started") return "已创建 Codex 会话";
|
||||||
@@ -1064,6 +1164,7 @@ function toolName(name: string) {
|
|||||||
if (name === "canvas_get_selection") return "读取选区";
|
if (name === "canvas_get_selection") return "读取选区";
|
||||||
if (name === "canvas_export_snapshot") return "导出快照";
|
if (name === "canvas_export_snapshot") return "导出快照";
|
||||||
if (name === "canvas_create_node") return "创建节点";
|
if (name === "canvas_create_node") return "创建节点";
|
||||||
|
if (name === "canvas_create_attachment_nodes") return "添加附件图片";
|
||||||
if (name === "canvas_create_text_node") return "创建文本";
|
if (name === "canvas_create_text_node") return "创建文本";
|
||||||
if (name === "canvas_create_text_nodes") return "批量创建文本";
|
if (name === "canvas_create_text_nodes") return "批量创建文本";
|
||||||
if (name === "canvas_create_config_node") return "创建生成配置";
|
if (name === "canvas_create_config_node") return "创建生成配置";
|
||||||
@@ -1093,6 +1194,10 @@ function siteToolSummary(name: string, result: unknown) {
|
|||||||
if (name === "prompts_search") return `找到 ${numberField(data, "total")} 条提示词`;
|
if (name === "prompts_search") return `找到 ${numberField(data, "total")} 条提示词`;
|
||||||
if (name === "assets_list") return `共 ${numberField(data, "total")} 个资产`;
|
if (name === "assets_list") return `共 ${numberField(data, "total")} 个资产`;
|
||||||
if (name === "assets_add") return "已加入我的资产";
|
if (name === "assets_add") return "已加入我的资产";
|
||||||
|
if (name === "generation_get_status") {
|
||||||
|
const summary = data.summary && typeof data.summary === "object" ? (data.summary as Record<string, unknown>) : {};
|
||||||
|
return `共 ${numberField(data, "total")} 个任务,排队 ${numberField(summary, "queued")},运行中 ${numberField(summary, "running")},成功 ${numberField(summary, "succeeded")},失败 ${numberField(summary, "failed")}`;
|
||||||
|
}
|
||||||
if (name === "workbench_image_generate" || name === "workbench_video_generate") return typeof data.note === "string" ? data.note : "已在工作台执行";
|
if (name === "workbench_image_generate" || name === "workbench_video_generate") return typeof data.note === "string" ? data.note : "已在工作台执行";
|
||||||
if (name === "workbench_image_get_config" || name === "workbench_video_get_config") return "已读取工作台配置";
|
if (name === "workbench_image_get_config" || name === "workbench_video_get_config") return "已读取工作台配置";
|
||||||
return "已完成";
|
return "已完成";
|
||||||
@@ -1167,9 +1272,7 @@ function mergeAgentText(prev: string, next: string) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
function promptWithAttachments(text: string, attachments: AgentAttachment[]) {
|
function promptWithAttachments(text: string, attachments: AgentAttachment[]) {
|
||||||
if (!attachments.length) return text;
|
return text || (attachments.length ? "请处理上传的图片附件。" : "");
|
||||||
const names = attachments.map((item) => item.name).join("、");
|
|
||||||
return [text, `用户上传了 ${attachments.length} 张图片附件:${names}。`].filter(Boolean).join("\n\n");
|
|
||||||
}
|
}
|
||||||
|
|
||||||
function attachmentPayloadBytes(attachments: AgentAttachment[]) {
|
function attachmentPayloadBytes(attachments: AgentAttachment[]) {
|
||||||
@@ -1180,6 +1283,41 @@ function formatBytes(bytes: number) {
|
|||||||
return bytes > 1024 * 1024 ? `${(bytes / 1024 / 1024).toFixed(1)}MB` : `${Math.ceil(bytes / 1024)}KB`;
|
return bytes > 1024 * 1024 ? `${(bytes / 1024 / 1024).toFixed(1)}MB` : `${Math.ceil(bytes / 1024)}KB`;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function isCanvasWriteTool(name: string) {
|
||||||
|
return name === "canvas_apply_ops" || name === "canvas_create_attachment_nodes";
|
||||||
|
}
|
||||||
|
|
||||||
|
async function attachmentNodeOps(endpoint: string, token: string, clientId: string, value: unknown): Promise<CanvasAgentOp[]> {
|
||||||
|
const nodes = Array.isArray(value) ? value : [];
|
||||||
|
if (!nodes.length) throw new Error("没有可添加的图片附件");
|
||||||
|
return await Promise.all(
|
||||||
|
nodes.map(async (value) => {
|
||||||
|
const item = value as { id?: unknown; attachmentId?: unknown; title?: unknown; position?: unknown };
|
||||||
|
const id = String(item.id || "");
|
||||||
|
const attachmentId = String(item.attachmentId || "");
|
||||||
|
if (!id || !attachmentId) throw new Error("图片附件节点参数无效");
|
||||||
|
const res = await fetch(`${endpoint}/agent/attachments/${encodeURIComponent(attachmentId)}?token=${encodeURIComponent(token)}&clientId=${encodeURIComponent(clientId)}`);
|
||||||
|
if (!res.ok) {
|
||||||
|
const body = (await res.json().catch(() => null)) as { error?: string } | null;
|
||||||
|
throw new Error(body?.error || "读取图片附件失败");
|
||||||
|
}
|
||||||
|
const image = await uploadImage(await res.blob());
|
||||||
|
const size = fitNodeSize(image.width, image.height);
|
||||||
|
const position = item.position && typeof item.position === "object" ? (item.position as { x?: unknown; y?: unknown }) : {};
|
||||||
|
return {
|
||||||
|
type: "add_node" as const,
|
||||||
|
id,
|
||||||
|
nodeType: "image" as const,
|
||||||
|
title: String(item.title || "参考图"),
|
||||||
|
position: { x: Number(position.x) || 0, y: Number(position.y) || 0 },
|
||||||
|
width: size.width,
|
||||||
|
height: size.height,
|
||||||
|
metadata: imageMetadata(image),
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
async function fetchAgentJson<T>(endpoint: string, token: string, path: string, init?: RequestInit) {
|
async function fetchAgentJson<T>(endpoint: string, token: string, path: string, init?: RequestInit) {
|
||||||
const url = `${endpoint}${path}${path.includes("?") ? "&" : "?"}token=${encodeURIComponent(token)}`;
|
const url = `${endpoint}${path}${path.includes("?") ? "&" : "?"}token=${encodeURIComponent(token)}`;
|
||||||
const res = await fetch(url, init);
|
const res = await fetch(url, init);
|
||||||
|
|||||||
@@ -39,9 +39,11 @@ export function CanvasNodePromptPanel({ node, isRunning, onPromptChange, onConfi
|
|||||||
const isEditingExistingContent = hasTextContent || hasImageContent;
|
const isEditingExistingContent = hasTextContent || hasImageContent;
|
||||||
const [prompt, setPrompt] = useState(isEditingExistingContent ? "" : node.metadata?.prompt || "");
|
const [prompt, setPrompt] = useState(isEditingExistingContent ? "" : node.metadata?.prompt || "");
|
||||||
|
|
||||||
|
// 仅在切换到其它节点时重置输入框;同一节点生成完成后(内容写回自身导致 isEditingExistingContent 变化)保留用户输入
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
setPrompt(isEditingExistingContent ? "" : node.metadata?.prompt || "");
|
setPrompt(isEditingExistingContent ? "" : node.metadata?.prompt || "");
|
||||||
}, [isEditingExistingContent, node.id]);
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||||||
|
}, [node.id]);
|
||||||
|
|
||||||
const updatePrompt = (value: string) => {
|
const updatePrompt = (value: string) => {
|
||||||
setPrompt(value);
|
setPrompt(value);
|
||||||
@@ -52,7 +54,6 @@ export function CanvasNodePromptPanel({ node, isRunning, onPromptChange, onConfi
|
|||||||
const text = prompt.trim();
|
const text = prompt.trim();
|
||||||
if (!text || isRunning) return;
|
if (!text || isRunning) return;
|
||||||
onGenerate(node.id, mode, text);
|
onGenerate(node.id, mode, text);
|
||||||
setPrompt("");
|
|
||||||
};
|
};
|
||||||
|
|
||||||
return (
|
return (
|
||||||
|
|||||||
@@ -1,21 +1,21 @@
|
|||||||
import { useMemo, useRef, useState, type PointerEvent as ReactPointerEvent } from "react";
|
import { memo, useMemo, useRef, useState, type PointerEvent as ReactPointerEvent } from "react";
|
||||||
import { App, Empty, Input, Popconfirm, Select, Tag } from "antd";
|
import { App, Empty, Input, Popconfirm, Select, Spin, Tag } from "antd";
|
||||||
import { Check, ChevronRight, Download, FileText, Image as ImageIcon, ListChecks, Music2, Plus, Search, Settings2, Square, Trash2, Type, Video } from "lucide-react";
|
import { useQuery } from "@tanstack/react-query";
|
||||||
|
import { BookOpen, Check, ChevronRight, Download, Eye, FileText, Image as ImageIcon, ListChecks, Music2, Plus, Search, Settings2, Square, Trash2, Type, Video } from "lucide-react";
|
||||||
import { motion } from "motion/react";
|
import { motion } from "motion/react";
|
||||||
|
|
||||||
import { canvasThemes, type CanvasTheme } from "@/lib/canvas-theme";
|
import { canvasThemes, type CanvasTheme } from "@/lib/canvas-theme";
|
||||||
import { exportCanvasNodes } from "@/lib/canvas/canvas-export";
|
import { exportCanvasNodes } from "@/lib/canvas/canvas-export";
|
||||||
import { getNodeDefinition } from "@/lib/canvas/node-registry";
|
import { getNodeDefinition } from "@/lib/canvas/node-registry";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
|
import { PromptDetailDialog } from "@/pages/prompts/components/prompt-detail-dialog";
|
||||||
|
import { fetchSourcePrompts, personalPromptToPrompt, type Prompt } from "@/services/api/prompts";
|
||||||
import { uploadMediaFile } from "@/services/file-storage";
|
import { uploadMediaFile } from "@/services/file-storage";
|
||||||
import { uploadImage } from "@/services/image-storage";
|
import { uploadImage } from "@/services/image-storage";
|
||||||
import { useAssetStore, type Asset, type AssetKind } from "@/stores/use-asset-store";
|
import { useAssetStore, type Asset, type AssetKind } from "@/stores/use-asset-store";
|
||||||
import {
|
import { usePromptSourceStore } from "@/stores/use-prompt-source-store";
|
||||||
CANVAS_SIDE_PANEL_MAX_WIDTH,
|
import { usePromptStore } from "@/stores/use-prompt-store";
|
||||||
CANVAS_SIDE_PANEL_MIN_WIDTH,
|
import { CANVAS_SIDE_PANEL_MAX_WIDTH, CANVAS_SIDE_PANEL_MIN_WIDTH, CANVAS_SIDE_PANEL_MOTION_MS, useCanvasSidePanelStore } from "@/stores/use-canvas-side-panel-store";
|
||||||
CANVAS_SIDE_PANEL_MOTION_MS,
|
|
||||||
useCanvasSidePanelStore,
|
|
||||||
} from "@/stores/use-canvas-side-panel-store";
|
|
||||||
import { useThemeStore } from "@/stores/use-theme-store";
|
import { useThemeStore } from "@/stores/use-theme-store";
|
||||||
import { CanvasNodeType, type CanvasNodeData } from "@/types/canvas";
|
import { CanvasNodeType, type CanvasNodeData } from "@/types/canvas";
|
||||||
|
|
||||||
@@ -24,7 +24,7 @@ import type { InsertAssetPayload } from "./asset-picker-modal";
|
|||||||
const PANEL_MOTION_SECONDS = CANVAS_SIDE_PANEL_MOTION_MS / 1000;
|
const PANEL_MOTION_SECONDS = CANVAS_SIDE_PANEL_MOTION_MS / 1000;
|
||||||
const PANEL_EASE = [0.22, 1, 0.36, 1] as const;
|
const PANEL_EASE = [0.22, 1, 0.36, 1] as const;
|
||||||
|
|
||||||
type PanelTab = "canvas" | "assets";
|
type PanelTab = "canvas" | "assets" | "prompts";
|
||||||
|
|
||||||
type Props = {
|
type Props = {
|
||||||
nodes: CanvasNodeData[];
|
nodes: CanvasNodeData[];
|
||||||
@@ -100,8 +100,17 @@ export function CanvasSidePanel({ nodes, selectedNodeIds, onFocusNode, onInsertA
|
|||||||
<div className="flex items-center gap-5 px-4 pt-3.5">
|
<div className="flex items-center gap-5 px-4 pt-3.5">
|
||||||
<TabButton label="画布" active={tab === "canvas"} theme={theme} onClick={() => setTab("canvas")} />
|
<TabButton label="画布" active={tab === "canvas"} theme={theme} onClick={() => setTab("canvas")} />
|
||||||
<TabButton label="资产" active={tab === "assets"} theme={theme} onClick={() => setTab("assets")} />
|
<TabButton label="资产" active={tab === "assets"} theme={theme} onClick={() => setTab("assets")} />
|
||||||
|
<TabButton label="提示词库" active={tab === "prompts"} theme={theme} onClick={() => setTab("prompts")} />
|
||||||
|
</div>
|
||||||
|
<div className="mt-2 min-h-0 flex-1 overflow-hidden">
|
||||||
|
{tab === "canvas" ? (
|
||||||
|
<CanvasNodesTab nodes={nodes} selectedNodeIds={selectedNodeIds} onFocusNode={onFocusNode} theme={theme} />
|
||||||
|
) : tab === "assets" ? (
|
||||||
|
<CanvasAssetsTab onInsert={onInsertAsset} theme={theme} />
|
||||||
|
) : (
|
||||||
|
<CanvasPromptsTab onInsert={onInsertAsset} theme={theme} />
|
||||||
|
)}
|
||||||
</div>
|
</div>
|
||||||
<div className="mt-2 min-h-0 flex-1 overflow-hidden">{tab === "canvas" ? <CanvasNodesTab nodes={nodes} selectedNodeIds={selectedNodeIds} onFocusNode={onFocusNode} theme={theme} /> : <CanvasAssetsTab onInsert={onInsertAsset} theme={theme} />}</div>
|
|
||||||
<button type="button" className="absolute inset-y-0 right-0 z-40 w-4 translate-x-1/2 cursor-col-resize" onPointerDown={startResize} aria-label="调整左侧面板宽度" />
|
<button type="button" className="absolute inset-y-0 right-0 z-40 w-4 translate-x-1/2 cursor-col-resize" onPointerDown={startResize} aria-label="调整左侧面板宽度" />
|
||||||
</motion.aside>
|
</motion.aside>
|
||||||
</motion.div>
|
</motion.div>
|
||||||
@@ -278,7 +287,7 @@ function buildInsertPayload(asset: Asset): InsertAssetPayload {
|
|||||||
return { kind: "image", dataUrl: asset.data.dataUrl, storageKey: asset.data.storageKey, title: asset.title };
|
return { kind: "image", dataUrl: asset.data.dataUrl, storageKey: asset.data.storageKey, title: asset.title };
|
||||||
}
|
}
|
||||||
|
|
||||||
function CanvasAssetsTab({ onInsert, theme }: { onInsert: (payload: InsertAssetPayload) => void; theme: CanvasTheme }) {
|
const CanvasAssetsTab = memo(function CanvasAssetsTab({ onInsert, theme }: { onInsert: (payload: InsertAssetPayload) => void; theme: CanvasTheme }) {
|
||||||
const { message } = App.useApp();
|
const { message } = App.useApp();
|
||||||
const assets = useAssetStore((state) => state.assets);
|
const assets = useAssetStore((state) => state.assets);
|
||||||
const addAsset = useAssetStore((state) => state.addAsset);
|
const addAsset = useAssetStore((state) => state.addAsset);
|
||||||
@@ -363,7 +372,11 @@ function CanvasAssetsTab({ onInsert, theme }: { onInsert: (payload: InsertAssetP
|
|||||||
const isCollapsed = collapsed[group.kind];
|
const isCollapsed = collapsed[group.kind];
|
||||||
return (
|
return (
|
||||||
<div key={group.kind}>
|
<div key={group.kind}>
|
||||||
<button type="button" onClick={() => setCollapsed((prev) => ({ ...prev, [group.kind]: !prev[group.kind] }))} className="flex w-full items-center gap-1.5 rounded-md px-1.5 py-1.5 text-left text-xs font-semibold opacity-75 transition hover:opacity-100">
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={() => setCollapsed((prev) => ({ ...prev, [group.kind]: !prev[group.kind] }))}
|
||||||
|
className="flex w-full items-center gap-1.5 rounded-md px-1.5 py-1.5 text-left text-xs font-semibold opacity-75 transition hover:opacity-100"
|
||||||
|
>
|
||||||
<ChevronRight className={cn("size-3.5 transition-transform", !isCollapsed && "rotate-90")} />
|
<ChevronRight className={cn("size-3.5 transition-transform", !isCollapsed && "rotate-90")} />
|
||||||
<group.icon className="size-3.5" />
|
<group.icon className="size-3.5" />
|
||||||
<span>{group.label}</span>
|
<span>{group.label}</span>
|
||||||
@@ -386,7 +399,7 @@ function CanvasAssetsTab({ onInsert, theme }: { onInsert: (payload: InsertAssetP
|
|||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
});
|
||||||
|
|
||||||
function AssetCard({ asset, theme, onInsert, onRemove }: { asset: Asset; theme: CanvasTheme; onInsert: () => void; onRemove: () => void }) {
|
function AssetCard({ asset, theme, onInsert, onRemove }: { asset: Asset; theme: CanvasTheme; onInsert: () => void; onRemove: () => void }) {
|
||||||
return (
|
return (
|
||||||
@@ -423,3 +436,188 @@ function AssetCover({ asset }: { asset: Asset }) {
|
|||||||
}
|
}
|
||||||
return <img src={asset.coverUrl || asset.data.dataUrl} alt="" className="size-full object-cover transition duration-300 group-hover:scale-[1.04]" />;
|
return <img src={asset.coverUrl || asset.data.dataUrl} alt="" className="size-full object-cover transition duration-300 group-hover:scale-[1.04]" />;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
// 提示词库 Tab —— 按来源折叠分组,展开时按需加载,点击复制 / 插入文本节点
|
||||||
|
// ---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
const CanvasPromptsTab = memo(function CanvasPromptsTab({ onInsert, theme }: { onInsert: (payload: InsertAssetPayload) => void; theme: CanvasTheme }) {
|
||||||
|
const { message } = App.useApp();
|
||||||
|
const sources = usePromptSourceStore((state) => state.sources);
|
||||||
|
const personalPrompts = usePromptStore((state) => state.prompts);
|
||||||
|
const enabledSources = useMemo(() => sources.filter((source) => source.enabled), [sources]);
|
||||||
|
const [keyword, setKeyword] = useState("");
|
||||||
|
const [expanded, setExpanded] = useState<Record<string, boolean>>({ personal: true });
|
||||||
|
const [detail, setDetail] = useState<Prompt | null>(null);
|
||||||
|
|
||||||
|
const copyPrompt = async (prompt: string) => {
|
||||||
|
try {
|
||||||
|
await navigator.clipboard.writeText(prompt);
|
||||||
|
message.success("已复制提示词");
|
||||||
|
} catch {
|
||||||
|
message.error("复制失败");
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="flex h-full flex-col">
|
||||||
|
<div className="px-3 pb-2.5 pt-1">
|
||||||
|
<Input size="small" allowClear prefix={<Search className="size-3.5 text-stone-400" />} placeholder="搜索提示词" value={keyword} onChange={(e) => setKeyword(e.target.value)} />
|
||||||
|
</div>
|
||||||
|
<div className="min-h-0 flex-1 overflow-y-auto px-2 pb-3">
|
||||||
|
<div className="space-y-1">
|
||||||
|
<PersonalPromptGroup
|
||||||
|
items={personalPrompts.map(personalPromptToPrompt)}
|
||||||
|
keyword={keyword}
|
||||||
|
open={!!expanded.personal}
|
||||||
|
theme={theme}
|
||||||
|
onToggle={() => setExpanded((prev) => ({ ...prev, personal: !prev.personal }))}
|
||||||
|
onInsert={onInsert}
|
||||||
|
onView={setDetail}
|
||||||
|
/>
|
||||||
|
{enabledSources.length ? (
|
||||||
|
<>
|
||||||
|
{enabledSources.map((source) => (
|
||||||
|
<PromptSourceGroup
|
||||||
|
key={source.id}
|
||||||
|
sourceId={source.id}
|
||||||
|
sourceName={source.name}
|
||||||
|
keyword={keyword}
|
||||||
|
open={!!expanded[source.id]}
|
||||||
|
theme={theme}
|
||||||
|
onToggle={() => setExpanded((prev) => ({ ...prev, [source.id]: !prev[source.id] }))}
|
||||||
|
onInsert={onInsert}
|
||||||
|
onView={setDetail}
|
||||||
|
/>
|
||||||
|
))}
|
||||||
|
</>
|
||||||
|
) : personalPrompts.length === 0 ? <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} description="暂无提示词" className="pt-12" /> : null}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<PromptDetailDialog prompt={detail} onClose={() => setDetail(null)} onCopy={(prompt) => void copyPrompt(prompt)} />
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
});
|
||||||
|
|
||||||
|
function PersonalPromptGroup({ items, keyword, open, theme, onToggle, onInsert, onView }: { items: Prompt[]; keyword: string; open: boolean; theme: CanvasTheme; onToggle: () => void; onInsert: (payload: InsertAssetPayload) => void; onView: (prompt: Prompt) => void }) {
|
||||||
|
const showResults = open || !!keyword.trim();
|
||||||
|
const filtered = useMemo(() => {
|
||||||
|
const query = keyword.trim().toLowerCase();
|
||||||
|
return query ? items.filter((item) => [item.title, item.prompt, item.description, ...item.tags].join(" ").toLowerCase().includes(query)) : items;
|
||||||
|
}, [items, keyword]);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div>
|
||||||
|
<button type="button" onClick={onToggle} className="flex w-full items-center gap-1.5 rounded-md px-1.5 py-1.5 text-left text-xs font-semibold opacity-75 transition hover:opacity-100">
|
||||||
|
<ChevronRight className={cn("size-3.5 transition-transform", showResults && "rotate-90")} />
|
||||||
|
<BookOpen className="size-3.5" />
|
||||||
|
<span className="min-w-0 flex-1 truncate">我的提示词</span>
|
||||||
|
<span className="opacity-50">{filtered.length}</span>
|
||||||
|
</button>
|
||||||
|
{showResults ? (
|
||||||
|
<div className="space-y-1.5 px-1 pb-2 pt-1">
|
||||||
|
{filtered.map((item) => <PromptRow key={item.id} item={item} theme={theme} onInsert={() => onInsert({ kind: "text", content: item.prompt, title: item.title })} onView={() => onView(item)} />)}
|
||||||
|
{!filtered.length ? <div className="py-4 text-center text-xs opacity-40">{keyword.trim() ? "无匹配提示词" : "还没有保存提示词"}</div> : null}
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function PromptSourceGroup({
|
||||||
|
sourceId,
|
||||||
|
sourceName,
|
||||||
|
keyword,
|
||||||
|
open,
|
||||||
|
theme,
|
||||||
|
onToggle,
|
||||||
|
onInsert,
|
||||||
|
onView,
|
||||||
|
}: {
|
||||||
|
sourceId: string;
|
||||||
|
sourceName: string;
|
||||||
|
keyword: string;
|
||||||
|
open: boolean;
|
||||||
|
theme: CanvasTheme;
|
||||||
|
onToggle: () => void;
|
||||||
|
onInsert: (payload: InsertAssetPayload) => void;
|
||||||
|
onView: (prompt: Prompt) => void;
|
||||||
|
}) {
|
||||||
|
// 展开过一次即缓存,避免收起后重复请求;搜索命中时也需要拿到数据来计数。
|
||||||
|
const showResults = open || !!keyword.trim();
|
||||||
|
const query = useQuery({ queryKey: ["side-panel-prompts", sourceId], queryFn: () => fetchSourcePrompts(sourceId), enabled: showResults, staleTime: 1000 * 60 * 60 });
|
||||||
|
|
||||||
|
const filtered = useMemo(() => {
|
||||||
|
const items = query.data || [];
|
||||||
|
const q = keyword.trim().toLowerCase();
|
||||||
|
if (!q) return items;
|
||||||
|
return items.filter((item) => [item.title, item.prompt, ...item.tags].join(" ").toLowerCase().includes(q));
|
||||||
|
}, [query.data, keyword]);
|
||||||
|
|
||||||
|
const insertPrompt = (item: Prompt) => onInsert({ kind: "text", content: item.prompt, title: item.title });
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div>
|
||||||
|
<button type="button" onClick={onToggle} className="flex w-full items-center gap-1.5 rounded-md px-1.5 py-1.5 text-left text-xs font-semibold opacity-75 transition hover:opacity-100">
|
||||||
|
<ChevronRight className={cn("size-3.5 transition-transform", showResults && "rotate-90")} />
|
||||||
|
<BookOpen className="size-3.5" />
|
||||||
|
<span className="min-w-0 flex-1 truncate">{sourceName}</span>
|
||||||
|
{showResults && query.isSuccess ? <span className="opacity-50">{filtered.length}</span> : null}
|
||||||
|
</button>
|
||||||
|
{showResults ? (
|
||||||
|
<div className="px-1 pb-2 pt-1">
|
||||||
|
{query.isLoading ? (
|
||||||
|
<div className="flex justify-center py-6">
|
||||||
|
<Spin size="small" />
|
||||||
|
</div>
|
||||||
|
) : query.isError ? (
|
||||||
|
<button type="button" onClick={() => void query.refetch()} className="block w-full py-4 text-center text-xs text-red-500 opacity-80 transition hover:opacity-100">
|
||||||
|
加载失败,点击重试
|
||||||
|
</button>
|
||||||
|
) : filtered.length ? (
|
||||||
|
<div className="space-y-1.5">
|
||||||
|
{filtered.map((item) => (
|
||||||
|
<PromptRow key={item.id} item={item} theme={theme} onInsert={() => insertPrompt(item)} onView={() => onView(item)} />
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="py-4 text-center text-xs opacity-40">{keyword.trim() ? "无匹配提示词" : "该来源暂无提示词"}</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
) : null}
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function PromptRow({ item, theme, onInsert, onView }: { item: Prompt; theme: CanvasTheme; onInsert: () => void; onView: () => void }) {
|
||||||
|
return (
|
||||||
|
<div className="group relative flex items-center gap-2.5 rounded-lg px-2 py-2 transition hover:bg-black/5 dark:hover:bg-white/5">
|
||||||
|
{item.coverUrl ? (
|
||||||
|
<img src={item.coverUrl} alt="" className="size-10 shrink-0 rounded-md object-cover" loading="lazy" />
|
||||||
|
) : (
|
||||||
|
<span className="grid size-10 shrink-0 place-items-center rounded-md" style={{ background: theme.node.panel }}>
|
||||||
|
<FileText className="size-4 opacity-50" />
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
<button type="button" onClick={onView} className="min-w-0 flex-1 text-left">
|
||||||
|
<div className="truncate text-sm font-medium leading-snug">{item.title}</div>
|
||||||
|
<div className="mt-0.5 truncate text-xs leading-snug opacity-50">{item.prompt}</div>
|
||||||
|
</button>
|
||||||
|
<div className="flex shrink-0 flex-col items-center gap-0.5">
|
||||||
|
<button type="button" onClick={onView} className="grid size-6 place-items-center rounded-md opacity-60 transition hover:bg-black/10 hover:opacity-100 dark:hover:bg-white/10" aria-label="查看详情" title="查看详情">
|
||||||
|
<Eye className="size-3.5" />
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={onInsert}
|
||||||
|
className="grid size-6 place-items-center rounded-md opacity-60 transition hover:bg-black/10 hover:opacity-100 dark:hover:bg-white/10"
|
||||||
|
style={{ color: theme.toolbar.activeText }}
|
||||||
|
aria-label="插入画布"
|
||||||
|
title="插入画布"
|
||||||
|
>
|
||||||
|
<Plus className="size-3.5" />
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|||||||
@@ -4,6 +4,7 @@ import { useEffect, useState } from "react";
|
|||||||
|
|
||||||
import { ModelPicker } from "@/components/model-picker";
|
import { ModelPicker } from "@/components/model-picker";
|
||||||
import { ChannelEditorDrawer } from "@/components/layout/channel-editor-drawer";
|
import { ChannelEditorDrawer } from "@/components/layout/channel-editor-drawer";
|
||||||
|
import { ConfigPromptSources } from "@/components/layout/config-prompt-sources";
|
||||||
import { syncAppDataToWebdav, type AppSyncDomainKey, type AppSyncProgressEvent } from "@/services/app-sync";
|
import { syncAppDataToWebdav, type AppSyncDomainKey, type AppSyncProgressEvent } from "@/services/app-sync";
|
||||||
import { testWebdavConnection, WEBDAV_MANIFEST_FILE_NAME } from "@/services/webdav-sync";
|
import { testWebdavConnection, WEBDAV_MANIFEST_FILE_NAME } from "@/services/webdav-sync";
|
||||||
import { audioFormatOptions, audioVoiceOptions, normalizeAudioSpeedValue } from "@/lib/audio-generation";
|
import { audioFormatOptions, audioVoiceOptions, normalizeAudioSpeedValue } from "@/lib/audio-generation";
|
||||||
@@ -240,6 +241,11 @@ export function AppConfigPanel({ showDoneButton = false, initialTab = "channels"
|
|||||||
</Form>
|
</Form>
|
||||||
),
|
),
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
key: "prompt-sources",
|
||||||
|
label: "提示词来源",
|
||||||
|
children: <ConfigPromptSources />,
|
||||||
|
},
|
||||||
{
|
{
|
||||||
key: "webdav",
|
key: "webdav",
|
||||||
label: "WebDAV",
|
label: "WebDAV",
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ import { useEffect, useRef } from "react";
|
|||||||
import { App } from "antd";
|
import { App } from "antd";
|
||||||
|
|
||||||
import { createModelChannel, useConfigStore } from "@/stores/use-config-store";
|
import { createModelChannel, useConfigStore } from "@/stores/use-config-store";
|
||||||
|
import { usePromptSourceScheduler } from "@/hooks/use-prompt-source-scheduler";
|
||||||
|
|
||||||
export function ClientRootInit({ children }: { children: ReactNode }) {
|
export function ClientRootInit({ children }: { children: ReactNode }) {
|
||||||
const { message } = App.useApp();
|
const { message } = App.useApp();
|
||||||
@@ -11,6 +12,8 @@ export function ClientRootInit({ children }: { children: ReactNode }) {
|
|||||||
const config = useConfigStore((state) => state.config);
|
const config = useConfigStore((state) => state.config);
|
||||||
const openConfigDialog = useConfigStore((state) => state.openConfigDialog);
|
const openConfigDialog = useConfigStore((state) => state.openConfigDialog);
|
||||||
|
|
||||||
|
usePromptSourceScheduler();
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (handledConfigParams.current) return;
|
if (handledConfigParams.current) return;
|
||||||
const searchParams = new URLSearchParams(window.location.search);
|
const searchParams = new URLSearchParams(window.location.search);
|
||||||
|
|||||||
@@ -0,0 +1,155 @@
|
|||||||
|
import { App, Button, Select, Switch, Tag } from "antd";
|
||||||
|
import { useQuery, useQueryClient } from "@tanstack/react-query";
|
||||||
|
import { Eye, Pencil, Plus, RefreshCw, Trash2 } from "lucide-react";
|
||||||
|
import { useState } from "react";
|
||||||
|
|
||||||
|
import { PromptSourceEditorDrawer } from "./prompt-source-editor-drawer";
|
||||||
|
import { PromptSourceContentModal } from "./prompt-source-content-modal";
|
||||||
|
import { fetchPromptSourceStatuses, refreshAllSources, refreshSource } from "@/services/api/prompts";
|
||||||
|
import { PROMPT_SOURCE_INTERVAL_OPTIONS, usePromptSourceStore } from "@/stores/use-prompt-source-store";
|
||||||
|
import type { PromptSource } from "@/services/api/prompt-source-presets";
|
||||||
|
|
||||||
|
const STATUS_QUERY_KEY = ["prompt-source-statuses"];
|
||||||
|
|
||||||
|
export function ConfigPromptSources() {
|
||||||
|
const { message, modal } = App.useApp();
|
||||||
|
const queryClient = useQueryClient();
|
||||||
|
const sources = usePromptSourceStore((state) => state.sources);
|
||||||
|
const schedule = usePromptSourceStore((state) => state.schedule);
|
||||||
|
const addSource = usePromptSourceStore((state) => state.addSource);
|
||||||
|
const saveSource = usePromptSourceStore((state) => state.saveSource);
|
||||||
|
const removeSource = usePromptSourceStore((state) => state.removeSource);
|
||||||
|
const toggleSource = usePromptSourceStore((state) => state.toggleSource);
|
||||||
|
const updateSchedule = usePromptSourceStore((state) => state.updateSchedule);
|
||||||
|
const statusQuery = useQuery({ queryKey: STATUS_QUERY_KEY, queryFn: fetchPromptSourceStatuses });
|
||||||
|
|
||||||
|
const [editingSource, setEditingSource] = useState<PromptSource | null>(null);
|
||||||
|
const [viewingId, setViewingId] = useState("");
|
||||||
|
const [refreshingId, setRefreshingId] = useState("");
|
||||||
|
const [refreshingAll, setRefreshingAll] = useState(false);
|
||||||
|
const viewingSource = sources.find((item) => item.id === viewingId) || null;
|
||||||
|
|
||||||
|
const invalidatePrompts = async () => {
|
||||||
|
await Promise.all([
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["prompts"] }),
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["side-panel-prompts"] }),
|
||||||
|
queryClient.invalidateQueries({ queryKey: STATUS_QUERY_KEY }),
|
||||||
|
]);
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleSave = (source: PromptSource) => {
|
||||||
|
saveSource(source);
|
||||||
|
void invalidatePrompts();
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleDelete = (source: PromptSource) => {
|
||||||
|
modal.confirm({
|
||||||
|
title: `删除「${source.name}」?`,
|
||||||
|
content: "来源配置会被移除,已经保存到我的提示词的内容不受影响。",
|
||||||
|
okText: "删除",
|
||||||
|
okButtonProps: { danger: true },
|
||||||
|
cancelText: "取消",
|
||||||
|
onOk: async () => {
|
||||||
|
removeSource(source.id);
|
||||||
|
await invalidatePrompts();
|
||||||
|
},
|
||||||
|
});
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleRefreshOne = async (source: PromptSource) => {
|
||||||
|
setRefreshingId(source.id);
|
||||||
|
try {
|
||||||
|
const result = await refreshSource(source.id);
|
||||||
|
await invalidatePrompts();
|
||||||
|
message.success(`「${source.name}」已更新 ${result.count} 条`);
|
||||||
|
} catch (error) {
|
||||||
|
await queryClient.invalidateQueries({ queryKey: STATUS_QUERY_KEY });
|
||||||
|
message.error(error instanceof Error ? error.message : "更新失败,已保留旧缓存");
|
||||||
|
} finally {
|
||||||
|
setRefreshingId("");
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleRefreshAll = async () => {
|
||||||
|
setRefreshingAll(true);
|
||||||
|
try {
|
||||||
|
const result = await refreshAllSources();
|
||||||
|
updateSchedule("lastFetchedAt", new Date().toISOString());
|
||||||
|
await invalidatePrompts();
|
||||||
|
if (result.failureCount) message.warning(`更新完成:${result.successCount} 个成功,${result.failureCount} 个失败,失败来源已保留旧缓存`);
|
||||||
|
else message.success(`已更新 ${result.successCount} 个来源,共 ${result.total} 条`);
|
||||||
|
} catch (error) {
|
||||||
|
message.error(error instanceof Error ? error.message : "更新失败");
|
||||||
|
} finally {
|
||||||
|
setRefreshingAll(false);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div>
|
||||||
|
<div className="mb-4 flex flex-wrap items-center justify-end gap-3">
|
||||||
|
<Button type="primary" icon={<Plus className="size-4" />} onClick={() => setEditingSource(addSource())}>
|
||||||
|
新增来源
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="space-y-2">
|
||||||
|
{sources.map((source) => {
|
||||||
|
const status = statusQuery.data?.[source.id];
|
||||||
|
return (
|
||||||
|
<div key={source.id} className="flex flex-wrap items-center gap-3 rounded-lg border border-stone-200 px-4 py-3 dark:border-stone-800">
|
||||||
|
<Switch size="small" checked={source.enabled} onChange={(checked) => { toggleSource(source.id, checked); void invalidatePrompts(); }} />
|
||||||
|
<div className="min-w-[220px] flex-1">
|
||||||
|
<div className="flex min-w-0 items-center gap-2">
|
||||||
|
<span className="truncate text-sm font-semibold">{source.name}</span>
|
||||||
|
{source.builtIn ? <Tag className="m-0 shrink-0 text-[10px]">内置</Tag> : null}
|
||||||
|
</div>
|
||||||
|
<div className="mt-1 flex min-w-0 flex-wrap items-center gap-x-3 gap-y-1 text-xs text-stone-500">
|
||||||
|
<a className="max-w-full truncate hover:text-stone-800 hover:underline dark:hover:text-stone-200" href={source.homepage || source.url} target="_blank" rel="noreferrer">
|
||||||
|
{source.homepage || source.url}
|
||||||
|
</a>
|
||||||
|
<span className="tabular-nums">{status?.count ?? 0} 条</span>
|
||||||
|
{status?.lastError ? <Tag color="error" className="m-0 text-[10px]" title={status.lastError}>失败</Tag> : status?.lastSuccessAt ? <Tag color="success" className="m-0 text-[10px]">正常</Tag> : <Tag className="m-0 text-[10px]">未同步</Tag>}
|
||||||
|
<span>{status?.lastSuccessAt ? `上次成功 ${formatTime(status.lastSuccessAt)}` : "尚未拉取"}</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
<div className="ml-auto flex flex-wrap justify-end gap-2">
|
||||||
|
<Button size="small" icon={<Eye className="size-3.5" />} onClick={() => setViewingId(source.id)}>
|
||||||
|
查看内容
|
||||||
|
</Button>
|
||||||
|
<Button size="small" icon={<RefreshCw className="size-3.5" />} loading={refreshingId === source.id} onClick={() => void handleRefreshOne(source)}>
|
||||||
|
立即拉取
|
||||||
|
</Button>
|
||||||
|
{!source.builtIn ? <Button size="small" icon={<Pencil className="size-3.5" />} onClick={() => setEditingSource(source)}>编辑来源</Button> : null}
|
||||||
|
{!source.builtIn ? <Button size="small" danger icon={<Trash2 className="size-3.5" />} onClick={() => handleDelete(source)}>删除</Button> : null}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
})}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<section className="mt-5 rounded-lg border border-stone-200 p-4 dark:border-stone-800">
|
||||||
|
<div className="mb-3 text-sm font-semibold">定时拉取</div>
|
||||||
|
<div className="flex flex-wrap items-center gap-3">
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<span className="text-xs text-stone-500">拉取周期</span>
|
||||||
|
<Select size="small" className="w-36" value={schedule.intervalMinutes} options={PROMPT_SOURCE_INTERVAL_OPTIONS} onChange={(value) => updateSchedule("intervalMinutes", value)} />
|
||||||
|
</div>
|
||||||
|
<Button size="small" type="primary" icon={<RefreshCw className="size-3.5" />} loading={refreshingAll} onClick={() => void handleRefreshAll()}>
|
||||||
|
全部立即拉取
|
||||||
|
</Button>
|
||||||
|
<span className="text-xs text-stone-500">{schedule.lastFetchedAt ? `上次拉取 ${formatTime(schedule.lastFetchedAt)}` : "尚未定时拉取"}</span>
|
||||||
|
</div>
|
||||||
|
<div className="mt-2 text-xs text-stone-400">开启周期后,页面打开期间会按周期自动拉取所有启用的来源。</div>
|
||||||
|
</section>
|
||||||
|
|
||||||
|
<PromptSourceEditorDrawer open={Boolean(editingSource)} source={editingSource} onSave={handleSave} onClose={() => setEditingSource(null)} />
|
||||||
|
<PromptSourceContentModal source={viewingSource} onClose={() => setViewingId("")} />
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function formatTime(value: string) {
|
||||||
|
const date = new Date(value);
|
||||||
|
return Number.isNaN(date.getTime()) ? "-" : date.toLocaleString("zh-CN", { month: "2-digit", day: "2-digit", hour: "2-digit", minute: "2-digit" });
|
||||||
|
}
|
||||||
@@ -0,0 +1,130 @@
|
|||||||
|
import { App, Button, Empty, Modal, Space, Table, Tag } from "antd";
|
||||||
|
import { Copy, FolderPlus, RefreshCw } from "lucide-react";
|
||||||
|
import { useCallback, useEffect, useState } from "react";
|
||||||
|
|
||||||
|
import { PromptDetailDialog } from "@/pages/prompts/components/prompt-detail-dialog";
|
||||||
|
import { useCopyText } from "@/hooks/use-copy-text";
|
||||||
|
import { useAssetStore } from "@/stores/use-asset-store";
|
||||||
|
import { fetchSourcePrompts, refreshSource, type Prompt } from "@/services/api/prompts";
|
||||||
|
import type { PromptSource } from "@/services/api/prompt-source-presets";
|
||||||
|
|
||||||
|
export function PromptSourceContentModal({ source, onClose }: { source: PromptSource | null; onClose: () => void }) {
|
||||||
|
const { message } = App.useApp();
|
||||||
|
const [items, setItems] = useState<Prompt[]>([]);
|
||||||
|
const [loading, setLoading] = useState(false);
|
||||||
|
const [detail, setDetail] = useState<Prompt | null>(null);
|
||||||
|
const copyText = useCopyText();
|
||||||
|
const addAsset = useAssetStore((state) => state.addAsset);
|
||||||
|
|
||||||
|
const load = useCallback(
|
||||||
|
async (force: boolean) => {
|
||||||
|
if (!source) return;
|
||||||
|
setLoading(true);
|
||||||
|
try {
|
||||||
|
setItems(force ? await refreshSourceItems(source.id) : await fetchSourcePrompts(source.id));
|
||||||
|
} catch (error) {
|
||||||
|
message.error(error instanceof Error ? error.message : "拉取提示词失败");
|
||||||
|
} finally {
|
||||||
|
setLoading(false);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
[source, message],
|
||||||
|
);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (source) void load(false);
|
||||||
|
else setItems([]);
|
||||||
|
}, [source, load]);
|
||||||
|
|
||||||
|
const saveAsset = (item: Prompt) => {
|
||||||
|
addAsset({ kind: "text", title: item.title, coverUrl: item.coverUrl, tags: item.tags, source: item.category, data: { content: item.prompt }, metadata: { source: "prompt-library", promptId: item.id, githubUrl: item.githubUrl } });
|
||||||
|
message.success("已加入我的资产");
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<>
|
||||||
|
<Modal
|
||||||
|
open={Boolean(source)}
|
||||||
|
onCancel={onClose}
|
||||||
|
width={980}
|
||||||
|
footer={null}
|
||||||
|
title={
|
||||||
|
<div className="flex flex-wrap items-center justify-between gap-2 pr-6">
|
||||||
|
<div>
|
||||||
|
<div className="text-base font-semibold">{source?.name || ""} · 提示词内容</div>
|
||||||
|
<div className="mt-0.5 text-xs font-normal text-stone-500">共 {items.length} 条</div>
|
||||||
|
</div>
|
||||||
|
<Button size="small" icon={<RefreshCw className="size-3.5" />} loading={loading} onClick={() => void load(true)}>
|
||||||
|
立即更新
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
<Table<Prompt>
|
||||||
|
rowKey="id"
|
||||||
|
size="small"
|
||||||
|
loading={loading}
|
||||||
|
dataSource={items}
|
||||||
|
pagination={{ pageSize: 10, showSizeChanger: false, size: "small" }}
|
||||||
|
scroll={{ y: "56vh" }}
|
||||||
|
locale={{ emptyText: <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} description="暂无提示词" /> }}
|
||||||
|
columns={[
|
||||||
|
{
|
||||||
|
title: "封面",
|
||||||
|
dataIndex: "coverUrl",
|
||||||
|
width: 72,
|
||||||
|
render: (coverUrl: string) => (coverUrl ? <img src={coverUrl} alt="" className="size-12 rounded object-cover" /> : <div className="size-12 rounded bg-stone-100 dark:bg-stone-800" />),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: "标题",
|
||||||
|
dataIndex: "title",
|
||||||
|
render: (title: string, item) => (
|
||||||
|
<div className="min-w-0">
|
||||||
|
<div className="truncate font-medium">{title}</div>
|
||||||
|
<div className="mt-0.5 line-clamp-2 text-xs text-stone-500">{item.prompt}</div>
|
||||||
|
</div>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: "标签",
|
||||||
|
dataIndex: "tags",
|
||||||
|
width: 200,
|
||||||
|
render: (tags: string[]) => (
|
||||||
|
<div className="flex flex-wrap gap-1">
|
||||||
|
{tags.slice(0, 4).map((tag) => (
|
||||||
|
<Tag key={tag} className="m-0">
|
||||||
|
{tag}
|
||||||
|
</Tag>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
title: "操作",
|
||||||
|
width: 210,
|
||||||
|
render: (_, item) => (
|
||||||
|
<Space size={4} wrap>
|
||||||
|
<Button size="small" type="text" icon={<Copy className="size-3.5" />} onClick={() => copyText(item.prompt, "提示词已复制")}>
|
||||||
|
复制
|
||||||
|
</Button>
|
||||||
|
<Button size="small" type="text" onClick={() => setDetail(item)}>
|
||||||
|
详情
|
||||||
|
</Button>
|
||||||
|
<Button size="small" type="text" icon={<FolderPlus className="size-3.5" />} onClick={() => saveAsset(item)}>
|
||||||
|
加入资产
|
||||||
|
</Button>
|
||||||
|
</Space>
|
||||||
|
),
|
||||||
|
},
|
||||||
|
]}
|
||||||
|
/>
|
||||||
|
</Modal>
|
||||||
|
<PromptDetailDialog prompt={detail} onClose={() => setDetail(null)} onCopy={(prompt) => copyText(prompt, "提示词已复制")} onSaveAsset={saveAsset} />
|
||||||
|
</>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function refreshSourceItems(sourceId: string) {
|
||||||
|
await refreshSource(sourceId);
|
||||||
|
return fetchSourcePrompts(sourceId);
|
||||||
|
}
|
||||||
@@ -0,0 +1,86 @@
|
|||||||
|
import { App, Button, Drawer, Input, Space, Switch } from "antd";
|
||||||
|
import { useEffect, useState } from "react";
|
||||||
|
|
||||||
|
import type { PromptSource } from "@/services/api/prompt-source-presets";
|
||||||
|
|
||||||
|
export function PromptSourceEditorDrawer({ open, source, onSave, onClose }: { open: boolean; source: PromptSource | null; onSave: (source: PromptSource) => void; onClose: () => void }) {
|
||||||
|
const { message } = App.useApp();
|
||||||
|
const [draft, setDraft] = useState<PromptSource | null>(source);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (open && source) setDraft(source);
|
||||||
|
}, [open, source]);
|
||||||
|
|
||||||
|
if (!draft) return null;
|
||||||
|
|
||||||
|
const patch = (value: Partial<PromptSource>) => setDraft((current) => (current ? { ...current, ...value } : current));
|
||||||
|
|
||||||
|
const save = () => {
|
||||||
|
const name = draft.name.trim();
|
||||||
|
const url = draft.url.trim();
|
||||||
|
if (!name) return message.warning("请输入来源名称");
|
||||||
|
if (!isHttpUrl(url)) return message.warning("请输入有效的 JSON URL");
|
||||||
|
if (draft.homepage.trim() && !isHttpUrl(draft.homepage.trim())) return message.warning("请输入有效的主页地址");
|
||||||
|
onSave({ ...draft, name, url, homepage: draft.homepage.trim(), builtIn: false });
|
||||||
|
onClose();
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Drawer
|
||||||
|
open={open}
|
||||||
|
width={560}
|
||||||
|
title={source?.name === "新来源" ? "新增提示词来源" : "编辑提示词来源"}
|
||||||
|
onClose={onClose}
|
||||||
|
styles={{ body: { paddingTop: 16 } }}
|
||||||
|
extra={
|
||||||
|
<Space>
|
||||||
|
<Button onClick={onClose}>取消</Button>
|
||||||
|
<Button type="primary" onClick={save}>
|
||||||
|
保存
|
||||||
|
</Button>
|
||||||
|
</Space>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
<div className="space-y-5">
|
||||||
|
<label className="block">
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">来源名称</span>
|
||||||
|
<Input value={draft.name} onChange={(event) => patch({ name: event.target.value })} placeholder="用于分类展示" />
|
||||||
|
</label>
|
||||||
|
<label className="block">
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">JSON URL</span>
|
||||||
|
<Input value={draft.url} onChange={(event) => patch({ url: event.target.value })} placeholder="https://example.com/prompts.json" />
|
||||||
|
</label>
|
||||||
|
<label className="block">
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">来源主页(可选)</span>
|
||||||
|
<Input value={draft.homepage} onChange={(event) => patch({ homepage: event.target.value })} placeholder="https://example.com" />
|
||||||
|
</label>
|
||||||
|
<div className="flex items-center justify-between border-y border-stone-200 py-3 dark:border-stone-800">
|
||||||
|
<span className="text-sm font-medium">启用来源</span>
|
||||||
|
<Switch checked={draft.enabled} onChange={(enabled) => patch({ enabled })} />
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<div className="mb-2 text-sm font-medium">JSON 格式</div>
|
||||||
|
<pre className="overflow-x-auto rounded-md bg-stone-100 p-3 text-xs leading-5 text-stone-600 dark:bg-stone-900 dark:text-stone-300">{`[
|
||||||
|
{
|
||||||
|
"id": "product-photo-1",
|
||||||
|
"title": "白底商品图",
|
||||||
|
"prompt": "生成专业白底商品摄影图",
|
||||||
|
"description": "",
|
||||||
|
"coverUrl": "",
|
||||||
|
"referenceImageUrls": [],
|
||||||
|
"tags": ["商品", "摄影"]
|
||||||
|
}
|
||||||
|
]`}</pre>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</Drawer>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function isHttpUrl(value: string) {
|
||||||
|
try {
|
||||||
|
return ["http:", "https:"].includes(new URL(value).protocol);
|
||||||
|
} catch {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,4 +1,4 @@
|
|||||||
import { Copy } from "lucide-react";
|
import { Copy, FileText } from "lucide-react";
|
||||||
import type { ReactNode } from "react";
|
import type { ReactNode } from "react";
|
||||||
import { Button, Card, Tag } from "antd";
|
import { Button, Card, Tag } from "antd";
|
||||||
|
|
||||||
@@ -28,7 +28,7 @@ export function PromptCard({
|
|||||||
styles={{ body: { padding: 0 } }}
|
styles={{ body: { padding: 0 } }}
|
||||||
cover={
|
cover={
|
||||||
<button type="button" className="block w-full text-left" onClick={onOpen}>
|
<button type="button" className="block w-full text-left" onClick={onOpen}>
|
||||||
<img src={item.coverUrl} alt={item.title} className="aspect-[4/3] w-full object-cover" />
|
{item.coverUrl ? <img src={item.coverUrl} alt={item.title} className="aspect-[4/3] w-full object-cover" loading="lazy" /> : <span className="grid aspect-[4/3] w-full place-items-center bg-stone-100 text-stone-400 dark:bg-stone-900 dark:text-stone-600"><FileText className="size-8" /></span>}
|
||||||
</button>
|
</button>
|
||||||
}
|
}
|
||||||
>
|
>
|
||||||
@@ -38,7 +38,7 @@ export function PromptCard({
|
|||||||
<h2 className="line-clamp-1 text-sm font-semibold text-stone-950 dark:text-stone-100">{item.title}</h2>
|
<h2 className="line-clamp-1 text-sm font-semibold text-stone-950 dark:text-stone-100">{item.title}</h2>
|
||||||
<span className="shrink-0 text-xs text-stone-400 dark:text-stone-500">{formatPromptDate(item.updatedAt)}</span>
|
<span className="shrink-0 text-xs text-stone-400 dark:text-stone-500">{formatPromptDate(item.updatedAt)}</span>
|
||||||
</div>
|
</div>
|
||||||
<p className="mt-2 line-clamp-3 text-xs leading-5 text-stone-600 dark:text-stone-400">{item.prompt}</p>
|
<p className="mt-2 line-clamp-3 text-xs leading-5 text-stone-600 dark:text-stone-400">{item.description || item.prompt}</p>
|
||||||
<div className="mt-3 flex flex-wrap gap-1.5">
|
<div className="mt-3 flex flex-wrap gap-1.5">
|
||||||
{item.tags.map((tag) => (
|
{item.tags.map((tag) => (
|
||||||
<Tag key={tag} className="m-0 text-[11px]">
|
<Tag key={tag} className="m-0 text-[11px]">
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ export function PromptSelectDialog({ open, onOpenChange, onSelect }: { open: boo
|
|||||||
const [keyword, setKeyword] = useState("");
|
const [keyword, setKeyword] = useState("");
|
||||||
const [selectedTags, setSelectedTags] = useState<string[]>([]);
|
const [selectedTags, setSelectedTags] = useState<string[]>([]);
|
||||||
const [selectedCategory, setSelectedCategory] = useState(ALL_PROMPTS_OPTION);
|
const [selectedCategory, setSelectedCategory] = useState(ALL_PROMPTS_OPTION);
|
||||||
const { query, items, tags: promptTags, categories: promptCategories } = usePromptList({ keyword, tags: selectedTags, category: selectedCategory, enabled: open });
|
const { query, items, tags: promptTags, categories: promptCategories } = usePromptList({ keyword, tags: selectedTags, category: selectedCategory, enabled: open, includePersonal: true });
|
||||||
const toggleTag = (tag: string) => {
|
const toggleTag = (tag: string) => {
|
||||||
if (tag === ALL_PROMPTS_OPTION) return setSelectedTags([]);
|
if (tag === ALL_PROMPTS_OPTION) return setSelectedTags([]);
|
||||||
setSelectedTags((items) => (items.includes(tag) ? items.filter((item) => item !== tag) : [...items, tag]));
|
setSelectedTags((items) => (items.includes(tag) ? items.filter((item) => item !== tag) : [...items, tag]));
|
||||||
|
|||||||
@@ -5,10 +5,10 @@ import { ALL_PROMPTS_OPTION, fetchPrompts } from "@/services/api/prompts";
|
|||||||
|
|
||||||
export const PROMPT_PAGE_SIZE = 20;
|
export const PROMPT_PAGE_SIZE = 20;
|
||||||
|
|
||||||
export function usePromptList({ keyword, tags, category, enabled = true }: { keyword: string; tags: string[]; category: string; enabled?: boolean }) {
|
export function usePromptList({ keyword, tags, category, enabled = true, includePersonal = false }: { keyword: string; tags: string[]; category: string; enabled?: boolean; includePersonal?: boolean }) {
|
||||||
const query = useInfiniteQuery({
|
const query = useInfiniteQuery({
|
||||||
queryKey: ["prompts", keyword, tags, category],
|
queryKey: ["prompts", keyword, tags, category, includePersonal],
|
||||||
queryFn: ({ pageParam }) => fetchPrompts({ keyword, tag: tags, category, page: pageParam, pageSize: PROMPT_PAGE_SIZE }),
|
queryFn: ({ pageParam }) => fetchPrompts({ keyword, tag: tags, category, page: pageParam, pageSize: PROMPT_PAGE_SIZE, includePersonal }),
|
||||||
initialPageParam: 1,
|
initialPageParam: 1,
|
||||||
getNextPageParam: (lastPage, pages) => (pages.reduce((total, page) => total + page.items.length, 0) < lastPage.total ? pages.length + 1 : undefined),
|
getNextPageParam: (lastPage, pages) => (pages.reduce((total, page) => total + page.items.length, 0) < lastPage.total ? pages.length + 1 : undefined),
|
||||||
enabled,
|
enabled,
|
||||||
|
|||||||
@@ -0,0 +1,40 @@
|
|||||||
|
import { useEffect } from "react";
|
||||||
|
import { useQueryClient } from "@tanstack/react-query";
|
||||||
|
|
||||||
|
import { refreshDueSources } from "@/services/api/prompts";
|
||||||
|
import { usePromptSourceStore } from "@/stores/use-prompt-source-store";
|
||||||
|
|
||||||
|
const CHECK_INTERVAL_MS = 60_000;
|
||||||
|
|
||||||
|
/** Periodically update only the sources whose last successful refresh is due. */
|
||||||
|
export function usePromptSourceScheduler() {
|
||||||
|
const queryClient = useQueryClient();
|
||||||
|
const intervalMinutes = usePromptSourceStore((state) => state.schedule.intervalMinutes);
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (!intervalMinutes) return;
|
||||||
|
let running = false;
|
||||||
|
const tick = async () => {
|
||||||
|
if (running) return;
|
||||||
|
const { updateSchedule } = usePromptSourceStore.getState();
|
||||||
|
running = true;
|
||||||
|
try {
|
||||||
|
const result = await refreshDueSources(intervalMinutes * 60_000);
|
||||||
|
if (!result.results.length) return;
|
||||||
|
updateSchedule("lastFetchedAt", new Date().toISOString());
|
||||||
|
await Promise.all([
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["prompts"] }),
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["side-panel-prompts"] }),
|
||||||
|
queryClient.invalidateQueries({ queryKey: ["prompt-source-statuses"] }),
|
||||||
|
]);
|
||||||
|
} catch {
|
||||||
|
// 单个来源的错误已写入来源状态,下一个检查周期会继续尝试。
|
||||||
|
} finally {
|
||||||
|
running = false;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
void tick();
|
||||||
|
const timer = window.setInterval(() => void tick(), CHECK_INTERVAL_MS);
|
||||||
|
return () => window.clearInterval(timer);
|
||||||
|
}, [intervalMinutes, queryClient]);
|
||||||
|
}
|
||||||
@@ -4,6 +4,7 @@ import { fetchPrompts } from "@/services/api/prompts";
|
|||||||
import { uploadImage } from "@/services/image-storage";
|
import { uploadImage } from "@/services/image-storage";
|
||||||
import { imageAspectOptions, imageQualityOptions } from "@/components/image-settings-panel";
|
import { imageAspectOptions, imageQualityOptions } from "@/components/image-settings-panel";
|
||||||
import { videoResolutionOptions, videoSecondOptions, videoSizeOptions } from "@/components/video-settings-panel";
|
import { videoResolutionOptions, videoSecondOptions, videoSizeOptions } from "@/components/video-settings-panel";
|
||||||
|
import type { CanvasAgentSnapshot } from "@/lib/canvas/canvas-agent-ops";
|
||||||
import { useCanvasStore } from "@/stores/canvas/use-canvas-store";
|
import { useCanvasStore } from "@/stores/canvas/use-canvas-store";
|
||||||
import { useAssetStore } from "@/stores/use-asset-store";
|
import { useAssetStore } from "@/stores/use-asset-store";
|
||||||
import { modelOptionLabel, modelOptionName, normalizeModelOptionValue, selectableModelsByCapability, useConfigStore } from "@/stores/use-config-store";
|
import { modelOptionLabel, modelOptionName, normalizeModelOptionValue, selectableModelsByCapability, useConfigStore } from "@/stores/use-config-store";
|
||||||
@@ -14,6 +15,7 @@ import { useWorkbenchAgentStore } from "@/stores/use-workbench-agent-store";
|
|||||||
|
|
||||||
export const SITE_TOOL_NAMES = [
|
export const SITE_TOOL_NAMES = [
|
||||||
"canvas_list_projects",
|
"canvas_list_projects",
|
||||||
|
"generation_get_status",
|
||||||
"workbench_image_get_config",
|
"workbench_image_get_config",
|
||||||
"workbench_image_generate",
|
"workbench_image_generate",
|
||||||
"workbench_video_get_config",
|
"workbench_video_get_config",
|
||||||
@@ -31,6 +33,7 @@ export function isSiteTool(name: string): name is SiteToolName {
|
|||||||
|
|
||||||
export const SITE_TOOL_LABELS: Record<SiteToolName, string> = {
|
export const SITE_TOOL_LABELS: Record<SiteToolName, string> = {
|
||||||
canvas_list_projects: "画布列表",
|
canvas_list_projects: "画布列表",
|
||||||
|
generation_get_status: "生成任务状态",
|
||||||
workbench_image_get_config: "生图配置",
|
workbench_image_get_config: "生图配置",
|
||||||
workbench_image_generate: "生图工作台生成",
|
workbench_image_generate: "生图工作台生成",
|
||||||
workbench_video_get_config: "视频配置",
|
workbench_video_get_config: "视频配置",
|
||||||
@@ -41,11 +44,16 @@ export const SITE_TOOL_LABELS: Record<SiteToolName, string> = {
|
|||||||
};
|
};
|
||||||
|
|
||||||
type SiteToolInput = Record<string, unknown>;
|
type SiteToolInput = Record<string, unknown>;
|
||||||
|
type SiteToolContext = { canvasSnapshot?: CanvasAgentSnapshot | null };
|
||||||
|
type GenerationStatus = "idle" | "queued" | "running" | "succeeded" | "failed";
|
||||||
|
type GenerationStatusItem = { id: string; source: "canvas" | "image" | "video"; status: GenerationStatus; kind?: string; title?: string; prompt?: string; projectId?: string; createdAt?: string; updatedAt?: string; successCount?: number; failCount?: number; error?: string };
|
||||||
|
|
||||||
export async function runSiteTool(name: SiteToolName, input: SiteToolInput, navigate: NavigateFunction): Promise<unknown> {
|
export async function runSiteTool(name: SiteToolName, input: SiteToolInput, navigate: NavigateFunction, context: SiteToolContext = {}): Promise<unknown> {
|
||||||
switch (name) {
|
switch (name) {
|
||||||
case "canvas_list_projects":
|
case "canvas_list_projects":
|
||||||
return listCanvasProjects(input);
|
return listCanvasProjects(input);
|
||||||
|
case "generation_get_status":
|
||||||
|
return getGenerationStatus(input, context.canvasSnapshot);
|
||||||
case "workbench_image_get_config":
|
case "workbench_image_get_config":
|
||||||
return getImageConfig();
|
return getImageConfig();
|
||||||
case "workbench_image_generate":
|
case "workbench_image_generate":
|
||||||
@@ -65,6 +73,56 @@ export async function runSiteTool(name: SiteToolName, input: SiteToolInput, navi
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function getGenerationStatus(input: SiteToolInput, canvasSnapshot?: CanvasAgentSnapshot | null) {
|
||||||
|
const scope = input.scope === "canvas" || input.scope === "image" || input.scope === "video" ? input.scope : "all";
|
||||||
|
const taskId = typeof input.taskId === "string" ? input.taskId : "";
|
||||||
|
const nodeIds = new Set(Array.isArray(input.nodeIds) ? input.nodeIds.filter((id): id is string => typeof id === "string") : []);
|
||||||
|
const limit = Math.max(1, Math.min(100, Math.floor(Number(input.limit)) || 20));
|
||||||
|
const tasks: GenerationStatusItem[] = [];
|
||||||
|
const includeCanvas = (scope === "all" || scope === "canvas") && (!taskId || nodeIds.size > 0);
|
||||||
|
const includeWorkbench = !nodeIds.size || Boolean(taskId);
|
||||||
|
|
||||||
|
if (includeCanvas && canvasSnapshot) {
|
||||||
|
canvasSnapshot.nodes.forEach((node) => {
|
||||||
|
const status = normalizeCanvasGenerationStatus(node.metadata?.status);
|
||||||
|
if (!status || (nodeIds.size && !nodeIds.has(node.id))) return;
|
||||||
|
const metadata = node.metadata || {};
|
||||||
|
if (!nodeIds.size && node.type !== "config" && status !== "running" && status !== "failed" && !metadata.generationMode && !metadata.generationType && !metadata.model) return;
|
||||||
|
tasks.push({ id: node.id, source: "canvas", status, kind: metadata.generationMode || node.type, title: node.title, prompt: compactPrompt(metadata.prompt || metadata.composerContent), projectId: canvasSnapshot.projectId, error: metadata.errorDetails });
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
if (includeWorkbench) {
|
||||||
|
useWorkbenchAgentStore.getState().tasks.forEach((task) => {
|
||||||
|
if ((scope === "image" || scope === "video") && task.kind !== scope) return;
|
||||||
|
if (scope === "canvas" || (taskId && task.id !== taskId)) return;
|
||||||
|
tasks.push({ ...task, source: task.kind, prompt: compactPrompt(task.prompt) });
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
tasks.sort((a, b) => generationStatusOrder(a.status) - generationStatusOrder(b.status) || (b.updatedAt || "").localeCompare(a.updatedAt || ""));
|
||||||
|
const summary: Record<GenerationStatus, number> = { idle: 0, queued: 0, running: 0, succeeded: 0, failed: 0 };
|
||||||
|
tasks.forEach((task) => (summary[task.status] += 1));
|
||||||
|
return { total: tasks.length, summary, tasks: tasks.slice(0, limit) };
|
||||||
|
}
|
||||||
|
|
||||||
|
function generationStatusOrder(status: GenerationStatus) {
|
||||||
|
return status === "running" ? 0 : status === "queued" ? 1 : 2;
|
||||||
|
}
|
||||||
|
|
||||||
|
function normalizeCanvasGenerationStatus(status: unknown): GenerationStatus | null {
|
||||||
|
if (status === "idle") return "idle";
|
||||||
|
if (status === "loading") return "running";
|
||||||
|
if (status === "success") return "succeeded";
|
||||||
|
if (status === "error") return "failed";
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function compactPrompt(prompt: unknown) {
|
||||||
|
const value = typeof prompt === "string" ? prompt.trim() : "";
|
||||||
|
return value ? `${value.slice(0, 200)}${value.length > 200 ? "..." : ""}` : undefined;
|
||||||
|
}
|
||||||
|
|
||||||
function listCanvasProjects(input: SiteToolInput) {
|
function listCanvasProjects(input: SiteToolInput) {
|
||||||
const { projects, hydrated } = useCanvasStore.getState();
|
const { projects, hydrated } = useCanvasStore.getState();
|
||||||
if (!hydrated) throw new Error("画布还在加载中,请稍后重试");
|
if (!hydrated) throw new Error("画布还在加载中,请稍后重试");
|
||||||
@@ -118,8 +176,8 @@ function runImageWorkbench(input: SiteToolInput, navigate: NavigateFunction) {
|
|||||||
const prompt = typeof input.prompt === "string" ? input.prompt : undefined;
|
const prompt = typeof input.prompt === "string" ? input.prompt : undefined;
|
||||||
const run = input.run !== false;
|
const run = input.run !== false;
|
||||||
navigate("/image");
|
navigate("/image");
|
||||||
useWorkbenchAgentStore.getState().dispatchImage({ prompt, run });
|
const taskId = useWorkbenchAgentStore.getState().dispatchImage({ prompt, run });
|
||||||
return { ok: true, navigated: "/image", prompt, run, applied, note: run ? "已跳转生图工作台并触发生成,结果请稍后在工作台查看" : "已跳转生图工作台并填入参数,未触发生成" };
|
return { ok: true, navigated: "/image", prompt, run, taskId, applied, note: run ? "已跳转生图工作台并触发生成,可用 generation_get_status 查询任务" : "已跳转生图工作台并填入参数,未触发生成" };
|
||||||
}
|
}
|
||||||
|
|
||||||
function getVideoConfig() {
|
function getVideoConfig() {
|
||||||
@@ -173,8 +231,8 @@ function runVideoWorkbench(input: SiteToolInput, navigate: NavigateFunction) {
|
|||||||
const prompt = typeof input.prompt === "string" ? input.prompt : undefined;
|
const prompt = typeof input.prompt === "string" ? input.prompt : undefined;
|
||||||
const run = input.run !== false;
|
const run = input.run !== false;
|
||||||
navigate("/video");
|
navigate("/video");
|
||||||
useWorkbenchAgentStore.getState().dispatchVideo({ prompt, run });
|
const taskId = useWorkbenchAgentStore.getState().dispatchVideo({ prompt, run });
|
||||||
return { ok: true, navigated: "/video", prompt, run, applied, note: run ? "已跳转视频创作台并触发生成,结果请稍后在工作台查看" : "已跳转视频创作台并填入参数,未触发生成" };
|
return { ok: true, navigated: "/video", prompt, run, taskId, applied, note: run ? "已跳转视频创作台并触发生成,可用 generation_get_status 查询任务" : "已跳转视频创作台并填入参数,未触发生成" };
|
||||||
}
|
}
|
||||||
|
|
||||||
async function searchPrompts(input: SiteToolInput) {
|
async function searchPrompts(input: SiteToolInput) {
|
||||||
|
|||||||
@@ -2591,11 +2591,11 @@ function InfiniteCanvasPage() {
|
|||||||
);
|
);
|
||||||
|
|
||||||
const insertAssistantText = useCallback(
|
const insertAssistantText = useCallback(
|
||||||
(text: string) => {
|
(text: string, title?: string) => {
|
||||||
const center = screenToCanvas((containerRef.current?.getBoundingClientRect().left || 0) + size.width / 2, (containerRef.current?.getBoundingClientRect().top || 0) + size.height / 2);
|
const center = screenToCanvas((containerRef.current?.getBoundingClientRect().left || 0) + size.width / 2, (containerRef.current?.getBoundingClientRect().top || 0) + size.height / 2);
|
||||||
const node = {
|
const node = {
|
||||||
...createCanvasNode(CanvasNodeType.Text, center, { content: text, status: NODE_STATUS_SUCCESS }),
|
...createCanvasNode(CanvasNodeType.Text, center, { content: text, status: NODE_STATUS_SUCCESS }),
|
||||||
title: text.slice(0, 32) || "Assistant Text",
|
title: title || text.slice(0, 32) || "Assistant Text",
|
||||||
};
|
};
|
||||||
|
|
||||||
setNodes((prev) => [...prev, node]);
|
setNodes((prev) => [...prev, node]);
|
||||||
@@ -2608,7 +2608,7 @@ function InfiniteCanvasPage() {
|
|||||||
const handleAssetInsert = useCallback(
|
const handleAssetInsert = useCallback(
|
||||||
(payload: InsertAssetPayload) => {
|
(payload: InsertAssetPayload) => {
|
||||||
if (payload.kind === "text") {
|
if (payload.kind === "text") {
|
||||||
insertAssistantText(payload.content);
|
insertAssistantText(payload.content, payload.title);
|
||||||
} else if (payload.kind === "video") {
|
} else if (payload.kind === "video") {
|
||||||
const spec = NODE_DEFAULT_SIZE[CanvasNodeType.Video];
|
const spec = NODE_DEFAULT_SIZE[CanvasNodeType.Video];
|
||||||
const center = screenToCanvas((containerRef.current?.getBoundingClientRect().left || 0) + size.width / 2, (containerRef.current?.getBoundingClientRect().top || 0) + size.height / 2);
|
const center = screenToCanvas((containerRef.current?.getBoundingClientRect().left || 0) + size.width / 2, (containerRef.current?.getBoundingClientRect().top || 0) + size.height / 2);
|
||||||
|
|||||||
@@ -92,7 +92,9 @@ export default function ImagePage() {
|
|||||||
const [autoRunToken, setAutoRunToken] = useState(0);
|
const [autoRunToken, setAutoRunToken] = useState(0);
|
||||||
const imageCommand = useWorkbenchAgentStore((state) => state.imageCommand);
|
const imageCommand = useWorkbenchAgentStore((state) => state.imageCommand);
|
||||||
const clearImageCommand = useWorkbenchAgentStore((state) => state.clearImageCommand);
|
const clearImageCommand = useWorkbenchAgentStore((state) => state.clearImageCommand);
|
||||||
|
const updateAgentTask = useWorkbenchAgentStore((state) => state.updateTask);
|
||||||
const processedCommandRef = useRef(0);
|
const processedCommandRef = useRef(0);
|
||||||
|
const agentTaskIdRef = useRef<string | undefined>(undefined);
|
||||||
|
|
||||||
const model = effectiveConfig.imageModel || effectiveConfig.model;
|
const model = effectiveConfig.imageModel || effectiveConfig.model;
|
||||||
const canGenerate = Boolean(prompt.trim());
|
const canGenerate = Boolean(prompt.trim());
|
||||||
@@ -141,22 +143,30 @@ export default function ImagePage() {
|
|||||||
};
|
};
|
||||||
|
|
||||||
const generate = async () => {
|
const generate = async () => {
|
||||||
|
const agentTaskId = agentTaskIdRef.current;
|
||||||
|
agentTaskIdRef.current = undefined;
|
||||||
const text = prompt.trim();
|
const text = prompt.trim();
|
||||||
if (!text) {
|
if (!text) {
|
||||||
message.error("请输入生图提示词");
|
message.error("请输入生图提示词");
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "failed", error: "请输入生图提示词" });
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
if (!isAiConfigReady(effectiveConfig, model)) {
|
if (!isAiConfigReady(effectiveConfig, model)) {
|
||||||
message.warning("请先完成配置");
|
message.warning("请先完成配置");
|
||||||
openConfigDialog(true);
|
openConfigDialog(true);
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "failed", error: "生图配置不完整" });
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const snapshot = buildRequestSnapshot();
|
const snapshot = buildRequestSnapshot();
|
||||||
if (!snapshot) return;
|
if (!snapshot) {
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "failed", error: "生图参数无效" });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
setElapsedMs(0);
|
setElapsedMs(0);
|
||||||
setRunning(true);
|
setRunning(true);
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "running", error: undefined });
|
||||||
setPreviewLog(null);
|
setPreviewLog(null);
|
||||||
setResults(Array.from({ length: generationCount }, () => ({ id: nanoid(), status: "pending" })));
|
setResults(Array.from({ length: generationCount }, () => ({ id: nanoid(), status: "pending" })));
|
||||||
const batchStartedAt = performance.now();
|
const batchStartedAt = performance.now();
|
||||||
@@ -169,6 +179,8 @@ export default function ImagePage() {
|
|||||||
const successCount = successImages.length;
|
const successCount = successImages.length;
|
||||||
const failCount = generationCount - successCount;
|
const failCount = generationCount - successCount;
|
||||||
const failed = result.find((item): item is PromiseRejectedResult => item.status === "rejected");
|
const failed = result.find((item): item is PromiseRejectedResult => item.status === "rejected");
|
||||||
|
const error = failed?.reason instanceof Error ? failed.reason.message : failCount ? "生成失败" : undefined;
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: successCount ? "succeeded" : "failed", successCount, failCount, error: successCount ? undefined : error });
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const logImages = await Promise.all(
|
const logImages = await Promise.all(
|
||||||
@@ -202,8 +214,15 @@ export default function ImagePage() {
|
|||||||
processedCommandRef.current = imageCommand.nonce;
|
processedCommandRef.current = imageCommand.nonce;
|
||||||
clearImageCommand();
|
clearImageCommand();
|
||||||
if (typeof imageCommand.prompt === "string") setPrompt(imageCommand.prompt);
|
if (typeof imageCommand.prompt === "string") setPrompt(imageCommand.prompt);
|
||||||
if (imageCommand.run && !running) setAutoRunToken((value) => value + 1);
|
if (imageCommand.run && running) {
|
||||||
}, [imageCommand, clearImageCommand, running]);
|
if (imageCommand.taskId) updateAgentTask(imageCommand.taskId, { status: "failed", error: "生图工作台已有任务正在运行" });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (imageCommand.run) {
|
||||||
|
agentTaskIdRef.current = imageCommand.taskId;
|
||||||
|
setAutoRunToken((value) => value + 1);
|
||||||
|
}
|
||||||
|
}, [imageCommand, clearImageCommand, running, updateAgentTask]);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (!autoRunToken) return;
|
if (!autoRunToken) return;
|
||||||
|
|||||||
@@ -0,0 +1,108 @@
|
|||||||
|
import { App, Button, Input, Modal, Space } from "antd";
|
||||||
|
import { useEffect, useState } from "react";
|
||||||
|
|
||||||
|
import type { PersonalPrompt, PersonalPromptInput } from "@/stores/use-prompt-store";
|
||||||
|
|
||||||
|
const EMPTY_PROMPT: PersonalPromptInput = {
|
||||||
|
title: "",
|
||||||
|
prompt: "",
|
||||||
|
description: "",
|
||||||
|
coverUrl: "",
|
||||||
|
referenceImageUrls: [],
|
||||||
|
tags: [],
|
||||||
|
};
|
||||||
|
|
||||||
|
export function MyPromptEditorDialog({ open, prompt, onSave, onClose }: { open: boolean; prompt: PersonalPrompt | null; onSave: (value: PersonalPromptInput) => void; onClose: () => void }) {
|
||||||
|
const { message } = App.useApp();
|
||||||
|
const [draft, setDraft] = useState<PersonalPromptInput>(EMPTY_PROMPT);
|
||||||
|
const [tags, setTags] = useState("");
|
||||||
|
const [referenceImages, setReferenceImages] = useState("");
|
||||||
|
|
||||||
|
useEffect(() => {
|
||||||
|
if (!open) return;
|
||||||
|
setDraft(prompt ? toInput(prompt) : EMPTY_PROMPT);
|
||||||
|
setTags(prompt?.tags.join(", ") || "");
|
||||||
|
setReferenceImages(prompt?.referenceImageUrls.join("\n") || "");
|
||||||
|
}, [open, prompt]);
|
||||||
|
|
||||||
|
const patch = (value: Partial<PersonalPromptInput>) => setDraft((current) => ({ ...current, ...value }));
|
||||||
|
const save = () => {
|
||||||
|
if (!draft.title.trim()) return message.warning("请输入标题");
|
||||||
|
if (!draft.prompt.trim()) return message.warning("请输入提示词");
|
||||||
|
onSave({
|
||||||
|
...draft,
|
||||||
|
title: draft.title.trim(),
|
||||||
|
prompt: draft.prompt.trim(),
|
||||||
|
description: draft.description.trim(),
|
||||||
|
coverUrl: draft.coverUrl.trim(),
|
||||||
|
tags: splitValues(tags, /[,,\n]/),
|
||||||
|
referenceImageUrls: splitValues(referenceImages, /\n/),
|
||||||
|
});
|
||||||
|
onClose();
|
||||||
|
};
|
||||||
|
|
||||||
|
return (
|
||||||
|
<Modal
|
||||||
|
title={prompt ? "编辑提示词" : "新增提示词"}
|
||||||
|
open={open}
|
||||||
|
onCancel={onClose}
|
||||||
|
width={680}
|
||||||
|
footer={
|
||||||
|
<Space>
|
||||||
|
<Button onClick={onClose}>取消</Button>
|
||||||
|
<Button type="primary" onClick={save}>
|
||||||
|
保存
|
||||||
|
</Button>
|
||||||
|
</Space>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
<div className="grid gap-4 pt-2">
|
||||||
|
<label>
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">标题</span>
|
||||||
|
<Input value={draft.title} onChange={(event) => patch({ title: event.target.value })} placeholder="例如:白底商品图" />
|
||||||
|
</label>
|
||||||
|
<label>
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">提示词</span>
|
||||||
|
<Input.TextArea rows={7} value={draft.prompt} onChange={(event) => patch({ prompt: event.target.value })} placeholder="输入可直接使用的提示词" />
|
||||||
|
</label>
|
||||||
|
<label>
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">说明(可选)</span>
|
||||||
|
<Input.TextArea rows={2} value={draft.description} onChange={(event) => patch({ description: event.target.value })} />
|
||||||
|
</label>
|
||||||
|
<div className="grid gap-4 sm:grid-cols-2">
|
||||||
|
<label>
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">封面 URL(可选)</span>
|
||||||
|
<Input value={draft.coverUrl} onChange={(event) => patch({ coverUrl: event.target.value })} placeholder="https://..." />
|
||||||
|
</label>
|
||||||
|
<label>
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">标签(可选)</span>
|
||||||
|
<Input value={tags} onChange={(event) => setTags(event.target.value)} placeholder="商品, 摄影" />
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
<label>
|
||||||
|
<span className="mb-1.5 block text-sm font-medium">参考图 URL(可选,每行一个)</span>
|
||||||
|
<Input.TextArea rows={3} value={referenceImages} onChange={(event) => setReferenceImages(event.target.value)} placeholder={"https://...\nhttps://..."} />
|
||||||
|
</label>
|
||||||
|
</div>
|
||||||
|
</Modal>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
function toInput(prompt: PersonalPrompt): PersonalPromptInput {
|
||||||
|
return {
|
||||||
|
title: prompt.title,
|
||||||
|
prompt: prompt.prompt,
|
||||||
|
description: prompt.description,
|
||||||
|
coverUrl: prompt.coverUrl,
|
||||||
|
referenceImageUrls: prompt.referenceImageUrls,
|
||||||
|
tags: prompt.tags,
|
||||||
|
imageMode: prompt.imageMode,
|
||||||
|
imageModel: prompt.imageModel,
|
||||||
|
imageSize: prompt.imageSize,
|
||||||
|
imageCount: prompt.imageCount,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
function splitValues(value: string, separator: RegExp) {
|
||||||
|
return Array.from(new Set(value.split(separator).map((item) => item.trim()).filter(Boolean)));
|
||||||
|
}
|
||||||
@@ -1,9 +1,9 @@
|
|||||||
import { Copy, FolderPlus } from "lucide-react";
|
import { BookmarkPlus, Copy, FileText, FolderPlus } from "lucide-react";
|
||||||
import { Button, Modal, Space, Tag } from "antd";
|
import { Button, Modal, Space, Tag } from "antd";
|
||||||
|
|
||||||
import { formatPromptDate, type Prompt } from "@/services/api/prompts";
|
import { formatPromptDate, type Prompt } from "@/services/api/prompts";
|
||||||
|
|
||||||
export function PromptDetailDialog({ prompt, onClose, onCopy, onSaveAsset }: { prompt: Prompt | null; onClose: () => void; onCopy: (prompt: string) => void; onSaveAsset?: (prompt: Prompt) => void }) {
|
export function PromptDetailDialog({ prompt, onClose, onCopy, onSaveAsset, onSavePrompt }: { prompt: Prompt | null; onClose: () => void; onCopy: (prompt: string) => void; onSaveAsset?: (prompt: Prompt) => void; onSavePrompt?: (prompt: Prompt) => void }) {
|
||||||
return (
|
return (
|
||||||
<>
|
<>
|
||||||
<Modal title={prompt?.title} open={Boolean(prompt)} onCancel={onClose} footer={null} width={860}>
|
<Modal title={prompt?.title} open={Boolean(prompt)} onCancel={onClose} footer={null} width={860}>
|
||||||
@@ -11,7 +11,8 @@ export function PromptDetailDialog({ prompt, onClose, onCopy, onSaveAsset }: { p
|
|||||||
<>
|
<>
|
||||||
<div className="grid gap-5 md:grid-cols-[300px_minmax(0,1fr)]">
|
<div className="grid gap-5 md:grid-cols-[300px_minmax(0,1fr)]">
|
||||||
<div className="space-y-3">
|
<div className="space-y-3">
|
||||||
<img src={prompt.coverUrl} alt={prompt.title} className="aspect-[4/3] w-full rounded-lg object-cover" />
|
{prompt.coverUrl ? <img src={prompt.coverUrl} alt={prompt.title} className="aspect-[4/3] w-full rounded-lg object-cover" /> : <div className="grid aspect-[4/3] w-full place-items-center rounded-lg bg-stone-100 text-stone-400 dark:bg-stone-900 dark:text-stone-600"><FileText className="size-9" /></div>}
|
||||||
|
{prompt.referenceImageUrls.length > 1 ? <div className="grid grid-cols-3 gap-2">{prompt.referenceImageUrls.filter((url) => url !== prompt.coverUrl).slice(0, 6).map((url) => <img key={url} src={url} alt="" className="aspect-square w-full rounded-md object-cover" loading="lazy" />)}</div> : null}
|
||||||
{prompt.preview ? <pre className="max-h-60 overflow-auto whitespace-pre-wrap rounded-lg bg-stone-100 p-3 text-xs leading-5 text-stone-600 dark:bg-stone-900 dark:text-stone-300">{prompt.preview}</pre> : null}
|
{prompt.preview ? <pre className="max-h-60 overflow-auto whitespace-pre-wrap rounded-lg bg-stone-100 p-3 text-xs leading-5 text-stone-600 dark:bg-stone-900 dark:text-stone-300">{prompt.preview}</pre> : null}
|
||||||
</div>
|
</div>
|
||||||
<div className="min-w-0">
|
<div className="min-w-0">
|
||||||
@@ -22,10 +23,9 @@ export function PromptDetailDialog({ prompt, onClose, onCopy, onSaveAsset }: { p
|
|||||||
</Tag>
|
</Tag>
|
||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
|
{prompt.description ? <p className="mt-4 text-sm leading-6 text-stone-500 dark:text-stone-400">{prompt.description}</p> : null}
|
||||||
<p className="mt-4 whitespace-pre-wrap text-sm leading-7 text-stone-800 dark:text-stone-300">{prompt.prompt}</p>
|
<p className="mt-4 whitespace-pre-wrap text-sm leading-7 text-stone-800 dark:text-stone-300">{prompt.prompt}</p>
|
||||||
<div className="mt-4 text-xs text-stone-500 dark:text-stone-400">
|
{prompt.createdAt || prompt.updatedAt ? <div className="mt-4 text-xs text-stone-500 dark:text-stone-400">{prompt.createdAt ? `创建:${formatPromptDate(prompt.createdAt)}` : null}{prompt.createdAt && prompt.updatedAt ? " · " : null}{prompt.updatedAt ? `更新:${formatPromptDate(prompt.updatedAt)}` : null}</div> : null}
|
||||||
创建:{formatPromptDate(prompt.createdAt)} · 更新:{formatPromptDate(prompt.updatedAt)}
|
|
||||||
</div>
|
|
||||||
<Space wrap className="mt-5">
|
<Space wrap className="mt-5">
|
||||||
<Button type="primary" icon={<Copy className="size-4" />} onClick={() => onCopy(prompt.prompt)}>
|
<Button type="primary" icon={<Copy className="size-4" />} onClick={() => onCopy(prompt.prompt)}>
|
||||||
复制提示词
|
复制提示词
|
||||||
@@ -35,6 +35,11 @@ export function PromptDetailDialog({ prompt, onClose, onCopy, onSaveAsset }: { p
|
|||||||
加入我的资产
|
加入我的资产
|
||||||
</Button>
|
</Button>
|
||||||
) : null}
|
) : null}
|
||||||
|
{onSavePrompt ? (
|
||||||
|
<Button icon={<BookmarkPlus className="size-4" />} onClick={() => onSavePrompt(prompt)}>
|
||||||
|
保存到我的提示词
|
||||||
|
</Button>
|
||||||
|
) : null}
|
||||||
</Space>
|
</Space>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
+105
-80
@@ -1,29 +1,41 @@
|
|||||||
import { FolderPlus, Search } from "lucide-react";
|
import { BookmarkPlus, FolderPlus, Pencil, Plus, Search, Trash2 } from "lucide-react";
|
||||||
import { type UIEvent, useEffect, useState } from "react";
|
import { type ReactNode, type UIEvent, useEffect, useMemo, useState } from "react";
|
||||||
import { App, Button, Empty, Input, Spin, Tag } from "antd";
|
import { App, Button, Empty, Input, Popconfirm, Space, Spin, Tabs, Tag, Tooltip } from "antd";
|
||||||
|
|
||||||
import { PromptCard } from "@/components/prompts/prompt-card";
|
import { PromptCard } from "@/components/prompts/prompt-card";
|
||||||
import { usePromptList } from "@/components/prompts/use-prompt-list";
|
import { usePromptList } from "@/components/prompts/use-prompt-list";
|
||||||
|
import { MyPromptEditorDialog } from "./components/my-prompt-editor-dialog";
|
||||||
import { PromptDetailDialog } from "./components/prompt-detail-dialog";
|
import { PromptDetailDialog } from "./components/prompt-detail-dialog";
|
||||||
import { useCopyText } from "@/hooks/use-copy-text";
|
import { useCopyText } from "@/hooks/use-copy-text";
|
||||||
import { cn } from "@/lib/utils";
|
import { cn } from "@/lib/utils";
|
||||||
import { useAssetStore } from "@/stores/use-asset-store";
|
import { useAssetStore } from "@/stores/use-asset-store";
|
||||||
import { ALL_PROMPTS_OPTION, type Prompt } from "@/services/api/prompts";
|
import { usePromptStore, type PersonalPrompt, type PersonalPromptInput } from "@/stores/use-prompt-store";
|
||||||
|
import { ALL_PROMPTS_OPTION, personalPromptToPrompt, type Prompt } from "@/services/api/prompts";
|
||||||
|
|
||||||
export default function PromptsPage() {
|
export default function PromptsPage() {
|
||||||
const { message } = App.useApp();
|
const { message } = App.useApp();
|
||||||
|
const [activeTab, setActiveTab] = useState("library");
|
||||||
const [titleKeyword, setTitleKeyword] = useState("");
|
const [titleKeyword, setTitleKeyword] = useState("");
|
||||||
const [selectedTags, setSelectedTags] = useState<string[]>([]);
|
const [selectedTags, setSelectedTags] = useState<string[]>([]);
|
||||||
const [selectedCategory, setSelectedCategory] = useState(ALL_PROMPTS_OPTION);
|
const [selectedCategory, setSelectedCategory] = useState(ALL_PROMPTS_OPTION);
|
||||||
const [selectedPrompt, setSelectedPrompt] = useState<Prompt | null>(null);
|
const [selectedPrompt, setSelectedPrompt] = useState<Prompt | null>(null);
|
||||||
|
const [editorOpen, setEditorOpen] = useState(false);
|
||||||
|
const [editingPrompt, setEditingPrompt] = useState<PersonalPrompt | null>(null);
|
||||||
const addAsset = useAssetStore((state) => state.addAsset);
|
const addAsset = useAssetStore((state) => state.addAsset);
|
||||||
|
const personalPrompts = usePromptStore((state) => state.prompts);
|
||||||
|
const addPrompt = usePromptStore((state) => state.addPrompt);
|
||||||
|
const updatePrompt = usePromptStore((state) => state.updatePrompt);
|
||||||
|
const removePrompt = usePromptStore((state) => state.removePrompt);
|
||||||
const copyText = useCopyText();
|
const copyText = useCopyText();
|
||||||
const { query, items: promptItems, tags: promptTags, categories: promptCategoryOptions, total: totalPrompts } = usePromptList({ keyword: titleKeyword, tags: selectedTags, category: selectedCategory });
|
const { query, items: promptItems, tags: promptTags, categories: promptCategoryOptions, total: totalPrompts } = usePromptList({ keyword: titleKeyword, tags: selectedTags, category: selectedCategory, enabled: activeTab === "library" });
|
||||||
|
const filteredPersonalPrompts = useMemo(() => {
|
||||||
|
const keyword = titleKeyword.trim().toLowerCase();
|
||||||
|
if (!keyword) return personalPrompts;
|
||||||
|
return personalPrompts.filter((item) => [item.title, item.prompt, item.description, ...item.tags].join(" ").toLowerCase().includes(keyword));
|
||||||
|
}, [personalPrompts, titleKeyword]);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (query.isError) {
|
if (query.isError) message.error(query.error instanceof Error ? query.error.message : "获取提示词失败");
|
||||||
message.error(query.error instanceof Error ? query.error.message : "获取提示词失败");
|
|
||||||
}
|
|
||||||
}, [message, query.error, query.isError]);
|
}, [message, query.error, query.isError]);
|
||||||
|
|
||||||
const toggleTag = (tag: string) => {
|
const toggleTag = (tag: string) => {
|
||||||
@@ -36,91 +48,104 @@ export default function PromptsPage() {
|
|||||||
message.success("已加入我的资产");
|
message.success("已加入我的资产");
|
||||||
};
|
};
|
||||||
|
|
||||||
const handleListScroll = (event: UIEvent<HTMLDivElement>) => {
|
const saveToMyPrompts = (item: Prompt) => {
|
||||||
const target = event.currentTarget;
|
addPrompt(toPersonalInput(item));
|
||||||
if (query.hasNextPage && !query.isFetchingNextPage && target.scrollTop + target.clientHeight >= target.scrollHeight - 160) {
|
message.success("已保存到我的提示词");
|
||||||
void query.fetchNextPage();
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
const openNewPrompt = () => {
|
||||||
|
setEditingPrompt(null);
|
||||||
|
setEditorOpen(true);
|
||||||
|
};
|
||||||
|
|
||||||
|
const openEditPrompt = (item: PersonalPrompt) => {
|
||||||
|
setEditingPrompt(item);
|
||||||
|
setEditorOpen(true);
|
||||||
|
};
|
||||||
|
|
||||||
|
const savePersonalPrompt = (input: PersonalPromptInput) => {
|
||||||
|
if (editingPrompt) updatePrompt(editingPrompt.id, input);
|
||||||
|
else addPrompt(input);
|
||||||
|
message.success(editingPrompt ? "提示词已更新" : "提示词已添加");
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleListScroll = (event: UIEvent<HTMLDivElement>) => {
|
||||||
|
if (activeTab !== "library") return;
|
||||||
|
const target = event.currentTarget;
|
||||||
|
if (query.hasNextPage && !query.isFetchingNextPage && target.scrollTop + target.clientHeight >= target.scrollHeight - 160) void query.fetchNextPage();
|
||||||
|
};
|
||||||
|
|
||||||
|
const personalItems = filteredPersonalPrompts.map(personalPromptToPrompt);
|
||||||
|
const visibleCount = activeTab === "library" ? totalPrompts : filteredPersonalPrompts.length;
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="flex h-full flex-col overflow-hidden bg-background text-stone-800 dark:text-stone-100">
|
<div className="flex h-full flex-col overflow-hidden bg-background text-stone-800 dark:text-stone-100">
|
||||||
<main
|
<main className="min-h-0 flex-1 overflow-y-auto bg-background bg-[radial-gradient(#e5e7eb_1px,transparent_1px)] px-6 py-8 [background-size:16px_16px] dark:bg-[radial-gradient(rgba(245,245,244,.16)_1px,transparent_1px)]" onScroll={handleListScroll}>
|
||||||
className="min-h-0 flex-1 overflow-y-auto bg-background bg-[radial-gradient(#e5e7eb_1px,transparent_1px)] px-6 py-8 [background-size:16px_16px] dark:bg-[radial-gradient(rgba(245,245,244,.16)_1px,transparent_1px)]"
|
<div className="mx-auto max-w-7xl pb-8">
|
||||||
onScroll={handleListScroll}
|
<div className="flex flex-wrap items-end justify-between gap-4">
|
||||||
>
|
<div>
|
||||||
<div className="pb-8">
|
<h1 className="text-3xl font-semibold text-stone-950 dark:text-stone-100">提示词中心</h1>
|
||||||
<div className="mx-auto max-w-5xl text-center">
|
<p className="mt-2 text-sm text-stone-500 dark:text-stone-400">当前共 {visibleCount} 条提示词</p>
|
||||||
<h1 className="text-4xl font-semibold tracking-tight text-stone-950 dark:text-stone-100">提示词中心</h1>
|
|
||||||
<p className="mt-3 text-sm text-stone-500 dark:text-stone-400">共 {totalPrompts} 条提示词,按标题、标签与分类快速查找灵感。</p>
|
|
||||||
</div>
|
|
||||||
{query.isLoading ? (
|
|
||||||
<div className="flex h-60 items-center justify-center">
|
|
||||||
<Spin />
|
|
||||||
</div>
|
</div>
|
||||||
) : null}
|
{activeTab === "personal" ? (
|
||||||
{!query.isLoading ? (
|
<Button type="primary" icon={<Plus className="size-4" />} onClick={openNewPrompt}>
|
||||||
<>
|
新增提示词
|
||||||
<div className="mx-auto mt-8 w-full max-w-2xl">
|
</Button>
|
||||||
<Input size="large" className="w-full" prefix={<Search className="size-4 text-stone-400" />} value={titleKeyword} placeholder="按标题查询" onChange={(event) => setTitleKeyword(event.target.value)} />
|
) : null}
|
||||||
</div>
|
</div>
|
||||||
<div className="mx-auto mt-6 grid max-w-6xl gap-3 text-left">
|
<Tabs className="mt-5" activeKey={activeTab} onChange={setActiveTab} items={[{ key: "library", label: "提示词库" }, { key: "personal", label: `我的提示词 (${personalPrompts.length})` }]} />
|
||||||
<div className="grid gap-2 sm:grid-cols-[56px_minmax(0,1fr)] sm:items-start">
|
<div className="mx-auto mt-2 w-full max-w-2xl">
|
||||||
<div className="pt-2 text-xs font-medium text-stone-500 dark:text-stone-400">分类</div>
|
<Input size="large" prefix={<Search className="size-4 text-stone-400" />} value={titleKeyword} placeholder="搜索标题、内容或标签" onChange={(event) => setTitleKeyword(event.target.value)} />
|
||||||
<div className="flex flex-wrap gap-2">
|
</div>
|
||||||
{promptCategoryOptions.map((category) => (
|
{activeTab === "library" ? (
|
||||||
<Tag.CheckableTag key={category} checked={selectedCategory === category} className={cn("prompt-filter-tag", selectedCategory === category && "is-active")} onChange={() => setSelectedCategory(category)}>
|
<div className="mx-auto mt-6 grid max-w-6xl gap-3 text-left">
|
||||||
{category}
|
<PromptFilter label="分类" options={promptCategoryOptions} selected={selectedCategory} onChange={setSelectedCategory} />
|
||||||
</Tag.CheckableTag>
|
<div className="grid gap-2 sm:grid-cols-[56px_minmax(0,1fr)] sm:items-start">
|
||||||
))}
|
<div className="pt-2 text-xs font-medium text-stone-500 dark:text-stone-400">标签</div>
|
||||||
</div>
|
<div className="flex flex-wrap gap-2">
|
||||||
</div>
|
{promptTags.map((tag) => {
|
||||||
<div className="grid gap-2 sm:grid-cols-[56px_minmax(0,1fr)] sm:items-start">
|
const active = tag === ALL_PROMPTS_OPTION ? selectedTags.length === 0 : selectedTags.includes(tag);
|
||||||
<div className="pt-2 text-xs font-medium text-stone-500 dark:text-stone-400">标签</div>
|
return <Tag.CheckableTag key={tag} checked={active} className={cn("prompt-filter-tag", active && "is-active")} onChange={() => toggleTag(tag)}>{tag}</Tag.CheckableTag>;
|
||||||
<div className="flex flex-wrap gap-2">
|
})}
|
||||||
{promptTags.map((tag) => (
|
|
||||||
<Tag.CheckableTag
|
|
||||||
key={tag}
|
|
||||||
checked={tag === ALL_PROMPTS_OPTION ? selectedTags.length === 0 : selectedTags.includes(tag)}
|
|
||||||
className={cn("prompt-filter-tag", (tag === ALL_PROMPTS_OPTION ? selectedTags.length === 0 : selectedTags.includes(tag)) && "is-active")}
|
|
||||||
onChange={() => toggleTag(tag)}
|
|
||||||
>
|
|
||||||
{tag}
|
|
||||||
</Tag.CheckableTag>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</>
|
</div>
|
||||||
) : null}
|
) : null}
|
||||||
</div>
|
</div>
|
||||||
|
|
||||||
{!query.isLoading ? (
|
{activeTab === "library" && query.isLoading ? <div className="flex h-60 items-center justify-center"><Spin /></div> : null}
|
||||||
<div>
|
{activeTab === "library" && !query.isLoading ? (
|
||||||
<div className="mx-auto grid max-w-7xl gap-5 sm:grid-cols-2 xl:grid-cols-3 2xl:grid-cols-4">
|
<PromptGrid items={promptItems} onOpen={setSelectedPrompt} renderActions={(item) => <><Button size="small" icon={<BookmarkPlus className="size-3.5" />} onClick={() => saveToMyPrompts(item)}>保存</Button><Tooltip title="加入我的资产"><Button type="text" size="small" icon={<FolderPlus className="size-3.5" />} onClick={() => savePromptAsset(item)} /></Tooltip></>} onCopy={(item) => copyText(item.prompt, "提示词已复制")} emptyText="没有找到匹配的提示词" />
|
||||||
{promptItems.map((item) => (
|
|
||||||
<PromptCard
|
|
||||||
key={item.id}
|
|
||||||
item={item}
|
|
||||||
onOpen={() => setSelectedPrompt(item)}
|
|
||||||
onCopy={() => copyText(item.prompt, "提示词已复制")}
|
|
||||||
extraAction={
|
|
||||||
<Button size="small" icon={<FolderPlus className="size-3.5" />} onClick={() => savePromptAsset(item)}>
|
|
||||||
加入我的资产
|
|
||||||
</Button>
|
|
||||||
}
|
|
||||||
/>
|
|
||||||
))}
|
|
||||||
</div>
|
|
||||||
{promptItems.length === 0 ? <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} description="没有找到匹配的提示词" className="py-16" /> : null}
|
|
||||||
<div className="mx-auto mt-6 max-w-7xl text-center text-xs text-stone-500 dark:text-stone-400">
|
|
||||||
{query.isFetchingNextPage ? "加载中..." : query.hasNextPage ? "继续向下滚动加载更多" : promptItems.length > 0 ? "已经到底了" : null}
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
) : null}
|
) : null}
|
||||||
|
{activeTab === "personal" ? (
|
||||||
|
<PromptGrid
|
||||||
|
items={personalItems}
|
||||||
|
onOpen={setSelectedPrompt}
|
||||||
|
onCopy={(item) => copyText(item.prompt, "提示词已复制")}
|
||||||
|
renderActions={(item) => {
|
||||||
|
const personal = personalPrompts.find((prompt) => prompt.id === item.id)!;
|
||||||
|
return <Space size={0}><Tooltip title="编辑"><Button type="text" size="small" icon={<Pencil className="size-3.5" />} onClick={() => openEditPrompt(personal)} /></Tooltip><Popconfirm title="删除这条提示词?" okText="删除" cancelText="取消" onConfirm={() => removePrompt(item.id)}><Tooltip title="删除"><Button type="text" danger size="small" icon={<Trash2 className="size-3.5" />} /></Tooltip></Popconfirm></Space>;
|
||||||
|
}}
|
||||||
|
emptyText="还没有保存提示词"
|
||||||
|
/>
|
||||||
|
) : null}
|
||||||
|
{activeTab === "library" ? <div className="mx-auto mt-6 max-w-7xl text-center text-xs text-stone-500 dark:text-stone-400">{query.isFetchingNextPage ? "加载中..." : query.hasNextPage ? "继续向下滚动加载更多" : promptItems.length > 0 ? "已经到底了" : null}</div> : null}
|
||||||
</main>
|
</main>
|
||||||
|
|
||||||
<PromptDetailDialog prompt={selectedPrompt} onClose={() => setSelectedPrompt(null)} onCopy={(prompt) => copyText(prompt, "提示词已复制")} onSaveAsset={savePromptAsset} />
|
<PromptDetailDialog prompt={selectedPrompt} onClose={() => setSelectedPrompt(null)} onCopy={(prompt) => copyText(prompt, "提示词已复制")} onSaveAsset={selectedPrompt?.sourceId === "personal" ? undefined : savePromptAsset} onSavePrompt={selectedPrompt?.sourceId === "personal" ? undefined : saveToMyPrompts} />
|
||||||
|
<MyPromptEditorDialog open={editorOpen} prompt={editingPrompt} onSave={savePersonalPrompt} onClose={() => setEditorOpen(false)} />
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function PromptFilter({ label, options, selected, onChange }: { label: string; options: string[]; selected: string; onChange: (value: string) => void }) {
|
||||||
|
return <div className="grid gap-2 sm:grid-cols-[56px_minmax(0,1fr)] sm:items-start"><div className="pt-2 text-xs font-medium text-stone-500 dark:text-stone-400">{label}</div><div className="flex flex-wrap gap-2">{options.map((option) => <Tag.CheckableTag key={option} checked={selected === option} className={cn("prompt-filter-tag", selected === option && "is-active")} onChange={() => onChange(option)}>{option}</Tag.CheckableTag>)}</div></div>;
|
||||||
|
}
|
||||||
|
|
||||||
|
function PromptGrid({ items, onOpen, onCopy, renderActions, emptyText }: { items: Prompt[]; onOpen: (item: Prompt) => void; onCopy: (item: Prompt) => void; renderActions: (item: Prompt) => ReactNode; emptyText: string }) {
|
||||||
|
return <div><div className="mx-auto grid max-w-7xl gap-5 sm:grid-cols-2 xl:grid-cols-3 2xl:grid-cols-4">{items.map((item) => <PromptCard key={`${item.sourceId}:${item.id}`} item={item} onOpen={() => onOpen(item)} onCopy={() => onCopy(item)} extraAction={renderActions(item)} />)}</div>{items.length === 0 ? <Empty image={Empty.PRESENTED_IMAGE_SIMPLE} description={emptyText} className="py-16" /> : null}</div>;
|
||||||
|
}
|
||||||
|
|
||||||
|
function toPersonalInput(item: Prompt): PersonalPromptInput {
|
||||||
|
return { title: item.title, prompt: item.prompt, description: item.description, coverUrl: item.coverUrl, referenceImageUrls: item.referenceImageUrls, tags: item.tags, imageMode: item.imageMode, imageModel: item.imageModel, imageSize: item.imageSize, imageCount: item.imageCount };
|
||||||
|
}
|
||||||
|
|||||||
@@ -97,7 +97,9 @@ export default function VideoPage() {
|
|||||||
const [autoRunToken, setAutoRunToken] = useState(0);
|
const [autoRunToken, setAutoRunToken] = useState(0);
|
||||||
const videoCommand = useWorkbenchAgentStore((state) => state.videoCommand);
|
const videoCommand = useWorkbenchAgentStore((state) => state.videoCommand);
|
||||||
const clearVideoCommand = useWorkbenchAgentStore((state) => state.clearVideoCommand);
|
const clearVideoCommand = useWorkbenchAgentStore((state) => state.clearVideoCommand);
|
||||||
|
const updateAgentTask = useWorkbenchAgentStore((state) => state.updateTask);
|
||||||
const processedCommandRef = useRef(0);
|
const processedCommandRef = useRef(0);
|
||||||
|
const agentTaskIdRef = useRef<string | undefined>(undefined);
|
||||||
|
|
||||||
const model = effectiveConfig.videoModel || effectiveConfig.model;
|
const model = effectiveConfig.videoModel || effectiveConfig.model;
|
||||||
const canGenerate = Boolean(prompt.trim());
|
const canGenerate = Boolean(prompt.trim());
|
||||||
@@ -170,10 +172,16 @@ export default function VideoPage() {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
const generate = async () => {
|
const generate = async () => {
|
||||||
|
const agentTaskId = agentTaskIdRef.current;
|
||||||
|
agentTaskIdRef.current = undefined;
|
||||||
const snapshot = buildRequestSnapshot();
|
const snapshot = buildRequestSnapshot();
|
||||||
if (!snapshot) return;
|
if (!snapshot) {
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "failed", error: "视频生成参数无效" });
|
||||||
|
return;
|
||||||
|
}
|
||||||
setElapsedMs(0);
|
setElapsedMs(0);
|
||||||
setRunning(true);
|
setRunning(true);
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "running", error: undefined });
|
||||||
setPreviewLog(null);
|
setPreviewLog(null);
|
||||||
setResults([{ id: nanoid(), status: "pending" }]);
|
setResults([{ id: nanoid(), status: "pending" }]);
|
||||||
const batchStartedAt = performance.now();
|
const batchStartedAt = performance.now();
|
||||||
@@ -181,11 +189,12 @@ export default function VideoPage() {
|
|||||||
try {
|
try {
|
||||||
const task = await createVideoGenerationTask(snapshot.config, snapshot.text, snapshot.references, snapshot.videoReferences, snapshot.audioReferences);
|
const task = await createVideoGenerationTask(snapshot.config, snapshot.text, snapshot.references, snapshot.videoReferences, snapshot.audioReferences);
|
||||||
const log = buildLog({ prompt: snapshot.text, model, config: snapshot.config, references: snapshot.references, videoReferences: snapshot.videoReferences, audioReferences: snapshot.audioReferences, durationMs: 0, status: "生成中", task });
|
const log = buildLog({ prompt: snapshot.text, model, config: snapshot.config, references: snapshot.references, videoReferences: snapshot.videoReferences, audioReferences: snapshot.audioReferences, durationMs: 0, status: "生成中", task });
|
||||||
await saveLog(log);
|
await saveLog(log, false);
|
||||||
void pollGenerationLog(log, snapshot.config);
|
void pollGenerationLog(log, snapshot.config, agentTaskId);
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
const errorMessage = error instanceof Error ? error.message : "生成失败";
|
const errorMessage = error instanceof Error ? error.message : "生成失败";
|
||||||
setResults([{ id: nanoid(), status: "failed", error: errorMessage }]);
|
setResults([{ id: nanoid(), status: "failed", error: errorMessage }]);
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "failed", successCount: 0, failCount: 1, error: errorMessage });
|
||||||
await saveLog(buildLog({ prompt: snapshot.text, model, config: snapshot.config, references: snapshot.references, videoReferences: snapshot.videoReferences, audioReferences: snapshot.audioReferences, durationMs: performance.now() - batchStartedAt, status: "失败", error: errorMessage }));
|
await saveLog(buildLog({ prompt: snapshot.text, model, config: snapshot.config, references: snapshot.references, videoReferences: snapshot.videoReferences, audioReferences: snapshot.audioReferences, durationMs: performance.now() - batchStartedAt, status: "失败", error: errorMessage }));
|
||||||
message.error(errorMessage);
|
message.error(errorMessage);
|
||||||
setRunning(false);
|
setRunning(false);
|
||||||
@@ -198,8 +207,15 @@ export default function VideoPage() {
|
|||||||
processedCommandRef.current = videoCommand.nonce;
|
processedCommandRef.current = videoCommand.nonce;
|
||||||
clearVideoCommand();
|
clearVideoCommand();
|
||||||
if (typeof videoCommand.prompt === "string") setPrompt(videoCommand.prompt);
|
if (typeof videoCommand.prompt === "string") setPrompt(videoCommand.prompt);
|
||||||
if (videoCommand.run && !running) setAutoRunToken((value) => value + 1);
|
if (videoCommand.run && running) {
|
||||||
}, [videoCommand, clearVideoCommand, running]);
|
if (videoCommand.taskId) updateAgentTask(videoCommand.taskId, { status: "failed", error: "视频工作台已有任务正在运行" });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (videoCommand.run) {
|
||||||
|
agentTaskIdRef.current = videoCommand.taskId;
|
||||||
|
setAutoRunToken((value) => value + 1);
|
||||||
|
}
|
||||||
|
}, [videoCommand, clearVideoCommand, running, updateAgentTask]);
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
if (!autoRunToken) return;
|
if (!autoRunToken) return;
|
||||||
@@ -276,7 +292,7 @@ export default function VideoPage() {
|
|||||||
.filter((log) => selectedLogIds.includes(log.id))
|
.filter((log) => selectedLogIds.includes(log.id))
|
||||||
.map((log) => log.video?.storageKey)
|
.map((log) => log.video?.storageKey)
|
||||||
.filter((key): key is string => Boolean(key));
|
.filter((key): key is string => Boolean(key));
|
||||||
void Promise.all([deleteStoredMedia(mediaKeys), ...selectedLogIds.map((id) => logStore.removeItem(id))]).then(refreshLogs);
|
void Promise.all([deleteStoredMedia(mediaKeys), ...selectedLogIds.map((id) => logStore.removeItem(id))]).then(() => refreshLogs());
|
||||||
if (previewLog && selectedLogIds.includes(previewLog.id)) {
|
if (previewLog && selectedLogIds.includes(previewLog.id)) {
|
||||||
setPreviewLog(null);
|
setPreviewLog(null);
|
||||||
setResults([]);
|
setResults([]);
|
||||||
@@ -285,15 +301,15 @@ export default function VideoPage() {
|
|||||||
setDeleteConfirmOpen(false);
|
setDeleteConfirmOpen(false);
|
||||||
};
|
};
|
||||||
|
|
||||||
const saveLog = async (log: GenerationLog) => {
|
const saveLog = async (log: GenerationLog, resumePending = true) => {
|
||||||
await logStore.setItem(log.id, serializeLog(log));
|
await logStore.setItem(log.id, serializeLog(log));
|
||||||
await refreshLogs();
|
await refreshLogs(resumePending);
|
||||||
};
|
};
|
||||||
|
|
||||||
const refreshLogs = async () => {
|
const refreshLogs = async (resumePending = true) => {
|
||||||
const nextLogs = await readStoredLogs();
|
const nextLogs = await readStoredLogs();
|
||||||
setLogs(nextLogs);
|
setLogs(nextLogs);
|
||||||
resumePendingLogs(nextLogs);
|
if (resumePending) resumePendingLogs(nextLogs);
|
||||||
return nextLogs;
|
return nextLogs;
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -303,7 +319,7 @@ export default function VideoPage() {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
const pollGenerationLog = async (log: GenerationLog, configOverride?: AiConfig) => {
|
const pollGenerationLog = async (log: GenerationLog, configOverride?: AiConfig, agentTaskId?: string) => {
|
||||||
if (!log.task || activeLogIdsRef.current.has(log.id)) return;
|
if (!log.task || activeLogIdsRef.current.has(log.id)) return;
|
||||||
activeLogIdsRef.current.add(log.id);
|
activeLogIdsRef.current.add(log.id);
|
||||||
setRunning(true);
|
setRunning(true);
|
||||||
@@ -326,6 +342,7 @@ export default function VideoPage() {
|
|||||||
mimeType: stored.mimeType,
|
mimeType: stored.mimeType,
|
||||||
};
|
};
|
||||||
setResults([{ id: nextVideo.id, status: "success", video: nextVideo }]);
|
setResults([{ id: nextVideo.id, status: "success", video: nextVideo }]);
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "succeeded", successCount: 1, failCount: 0, error: undefined });
|
||||||
await saveLog({ ...log, status: "成功", durationMs: nextVideo.durationMs, video: nextVideo, error: undefined });
|
await saveLog({ ...log, status: "成功", durationMs: nextVideo.durationMs, video: nextVideo, error: undefined });
|
||||||
message.success("视频已生成");
|
message.success("视频已生成");
|
||||||
return;
|
return;
|
||||||
@@ -337,6 +354,7 @@ export default function VideoPage() {
|
|||||||
} catch (error) {
|
} catch (error) {
|
||||||
const errorMessage = error instanceof Error ? error.message : "生成失败";
|
const errorMessage = error instanceof Error ? error.message : "生成失败";
|
||||||
setResults([{ id: log.id, status: "failed", error: errorMessage }]);
|
setResults([{ id: log.id, status: "failed", error: errorMessage }]);
|
||||||
|
if (agentTaskId) updateAgentTask(agentTaskId, { status: "failed", successCount: 0, failCount: 1, error: errorMessage });
|
||||||
await saveLog({ ...log, status: "失败", durationMs: Date.now() - log.createdAt, error: errorMessage });
|
await saveLog({ ...log, status: "失败", durationMs: Date.now() - log.createdAt, error: errorMessage });
|
||||||
message.error(errorMessage);
|
message.error(errorMessage);
|
||||||
} finally {
|
} finally {
|
||||||
|
|||||||
@@ -0,0 +1,37 @@
|
|||||||
|
import { nanoid } from "nanoid";
|
||||||
|
|
||||||
|
export type PromptSource = {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
url: string;
|
||||||
|
homepage: string;
|
||||||
|
enabled: boolean;
|
||||||
|
builtIn: boolean;
|
||||||
|
};
|
||||||
|
|
||||||
|
export const PROMPT_REGISTRY_HOMEPAGE = "https://github.com/yukkcat/image-prompts";
|
||||||
|
const PROMPT_REGISTRY_SOURCE_BASE = "https://raw.githubusercontent.com/yukkcat/image-prompts/main/dist/sources";
|
||||||
|
|
||||||
|
export function createPromptSource(source?: Partial<PromptSource>): PromptSource {
|
||||||
|
return {
|
||||||
|
id: source?.id?.trim() || nanoid(),
|
||||||
|
name: source?.name?.trim() || "新来源",
|
||||||
|
url: source?.url?.trim() || "",
|
||||||
|
homepage: source?.homepage?.trim() || "",
|
||||||
|
enabled: source?.enabled ?? true,
|
||||||
|
builtIn: source?.builtIn ?? false,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export const DEFAULT_PROMPT_SOURCES: PromptSource[] = [
|
||||||
|
registrySource("banana-prompt-quicker", "Banana Prompt Quicker", "https://glidea.github.io/banana-prompt-quicker/"),
|
||||||
|
registrySource("davidwu-gpt-image2-prompts", "DavidWu GPT Image 2", "https://github.com/davidwuw0811-boop/awesome-gpt-image2-prompts"),
|
||||||
|
registrySource("awesome-gpt-image", "Awesome GPT Image", "https://github.com/ZeroLu/awesome-gpt-image"),
|
||||||
|
registrySource("awesome-gpt4o-image-prompts", "Awesome GPT-4o", "https://github.com/ImgEdify/Awesome-GPT4o-Image-Prompts"),
|
||||||
|
registrySource("youmind-gpt-image-2", "YouMind GPT Image 2", "https://github.com/YouMind-OpenLab/awesome-gpt-image-2"),
|
||||||
|
registrySource("youmind-nano-banana-pro", "YouMind Nano Banana Pro", "https://github.com/YouMind-OpenLab/awesome-nano-banana-pro-prompts"),
|
||||||
|
];
|
||||||
|
|
||||||
|
function registrySource(id: string, name: string, homepage: string): PromptSource {
|
||||||
|
return { id, name, url: `${PROMPT_REGISTRY_SOURCE_BASE}/${id}.json`, homepage, enabled: true, builtIn: true };
|
||||||
|
}
|
||||||
@@ -0,0 +1,118 @@
|
|||||||
|
import type { PromptSource } from "./prompt-source-presets";
|
||||||
|
|
||||||
|
export type RawPrompt = {
|
||||||
|
id: string;
|
||||||
|
title: string;
|
||||||
|
prompt: string;
|
||||||
|
description: string;
|
||||||
|
coverUrl: string;
|
||||||
|
referenceImageUrls: string[];
|
||||||
|
tags: string[];
|
||||||
|
preview: string;
|
||||||
|
createdAt: string;
|
||||||
|
updatedAt: string;
|
||||||
|
author?: string;
|
||||||
|
sourceUrl?: string;
|
||||||
|
imageMode?: string;
|
||||||
|
imageModel?: string;
|
||||||
|
imageSize?: string;
|
||||||
|
imageCount?: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
type RunOptions = { signal?: AbortSignal };
|
||||||
|
|
||||||
|
async function fetchSource(source: PromptSource, options?: RunOptions) {
|
||||||
|
const response = await fetch(source.url, { cache: "no-store", signal: options?.signal });
|
||||||
|
if (!response.ok) throw new Error(`请求失败(${response.status})`);
|
||||||
|
return response.json();
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function runPromptSource(source: PromptSource, options?: RunOptions): Promise<RawPrompt[]> {
|
||||||
|
if (!source.url.trim()) throw new Error("JSON URL 不能为空");
|
||||||
|
let data: unknown;
|
||||||
|
try {
|
||||||
|
data = await fetchSource(source, options);
|
||||||
|
} catch (error) {
|
||||||
|
if (error instanceof DOMException && error.name === "AbortError") throw error;
|
||||||
|
throw new Error(`「${source.name}」拉取失败:${error instanceof Error ? error.message : String(error)}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
const items = parseJsonSource(data, source);
|
||||||
|
if (source.builtIn && !items.length) throw new Error(`「${source.name}」未解析到有效提示词`);
|
||||||
|
return items;
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseJsonSource(data: unknown, source: PromptSource) {
|
||||||
|
if (!Array.isArray(data)) throw new Error(`「${source.name}」格式错误:根节点必须是数组`);
|
||||||
|
return normalizeItems(data, source);
|
||||||
|
}
|
||||||
|
|
||||||
|
function normalizeItems(values: unknown[], source: PromptSource) {
|
||||||
|
const seen = new Set<string>();
|
||||||
|
const items: RawPrompt[] = [];
|
||||||
|
values.forEach((value, index) => {
|
||||||
|
const record = asRecord(value);
|
||||||
|
const title = stringValue(record.title).trim();
|
||||||
|
const prompt = stringValue(record.prompt).trim();
|
||||||
|
if (!title || !prompt) return;
|
||||||
|
const id = stringValue(record.id).trim() || `${source.id}-${leftPad(index + 1)}`;
|
||||||
|
if (seen.has(id)) return;
|
||||||
|
seen.add(id);
|
||||||
|
const referenceImageUrls = stringArray(record.referenceImageUrls).map((url) => absoluteUrl(source.url, url));
|
||||||
|
const coverUrl = absoluteUrl(source.url, stringValue(record.coverUrl)) || referenceImageUrls[0] || "";
|
||||||
|
items.push({
|
||||||
|
id,
|
||||||
|
title,
|
||||||
|
prompt,
|
||||||
|
description: stringValue(record.description),
|
||||||
|
coverUrl,
|
||||||
|
referenceImageUrls,
|
||||||
|
tags: stringArray(record.tags),
|
||||||
|
preview: stringValue(record.preview),
|
||||||
|
createdAt: stringValue(record.createdAt),
|
||||||
|
updatedAt: stringValue(record.updatedAt),
|
||||||
|
author: stringValue(record.author),
|
||||||
|
sourceUrl: absoluteUrl(source.url, stringValue(record.sourceUrl)),
|
||||||
|
imageMode: optionalString(record.imageMode),
|
||||||
|
imageModel: optionalString(record.imageModel),
|
||||||
|
imageSize: optionalString(record.imageSize),
|
||||||
|
imageCount: optionalNumber(record.imageCount),
|
||||||
|
});
|
||||||
|
});
|
||||||
|
return items;
|
||||||
|
}
|
||||||
|
|
||||||
|
function asRecord(value: unknown): Record<string, unknown> {
|
||||||
|
return value && typeof value === "object" ? (value as Record<string, unknown>) : {};
|
||||||
|
}
|
||||||
|
|
||||||
|
function stringValue(value: unknown) {
|
||||||
|
return typeof value === "string" || typeof value === "number" ? String(value) : "";
|
||||||
|
}
|
||||||
|
|
||||||
|
function stringArray(value: unknown) {
|
||||||
|
return Array.isArray(value) ? value.map(stringValue).map((item) => item.trim()).filter(Boolean) : [];
|
||||||
|
}
|
||||||
|
|
||||||
|
function optionalString(value: unknown) {
|
||||||
|
const result = stringValue(value).trim();
|
||||||
|
return result || undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
function optionalNumber(value: unknown) {
|
||||||
|
const result = Number(value);
|
||||||
|
return Number.isFinite(result) && result > 0 ? result : undefined;
|
||||||
|
}
|
||||||
|
|
||||||
|
function absoluteUrl(baseUrl: string, path: string) {
|
||||||
|
if (!path) return "";
|
||||||
|
try {
|
||||||
|
return new URL(path, baseUrl).toString();
|
||||||
|
} catch {
|
||||||
|
return path;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function leftPad(value: number) {
|
||||||
|
return String(value).padStart(4, "0");
|
||||||
|
}
|
||||||
+184
-188
@@ -1,25 +1,18 @@
|
|||||||
import localforage from "localforage";
|
import localforage from "localforage";
|
||||||
|
|
||||||
export type Prompt = {
|
import { runPromptSource, type RawPrompt } from "./prompt-source-runtime";
|
||||||
id: string;
|
import { usePromptSourceStore } from "@/stores/use-prompt-source-store";
|
||||||
title: string;
|
import { usePromptStore, type PersonalPrompt } from "@/stores/use-prompt-store";
|
||||||
coverUrl: string;
|
import type { PromptSource } from "./prompt-source-presets";
|
||||||
prompt: string;
|
|
||||||
tags: string[];
|
|
||||||
category: string;
|
|
||||||
githubUrl: string;
|
|
||||||
preview: string;
|
|
||||||
createdAt: string;
|
|
||||||
updatedAt: string;
|
|
||||||
};
|
|
||||||
|
|
||||||
type PromptCategory = {
|
export type Prompt = RawPrompt & {
|
||||||
|
sourceId: string;
|
||||||
category: string;
|
category: string;
|
||||||
githubUrl: string;
|
githubUrl: string;
|
||||||
build: () => Promise<Omit<Prompt, "category" | "githubUrl">[]>;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
export const ALL_PROMPTS_OPTION = "全部";
|
export const ALL_PROMPTS_OPTION = "全部";
|
||||||
|
export const PERSONAL_PROMPTS_CATEGORY = "我的提示词";
|
||||||
|
|
||||||
export type PromptListResponse = {
|
export type PromptListResponse = {
|
||||||
items: Prompt[];
|
items: Prompt[];
|
||||||
@@ -28,65 +21,200 @@ export type PromptListResponse = {
|
|||||||
total: number;
|
total: number;
|
||||||
};
|
};
|
||||||
|
|
||||||
const awesomeGptImageRawBase = "https://raw.githubusercontent.com/ZeroLu/awesome-gpt-image/main";
|
export type PromptSourceStatus = {
|
||||||
const awesomeGpt4oImagePromptsBase = "https://raw.githubusercontent.com/ImgEdify/Awesome-GPT4o-Image-Prompts/main";
|
sourceId: string;
|
||||||
const youMindGptImage2RawBase = "https://raw.githubusercontent.com/YouMind-OpenLab/awesome-gpt-image-2/main";
|
count: number;
|
||||||
const youMindNanoBananaProRawBase = "https://raw.githubusercontent.com/YouMind-OpenLab/awesome-nano-banana-pro-prompts/main";
|
lastSuccessAt: string;
|
||||||
const davidWuGptImage2RawBase = "https://raw.githubusercontent.com/davidwuw0811-boop/awesome-gpt-image2-prompts/main";
|
lastError: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type PromptSourceRefreshResult = PromptSourceStatus & {
|
||||||
|
sourceName: string;
|
||||||
|
success: boolean;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type PromptSourceRefreshSummary = {
|
||||||
|
results: PromptSourceRefreshResult[];
|
||||||
|
total: number;
|
||||||
|
successCount: number;
|
||||||
|
failureCount: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
type SourceCache = PromptSourceStatus & {
|
||||||
|
items: Prompt[];
|
||||||
|
fetchedAt: number;
|
||||||
|
signature: string;
|
||||||
|
};
|
||||||
|
|
||||||
const cacheTtlMs = 1000 * 60 * 60;
|
const cacheTtlMs = 1000 * 60 * 60;
|
||||||
const promptCacheKey = "third-party-prompts";
|
|
||||||
const promptCacheStore = localforage.createInstance({ name: "infinite-canvas", storeName: "prompt_cache" });
|
const promptCacheStore = localforage.createInstance({ name: "infinite-canvas", storeName: "prompt_cache" });
|
||||||
|
const loadingSources = new Map<string, Promise<PromptSourceRefreshResult>>();
|
||||||
|
|
||||||
const categories: PromptCategory[] = [
|
function enabledSources() {
|
||||||
{ category: "awesome-gpt-image", githubUrl: "https://github.com/ZeroLu/awesome-gpt-image", build: buildAwesomeGptImagePrompts },
|
return usePromptSourceStore.getState().sources.filter((source) => source.enabled);
|
||||||
{ category: "awesome-gpt4o-image-prompts", githubUrl: "https://github.com/ImgEdify/Awesome-GPT4o-Image-Prompts", build: buildAwesomeGpt4oImagePrompts },
|
}
|
||||||
{ category: "youmind-gpt-image-2", githubUrl: "https://github.com/YouMind-OpenLab/awesome-gpt-image-2", build: () => buildYouMindPrompts(youMindGptImage2RawBase, "youmind-gpt-image-2", "gpt-image-2") },
|
|
||||||
{ category: "youmind-nano-banana-pro", githubUrl: "https://github.com/YouMind-OpenLab/awesome-nano-banana-pro-prompts", build: () => buildYouMindPrompts(youMindNanoBananaProRawBase, "youmind-nano-banana-pro", "nano-banana-pro") },
|
|
||||||
{ category: "davidwu-gpt-image2-prompts", githubUrl: "https://github.com/davidwuw0811-boop/awesome-gpt-image2-prompts", build: buildDavidWuGptImage2Prompts },
|
|
||||||
];
|
|
||||||
|
|
||||||
let loadingPrompts: Promise<Prompt[]> | null = null;
|
function cacheKey(sourceId: string) {
|
||||||
|
return `prompt-source:${sourceId}`;
|
||||||
|
}
|
||||||
|
|
||||||
export async function fetchPrompts({ keyword = "", tag = [], category = ALL_PROMPTS_OPTION, page = 1, pageSize = 20 }: { keyword?: string; tag?: string[]; category?: string; page?: number; pageSize?: number } = {}) {
|
function sourceSignature(source: PromptSource) {
|
||||||
const items = await getPrompts();
|
const value = `${source.name}\n${source.url}\n${source.homepage}`;
|
||||||
const normalizedKeyword = keyword.trim().toLowerCase();
|
let hash = 0;
|
||||||
const normalizedPage = Math.max(1, page);
|
for (let i = 0; i < value.length; i += 1) hash = (hash * 31 + value.charCodeAt(i)) | 0;
|
||||||
const normalizedPageSize = Math.max(1, Math.min(100, pageSize));
|
return `${value.length}:${hash}`;
|
||||||
const withoutTagFilter = filterPrompts(items, { keyword: normalizedKeyword, category, tags: [] });
|
}
|
||||||
const filtered = filterPrompts(items, { keyword: normalizedKeyword, category, tags: tag });
|
|
||||||
|
|
||||||
|
function withSourceMeta(source: PromptSource, items: RawPrompt[]): Prompt[] {
|
||||||
|
return items.map((item) => ({
|
||||||
|
...item,
|
||||||
|
description: item.description || "",
|
||||||
|
referenceImageUrls: Array.isArray(item.referenceImageUrls) ? item.referenceImageUrls : [],
|
||||||
|
sourceId: source.id,
|
||||||
|
category: source.name,
|
||||||
|
githubUrl: item.sourceUrl || source.homepage,
|
||||||
|
}));
|
||||||
|
}
|
||||||
|
|
||||||
|
export function personalPromptToPrompt(item: PersonalPrompt): Prompt {
|
||||||
return {
|
return {
|
||||||
items: filtered.slice((normalizedPage - 1) * normalizedPageSize, normalizedPage * normalizedPageSize),
|
...item,
|
||||||
tags: collectTags(withoutTagFilter),
|
coverUrl: item.coverUrl || item.referenceImageUrls[0] || "",
|
||||||
categories: categories.map((item) => item.category),
|
sourceId: "personal",
|
||||||
total: filtered.length,
|
category: PERSONAL_PROMPTS_CATEGORY,
|
||||||
|
githubUrl: "",
|
||||||
|
preview: "",
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
async function getPrompts() {
|
async function readSourceCache(sourceId: string) {
|
||||||
const cached = await promptCacheStore.getItem<{ items?: Prompt[]; fetchedAt?: number }>(promptCacheKey);
|
return promptCacheStore.getItem<SourceCache>(cacheKey(sourceId));
|
||||||
if (cached?.items?.length && cached.fetchedAt && Date.now() - cached.fetchedAt < cacheTtlMs) return cached.items;
|
|
||||||
if (loadingPrompts) return loadingPrompts;
|
|
||||||
loadingPrompts = loadPrompts().finally(() => {
|
|
||||||
loadingPrompts = null;
|
|
||||||
});
|
|
||||||
return loadingPrompts;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async function loadPrompts() {
|
async function refreshSourceRecord(source: PromptSource): Promise<PromptSourceRefreshResult> {
|
||||||
|
const previous = await readSourceCache(source.id);
|
||||||
|
try {
|
||||||
|
const items = withSourceMeta(source, await runPromptSource(source));
|
||||||
|
const lastSuccessAt = new Date().toISOString();
|
||||||
|
const cache: SourceCache = { sourceId: source.id, items, count: items.length, fetchedAt: Date.now(), lastSuccessAt, lastError: "", signature: sourceSignature(source) };
|
||||||
|
await promptCacheStore.setItem(cacheKey(source.id), cache);
|
||||||
|
return { sourceId: source.id, sourceName: source.name, count: items.length, lastSuccessAt, lastError: "", success: true };
|
||||||
|
} catch (error) {
|
||||||
|
const lastError = error instanceof Error ? error.message : String(error);
|
||||||
|
const cache: SourceCache = {
|
||||||
|
sourceId: source.id,
|
||||||
|
items: previous?.items || [],
|
||||||
|
count: previous?.items?.length || 0,
|
||||||
|
fetchedAt: previous?.fetchedAt || 0,
|
||||||
|
lastSuccessAt: previous?.lastSuccessAt || "",
|
||||||
|
lastError,
|
||||||
|
signature: previous?.signature || sourceSignature(source),
|
||||||
|
};
|
||||||
|
await promptCacheStore.setItem(cacheKey(source.id), cache);
|
||||||
|
return { sourceId: source.id, sourceName: source.name, count: cache.count, lastSuccessAt: cache.lastSuccessAt, lastError, success: false };
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function getOrStartRefresh(source: PromptSource) {
|
||||||
|
const current = loadingSources.get(source.id);
|
||||||
|
if (current) return current;
|
||||||
|
const loading = refreshSourceRecord(source).finally(() => loadingSources.delete(source.id));
|
||||||
|
loadingSources.set(source.id, loading);
|
||||||
|
return loading;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function getSourcePrompts(source: PromptSource): Promise<Prompt[]> {
|
||||||
|
const cached = await readSourceCache(source.id);
|
||||||
|
if (cached) {
|
||||||
|
const stale = cached.signature !== sourceSignature(source) || Date.now() - cached.fetchedAt >= cacheTtlMs;
|
||||||
|
if (stale) void getOrStartRefresh(source).catch(() => undefined);
|
||||||
|
return withSourceMeta(source, cached.items);
|
||||||
|
}
|
||||||
|
const result = await getOrStartRefresh(source);
|
||||||
|
if (!result.success) throw new Error(result.lastError);
|
||||||
|
return (await readSourceCache(source.id))?.items || [];
|
||||||
|
}
|
||||||
|
|
||||||
|
async function getAllPrompts(includePersonal: boolean): Promise<Prompt[]> {
|
||||||
const settled = await Promise.all(
|
const settled = await Promise.all(
|
||||||
categories.map(async (category) => {
|
enabledSources().map(async (source) => {
|
||||||
try {
|
try {
|
||||||
const items = await category.build();
|
return await getSourcePrompts(source);
|
||||||
return items.map((item) => ({ ...item, category: category.category, githubUrl: category.githubUrl }));
|
|
||||||
} catch {
|
} catch {
|
||||||
return [];
|
return [];
|
||||||
}
|
}
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
const items = settled.flat();
|
const personal = includePersonal ? usePromptStore.getState().prompts.map(personalPromptToPrompt) : [];
|
||||||
await promptCacheStore.setItem(promptCacheKey, { items, fetchedAt: Date.now() });
|
return [...personal, ...settled.flat()];
|
||||||
return items;
|
}
|
||||||
|
|
||||||
|
export async function fetchPrompts({ keyword = "", tag = [], category = ALL_PROMPTS_OPTION, page = 1, pageSize = 20, includePersonal = true }: { keyword?: string; tag?: string[]; category?: string; page?: number; pageSize?: number; includePersonal?: boolean } = {}) {
|
||||||
|
const items = await getAllPrompts(includePersonal);
|
||||||
|
const normalizedKeyword = keyword.trim().toLowerCase();
|
||||||
|
const normalizedPage = Math.max(1, page);
|
||||||
|
const normalizedPageSize = Math.max(1, Math.min(100, pageSize));
|
||||||
|
const withoutTagFilter = filterPrompts(items, { keyword: normalizedKeyword, category, tags: [] });
|
||||||
|
const filtered = filterPrompts(items, { keyword: normalizedKeyword, category, tags: tag });
|
||||||
|
const categories = enabledSources().map((source) => source.name);
|
||||||
|
if (includePersonal && usePromptStore.getState().prompts.length) categories.unshift(PERSONAL_PROMPTS_CATEGORY);
|
||||||
|
|
||||||
|
return {
|
||||||
|
items: filtered.slice((normalizedPage - 1) * normalizedPageSize, normalizedPage * normalizedPageSize),
|
||||||
|
tags: collectTags(withoutTagFilter),
|
||||||
|
categories,
|
||||||
|
total: filtered.length,
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function fetchSourcePrompts(sourceId: string): Promise<Prompt[]> {
|
||||||
|
const source = usePromptSourceStore.getState().sources.find((item) => item.id === sourceId);
|
||||||
|
if (!source) throw new Error("提示词来源不存在");
|
||||||
|
return getSourcePrompts(source);
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function refreshSource(sourceId: string): Promise<PromptSourceRefreshResult> {
|
||||||
|
const source = usePromptSourceStore.getState().sources.find((item) => item.id === sourceId);
|
||||||
|
if (!source) throw new Error("提示词来源不存在");
|
||||||
|
const result = await getOrStartRefresh(source);
|
||||||
|
if (!result.success) throw new Error(result.lastError);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function refreshAllSources(): Promise<PromptSourceRefreshSummary> {
|
||||||
|
const results = await Promise.all(enabledSources().map(getOrStartRefresh));
|
||||||
|
return summarizeRefresh(results);
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function refreshDueSources(maxAgeMs: number): Promise<PromptSourceRefreshSummary> {
|
||||||
|
const sources = await Promise.all(
|
||||||
|
enabledSources().map(async (source) => {
|
||||||
|
const cached = await readSourceCache(source.id);
|
||||||
|
const lastSuccess = cached?.lastSuccessAt ? new Date(cached.lastSuccessAt).getTime() : 0;
|
||||||
|
return !lastSuccess || Boolean(cached?.lastError) || Date.now() - lastSuccess >= maxAgeMs || cached?.signature !== sourceSignature(source) ? source : null;
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
const results = await Promise.all(sources.filter((source): source is PromptSource => Boolean(source)).map(getOrStartRefresh));
|
||||||
|
return summarizeRefresh(results);
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function fetchPromptSourceStatuses(): Promise<Record<string, PromptSourceStatus>> {
|
||||||
|
const entries = await Promise.all(
|
||||||
|
usePromptSourceStore.getState().sources.map(async (source) => {
|
||||||
|
const cache = await readSourceCache(source.id);
|
||||||
|
return [source.id, { sourceId: source.id, count: cache?.items?.length || 0, lastSuccessAt: cache?.lastSuccessAt || "", lastError: cache?.lastError || "" }] as const;
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
return Object.fromEntries(entries);
|
||||||
|
}
|
||||||
|
|
||||||
|
function summarizeRefresh(results: PromptSourceRefreshResult[]): PromptSourceRefreshSummary {
|
||||||
|
return {
|
||||||
|
results,
|
||||||
|
total: results.reduce((total, item) => total + item.count, 0),
|
||||||
|
successCount: results.filter((item) => item.success).length,
|
||||||
|
failureCount: results.filter((item) => !item.success).length,
|
||||||
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
function filterPrompts(items: Prompt[], options: { keyword: string; category: string; tags: string[] }) {
|
function filterPrompts(items: Prompt[], options: { keyword: string; category: string; tags: string[] }) {
|
||||||
@@ -94,148 +222,16 @@ function filterPrompts(items: Prompt[], options: { keyword: string; category: st
|
|||||||
if (isActiveOption(options.category) && item.category !== options.category) return false;
|
if (isActiveOption(options.category) && item.category !== options.category) return false;
|
||||||
if (options.tags.length && !options.tags.some((tag) => item.tags.includes(tag))) return false;
|
if (options.tags.length && !options.tags.some((tag) => item.tags.includes(tag))) return false;
|
||||||
if (!options.keyword) return true;
|
if (!options.keyword) return true;
|
||||||
return [item.title, item.prompt, item.category, ...item.tags].join(" ").toLowerCase().includes(options.keyword);
|
return [item.title, item.prompt, item.description, item.category, ...item.tags].join(" ").toLowerCase().includes(options.keyword);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
async function buildAwesomeGptImagePrompts() {
|
|
||||||
const markdown = await fetchText(awesomeGptImageRawBase, "README.zh-CN.md");
|
|
||||||
const items: Omit<Prompt, "category" | "githubUrl">[] = [];
|
|
||||||
for (const section of splitBeforeHeading(markdown, "## ")) {
|
|
||||||
const tags = tagsFromHeading(firstMatch(section, /^##\s+(.+)$/m));
|
|
||||||
for (const block of splitBeforeHeading(section, "### ")) {
|
|
||||||
const title = firstMatch(block, /^###\s+(.+)$/m).replace(/\[([^\]]+)]\([^)]+\)/g, "$1").trim();
|
|
||||||
const prompt = firstMatch(block, /\*\*提示词:\*\*\s*\r?\n\s*```[\w-]*\r?\n(.*?)\r?\n```/s).trim();
|
|
||||||
if (!title || !prompt) continue;
|
|
||||||
const images = extractMarkdownImages(awesomeGptImageRawBase, block);
|
|
||||||
items.push(defaultPrompt(`awesome-gpt-image-${leftPad(items.length + 1)}`, title, prompt, images[0] || "", tags, markdownPreview(images)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return items;
|
|
||||||
}
|
|
||||||
|
|
||||||
async function buildAwesomeGpt4oImagePrompts() {
|
|
||||||
const markdown = await fetchText(awesomeGpt4oImagePromptsBase, "README.zh-CN.md");
|
|
||||||
const items: Omit<Prompt, "category" | "githubUrl">[] = [];
|
|
||||||
for (const block of splitBeforeHeading(markdown, "### ")) {
|
|
||||||
const title = firstMatch(block, /^###\s+(.+)$/m).trim();
|
|
||||||
const prompt = firstMatch(block, /- \*\*提示词文本:\*\*\s*`(.*?)`/s).trim();
|
|
||||||
if (!title || !prompt) continue;
|
|
||||||
const images = extractMarkdownImages(awesomeGpt4oImagePromptsBase, block);
|
|
||||||
items.push(defaultPrompt(`awesome-gpt4o-image-prompts-${leftPad(items.length + 1)}`, title, prompt, images[0] || "", ["gpt4o"], markdownPreview(images)));
|
|
||||||
}
|
|
||||||
return items;
|
|
||||||
}
|
|
||||||
|
|
||||||
async function buildYouMindPrompts(baseUrl: string, idPrefix: string, modelTag: string) {
|
|
||||||
const markdown = await fetchText(baseUrl, "README_zh.md");
|
|
||||||
const items: Omit<Prompt, "category" | "githubUrl">[] = [];
|
|
||||||
for (const block of splitBeforeHeading(markdown, "### ")) {
|
|
||||||
const title = firstMatch(block, /^###\s+No\.\s*\d+:\s*(.+)$/m).trim();
|
|
||||||
const prompt = firstMatch(block, /#### .*?提示词\s*\r?\n\s*```[\w-]*\r?\n(.*?)\r?\n```/s).trim();
|
|
||||||
if (!title || !prompt) continue;
|
|
||||||
const images = extractMarkdownImages(baseUrl, block);
|
|
||||||
items.push(defaultPrompt(`${idPrefix}-${leftPad(items.length + 1)}`, title, prompt, images[0] || "", youMindTags(title, modelTag), markdownPreview(images)));
|
|
||||||
}
|
|
||||||
return items;
|
|
||||||
}
|
|
||||||
|
|
||||||
async function buildDavidWuGptImage2Prompts() {
|
|
||||||
const data = await fetchJson<Array<{ id?: number; title_en?: string; title_cn?: string; category?: string; category_cn?: string; prompt?: string; note?: string; author?: string; source?: string; needs_ref?: boolean; image?: string }>>(davidWuGptImage2RawBase, "prompts.json");
|
|
||||||
return data
|
|
||||||
.map((item, index) => {
|
|
||||||
const title = (item.title_cn || item.title_en || "").trim();
|
|
||||||
const prompt = (item.prompt || "").trim();
|
|
||||||
if (!title || !prompt) return null;
|
|
||||||
const image = absoluteImage(davidWuGptImage2RawBase, item.image || "");
|
|
||||||
const preview = [item.title_en, item.note, image ? `` : ""].filter(Boolean).join("\n\n");
|
|
||||||
return defaultPrompt(`davidwu-gpt-image2-prompts-${leftPad(item.id || index + 1)}`, title, prompt, image, davidWuTags(item), preview);
|
|
||||||
})
|
|
||||||
.filter((item): item is Omit<Prompt, "category" | "githubUrl"> => Boolean(item));
|
|
||||||
}
|
|
||||||
|
|
||||||
function defaultPrompt(id: string, title: string, prompt: string, coverUrl: string, tags: string[], preview: string): Omit<Prompt, "category" | "githubUrl"> {
|
|
||||||
return { id, title, coverUrl, prompt, tags, preview, createdAt: "", updatedAt: "" };
|
|
||||||
}
|
|
||||||
|
|
||||||
async function fetchText(baseUrl: string, file: string) {
|
|
||||||
const response = await fetch(`${baseUrl}/${file}`, { cache: "no-store" });
|
|
||||||
if (!response.ok) throw new Error(`${file} 拉取失败`);
|
|
||||||
return response.text();
|
|
||||||
}
|
|
||||||
|
|
||||||
async function fetchJson<T>(baseUrl: string, file: string) {
|
|
||||||
return JSON.parse(await fetchText(baseUrl, file)) as T;
|
|
||||||
}
|
|
||||||
|
|
||||||
function splitBeforeHeading(markdown: string, prefix: string) {
|
|
||||||
const blocks: string[] = [];
|
|
||||||
let current: string[] = [];
|
|
||||||
for (const line of markdown.split("\n")) {
|
|
||||||
if (line.startsWith(prefix) && current.length) {
|
|
||||||
blocks.push(current.join("\n"));
|
|
||||||
current = [];
|
|
||||||
}
|
|
||||||
current.push(line);
|
|
||||||
}
|
|
||||||
blocks.push(current.join("\n"));
|
|
||||||
return blocks;
|
|
||||||
}
|
|
||||||
|
|
||||||
function firstMatch(value: string, pattern: RegExp) {
|
|
||||||
return pattern.exec(value)?.[1] || "";
|
|
||||||
}
|
|
||||||
|
|
||||||
function extractMarkdownImages(baseUrl: string, markdown: string) {
|
|
||||||
return Array.from(markdown.matchAll(/!\[[^\]]*]\(([^)]+)\)/g), (match) => absoluteImage(baseUrl, match[1])).filter(Boolean);
|
|
||||||
}
|
|
||||||
|
|
||||||
function absoluteImage(baseUrl: string, image: string) {
|
|
||||||
if (!image) return "";
|
|
||||||
if (/^https?:\/\//i.test(image)) return image;
|
|
||||||
return `${baseUrl}/${image.replace(/^\.?\//, "")}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
function tagsFromCategory(category: string) {
|
|
||||||
return splitTags(category.replace(/\s+Cases$/i, ""), /\s*(?:&|and)\s*/);
|
|
||||||
}
|
|
||||||
|
|
||||||
function tagsFromHeading(heading: string) {
|
|
||||||
return splitTags(heading.replace(/[^\p{L}\p{N}/&、与 ]/gu, ""), /\s*(?:\/|&|、|与)\s*/);
|
|
||||||
}
|
|
||||||
|
|
||||||
function youMindTags(title: string, modelTag: string) {
|
|
||||||
const [, prefix] = title.match(/^(.+?) - /) || [];
|
|
||||||
return [modelTag, ...tagsFromHeading(prefix || "")];
|
|
||||||
}
|
|
||||||
|
|
||||||
function davidWuTags(item: { category_cn?: string; category?: string; author?: string; source?: string; needs_ref?: boolean }) {
|
|
||||||
const tags = splitTags([item.category_cn, item.category, item.author, item.source].filter(Boolean).join("/"), /\//);
|
|
||||||
if (item.needs_ref) tags.push("需要参考图");
|
|
||||||
return tags;
|
|
||||||
}
|
|
||||||
|
|
||||||
function splitTags(value: string, pattern: RegExp) {
|
|
||||||
return value
|
|
||||||
.split(pattern)
|
|
||||||
.map((tag) => tag.trim().toLowerCase())
|
|
||||||
.filter(Boolean);
|
|
||||||
}
|
|
||||||
|
|
||||||
function markdownPreview(images: string[]) {
|
|
||||||
return images.filter(Boolean).map((image) => ``).join("\n\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
function collectTags(items: Prompt[]) {
|
function collectTags(items: Prompt[]) {
|
||||||
return Array.from(new Set(items.flatMap((item) => item.tags).filter(Boolean)));
|
return Array.from(new Set(items.flatMap((item) => item.tags).filter(Boolean)));
|
||||||
}
|
}
|
||||||
|
|
||||||
function leftPad(value: number) {
|
|
||||||
return String(value).padStart(4, "0");
|
|
||||||
}
|
|
||||||
|
|
||||||
function isActiveOption(value: string) {
|
function isActiveOption(value: string) {
|
||||||
return value && value !== "全部" && value !== "all";
|
return value && value !== ALL_PROMPTS_OPTION && value !== "all";
|
||||||
}
|
}
|
||||||
|
|
||||||
export function formatPromptDate(value: string) {
|
export function formatPromptDate(value: string) {
|
||||||
|
|||||||
@@ -3,7 +3,7 @@ import { create } from "zustand";
|
|||||||
import type { CanvasAgentOp, CanvasAgentSnapshot } from "@/lib/canvas/canvas-agent-ops";
|
import type { CanvasAgentOp, CanvasAgentSnapshot } from "@/lib/canvas/canvas-agent-ops";
|
||||||
|
|
||||||
export type AgentChatRole = "user" | "assistant" | "system" | "tool" | "error";
|
export type AgentChatRole = "user" | "assistant" | "system" | "tool" | "error";
|
||||||
export type AgentAttachment = { id: string; name: string; type: string; size: number; url: string; dataUrl: string };
|
export type AgentAttachment = { id: string; name: string; type: string; size: number; width: number; height: number; url: string; dataUrl: string };
|
||||||
export type AgentChatItem = { id: string; role: AgentChatRole; title?: string; text: string; meta?: string; detail?: unknown; attachments?: AgentAttachment[]; streamId?: string };
|
export type AgentChatItem = { id: string; role: AgentChatRole; title?: string; text: string; meta?: string; detail?: unknown; attachments?: AgentAttachment[]; streamId?: string };
|
||||||
export type AgentEventLog = { id: string; time: string; title: string; text: string; raw?: unknown };
|
export type AgentEventLog = { id: string; time: string; title: string; text: string; raw?: unknown };
|
||||||
export type AgentPendingToolCall = { requestId: string; name: string; input?: { ops?: CanvasAgentOp[]; path?: string } & Record<string, unknown> };
|
export type AgentPendingToolCall = { requestId: string; name: string; input?: { ops?: CanvasAgentOp[]; path?: string } & Record<string, unknown> };
|
||||||
|
|||||||
@@ -56,7 +56,7 @@ export type WebdavSyncConfig = {
|
|||||||
directory: string;
|
directory: string;
|
||||||
lastSyncedAt: string;
|
lastSyncedAt: string;
|
||||||
};
|
};
|
||||||
export type ConfigTabKey = "channels" | "preferences" | "webdav";
|
export type ConfigTabKey = "channels" | "preferences" | "prompt-sources" | "webdav";
|
||||||
|
|
||||||
export const CONFIG_STORE_KEY = "infinite-canvas:ai_config_store";
|
export const CONFIG_STORE_KEY = "infinite-canvas:ai_config_store";
|
||||||
const CHANNEL_MODEL_SEPARATOR = "::";
|
const CHANNEL_MODEL_SEPARATOR = "::";
|
||||||
|
|||||||
@@ -0,0 +1,65 @@
|
|||||||
|
import { create } from "zustand";
|
||||||
|
import { persist } from "zustand/middleware";
|
||||||
|
|
||||||
|
import { DEFAULT_PROMPT_SOURCES, createPromptSource, type PromptSource } from "@/services/api/prompt-source-presets";
|
||||||
|
|
||||||
|
export type PromptSourceSchedule = {
|
||||||
|
intervalMinutes: number;
|
||||||
|
lastFetchedAt: string;
|
||||||
|
};
|
||||||
|
|
||||||
|
const PROMPT_SOURCE_STORE_KEY = "infinite-canvas:prompt_source_store_v2";
|
||||||
|
|
||||||
|
const defaultSchedule: PromptSourceSchedule = {
|
||||||
|
intervalMinutes: 30,
|
||||||
|
lastFetchedAt: "",
|
||||||
|
};
|
||||||
|
|
||||||
|
export const PROMPT_SOURCE_INTERVAL_OPTIONS = [
|
||||||
|
{ label: "关闭定时", value: 0 },
|
||||||
|
{ label: "每 30 分钟", value: 30 },
|
||||||
|
{ label: "每 1 小时", value: 60 },
|
||||||
|
{ label: "每 6 小时", value: 360 },
|
||||||
|
{ label: "每 24 小时", value: 1440 },
|
||||||
|
];
|
||||||
|
|
||||||
|
type PromptSourceStore = {
|
||||||
|
sources: PromptSource[];
|
||||||
|
schedule: PromptSourceSchedule;
|
||||||
|
addSource: () => PromptSource;
|
||||||
|
saveSource: (source: PromptSource) => void;
|
||||||
|
removeSource: (id: string) => void;
|
||||||
|
toggleSource: (id: string, enabled: boolean) => void;
|
||||||
|
updateSchedule: <K extends keyof PromptSourceSchedule>(key: K, value: PromptSourceSchedule[K]) => void;
|
||||||
|
};
|
||||||
|
|
||||||
|
export const usePromptSourceStore = create<PromptSourceStore>()(
|
||||||
|
persist(
|
||||||
|
(set) => ({
|
||||||
|
sources: DEFAULT_PROMPT_SOURCES,
|
||||||
|
schedule: defaultSchedule,
|
||||||
|
addSource: () => createPromptSource(),
|
||||||
|
saveSource: (source) =>
|
||||||
|
set((state) => ({
|
||||||
|
sources: state.sources.some((item) => item.id === source.id)
|
||||||
|
? state.sources.map((item) => (item.id === source.id && !item.builtIn ? createPromptSource(source) : item))
|
||||||
|
: [...state.sources, createPromptSource(source)],
|
||||||
|
})),
|
||||||
|
removeSource: (id) => set((state) => ({ sources: state.sources.filter((item) => item.id !== id || item.builtIn) })),
|
||||||
|
toggleSource: (id, enabled) => set((state) => ({ sources: state.sources.map((item) => (item.id === id ? { ...item, enabled } : item)) })),
|
||||||
|
updateSchedule: (key, value) => set((state) => ({ schedule: { ...state.schedule, [key]: value } })),
|
||||||
|
}),
|
||||||
|
{
|
||||||
|
name: PROMPT_SOURCE_STORE_KEY,
|
||||||
|
partialize: (state) => ({ sources: state.sources, schedule: state.schedule }),
|
||||||
|
merge: (persisted, current) => {
|
||||||
|
const persistedState = (persisted || {}) as Partial<PromptSourceStore>;
|
||||||
|
const savedSources = Array.isArray(persistedState.sources) ? persistedState.sources : [];
|
||||||
|
const enabledById = new Map(savedSources.map((source) => [source.id, source.enabled]));
|
||||||
|
const builtIn = DEFAULT_PROMPT_SOURCES.map((source) => ({ ...source, enabled: enabledById.get(source.id) ?? source.enabled }));
|
||||||
|
const custom = savedSources.filter((source) => !source.builtIn).map((source) => createPromptSource(source));
|
||||||
|
return { ...current, sources: [...builtIn, ...custom], schedule: { ...defaultSchedule, ...(persistedState.schedule || {}) } };
|
||||||
|
},
|
||||||
|
},
|
||||||
|
),
|
||||||
|
);
|
||||||
@@ -0,0 +1,53 @@
|
|||||||
|
import { nanoid } from "nanoid";
|
||||||
|
import { create } from "zustand";
|
||||||
|
import { createJSONStorage, persist } from "zustand/middleware";
|
||||||
|
|
||||||
|
import { localForageStorage } from "@/lib/localforage-storage";
|
||||||
|
|
||||||
|
export type PersonalPrompt = {
|
||||||
|
id: string;
|
||||||
|
title: string;
|
||||||
|
prompt: string;
|
||||||
|
description: string;
|
||||||
|
coverUrl: string;
|
||||||
|
referenceImageUrls: string[];
|
||||||
|
tags: string[];
|
||||||
|
createdAt: string;
|
||||||
|
updatedAt: string;
|
||||||
|
imageMode?: string;
|
||||||
|
imageModel?: string;
|
||||||
|
imageSize?: string;
|
||||||
|
imageCount?: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
export type PersonalPromptInput = Omit<PersonalPrompt, "id" | "createdAt" | "updatedAt">;
|
||||||
|
|
||||||
|
type PromptStore = {
|
||||||
|
hydrated: boolean;
|
||||||
|
prompts: PersonalPrompt[];
|
||||||
|
addPrompt: (prompt: PersonalPromptInput) => string;
|
||||||
|
updatePrompt: (id: string, prompt: PersonalPromptInput) => void;
|
||||||
|
removePrompt: (id: string) => void;
|
||||||
|
};
|
||||||
|
|
||||||
|
export const usePromptStore = create<PromptStore>()(
|
||||||
|
persist(
|
||||||
|
(set) => ({
|
||||||
|
hydrated: false,
|
||||||
|
prompts: [],
|
||||||
|
addPrompt: (prompt) => {
|
||||||
|
const id = nanoid();
|
||||||
|
const now = new Date().toISOString();
|
||||||
|
set((state) => ({ prompts: [{ ...prompt, id, createdAt: now, updatedAt: now }, ...state.prompts] }));
|
||||||
|
return id;
|
||||||
|
},
|
||||||
|
updatePrompt: (id, prompt) => set((state) => ({ prompts: state.prompts.map((item) => (item.id === id ? { ...item, ...prompt, updatedAt: new Date().toISOString() } : item)) })),
|
||||||
|
removePrompt: (id) => set((state) => ({ prompts: state.prompts.filter((item) => item.id !== id) })),
|
||||||
|
}),
|
||||||
|
{
|
||||||
|
name: "infinite-canvas:prompt_store",
|
||||||
|
storage: createJSONStorage(() => localForageStorage),
|
||||||
|
onRehydrateStorage: () => () => usePromptStore.setState({ hydrated: true }),
|
||||||
|
},
|
||||||
|
),
|
||||||
|
);
|
||||||
@@ -6,15 +6,30 @@ import { create } from "zustand";
|
|||||||
|
|
||||||
export type WorkbenchCommand = {
|
export type WorkbenchCommand = {
|
||||||
nonce: number;
|
nonce: number;
|
||||||
|
taskId?: string;
|
||||||
prompt?: string;
|
prompt?: string;
|
||||||
run: boolean;
|
run: boolean;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
export type WorkbenchGenerationTask = {
|
||||||
|
id: string;
|
||||||
|
kind: "image" | "video";
|
||||||
|
status: "queued" | "running" | "succeeded" | "failed";
|
||||||
|
prompt?: string;
|
||||||
|
createdAt: string;
|
||||||
|
updatedAt: string;
|
||||||
|
successCount?: number;
|
||||||
|
failCount?: number;
|
||||||
|
error?: string;
|
||||||
|
};
|
||||||
|
|
||||||
type WorkbenchAgentStore = {
|
type WorkbenchAgentStore = {
|
||||||
imageCommand: WorkbenchCommand | null;
|
imageCommand: WorkbenchCommand | null;
|
||||||
videoCommand: WorkbenchCommand | null;
|
videoCommand: WorkbenchCommand | null;
|
||||||
dispatchImage: (command: Omit<WorkbenchCommand, "nonce">) => void;
|
tasks: WorkbenchGenerationTask[];
|
||||||
dispatchVideo: (command: Omit<WorkbenchCommand, "nonce">) => void;
|
dispatchImage: (command: Omit<WorkbenchCommand, "nonce" | "taskId">) => string | undefined;
|
||||||
|
dispatchVideo: (command: Omit<WorkbenchCommand, "nonce" | "taskId">) => string | undefined;
|
||||||
|
updateTask: (id: string, patch: Partial<Pick<WorkbenchGenerationTask, "status" | "successCount" | "failCount" | "error">>) => void;
|
||||||
clearImageCommand: () => void;
|
clearImageCommand: () => void;
|
||||||
clearVideoCommand: () => void;
|
clearVideoCommand: () => void;
|
||||||
};
|
};
|
||||||
@@ -25,8 +40,25 @@ const nextNonce = () => (nonce += 1);
|
|||||||
export const useWorkbenchAgentStore = create<WorkbenchAgentStore>((set) => ({
|
export const useWorkbenchAgentStore = create<WorkbenchAgentStore>((set) => ({
|
||||||
imageCommand: null,
|
imageCommand: null,
|
||||||
videoCommand: null,
|
videoCommand: null,
|
||||||
dispatchImage: (command) => set({ imageCommand: { ...command, nonce: nextNonce() } }),
|
tasks: [],
|
||||||
dispatchVideo: (command) => set({ videoCommand: { ...command, nonce: nextNonce() } }),
|
dispatchImage: (command) => {
|
||||||
|
const commandNonce = nextNonce();
|
||||||
|
const task = command.run ? createTask("image", commandNonce, command.prompt) : undefined;
|
||||||
|
set((state) => ({ imageCommand: { ...command, nonce: commandNonce, taskId: task?.id }, tasks: task ? [task, ...state.tasks].slice(0, 30) : state.tasks }));
|
||||||
|
return task?.id;
|
||||||
|
},
|
||||||
|
dispatchVideo: (command) => {
|
||||||
|
const commandNonce = nextNonce();
|
||||||
|
const task = command.run ? createTask("video", commandNonce, command.prompt) : undefined;
|
||||||
|
set((state) => ({ videoCommand: { ...command, nonce: commandNonce, taskId: task?.id }, tasks: task ? [task, ...state.tasks].slice(0, 30) : state.tasks }));
|
||||||
|
return task?.id;
|
||||||
|
},
|
||||||
|
updateTask: (id, patch) => set((state) => ({ tasks: state.tasks.map((task) => (task.id === id ? { ...task, ...patch, updatedAt: new Date().toISOString() } : task)) })),
|
||||||
clearImageCommand: () => set({ imageCommand: null }),
|
clearImageCommand: () => set({ imageCommand: null }),
|
||||||
clearVideoCommand: () => set({ videoCommand: null }),
|
clearVideoCommand: () => set({ videoCommand: null }),
|
||||||
}));
|
}));
|
||||||
|
|
||||||
|
function createTask(kind: "image" | "video", commandNonce: number, prompt?: string): WorkbenchGenerationTask {
|
||||||
|
const now = new Date().toISOString();
|
||||||
|
return { id: `${kind}-${commandNonce}`, kind, status: "queued", prompt, createdAt: now, updatedAt: now };
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user