LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-05-20 05:07:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: read truncated transcripts from sqlite

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-08 15:39:05 +01:00
parent fa53d1193e
commit 06d03ae1fb
1 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/agents/pi-embedded-runner/tool-result-truncation.test.ts
View File

@@ -4,9 +4,12 @@ import path from "node:path";
import type { AgentMessage } from "openclaw/plugin-sdk/agent-core";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { onSessionTranscriptUpdate } from "../../sessions/transcript-events.js";
import { closeOpenClawAgentDatabasesForTest } from "../../state/openclaw-agent-db.js";
import { closeOpenClawStateDatabaseForTest } from "../../state/openclaw-state-db.js";
import type { AssistantMessage, ToolResultMessage, UserMessage } from "../pi-ai-contract.js";
import { makeAgentAssistantMessage } from "../test-helpers/agent-message-fixtures.js";
import { SessionManager } from "../transcript/session-transcript-contract.js";
import { readTranscriptState } from "../transcript/transcript-state.js";
let truncateToolResultText: typeof import("./tool-result-truncation.js").truncateToolResultText;
let truncateToolResultMessage: typeof import("./tool-result-truncation.js").truncateToolResultMessage;
@@ -50,6 +53,9 @@ beforeEach(async () => {
});
afterEach(async () => {
closeOpenClawAgentDatabasesForTest();
closeOpenClawStateDatabaseForTest();
vi.unstubAllEnvs();
if (tmpDir) {
await fs.rm(tmpDir, { recursive: true, force: true }).catch(() => {});
tmpDir = undefined;
@@ -94,9 +100,14 @@ function getFirstToolResultText(message: AgentMessage | ToolResultMessage): stri
async function createTmpDir(): Promise<string> {
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), "tool-result-truncation-test-"));
vi.stubEnv("OPENCLAW_STATE_DIR", tmpDir);
return tmpDir;
}
async function loadBranch(sessionFile: string) {
return (await readTranscriptState(sessionFile)).getBranch();
}
describe("truncateToolResultText", () => {
it("returns text unchanged when under limit", () => {
const text = "hello world";
@@ -432,7 +443,7 @@ describe("truncateOversizedToolResultsInSession", () => {
sm.appendMessage(makeToolResult(medium, "call_3"));
const sessionFile = sm.getSessionFile()!;
const beforeBranch = SessionManager.open(sessionFile).getBranch();
const beforeBranch = await loadBranch(sessionFile);
const beforeLengths = beforeBranch
.filter((entry) => entry.type === "message")
.map((entry) =>
@@ -459,7 +470,7 @@ describe("truncateOversizedToolResultsInSession", () => {
expect(result.truncatedCount).toBeGreaterThan(0);
expect(listener).toHaveBeenCalledWith({ sessionFile, sessionKey: "agent:main:test" });
const afterBranch = SessionManager.open(sessionFile).getBranch();
const afterBranch = await loadBranch(sessionFile);
const afterToolResults = afterBranch.filter(
(entry) => entry.type === "message" && entry.message.role === "toolResult",
);
@@ -497,7 +508,7 @@ describe("truncateOversizedToolResultsInSession", () => {
sm.appendMessage(makeToolResult(newerEnough, "call_2"));
const sessionFile = sm.getSessionFile()!;
const beforeBranch = SessionManager.open(sessionFile).getBranch();
const beforeBranch = await loadBranch(sessionFile);
const beforeToolResults = beforeBranch.filter(
(entry) => entry.type === "message" && entry.message.role === "toolResult",
);
@@ -513,7 +524,7 @@ describe("truncateOversizedToolResultsInSession", () => {
expect(result.truncated).toBe(true);
expect(result.truncatedCount).toBe(1);
const afterBranch = SessionManager.open(sessionFile).getBranch();
const afterBranch = await loadBranch(sessionFile);
const afterToolResults = afterBranch.filter(
(entry) => entry.type === "message" && entry.message.role === "toolResult",
);
@@ -540,7 +551,7 @@ describe("truncateOversizedToolResultsInSession", () => {
});
expect(result.truncated).toBe(true);
const afterBranch = SessionManager.open(sessionFile).getBranch();
const afterBranch = await loadBranch(sessionFile);
const toolResult = afterBranch.find(
(entry) => entry.type === "message" && entry.message.role === "toolResult",
);
@@ -571,7 +582,7 @@ describe("truncateOversizedToolResultsInSession", () => {
expect(result.truncated).toBe(true);
expect(result.truncatedCount).toBe(3);
const afterBranch = SessionManager.open(sessionFile).getBranch();
const afterBranch = await loadBranch(sessionFile);
const toolResults = afterBranch.filter(
(entry) => entry.type === "message" && entry.message.role === "toolResult",
);
@@ -602,7 +613,7 @@ describe("truncateOversizedToolResultsInSession", () => {
});
expect(result.truncated).toBe(true);
const afterBranch = SessionManager.open(sessionFile).getBranch();
const afterBranch = await loadBranch(sessionFile);
const toolResults = afterBranch.filter(
(entry) => entry.type === "message" && entry.message.role === "toolResult",
);