@nocobase/plugin-ai 2.0.0-beta.6 → 2.0.0-beta.7

package/dist/node_modules/@langchain/google-genai/dist/chat_models.d.ts

@@ -1,127 +1,136 @@
-import { GenerateContentRequest, SafetySetting, Part as GenerativeAIPart, ModelParams, RequestOptions, type CachedContent, Schema } from "@google/generative-ai";
-import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
+import { GoogleGenerativeAIThinkingConfig, GoogleGenerativeAIToolType } from "./types.js";
+import * as _google_generative_ai0 from "@google/generative-ai";
+import { CachedContent, GenerateContentRequest, ModelParams, Part, RequestOptions, SafetySetting, Schema } from "@google/generative-ai";
+import { BaseChatModel, BaseChatModelCallOptions, BaseChatModelParams, LangSmithParams } from "@langchain/core/language_models/chat_models";
+import { Runnable } from "@langchain/core/runnables";
+import { InteropZodType } from "@langchain/core/utils/types";
 import { AIMessageChunk, BaseMessage } from "@langchain/core/messages";
 import { ChatGenerationChunk, ChatResult } from "@langchain/core/outputs";
-import { BaseChatModel, type BaseChatModelCallOptions, type LangSmithParams, type BaseChatModelParams } from "@langchain/core/language_models/chat_models";
 import { BaseLanguageModelInput, StructuredOutputMethodOptions } from "@langchain/core/language_models/base";
-import { Runnable } from "@langchain/core/runnables";
-import { InteropZodType } from "@langchain/core/utils/types";
-import { GoogleGenerativeAIToolType } from "./types.js";
-export type BaseMessageExamplePair = {
-    input: BaseMessage;
-    output: BaseMessage;
+import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
+import { ModelProfile } from "@langchain/core/language_models/profile";
+
+//#region src/chat_models.d.ts
+type BaseMessageExamplePair = {
+  input: BaseMessage;
+  output: BaseMessage;
 };
-export interface GoogleGenerativeAIChatCallOptions extends BaseChatModelCallOptions {
-    tools?: GoogleGenerativeAIToolType[];
-    /**
-     * Allowed functions to call when the mode is "any".
-     * If empty, any one of the provided functions are called.
-     */
-    allowedFunctionNames?: string[];
-    /**
-     * Whether or not to include usage data, like token counts
-     * in the streamed response chunks.
-     * @default true
-     */
-    streamUsage?: boolean;
-    /**
-     * JSON schema to be returned by the model.
-     */
-    responseSchema?: Schema;
+interface GoogleGenerativeAIChatCallOptions extends BaseChatModelCallOptions {
+  tools?: GoogleGenerativeAIToolType[];
+  /**
+   * Allowed functions to call when the mode is "any".
+   * If empty, any one of the provided functions are called.
+   */
+  allowedFunctionNames?: string[];
+  /**
+   * Whether or not to include usage data, like token counts
+   * in the streamed response chunks.
+   * @default true
+   */
+  streamUsage?: boolean;
+  /**
+   * JSON schema to be returned by the model.
+   */
+  responseSchema?: Schema;
 }
 /**
  * An interface defining the input to the ChatGoogleGenerativeAI class.
  */
-export interface GoogleGenerativeAIChatInput extends BaseChatModelParams, Pick<GoogleGenerativeAIChatCallOptions, "streamUsage"> {
-    /**
-     * Model Name to use
-     *
-     * Note: The format must follow the pattern - `{model}`
-     */
-    model: string;
-    /**
-     * Controls the randomness of the output.
-     *
-     * Values can range from [0.0,2.0], inclusive. A value closer to 2.0
-     * will produce responses that are more varied and creative, while
-     * a value closer to 0.0 will typically result in less surprising
-     * responses from the model.
-     *
-     * Note: The default value varies by model
-     */
-    temperature?: number;
-    /**
-     * Maximum number of tokens to generate in the completion.
-     */
-    maxOutputTokens?: number;
-    /**
-     * Top-p changes how the model selects tokens for output.
-     *
-     * Tokens are selected from most probable to least until the sum
-     * of their probabilities equals the top-p value.
-     *
-     * For example, if tokens A, B, and C have a probability of
-     * .3, .2, and .1 and the top-p value is .5, then the model will
-     * select either A or B as the next token (using temperature).
-     *
-     * Note: The default value varies by model
-     */
-    topP?: number;
-    /**
-     * Top-k changes how the model selects tokens for output.
-     *
-     * A top-k of 1 means the selected token is the most probable among
-     * all tokens in the model's vocabulary (also called greedy decoding),
-     * while a top-k of 3 means that the next token is selected from
-     * among the 3 most probable tokens (using temperature).
-     *
-     * Note: The default value varies by model
-     */
-    topK?: number;
-    /**
-     * The set of character sequences (up to 5) that will stop output generation.
-     * If specified, the API will stop at the first appearance of a stop
-     * sequence.
-     *
-     * Note: The stop sequence will not be included as part of the response.
-     * Note: stopSequences is only supported for Gemini models
-     */
-    stopSequences?: string[];
-    /**
-     * A list of unique `SafetySetting` instances for blocking unsafe content. The API will block
-     * any prompts and responses that fail to meet the thresholds set by these settings. If there
-     * is no `SafetySetting` for a given `SafetyCategory` provided in the list, the API will use
-     * the default safety setting for that category.
-     */
-    safetySettings?: SafetySetting[];
-    /**
-     * Google API key to use
-     */
-    apiKey?: string;
-    /**
-     * Google API version to use
-     */
-    apiVersion?: string;
-    /**
-     * Google API base URL to use
-     */
-    baseUrl?: string;
-    /** Whether to stream the results or not */
-    streaming?: boolean;
-    /**
-     * Whether or not to force the model to respond with JSON.
-     * Available for `gemini-1.5` models and later.
-     * @default false
-     */
-    json?: boolean;
-    /**
-     * Whether or not model supports system instructions.
-     * The following models support system instructions:
-     * - All Gemini 1.5 Pro model versions
-     * - All Gemini 1.5 Flash model versions
-     * - Gemini 1.0 Pro version gemini-1.0-pro-002
-     */
-    convertSystemMessageToHumanContent?: boolean | undefined;
+interface GoogleGenerativeAIChatInput extends BaseChatModelParams, Pick<GoogleGenerativeAIChatCallOptions, "streamUsage"> {
+  /**
+   * Model Name to use
+   *
+   * Note: The format must follow the pattern - `{model}`
+   */
+  model: string;
+  /**
+   * Controls the randomness of the output.
+   *
+   * Values can range from [0.0,2.0], inclusive. A value closer to 2.0
+   * will produce responses that are more varied and creative, while
+   * a value closer to 0.0 will typically result in less surprising
+   * responses from the model.
+   *
+   * Note: The default value varies by model
+   */
+  temperature?: number;
+  /**
+   * Maximum number of tokens to generate in the completion.
+   */
+  maxOutputTokens?: number;
+  /**
+   * Top-p changes how the model selects tokens for output.
+   *
+   * Tokens are selected from most probable to least until the sum
+   * of their probabilities equals the top-p value.
+   *
+   * For example, if tokens A, B, and C have a probability of
+   * .3, .2, and .1 and the top-p value is .5, then the model will
+   * select either A or B as the next token (using temperature).
+   *
+   * Note: The default value varies by model
+   */
+  topP?: number;
+  /**
+   * Top-k changes how the model selects tokens for output.
+   *
+   * A top-k of 1 means the selected token is the most probable among
+   * all tokens in the model's vocabulary (also called greedy decoding),
+   * while a top-k of 3 means that the next token is selected from
+   * among the 3 most probable tokens (using temperature).
+   *
+   * Note: The default value varies by model
+   */
+  topK?: number;
+  /**
+   * The set of character sequences (up to 5) that will stop output generation.
+   * If specified, the API will stop at the first appearance of a stop
+   * sequence.
+   *
+   * Note: The stop sequence will not be included as part of the response.
+   * Note: stopSequences is only supported for Gemini models
+   */
+  stopSequences?: string[];
+  /**
+   * A list of unique `SafetySetting` instances for blocking unsafe content. The API will block
+   * any prompts and responses that fail to meet the thresholds set by these settings. If there
+   * is no `SafetySetting` for a given `SafetyCategory` provided in the list, the API will use
+   * the default safety setting for that category.
+   */
+  safetySettings?: SafetySetting[];
+  /**
+   * Google API key to use
+   */
+  apiKey?: string;
+  /**
+   * Google API version to use
+   */
+  apiVersion?: string;
+  /**
+   * Google API base URL to use
+   */
+  baseUrl?: string;
+  /** Whether to stream the results or not */
+  streaming?: boolean;
+  /**
+   * Whether or not to force the model to respond with JSON.
+   * Available for `gemini-1.5` models and later.
+   * @default false
+   */
+  json?: boolean;
+  /**
+   * Whether or not model supports system instructions.
+   * The following models support system instructions:
+   * - All Gemini 1.5 Pro model versions
+   * - All Gemini 1.5 Flash model versions
+   * - Gemini 1.0 Pro version gemini-1.0-pro-002
+   */
+  convertSystemMessageToHumanContent?: boolean | undefined;
+  /**
+   * Optional. Config for thinking features. An error will be returned if this
+   * field is set for models that don't support thinking.
+   */
+  thinkingConfig?: GoogleGenerativeAIThinkingConfig;
 }
 /**
  * Google Generative AI chat model integration.
@@ -498,45 +507,75 @@ export interface GoogleGenerativeAIChatInput extends BaseChatModelParams, Pick<G
  *
  * <br />
  */
-export declare class ChatGoogleGenerativeAI extends BaseChatModel<GoogleGenerativeAIChatCallOptions, AIMessageChunk> implements GoogleGenerativeAIChatInput {
-    static lc_name(): string;
-    lc_serializable: boolean;
-    get lc_secrets(): {
-        [key: string]: string;
-    } | undefined;
-    lc_namespace: string[];
-    get lc_aliases(): {
-        apiKey: string;
-    };
-    model: string;
-    temperature?: number;
-    maxOutputTokens?: number;
-    topP?: number;
-    topK?: number;
-    stopSequences: string[];
-    safetySettings?: SafetySetting[];
-    apiKey?: string;
-    streaming: boolean;
-    json?: boolean;
-    streamUsage: boolean;
-    convertSystemMessageToHumanContent: boolean | undefined;
-    private client;
-    get _isMultimodalModel(): boolean;
-    constructor(fields: GoogleGenerativeAIChatInput);
-    useCachedContent(cachedContent: CachedContent, modelParams?: ModelParams, requestOptions?: RequestOptions): void;
-    get useSystemInstruction(): boolean;
-    get computeUseSystemInstruction(): boolean;
-    getLsParams(options: this["ParsedCallOptions"]): LangSmithParams;
-    _combineLLMOutput(): never[];
-    _llmType(): string;
-    bindTools(tools: GoogleGenerativeAIToolType[], kwargs?: Partial<GoogleGenerativeAIChatCallOptions>): Runnable<BaseLanguageModelInput, AIMessageChunk, GoogleGenerativeAIChatCallOptions>;
-    invocationParams(options?: this["ParsedCallOptions"]): Omit<GenerateContentRequest, "contents">;
-    _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
-    _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
-    completionWithRetry(request: string | GenerateContentRequest | (string | GenerativeAIPart)[], options?: this["ParsedCallOptions"]): Promise<import("@google/generative-ai").GenerateContentResult>;
-    withStructuredOutput<RunOutput extends Record<string, any> = Record<string, any>>(outputSchema: InteropZodType<RunOutput> | Record<string, any>, config?: StructuredOutputMethodOptions<false>): Runnable<BaseLanguageModelInput, RunOutput>;
-    withStructuredOutput<RunOutput extends Record<string, any> = Record<string, any>>(outputSchema: InteropZodType<RunOutput> | Record<string, any>, config?: StructuredOutputMethodOptions<true>): Runnable<BaseLanguageModelInput, {
-        raw: BaseMessage;
-        parsed: RunOutput;
-    }>;
+declare class ChatGoogleGenerativeAI extends BaseChatModel<GoogleGenerativeAIChatCallOptions, AIMessageChunk> implements GoogleGenerativeAIChatInput {
+  static lc_name(): string;
+  lc_serializable: boolean;
+  get lc_secrets(): {
+    [key: string]: string;
+  } | undefined;
+  lc_namespace: string[];
+  get lc_aliases(): {
+    apiKey: string;
+  };
+  model: string;
+  temperature?: number; // default value chosen based on model
+  maxOutputTokens?: number;
+  topP?: number; // default value chosen based on model
+  topK?: number; // default value chosen based on model
+  stopSequences: string[];
+  safetySettings?: SafetySetting[];
+  apiKey?: string;
+  streaming: boolean;
+  json?: boolean;
+  streamUsage: boolean;
+  convertSystemMessageToHumanContent: boolean | undefined;
+  thinkingConfig?: GoogleGenerativeAIThinkingConfig;
+  private client;
+  get _isMultimodalModel(): boolean;
+  constructor(fields: GoogleGenerativeAIChatInput);
+  useCachedContent(cachedContent: CachedContent, modelParams?: ModelParams, requestOptions?: RequestOptions): void;
+  get useSystemInstruction(): boolean;
+  get computeUseSystemInstruction(): boolean;
+  getLsParams(options: this["ParsedCallOptions"]): LangSmithParams;
+  _combineLLMOutput(): never[];
+  _llmType(): string;
+  bindTools(tools: GoogleGenerativeAIToolType[], kwargs?: Partial<GoogleGenerativeAIChatCallOptions>): Runnable<BaseLanguageModelInput, AIMessageChunk, GoogleGenerativeAIChatCallOptions>;
+  invocationParams(options?: this["ParsedCallOptions"]): Omit<GenerateContentRequest, "contents">;
+  _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
+  _streamResponseChunks(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
+  completionWithRetry(request: string | GenerateContentRequest | (string | Part)[], options?: this["ParsedCallOptions"]): Promise<_google_generative_ai0.GenerateContentResult>;
+  /**
+   * Return profiling information for the model.
+   *
+   * Provides information about the model's capabilities and constraints,
+   * including token limits, multimodal support, and advanced features like
+   * tool calling and structured output.
+   *
+   * @returns {ModelProfile} An object describing the model's capabilities and constraints
+   *
+   * @example
+   * ```typescript
+   * const model = new ChatGoogleGenerativeAI({ model: "gemini-1.5-flash" });
+   * const profile = model.profile;
+   * console.log(profile.maxInputTokens); // 2000000
+   * console.log(profile.imageInputs); // true
+   * ```
+   */
+  get profile(): ModelProfile;
+  withStructuredOutput<
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  RunOutput extends Record<string, any> = Record<string, any>>(outputSchema: InteropZodType<RunOutput>
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  | Record<string, any>, config?: StructuredOutputMethodOptions<false>): Runnable<BaseLanguageModelInput, RunOutput>;
+  withStructuredOutput<
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  RunOutput extends Record<string, any> = Record<string, any>>(outputSchema: InteropZodType<RunOutput>
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  | Record<string, any>, config?: StructuredOutputMethodOptions<true>): Runnable<BaseLanguageModelInput, {
+    raw: BaseMessage;
+    parsed: RunOutput;
+  }>;
 }
+//#endregion
+export { BaseMessageExamplePair, ChatGoogleGenerativeAI, GoogleGenerativeAIChatCallOptions, GoogleGenerativeAIChatInput };
+//# sourceMappingURL=chat_models.d.ts.map