@oh-my-pi/pi-agent-core 3.3.1337 → 3.5.1337

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "3.3.1337",
+	"version": "3.5.1337",
 	"description": "General-purpose agent with transport abstraction, state management, and attachment support",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -13,8 +13,8 @@
 		"test": "vitest --run"
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-ai": "3.3.1337",
-		"@oh-my-pi/pi-tui": "3.3.1337"
+		"@oh-my-pi/pi-ai": "3.5.1337",
+		"@oh-my-pi/pi-tui": "3.5.1337"
 	},
 	"keywords": [
 		"ai",

package/src/agent-loop.ts

@@ -225,6 +225,11 @@ async function streamAssistantResponse(
 	let addedPartial = false;
 
 	for await (const event of response) {
+		// Check abort early - allows TTSR and other abort sources to break immediately
+		if (signal?.aborted) {
+			break;
+		}
+
 		switch (event.type) {
 			case "start":
 				partialMessage = event.partial;
@@ -268,6 +273,22 @@ async function streamAssistantResponse(
 				return finalMessage;
 			}
 		}
+
+		// Check abort after processing - allows handlers to abort mid-stream
+		if (signal?.aborted) {
+			break;
+		}
+	}
+
+	// If we broke out due to abort, return an aborted message
+	if (signal?.aborted && partialMessage) {
+		const abortedMessage: AssistantMessage = {
+			...partialMessage,
+			stopReason: "aborted",
+		};
+		context.messages[context.messages.length - 1] = abortedMessage;
+		stream.push({ type: "message_end", message: abortedMessage });
+		return abortedMessage;
 	}
 
 	return await response.result();

package/src/agent.ts

@@ -174,6 +174,14 @@ export class Agent {
 		this._state.messages = [];
 	}
 
+	/** Remove and return the last message from the message list */
+	popMessage(): AgentMessage | undefined {
+		if (this._state.messages.length === 0) return undefined;
+		const popped = this._state.messages[this._state.messages.length - 1];
+		this._state.messages = this._state.messages.slice(0, -1);
+		return popped;
+	}
+
 	abort() {
 		this.abortController?.abort();
 	}