refactor: extract canvas agent stream helpers
This commit is contained in:
Your Name
parent
0f058207bb
commit
ef837f9985
@ -104,6 +104,17 @@ import {
|
||||
type CanvasToolbarConfig,
|
||||
type CanvasToolbarItemKey,
|
||||
} from "@/lib/canvas-toolbar-config";
|
||||
import {
|
||||
AGENT_PROGRESS_STEPS,
|
||||
type AgentProgressStreamPayload,
|
||||
type AgentTargetLockRefPreview,
|
||||
type AgentTargetLockStreamPayload,
|
||||
type CanvasMediaStreamPayload,
|
||||
getAgentProgressFallbackTitle,
|
||||
getAgentTargetLockTitle,
|
||||
getAgentTargetResolutionLabel,
|
||||
type ImageGenerationStatusStreamPayload,
|
||||
} from "@/components/canvas/chat-panel/canvas-agent-stream";
|
||||
import {
|
||||
clampChatRequestText,
|
||||
deriveSessionTitleFromMessages,
|
||||
@ -170,76 +181,6 @@ interface TagPreviewInfo {
|
||||
};
|
||||
}
|
||||
|
||||
type CanvasMediaStreamPayload = {
|
||||
messageId?: string;
|
||||
imageUrls?: string[];
|
||||
videoUrls?: string[];
|
||||
prompt?: string;
|
||||
sourceImageUrls?: string[];
|
||||
agentPresetId?: string;
|
||||
agentPresetName?: string;
|
||||
agentCategoryId?: string;
|
||||
agentCategoryName?: string;
|
||||
agentExecutionSnapshotId?: string;
|
||||
};
|
||||
|
||||
type ImageGenerationStatusStreamPayload = {
|
||||
phase?: "queued" | "processing" | "finalizing" | "failed" | "completed";
|
||||
providerType?: string;
|
||||
modelName?: string;
|
||||
imageCount?: number;
|
||||
queuePosition?: number;
|
||||
sameModelRunningCount?: number;
|
||||
concurrencyLimit?: number;
|
||||
estimatedWaitMs?: number;
|
||||
};
|
||||
|
||||
type AgentTargetLockRefPreview = {
|
||||
url: string;
|
||||
title?: string;
|
||||
nodeId?: string;
|
||||
assetId?: string;
|
||||
source:
|
||||
| "current_message"
|
||||
| "canvas_selection"
|
||||
| "canvas_context"
|
||||
| "history_user_upload"
|
||||
| "history_assistant_result"
|
||||
| "working_memory";
|
||||
role: "upload" | "selected" | "reference" | "history_result" | "working_memory";
|
||||
isSelected?: boolean;
|
||||
positionLabel?: string;
|
||||
};
|
||||
|
||||
type AgentTargetLockStreamPayload = {
|
||||
stage?: "locked" | "clarify" | "unresolved";
|
||||
resolutionMode?:
|
||||
| "explicit_selection"
|
||||
| "current_upload"
|
||||
| "text_reference"
|
||||
| "working_memory"
|
||||
| "latest_result"
|
||||
| "clarification"
|
||||
| "none";
|
||||
confidence?: number;
|
||||
reason?: string;
|
||||
needsClarification?: boolean;
|
||||
clarificationMessage?: string;
|
||||
targetCount?: number;
|
||||
candidateCount?: number;
|
||||
targets?: AgentTargetLockRefPreview[];
|
||||
candidates?: AgentTargetLockRefPreview[];
|
||||
};
|
||||
|
||||
type AgentProgressStreamPayload = {
|
||||
stage?: "planning" | "clarify" | "executing" | "completed" | "failed";
|
||||
currentStep?: "lock_target" | "analyze" | "select_tool" | "execute" | "finalize";
|
||||
title?: string;
|
||||
message?: string;
|
||||
toolName?: string;
|
||||
capability?: string;
|
||||
};
|
||||
|
||||
interface CanvasChatPanelProps {
|
||||
chatId: string;
|
||||
initialMessages?: ChatMessage[];
|
||||
@ -426,59 +367,6 @@ function buildPreviewImageUrl(url: string, width: number): string {
|
||||
});
|
||||
}
|
||||
|
||||
function getAgentTargetLockTitle(payload?: AgentTargetLockStreamPayload | null) {
|
||||
if (payload?.stage === "clarify") return "需要确认目标";
|
||||
if (payload?.stage === "locked") return "当前锁定目标";
|
||||
return "待锁定目标";
|
||||
}
|
||||
|
||||
function getAgentTargetResolutionLabel(mode?: AgentTargetLockStreamPayload["resolutionMode"]) {
|
||||
switch (mode) {
|
||||
case "explicit_selection":
|
||||
return "画布选中";
|
||||
case "current_upload":
|
||||
return "当前上传";
|
||||
case "text_reference":
|
||||
return "文本定位";
|
||||
case "working_memory":
|
||||
return "工作记忆";
|
||||
case "latest_result":
|
||||
return "最新结果";
|
||||
case "clarification":
|
||||
return "待确认";
|
||||
case "none":
|
||||
return "未解析";
|
||||
default:
|
||||
return "自动解析";
|
||||
}
|
||||
}
|
||||
|
||||
const AGENT_PROGRESS_STEPS: Array<{
|
||||
key: NonNullable<AgentProgressStreamPayload["currentStep"]>;
|
||||
label: string;
|
||||
}> = [
|
||||
{ key: "lock_target", label: "整理素材" },
|
||||
{ key: "analyze", label: "理解需求" },
|
||||
{ key: "select_tool", label: "优化提示" },
|
||||
{ key: "execute", label: "提交生成" },
|
||||
{ key: "finalize", label: "整理结果" },
|
||||
];
|
||||
|
||||
function getAgentProgressFallbackTitle(payload?: AgentProgressStreamPayload | null) {
|
||||
switch (payload?.stage) {
|
||||
case "clarify":
|
||||
return "需要补充信息";
|
||||
case "executing":
|
||||
return "正在生成结果";
|
||||
case "completed":
|
||||
return "创作完成";
|
||||
case "failed":
|
||||
return "创作失败";
|
||||
default:
|
||||
return "正在理解需求";
|
||||
}
|
||||
}
|
||||
|
||||
const CANVAS_IMAGE_ANALYSIS_TITLE = "图片分析";
|
||||
const CANVAS_IMAGE_DONE_SUMMARY = "已完成本次生成，如需微调可继续告诉我。";
|
||||
const CANVAS_AGENT_SECTION_TITLES = new Set([
|
||||
|
||||
124
components/canvas/chat-panel/canvas-agent-stream.ts
Normal file
124
components/canvas/chat-panel/canvas-agent-stream.ts
Normal file
@ -0,0 +1,124 @@
|
||||
export type CanvasMediaStreamPayload = {
|
||||
messageId?: string;
|
||||
imageUrls?: string[];
|
||||
videoUrls?: string[];
|
||||
prompt?: string;
|
||||
sourceImageUrls?: string[];
|
||||
agentPresetId?: string;
|
||||
agentPresetName?: string;
|
||||
agentCategoryId?: string;
|
||||
agentCategoryName?: string;
|
||||
agentExecutionSnapshotId?: string;
|
||||
};
|
||||
|
||||
export type ImageGenerationStatusStreamPayload = {
|
||||
phase?: "queued" | "processing" | "finalizing" | "failed" | "completed";
|
||||
providerType?: string;
|
||||
modelName?: string;
|
||||
imageCount?: number;
|
||||
queuePosition?: number;
|
||||
sameModelRunningCount?: number;
|
||||
concurrencyLimit?: number;
|
||||
estimatedWaitMs?: number;
|
||||
};
|
||||
|
||||
export type AgentTargetLockRefPreview = {
|
||||
url: string;
|
||||
title?: string;
|
||||
nodeId?: string;
|
||||
assetId?: string;
|
||||
source:
|
||||
| "current_message"
|
||||
| "canvas_selection"
|
||||
| "canvas_context"
|
||||
| "history_user_upload"
|
||||
| "history_assistant_result"
|
||||
| "working_memory";
|
||||
role: "upload" | "selected" | "reference" | "history_result" | "working_memory";
|
||||
isSelected?: boolean;
|
||||
positionLabel?: string;
|
||||
};
|
||||
|
||||
export type AgentTargetLockStreamPayload = {
|
||||
stage?: "locked" | "clarify" | "unresolved";
|
||||
resolutionMode?:
|
||||
| "explicit_selection"
|
||||
| "current_upload"
|
||||
| "text_reference"
|
||||
| "working_memory"
|
||||
| "latest_result"
|
||||
| "clarification"
|
||||
| "none";
|
||||
confidence?: number;
|
||||
reason?: string;
|
||||
needsClarification?: boolean;
|
||||
clarificationMessage?: string;
|
||||
targetCount?: number;
|
||||
candidateCount?: number;
|
||||
targets?: AgentTargetLockRefPreview[];
|
||||
candidates?: AgentTargetLockRefPreview[];
|
||||
};
|
||||
|
||||
export type AgentProgressStreamPayload = {
|
||||
stage?: "planning" | "clarify" | "executing" | "completed" | "failed";
|
||||
currentStep?: "lock_target" | "analyze" | "select_tool" | "execute" | "finalize";
|
||||
title?: string;
|
||||
message?: string;
|
||||
toolName?: string;
|
||||
capability?: string;
|
||||
};
|
||||
|
||||
export const AGENT_PROGRESS_STEPS: Array<{
|
||||
key: NonNullable<AgentProgressStreamPayload["currentStep"]>;
|
||||
label: string;
|
||||
}> = [
|
||||
{ key: "lock_target", label: "整理素材" },
|
||||
{ key: "analyze", label: "理解需求" },
|
||||
{ key: "select_tool", label: "优化提示" },
|
||||
{ key: "execute", label: "提交生成" },
|
||||
{ key: "finalize", label: "整理结果" },
|
||||
];
|
||||
|
||||
export function getAgentTargetLockTitle(payload?: AgentTargetLockStreamPayload | null) {
|
||||
if (payload?.stage === "clarify") return "需要确认目标";
|
||||
if (payload?.stage === "locked") return "当前锁定目标";
|
||||
return "待锁定目标";
|
||||
}
|
||||
|
||||
export function getAgentTargetResolutionLabel(
|
||||
mode?: AgentTargetLockStreamPayload["resolutionMode"]
|
||||
) {
|
||||
switch (mode) {
|
||||
case "explicit_selection":
|
||||
return "画布选中";
|
||||
case "current_upload":
|
||||
return "当前上传";
|
||||
case "text_reference":
|
||||
return "文本定位";
|
||||
case "working_memory":
|
||||
return "工作记忆";
|
||||
case "latest_result":
|
||||
return "最新结果";
|
||||
case "clarification":
|
||||
return "待确认";
|
||||
case "none":
|
||||
return "未解析";
|
||||
default:
|
||||
return "自动解析";
|
||||
}
|
||||
}
|
||||
|
||||
export function getAgentProgressFallbackTitle(payload?: AgentProgressStreamPayload | null) {
|
||||
switch (payload?.stage) {
|
||||
case "clarify":
|
||||
return "需要补充信息";
|
||||
case "executing":
|
||||
return "正在生成结果";
|
||||
case "completed":
|
||||
return "创作完成";
|
||||
case "failed":
|
||||
return "创作失败";
|
||||
default:
|
||||
return "正在理解需求";
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user