LLM/moltbot
1
0
Fork 0
mirror of https://github.com/moltbot/moltbot.git synced 2026-03-21 16:41:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(test): tighten tool result typing in context pruning tests

Browse Source
This commit is contained in:
Peter Steinberger
2026-03-03 01:18:23 +00:00
parent 5ee6ca13b7
commit 61adcea68e
1 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/agents/pi-extensions/context-pruning.test.ts
View File

@@ -1,4 +1,5 @@
import type { AgentMessage } from "@mariozechner/pi-agent-core";
import type { ToolResultMessage } from "@mariozechner/pi-ai";
import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
import { describe, expect, it } from "vitest";
import {
@@ -9,10 +10,11 @@ import {
} from "./context-pruning.js";
import { getContextPruningRuntime, setContextPruningRuntime } from "./context-pruning/runtime.js";
function toolText(msg: AgentMessage): string {
if (msg.role !== "toolResult") {
throw new Error("expected toolResult");
}
function isToolResultMessage(msg: AgentMessage): msg is ToolResultMessage {
return msg.role === "toolResult";
}
function toolText(msg: ToolResultMessage): string {
const first = msg.content.find((b) => b.type === "text");
if (!first || first.type !== "text") {
return "";
@@ -20,8 +22,10 @@ function toolText(msg: AgentMessage): string {
return first.text;
}
function findToolResult(messages: AgentMessage[], toolCallId: string): AgentMessage {
const msg = messages.find((m) => m.role === "toolResult" && m.toolCallId === toolCallId);
function findToolResult(messages: AgentMessage[], toolCallId: string): ToolResultMessage {
const msg = messages.find((m): m is ToolResultMessage => {
return isToolResultMessage(m) && m.toolCallId === toolCallId;
});
if (!msg) {
throw new Error(`missing toolResult: ${toolCallId}`);
}
@@ -32,7 +36,7 @@ function makeToolResult(params: {
toolCallId: string;
toolName: string;
text: string;
}): AgentMessage {
}): ToolResultMessage {
return {
role: "toolResult",
toolCallId: params.toolCallId,
@@ -47,7 +51,7 @@ function makeImageToolResult(params: {
toolCallId: string;
toolName: string;
text: string;
}): AgentMessage {
}): ToolResultMessage {
const base = makeToolResult(params);
return {
...base,
@@ -367,9 +371,6 @@ describe("context-pruning", () => {
const next = pruneWithAggressiveDefaults(messages);
const tool = findToolResult(next, "t1");
if (!tool || tool.role !== "toolResult") {
throw new Error("unexpected pruned message list shape");
}
expect(tool.content.some((b) => b.type === "image")).toBe(true);
expect(toolText(tool)).toContain("x".repeat(20_000));
});
@@ -387,7 +388,7 @@ describe("context-pruning", () => {
],
isError: false,
timestamp: Date.now(),
} as unknown as AgentMessage,
} as ToolResultMessage,
];
const next = pruneWithAggressiveDefaults(messages, {