@paean-ai/adk 0.2.17 → 0.2.19

package/dist/esm/models/google_llm.js

@@ -1,533 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-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";
-import { BaseLlm } from "./base_llm.js";
-import { GeminiLlmConnection } from "./gemini_llm_connection.js";
-import { createLlmResponse } from "./llm_response.js";
-const AGENT_ENGINE_TELEMETRY_TAG = "remote_reasoning_engine";
-const AGENT_ENGINE_TELEMETRY_ENV_VARIABLE_NAME = "GOOGLE_CLOUD_AGENT_ENGINE_ID";
-const GEMINI3_PREVIEW_API_ENDPOINT = "https://aiplatform.googleapis.com/v1/publishers/google";
-class Gemini extends BaseLlm {
-  /**
-   * @param params The parameters for creating a Gemini instance.
-   */
-  constructor({
-    model,
-    apiKey,
-    vertexai,
-    project,
-    location,
-    headers,
-    apiEndpoint
-  }) {
-    if (!model) {
-      model = "gemini-2.5-flash";
-    }
-    super({ model });
-    this.project = project;
-    this.location = location;
-    this.apiKey = apiKey;
-    this.headers = headers;
-    this.isGemini3Preview = isGemini3PreviewModel(model);
-    const canReadEnv = typeof process === "object";
-    this.apiEndpoint = apiEndpoint;
-    if (!this.apiEndpoint && canReadEnv) {
-      this.apiEndpoint = process.env["GEMINI_API_ENDPOINT"];
-    }
-    if (!this.apiEndpoint && this.isGemini3Preview) {
-      this.apiEndpoint = GEMINI3_PREVIEW_API_ENDPOINT;
-      logger.info(`Using Gemini 3 preview endpoint: ${this.apiEndpoint}`);
-    }
-    let useVertexAI = !!vertexai;
-    if (!useVertexAI && canReadEnv) {
-      const vertexAIfromEnv = process.env["GOOGLE_GENAI_USE_VERTEXAI"];
-      if (vertexAIfromEnv) {
-        useVertexAI = vertexAIfromEnv.toLowerCase() === "true" || vertexAIfromEnv === "1";
-      }
-    }
-    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."
-        );
-        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."
-        );
-      }
-    }
-    this.vertexai = useVertexAI;
-    if (this.vertexai) {
-      if (canReadEnv && !this.project) {
-        this.project = process.env["GOOGLE_CLOUD_PROJECT"];
-      }
-      if (canReadEnv && !this.location) {
-        this.location = process.env["GOOGLE_CLOUD_LOCATION"];
-      }
-      if (!this.project) {
-        throw new Error(
-          "VertexAI project must be provided via constructor or GOOGLE_CLOUD_PROJECT environment variable."
-        );
-      }
-      if (!this.location) {
-        throw new Error(
-          "VertexAI location must be provided via constructor or GOOGLE_CLOUD_LOCATION environment variable."
-        );
-      }
-    } else {
-      if (!this.apiKey && canReadEnv) {
-        this.apiKey = process.env["GOOGLE_GENAI_API_KEY"] || process.env["GEMINI_API_KEY"];
-      }
-      if (!this.apiKey) {
-        throw new Error(
-          "API key must be provided via constructor or GOOGLE_GENAI_API_KEY or GEMINI_API_KEY environment variable."
-        );
-      }
-    }
-  }
-  /**
-   * Sends a request to the Gemini model.
-   *
-   * @param llmRequest LlmRequest, the request to send to the Gemini model.
-   * @param stream bool = false, whether to do streaming call.
-   * @yields LlmResponse: The model response.
-   */
-  async *generateContentAsync(llmRequest, stream = false) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
-    this.preprocessRequest(llmRequest);
-    this.maybeAppendUserContent(llmRequest);
-    logger.info(
-      `Sending out request, model: ${llmRequest.model}, backend: ${this.apiBackend}, stream: ${stream}`
-    );
-    if ((_a = llmRequest.config) == null ? void 0 : _a.httpOptions) {
-      llmRequest.config.httpOptions.headers = {
-        ...llmRequest.config.httpOptions.headers,
-        ...this.trackingHeaders
-      };
-    }
-    if (stream) {
-      const streamResult = await this.apiClient.models.generateContentStream({
-        model: (_b = llmRequest.model) != null ? _b : this.model,
-        contents: llmRequest.contents,
-        config: llmRequest.config
-      });
-      let thoughtText = "";
-      let thoughtSignature;
-      let text = "";
-      let usageMetadata;
-      let lastResponse;
-      if (this.isGemini3Preview && this.cachedThoughtSignature) {
-        logger.info(
-          `[Gemini3] Starting new request with CACHED thoughtSignature (length: ${this.cachedThoughtSignature.length})`
-        );
-      }
-      for await (const response of streamResult) {
-        lastResponse = response;
-        const llmResponse = createLlmResponse(response);
-        usageMetadata = llmResponse.usageMetadata;
-        const firstPart = (_d = (_c = llmResponse.content) == null ? void 0 : _c.parts) == null ? void 0 : _d[0];
-        const hasFunctionCalls = (_f = (_e = llmResponse.content) == null ? void 0 : _e.parts) == null ? void 0 : _f.some(
-          (part) => part.functionCall
-        );
-        if (this.isGemini3Preview && ((_g = llmResponse.content) == null ? void 0 : _g.parts)) {
-          for (const part of llmResponse.content.parts) {
-            if (part.thoughtSignature && !thoughtSignature) {
-              thoughtSignature = part.thoughtSignature;
-              this.cachedThoughtSignature = thoughtSignature;
-              logger.info(
-                `[Gemini3] Captured and CACHED thoughtSignature from response part (length: ${thoughtSignature.length})`
-              );
-              break;
-            }
-          }
-        }
-        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;
-              this.cachedThoughtSignature = thoughtSignature;
-              logger.info(
-                `[Gemini3] Captured and CACHED thoughtSignature from thought part (length: ${thoughtSignature.length})`
-              );
-            }
-          } else {
-            text += firstPart.text;
-          }
-          llmResponse.partial = true;
-          if (this.isGemini3Preview && hasFunctionCalls) {
-            const responseHasSignature = (_i = (_h = llmResponse.content) == null ? void 0 : _h.parts) == null ? void 0 : _i.some(
-              (p) => p.thoughtSignature
-            );
-            logger.info(
-              `[Gemini3] Chunk has thought AND function calls. Response has signature: ${responseHasSignature}`
-            );
-            thoughtText = "";
-            thoughtSignature = void 0;
-            text = "";
-          }
-        } else if ((thoughtText || text) && (!firstPart || !firstPart.inlineData)) {
-          if (this.isGemini3Preview && hasFunctionCalls && llmResponse.content) {
-            logger.info(
-              `[Gemini3] Merging thought with function calls. Has accumulated signature: ${!!thoughtSignature}`
-            );
-            const prependParts = [];
-            if (thoughtText) {
-              const thoughtPart = { text: thoughtText, thought: true };
-              if (thoughtSignature) {
-                thoughtPart.thoughtSignature = thoughtSignature;
-              }
-              prependParts.push(thoughtPart);
-              logger.info(
-                `[Gemini3] Created thought part with signature: ${!!thoughtSignature}`
-              );
-            }
-            if (text) {
-              prependParts.push(createPartFromText(text));
-            }
-            if (!thoughtText && thoughtSignature) {
-              let signatureApplied = false;
-              for (const part of llmResponse.content.parts || []) {
-                if (part.functionCall && !signatureApplied) {
-                  if (!part.thoughtSignature) {
-                    part.thoughtSignature = thoughtSignature;
-                    logger.info(
-                      `[Gemini3] Applied accumulated signature to first function call: ${part.functionCall.name}`
-                    );
-                  }
-                  signatureApplied = true;
-                }
-              }
-            }
-            llmResponse.content.parts = [
-              ...prependParts,
-              ...llmResponse.content.parts || []
-            ];
-            logger.info(
-              `[Gemini3] Final merged content has ${llmResponse.content.parts.length} parts`
-            );
-            for (let i = 0; i < llmResponse.content.parts.length; i++) {
-              const p = llmResponse.content.parts[i];
-              logger.info(
-                `[Gemini3] Part ${i}: thought=${!!p.thought}, functionCall=${((_j = p.functionCall) == null ? void 0 : _j.name) || "none"}, hasSignature=${!!p.thoughtSignature}`
-              );
-            }
-            thoughtText = "";
-            thoughtSignature = void 0;
-            text = "";
-          } else {
-            const parts = [];
-            if (thoughtText) {
-              const thoughtPart = { text: thoughtText, thought: true };
-              if (thoughtSignature) {
-                thoughtPart.thoughtSignature = thoughtSignature;
-              }
-              parts.push(thoughtPart);
-            }
-            if (text) {
-              parts.push(createPartFromText(text));
-            }
-            yield {
-              content: {
-                role: "model",
-                parts
-              },
-              usageMetadata: llmResponse.usageMetadata
-            };
-            thoughtText = "";
-            thoughtSignature = void 0;
-            text = "";
-          }
-        }
-        if (this.isGemini3Preview && hasFunctionCalls && ((_k = llmResponse.content) == null ? void 0 : _k.parts)) {
-          const hasExistingSignature = llmResponse.content.parts.some(
-            (p) => p.thoughtSignature
-          );
-          if (!hasExistingSignature && thoughtSignature) {
-            for (const part of llmResponse.content.parts) {
-              if (part.functionCall) {
-                part.thoughtSignature = thoughtSignature;
-                logger.info(
-                  `[Gemini3] Applied cached signature to first function call: ${part.functionCall.name}`
-                );
-                break;
-              }
-            }
-          }
-          const functionCallNames = llmResponse.content.parts.filter((p) => p.functionCall).map((p) => p.functionCall.name);
-          let partsWithSig = llmResponse.content.parts.filter(
-            (p) => p.thoughtSignature
-          ).length;
-          if (partsWithSig === 0) {
-            const signatureToUse = thoughtSignature || this.cachedThoughtSignature;
-            if (signatureToUse) {
-              for (const part of llmResponse.content.parts) {
-                if (part.functionCall && !part.thoughtSignature) {
-                  part.thoughtSignature = signatureToUse;
-                  logger.info(
-                    `[Gemini3] Applied CACHED signature to function call: ${part.functionCall.name} (API didn't return new signature, using ${thoughtSignature ? "local" : "class-level"} cache)`
-                  );
-                  partsWithSig = 1;
-                  break;
-                }
-              }
-            }
-          }
-          logger.info(
-            `[Gemini3] Yielding function call response: calls=[${functionCallNames.join(", ")}], partsWithSignature=${partsWithSig}`
-          );
-          if (partsWithSig === 0) {
-            logger.warn(
-              `[Gemini3] WARNING: No thoughtSignature found and no cached signature available! This may cause 400 errors on next request.`
-            );
-          }
-        }
-        yield llmResponse;
-      }
-      if ((text || thoughtText) && ((_m = (_l = lastResponse == null ? void 0 : lastResponse.candidates) == null ? void 0 : _l[0]) == null ? void 0 : _m.finishReason) === FinishReason.STOP) {
-        const parts = [];
-        if (thoughtText) {
-          const thoughtPart = { text: thoughtText, thought: true };
-          if (thoughtSignature) {
-            thoughtPart.thoughtSignature = thoughtSignature;
-          }
-          parts.push(thoughtPart);
-        }
-        if (text) {
-          parts.push({ text });
-        }
-        yield {
-          content: {
-            role: "model",
-            parts
-          },
-          usageMetadata
-        };
-      }
-    } else {
-      const response = await this.apiClient.models.generateContent({
-        model: (_n = llmRequest.model) != null ? _n : this.model,
-        contents: llmRequest.contents,
-        config: llmRequest.config
-      });
-      const llmResponse = createLlmResponse(response);
-      if (this.isGemini3Preview && ((_o = llmResponse.content) == null ? void 0 : _o.parts)) {
-        let thoughtSig;
-        let hasThoughtPartWithSignature = false;
-        for (const part of llmResponse.content.parts) {
-          if (part.thoughtSignature) {
-            thoughtSig = part.thoughtSignature;
-            this.cachedThoughtSignature = thoughtSig;
-            if (part.thought) {
-              hasThoughtPartWithSignature = true;
-            }
-            break;
-          }
-        }
-        logger.info(
-          `[Gemini3] Non-streaming response: hasSignature=${!!thoughtSig}, hasThoughtPart=${hasThoughtPartWithSignature}, hasCachedSig=${!!this.cachedThoughtSignature}`
-        );
-        if (thoughtSig && !hasThoughtPartWithSignature) {
-          let signatureApplied = false;
-          for (const part of llmResponse.content.parts) {
-            if (part.functionCall) {
-              if (!signatureApplied && !part.thoughtSignature) {
-                part.thoughtSignature = thoughtSig;
-                logger.info(
-                  `[Gemini3] Applied signature to first function call: ${part.functionCall.name}`
-                );
-              }
-              signatureApplied = true;
-            }
-          }
-        }
-        const hasFunctionCalls = llmResponse.content.parts.some(
-          (p) => p.functionCall
-        );
-        if (hasFunctionCalls) {
-          let partsWithSig = llmResponse.content.parts.filter(
-            (p) => p.thoughtSignature
-          ).length;
-          if (partsWithSig === 0 && this.cachedThoughtSignature) {
-            for (const part of llmResponse.content.parts) {
-              if (part.functionCall && !part.thoughtSignature) {
-                part.thoughtSignature = this.cachedThoughtSignature;
-                logger.info(
-                  `[Gemini3] Applied CACHED signature to function call: ${part.functionCall.name} (API didn't return new signature)`
-                );
-                partsWithSig = 1;
-                break;
-              }
-            }
-          }
-          for (let i = 0; i < llmResponse.content.parts.length; i++) {
-            const p = llmResponse.content.parts[i];
-            logger.info(
-              `[Gemini3] Response Part ${i}: thought=${!!p.thought}, functionCall=${((_p = p.functionCall) == null ? void 0 : _p.name) || "none"}, hasSignature=${!!p.thoughtSignature}`
-            );
-          }
-          if (partsWithSig === 0) {
-            logger.warn(
-              `[Gemini3] WARNING: No thoughtSignature found and no cached signature available! This may cause 400 errors on next request.`
-            );
-          }
-        }
-      }
-      yield llmResponse;
-    }
-  }
-  get apiClient() {
-    if (this._apiClient) {
-      return this._apiClient;
-    }
-    const combinedHeaders = {
-      ...this.trackingHeaders,
-      ...this.headers
-    };
-    if (this.vertexai) {
-      this._apiClient = new GoogleGenAI({
-        vertexai: this.vertexai,
-        project: this.project,
-        location: this.location,
-        httpOptions: { headers: combinedHeaders }
-      });
-    } else {
-      const httpOptions = { headers: combinedHeaders };
-      if (this.apiEndpoint) {
-        httpOptions.baseUrl = this.apiEndpoint;
-        logger.debug(`Using custom API endpoint: ${this.apiEndpoint}`);
-        if (this.isGemini3Preview) {
-          httpOptions.apiVersion = "";
-          logger.info(
-            `Gemini 3 preview mode: using direct API path without version prefix`
-          );
-        }
-      }
-      this._apiClient = new GoogleGenAI({
-        apiKey: this.apiKey,
-        vertexai: false,
-        httpOptions
-      });
-    }
-    return this._apiClient;
-  }
-  get apiBackend() {
-    if (!this._apiBackend) {
-      this._apiBackend = this.apiClient.vertexai ? GoogleLLMVariant.VERTEX_AI : GoogleLLMVariant.GEMINI_API;
-    }
-    return this._apiBackend;
-  }
-  get liveApiVersion() {
-    if (!this._liveApiVersion) {
-      this._liveApiVersion = this.apiBackend === GoogleLLMVariant.VERTEX_AI ? "v1beta1" : "v1alpha";
-    }
-    return this._liveApiVersion;
-  }
-  get liveApiClient() {
-    if (!this._liveApiClient) {
-      const httpOptions = {
-        headers: this.trackingHeaders,
-        apiVersion: this.liveApiVersion
-      };
-      if (this.apiEndpoint) {
-        httpOptions.baseUrl = this.apiEndpoint;
-        if (this.isGemini3Preview) {
-          httpOptions.apiVersion = "";
-        }
-      }
-      this._liveApiClient = new GoogleGenAI({
-        apiKey: this.apiKey,
-        httpOptions
-      });
-    }
-    return this._liveApiClient;
-  }
-  /**
-   * Connects to the Gemini model and returns an llm connection.
-   *
-   * @param llmRequest LlmRequest, the request to send to the Gemini model.
-   * @returns BaseLlmConnection, the connection to the Gemini model.
-   */
-  async connect(llmRequest) {
-    var _a, _b, _c, _d;
-    if ((_a = llmRequest.liveConnectConfig) == null ? void 0 : _a.httpOptions) {
-      if (!llmRequest.liveConnectConfig.httpOptions.headers) {
-        llmRequest.liveConnectConfig.httpOptions.headers = {};
-      }
-      Object.assign(
-        llmRequest.liveConnectConfig.httpOptions.headers,
-        this.trackingHeaders
-      );
-      llmRequest.liveConnectConfig.httpOptions.apiVersion = this.isGemini3Preview ? "" : this.liveApiVersion;
-    }
-    if ((_b = llmRequest.config) == null ? void 0 : _b.systemInstruction) {
-      llmRequest.liveConnectConfig.systemInstruction = {
-        role: "system",
-        // TODO - b/425992518: validate type casting works well.
-        parts: [
-          createPartFromText(llmRequest.config.systemInstruction)
-        ]
-      };
-    }
-    llmRequest.liveConnectConfig.tools = (_c = llmRequest.config) == null ? void 0 : _c.tools;
-    const liveSession = await this.liveApiClient.live.connect({
-      model: (_d = llmRequest.model) != null ? _d : this.model,
-      config: llmRequest.liveConnectConfig,
-      callbacks: {
-        // TODO - b/425992518: GenAI SDK inconsistent API, missing methods.
-        onmessage: () => {
-        }
-      }
-    });
-    return new GeminiLlmConnection(liveSession);
-  }
-  preprocessRequest(llmRequest) {
-    if (this.apiBackend === GoogleLLMVariant.GEMINI_API) {
-      if (llmRequest.config) {
-        llmRequest.config.labels = void 0;
-      }
-      if (llmRequest.contents) {
-        for (const content of llmRequest.contents) {
-          if (!content.parts) continue;
-          for (const part of content.parts) {
-            removeDisplayNameIfPresent(part.inlineData);
-            removeDisplayNameIfPresent(part.fileData);
-          }
-        }
-      }
-    }
-  }
-}
-/**
- * A list of model name patterns that are supported by this LLM.
- *
- * @returns A list of supported models.
- */
-Gemini.supportedModels = [
-  /gemini-.*/,
-  // fine-tuned vertex endpoint pattern
-  /projects\/.+\/locations\/.+\/endpoints\/.+/,
-  // vertex gemini long name
-  /projects\/.+\/locations\/.+\/publishers\/google\/models\/gemini.+/
-];
-function removeDisplayNameIfPresent(dataObj) {
-  if (dataObj && dataObj.displayName) {
-    dataObj.displayName = void 0;
-  }
-}
-export {
-  Gemini
-};

package/dist/esm/models/llm_request.js

@@ -1,50 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-function appendInstructions(llmRequest, instructions) {
-  if (!llmRequest.config) {
-    llmRequest.config = {};
-  }
-  const newInstructions = instructions.join("\n\n");
-  if (llmRequest.config.systemInstruction) {
-    llmRequest.config.systemInstruction += "\n\n" + newInstructions;
-  } else {
-    llmRequest.config.systemInstruction = newInstructions;
-  }
-}
-function appendTools(llmRequest, tools) {
-  if (!(tools == null ? void 0 : tools.length)) {
-    return;
-  }
-  const functionDeclarations = [];
-  for (const tool of tools) {
-    const declaration = tool._getDeclaration();
-    if (declaration) {
-      functionDeclarations.push(declaration);
-      llmRequest.toolsDict[tool.name] = tool;
-    }
-  }
-  if (functionDeclarations.length) {
-    if (!llmRequest.config) {
-      llmRequest.config = {};
-    }
-    if (!llmRequest.config.tools) {
-      llmRequest.config.tools = [];
-    }
-    llmRequest.config.tools.push({ functionDeclarations });
-  }
-}
-function setOutputSchema(llmRequest, schema) {
-  if (!llmRequest.config) {
-    llmRequest.config = {};
-  }
-  llmRequest.config.responseSchema = schema;
-  llmRequest.config.responseMimeType = "application/json";
-}
-export {
-  appendInstructions,
-  appendTools,
-  setOutputSchema
-};

package/dist/esm/models/llm_response.js

@@ -1,41 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-function createLlmResponse(response) {
-  var _a;
-  const usageMetadata = 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) {
-      return {
-        content: candidate.content,
-        groundingMetadata: candidate.groundingMetadata,
-        usageMetadata,
-        finishReason: candidate.finishReason
-      };
-    }
-    return {
-      errorCode: candidate.finishReason,
-      errorMessage: candidate.finishMessage,
-      usageMetadata,
-      finishReason: candidate.finishReason
-    };
-  }
-  if (response.promptFeedback) {
-    return {
-      errorCode: response.promptFeedback.blockReason,
-      errorMessage: response.promptFeedback.blockReasonMessage,
-      usageMetadata
-    };
-  }
-  return {
-    errorCode: "UNKNOWN_ERROR",
-    errorMessage: "Unknown error.",
-    usageMetadata
-  };
-}
-export {
-  createLlmResponse
-};