@paean-ai/adk 0.2.25 → 0.2.26

package/dist/cjs/models/google_llm.js

@@ -65,11 +65,13 @@ class Gemini extends import_base_llm.BaseLlm {
     this.headers = headers;
     this.isGemini3Preview = (0, import_model_name.isGemini3PreviewModel)(model);
     const canReadEnv = typeof process === "object";
+    const aiStudioApiKey = canReadEnv ? process.env["AI_STUDIO_API_KEY"] : void 0;
+    const useAiStudioMode = !!aiStudioApiKey;
     this.apiEndpoint = apiEndpoint;
     if (!this.apiEndpoint && canReadEnv) {
       this.apiEndpoint = process.env["GEMINI_API_ENDPOINT"];
     }
-    if (!this.apiEndpoint && this.isGemini3Preview) {
+    if (!this.apiEndpoint && this.isGemini3Preview && !useAiStudioMode) {
       this.apiEndpoint = GEMINI3_PREVIEW_API_ENDPOINT;
       import_logger.logger.info(`Using Gemini 3 preview endpoint: ${this.apiEndpoint}`);
     }
@@ -80,6 +82,15 @@ class Gemini extends import_base_llm.BaseLlm {
         useVertexAI = vertexAIfromEnv.toLowerCase() === "true" || vertexAIfromEnv === "1";
       }
     }
+    if (useAiStudioMode) {
+      if (useVertexAI) {
+        import_logger.logger.info(
+          "AI_STUDIO_API_KEY set \u2014 overriding Vertex AI mode to use AI Studio (generativelanguage.googleapis.com)"
+        );
+      }
+      useVertexAI = false;
+      this.apiKey = aiStudioApiKey;
+    }
     if (this.isGemini3Preview && useVertexAI) {
       const availableApiKey = apiKey || (canReadEnv ? process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"] : void 0);
       if (availableApiKey) {
@@ -258,10 +269,10 @@ class Gemini extends import_base_llm.BaseLlm {
         }
         if (hasFunctionCalls) {
           if (pendingFCResponse && ((_i = pendingFCResponse.content) == null ? void 0 : _i.parts)) {
-            const newFCParts = (((_j = llmResponse.content) == null ? void 0 : _j.parts) || []).filter(
-              (p) => p.functionCall
+            const newParts = (((_j = llmResponse.content) == null ? void 0 : _j.parts) || []).filter(
+              (p) => p.functionCall || p.thoughtSignature
             );
-            pendingFCResponse.content.parts.push(...newFCParts);
+            pendingFCResponse.content.parts.push(...newParts);
             pendingFCResponse.usageMetadata = llmResponse.usageMetadata;
           } else {
             pendingFCResponse = llmResponse;
@@ -282,7 +293,14 @@ class Gemini extends import_base_llm.BaseLlm {
           const partsWithSig = pendingFCResponse.content.parts.filter(
             (p) => p.thoughtSignature
           ).length;
-          if (partsWithSig === 0) {
+          if (partsWithSig === 0 && thoughtSignature) {
+            for (const part of pendingFCResponse.content.parts) {
+              if (part.functionCall) {
+                part.thoughtSignature = thoughtSignature;
+                break;
+              }
+            }
+          } else if (partsWithSig === 0) {
             import_logger.logger.warn(
               `[Gemini3] No thoughtSignature on merged function call parts \u2014 may cause 400 on next request`
             );

package/dist/cjs/models/llm_response.js

@@ -27,6 +27,7 @@ __export(llm_response_exports, {
   createLlmResponse: () => createLlmResponse
 });
 module.exports = __toCommonJS(llm_response_exports);
+var import_genai = require("@google/genai");
 /**
  * @license
  * Copyright 2025 Google LLC
@@ -45,6 +46,12 @@ function createLlmResponse(response) {
         finishReason: candidate.finishReason
       };
     }
+    if (candidate.finishReason === import_genai.FinishReason.STOP) {
+      return {
+        usageMetadata,
+        finishReason: candidate.finishReason
+      };
+    }
     return {
       errorCode: candidate.finishReason,
       errorMessage: candidate.finishMessage,

package/dist/esm/agents/functions.js

@@ -371,6 +371,7 @@ function mergeParallelFunctionResponseEvents(functionResponseEvents) {
   const actionsList = functionResponseEvents.map((event) => event.actions || {});
   const mergedActions = mergeEventActions(actionsList);
   return createEvent({
+    invocationId: baseEvent.invocationId,
     author: baseEvent.author,
     branch: baseEvent.branch,
     content: { role: "user", parts: mergedParts },

package/dist/esm/agents/llm_agent.js

@@ -1012,6 +1012,7 @@ const _LlmAgent = class _LlmAgent extends BaseAgent {
       const allEmpty = llmResponse.content.parts.every(
         (p) => {
           if (p.functionCall || p.functionResponse || p.executableCode || p.codeExecutionResult) return false;
+          if (p.inlineData || p.fileData) return false;
           if ("text" in p && typeof p.text === "string" && p.text.length > 0) return false;
           return true;
         }

package/dist/esm/events/event.js

@@ -5,7 +5,7 @@
  */
 import { createEventActions } from "./event_actions.js";
 function createEvent(params = {}) {
-  const event = {
+  return {
     ...params,
     id: params.id || createNewEventId(),
     invocationId: params.invocationId || "",
@@ -15,7 +15,6 @@ function createEvent(params = {}) {
     branch: params.branch,
     timestamp: params.timestamp || Date.now()
   };
-  return event;
 }
 function isFinalResponse(event) {
   if (event.actions.skipSummarization || event.longRunningToolIds && event.longRunningToolIds.length > 0) {