@google/adk 2.3.0 → 2.4.0

package/dist/cjs/agents/agent_engine_client.js

@@ -0,0 +1,172 @@
+/**
+  * @license
+  * Copyright 2026 Google LLC
+  * SPDX-License-Identifier: Apache-2.0
+  */
+
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var agent_engine_client_exports = {};
+__export(agent_engine_client_exports, {
+  AgentEngineClient: () => AgentEngineClient,
+  AgentExecutionError: () => AgentExecutionError
+});
+module.exports = __toCommonJS(agent_engine_client_exports);
+var import_aiplatform = require("@google-cloud/aiplatform");
+/**
+ * @license
+ * Copyright 2026 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+const _AgentExecutionError = class _AgentExecutionError extends Error {
+  constructor(message, cause) {
+    super(message);
+    this.cause = cause;
+    this.name = "AgentExecutionError";
+  }
+};
+__name(_AgentExecutionError, "AgentExecutionError");
+let AgentExecutionError = _AgentExecutionError;
+function buildInputStruct(config) {
+  var _a;
+  const value = import_aiplatform.helpers.toValue(config);
+  return (_a = value == null ? void 0 : value.structValue) != null ? _a : void 0;
+}
+__name(buildInputStruct, "buildInputStruct");
+function parseOutput(output) {
+  return import_aiplatform.helpers.fromValue(output);
+}
+__name(parseOutput, "parseOutput");
+function parseFragment(line) {
+  const trimmed = line.trim();
+  if (!trimmed) return void 0;
+  const payload = trimmed.startsWith("data:") ? trimmed.slice(5).trim() : trimmed;
+  if (payload === "[DONE]") return void 0;
+  try {
+    return JSON.parse(payload);
+  } catch (e) {
+    throw new AgentExecutionError(
+      `Failed to parse stream fragment: ${payload}`,
+      e
+    );
+  }
+}
+__name(parseFragment, "parseFragment");
+async function* parseStream(stream) {
+  let buffer = "";
+  for await (const chunk of stream) {
+    const chunkData = chunk == null ? void 0 : chunk.data;
+    if (chunkData === void 0) {
+      if (typeof chunk === "object" && chunk !== null) {
+        yield chunk;
+      }
+      continue;
+    }
+    const dataStr = chunkData instanceof Uint8Array ? new TextDecoder().decode(chunkData) : String(chunkData);
+    buffer += dataStr;
+    const lines = buffer.split("\n");
+    buffer = lines.pop() || "";
+    for (const line of lines) {
+      const fragment = parseFragment(line);
+      if (fragment !== void 0) {
+        yield fragment;
+      }
+    }
+  }
+  const last = parseFragment(buffer);
+  if (last !== void 0) {
+    yield last;
+  }
+}
+__name(parseStream, "parseStream");
+const _AgentEngineClient = class _AgentEngineClient {
+  constructor(config) {
+    this.client = new import_aiplatform.v1.ReasoningEngineExecutionServiceClient({
+      apiEndpoint: `${config.location}-aiplatform.googleapis.com`
+    });
+    this.reasoningEnginePath = this.client.reasoningEnginePath(
+      config.project,
+      config.location,
+      config.reasoningEngineId
+    );
+  }
+  /**
+   * Creates a remote session.
+   *
+   * @returns The `create_session` payload returned by the Agent Engine. When
+   *     `config.sessionId` is omitted the engine generates one, and this is the
+   *     only place the caller can read it back from.
+   */
+  async createSession(config) {
+    try {
+      const [response] = await this.client.queryReasoningEngine({
+        name: this.reasoningEnginePath,
+        classMethod: "create_session",
+        input: buildInputStruct(config)
+      });
+      return response.output ? parseOutput(response.output) : void 0;
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      throw new AgentExecutionError(
+        `Failed to create session: ${message}`,
+        err
+      );
+    }
+  }
+  async query(config) {
+    try {
+      const [response] = await this.client.queryReasoningEngine({
+        name: this.reasoningEnginePath,
+        classMethod: "query",
+        input: buildInputStruct(config)
+      });
+      return response.output ? parseOutput(response.output) : void 0;
+    } catch (err) {
+      const message = err instanceof Error ? err.message : String(err);
+      throw new AgentExecutionError(`Failed to execute query: ${message}`, err);
+    }
+  }
+  async *streamQuery(config) {
+    try {
+      const stream = this.client.streamQueryReasoningEngine({
+        name: this.reasoningEnginePath,
+        classMethod: "query",
+        input: buildInputStruct(config)
+      });
+      yield* parseStream(stream);
+    } catch (err) {
+      if (err instanceof AgentExecutionError) {
+        throw err;
+      }
+      const message = err instanceof Error ? err.message : String(err);
+      throw new AgentExecutionError(
+        `Failed to execute stream query: ${message}`,
+        err
+      );
+    }
+  }
+};
+__name(_AgentEngineClient, "AgentEngineClient");
+let AgentEngineClient = _AgentEngineClient;
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  AgentEngineClient,
+  AgentExecutionError
+});

package/dist/cjs/agents/base_agent.js

@@ -259,6 +259,22 @@ const _BaseAgent = class _BaseAgent extends (_b = import_base_node.BaseNode, _a
       agent: this
     });
   }
+  /**
+   * The resolved beforeAgentCallback field as a list of SingleAgentCallback.
+   *
+   * This method is only for use by Agent Development Kit.
+   */
+  get canonicalBeforeAgentCallbacks() {
+    return this.beforeAgentCallback;
+  }
+  /**
+   * The resolved afterAgentCallback field as a list of SingleAgentCallback.
+   *
+   * This method is only for use by Agent Development Kit.
+   */
+  get canonicalAfterAgentCallbacks() {
+    return this.afterAgentCallback;
+  }
   /**
    * Runs the before agent callback if it exists.
    *
@@ -268,11 +284,11 @@ const _BaseAgent = class _BaseAgent extends (_b = import_base_node.BaseNode, _a
    */
   async handleBeforeAgentCallback(invocationContext) {
     var _a2;
-    if (this.beforeAgentCallback.length === 0) {
+    if (this.canonicalBeforeAgentCallbacks.length === 0) {
       return void 0;
     }
     const callbackContext = new import_context.Context({ invocationContext });
-    for (const callback of this.beforeAgentCallback) {
+    for (const callback of this.canonicalBeforeAgentCallbacks) {
       const content = await callback(callbackContext);
       if ((_a2 = invocationContext.abortSignal) == null ? void 0 : _a2.aborted) {
         return;
@@ -307,11 +323,11 @@ const _BaseAgent = class _BaseAgent extends (_b = import_base_node.BaseNode, _a
    */
   async handleAfterAgentCallback(invocationContext) {
     var _a2;
-    if (this.afterAgentCallback.length === 0) {
+    if (this.canonicalAfterAgentCallbacks.length === 0) {
       return void 0;
     }
     const callbackContext = new import_context.Context({ invocationContext });
-    for (const callback of this.afterAgentCallback) {
+    for (const callback of this.canonicalAfterAgentCallbacks) {
       const content = await callback(callbackContext);
       if ((_a2 = invocationContext.abortSignal) == null ? void 0 : _a2.aborted) {
         return;

package/dist/cjs/agents/functions.js

@@ -172,21 +172,49 @@ async function callToolAsync(tool, args, toolContext) {
       (0, import_tracing.traceToolCall)({
         tool,
         args,
+        span,
         functionResponseEvent: buildResponseEvent(
           tool,
           result,
           toolContext,
           toolContext.invocationContext
         ),
+        errorType: getToolResponseErrorType(result),
         invocationContext: toolContext.invocationContext
       });
       return result;
+    } catch (error) {
+      (0, import_tracing.traceToolCall)({
+        tool,
+        args,
+        span,
+        error,
+        invocationContext: toolContext.invocationContext
+      });
+      throw error;
     } finally {
       span.end();
     }
   });
 }
 __name(callToolAsync, "callToolAsync");
+function getToolResponseErrorType(result) {
+  var _a2;
+  if (result === null || typeof result !== "object" || Array.isArray(result)) {
+    return void 0;
+  }
+  try {
+    if (!Object.hasOwn(result, "error")) {
+      return void 0;
+    }
+    const response = result;
+    const errorType = (_a2 = response.errorType) != null ? _a2 : response.error_type;
+    return typeof errorType === "string" ? errorType : void 0;
+  } catch {
+    return void 0;
+  }
+}
+__name(getToolResponseErrorType, "getToolResponseErrorType");
 function buildResponseEvent(tool, functionResult, toolContext, invocationContext) {
   let responseResult;
   if (typeof functionResult !== "object" || functionResult == null) {

package/dist/cjs/agents/instructions.js

@@ -29,6 +29,7 @@ __export(instructions_exports, {
 });
 module.exports = __toCommonJS(instructions_exports);
 var import_state = require("../sessions/state.js");
+var import_load_artifacts_tool = require("../tools/load_artifacts_tool.js");
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -62,7 +63,7 @@ async function resolveKey(key, isOptional, rawMatch, readonlyContext) {
       }
       throw new Error(`Artifact ${fileName} not found.`);
     }
-    return formatValue(artifact, true);
+    return formatValue((0, import_load_artifacts_tool.asSafePartForLlm)(artifact, fileName), true);
   }
   if (isValidStateName(key)) {
     if (key in invocationContext.session.state) {

package/dist/cjs/agents/llm_agent.js

@@ -43,6 +43,7 @@ var import_base_llm = require("../models/base_llm.js");
 var import_registry = require("../models/registry.js");
 var import_built_in_planner = require("../planners/built_in_planner.js");
 var import_base_tool = require("../tools/base_tool.js");
+var import_base_toolset = require("../tools/base_toolset.js");
 var import_logger = require("../utils/logger.js");
 var import_output_schema_utils = require("../utils/output_schema_utils.js");
 var import_context = require("./context.js");
@@ -658,7 +659,7 @@ const _LlmAgent = class _LlmAgent extends (_b = import_base_agent.BaseAgent, _a
     }
   }
   async *runLivePreprocess(invocationContext, llmRequest) {
-    var _a2, _b2;
+    var _a2, _b2, _c;
     for (const processor of this.requestProcessors) {
       for await (const event of processor.runAsync(
         invocationContext,
@@ -672,6 +673,12 @@ const _LlmAgent = class _LlmAgent extends (_b = import_base_agent.BaseAgent, _a
     }
     for (const toolUnion of this.tools) {
       const toolContext = new import_context.Context({ invocationContext });
+      if ((0, import_base_toolset.isBaseToolset)(toolUnion)) {
+        await toolUnion.processLlmRequest(toolContext, llmRequest);
+        if ((_b2 = invocationContext.abortSignal) == null ? void 0 : _b2.aborted) {
+          return;
+        }
+      }
       const tools = (await convertToolUnionToTools(
         toolUnion,
         new import_readonly_context.ReadonlyContext(invocationContext)
@@ -680,7 +687,7 @@ const _LlmAgent = class _LlmAgent extends (_b = import_base_agent.BaseAgent, _a
       );
       for (const tool of tools) {
         await tool.processLlmRequest({ toolContext, llmRequest });
-        if ((_b2 = invocationContext.abortSignal) == null ? void 0 : _b2.aborted) {
+        if ((_c = invocationContext.abortSignal) == null ? void 0 : _c.aborted) {
           return;
         }
       }
@@ -903,7 +910,7 @@ const _LlmAgent = class _LlmAgent extends (_b = import_base_agent.BaseAgent, _a
     yield functionResponseEvent;
   }
   async *runOneStepAsync(invocationContext) {
-    var _a2, _b2, _c;
+    var _a2, _b2, _c, _d;
     const llmRequest = {
       contents: [],
       toolsDict: {},
@@ -923,7 +930,7 @@ const _LlmAgent = class _LlmAgent extends (_b = import_base_agent.BaseAgent, _a
     const allTools = [...this.tools];
     if (this.mode === "task") {
       allTools.push(this.finishTaskTool);
-    } else if (this.outputSchema && allTools.length > 0 && !(0, import_output_schema_utils.canUseOutputSchemaWithTools)(this.canonicalModel.model)) {
+    } else if (this.outputSchema && allTools.length > 0 && !(0, import_output_schema_utils.canUseOutputSchemaWithTools)(this.canonicalModel)) {
       const setModelResponseTool = new import_function_tool.FunctionTool({
         name: "set_model_response",
         description: "Call this tool to submit your final response conforming to the output schema. Use this tool only when you have collected all the information and are ready to return the final answer.",
@@ -947,6 +954,12 @@ const _LlmAgent = class _LlmAgent extends (_b = import_base_agent.BaseAgent, _a
         invocationContext,
         eventActions: modelResponseEvent.actions
       });
+      if ((0, import_base_toolset.isBaseToolset)(toolUnion)) {
+        await toolUnion.processLlmRequest(toolContext, llmRequest);
+        if ((_b2 = invocationContext.abortSignal) == null ? void 0 : _b2.aborted) {
+          return;
+        }
+      }
       const tools = (await convertToolUnionToTools(
         toolUnion,
         new import_readonly_context.ReadonlyContext(invocationContext)
@@ -955,12 +968,12 @@ const _LlmAgent = class _LlmAgent extends (_b = import_base_agent.BaseAgent, _a
       });
       for (const tool of tools) {
         await tool.processLlmRequest({ toolContext, llmRequest });
-        if ((_b2 = invocationContext.abortSignal) == null ? void 0 : _b2.aborted) {
+        if ((_c = invocationContext.abortSignal) == null ? void 0 : _c.aborted) {
           return;
         }
       }
     }
-    if (invocationContext.endInvocation || ((_c = invocationContext.abortSignal) == null ? void 0 : _c.aborted)) {
+    if (invocationContext.endInvocation || ((_d = invocationContext.abortSignal) == null ? void 0 : _d.aborted)) {
       return;
     }
     const span = import_tracing.tracer.startSpan("call_llm");

package/dist/cjs/agents/processors/basic_llm_request_processor.js

@@ -55,7 +55,7 @@ const _BasicLlmRequestProcessor = class _BasicLlmRequestProcessor extends import
     }
     llmRequest.model = agent.canonicalModel.model;
     llmRequest.config = { ...(_a = agent.generateContentConfig) != null ? _a : {} };
-    if (agent.outputSchema && agent.mode !== "task" && (!((_b = agent.tools) == null ? void 0 : _b.length) || (0, import_output_schema_utils.canUseOutputSchemaWithTools)(agent.canonicalModel.model))) {
+    if (agent.outputSchema && agent.mode !== "task" && (!((_b = agent.tools) == null ? void 0 : _b.length) || (0, import_output_schema_utils.canUseOutputSchemaWithTools)(agent.canonicalModel))) {
       (0, import_llm_request.setOutputSchema)(llmRequest, agent.outputSchema);
     }
     if (invocationContext.runConfig) {

package/dist/cjs/agents/processors/code_execution_request_processor.js

@@ -83,6 +83,14 @@ const DATA_FILE_UTIL_MAP = {
 const DATA_FILE_HELPER_LIB = `
 import pandas as pd
 
+def crop(s: str, max_chars: int = 64) -> str:
+  """Crops a string to max_chars characters."""
+  if len(s) <= max_chars:
+    return s
+  if max_chars >= 3:
+    return s[:max_chars - 3] + '...'
+  return s[:max_chars]
+
 def explore_df(df: pd.DataFrame) -> None:
   """Prints some information about a pandas DataFrame."""
 

package/dist/cjs/agents/processors/content_processor_utils.js

@@ -56,7 +56,7 @@ function removeClientFunctionCallId(content) {
   }
 }
 __name(removeClientFunctionCallId, "removeClientFunctionCallId");
-function getContents(events, agentName, currentBranch, currentIsolationScope) {
+function getContents(events, agentName, currentBranch, currentIsolationScope, preserveFunctionCallIds = false) {
   const filteredEvents = [];
   for (const event of events) {
     if (!(0, import_compaction_utils.isEventVisibleInIsolationScope)(event, currentIsolationScope)) {
@@ -69,16 +69,23 @@ function getContents(events, agentName, currentBranch, currentIsolationScope) {
     if (!shouldIncludeEventInContext(event, currentBranch, currentIsolationScope)) {
       continue;
     }
-    filteredEvents.push(
-      isEventFromAnotherAgent(agentName, event) ? convertForeignEvent(event) : event
-    );
+    if (isEventFromAnotherAgent(agentName, event)) {
+      const converted = convertForeignEvent(event);
+      if (converted) {
+        filteredEvents.push(converted);
+      }
+    } else {
+      filteredEvents.push(event);
+    }
   }
   let resultEvents = rearrangeEventsForLatestFunctionResponse(filteredEvents);
   resultEvents = rearrangeEventsForAsyncFunctionResponsesInHistory(resultEvents);
   const contents = [];
   for (const event of resultEvents) {
     const content = (0, import_lodash_es.cloneDeep)(event.content);
-    removeClientFunctionCallId(content);
+    if (!preserveFunctionCallIds) {
+      removeClientFunctionCallId(content);
+    }
     contents.push(content);
   }
   return contents;
@@ -111,18 +118,20 @@ function isRequestInputEvent(event) {
   return false;
 }
 __name(isRequestInputEvent, "isRequestInputEvent");
-function getCurrentTurnContents(events, agentName, currentBranch, currentIsolationScope) {
+function getCurrentTurnContents(events, agentName, currentBranch, currentIsolationScope, preserveFunctionCallIds = false) {
+  var _a;
   for (let i = events.length - 1; i >= 0; i--) {
     const event = events[i];
     if (!shouldIncludeEventInContext(event, currentBranch, currentIsolationScope)) {
       continue;
     }
-    if (event.author === "user" || isEventFromAnotherAgent(agentName, event)) {
+    if (event.author === "user" || isEventFromAnotherAgent(agentName, event) && ((_a = event.content.parts) == null ? void 0 : _a.some(isVisibleForeignPart))) {
       return getContents(
         events.slice(turnStart(events, i)),
         agentName,
         currentBranch,
-        currentIsolationScope
+        currentIsolationScope,
+        preserveFunctionCallIds
       );
     }
   }
@@ -187,6 +196,13 @@ function isEventFromAnotherAgent(agentName, event) {
   return !!agentName && event.author !== agentName && event.author !== "user";
 }
 __name(isEventFromAnotherAgent, "isEventFromAnotherAgent");
+function isVisibleForeignPart(part) {
+  if (part.functionCall || part.functionResponse) {
+    return true;
+  }
+  return !part.thought && part.text !== "";
+}
+__name(isVisibleForeignPart, "isVisibleForeignPart");
 function convertForeignEvent(event) {
   var _a, _b, _c, _d, _e, _f;
   if (!((_b = (_a = event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.length)) {
@@ -201,7 +217,10 @@ function convertForeignEvent(event) {
     ]
   };
   for (const part of event.content.parts) {
-    if (part.text && !part.thought) {
+    if (!isVisibleForeignPart(part)) {
+      continue;
+    }
+    if (part.text) {
       (_c = content.parts) == null ? void 0 : _c.push({
         text: `[${event.author}] said: ${part.text}`
       });
@@ -218,9 +237,18 @@ function convertForeignEvent(event) {
         )}`
       });
     } else {
-      (_f = content.parts) == null ? void 0 : _f.push((0, import_lodash_es.cloneDeep)(part));
+      const {
+        thought: _thought,
+        thoughtSignature: _thoughtSignature,
+        ...rest
+      } = part;
+      (_f = content.parts) == null ? void 0 : _f.push((0, import_lodash_es.cloneDeep)(rest));
     }
   }
+  const preambleOnly = content.parts.length === 1;
+  if (preambleOnly) {
+    return void 0;
+  }
   return (0, import_event.createEvent)({
     invocationId: event.invocationId,
     author: "user",

package/dist/cjs/agents/processors/content_request_processor.js

@@ -30,6 +30,7 @@ __export(content_request_processor_exports, {
 });
 module.exports = __toCommonJS(content_request_processor_exports);
 var import_compaction_utils = require("../../context/compaction_utils.js");
+var import_base_llm = require("../../models/base_llm.js");
 var import_llm_agent = require("../llm_agent.js");
 var import_content_processor_utils = require("./content_processor_utils.js");
 /**
@@ -37,6 +38,31 @@ var import_content_processor_utils = require("./content_processor_utils.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+const CLAUDE_MODEL_SYMBOL = /* @__PURE__ */ Symbol.for("google.adk.claudeModel");
+const CLAUDE_MODEL_NAME_PATTERN = /^claude-/i;
+function isClaudeModelInstance(model) {
+  return CLAUDE_MODEL_SYMBOL in model && model[CLAUDE_MODEL_SYMBOL] === true || CLAUDE_MODEL_NAME_PATTERN.test(model.model);
+}
+__name(isClaudeModelInstance, "isClaudeModelInstance");
+function shouldPreserveFunctionCallIds(agent, llmRequest) {
+  if (llmRequest.model && CLAUDE_MODEL_NAME_PATTERN.test(llmRequest.model)) {
+    return true;
+  }
+  let current = agent;
+  while (current) {
+    if ((0, import_llm_agent.isLlmAgent)(current)) {
+      if ((0, import_base_llm.isBaseLlm)(current.model)) {
+        return isClaudeModelInstance(current.model);
+      }
+      if (typeof current.model === "string" && current.model.length > 0) {
+        return CLAUDE_MODEL_NAME_PATTERN.test(current.model);
+      }
+    }
+    current = current.parentAgent;
+  }
+  return false;
+}
+__name(shouldPreserveFunctionCallIds, "shouldPreserveFunctionCallIds");
 const _ContentRequestProcessor = class _ContentRequestProcessor {
   /**
    * Fills {@link LlmRequest.contents} based on the session event history and
@@ -55,19 +81,25 @@ const _ContentRequestProcessor = class _ContentRequestProcessor {
       invocationContext.session.events,
       invocationContext.isolationScope
     );
+    const preserveFunctionCallIds = shouldPreserveFunctionCallIds(
+      agent,
+      llmRequest
+    );
     if (agent.includeContents === "default") {
       llmRequest.contents = (0, import_content_processor_utils.getContents)(
         events,
         agent.name,
         invocationContext.branch,
-        invocationContext.isolationScope
+        invocationContext.isolationScope,
+        preserveFunctionCallIds
       );
     } else {
       llmRequest.contents = (0, import_content_processor_utils.getCurrentTurnContents)(
         events,
         agent.name,
         invocationContext.branch,
-        invocationContext.isolationScope
+        invocationContext.isolationScope,
+        preserveFunctionCallIds
       );
     }
     return;

package/dist/cjs/agents/processors/instructions_llm_request_processor.js

@@ -78,7 +78,7 @@ const _InstructionsLlmRequestProcessor = class _InstructionsLlmRequestProcessor
       }
       (0, import_llm_request.appendInstructions)(llmRequest, [instructionWithState]);
     }
-    if (agent.outputSchema && ((_a = agent.tools) == null ? void 0 : _a.length) && !(0, import_output_schema_utils.canUseOutputSchemaWithTools)(agent.canonicalModel.model)) {
+    if (agent.outputSchema && ((_a = agent.tools) == null ? void 0 : _a.length) && !(0, import_output_schema_utils.canUseOutputSchemaWithTools)(agent.canonicalModel)) {
       (0, import_llm_request.appendInstructions)(llmRequest, [
         'To output the final result, you must call the "set_model_response" function with the appropriate values. Do not output anything else.'
       ]);

package/dist/cjs/agents/processors/nl_planning_processor.js

@@ -31,6 +31,7 @@ __export(nl_planning_processor_exports, {
   NlPlanningResponseProcessor: () => NlPlanningResponseProcessor
 });
 module.exports = __toCommonJS(nl_planning_processor_exports);
+var import_event = require("../../events/event.js");
 var import_llm_request = require("../../models/llm_request.js");
 var import_base_planner = require("../../planners/base_planner.js");
 var import_built_in_planner = require("../../planners/built_in_planner.js");
@@ -85,20 +86,28 @@ const _NlPlanningRequestProcessor = class _NlPlanningRequestProcessor extends im
 __name(_NlPlanningRequestProcessor, "NlPlanningRequestProcessor");
 let NlPlanningRequestProcessor = _NlPlanningRequestProcessor;
 const _NlPlanningResponseProcessor = class _NlPlanningResponseProcessor extends import_base_llm_processor.BaseLlmResponseProcessor {
-  // eslint-disable-next-line require-yield -- this processor only rewrites the response and has no event to emit
   async *runAsync(invocationContext, llmResponse) {
-    var _a, _b;
+    var _a, _b, _c;
     const planner = getPlanner(invocationContext);
     if (!planner || !((_b = (_a = llmResponse.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.length)) {
       return;
     }
+    const callbackContext = new import_context.Context({ invocationContext });
     const processedParts = planner.processPlanningResponse(
-      new import_context.Context({ invocationContext }),
+      callbackContext,
       llmResponse.content.parts
     );
     if (processedParts !== void 0) {
       llmResponse.content = { ...llmResponse.content, parts: processedParts };
     }
+    if (callbackContext.state.hasDelta()) {
+      yield (0, import_event.createEvent)({
+        invocationId: invocationContext.invocationId,
+        author: (_c = invocationContext.agent) == null ? void 0 : _c.name,
+        branch: invocationContext.branch,
+        actions: callbackContext.eventActions
+      });
+    }
   }
 };
 __name(_NlPlanningResponseProcessor, "NlPlanningResponseProcessor");

package/dist/cjs/cli/utils/logs.js

@@ -36,8 +36,8 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var logs_exports = {};
 __export(logs_exports, {
   LOGGING_FORMAT: () => LOGGING_FORMAT,
-  logToStderr: () => logToStderr,
-  logToTmpFolder: () => logToTmpFolder
+  logToTmpFolder: () => logToTmpFolder,
+  setupAdkLogger: () => setupAdkLogger
 });
 module.exports = __toCommonJS(logs_exports);
 var fs = __toESM(require("node:fs"), 1);
@@ -99,7 +99,7 @@ function createWriterLogger(initialLevel, writeLine) {
   };
 }
 __name(createWriterLogger, "createWriterLogger");
-function logToStderr(level = import_logger.LogLevel.INFO) {
+function setupAdkLogger(level = import_logger.LogLevel.INFO) {
   (0, import_logger.setLogLevel)(level);
   (0, import_logger.setLogger)(
     createWriterLogger(level, (line) => {
@@ -108,7 +108,7 @@ function logToStderr(level = import_logger.LogLevel.INFO) {
     })
   );
 }
-__name(logToStderr, "logToStderr");
+__name(setupAdkLogger, "setupAdkLogger");
 function logToTmpFolder(options = {}) {
   const {
     level = import_logger.LogLevel.INFO,
@@ -145,6 +145,6 @@ __name(logToTmpFolder, "logToTmpFolder");
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   LOGGING_FORMAT,
-  logToStderr,
-  logToTmpFolder
+  logToTmpFolder,
+  setupAdkLogger
 });