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

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-agent-core",
-	"version": "6.1.0",
+	"version": "6.2.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.2.0",
+		"@oh-my-pi/pi-tui": "6.2.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 }>;
 }
 
 /**