Luke The Dev
139 lines
3.9 KiB
TypeScript
139 lines
3.9 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
|
|
import {
|
|
agentStoreReducer,
|
|
initialAgentStoreState,
|
|
type AgentStoreSeed,
|
|
} from "@/features/agents/state/store";
|
|
import { createTranscriptEntryFromLine } from "@/features/agents/state/transcript";
|
|
|
|
describe("agent store transcript upsert", () => {
|
|
it("replaces appendOutput entries that share transcript entryId", () => {
|
|
const seed: AgentStoreSeed = {
|
|
agentId: "agent-1",
|
|
name: "Agent One",
|
|
sessionKey: "agent:agent-1:studio:test-session",
|
|
};
|
|
|
|
let state = agentStoreReducer(initialAgentStoreState, {
|
|
type: "hydrateAgents",
|
|
agents: [seed],
|
|
});
|
|
|
|
state = agentStoreReducer(state, {
|
|
type: "appendOutput",
|
|
agentId: "agent-1",
|
|
line: "fallback final",
|
|
transcript: {
|
|
source: "runtime-agent",
|
|
runId: "run-1",
|
|
sessionKey: seed.sessionKey,
|
|
timestampMs: 1000,
|
|
role: "assistant",
|
|
kind: "assistant",
|
|
entryId: "run:run-1:assistant:final",
|
|
confirmed: false,
|
|
},
|
|
});
|
|
|
|
state = agentStoreReducer(state, {
|
|
type: "appendOutput",
|
|
agentId: "agent-1",
|
|
line: "canonical final",
|
|
transcript: {
|
|
source: "runtime-chat",
|
|
runId: "run-1",
|
|
sessionKey: seed.sessionKey,
|
|
timestampMs: 1100,
|
|
role: "assistant",
|
|
kind: "assistant",
|
|
entryId: "run:run-1:assistant:final",
|
|
confirmed: true,
|
|
},
|
|
});
|
|
|
|
const agent = state.agents.find((entry) => entry.agentId === "agent-1");
|
|
const assistantEntries = (agent?.transcriptEntries ?? []).filter(
|
|
(entry) => entry.kind === "assistant"
|
|
);
|
|
|
|
expect(assistantEntries).toHaveLength(1);
|
|
expect(assistantEntries[0]?.text).toBe("canonical final");
|
|
expect(assistantEntries[0]?.confirmed).toBe(true);
|
|
expect(agent?.outputLines).toEqual(["canonical final"]);
|
|
});
|
|
|
|
it("collapses duplicate transcript entryIds when applying an upsert", () => {
|
|
const seed: AgentStoreSeed = {
|
|
agentId: "agent-1",
|
|
name: "Agent One",
|
|
sessionKey: "agent:agent-1:studio:test-session",
|
|
};
|
|
|
|
let state = agentStoreReducer(initialAgentStoreState, {
|
|
type: "hydrateAgents",
|
|
agents: [seed],
|
|
});
|
|
|
|
const first = createTranscriptEntryFromLine({
|
|
line: "first duplicate",
|
|
sessionKey: seed.sessionKey,
|
|
source: "runtime-agent",
|
|
sequenceKey: 1,
|
|
runId: "run-1",
|
|
role: "assistant",
|
|
kind: "assistant",
|
|
entryId: "run:run-1:assistant:final",
|
|
confirmed: false,
|
|
});
|
|
const second = createTranscriptEntryFromLine({
|
|
line: "second duplicate",
|
|
sessionKey: seed.sessionKey,
|
|
source: "runtime-chat",
|
|
sequenceKey: 2,
|
|
runId: "run-1",
|
|
role: "assistant",
|
|
kind: "assistant",
|
|
entryId: "run:run-1:assistant:final",
|
|
confirmed: false,
|
|
});
|
|
if (!first || !second) {
|
|
throw new Error("expected transcript entries");
|
|
}
|
|
|
|
state = agentStoreReducer(state, {
|
|
type: "updateAgent",
|
|
agentId: "agent-1",
|
|
patch: {
|
|
transcriptEntries: [first, second],
|
|
outputLines: [first.text, second.text],
|
|
},
|
|
});
|
|
|
|
state = agentStoreReducer(state, {
|
|
type: "appendOutput",
|
|
agentId: "agent-1",
|
|
line: "canonical final",
|
|
transcript: {
|
|
source: "runtime-chat",
|
|
runId: "run-1",
|
|
sessionKey: seed.sessionKey,
|
|
timestampMs: 1100,
|
|
role: "assistant",
|
|
kind: "assistant",
|
|
entryId: "run:run-1:assistant:final",
|
|
confirmed: true,
|
|
},
|
|
});
|
|
|
|
const agent = state.agents.find((entry) => entry.agentId === "agent-1");
|
|
const assistantEntries = (agent?.transcriptEntries ?? []).filter(
|
|
(entry) => entry.kind === "assistant"
|
|
);
|
|
|
|
expect(assistantEntries).toHaveLength(1);
|
|
expect(assistantEntries[0]?.text).toBe("canonical final");
|
|
expect(agent?.outputLines).toEqual(["canonical final"]);
|
|
});
|
|
});
|