@google/adk 0.2.5 → 0.4.0

package/dist/cjs/agents/llm_agent.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -30,6 +30,7 @@ __export(llm_agent_exports, {
   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");
@@ -48,6 +49,8 @@ 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_simple_zod_to_json = require("../utils/simple_zod_to_json.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");
@@ -63,13 +66,14 @@ var import_run_config = require("./run_config.js");
  */
 var _a, _b;
 const ADK_AGENT_NAME_LABEL_KEY = "adk_agent_name";
-async function convertToolUnionToTools(toolUnion, context) {
-  if (toolUnion instanceof import_base_tool.BaseTool) {
+async function convertToolUnionToTools(toolUnion, context2) {
+  if ((0, import_base_tool.isBaseTool)(toolUnion)) {
     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 _a2;
     const agent = invocationContext.agent;
@@ -94,6 +98,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}".`];
@@ -108,9 +113,10 @@ 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)) {
+    if (!isLlmAgent(agent) || !isLlmAgent(agent.rootAgent)) {
       return;
     }
     const rootAgent = agent.rootAgent;
@@ -144,6 +150,7 @@ 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 || !isLlmAgent(agent)) {
@@ -185,8 +192,9 @@ class AgentTransferLlmRequestProcessor extends import_base_llm_processor.BaseLlm
       }
     });
   }
+  // eslint-disable-next-line require-yield
   async *runAsync(invocationContext, llmRequest) {
-    if (!(invocationContext.agent instanceof LlmAgent)) {
+    if (!isLlmAgent(invocationContext.agent)) {
       return;
     }
     const transferTargets = this.getTransferTargets(invocationContext.agent);
@@ -253,7 +261,7 @@ 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 (!isLlmAgent(agent)) {
       return;
@@ -372,7 +380,7 @@ class RequestConfirmationLlmRequestProcessor extends import_base_llm_processor.B
 const REQUEST_CONFIRMATION_LLM_REQUEST_PROCESSOR = new RequestConfirmationLlmRequestProcessor();
 class CodeExecutionRequestProcessor extends import_base_llm_processor.BaseLlmRequestProcessor {
   async *runAsync(invocationContext, llmRequest) {
-    if (!(invocationContext.agent instanceof LlmAgent)) {
+    if (!isLlmAgent(invocationContext.agent)) {
       return;
     }
     if (!invocationContext.agent.codeExecutor) {
@@ -386,7 +394,7 @@ class CodeExecutionRequestProcessor extends import_base_llm_processor.BaseLlmReq
     }
     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
@@ -596,7 +604,7 @@ async function* runPostProcessor(invocationContext, llmResponse) {
     codeExecutorContext,
     codeExecutionResult
   );
-  llmResponse.content = null;
+  llmResponse.content = void 0;
 }
 function extractAndReplaceInlineFiles(codeExecutorContext, llmRequest) {
   var _a2;
@@ -722,8 +730,8 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
     this.disallowTransferToParent = (_d = config.disallowTransferToParent) != null ? _d : false;
     this.disallowTransferToPeers = (_e = config.disallowTransferToPeers) != null ? _e : false;
     this.includeContents = (_f = config.includeContents) != null ? _f : "default";
-    this.inputSchema = config.inputSchema;
-    this.outputSchema = config.outputSchema;
+    this.inputSchema = (0, import_simple_zod_to_json.isZodObject)(config.inputSchema) ? (0, import_simple_zod_to_json.zodObjectToSchema)(config.inputSchema) : config.inputSchema;
+    this.outputSchema = (0, import_simple_zod_to_json.isZodObject)(config.outputSchema) ? (0, import_simple_zod_to_json.zodObjectToSchema)(config.outputSchema) : config.outputSchema;
     this.outputKey = config.outputKey;
     this.beforeModelCallback = config.beforeModelCallback;
     this.afterModelCallback = config.afterModelCallback;
@@ -768,16 +776,6 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
         this.disallowTransferToParent = true;
         this.disallowTransferToPeers = true;
       }
-      if (this.subAgents && this.subAgents.length > 0) {
-        throw new Error(
-          `Invalid config for agent ${this.name}: if outputSchema is set, subAgents must be empty to disable agent transfer.`
-        );
-      }
-      if (this.tools && this.tools.length > 0) {
-        throw new Error(
-          `Invalid config for agent ${this.name}: if outputSchema is set, tools must be empty`
-        );
-      }
     }
   }
   /**
@@ -809,12 +807,12 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
    * @param context The context to retrieve the session state.
    * @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
     };
   }
@@ -825,12 +823,15 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
    * @param context The context to retrieve the session state.
    * @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
     };
   }
@@ -839,10 +840,10 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
    *
    * 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;
@@ -946,10 +947,10 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
     }
     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;
@@ -963,19 +964,20 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
       }
     }
   }
-  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");
   }
@@ -1011,25 +1013,42 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
       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* () {
+        const responsesGenerator = 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;
+            }
+          }
+        };
+        yield* this.runAndHandleError(
+          responsesGenerator.call(this),
+          invocationContext,
+          llmRequest,
+          modelResponseEvent
+        );
       }
-    }
+    );
+    span.end();
   }
   async *postprocess(invocationContext, llmRequest, llmResponse, modelResponseEvent) {
-    var _a2;
+    var _a2, _b2;
     for (const processor of this.responseProcessors) {
       for await (const event of processor.runAsync(
         invocationContext,
@@ -1058,6 +1077,10 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
     if (!((_a2 = (0, import_event.getFunctionCalls)(mergedEvent)) == null ? void 0 : _a2.length)) {
       return;
     }
+    if ((_b2 = invocationContext.runConfig) == null ? void 0 : _b2.pauseOnToolCalls) {
+      invocationContext.endInvocation = true;
+      return;
+    }
     const functionResponseEvent = await (0, import_functions.handleFunctionCallsAsync)({
       invocationContext,
       functionCallEvent: mergedEvent,
@@ -1082,6 +1105,8 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
     });
     if (toolConfirmationEvent) {
       yield toolConfirmationEvent;
+      invocationContext.endInvocation = true;
+      return;
     }
     yield functionResponseEvent;
     const nextAgentName = functionResponseEvent.actions.transferToAgent;
@@ -1138,12 +1163,13 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
         /* stream= */
         ((_e = invocationContext.runConfig) == null ? void 0 : _e.streamingMode) === import_run_config.StreamingMode.SSE
       );
-      for await (const llmResponse of this.runAndHandleError(
-        responsesGenerator,
-        invocationContext,
-        llmRequest,
-        modelResponseEvent
-      )) {
+      for await (const llmResponse of responsesGenerator) {
+        (0, import_tracing.traceCallLlm)({
+          invocationContext,
+          eventId: modelResponseEvent.id,
+          llmRequest,
+          llmResponse
+        });
         const alteredLlmResponse = await this.handleAfterModelCallback(
           invocationContext,
           llmResponse,
@@ -1218,11 +1244,29 @@ class LlmAgent extends (_b = import_base_agent.BaseAgent, _a = LLM_AGENT_SIGNATU
         if (onModelErrorCallbackResponse) {
           yield onModelErrorCallbackResponse;
         } else {
-          const errorResponse = JSON.parse(modelError.message);
-          yield {
-            errorCode: String(errorResponse.error.code),
-            errorMessage: errorResponse.error.message
-          };
+          let errorCode = "UNKNOWN_ERROR";
+          let errorMessage = modelError.message;
+          try {
+            const errorResponse = JSON.parse(modelError.message);
+            if (errorResponse == null ? void 0 : errorResponse.error) {
+              errorCode = String(errorResponse.error.code || "UNKNOWN_ERROR");
+              errorMessage = errorResponse.error.message || errorMessage;
+            }
+          } catch {
+          }
+          if (modelResponseEvent.actions) {
+            yield (0, import_event.createEvent)({
+              invocationId: invocationContext.invocationId,
+              author: this.name,
+              errorCode,
+              errorMessage
+            });
+          } else {
+            yield {
+              errorCode,
+              errorMessage
+            };
+          }
         }
       } else {
         import_logger.logger.error("Unknown error during response generation", modelError);

package/dist/cjs/agents/loop_agent.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -68,7 +68,8 @@ class LoopAgent extends (_b = import_base_agent.BaseAgent, _a = LOOP_AGENT_SIGNA
     }
     return;
   }
-  async *runLiveImpl(context) {
+  // eslint-disable-next-line require-yield
+  async *runLiveImpl(_context) {
     throw new Error("This is not supported yet for LoopAgent.");
   }
 }

package/dist/cjs/agents/parallel_agent.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -50,15 +50,14 @@ class ParallelAgent extends (_b = import_base_agent.BaseAgent, _a = PARALLEL_AGE
   }
   async *runAsyncImpl(context) {
     const agentRuns = this.subAgents.map(
-      (subAgent) => subAgent.runAsync(
-        createBranchCtxForSubAgent(this, subAgent, context)
-      )
+      (subAgent) => subAgent.runAsync(createBranchCtxForSubAgent(this, subAgent, context))
     );
     for await (const event of mergeAgentRuns(agentRuns)) {
       yield event;
     }
   }
-  async *runLiveImpl(context) {
+  // eslint-disable-next-line require-yield
+  async *runLiveImpl(_context) {
     throw new Error("This is not supported yet for ParallelAgent.");
   }
 }

package/dist/cjs/agents/readonly_context.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -49,6 +49,18 @@ class ReadonlyContext {
   get invocationId() {
     return this.invocationContext.invocationId;
   }
+  /**
+   * The user ID of the current session.
+   */
+  get userId() {
+    return this.invocationContext.userId;
+  }
+  /**
+   * The ID of the current session.
+   */
+  get sessionId() {
+    return this.invocationContext.session.id;
+  }
   /**
    * The current agent name.
    */

package/dist/cjs/agents/run_config.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 
@@ -47,6 +47,7 @@ function createRunConfig(params = {}) {
     enableAffectiveDialog: false,
     streamingMode: "none" /* NONE */,
     maxLlmCalls: validateMaxLlmCalls(params.maxLlmCalls || 500),
+    pauseOnToolCalls: false,
     ...params
   };
 }

package/dist/cjs/agents/sequential_agent.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/agents/transcription_entry.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */
 

package/dist/cjs/artifacts/base_artifact_service.js

@@ -1,6 +1,6 @@
 /**
   * @license
-  * Copyright 2025 Google LLC
+  * Copyright 2026 Google LLC
   * SPDX-License-Identifier: Apache-2.0
   */