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,4 +1,4 @@
import type { AgentEvent, AgentMessage } from "@mariozechner/pi-agent-core";
import type { AgentEvent, AppMessage } 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: AgentMessage }).message;
const msg = (evt as AgentEvent & { message: AppMessage }).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: AgentMessage }).message;
const msg = (evt as AgentEvent & { message: AppMessage }).message;
if (msg?.role === "assistant") {
const cleaned = params.enforceFinalTag
? stripThinkingSegments(