@paean-ai/adk 0.2.24 → 0.2.26

package/dist/esm/models/google_llm.js

@@ -40,11 +40,13 @@ class Gemini extends BaseLlm {
     this.headers = headers;
     this.isGemini3Preview = 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;
       logger.info(`Using Gemini 3 preview endpoint: ${this.apiEndpoint}`);
     }
@@ -55,6 +57,15 @@ class Gemini extends BaseLlm {
         useVertexAI = vertexAIfromEnv.toLowerCase() === "true" || vertexAIfromEnv === "1";
       }
     }
+    if (useAiStudioMode) {
+      if (useVertexAI) {
+        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) {
@@ -106,7 +117,7 @@ class Gemini extends BaseLlm {
    * @yields LlmResponse: The model response.
    */
   async *generateContentAsync(llmRequest, stream = false) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q;
     this.preprocessRequest(llmRequest);
     this.maybeAppendUserContent(llmRequest);
     logger.info(
@@ -129,6 +140,7 @@ class Gemini extends BaseLlm {
       let text = "";
       let usageMetadata;
       let lastResponse;
+      let pendingFCResponse = null;
       for await (const response of streamResult) {
         lastResponse = response;
         const llmResponse = createLlmResponse(response);
@@ -157,7 +169,9 @@ class Gemini extends BaseLlm {
           } else {
             text += firstPart.text;
           }
-          llmResponse.partial = true;
+          if (!hasFunctionCalls) {
+            llmResponse.partial = true;
+          }
           if (this.isGemini3Preview && hasFunctionCalls) {
             thoughtText = "";
             thoughtSignature = void 0;
@@ -227,18 +241,50 @@ class Gemini extends BaseLlm {
               }
             }
           }
-          const partsWithSig = llmResponse.content.parts.filter(
+        }
+        if (hasFunctionCalls) {
+          if (pendingFCResponse && ((_i = pendingFCResponse.content) == null ? void 0 : _i.parts)) {
+            const newParts = (((_j = llmResponse.content) == null ? void 0 : _j.parts) || []).filter(
+              (p) => p.functionCall || p.thoughtSignature
+            );
+            pendingFCResponse.content.parts.push(...newParts);
+            pendingFCResponse.usageMetadata = llmResponse.usageMetadata;
+          } else {
+            pendingFCResponse = llmResponse;
+          }
+          continue;
+        }
+        if (pendingFCResponse) {
+          if (!((_l = (_k = llmResponse.content) == null ? void 0 : _k.parts) == null ? void 0 : _l.length)) {
+            continue;
+          }
+          yield pendingFCResponse;
+          pendingFCResponse = null;
+        }
+        yield llmResponse;
+      }
+      if (pendingFCResponse) {
+        if (this.isGemini3Preview && ((_m = pendingFCResponse.content) == null ? void 0 : _m.parts)) {
+          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) {
             logger.warn(
-              `[Gemini3] No thoughtSignature on function call parts \u2014 may cause 400 on next request`
+              `[Gemini3] No thoughtSignature on merged function call parts \u2014 may cause 400 on next request`
             );
           }
         }
-        yield llmResponse;
+        yield pendingFCResponse;
+        pendingFCResponse = null;
       }
-      if ((text || thoughtText) && ((_j = (_i = lastResponse == null ? void 0 : lastResponse.candidates) == null ? void 0 : _i[0]) == null ? void 0 : _j.finishReason) === FinishReason.STOP) {
+      if ((text || thoughtText) && ((_o = (_n = lastResponse == null ? void 0 : lastResponse.candidates) == null ? void 0 : _n[0]) == null ? void 0 : _o.finishReason) === FinishReason.STOP) {
         const parts = [];
         if (thoughtText) {
           const thoughtPart = { text: thoughtText, thought: true };
@@ -260,12 +306,12 @@ class Gemini extends BaseLlm {
       }
     } else {
       const response = await this.apiClient.models.generateContent({
-        model: (_k = llmRequest.model) != null ? _k : this.model,
+        model: (_p = llmRequest.model) != null ? _p : this.model,
         contents: llmRequest.contents,
         config: llmRequest.config
       });
       const llmResponse = createLlmResponse(response);
-      if (this.isGemini3Preview && ((_l = llmResponse.content) == null ? void 0 : _l.parts)) {
+      if (this.isGemini3Preview && ((_q = llmResponse.content) == null ? void 0 : _q.parts)) {
         let thoughtSig;
         let hasThoughtPartWithSignature = false;
         for (const part of llmResponse.content.parts) {

package/dist/esm/models/llm_response.js

@@ -3,6 +3,7 @@
  * Copyright 2025 Google LLC
  * SPDX-License-Identifier: Apache-2.0
  */
+import { FinishReason } from "@google/genai";
 function createLlmResponse(response) {
   var _a;
   const usageMetadata = response.usageMetadata;
@@ -16,6 +17,12 @@ function createLlmResponse(response) {
         finishReason: candidate.finishReason
       };
     }
+    if (candidate.finishReason === FinishReason.STOP) {
+      return {
+        usageMetadata,
+        finishReason: candidate.finishReason
+      };
+    }
     return {
       errorCode: candidate.finishReason,
       errorMessage: candidate.finishMessage,

package/dist/types/models/google_llm.d.ts

@@ -20,6 +20,10 @@ export interface GeminiParams {
     /**
      * The API key to use for the Gemini API. If not provided, it will look for
      * the GOOGLE_GENAI_API_KEY or GEMINI_API_KEY environment variable.
+     *
+     * Alternatively, set AI_STUDIO_API_KEY env var to force all models
+     * (including Gemini 3 preview) through the standard AI Studio endpoint
+     * (generativelanguage.googleapis.com) instead of aiplatform.googleapis.com.
      */
     apiKey?: string;
     /**

package/dist/web/agents/functions.js

@@ -89,6 +89,9 @@ function generateAuthEvent(invocationContext, functionResponseEvent) {
     longRunningToolIds.add(requestEucFunctionCall.id);
     parts.push({ functionCall: requestEucFunctionCall });
   }
+  if (parts.length === 0) {
+    return void 0;
+  }
   return createEvent({
     invocationId: invocationContext.invocationId,
     author: invocationContext.agent.name,
@@ -130,6 +133,9 @@ function generateRequestConfirmationEvent({
     longRunningToolIds.add(requestConfirmationFunctionCall.id);
     parts.push({ functionCall: requestConfirmationFunctionCall });
   }
+  if (parts.length === 0) {
+    return void 0;
+  }
   return createEvent({
     invocationId: invocationContext.invocationId,
     author: invocationContext.agent.name,
@@ -193,8 +199,9 @@ async function handleFunctionCallList({
       }
     );
     if (!toolAndContext) {
+      const argsPreview = functionCall.args ? JSON.stringify(functionCall.args).substring(0, 300) : "(none)";
       logger.warn(
-        'Function "'.concat(functionCall.name, '" not found in toolsDict ') + "(".concat(Object.keys(toolsDict).length, " tools registered).")
+        'Function "'.concat(functionCall.name, '" not found in toolsDict ') + "(".concat(Object.keys(toolsDict).length, " tools registered). ") + "Args: ".concat(argsPreview)
       );
       const errorResponseEvent = createEvent({
         invocationId: invocationContext.invocationId,
@@ -243,6 +250,10 @@ async function handleFunctionCallList({
           toolContext
         );
       } catch (e) {
+        const argsPreview = JSON.stringify(functionArgs).substring(0, 500);
+        logger.error(
+          'Tool execution error: "'.concat(tool.name, '" threw ').concat(e instanceof Error ? e.message : String(e), ". Args: ").concat(argsPreview)
+        );
         if (e instanceof Error) {
           const onToolErrorResponse = await invocationContext.pluginManager.runOnToolErrorCallback(
             {
@@ -360,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/web/agents/llm_agent.js

@@ -923,12 +923,12 @@ const _LlmAgent = class _LlmAgent extends BaseAgent {
           consecutiveErrors++;
           if (consecutiveErrors <= _LlmAgent.MAX_AGENT_LOOP_ERROR_RETRIES) {
             logger.warn(
-              "[runAsyncImpl] Error event (".concat(lastEvent.errorCode, "), ") + "retrying agent loop (".concat(consecutiveErrors, "/").concat(_LlmAgent.MAX_AGENT_LOOP_ERROR_RETRIES, ")")
+              "[runAsyncImpl] Error event (".concat(lastEvent.errorCode, ": ").concat(lastEvent.errorMessage || "no message", "), ") + "retrying agent loop (".concat(consecutiveErrors, "/").concat(_LlmAgent.MAX_AGENT_LOOP_ERROR_RETRIES, ")")
             );
             continue;
           }
           logger.error(
-            "[runAsyncImpl] Max agent-loop error retries exhausted for ".concat(lastEvent.errorCode)
+            "[runAsyncImpl] Max agent-loop error retries exhausted for " + "".concat(lastEvent.errorCode, ": ").concat(lastEvent.errorMessage || "no message")
           );
           break;
         }
@@ -1073,7 +1073,7 @@ const _LlmAgent = class _LlmAgent extends BaseAgent {
   }
   postprocess(invocationContext, llmRequest, llmResponse, modelResponseEvent) {
     return __asyncGenerator(this, null, function* () {
-      var _a, _b;
+      var _a, _b, _c, _d, _e, _f, _g, _h, _i;
       for (const processor of this.responseProcessors) {
         try {
           for (var iter = __forAwait(processor.runAsync(invocationContext, llmResponse)), more, temp, error; more = !(temp = yield new __await(iter.next())).done; more = false) {
@@ -1094,6 +1094,28 @@ const _LlmAgent = class _LlmAgent extends BaseAgent {
       if (!llmResponse.content && !llmResponse.errorCode && !llmResponse.interrupted) {
         return;
       }
+      if (llmResponse.content && !llmResponse.errorCode) {
+        if (!llmResponse.content.parts || llmResponse.content.parts.length === 0) {
+          logger.debug(
+            "[postprocess] Skipping LLM response with no parts (role=".concat(llmResponse.content.role, ")")
+          );
+          return;
+        }
+        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;
+          }
+        );
+        if (allEmpty) {
+          logger.debug(
+            "[postprocess] Skipping empty-content LLM response (".concat(llmResponse.content.parts.length, " empty parts)")
+          );
+          return;
+        }
+      }
       const mergedEvent = createEvent(__spreadValues(__spreadValues({}, modelResponseEvent), llmResponse));
       if (mergedEvent.content) {
         const functionCalls = getFunctionCalls(mergedEvent);
@@ -1112,8 +1134,11 @@ const _LlmAgent = class _LlmAgent extends BaseAgent {
           }
         }
       }
+      logger.debug(
+        "[postprocess] Yielding mergedEvent: role=".concat((_b = mergedEvent.content) == null ? void 0 : _b.role, ", parts=").concat((_d = (_c = mergedEvent.content) == null ? void 0 : _c.parts) == null ? void 0 : _d.length, ", ") + "hasFCs=".concat(((_e = getFunctionCalls(mergedEvent)) == null ? void 0 : _e.length) || 0, ", partial=").concat(mergedEvent.partial)
+      );
       yield mergedEvent;
-      if (!((_b = getFunctionCalls(mergedEvent)) == null ? void 0 : _b.length)) {
+      if (!((_f = getFunctionCalls(mergedEvent)) == null ? void 0 : _f.length)) {
         return;
       }
       const functionResponseEvent = yield new __await(handleFunctionCallsAsync({
@@ -1138,6 +1163,9 @@ const _LlmAgent = class _LlmAgent extends BaseAgent {
       if (toolConfirmationEvent) {
         yield toolConfirmationEvent;
       }
+      logger.debug(
+        "[postprocess] Yielding functionResponseEvent: role=".concat((_g = functionResponseEvent.content) == null ? void 0 : _g.role, ", parts=").concat((_i = (_h = functionResponseEvent.content) == null ? void 0 : _h.parts) == null ? void 0 : _i.length)
+      );
       yield functionResponseEvent;
       const nextAgentName = functionResponseEvent.actions.transferToAgent;
       if (nextAgentName) {
@@ -1235,7 +1263,7 @@ const _LlmAgent = class _LlmAgent extends BaseAgent {
               if (llmResponse.errorCode && _LlmAgent.LLM_RETRYABLE_ERROR_CODES.has(llmResponse.errorCode) && !contentYielded && attempt < maxRetries) {
                 shouldRetry = true;
                 logger.warn(
-                  "[callLlmAsync] Transient LLM error: ".concat(llmResponse.errorCode, ", ") + "usage: ".concat(JSON.stringify(llmResponse.usageMetadata))
+                  "[callLlmAsync] Transient LLM error: ".concat(llmResponse.errorCode) + "".concat(llmResponse.errorMessage ? ": " + llmResponse.errorMessage : "", ", ") + "finishReason: ".concat(llmResponse.finishReason || "none", ", ") + "usage: ".concat(JSON.stringify(llmResponse.usageMetadata))
                 );
                 break;
               }