@paean-ai/adk 0.2.6 → 0.2.8

package/dist/web/models/google_llm.js

@@ -36,7 +36,11 @@ var __forAwait = (obj, it, method) => (it = obj[__knownSymbol("asyncIterator")])
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
-import { createPartFromText, FinishReason, GoogleGenAI } from "@google/genai";
+import {
+  createPartFromText,
+  FinishReason,
+  GoogleGenAI
+} from "@google/genai";
 import { logger } from "../utils/logger.js";
 import { isGemini3PreviewModel } from "../utils/model_name.js";
 import { GoogleLLMVariant } from "../utils/variant_utils.js";
@@ -87,11 +91,15 @@ class Gemini extends BaseLlm {
     if (this.isGemini3Preview && useVertexAI) {
       const availableApiKey = apiKey || (canReadEnv ? process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"] : void 0);
       if (availableApiKey) {
-        logger.info("Gemini 3 preview detected with Vertex AI mode. Switching to API key mode for correct endpoint handling.");
+        logger.info(
+          "Gemini 3 preview detected with Vertex AI mode. Switching to API key mode for correct endpoint handling."
+        );
         useVertexAI = false;
         this.apiKey = availableApiKey;
       } else {
-        logger.warn("Gemini 3 preview requires API key authentication for correct endpoint handling. Set GEMINI_API_KEY or GOOGLE_GENAI_API_KEY environment variable for best compatibility.");
+        logger.warn(
+          "Gemini 3 preview requires API key authentication for correct endpoint handling. Set GEMINI_API_KEY or GOOGLE_GENAI_API_KEY environment variable for best compatibility."
+        );
       }
     }
     this.vertexai = useVertexAI;
@@ -132,7 +140,7 @@ class Gemini extends BaseLlm {
    */
   generateContentAsync(llmRequest, stream = false) {
     return __asyncGenerator(this, null, function* () {
-      var _a, _b, _c, _d, _e, _f, _g;
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
       this.preprocessRequest(llmRequest);
       this.maybeAppendUserContent(llmRequest);
       logger.info(
@@ -148,6 +156,7 @@ class Gemini extends BaseLlm {
           config: llmRequest.config
         }));
         let thoughtText = "";
+        let thoughtSignature;
         let text = "";
         let usageMetadata;
         let lastResponse;
@@ -161,6 +170,9 @@ class Gemini extends BaseLlm {
             if (firstPart == null ? void 0 : firstPart.text) {
               if ("thought" in firstPart && firstPart.thought) {
                 thoughtText += firstPart.text;
+                if ("thoughtSignature" in firstPart && firstPart.thoughtSignature) {
+                  thoughtSignature = firstPart.thoughtSignature;
+                }
               } else {
                 text += firstPart.text;
               }
@@ -168,7 +180,11 @@ class Gemini extends BaseLlm {
             } else if ((thoughtText || text) && (!firstPart || !firstPart.inlineData)) {
               const parts = [];
               if (thoughtText) {
-                parts.push({ text: thoughtText, thought: true });
+                const thoughtPart = { text: thoughtText, thought: true };
+                if (thoughtSignature) {
+                  thoughtPart.thoughtSignature = thoughtSignature;
+                }
+                parts.push(thoughtPart);
               }
               if (text) {
                 parts.push(createPartFromText(text));
@@ -183,7 +199,17 @@ class Gemini extends BaseLlm {
               thoughtText = "";
               text = "";
             }
+            if (this.isGemini3Preview && thoughtSignature && ((_e = llmResponse.content) == null ? void 0 : _e.parts)) {
+              for (const part of llmResponse.content.parts) {
+                if (part.functionCall && !part.thoughtSignature) {
+                  part.thoughtSignature = thoughtSignature;
+                }
+              }
+            }
             yield llmResponse;
+            if ((_g = (_f = llmResponse.content) == null ? void 0 : _f.parts) == null ? void 0 : _g.some((part) => part.functionCall)) {
+              thoughtSignature = void 0;
+            }
           }
         } catch (temp) {
           error = [temp];
@@ -195,10 +221,14 @@ class Gemini extends BaseLlm {
               throw error[0];
           }
         }
-        if ((text || thoughtText) && ((_f = (_e = lastResponse == null ? void 0 : lastResponse.candidates) == null ? void 0 : _e[0]) == null ? void 0 : _f.finishReason) === FinishReason.STOP) {
+        if ((text || thoughtText) && ((_i = (_h = lastResponse == null ? void 0 : lastResponse.candidates) == null ? void 0 : _h[0]) == null ? void 0 : _i.finishReason) === FinishReason.STOP) {
           const parts = [];
           if (thoughtText) {
-            parts.push({ text: thoughtText, thought: true });
+            const thoughtPart = { text: thoughtText, thought: true };
+            if (thoughtSignature) {
+              thoughtPart.thoughtSignature = thoughtSignature;
+            }
+            parts.push(thoughtPart);
           }
           if (text) {
             parts.push({ text });
@@ -213,11 +243,28 @@ class Gemini extends BaseLlm {
         }
       } else {
         const response = yield new __await(this.apiClient.models.generateContent({
-          model: (_g = llmRequest.model) != null ? _g : this.model,
+          model: (_j = llmRequest.model) != null ? _j : this.model,
           contents: llmRequest.contents,
           config: llmRequest.config
         }));
-        yield createLlmResponse(response);
+        const llmResponse = createLlmResponse(response);
+        if (this.isGemini3Preview && ((_k = llmResponse.content) == null ? void 0 : _k.parts)) {
+          let thoughtSig;
+          for (const part of llmResponse.content.parts) {
+            if (part.thought && part.thoughtSignature) {
+              thoughtSig = part.thoughtSignature;
+              break;
+            }
+          }
+          if (thoughtSig) {
+            for (const part of llmResponse.content.parts) {
+              if (part.functionCall && !part.thoughtSignature) {
+                part.thoughtSignature = thoughtSig;
+              }
+            }
+          }
+        }
+        yield llmResponse;
       }
     });
   }
@@ -240,7 +287,9 @@ class Gemini extends BaseLlm {
         logger.debug("Using custom API endpoint: ".concat(this.apiEndpoint));
         if (this.isGemini3Preview) {
           httpOptions.apiVersion = "";
-          logger.info("Gemini 3 preview mode: using direct API path without version prefix");
+          logger.info(
+            "Gemini 3 preview mode: using direct API path without version prefix"
+          );
         }
       }
       this._apiClient = new GoogleGenAI({
@@ -304,7 +353,9 @@ class Gemini extends BaseLlm {
       llmRequest.liveConnectConfig.systemInstruction = {
         role: "system",
         // TODO - b/425992518: validate type casting works well.
-        parts: [createPartFromText(llmRequest.config.systemInstruction)]
+        parts: [
+          createPartFromText(llmRequest.config.systemInstruction)
+        ]
       };
     }
     llmRequest.liveConnectConfig.tools = (_c = llmRequest.config) == null ? void 0 : _c.tools;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@paean-ai/adk",
-  "version": "0.2.6",
+  "version": "0.2.8",
   "description": "Google ADK JS (paean-ai fork with streaming fixes)",
   "author": "paean-ai",
   "license": "Apache-2.0",