93 lines
3.0 KiB
TypeScript
93 lines
3.0 KiB
TypeScript
import "server-only";
|
|
|
|
export interface PromptMemoryEntry {
|
|
optimizedPrompt: string;
|
|
sourceIntent?: string;
|
|
sourceText?: string;
|
|
reusable?: boolean;
|
|
}
|
|
|
|
function normalizeText(value: unknown, max = 2000): string | undefined {
|
|
const normalized = String(value || "").trim();
|
|
if (!normalized) return undefined;
|
|
return normalized.slice(0, max);
|
|
}
|
|
|
|
export function extractPromptMemoryFromWorkingState(
|
|
workingState?: {
|
|
lastTurnSummary?: unknown;
|
|
} | null,
|
|
): PromptMemoryEntry | null {
|
|
const summary = workingState?.lastTurnSummary;
|
|
if (!summary || typeof summary !== "object") return null;
|
|
const summaryRecord = summary as Record<string, unknown>;
|
|
const memory = summaryRecord.promptMemory;
|
|
if (memory && typeof memory === "object") {
|
|
const memoryRecord = memory as Record<string, unknown>;
|
|
const optimizedPrompt = normalizeText(memoryRecord.optimizedPrompt, 4000);
|
|
if (!optimizedPrompt) return null;
|
|
return {
|
|
optimizedPrompt,
|
|
sourceIntent: normalizeText(memoryRecord.sourceIntent, 64),
|
|
sourceText: normalizeText(memoryRecord.sourceText, 500),
|
|
reusable:
|
|
typeof memoryRecord.reusable === "boolean" ? memoryRecord.reusable : true,
|
|
};
|
|
}
|
|
|
|
const resolutionMode = normalizeText(summaryRecord.resolutionMode, 64);
|
|
const optimizedPrompt = normalizeText(summaryRecord.optimizedPrompt, 4000);
|
|
if (resolutionMode === "prompt_only" && optimizedPrompt) {
|
|
return {
|
|
optimizedPrompt,
|
|
sourceIntent: "image_gen",
|
|
sourceText: normalizeText(summaryRecord.text, 500),
|
|
reusable: true,
|
|
};
|
|
}
|
|
return null;
|
|
}
|
|
|
|
export function buildPromptMemoryUpdate(params: {
|
|
optimizedPrompt: string;
|
|
sourceIntent?: string;
|
|
sourceText?: string;
|
|
reusable?: boolean;
|
|
}): PromptMemoryEntry {
|
|
return {
|
|
optimizedPrompt: normalizeText(params.optimizedPrompt, 4000) || "",
|
|
sourceIntent: normalizeText(params.sourceIntent, 64),
|
|
sourceText: normalizeText(params.sourceText, 500),
|
|
reusable: params.reusable !== false,
|
|
};
|
|
}
|
|
|
|
export function isPromptCarryForwardRequest(params: {
|
|
text: string;
|
|
hasImage: boolean;
|
|
}): boolean {
|
|
const normalized = String(params.text || "").toLowerCase().replace(/\s+/g, "");
|
|
if (!normalized) return false;
|
|
|
|
const refersPreviousPrompt =
|
|
/根据(上一个|上一句|刚才|上面|之前|那个)?提示词|按(上一个|上一句|刚才|上面|之前|那个)?提示词|用(上一个|上一句|刚才|上面|之前|那个)?提示词|用刚才输出的提示词|按刚才输出的提示词|按上面的提示词|沿用上个提示词|继续用那个提示词/.test(
|
|
normalized,
|
|
);
|
|
|
|
if (!refersPreviousPrompt) return false;
|
|
|
|
const wantsExecution =
|
|
/生图|生成图片|出图|画出来|做图|继续生成|继续做|重新做|重绘|优化这张图|优化这个图|处理这张图|处理这个图|按这个做|照这个做|按此生成/.test(
|
|
normalized,
|
|
);
|
|
|
|
if (wantsExecution) return true;
|
|
|
|
if (params.hasImage) {
|
|
return /优化|处理|重绘|参考这张图|结合这张图|基于这张图/.test(normalized);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|