can1357
779ca4872b
- Removed Prettier configuration files (.prettierignore and .prettierrc) and migrated formatting to Biome. - Updated Biome configuration from version 2.3.11 to 2.3.12 and changed arrowParentheses rule from 'always' to 'asNeeded'. - Pinned @biomejs/biome dependency to exact version 2.3.12 in package.json and bun.lock. - Applied consistent arrow function formatting across 489 files by removing unnecessary parentheses around single parameters. - Removed blank lines after comment blocks and reorganized imports for consistency across the codebase.
53 lines
1.5 KiB
TypeScript
53 lines
1.5 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import type { AssistantMessage, ToolCall } from "@oh-my-pi/pi-ai";
|
|
import { SessionManager } from "@oh-my-pi/pi-coding-agent/session/session-manager";
|
|
|
|
function buildAssistantMessage(content: AssistantMessage["content"]): AssistantMessage {
|
|
return {
|
|
role: "assistant",
|
|
content,
|
|
api: "anthropic-messages",
|
|
provider: "anthropic",
|
|
model: "test",
|
|
usage: {
|
|
input: 1,
|
|
output: 1,
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
totalTokens: 2,
|
|
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
},
|
|
stopReason: "stop",
|
|
timestamp: 1,
|
|
};
|
|
}
|
|
|
|
describe("SessionManager tool call rewrite", () => {
|
|
it("rewrites assistant toolCall arguments in context", async () => {
|
|
const session = SessionManager.inMemory();
|
|
const toolCallId = "toolu_rewrite_1";
|
|
|
|
const toolCall = {
|
|
type: "toolCall",
|
|
id: toolCallId,
|
|
name: "edit",
|
|
arguments: { path: "file.ts", op: "update", diff: "@@\n-old\n+new" },
|
|
} satisfies ToolCall;
|
|
|
|
session.appendMessage(buildAssistantMessage([toolCall]));
|
|
|
|
const updated = await session.rewriteAssistantToolCallArgs(toolCallId, {
|
|
path: "file.ts",
|
|
op: "update",
|
|
diff: "@@\n-old\n+newer",
|
|
});
|
|
|
|
expect(updated).toBe(true);
|
|
|
|
const ctx = session.buildSessionContext();
|
|
const assistant = ctx.messages.find(m => m.role === "assistant") as AssistantMessage;
|
|
const updatedCall = assistant.content.find(b => b.type === "toolCall") as ToolCall;
|
|
expect(updatedCall.arguments).toEqual({ path: "file.ts", op: "update", diff: "@@\n-old\n+newer" });
|
|
});
|
|
});
|