mirror of https://github.com/openclaw/openclaw.git
67 lines
2.3 KiB
TypeScript
67 lines
2.3 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
||
import { sanitizeMemoryText, stripConversationMetadata } from "./prompt.js";
|
||
|
||
describe("stripConversationMetadata", () => {
|
||
it("removes conversation metadata block and keeps user content", () => {
|
||
const input = [
|
||
"Conversation info (untrusted metadata):",
|
||
"```json",
|
||
'{"message_id":"123","sender":"u1"}',
|
||
"```",
|
||
"",
|
||
"帮我看看 /data 目录下有哪些文件",
|
||
].join("\n");
|
||
expect(stripConversationMetadata(input)).toBe("帮我看看 /data 目录下有哪些文件");
|
||
});
|
||
|
||
it("returns original text when no metadata prefix exists", () => {
|
||
const input = "如何查看 openclaw 调用 tools 的日志？";
|
||
expect(stripConversationMetadata(input)).toBe(input);
|
||
});
|
||
|
||
it("removes sender untrusted metadata block", () => {
|
||
const input = [
|
||
"Sender (untrusted metadata):",
|
||
"```json",
|
||
'{"label":"Longman","id":"537121267"}',
|
||
"```",
|
||
"",
|
||
"帮我看看当前机器磁盘空间还有多少",
|
||
].join("\n");
|
||
expect(stripConversationMetadata(input)).toBe("帮我看看当前机器磁盘空间还有多少");
|
||
});
|
||
|
||
it("removes metadata when prefixed inside intent/user_feedback lines", () => {
|
||
const input = [
|
||
"intent: Sender (untrusted metadata):",
|
||
"```json",
|
||
'{\"label\":\"Longman\",\"id\":\"537121267\"}',
|
||
"```",
|
||
"",
|
||
"帮我看看/home目录下有哪些文件？",
|
||
"user_feedback: Sender (untrusted metadata):",
|
||
"```json",
|
||
'{\"label\":\"Longman\",\"id\":\"537121267\"}',
|
||
"```",
|
||
"",
|
||
"很好",
|
||
].join("\n");
|
||
expect(stripConversationMetadata(input)).toBe(
|
||
["intent:", "帮我看看/home目录下有哪些文件？", "user_feedback:", "很好"].join("\n"),
|
||
);
|
||
});
|
||
|
||
it("sanitizes message id and sender id leakage for memory fields", () => {
|
||
const input = [
|
||
"[message_id: om_x100b559d255a5904c2ef5b224426c1b]",
|
||
"ou_41b427ee3d1ca8304e83f6540c04a3cb: 你帮我看看/data 目录下有什么",
|
||
"",
|
||
"user_feedback: [message_id: om_x100b559d3ea5d444c117283ea62a031]",
|
||
"ou_41b427ee3d1ca8304e83f6540c04a3cb: 赞",
|
||
].join("\n");
|
||
expect(sanitizeMemoryText(input)).toBe(
|
||
["你帮我看看/data 目录下有什么", "", "user_feedback:", "赞"].join("\n"),
|
||
);
|
||
});
|
||
});
|