@oh-my-pi/pi-agent-core 3.37.1 → 4.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "3.37.1",
+	"version": "4.0.1",
 	"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.37.1",
-		"@oh-my-pi/pi-tui": "3.37.1"
+		"@oh-my-pi/pi-ai": "4.0.1",
+		"@oh-my-pi/pi-tui": "4.0.1"
 	},
 	"keywords": [
 		"ai",

package/src/agent.ts

@@ -3,7 +3,15 @@
  * No transport abstraction - calls streamSimple via the loop.
  */
 
-import { getModel, type ImageContent, type Message, type Model, streamSimple, type TextContent } from "@oh-my-pi/pi-ai";
+import {
+	getModel,
+	type ImageContent,
+	type Message,
+	type Model,
+	streamSimple,
+	type TextContent,
+	type ThinkingBudgets,
+} from "@oh-my-pi/pi-ai";
 import { agentLoop, agentLoopContinue } from "./agent-loop";
 import type {
 	AgentContext,
@@ -73,6 +81,11 @@ export interface AgentOptions {
 	 */
 	getApiKey?: (provider: string) => Promise<string | undefined> | string | undefined;
 
+	/**
+	 * Custom token budgets for thinking levels (token-based providers only).
+	 */
+	thinkingBudgets?: ThinkingBudgets;
+
 	/**
 	 * Provides tool execution context, resolved per tool call.
 	 * Use for late-bound UI or session state access.
@@ -104,6 +117,7 @@ export class Agent {
 	private interruptMode: "immediate" | "wait";
 	public streamFn: StreamFn;
 	private _sessionId?: string;
+	private _thinkingBudgets?: ThinkingBudgets;
 	public getApiKey?: (provider: string) => Promise<string | undefined> | string | undefined;
 	private getToolContext?: () => AgentToolContext | undefined;
 	private runningPrompt?: Promise<void>;
@@ -118,6 +132,7 @@ export class Agent {
 		this.interruptMode = opts.interruptMode || "immediate";
 		this.streamFn = opts.streamFn || streamSimple;
 		this._sessionId = opts.sessionId;
+		this._thinkingBudgets = opts.thinkingBudgets;
 		this.getApiKey = opts.getApiKey;
 		this.getToolContext = opts.getToolContext;
 	}
@@ -137,6 +152,20 @@ export class Agent {
 		this._sessionId = value;
 	}
 
+	/**
+	 * Get the current thinking budgets.
+	 */
+	get thinkingBudgets(): ThinkingBudgets | undefined {
+		return this._thinkingBudgets;
+	}
+
+	/**
+	 * Set custom thinking budgets for token-based providers.
+	 */
+	set thinkingBudgets(value: ThinkingBudgets | undefined) {
+		this._thinkingBudgets = value;
+	}
+
 	get state(): AgentState {
 		return this._state;
 	}
@@ -236,6 +265,22 @@ export class Agent {
 		this.followUpQueue = [];
 	}
 
+	/**
+	 * Remove and return the last steering message from the queue (LIFO).
+	 * Used by dequeue keybinding.
+	 */
+	popLastSteer(): AgentMessage | undefined {
+		return this.steeringQueue.pop();
+	}
+
+	/**
+	 * Remove and return the last follow-up message from the queue (LIFO).
+	 * Used by dequeue keybinding.
+	 */
+	popLastFollowUp(): AgentMessage | undefined {
+		return this.followUpQueue.pop();
+	}
+
 	clearMessages() {
 		this._state.messages = [];
 	}
@@ -342,6 +387,7 @@ export class Agent {
 			reasoning,
 			interruptMode: this.interruptMode,
 			sessionId: this._sessionId,
+			thinkingBudgets: this._thinkingBudgets,
 			convertToLlm: this.convertToLlm,
 			transformContext: this.transformContext,
 			getApiKey: this.getApiKey,