fix: avoid duplicate prompt context

This commit is contained in:
Peter Steinberger
2026-01-08 00:01:40 +00:00
parent 8b4bcc6b7a
commit badc1602c8
3 changed files with 64 additions and 44 deletions

View File

@@ -1,10 +1,14 @@
import type { AgentMessage, AgentTool } from "@mariozechner/pi-agent-core";
import { SessionManager } from "@mariozechner/pi-coding-agent";
import {
buildSystemPrompt,
SessionManager,
} from "@mariozechner/pi-coding-agent";
import { Type } from "@sinclair/typebox";
import { describe, expect, it, vi } from "vitest";
import {
applyGoogleTurnOrderingFix,
buildEmbeddedSandboxInfo,
createSystemPromptAppender,
splitSdkTools,
} from "./pi-embedded-runner.js";
import type { SandboxContext } from "./sandbox.js";
@@ -105,6 +109,22 @@ describe("splitSdkTools", () => {
});
});
describe("createSystemPromptAppender", () => {
it("appends without duplicating context files", () => {
const sentinel = "CONTEXT_SENTINEL_42";
const defaultPrompt = buildSystemPrompt({
cwd: "/tmp",
contextFiles: [{ path: "/tmp/AGENTS.md", content: sentinel }],
});
const appender = createSystemPromptAppender("APPEND_SECTION");
const finalPrompt = appender(defaultPrompt);
const occurrences = finalPrompt.split(sentinel).length - 1;
expect(typeof appender).toBe("function");
expect(occurrences).toBe(1);
expect(finalPrompt).toContain("APPEND_SECTION");
});
});
describe("applyGoogleTurnOrderingFix", () => {
const makeAssistantFirst = () =>
[