88 lines
3.8 KiB
TypeScript
88 lines
3.8 KiB
TypeScript
import { canonicalize } from "json-canonicalize";
|
|
|
|
import type { NodeGenerationContext } from "@/components/canvas/canvas-node-generation";
|
|
import type { CanvasNodeGenerationMode } from "@/components/canvas/canvas-node-prompt-panel";
|
|
import { generationPromptFingerprint } from "@/lib/generation-error";
|
|
|
|
type GenerationReference = NodeGenerationContext["referenceImages"][number] | NodeGenerationContext["referenceVideos"][number] | NodeGenerationContext["referenceAudios"][number];
|
|
|
|
export type CanvasGenerationRequestFingerprintInput = {
|
|
nodeId: string;
|
|
mode: CanvasNodeGenerationMode;
|
|
prompt: string;
|
|
model: string;
|
|
options: Record<string, unknown>;
|
|
workflow?: {
|
|
provider?: string;
|
|
workflowId?: string;
|
|
workflowKind?: string;
|
|
parameters?: Record<string, unknown>;
|
|
};
|
|
operation?: string;
|
|
audioInstructions?: string;
|
|
promptTemplateOperation?: string;
|
|
promptTemplateVariables?: Record<string, string>;
|
|
context: Pick<NodeGenerationContext, "referenceImages" | "referenceVideos" | "referenceAudios" | "characterReferences" | "resolvedCharacterVersions" | "resolvedCharacterVoices">;
|
|
};
|
|
|
|
/**
|
|
* 编辑器保留带 @ 槽位的原始提示词，任务请求单独保存经过技能/风格处理的模型提示词。
|
|
* 模型提示词也保留 @图片1 等槽位标记。
|
|
* 两者不能互相覆盖，否则刷新后富引用会退化为普通“图片1”文本。
|
|
*/
|
|
export function canvasGenerationPromptMetadata(composerContent: string, effectivePrompt: string) {
|
|
return { composerContent, prompt: effectivePrompt };
|
|
}
|
|
|
|
export function canvasGenerationRequestFingerprint(input: CanvasGenerationRequestFingerprintInput) {
|
|
const serialized = canonicalize({
|
|
version: 1,
|
|
nodeId: input.nodeId,
|
|
mode: input.mode,
|
|
prompt: input.prompt.trim(),
|
|
model: input.model,
|
|
options: input.options,
|
|
workflow: input.workflow,
|
|
operation: input.operation,
|
|
audioInstructions: input.audioInstructions,
|
|
promptTemplateOperation: input.promptTemplateOperation,
|
|
promptTemplateVariables: input.promptTemplateVariables,
|
|
references: {
|
|
images: input.context.referenceImages.map(referenceIdentity),
|
|
videos: input.context.referenceVideos.map(referenceIdentity),
|
|
audios: input.context.referenceAudios.map(referenceIdentity),
|
|
characters: input.context.characterReferences,
|
|
resolvedCharacterVersions: input.context.resolvedCharacterVersions,
|
|
resolvedCharacterVoices: input.context.resolvedCharacterVoices,
|
|
},
|
|
});
|
|
return `canvas-generation:v1:${generationPromptFingerprint(serialized)}`;
|
|
}
|
|
|
|
function referenceIdentity(reference: GenerationReference) {
|
|
const source = "storageKey" in reference && reference.storageKey ? reference.storageKey : "url" in reference && reference.url ? reference.url : "dataUrl" in reference ? reference.dataUrl : "";
|
|
return {
|
|
id: reference.id,
|
|
name: reference.name,
|
|
type: reference.type,
|
|
source: source.startsWith("data:") ? `data:${generationPromptFingerprint(source)}` : source,
|
|
derivedSource: "source" in reference ? reference.source : undefined,
|
|
durationMs: "durationMs" in reference ? reference.durationMs : undefined,
|
|
};
|
|
}
|
|
|
|
export function runCanvasGenerationSubmissionOnce<T>(locks: Map<string, Promise<unknown>>, nodeId: string, operation: () => Promise<T>, onDuplicate?: () => void): Promise<T> {
|
|
const existing = locks.get(nodeId) as Promise<T> | undefined;
|
|
if (existing) {
|
|
onDuplicate?.();
|
|
return existing;
|
|
}
|
|
const running = Promise.resolve()
|
|
.then(operation)
|
|
.finally(() => {
|
|
if (locks.get(nodeId) === running) locks.delete(nodeId);
|
|
});
|
|
locks.set(nodeId, running);
|
|
return running;
|
|
}
|