@langchain/google 0.2.11 → 0.2.12

package/dist/chat_models/base.d.ts

@@ -1,68 +1,18 @@
-import { Prettify } from "../utils/misc.js";
 import { ApiClient } from "../clients/index.js";
-import { Gemini } from "./api-types.js";
-import { ChatGoogleFields } from "./types.js";
-import { BaseChatModel, BaseChatModelCallOptions, BaseChatModelParams, BindToolsInput, LangSmithParams } from "@langchain/core/language_models/chat_models";
+import { GoogleChatApiHandler } from "./handlers/base.js";
+import { BaseChatGoogleCallOptions, BaseChatGoogleParams, GenerateContentApiHandlerParams, GooglePlatformType } from "./types.js";
+import "./handlers/index.js";
+import { BaseChatModel, BindToolsInput, LangSmithParams } from "@langchain/core/language_models/chat_models";
 import { AIMessageChunk, BaseMessage } from "@langchain/core/messages";
-import { ChatGenerationChunk, ChatResult } from "@langchain/core/outputs";
 import { Runnable } from "@langchain/core/runnables";
-import { JsonSchema7Type } from "@langchain/core/utils/json_schema";
+import { ChatGenerationChunk, ChatResult } from "@langchain/core/outputs";
 import { InteropZodType } from "@langchain/core/utils/types";
-import { BaseLanguageModelInput, StructuredOutputMethodOptions } from "@langchain/core/language_models/base";
 import { SerializableSchema } from "@langchain/core/utils/standard_schema";
+import { BaseLanguageModelInput, StructuredOutputMethodOptions } from "@langchain/core/language_models/base";
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
 import { ChatModelStreamEvent } from "@langchain/core/language_models/event";
 import { ModelProfile } from "@langchain/core/language_models/profile";
 //#region src/chat_models/base.d.ts
-import ServiceTier = Gemini.ServiceTier;
-export type GooglePlatformType = "gai" | "gcp";
-export interface BaseChatGoogleParams extends BaseChatModelParams, ChatGoogleFields {
-  /**
-   * The name of the Gemini model to use.
-   *
-   * Example: "gemini-3-pro-preview"
-   */
-  model: string;
-  /**
-   * Optional. The API client implementation for making HTTP requests to the Gemini API.
-   * If not set, a default client will be used based on the runtime environment.
-   */
-  apiClient?: ApiClient;
-  /**
-   * Hostname for the API call (if this is running on GCP)
-   * Usually this is computed based on location and platformType.
-   **/
-  endpoint?: string;
-  /**
-   * Region where the LLM is stored (if this is running on GCP)
-   * Defaults to "global"
-   **/
-  location?: string;
-  /**
-   * The version of the API functions. Part of the path.
-   * Usually this is computed based on platformType.
-   **/
-  apiVersion?: string;
-  /**
-   * What platform to run the service on.
-   * If not specified, the class should determine this from other
-   * means. Either way, the platform actually used will be in
-   * the "platform" getter.
-   */
-  platformType?: GooglePlatformType;
-  /**
-   * For compatibility with Google's libraries, should this use Vertex?
-   * The "platformType" parmeter takes precedence.
-   */
-  vertexai?: boolean;
-  /**
-   * Backwards compatibility.
-   * @deprecated in favor of using `disableStreaming` or `.stream()`
-   */
-  streaming?: boolean;
-  streamUsage?: boolean;
-}
-export interface BaseChatGoogleCallOptions extends BaseChatModelCallOptions, ChatGoogleFields {}
 export declare abstract class BaseChatGoogle<CallOptions extends BaseChatGoogleCallOptions = BaseChatGoogleCallOptions> extends BaseChatModel<CallOptions, AIMessageChunk> {
   protected params: BaseChatGoogleParams & {
     credentials?: unknown;
@@ -90,7 +40,10 @@ export declare abstract class BaseChatGoogle<CallOptions extends BaseChatGoogleC
   protected get location(): string;
   protected get endpoint(): string;
   protected get publisher(): string;
-  protected get urlMethod(): string;
+  protected _apiHandler?: GoogleChatApiHandler;
+  protected _handlerParams?: GenerateContentApiHandlerParams;
+  protected syncHandlerParams(): GenerateContentApiHandlerParams;
+  protected get apiHandler(): GoogleChatApiHandler;
   /**
    * Returns the model profile for this instance's model, describing its
    * capabilities and constraints (e.g. max tokens, input/output modalities,
@@ -105,151 +58,12 @@ export declare abstract class BaseChatGoogle<CallOptions extends BaseChatGoogleC
    * ```
    */
   get profile(): ModelProfile;
-  protected buildUrlGemini(urlMethod?: string): Promise<string>;
-  protected get endpointUrl(): string;
-  protected buildUrlVertexExpress(urlMethod?: string): Promise<string>;
-  protected buildUrlVertexLocation(urlMethod?: string): Promise<string>;
-  protected buildUrlVertex(urlMethod?: string): Promise<string>;
-  protected buildUrl(urlMethod?: string): Promise<string>;
-  invocationParams(options: this["ParsedCallOptions"]): {
-    serviceTier?: ServiceTier | undefined;
-    tools?: Gemini.Tool[] | undefined;
-    toolConfig?: Gemini.Tools.ToolConfig | undefined;
-    safetySettings: {
-      category: "HARM_CATEGORY_CIVIC_INTEGRITY" | "HARM_CATEGORY_DANGEROUS" | "HARM_CATEGORY_DANGEROUS_CONTENT" | "HARM_CATEGORY_DEROGATORY" | "HARM_CATEGORY_HARASSMENT" | "HARM_CATEGORY_HATE_SPEECH" | "HARM_CATEGORY_MEDICAL" | "HARM_CATEGORY_SEXUAL" | "HARM_CATEGORY_SEXUALLY_EXPLICIT" | "HARM_CATEGORY_TOXICITY" | "HARM_CATEGORY_UNSPECIFIED" | "HARM_CATEGORY_VIOLENCE" | {
-        readonly [x: number]: string;
-        toString: {};
-        charAt: {};
-        charCodeAt: {};
-        concat: {};
-        indexOf: {};
-        lastIndexOf: {};
-        localeCompare: {};
-        match: {};
-        replace: {};
-        search: {};
-        slice: {};
-        split: {};
-        substring: {};
-        toLowerCase: {};
-        toLocaleLowerCase: {};
-        toUpperCase: {};
-        toLocaleUpperCase: {};
-        trim: {};
-        readonly length: number;
-        substr: {};
-        valueOf: {};
-        codePointAt: {};
-        includes: {};
-        endsWith: {};
-        normalize: {};
-        repeat: {};
-        startsWith: {};
-        anchor: {};
-        big: {};
-        blink: {};
-        bold: {};
-        fixed: {};
-        fontcolor: {};
-        fontsize: {};
-        italics: {};
-        link: {};
-        small: {};
-        strike: {};
-        sub: {};
-        sup: {};
-        [Symbol.iterator]: {};
-        padStart: {};
-        padEnd: {};
-        trimEnd: {};
-        trimStart: {};
-        trimLeft: {};
-        trimRight: {};
-        matchAll: {};
-        replaceAll: {};
-      };
-      threshold: "BLOCK_LOW_AND_ABOVE" | "BLOCK_MEDIUM_AND_ABOVE" | "BLOCK_NONE" | "BLOCK_ONLY_HIGH" | "HARM_BLOCK_THRESHOLD_UNSPECIFIED" | "OFF" | {
-        readonly [x: number]: string;
-        toString: {};
-        charAt: {};
-        charCodeAt: {};
-        concat: {};
-        indexOf: {};
-        lastIndexOf: {};
-        localeCompare: {};
-        match: {};
-        replace: {};
-        search: {};
-        slice: {};
-        split: {};
-        substring: {};
-        toLowerCase: {};
-        toLocaleLowerCase: {};
-        toUpperCase: {};
-        toLocaleUpperCase: {};
-        trim: {};
-        readonly length: number;
-        substr: {};
-        valueOf: {};
-        codePointAt: {};
-        includes: {};
-        endsWith: {};
-        normalize: {};
-        repeat: {};
-        startsWith: {};
-        anchor: {};
-        big: {};
-        blink: {};
-        bold: {};
-        fixed: {};
-        fontcolor: {};
-        fontsize: {};
-        italics: {};
-        link: {};
-        small: {};
-        strike: {};
-        sub: {};
-        sup: {};
-        [Symbol.iterator]: {};
-        padStart: {};
-        padEnd: {};
-        trimEnd: {};
-        trimStart: {};
-        trimLeft: {};
-        trimRight: {};
-        matchAll: {};
-        replaceAll: {};
-      };
-    }[] | undefined;
-    generationConfig: {
-      temperature: number | undefined;
-      topP: number | undefined;
-      topK: number | undefined;
-      maxOutputTokens: number | undefined;
-      presencePenalty: number | undefined;
-      frequencyPenalty: number | undefined;
-      stopSequences: string[] | undefined;
-      responseMimeType?: string | undefined;
-      responseJsonSchema?: Record<string, unknown> | JsonSchema7Type | undefined;
-      responseModalities?: Prettify<Gemini.GenerativeLanguageModality>[] | undefined;
-      candidateCount: number;
-      seed: number | undefined;
-      responseLogprobs: boolean | undefined;
-      logprobs: number | undefined;
-      enableEnhancedCivicAnswers?: boolean | undefined;
-      thinkingConfig: Gemini.ThinkingConfig | undefined;
-      speechConfig: Gemini.SpeechConfig | undefined;
-      imageConfig?: {
-        aspectRatio?: string;
-        imageSize?: string;
-      } | undefined;
-      mediaResolution?: "MEDIA_RESOLUTION_HIGH" | "MEDIA_RESOLUTION_LOW" | "MEDIA_RESOLUTION_MEDIUM" | "MEDIA_RESOLUTION_UNSPECIFIED" | (string & {}) | undefined;
-    };
-  };
+  buildUrl(urlMethod?: string): Promise<string>;
+  invocationParams(options: this["ParsedCallOptions"]): Record<string, unknown>;
   getLsParams(options: this["ParsedCallOptions"]): LangSmithParams;
-  getHeaders(options: this["ParsedCallOptions"]): HeadersInit;
+  getHeaders(options?: this["ParsedCallOptions"]): HeadersInit;
   _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
-  _streamChatModelEvents(messages: BaseMessage[], options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatModelStreamEvent>;
+  _streamChatModelEvents(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatModelStreamEvent>;
   _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
   /**
    * Bind tool-like objects to this chat model.
@@ -270,4 +84,5 @@ export declare abstract class BaseChatGoogle<CallOptions extends BaseChatGoogleC
   }>;
 }
 //#endregion
+export type { BaseChatGoogleCallOptions, BaseChatGoogleParams };
 //# sourceMappingURL=base.d.ts.map