86 lines
4.4 KiB
TypeScript
86 lines
4.4 KiB
TypeScript
import { buildSkillMentionReferences, renderSkillPrompt } from "@/lib/canvas/canvas-skill-mentions";
|
|
import { canvasResourceMentionToken, type CanvasResourceReference } from "@/lib/canvas/canvas-resource-references";
|
|
import type { Skill } from "@/services/api/skills";
|
|
import { creationAttachmentKind, type CreationAttachment } from "./creation-assets";
|
|
|
|
export type CreationReference = CanvasResourceReference & {
|
|
attachmentId?: string;
|
|
};
|
|
|
|
export function buildCreationMentionReferences(skills: Skill[], attachments: CreationAttachment[] = [], snapshots: CreationReference[] = []) {
|
|
const attachmentReferences = attachments.map(attachmentReference);
|
|
const skillReferences = buildSkillMentionReferences(skills) as CreationReference[];
|
|
const current = [...attachmentReferences, ...skillReferences];
|
|
const currentIDs = new Set(current.map((reference) => reference.id));
|
|
const restored = snapshots.filter((reference) => reference.kind === "skill" && !currentIDs.has(reference.id));
|
|
return [...current, ...restored].map((reference) => ({ ...reference, active: true }));
|
|
}
|
|
|
|
export function selectedCreationReferences(prompt: string, references: CreationReference[]) {
|
|
return references.filter((reference) => prompt.includes(canvasResourceMentionToken(reference)));
|
|
}
|
|
|
|
export function reconcileCreationAttachmentLimit(attachments: CreationAttachment[], references: CreationReference[], maxReferences: number) {
|
|
const limit = Math.max(0, Math.floor(maxReferences));
|
|
if (attachments.length <= limit) return { attachments, removedReferences: [] as CreationReference[] };
|
|
|
|
const nextAttachments = attachments.slice(0, limit);
|
|
const removedAttachmentIds = new Set(attachments.slice(limit).map((attachment) => attachment.id));
|
|
const removedReferences = references.filter((reference) => reference.attachmentId && removedAttachmentIds.has(reference.attachmentId));
|
|
return { attachments: nextAttachments, removedReferences };
|
|
}
|
|
|
|
export function removeCreationReferenceTokens(value: string, references: CreationReference[]) {
|
|
return references.reduce((current, reference) => current.split(canvasResourceMentionToken(reference)).join(""), value);
|
|
}
|
|
|
|
export function displayCreationPrompt(prompt: string, references: CreationReference[]) {
|
|
return references.reduce((value, reference) => value.split(canvasResourceMentionToken(reference)).join(`@${reference.label}`), prompt);
|
|
}
|
|
|
|
export function expandCreationPrompt(prompt: string, references: CreationReference[], attachments: CreationAttachment[] = []) {
|
|
const visiblePrompt = displayCreationPrompt(prompt, references).trim();
|
|
if (!references.length) return visiblePrompt;
|
|
|
|
const contexts: string[] = [];
|
|
const mediaMappings: string[] = [];
|
|
const attachmentPositions = new Map(attachments.map((attachment, index) => [attachment.id, index + 1]));
|
|
references.forEach((reference) => {
|
|
if (reference.kind === "skill" && reference.skill) {
|
|
contexts.push(renderSkillPrompt(reference.skill));
|
|
return;
|
|
}
|
|
if (reference.attachmentId) {
|
|
const position = attachmentPositions.get(reference.attachmentId);
|
|
const kindLabel = reference.kind === "video" ? "视频" : reference.kind === "audio" ? "音频" : "图片";
|
|
mediaMappings.push(`- @${reference.label}：参考${kindLabel} ${position || 1}`);
|
|
return;
|
|
}
|
|
});
|
|
|
|
if (mediaMappings.length) contexts.push(`【资源对应关系】\n${mediaMappings.join("\n")}`);
|
|
return [...contexts, `【创作要求】\n${visiblePrompt}`].filter(Boolean).join("\n\n");
|
|
}
|
|
|
|
export function creationReferenceMetadata(references: CreationReference[]) {
|
|
return {
|
|
skillIds: references.flatMap((reference) => (reference.skill?.skill_id ? [reference.skill.skill_id] : [])),
|
|
};
|
|
}
|
|
|
|
function attachmentReference(attachment: CreationAttachment, index: number): CreationReference {
|
|
const kind = creationAttachmentKind(attachment);
|
|
const label = kind === "video" ? "视频" : kind === "audio" ? "音频" : "图片";
|
|
return {
|
|
id: `upload:${attachment.id}`,
|
|
nodeId: `upload:${attachment.id}`,
|
|
kind,
|
|
label: `${label}${index + 1}`,
|
|
title: "当前参考内容",
|
|
previewUrl: attachment.previewUrl || ("dataUrl" in attachment ? attachment.dataUrl : attachment.url),
|
|
storageKey: attachment.storageKey,
|
|
active: true,
|
|
attachmentId: attachment.id,
|
|
};
|
|
}
|