@google/adk 0.2.4 → 0.3.0

package/dist/cjs/agents/llm_agent.js

@@ -26,9 +26,11 @@ var llm_agent_exports = {};
 __export(llm_agent_exports, {
   LlmAgent: () => LlmAgent,
   REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR: () => REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR,
+  isLlmAgent: () => isLlmAgent,
   responseProcessor: () => responseProcessor
 });
 module.exports = __toCommonJS(llm_agent_exports);
+var import_api = require("@opentelemetry/api");
 var import_lodash_es = require("lodash-es");
 var import_zod = require("zod");
 var import_base_code_executor = require("../code_executors/base_code_executor.js");
@@ -47,6 +49,7 @@ var import_tool_confirmation = require("../tools/tool_confirmation.js");
 var import_tool_context = require("../tools/tool_context.js");
 var import_env_aware_utils = require("../utils/env_aware_utils.js");
 var import_logger = require("../utils/logger.js");
+var import_tracing = require("../telemetry/tracing.js");
 var import_base_agent = require("./base_agent.js");
 var import_base_llm_processor = require("./base_llm_processor.js");
 var import_callback_context = require("./callback_context.js");
@@ -60,22 +63,24 @@ var import_run_config = require("./run_config.js");
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+var _a, _b;
 const ADK_AGENT_NAME_LABEL_KEY = "adk_agent_name";
-async function convertToolUnionToTools(toolUnion, context) {
+async function convertToolUnionToTools(toolUnion, context2) {
   if (toolUnion instanceof import_base_tool.BaseTool) {
     return [toolUnion];
   }
-  return await toolUnion.getTools(context);
+  return await toolUnion.getTools(context2);
 }
 class BasicLlmRequestProcessor extends import_base_llm_processor.BaseLlmRequestProcessor {
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
-    var _a;
+    var _a2;
     const agent = invocationContext.agent;
-    if (!(agent instanceof LlmAgent)) {
+    if (!isLlmAgent(agent)) {
       return;
     }
     llmRequest.model = agent.canonicalModel.model;
-    llmRequest.config = { ...(_a = agent.generateContentConfig) != null ? _a : {} };
+    llmRequest.config = { ...(_a2 = agent.generateContentConfig) != null ? _a2 : {} };
     if (agent.outputSchema) {
       (0, import_llm_request.setOutputSchema)(llmRequest, agent.outputSchema);
     }
@@ -92,6 +97,7 @@ class BasicLlmRequestProcessor extends import_base_llm_processor.BaseLlmRequestP
 }
 const BASIC_LLM_REQUEST_PROCESSOR = new BasicLlmRequestProcessor();
 class IdentityLlmRequestProcessor extends import_base_llm_processor.BaseLlmRequestProcessor {
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     const si = [`You are an agent. Your internal name is "${agent.name}".`];
@@ -106,13 +112,14 @@ class InstructionsLlmRequestProcessor extends import_base_llm_processor.BaseLlmR
   /**
    * Handles instructions and global instructions for LLM flow.
    */
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
     if (!(agent instanceof LlmAgent) || !(agent.rootAgent instanceof LlmAgent)) {
       return;
     }
     const rootAgent = agent.rootAgent;
-    if (rootAgent instanceof LlmAgent && rootAgent.globalInstruction) {
+    if (isLlmAgent(rootAgent) && rootAgent.globalInstruction) {
       const { instruction, requireStateInjection } = await rootAgent.canonicalGlobalInstruction(
         new import_readonly_context.ReadonlyContext(invocationContext)
       );
@@ -142,9 +149,10 @@ class InstructionsLlmRequestProcessor extends import_base_llm_processor.BaseLlmR
 }
 const INSTRUCTIONS_LLM_REQUEST_PROCESSOR = new InstructionsLlmRequestProcessor();
 class ContentRequestProcessor {
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     const agent = invocationContext.agent;
-    if (!agent || !(agent instanceof LlmAgent)) {
+    if (!agent || !isLlmAgent(agent)) {
       return;
     }
     if (agent.includeContents === "default") {
@@ -183,6 +191,7 @@ class AgentTransferLlmRequestProcessor extends import_base_llm_processor.BaseLlm
       }
     });
   }
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
     if (!(invocationContext.agent instanceof LlmAgent)) {
       return;
@@ -232,7 +241,7 @@ to your parent agent.
   getTransferTargets(agent) {
     const targets = [];
     targets.push(...agent.subAgents);
-    if (!agent.parentAgent || !(agent.parentAgent instanceof LlmAgent)) {
+    if (!agent.parentAgent || !isLlmAgent(agent.parentAgent)) {
       return targets;
     }
     if (!agent.disallowTransferToParent) {
@@ -251,9 +260,9 @@ to your parent agent.
 const AGENT_TRANSFER_LLM_REQUEST_PROCESSOR = new AgentTransferLlmRequestProcessor();
 class RequestConfirmationLlmRequestProcessor extends import_base_llm_processor.BaseLlmRequestProcessor {
   /** Handles tool confirmation information to build the LLM request. */
-  async *runAsync(invocationContext, llmRequest) {
+  async *runAsync(invocationContext) {
     const agent = invocationContext.agent;
-    if (!(agent instanceof LlmAgent)) {
+    if (!isLlmAgent(agent)) {
       return;
     }
     const events = invocationContext.session.events;
@@ -279,7 +288,9 @@ class RequestConfirmationLlmRequestProcessor extends import_base_llm_processor.B
         foundConfirmation = true;
         let toolConfirmation = null;
         if (functionResponse.response && Object.keys(functionResponse.response).length === 1 && "response" in functionResponse.response) {
-          toolConfirmation = JSON.parse(functionResponse.response["response"]);
+          toolConfirmation = JSON.parse(
+            functionResponse.response["response"]
+          );
         } else if (functionResponse.response) {
           toolConfirmation = new import_tool_confirmation.ToolConfirmation({
             hint: functionResponse.response["hint"],
@@ -343,8 +354,12 @@ class RequestConfirmationLlmRequestProcessor extends import_base_llm_processor.B
       if (Object.keys(toolsToResumeWithConfirmation).length === 0) {
         continue;
       }
-      const toolsList = await agent.canonicalTools(new import_readonly_context.ReadonlyContext(invocationContext));
-      const toolsDict = Object.fromEntries(toolsList.map((tool) => [tool.name, tool]));
+      const toolsList = await agent.canonicalTools(
+        new import_readonly_context.ReadonlyContext(invocationContext)
+      );
+      const toolsDict = Object.fromEntries(
+        toolsList.map((tool) => [tool.name, tool])
+      );
       const functionResponseEvent = await (0, import_functions.handleFunctionCallList)({
         invocationContext,
         functionCalls: Object.values(toolsToResumeWithArgs),
@@ -373,12 +388,12 @@ class CodeExecutionRequestProcessor extends import_base_llm_processor.BaseLlmReq
     for await (const event of runPreProcessor(invocationContext, llmRequest)) {
       yield event;
     }
-    if (!(invocationContext.agent.codeExecutor instanceof import_base_code_executor.BaseCodeExecutor)) {
+    if (!(0, import_base_code_executor.isBaseCodeExecutor)(invocationContext.agent.codeExecutor)) {
       return;
     }
     for (const content of llmRequest.contents) {
       const delimeters = invocationContext.agent.codeExecutor.codeBlockDelimiters.length ? invocationContext.agent.codeExecutor.codeBlockDelimiters[0] : ["", ""];
-      const codeExecutionParts = (0, import_code_execution_utils.convertCodeExecutionParts)(
+      (0, import_code_execution_utils.convertCodeExecutionParts)(
         content,
         delimeters,
         invocationContext.agent.codeExecutor.executionResultDelimiters
@@ -442,7 +457,10 @@ class CodeExecutionResponseProcessor {
     if (llmResponse.partial) {
       return;
     }
-    for await (const event of runPostProcessor(invocationContext, llmResponse)) {
+    for await (const event of runPostProcessor(
+      invocationContext,
+      llmResponse
+    )) {
       yield event;
     }
   }
@@ -450,27 +468,36 @@ class CodeExecutionResponseProcessor {
 const responseProcessor = new CodeExecutionResponseProcessor();
 async function* runPreProcessor(invocationContext, llmRequest) {
   const agent = invocationContext.agent;
-  if (!(agent instanceof LlmAgent)) {
+  if (!isLlmAgent(agent)) {
     return;
   }
   const codeExecutor = agent.codeExecutor;
-  if (!codeExecutor || !(codeExecutor instanceof import_base_code_executor.BaseCodeExecutor)) {
+  if (!codeExecutor || !(0, import_base_code_executor.isBaseCodeExecutor)(codeExecutor)) {
     return;
   }
-  if (codeExecutor instanceof import_built_in_code_executor.BuiltInCodeExecutor) {
+  if ((0, import_built_in_code_executor.isBuiltInCodeExecutor)(codeExecutor)) {
     codeExecutor.processLlmRequest(llmRequest);
     return;
   }
   if (!codeExecutor.optimizeDataFile) {
     return;
   }
-  const codeExecutorContext = new import_code_executor_context.CodeExecutorContext(new import_state.State(invocationContext.session.state));
+  const codeExecutorContext = new import_code_executor_context.CodeExecutorContext(
+    new import_state.State(invocationContext.session.state)
+  );
   if (codeExecutorContext.getErrorCount(invocationContext.invocationId) >= codeExecutor.errorRetryAttempts) {
     return;
   }
-  const allInputFiles = extractAndReplaceInlineFiles(codeExecutorContext, llmRequest);
-  const processedFileNames = new Set(codeExecutorContext.getProcessedFileNames());
-  const filesToProcess = allInputFiles.filter((f) => !processedFileNames.has(f.name));
+  const allInputFiles = extractAndReplaceInlineFiles(
+    codeExecutorContext,
+    llmRequest
+  );
+  const processedFileNames = new Set(
+    codeExecutorContext.getProcessedFileNames()
+  );
+  const filesToProcess = allInputFiles.filter(
+    (f) => !processedFileNames.has(f.name)
+  );
   for (const file of filesToProcess) {
     const codeStr = getDataFilePreprocessingCode(file);
     if (!codeStr) {
@@ -490,7 +517,10 @@ async function* runPreProcessor(invocationContext, llmRequest) {
       branch: invocationContext.branch,
       content: codeContent
     });
-    const executionId = getOrSetExecutionId(invocationContext, codeExecutorContext);
+    const executionId = getOrSetExecutionId(
+      invocationContext,
+      codeExecutorContext
+    );
     const codeExecutionResult = await codeExecutor.executeCode({
       invocationContext,
       codeExecutionInput: {
@@ -517,20 +547,22 @@ async function* runPreProcessor(invocationContext, llmRequest) {
 }
 async function* runPostProcessor(invocationContext, llmResponse) {
   const agent = invocationContext.agent;
-  if (!(agent instanceof LlmAgent)) {
+  if (!isLlmAgent(agent)) {
     return;
   }
   const codeExecutor = agent.codeExecutor;
-  if (!codeExecutor || !(codeExecutor instanceof import_base_code_executor.BaseCodeExecutor)) {
+  if (!codeExecutor || !(0, import_base_code_executor.isBaseCodeExecutor)(codeExecutor)) {
     return;
   }
   if (!llmResponse || !llmResponse.content) {
     return;
   }
-  if (codeExecutor instanceof import_built_in_code_executor.BuiltInCodeExecutor) {
+  if ((0, import_built_in_code_executor.isBuiltInCodeExecutor)(codeExecutor)) {
     return;
   }
-  const codeExecutorContext = new import_code_executor_context.CodeExecutorContext(new import_state.State(invocationContext.session.state));
+  const codeExecutorContext = new import_code_executor_context.CodeExecutorContext(
+    new import_state.State(invocationContext.session.state)
+  );
   if (codeExecutorContext.getErrorCount(invocationContext.invocationId) >= codeExecutor.errorRetryAttempts) {
     return;
   }
@@ -548,7 +580,10 @@ async function* runPostProcessor(invocationContext, llmResponse) {
     branch: invocationContext.branch,
     content: responseContent
   });
-  const executionId = getOrSetExecutionId(invocationContext, codeExecutorContext);
+  const executionId = getOrSetExecutionId(
+    invocationContext,
+    codeExecutorContext
+  );
   const codeExecutionResult = await codeExecutor.executeCode({
     invocationContext,
     codeExecutionInput: {
@@ -568,10 +603,10 @@ async function* runPostProcessor(invocationContext, llmResponse) {
     codeExecutorContext,
     codeExecutionResult
   );
-  llmResponse.content = null;
+  llmResponse.content = void 0;
 }
 function extractAndReplaceInlineFiles(codeExecutorContext, llmRequest) {
-  var _a;
+  var _a2;
   const allInputFiles = codeExecutorContext.getInputFiles();
   const savedFileNames = new Set(allInputFiles.map((f) => f.name));
   for (let i = 0; i < llmRequest.contents.length; i++) {
@@ -581,7 +616,7 @@ function extractAndReplaceInlineFiles(codeExecutorContext, llmRequest) {
     }
     for (let j = 0; j < content.parts.length; j++) {
       const part = content.parts[j];
-      const mimeType = (_a = part.inlineData) == null ? void 0 : _a.mimeType;
+      const mimeType = (_a2 = part.inlineData) == null ? void 0 : _a2.mimeType;
       if (!mimeType || !part.inlineData || !DATA_FILE_UTIL_MAP[mimeType]) {
         continue;
       }
@@ -603,9 +638,9 @@ Available file: \`${fileName}\`
   return allInputFiles;
 }
 function getOrSetExecutionId(invocationContext, codeExecutorContext) {
-  var _a;
+  var _a2;
   const agent = invocationContext.agent;
-  if (!(agent instanceof LlmAgent) || !((_a = agent.codeExecutor) == null ? void 0 : _a.stateful)) {
+  if (!isLlmAgent(agent) || !((_a2 = agent.codeExecutor) == null ? void 0 : _a2.stateful)) {
     return void 0;
   }
   let executionId = codeExecutorContext.getExecutionId();
@@ -623,7 +658,9 @@ async function postProcessCodeExecutionResult(invocationContext, codeExecutorCon
     role: "model",
     parts: [(0, import_code_execution_utils.buildCodeExecutionResultPart)(codeExecutionResult)]
   };
-  const eventActions = (0, import_event_actions.createEventActions)({ stateDelta: codeExecutorContext.getStateDelta() });
+  const eventActions = (0, import_event_actions.createEventActions)({
+    stateDelta: codeExecutorContext.getStateDelta()
+  });
   if (codeExecutionResult.stderr) {
     codeExecutorContext.incrementErrorCount(invocationContext.invocationId);
   } else {
@@ -662,10 +699,7 @@ function getDataFilePreprocessingCode(file) {
     return void 0;
   }
   const varName = getNormalizedFileName(file.name);
-  const loaderCode = DATA_FILE_UTIL_MAP[file.mimeType].loaderCodeTemplate.replace(
-    "{filename}",
-    file.name
-  );
+  const loaderCode = DATA_FILE_UTIL_MAP[file.mimeType].loaderCodeTemplate.replace("{filename}", file.name);
   return `
 ${DATA_FILE_HELPER_LIB}
 
@@ -677,13 +711,19 @@ explore_df(${varName})
 `;
 }
 const CODE_EXECUTION_REQUEST_PROCESSOR = new CodeExecutionRequestProcessor();
-class LlmAgent extends import_base_agent.BaseAgent {
+const LLM_AGENT_SIGNATURE_SYMBOL = Symbol.for("google.adk.llmAgent");
+function isLlmAgent(obj) {
+  return typeof obj === "object" && obj !== null && LLM_AGENT_SIGNATURE_SYMBOL in obj && obj[LLM_AGENT_SIGNATURE_SYMBOL] === true;
+}
+class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATURE_SYMBOL, _b) {
   constructor(config) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    var _a2, _b2, _c, _d, _e, _f, _g, _h, _i;
     super(config);
+    /** A unique symbol to identify ADK LLM agent class. */
+    this[_a] = true;
     this.model = config.model;
-    this.instruction = (_a = config.instruction) != null ? _a : "";
-    this.globalInstruction = (_b = config.globalInstruction) != null ? _b : "";
+    this.instruction = (_a2 = config.instruction) != null ? _a2 : "";
+    this.globalInstruction = (_b2 = config.globalInstruction) != null ? _b2 : "";
     this.tools = (_c = config.tools) != null ? _c : [];
     this.generateContentConfig = config.generateContentConfig;
     this.disallowTransferToParent = (_d = config.disallowTransferToParent) != null ? _d : false;
@@ -761,7 +801,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
     }
     let ancestorAgent = this.parentAgent;
     while (ancestorAgent) {
-      if (ancestorAgent instanceof LlmAgent) {
+      if (isLlmAgent(ancestorAgent)) {
         return ancestorAgent.canonicalModel;
       }
       ancestorAgent = ancestorAgent.parentAgent;
@@ -769,47 +809,50 @@ class LlmAgent extends import_base_agent.BaseAgent {
     throw new Error(`No model found for ${this.name}.`);
   }
   /**
-   * The resolved self.instruction field to construct instruction for this
+   * The resolved instruction field to construct instruction for this
    * agent.
    *
    * This method is only for use by Agent Development Kit.
    * @param context The context to retrieve the session state.
-   * @returns The resolved self.instruction field.
+   * @returns The resolved instruction field.
    */
-  async canonicalInstruction(context) {
+  async canonicalInstruction(context2) {
     if (typeof this.instruction === "string") {
       return { instruction: this.instruction, requireStateInjection: true };
     }
     return {
-      instruction: await this.instruction(context),
+      instruction: await this.instruction(context2),
       requireStateInjection: false
     };
   }
   /**
-   * The resolved self.instruction field to construct global instruction.
+   * The resolved globalInstruction field to construct global instruction.
    *
    * This method is only for use by Agent Development Kit.
    * @param context The context to retrieve the session state.
-   * @returns The resolved self.global_instruction field.
+   * @returns The resolved globalInstruction field.
    */
-  async canonicalGlobalInstruction(context) {
+  async canonicalGlobalInstruction(context2) {
     if (typeof this.globalInstruction === "string") {
-      return { instruction: this.globalInstruction, requireStateInjection: true };
+      return {
+        instruction: this.globalInstruction,
+        requireStateInjection: true
+      };
     }
     return {
-      instruction: await this.globalInstruction(context),
+      instruction: await this.globalInstruction(context2),
       requireStateInjection: false
     };
   }
   /**
-   * The resolved self.tools field as a list of BaseTool based on the context.
+   * The resolved tools field as a list of BaseTool based on the context.
    *
    * This method is only for use by Agent Development Kit.
    */
-  async canonicalTools(context) {
+  async canonicalTools(context2) {
     const resolvedTools = [];
     for (const toolUnion of this.tools) {
-      const tools = await convertToolUnionToTools(toolUnion, context);
+      const tools = await convertToolUnionToTools(toolUnion, context2);
       resolvedTools.push(...tools);
     }
     return resolvedTools;
@@ -830,7 +873,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
     return [callback];
   }
   /**
-   * The resolved self.before_model_callback field as a list of
+   * The resolved beforeModelCallback field as a list of
    * SingleBeforeModelCallback.
    *
    * This method is only for use by Agent Development Kit.
@@ -839,7 +882,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
     return LlmAgent.normalizeCallbackArray(this.beforeModelCallback);
   }
   /**
-   * The resolved self.after_model_callback field as a list of
+   * The resolved afterModelCallback field as a list of
    * SingleAfterModelCallback.
    *
    * This method is only for use by Agent Development Kit.
@@ -848,7 +891,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
     return LlmAgent.normalizeCallbackArray(this.afterModelCallback);
   }
   /**
-   * The resolved self.before_tool_callback field as a list of
+   * The resolved beforeToolCallback field as a list of
    * BeforeToolCallback.
    *
    * This method is only for use by Agent Development Kit.
@@ -857,7 +900,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
     return LlmAgent.normalizeCallbackArray(this.beforeToolCallback);
   }
   /**
-   * The resolved self.after_tool_callback field as a list of AfterToolCallback.
+   * The resolved afterToolCallback field as a list of AfterToolCallback.
    *
    * This method is only for use by Agent Development Kit.
    */
@@ -874,7 +917,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
    * @param event The event to process.
    */
   maybeSaveOutputToState(event) {
-    var _a, _b;
+    var _a2, _b2;
     if (event.author !== this.name) {
       import_logger.logger.debug(
         `Skipping output save for agent ${this.name}: event authored by ${event.author}`
@@ -893,7 +936,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
       );
       return;
     }
-    if (!((_b = (_a = event.content) == null ? void 0 : _a.parts) == null ? void 0 : _b.length)) {
+    if (!((_b2 = (_a2 = event.content) == null ? void 0 : _a2.parts) == null ? void 0 : _b2.length)) {
       import_logger.logger.debug(
         `Skipping output save for agent ${this.name}: event content is empty`
       );
@@ -913,10 +956,10 @@ class LlmAgent extends import_base_agent.BaseAgent {
     }
     event.actions.stateDelta[this.outputKey] = result;
   }
-  async *runAsyncImpl(context) {
+  async *runAsyncImpl(context2) {
     while (true) {
       let lastEvent = void 0;
-      for await (const event of this.runOneStepAsync(context)) {
+      for await (const event of this.runOneStepAsync(context2)) {
         lastEvent = event;
         this.maybeSaveOutputToState(event);
         yield event;
@@ -930,19 +973,20 @@ class LlmAgent extends import_base_agent.BaseAgent {
       }
     }
   }
-  async *runLiveImpl(context) {
-    for await (const event of this.runLiveFlow(context)) {
+  async *runLiveImpl(context2) {
+    for await (const event of this.runLiveFlow(context2)) {
       this.maybeSaveOutputToState(event);
       yield event;
     }
-    if (context.endInvocation) {
+    if (context2.endInvocation) {
       return;
     }
   }
   // --------------------------------------------------------------------------
   // #START LlmFlow Logic
   // --------------------------------------------------------------------------
-  async *runLiveFlow(invocationContext) {
+  // eslint-disable-next-line require-yield
+  async *runLiveFlow(_invocationContext) {
     await Promise.resolve();
     throw new Error("LlmAgent.runLiveFlow not implemented");
   }
@@ -953,7 +997,10 @@ class LlmAgent extends import_base_agent.BaseAgent {
       liveConnectConfig: {}
     };
     for (const processor of this.requestProcessors) {
-      for await (const event of processor.runAsync(invocationContext, llmRequest)) {
+      for await (const event of processor.runAsync(
+        invocationContext,
+        llmRequest
+      )) {
         yield event;
       }
     }
@@ -975,27 +1022,39 @@ class LlmAgent extends import_base_agent.BaseAgent {
       author: this.name,
       branch: invocationContext.branch
     });
-    for await (const llmResponse of this.callLlmAsync(
-      invocationContext,
-      llmRequest,
-      modelResponseEvent
-    )) {
-      for await (const event of this.postprocess(
-        invocationContext,
-        llmRequest,
-        llmResponse,
-        modelResponseEvent
-      )) {
-        modelResponseEvent.id = (0, import_event.createNewEventId)();
-        modelResponseEvent.timestamp = (/* @__PURE__ */ new Date()).getTime();
-        yield event;
+    const span = import_tracing.tracer.startSpan("call_llm");
+    const ctx = import_api.trace.setSpan(import_api.context.active(), span);
+    yield* (0, import_tracing.runAsyncGeneratorWithOtelContext)(
+      ctx,
+      this,
+      async function* () {
+        for await (const llmResponse of this.callLlmAsync(
+          invocationContext,
+          llmRequest,
+          modelResponseEvent
+        )) {
+          for await (const event of this.postprocess(
+            invocationContext,
+            llmRequest,
+            llmResponse,
+            modelResponseEvent
+          )) {
+            modelResponseEvent.id = (0, import_event.createNewEventId)();
+            modelResponseEvent.timestamp = (/* @__PURE__ */ new Date()).getTime();
+            yield event;
+          }
+        }
       }
-    }
+    );
+    span.end();
   }
   async *postprocess(invocationContext, llmRequest, llmResponse, modelResponseEvent) {
-    var _a;
+    var _a2;
     for (const processor of this.responseProcessors) {
-      for await (const event of processor.runAsync(invocationContext, llmResponse)) {
+      for await (const event of processor.runAsync(
+        invocationContext,
+        llmResponse
+      )) {
         yield event;
       }
     }
@@ -1016,7 +1075,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
       }
     }
     yield mergedEvent;
-    if (!((_a = (0, import_event.getFunctionCalls)(mergedEvent)) == null ? void 0 : _a.length)) {
+    if (!((_a2 = (0, import_event.getFunctionCalls)(mergedEvent)) == null ? void 0 : _a2.length)) {
       return;
     }
     const functionResponseEvent = await (0, import_functions.handleFunctionCallsAsync)({
@@ -1029,7 +1088,10 @@ class LlmAgent extends import_base_agent.BaseAgent {
     if (!functionResponseEvent) {
       return;
     }
-    const authEvent = (0, import_functions.generateAuthEvent)(invocationContext, functionResponseEvent);
+    const authEvent = (0, import_functions.generateAuthEvent)(
+      invocationContext,
+      functionResponseEvent
+    );
     if (authEvent) {
       yield authEvent;
     }
@@ -1071,7 +1133,7 @@ class LlmAgent extends import_base_agent.BaseAgent {
     return agentToRun;
   }
   async *callLlmAsync(invocationContext, llmRequest, modelResponseEvent) {
-    var _a, _b, _c, _d, _e;
+    var _a2, _b2, _c, _d, _e;
     const beforeModelResponse = await this.handleBeforeModelCallback(
       invocationContext,
       llmRequest,
@@ -1081,8 +1143,8 @@ class LlmAgent extends import_base_agent.BaseAgent {
       yield beforeModelResponse;
       return;
     }
-    (_a = llmRequest.config) != null ? _a : llmRequest.config = {};
-    (_c = (_b = llmRequest.config).labels) != null ? _c : _b.labels = {};
+    (_a2 = llmRequest.config) != null ? _a2 : llmRequest.config = {};
+    (_c = (_b2 = llmRequest.config).labels) != null ? _c : _b2.labels = {};
     if (!llmRequest.config.labels[ADK_AGENT_NAME_LABEL_KEY]) {
       llmRequest.config.labels[ADK_AGENT_NAME_LABEL_KEY] = this.name;
     }
@@ -1102,6 +1164,12 @@ class LlmAgent extends import_base_agent.BaseAgent {
         llmRequest,
         modelResponseEvent
       )) {
+        (0, import_tracing.traceCallLlm)({
+          invocationContext,
+          eventId: modelResponseEvent.id,
+          llmRequest,
+          llmResponse
+        });
         const alteredLlmResponse = await this.handleAfterModelCallback(
           invocationContext,
           llmResponse,
@@ -1112,17 +1180,22 @@ class LlmAgent extends import_base_agent.BaseAgent {
     }
   }
   async handleBeforeModelCallback(invocationContext, llmRequest, modelResponseEvent) {
-    const callbackContext = new import_callback_context.CallbackContext(
-      { invocationContext, eventActions: modelResponseEvent.actions }
-    );
-    const beforeModelCallbackResponse = await invocationContext.pluginManager.runBeforeModelCallback(
-      { callbackContext, llmRequest }
-    );
+    const callbackContext = new import_callback_context.CallbackContext({
+      invocationContext,
+      eventActions: modelResponseEvent.actions
+    });
+    const beforeModelCallbackResponse = await invocationContext.pluginManager.runBeforeModelCallback({
+      callbackContext,
+      llmRequest
+    });
     if (beforeModelCallbackResponse) {
       return beforeModelCallbackResponse;
     }
     for (const callback of this.canonicalBeforeModelCallbacks) {
-      const callbackResponse = await callback({ context: callbackContext, request: llmRequest });
+      const callbackResponse = await callback({
+        context: callbackContext,
+        request: llmRequest
+      });
       if (callbackResponse) {
         return callbackResponse;
       }
@@ -1130,17 +1203,22 @@ class LlmAgent extends import_base_agent.BaseAgent {
     return void 0;
   }
   async handleAfterModelCallback(invocationContext, llmResponse, modelResponseEvent) {
-    const callbackContext = new import_callback_context.CallbackContext(
-      { invocationContext, eventActions: modelResponseEvent.actions }
-    );
-    const afterModelCallbackResponse = await invocationContext.pluginManager.runAfterModelCallback(
-      { callbackContext, llmResponse }
-    );
+    const callbackContext = new import_callback_context.CallbackContext({
+      invocationContext,
+      eventActions: modelResponseEvent.actions
+    });
+    const afterModelCallbackResponse = await invocationContext.pluginManager.runAfterModelCallback({
+      callbackContext,
+      llmResponse
+    });
     if (afterModelCallbackResponse) {
       return afterModelCallbackResponse;
     }
     for (const callback of this.canonicalAfterModelCallbacks) {
-      const callbackResponse = await callback({ context: callbackContext, response: llmResponse });
+      const callbackResponse = await callback({
+        context: callbackContext,
+        response: llmResponse
+      });
       if (callbackResponse) {
         return callbackResponse;
       }
@@ -1153,9 +1231,10 @@ class LlmAgent extends import_base_agent.BaseAgent {
         yield response;
       }
     } catch (modelError) {
-      const callbackContext = new import_callback_context.CallbackContext(
-        { invocationContext, eventActions: modelResponseEvent.actions }
-      );
+      const callbackContext = new import_callback_context.CallbackContext({
+        invocationContext,
+        eventActions: modelResponseEvent.actions
+      });
       if (modelError instanceof Error) {
         const onModelErrorCallbackResponse = await invocationContext.pluginManager.runOnModelErrorCallback({
           callbackContext,
@@ -1188,5 +1267,6 @@ class LlmAgent extends import_base_agent.BaseAgent {
 0 && (module.exports = {
   LlmAgent,
   REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR,
+  isLlmAgent,
   responseProcessor
 });