@oh-my-pi/pi-agent-core 6.7.670 → 6.8.1

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "6.7.670",
+	"version": "6.8.1",
 	"description": "General-purpose agent with transport abstraction, state management, and attachment support",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -13,8 +13,9 @@
 		"test": "vitest --run"
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-ai": "6.7.670",
-		"@oh-my-pi/pi-tui": "6.7.670"
+		"@oh-my-pi/pi-ai": "6.8.1",
+		"@oh-my-pi/pi-tui": "6.8.1",
+		"@oh-my-pi/pi-utils": "6.8.1"
 	},
 	"keywords": [
 		"ai",
@@ -34,6 +35,7 @@
 		"bun": ">=1.0.0"
 	},
 	"devDependencies": {
+		"@sinclair/typebox": "^0.34.46",
 		"@types/node": "^24.3.0",
 		"vitest": "^3.2.4"
 	}

package/src/proxy.ts

@@ -14,6 +14,7 @@ import {
 	type ToolCall,
 } from "@oh-my-pi/pi-ai";
 import { parseStreamingJson } from "@oh-my-pi/pi-ai/src/utils/json-parse";
+import { readSseEvents } from "@oh-my-pi/pi-utils";
 
 // Create stream class matching ProxyMessageEventStream
 class ProxyMessageEventStream extends EventStream<AssistantMessageEvent, AssistantMessage> {
@@ -104,20 +105,19 @@ export function streamProxy(model: Model<any>, context: Context, options: ProxyS
 			timestamp: Date.now(),
 		};
 
-		let reader: ReadableStreamDefaultReader<Uint8Array> | undefined;
-
+		let response: Response | null = null;
 		const abortHandler = () => {
-			if (reader) {
-				reader.cancel("Request aborted by user").catch(() => {});
+			const body = response?.body;
+			if (body) {
+				body.cancel("Request aborted by user").catch(() => {});
 			}
 		};
-
 		if (options.signal) {
-			options.signal.addEventListener("abort", abortHandler);
+			options.signal.addEventListener("abort", abortHandler, { once: true });
 		}
 
 		try {
-			const response = await fetch(`${options.proxyUrl}/api/stream`, {
+			response = await fetch(`${options.proxyUrl}/api/stream`, {
 				method: "POST",
 				headers: {
 					Authorization: `Bearer ${options.authToken}`,
@@ -148,33 +148,17 @@ export function streamProxy(model: Model<any>, context: Context, options: ProxyS
 				throw new Error(errorMessage);
 			}
 
-			reader = response.body!.getReader() as ReadableStreamDefaultReader<Uint8Array>;
-			const decoder = new TextDecoder();
-			let buffer = "";
-
-			while (true) {
-				const { done, value } = await reader!.read();
-				if (done) break;
-
+			for await (const event of readSseEvents(response.body!)) {
 				if (options.signal?.aborted) {
 					throw new Error("Request aborted by user");
 				}
 
-				buffer += decoder.decode(value, { stream: true });
-				const lines = buffer.split("\n");
-				buffer = lines.pop() || "";
-
-				for (const line of lines) {
-					if (line.startsWith("data: ")) {
-						const data = line.slice(6).trim();
-						if (data) {
-							const proxyEvent = JSON.parse(data) as ProxyAssistantMessageEvent;
-							const event = processProxyEvent(proxyEvent, partial);
-							if (event) {
-								stream.push(event);
-							}
-						}
-					}
+				const data = event.data?.trim();
+				if (!data || data === "[DONE]") continue;
+				const proxyEvent = JSON.parse(data) as ProxyAssistantMessageEvent;
+				const parsedEvent = processProxyEvent(proxyEvent, partial);
+				if (parsedEvent) {
+					stream.push(parsedEvent);
 				}
 			}