@oh-my-pi/pi-agent-core 13.3.13 → 13.4.0

package/CHANGELOG.md

@@ -2,6 +2,11 @@
 
 ## [Unreleased]
 
+## [13.4.0] - 2026-03-01
+### Added
+
+- Added `getToolChoice` option to dynamically override tool choice per LLM call
+
 ## [13.3.8] - 2026-02-28
 ### Changed
 

package/package.json

@@ -1,7 +1,7 @@
 {
 	"type": "module",
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "13.3.13",
+	"version": "13.4.0",
 	"description": "General-purpose agent with transport abstraction, state management, and attachment support",
 	"homepage": "https://github.com/can1357/oh-my-pi",
 	"author": "Can Boluk",
@@ -31,8 +31,8 @@
 		"test": "bun test"
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-ai": "13.3.13",
-		"@oh-my-pi/pi-utils": "13.3.13"
+		"@oh-my-pi/pi-ai": "13.4.0",
+		"@oh-my-pi/pi-utils": "13.4.0"
 	},
 	"devDependencies": {
 		"@sinclair/typebox": "^0.34",

package/src/agent-loop.ts

@@ -326,9 +326,11 @@ async function streamAssistantResponse(
 	const resolvedApiKey =
 		(config.getApiKey ? await config.getApiKey(config.model.provider) : undefined) || config.apiKey;
 
+	const dynamicToolChoice = config.getToolChoice?.();
 	const response = await streamFunction(config.model, llmContext, {
 		...config,
 		apiKey: resolvedApiKey,
+		toolChoice: dynamicToolChoice ?? config.toolChoice,
 		signal,
 	});
 

package/src/agent.ts

@@ -146,6 +146,8 @@ export interface AgentOptions {
 
 	/** Enable intent tracing schema injection/stripping in the harness. */
 	intentTracing?: boolean;
+	/** Dynamic tool choice override, resolved per LLM call. */
+	getToolChoice?: () => ToolChoice | undefined;
 
 	/**
 	 * Cursor exec handlers for local tool execution.
@@ -209,6 +211,7 @@ export class Agent {
 	#preferWebsockets?: boolean;
 	#transformToolCallArguments?: (args: Record<string, unknown>, toolName: string) => Record<string, unknown>;
 	#intentTracing: boolean;
+	#getToolChoice?: () => ToolChoice | undefined;
 
 	/** Buffered Cursor tool results with text length at time of call (for correct ordering) */
 	#cursorToolResultBuffer: CursorToolResultEntry[] = [];
@@ -242,6 +245,7 @@ export class Agent {
 		this.#preferWebsockets = opts.preferWebsockets;
 		this.#transformToolCallArguments = opts.transformToolCallArguments;
 		this.#intentTracing = opts.intentTracing === true;
+		this.#getToolChoice = opts.getToolChoice;
 	}
 
 	/**
@@ -709,6 +713,7 @@ export class Agent {
 			cursorOnToolResult,
 			transformToolCallArguments: this.#transformToolCallArguments,
 			intentTracing: this.#intentTracing,
+			getToolChoice: this.#getToolChoice,
 			getSteeringMessages: async () => {
 				if (skipInitialSteeringPoll) {
 					skipInitialSteeringPoll = false;

package/src/types.ts

@@ -8,6 +8,7 @@ import type {
 	streamSimple,
 	TextContent,
 	Tool,
+	ToolChoice,
 	ToolResultMessage,
 } from "@oh-my-pi/pi-ai";
 import type { Static, TSchema } from "@sinclair/typebox";
@@ -123,6 +124,12 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
 	 * then strips `_i` from arguments before executing tools.
 	 */
 	intentTracing?: boolean;
+
+	/**
+	 * Dynamic tool choice override, resolved per LLM call.
+	 * When set and returns a value, overrides the static `toolChoice`.
+	 */
+	getToolChoice?: () => ToolChoice | undefined;
 }
 
 export interface ToolCallContext {
@@ -178,19 +185,15 @@ export interface AgentState {
 	error?: string;
 }
 
-export interface AgentToolResult<T = any, TNormative extends TSchema = any> {
+export interface AgentToolResult<T = any, _TInput = unknown> {
 	// Content blocks supporting text and images
 	content: (TextContent | ImageContent)[];
 	// Details to be displayed in a UI or logged
 	details?: T;
-	/** Normative input for the tool result */
-	$normative?: Static<TNormative>;
 }
 
 // Callback for streaming tool execution updates
-export type AgentToolUpdateCallback<T = any, TNormative extends TSchema = any> = (
-	partialResult: AgentToolResult<T, TNormative>,
-) => void;
+export type AgentToolUpdateCallback<T = any, TInput = unknown> = (partialResult: AgentToolResult<T, TInput>) => void;
 
 /** Options passed to renderResult */
 export interface RenderResultOptions {
@@ -226,6 +229,8 @@ export interface AgentTool<TParameters extends TSchema = TSchema, TDetails = any
 	label: string;
 	/** If true, tool is excluded unless explicitly listed in --tools or agent's tools field */
 	hidden?: boolean;
+	/** If true, tool can stage a pending action that requires explicit resolution via the resolve tool. */
+	deferrable?: boolean;
 	/** If true, tool execution ignores abort signals (runs to completion) */
 	nonAbortable?: boolean;
 	/**