fix: update pi-agent-core integration

CHANGELOG.md

@@ -82,6 +82,7 @@
 - macOS menu: top status line now shows pending node pairing approvals (incl. repairs).
 - CLI: avoid spurious gateway close errors after successful request/response cycles.
 - Agent runtime: clamp tool-result images to the 5MB Anthropic limit to avoid hard request rejections.
+- Agent runtime: align pi-agent-core imports and write v2 session headers so Pi session branching stays in the Clawdis sessions dir.
 - Tests: add Swift Testing coverage for camera errors and Kotest coverage for Android bridge endpoints.
 
 ## 2.0.0-beta4 — 2025-12-27

src/agents/bash-tools.ts

@@ -1,6 +1,7 @@
 import { type ChildProcessWithoutNullStreams, spawn } from "node:child_process";
 import { randomUUID } from "node:crypto";
-import type { AgentTool, AgentToolResult } from "@mariozechner/pi-ai";
+import type { AgentTool, AgentToolResult } from "@mariozechner/pi-agent-core";
+import { StringEnum } from "@mariozechner/pi-ai";
 import { Type } from "@sinclair/typebox";
 
 import {

src/agents/clawdis-tools.ts

@@ -1,8 +1,8 @@
 import crypto from "node:crypto";
 import fs from "node:fs/promises";
 
-import type { AgentTool, AgentToolResult } from "@mariozechner/pi-ai";
-import { Type } from "@sinclair/typebox";
+import type { AgentTool, AgentToolResult } from "@mariozechner/pi-agent-core";
+import { type TSchema, Type } from "@sinclair/typebox";
 import {
   browserCloseTab,
   browserFocusTab,

src/agents/pi-embedded-helpers.ts

@@ -1,8 +1,8 @@
 import fs from "node:fs/promises";
 import path from "node:path";
 
-import type { AppMessage } from "@mariozechner/pi-agent-core";
-import type { AgentToolResult, AssistantMessage } from "@mariozechner/pi-ai";
+import type { AgentMessage, AgentToolResult } from "@mariozechner/pi-agent-core";
+import type { AssistantMessage } from "@mariozechner/pi-ai";
 
 import { sanitizeContentBlocksImages } from "./tool-images.js";
 import type { WorkspaceBootstrapFile } from "./workspace.js";
@@ -22,8 +22,10 @@ export async function ensureSessionHeader(params: {
     // create
   }
   await fs.mkdir(path.dirname(file), { recursive: true });
+  const sessionVersion = 2;
   const entry = {
     type: "session",
+    version: sessionVersion,
     id: params.sessionId,
     timestamp: new Date().toISOString(),
     cwd: params.cwd,
@@ -34,12 +36,12 @@ export async function ensureSessionHeader(params: {
 type ContentBlock = AgentToolResult<unknown>["content"][number];
 
 export async function sanitizeSessionMessagesImages(
-  messages: AppMessage[],
+  messages: AgentMessage[],
   label: string,
-): Promise<AppMessage[]> {
+): Promise<AgentMessage[]> {
   // 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: AppMessage[] = [];
+  const out: AgentMessage[] = [];
   for (const msg of messages) {
     if (!msg || typeof msg !== "object") {
       out.push(msg);
@@ -48,7 +50,7 @@ export async function sanitizeSessionMessagesImages(
 
     const role = (msg as { role?: unknown }).role;
     if (role === "toolResult") {
-      const toolMsg = msg as Extract<AppMessage, { role: "toolResult" }>;
+      const toolMsg = msg as Extract<AgentMessage, { role: "toolResult" }>;
       const content = Array.isArray(toolMsg.content) ? toolMsg.content : [];
       const nextContent = (await sanitizeContentBlocksImages(
         content as ContentBlock[],
@@ -59,7 +61,7 @@ export async function sanitizeSessionMessagesImages(
     }
 
     if (role === "user") {
-      const userMsg = msg as Extract<AppMessage, { role: "user" }>;
+      const userMsg = msg as Extract<AgentMessage, { role: "user" }>;
       const content = userMsg.content;
       if (Array.isArray(content)) {
         const nextContent = (await sanitizeContentBlocksImages(

src/agents/pi-embedded-runner.ts

@@ -3,7 +3,7 @@ import fs from "node:fs/promises";
 import os from "node:os";
 import path from "node:path";
 
-import type { AppMessage, ThinkingLevel } from "@mariozechner/pi-agent-core";
+import type { AgentMessage, ThinkingLevel } from "@mariozechner/pi-agent-core";
 import {
   type Api,
   type AssistantMessage,
@@ -433,10 +433,7 @@ export async function runEmbeddedPiAgent(params: {
           tools,
         });
 
-        const sessionManager = SessionManager.open(
-          params.sessionFile,
-          agentDir,
-        );
+        const sessionManager = SessionManager.open(params.sessionFile);
         const settingsManager = SettingsManager.create(
           resolvedWorkspace,
           agentDir,
@@ -504,7 +501,7 @@ export async function runEmbeddedPiAgent(params: {
           Math.max(1, params.timeoutMs),
         );
 
-        let messagesSnapshot: AppMessage[] = [];
+        let messagesSnapshot: AgentMessage[] = [];
         let sessionIdUsed = session.sessionId;
         const onAbort = () => {
           abortRun();
@@ -545,7 +542,7 @@ export async function runEmbeddedPiAgent(params: {
         const lastAssistant = messagesSnapshot
           .slice()
           .reverse()
-          .find((m) => (m as AppMessage)?.role === "assistant") as
+          .find((m) => (m as AgentMessage)?.role === "assistant") as
           | AssistantMessage
           | undefined;
 

src/agents/pi-embedded-subscribe.ts

@@ -1,4 +1,4 @@
-import type { AgentEvent, AppMessage } from "@mariozechner/pi-agent-core";
+import type { AgentEvent, AgentMessage } from "@mariozechner/pi-agent-core";
 import type { AssistantMessage } from "@mariozechner/pi-ai";
 import type { AgentSession } from "@mariozechner/pi-coding-agent";
 
@@ -234,7 +234,7 @@ export function subscribeEmbeddedPiSession(params: {
       }
 
       if (evt.type === "message_update") {
-        const msg = (evt as AgentEvent & { message: AppMessage }).message;
+        const msg = (evt as AgentEvent & { message: AgentMessage }).message;
         if (msg?.role === "assistant") {
           const assistantEvent = (
             evt as AgentEvent & { assistantMessageEvent?: unknown }
@@ -298,7 +298,7 @@ export function subscribeEmbeddedPiSession(params: {
       }
 
       if (evt.type === "message_end") {
-        const msg = (evt as AgentEvent & { message: AppMessage }).message;
+        const msg = (evt as AgentEvent & { message: AgentMessage }).message;
         if (msg?.role === "assistant") {
           const cleaned = params.enforceFinalTag
             ? stripThinkingSegments(

src/agents/pi-tools.ts

@@ -1,4 +1,4 @@
-import type { AgentTool, AgentToolResult } from "@mariozechner/pi-ai";
+import type { AgentTool, AgentToolResult } from "@mariozechner/pi-agent-core";
 import { codingTools, readTool } from "@mariozechner/pi-coding-agent";
 import { Type } from "@sinclair/typebox";
 

src/agents/tool-images.ts

@@ -1,4 +1,4 @@
-import type { AgentToolResult } from "@mariozechner/pi-ai";
+import type { AgentToolResult } from "@mariozechner/pi-agent-core";
 
 import { getImageMetadata, resizeToJpeg } from "../media/image-ops.js";