@google/adk 2.2.0 → 2.3.0

package/dist/cjs/models/google_llm.js

@@ -101,7 +101,7 @@ const _Gemini = class _Gemini extends (_b = import_base_llm.BaseLlm, _a = GEMINI
    *     metadata and no `content`; only a stream with neither ends silently.
    */
   async *generateContentAsync(llmRequest, stream = false, abortSignal) {
-    var _a2, _b2, _c;
+    var _a2, _b2;
     if (this.useInteractionsApi) {
       yield* (0, import_interactions_utils.generateContentViaInteractions)(this.apiClient, llmRequest, stream);
       return;
@@ -123,28 +123,23 @@ const _Gemini = class _Gemini extends (_b = import_base_llm.BaseLlm, _a = GEMINI
     if (abortSignal) {
       llmRequest.config.abortSignal = abortSignal;
     }
+    const params = {
+      model: (_b2 = llmRequest.model) != null ? _b2 : this.model,
+      contents: llmRequest.contents,
+      config: llmRequest.config
+    };
     if (stream) {
-      const streamResult = await this.apiClient.models.generateContentStream({
-        model: (_b2 = llmRequest.model) != null ? _b2 : this.model,
-        contents: llmRequest.contents,
-        config: llmRequest.config
-      });
-      const aggregator = new import_streaming_utils.StreamingResponseAggregator();
+      const streamResult = await this.apiClient.models.generateContentStream(params);
+      const aggregator = new import_streaming_utils.GeminiStreamingResponseAggregator();
       for await (const response of streamResult) {
-        for await (const llmResponse of aggregator.processResponse(response)) {
-          yield llmResponse;
-        }
+        yield* aggregator.processResponse(response);
       }
       const finalResponse = aggregator.close();
       if (finalResponse) {
         yield finalResponse;
       }
     } else {
-      const response = await this.apiClient.models.generateContent({
-        model: (_c = llmRequest.model) != null ? _c : this.model,
-        contents: llmRequest.contents,
-        config: llmRequest.config
-      });
+      const response = await this.apiClient.models.generateContent(params);
       yield (0, import_llm_response.createLlmResponse)(response);
     }
   }
@@ -164,6 +159,7 @@ const _Gemini = class _Gemini extends (_b = import_base_llm.BaseLlm, _a = GEMINI
       });
     } else {
       this._apiClient = new import_genai.GoogleGenAI({
+        vertexai: false,
         apiKey: this.apiKey,
         httpOptions: this.getHttpOptions()
       });
@@ -199,6 +195,7 @@ const _Gemini = class _Gemini extends (_b = import_base_llm.BaseLlm, _a = GEMINI
         });
       } else {
         this._liveApiClient = new import_genai.GoogleGenAI({
+          vertexai: false,
           apiKey: this.apiKey,
           httpOptions: this.getLiveHttpOptions()
         });
@@ -305,7 +302,7 @@ function geminiInitParams({
 }) {
   const params = { model, vertexai, project, location, apiKey };
   params.vertexai = !!vertexai;
-  if (!params.vertexai && !(0, import_env_aware_utils.isBrowser)()) {
+  if (vertexai === void 0 && !(0, import_env_aware_utils.isBrowser)()) {
     params.vertexai = (0, import_env_aware_utils.isEnterpriseModeEnabled)();
   }
   if (params.vertexai) {

package/dist/cjs/models/llm_response.js

@@ -33,9 +33,17 @@ module.exports = __toCommonJS(llm_response_exports);
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+function withOwnInitialPromptTokenCount(usage) {
+  const initialPromptTokenCount = usage == null ? void 0 : usage.initialPromptTokenCount;
+  if (!usage || initialPromptTokenCount === void 0 || initialPromptTokenCount === usage.promptTokenCount || Object.prototype.propertyIsEnumerable.call(usage, "initialPromptTokenCount")) {
+    return usage;
+  }
+  return { ...usage, initialPromptTokenCount };
+}
+__name(withOwnInitialPromptTokenCount, "withOwnInitialPromptTokenCount");
 function createLlmResponse(response) {
   var _a;
-  const usageMetadata = response.usageMetadata;
+  const usageMetadata = withOwnInitialPromptTokenCount(response.usageMetadata);
   if (response.candidates && response.candidates.length > 0) {
     const candidate = response.candidates[0];
     if (((_a = candidate.content) == null ? void 0 : _a.parts) && candidate.content.parts.length > 0) {

package/dist/cjs/models/registry.js

@@ -101,7 +101,7 @@ const _LLMRegistry = class _LLMRegistry {
     }
     for (const [regex, llmClass] of _LLMRegistry.llmRegistryDict.entries()) {
       const pattern = new RegExp(
-        `^${regex instanceof RegExp ? regex.source : regex}$`,
+        `^(?:${regex instanceof RegExp ? regex.source : regex})$`,
         regex instanceof RegExp ? regex.flags : void 0
       );
       if (pattern.test(model)) {

package/dist/cjs/models/routed_llm.js

@@ -54,14 +54,12 @@ const _RoutedLlm = class _RoutedLlm extends import_base_llm.BaseLlm {
   /**
    * Generates content by delegating to the selected model.
    */
-  async *generateContentAsync(llmRequest, stream) {
+  async *generateContentAsync(llmRequest, stream, abortSignal) {
     import_logger.logger.info(`Routing request via ${this.model}`);
-    yield* (0, import_failover_utils.runWithRouting)(
-      this.models,
-      llmRequest,
-      this.router,
-      (model) => model.generateContentAsync(llmRequest, stream)
-    );
+    yield* (0, import_failover_utils.runWithRouting)(this.models, llmRequest, this.router, (model) => {
+      llmRequest.model = model.model;
+      return model.generateContentAsync(llmRequest, stream, abortSignal);
+    });
   }
   /**
    * Creates a live connection to the LLM by delegating to the selected model.
@@ -73,7 +71,10 @@ const _RoutedLlm = class _RoutedLlm extends import_base_llm.BaseLlm {
       this.models,
       llmRequest,
       this.router,
-      (model) => model.connect(llmRequest)
+      (model) => {
+        llmRequest.model = model.model;
+        return model.connect(llmRequest);
+      }
     );
     const result = await generator.next();
     if (result.done || result.value === void 0) {

package/dist/cjs/planners/plan_re_act_planner.js

@@ -67,7 +67,7 @@ const _PlanReActPlanner = class _PlanReActPlanner extends import_base_planner.Ba
       }
       handleNonFunctionCallPart(part, preservedParts);
     }
-    if (firstFcPartIndex > 0) {
+    if (firstFcPartIndex >= 0) {
       for (let j = firstFcPartIndex + 1; j < responseParts.length; j++) {
         if (!responseParts[j].functionCall) {
           break;