fix: update pi-agent-core integration

This commit is contained in:
Peter Steinberger
2026-01-01 16:46:40 +00:00
parent 4e00edf8a7
commit f0f5acfa42
8 changed files with 23 additions and 22 deletions

View File

@@ -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(