@oh-my-pi/pi-agent-core 6.1.0 → 6.7.0

package/package.json

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

package/src/agent-loop.ts

@@ -377,6 +377,8 @@ async function executeToolCalls(
 	const results: ToolResultMessage[] = [];
 	let steeringMessages: AgentMessage[] | undefined;
 	const shouldInterruptImmediately = interruptMode !== "wait";
+	const toolCallInfos = toolCalls.map((call) => ({ id: call.id, name: call.name }));
+	const batchId = `${assistantMessage.timestamp ?? Date.now()}_${toolCalls[0]?.id ?? "batch"}`;
 
 	for (let index = 0; index < toolCalls.length; index++) {
 		const toolCall = toolCalls[index];
@@ -397,7 +399,14 @@ async function executeToolCalls(
 
 			const validatedArgs = validateToolArguments(tool, toolCall);
 
-			const toolContext = getToolContext ? getToolContext() : undefined;
+			const toolContext = getToolContext
+				? getToolContext({
+						batchId,
+						index,
+						total: toolCalls.length,
+						toolCalls: toolCallInfos,
+					})
+				: undefined;
 			result = await tool.execute(
 				toolCall.id,
 				validatedArgs,

package/src/agent.ts

@@ -27,6 +27,7 @@ import type {
 	AgentToolContext,
 	StreamFn,
 	ThinkingLevel,
+	ToolCallContext,
 } from "./types";
 
 /**
@@ -94,7 +95,7 @@ export interface AgentOptions {
 	 * Provides tool execution context, resolved per tool call.
 	 * Use for late-bound UI or session state access.
 	 */
-	getToolContext?: () => AgentToolContext | undefined;
+	getToolContext?: (toolCall?: ToolCallContext) => AgentToolContext | undefined;
 
 	/**
 	 * Cursor exec handlers for local tool execution.
@@ -139,7 +140,7 @@ export class Agent {
 	private _sessionId?: string;
 	private _thinkingBudgets?: ThinkingBudgets;
 	public getApiKey?: (provider: string) => Promise<string | undefined> | string | undefined;
-	private getToolContext?: () => AgentToolContext | undefined;
+	private getToolContext?: (toolCall?: ToolCallContext) => AgentToolContext | undefined;
 	private cursorExecHandlers?: CursorExecHandlers;
 	private cursorOnToolResult?: CursorToolResultHandler;
 	private runningPrompt?: Promise<void>;

package/src/types.ts

@@ -109,7 +109,14 @@ export interface AgentLoopConfig extends SimpleStreamOptions {
 	 * Provides tool execution context, resolved per tool call.
 	 * Use for late-bound UI or session state access.
 	 */
-	getToolContext?: () => AgentToolContext | undefined;
+	getToolContext?: (toolCall?: ToolCallContext) => AgentToolContext | undefined;
+}
+
+export interface ToolCallContext {
+	batchId: string;
+	index: number;
+	total: number;
+	toolCalls: Array<{ id: string; name: string }>;
 }
 
 /**
@@ -158,15 +165,19 @@ export interface AgentState {
 	error?: string;
 }
 
-export interface AgentToolResult<T> {
+export interface AgentToolResult<T, TNormative extends TSchema = any> {
 	// Content blocks supporting text and images
 	content: (TextContent | ImageContent)[];
 	// Details to be displayed in a UI or logged
-	details: T;
+	details?: T;
+	/** Normative input for the tool result */
+	$normative?: Static<TNormative>;
 }
 
 // Callback for streaming tool execution updates
-export type AgentToolUpdateCallback<T = any> = (partialResult: AgentToolResult<T>) => void;
+export type AgentToolUpdateCallback<T = any, TNormative extends TSchema = any> = (
+	partialResult: AgentToolResult<T, TNormative>,
+) => void;
 
 /** Options passed to renderResult */
 export interface RenderResultOptions {
@@ -197,15 +208,19 @@ export interface AgentTool<TParameters extends TSchema = TSchema, TDetails = any
 		toolCallId: string,
 		params: Static<TParameters>,
 		signal?: AbortSignal,
-		onUpdate?: AgentToolUpdateCallback<TDetails>,
+		onUpdate?: AgentToolUpdateCallback<TDetails, TParameters>,
 		context?: AgentToolContext,
-	) => Promise<AgentToolResult<TDetails>>;
+	) => Promise<AgentToolResult<TDetails, TParameters>>;
 
 	/** Optional custom rendering for tool call display (returns UI component) */
 	renderCall?: (args: Static<TParameters>, theme: TTheme) => unknown;
 
 	/** Optional custom rendering for tool result display (returns UI component) */
-	renderResult?: (result: AgentToolResult<TDetails>, options: RenderResultOptions, theme: TTheme) => unknown;
+	renderResult?: (
+		result: AgentToolResult<TDetails, TParameters>,
+		options: RenderResultOptions,
+		theme: TTheme,
+	) => unknown;
 }
 
 // AgentContext is like Context but uses AgentTool