115 lines
3.0 KiB
TypeScript
115 lines
3.0 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it } from "bun:test";
|
|
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
|
|
import { SessionTranscriptStore } from "../../src/sessions/transcriptStore.js";
|
|
|
|
describe("SessionTranscriptStore", () => {
|
|
let tempDir: string;
|
|
let store: SessionTranscriptStore;
|
|
|
|
beforeEach(async () => {
|
|
tempDir = await mkdtemp(join(tmpdir(), "tjwater-transcript-"));
|
|
store = new SessionTranscriptStore(tempDir);
|
|
await store.initialize();
|
|
});
|
|
|
|
afterEach(async () => {
|
|
await rm(tempDir, { force: true, recursive: true });
|
|
});
|
|
|
|
it("clones only the kept prefix when forking a thread", async () => {
|
|
await store.appendTurn(
|
|
{
|
|
actorKey: "actor-2",
|
|
clientSessionId: "thread-source",
|
|
projectKey: "project-2",
|
|
sessionId: "thread-source",
|
|
},
|
|
{
|
|
assistantMessage: "第一轮回复",
|
|
toolCallCount: 0,
|
|
userMessage: "第一轮提问",
|
|
},
|
|
);
|
|
await store.appendTurn(
|
|
{
|
|
actorKey: "actor-2",
|
|
clientSessionId: "thread-source",
|
|
projectKey: "project-2",
|
|
sessionId: "thread-source",
|
|
},
|
|
{
|
|
assistantMessage: "第二轮回复",
|
|
toolCallCount: 0,
|
|
userMessage: "第二轮提问",
|
|
},
|
|
);
|
|
|
|
const cloned = await store.cloneThread(
|
|
{
|
|
actorKey: "actor-2",
|
|
clientSessionId: "thread-source",
|
|
projectKey: "project-2",
|
|
sessionId: "thread-source",
|
|
},
|
|
{
|
|
actorKey: "actor-2",
|
|
clientSessionId: "thread-fork",
|
|
projectKey: "project-2",
|
|
sessionId: "thread-fork",
|
|
},
|
|
2,
|
|
);
|
|
|
|
expect(cloned.turns).toHaveLength(1);
|
|
expect(cloned.turns[0]?.userMessage).toBe("第一轮提问");
|
|
|
|
const forkRecentTurns = await store.getRecentTurns(
|
|
{
|
|
actorKey: "actor-2",
|
|
clientSessionId: "thread-fork",
|
|
projectKey: "project-2",
|
|
sessionId: "thread-fork",
|
|
},
|
|
5,
|
|
);
|
|
expect(forkRecentTurns).toHaveLength(1);
|
|
expect(forkRecentTurns[0]?.assistantMessage).toBe("第一轮回复");
|
|
});
|
|
|
|
it("does not duplicate the latest turn when the frontend state is saved again", async () => {
|
|
await store.appendTurn(
|
|
{
|
|
actorKey: "actor-3",
|
|
clientSessionId: "thread-3",
|
|
projectKey: "project-3",
|
|
sessionId: "thread-3",
|
|
},
|
|
{
|
|
assistantMessage: "已完成压力波动分析。",
|
|
toolCallCount: 1,
|
|
userMessage: "分析压力波动。",
|
|
},
|
|
);
|
|
|
|
const transcript = await store.appendTurn(
|
|
{
|
|
actorKey: "actor-3",
|
|
clientSessionId: "thread-3",
|
|
projectKey: "project-3",
|
|
sessionId: "thread-3",
|
|
},
|
|
{
|
|
assistantMessage: "已完成压力波动分析。",
|
|
toolCallCount: 2,
|
|
userMessage: "分析压力波动。",
|
|
},
|
|
);
|
|
|
|
expect(transcript.turns).toHaveLength(1);
|
|
expect(transcript.turns[0]?.toolCallCount).toBe(2);
|
|
});
|
|
});
|