fix: keep pi-ai tool types for published sdk

This commit is contained in:
Peter Steinberger
2026-01-01 17:01:56 +00:00
parent 3eb3f38adf
commit 956db9c182
8 changed files with 18 additions and 21 deletions

View File

@@ -1,11 +1,8 @@
import fs from "node:fs/promises";
import path from "node:path";
import type {
AgentMessage,
AgentToolResult,
} from "@mariozechner/pi-agent-core";
import type { AssistantMessage } from "@mariozechner/pi-ai";
import type { AppMessage } from "@mariozechner/pi-agent-core";
import type { AgentToolResult, AssistantMessage } from "@mariozechner/pi-ai";
import { sanitizeContentBlocksImages } from "./tool-images.js";
import type { WorkspaceBootstrapFile } from "./workspace.js";
@@ -39,12 +36,12 @@ export async function ensureSessionHeader(params: {
type ContentBlock = AgentToolResult<unknown>["content"][number];
export async function sanitizeSessionMessagesImages(
messages: AgentMessage[],
messages: AppMessage[],
label: string,
): Promise<AgentMessage[]> {
): Promise<AppMessage[]> {
// We sanitize historical session messages because Anthropic can reject a request
// if the transcript contains oversized base64 images (see MAX_IMAGE_DIMENSION_PX).
const out: AgentMessage[] = [];
const out: AppMessage[] = [];
for (const msg of messages) {
if (!msg || typeof msg !== "object") {
out.push(msg);
@@ -53,7 +50,7 @@ export async function sanitizeSessionMessagesImages(
const role = (msg as { role?: unknown }).role;
if (role === "toolResult") {
const toolMsg = msg as Extract<AgentMessage, { role: "toolResult" }>;
const toolMsg = msg as Extract<AppMessage, { role: "toolResult" }>;
const content = Array.isArray(toolMsg.content) ? toolMsg.content : [];
const nextContent = (await sanitizeContentBlocksImages(
content as ContentBlock[],
@@ -64,7 +61,7 @@ export async function sanitizeSessionMessagesImages(
}
if (role === "user") {
const userMsg = msg as Extract<AgentMessage, { role: "user" }>;
const userMsg = msg as Extract<AppMessage, { role: "user" }>;
const content = userMsg.content;
if (Array.isArray(content)) {
const nextContent = (await sanitizeContentBlocksImages(