@oh-my-pi/pi-agent-core 3.32.0 → 3.34.0

package/README.md

@@ -1,6 +1,6 @@
 # @oh-my-pi/pi-agent
 
-Stateful agent with tool execution and event streaming. Built on `@mariozechner/pi-ai`.
+Stateful agent with tool execution and event streaming. Built on `@oh-my-pi/pi-ai`.
 
 ## Installation
 
@@ -12,7 +12,7 @@ npm install @oh-my-pi/pi-agent
 
 ```typescript
 import { Agent } from "@oh-my-pi/pi-agent";
-import { getModel } from "@mariozechner/pi-ai";
+import { getModel } from "@oh-my-pi/pi-ai";
 
 const agent = new Agent({
 	initialState: {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "3.32.0",
+	"version": "3.34.0",
 	"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": {
-		"@mariozechner/pi-ai": "^0.37.8",
-		"@oh-my-pi/pi-tui": "3.32.0"
+		"@oh-my-pi/pi-ai": "3.34.0",
+		"@oh-my-pi/pi-tui": "3.34.0"
 	},
 	"keywords": [
 		"ai",

package/src/agent-loop.ts

@@ -10,7 +10,7 @@ import {
 	streamSimple,
 	type ToolResultMessage,
 	validateToolArguments,
-} from "@mariozechner/pi-ai";
+} from "@oh-my-pi/pi-ai";
 import type {
 	AgentContext,
 	AgentEvent,
@@ -98,6 +98,32 @@ function createAgentStream(): EventStream<AgentEvent, AgentMessage[]> {
 	);
 }
 
+function normalizeMessagesForProvider(
+	messages: Context["messages"],
+	model: AgentLoopConfig["model"],
+): Context["messages"] {
+	if (model.provider !== "cerebras") {
+		return messages;
+	}
+
+	let changed = false;
+	const normalized = messages.map((message) => {
+		if (message.role !== "assistant" || !Array.isArray(message.content)) {
+			return message;
+		}
+
+		const filtered = message.content.filter((block) => block.type !== "thinking");
+		if (filtered.length === message.content.length) {
+			return message;
+		}
+
+		changed = true;
+		return { ...message, content: filtered };
+	});
+
+	return changed ? normalized : messages;
+}
+
 /**
  * Main loop logic shared by agentLoop and agentLoopContinue.
  */
@@ -218,11 +244,12 @@ async function streamAssistantResponse(
 
 	// Convert to LLM-compatible messages (AgentMessage[] → Message[])
 	const llmMessages = await config.convertToLlm(messages);
+	const normalizedMessages = normalizeMessagesForProvider(llmMessages, config.model);
 
 	// Build LLM context
 	const llmContext: Context = {
 		systemPrompt: context.systemPrompt,
-		messages: llmMessages,
+		messages: normalizedMessages,
 		tools: context.tools,
 	};
 
@@ -244,8 +271,9 @@ async function streamAssistantResponse(
 	for await (const event of response) {
 		// Check for abort signal before processing each event
 		if (signal?.aborted) {
+			const errorMessage = "Request was aborted";
 			const abortedMessage: AssistantMessage = partialMessage
-				? { ...partialMessage, stopReason: "aborted" }
+				? { ...partialMessage, stopReason: "aborted", errorMessage }
 				: {
 						role: "assistant",
 						content: [],
@@ -261,6 +289,7 @@ async function streamAssistantResponse(
 							cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
 						},
 						stopReason: "aborted",
+						errorMessage,
 						timestamp: Date.now(),
 					};
 			if (addedPartial) {

package/src/agent.ts

@@ -3,14 +3,7 @@
  * No transport abstraction - calls streamSimple via the loop.
  */
 
-import {
-	getModel,
-	type ImageContent,
-	type Message,
-	type Model,
-	streamSimple,
-	type TextContent,
-} from "@mariozechner/pi-ai";
+import { getModel, type ImageContent, type Message, type Model, streamSimple, type TextContent } from "@oh-my-pi/pi-ai";
 import { agentLoop, agentLoopContinue } from "./agent-loop";
 import type {
 	AgentContext,

package/src/proxy.ts

@@ -12,8 +12,8 @@ import {
 	type SimpleStreamOptions,
 	type StopReason,
 	type ToolCall,
-} from "@mariozechner/pi-ai";
-import { parseStreamingJson } from "@mariozechner/pi-ai/dist/utils/json-parse.js";
+} from "@oh-my-pi/pi-ai";
+import { parseStreamingJson } from "@oh-my-pi/pi-ai/src/utils/json-parse";
 
 // Create stream class matching ProxyMessageEventStream
 class ProxyMessageEventStream extends EventStream<AssistantMessageEvent, AssistantMessage> {

package/src/types.ts

@@ -8,7 +8,7 @@ import type {
 	TextContent,
 	Tool,
 	ToolResultMessage,
-} from "@mariozechner/pi-ai";
+} from "@oh-my-pi/pi-ai";
 import type { Static, TSchema } from "@sinclair/typebox";
 
 /** Stream function - can return sync or Promise for async config lookup */