@langchain/google-genai 0.2.17 → 1.0.0

package/dist/chat_models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat_models.js","names":["fields: GoogleGenerativeAIChatInput","GenerativeAI","cachedContent: CachedContent","modelParams?: ModelParams","requestOptions?: RequestOptions","options: this[\"ParsedCallOptions\"]","tools: GoogleGenerativeAIToolType[]","kwargs?: Partial<GoogleGenerativeAIChatCallOptions>","options?: this[\"ParsedCallOptions\"]","messages: BaseMessage[]","runManager?: CallbackManagerForLLMRun","tokenUsage: TokenUsage","finalChunks: Record<number, ChatGenerationChunk>","usageMetadata: UsageMetadata | undefined","stream","request: string | GenerateContentRequest | (string | GenerativeAIPart)[]","e: any","outputSchema:\n      | InteropZodType<RunOutput>\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      | Record<string, any>","config?: StructuredOutputMethodOptions<boolean>","schema: InteropZodType<RunOutput> | Record<string, any>","outputParser: BaseLLMOutputParser<RunOutput>","tools: GoogleGenerativeAIFunctionDeclarationsTool[]","geminiFunctionDefinition: GenerativeAIFunctionDeclaration","input: any","config"],"sources":["../src/chat_models.ts"],"sourcesContent":["import {\n  GenerativeModel,\n  GoogleGenerativeAI as GenerativeAI,\n  FunctionDeclarationsTool as GoogleGenerativeAIFunctionDeclarationsTool,\n  FunctionDeclaration as GenerativeAIFunctionDeclaration,\n  type FunctionDeclarationSchema as GenerativeAIFunctionDeclarationSchema,\n  GenerateContentRequest,\n  SafetySetting,\n  Part as GenerativeAIPart,\n  ModelParams,\n  RequestOptions,\n  type CachedContent,\n  Schema,\n} from \"@google/generative-ai\";\nimport { CallbackManagerForLLMRun } from \"@langchain/core/callbacks/manager\";\nimport {\n  AIMessageChunk,\n  BaseMessage,\n  UsageMetadata,\n} from \"@langchain/core/messages\";\nimport { ChatGenerationChunk, ChatResult } from \"@langchain/core/outputs\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\nimport {\n  BaseChatModel,\n  type BaseChatModelCallOptions,\n  type LangSmithParams,\n  type BaseChatModelParams,\n} from \"@langchain/core/language_models/chat_models\";\nimport { NewTokenIndices } from \"@langchain/core/callbacks/base\";\nimport {\n  BaseLanguageModelInput,\n  StructuredOutputMethodOptions,\n} from \"@langchain/core/language_models/base\";\nimport {\n  Runnable,\n  RunnablePassthrough,\n  RunnableSequence,\n} from \"@langchain/core/runnables\";\nimport {\n  InferInteropZodOutput,\n  InteropZodType,\n  isInteropZodSchema,\n} from \"@langchain/core/utils/types\";\nimport {\n  BaseLLMOutputParser,\n  JsonOutputParser,\n} from \"@langchain/core/output_parsers\";\nimport {\n  schemaToGenerativeAIParameters,\n  removeAdditionalProperties,\n} from \"./utils/zod_to_genai_parameters.js\";\nimport {\n  convertBaseMessagesToContent,\n  convertResponseContentToChatGenerationChunk,\n  mapGenerateContentResultToChatResult,\n} from \"./utils/common.js\";\nimport { GoogleGenerativeAIToolsOutputParser } from \"./output_parsers.js\";\nimport { GoogleGenerativeAIToolType } from \"./types.js\";\nimport { convertToolsToGenAI } from \"./utils/tools.js\";\n\ninterface TokenUsage {\n  completionTokens?: number;\n  promptTokens?: number;\n  totalTokens?: number;\n}\n\nexport type BaseMessageExamplePair = {\n  input: BaseMessage;\n  output: BaseMessage;\n};\n\nexport interface GoogleGenerativeAIChatCallOptions\n  extends BaseChatModelCallOptions {\n  tools?: GoogleGenerativeAIToolType[];\n  /**\n   * Allowed functions to call when the mode is \"any\".\n   * If empty, any one of the provided functions are called.\n   */\n  allowedFunctionNames?: string[];\n  /**\n   * Whether or not to include usage data, like token counts\n   * in the streamed response chunks.\n   * @default true\n   */\n  streamUsage?: boolean;\n\n  /**\n   * JSON schema to be returned by the model.\n   */\n  responseSchema?: Schema;\n}\n\n/**\n * An interface defining the input to the ChatGoogleGenerativeAI class.\n */\nexport interface GoogleGenerativeAIChatInput\n  extends BaseChatModelParams,\n    Pick<GoogleGenerativeAIChatCallOptions, \"streamUsage\"> {\n  /**\n   * Model Name to use\n   *\n   * Note: The format must follow the pattern - `{model}`\n   */\n  model: string;\n\n  /**\n   * Controls the randomness of the output.\n   *\n   * Values can range from [0.0,2.0], inclusive. A value closer to 2.0\n   * will produce responses that are more varied and creative, while\n   * a value closer to 0.0 will typically result in less surprising\n   * responses from the model.\n   *\n   * Note: The default value varies by model\n   */\n  temperature?: number;\n\n  /**\n   * Maximum number of tokens to generate in the completion.\n   */\n  maxOutputTokens?: number;\n\n  /**\n   * Top-p changes how the model selects tokens for output.\n   *\n   * Tokens are selected from most probable to least until the sum\n   * of their probabilities equals the top-p value.\n   *\n   * For example, if tokens A, B, and C have a probability of\n   * .3, .2, and .1 and the top-p value is .5, then the model will\n   * select either A or B as the next token (using temperature).\n   *\n   * Note: The default value varies by model\n   */\n  topP?: number;\n\n  /**\n   * Top-k changes how the model selects tokens for output.\n   *\n   * A top-k of 1 means the selected token is the most probable among\n   * all tokens in the model's vocabulary (also called greedy decoding),\n   * while a top-k of 3 means that the next token is selected from\n   * among the 3 most probable tokens (using temperature).\n   *\n   * Note: The default value varies by model\n   */\n  topK?: number;\n\n  /**\n   * The set of character sequences (up to 5) that will stop output generation.\n   * If specified, the API will stop at the first appearance of a stop\n   * sequence.\n   *\n   * Note: The stop sequence will not be included as part of the response.\n   * Note: stopSequences is only supported for Gemini models\n   */\n  stopSequences?: string[];\n\n  /**\n   * A list of unique `SafetySetting` instances for blocking unsafe content. The API will block\n   * any prompts and responses that fail to meet the thresholds set by these settings. If there\n   * is no `SafetySetting` for a given `SafetyCategory` provided in the list, the API will use\n   * the default safety setting for that category.\n   */\n  safetySettings?: SafetySetting[];\n\n  /**\n   * Google API key to use\n   */\n  apiKey?: string;\n\n  /**\n   * Google API version to use\n   */\n  apiVersion?: string;\n\n  /**\n   * Google API base URL to use\n   */\n  baseUrl?: string;\n\n  /** Whether to stream the results or not */\n  streaming?: boolean;\n\n  /**\n   * Whether or not to force the model to respond with JSON.\n   * Available for `gemini-1.5` models and later.\n   * @default false\n   */\n  json?: boolean;\n\n  /**\n   * Whether or not model supports system instructions.\n   * The following models support system instructions:\n   * - All Gemini 1.5 Pro model versions\n   * - All Gemini 1.5 Flash model versions\n   * - Gemini 1.0 Pro version gemini-1.0-pro-002\n   */\n  convertSystemMessageToHumanContent?: boolean | undefined;\n}\n\n/**\n * Google Generative AI chat model integration.\n *\n * Setup:\n * Install `@langchain/google-genai` and set an environment variable named `GOOGLE_API_KEY`.\n *\n * ```bash\n * npm install @langchain/google-genai\n * export GOOGLE_API_KEY=\"your-api-key\"\n * ```\n *\n * ## [Constructor args](https://api.js.langchain.com/classes/langchain_google_genai.ChatGoogleGenerativeAI.html#constructor)\n *\n * ## [Runtime args](https://api.js.langchain.com/interfaces/langchain_google_genai.GoogleGenerativeAIChatCallOptions.html)\n *\n * Runtime args can be passed as the second argument to any of the base runnable methods `.invoke`. `.stream`, `.batch`, etc.\n * They can also be passed via `.withConfig`, or the second arg in `.bindTools`, like shown in the examples below:\n *\n * ```typescript\n * // When calling `.withConfig`, call options should be passed via the first argument\n * const llmWithArgsBound = llm.withConfig({\n *   stop: [\"\\n\"],\n * });\n *\n * // When calling `.bindTools`, call options should be passed via the second argument\n * const llmWithTools = llm.bindTools(\n *   [...],\n *   {\n *     stop: [\"\\n\"],\n *   }\n * );\n * ```\n *\n * ## Examples\n *\n * <details open>\n * <summary><strong>Instantiate</strong></summary>\n *\n * ```typescript\n * import { ChatGoogleGenerativeAI } from '@langchain/google-genai';\n *\n * const llm = new ChatGoogleGenerativeAI({\n *   model: \"gemini-1.5-flash\",\n *   temperature: 0,\n *   maxRetries: 2,\n *   // apiKey: \"...\",\n *   // other params...\n * });\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Invoking</strong></summary>\n *\n * ```typescript\n * const input = `Translate \"I love programming\" into French.`;\n *\n * // Models also accept a list of chat messages or a formatted prompt\n * const result = await llm.invoke(input);\n * console.log(result);\n * ```\n *\n * ```txt\n * AIMessage {\n *   \"content\": \"There are a few ways to translate \\\"I love programming\\\" into French, depending on the level of formality and nuance you want to convey:\\n\\n**Formal:**\\n\\n* **J'aime la programmation.** (This is the most literal and formal translation.)\\n\\n**Informal:**\\n\\n* **J'adore programmer.** (This is a more enthusiastic and informal translation.)\\n* **J'aime beaucoup programmer.** (This is a slightly less enthusiastic but still informal translation.)\\n\\n**More specific:**\\n\\n* **J'aime beaucoup coder.** (This specifically refers to writing code.)\\n* **J'aime beaucoup développer des logiciels.** (This specifically refers to developing software.)\\n\\nThe best translation will depend on the context and your intended audience. \\n\",\n *   \"response_metadata\": {\n *     \"finishReason\": \"STOP\",\n *     \"index\": 0,\n *     \"safetyRatings\": [\n *       {\n *         \"category\": \"HARM_CATEGORY_SEXUALLY_EXPLICIT\",\n *         \"probability\": \"NEGLIGIBLE\"\n *       },\n *       {\n *         \"category\": \"HARM_CATEGORY_HATE_SPEECH\",\n *         \"probability\": \"NEGLIGIBLE\"\n *       },\n *       {\n *         \"category\": \"HARM_CATEGORY_HARASSMENT\",\n *         \"probability\": \"NEGLIGIBLE\"\n *       },\n *       {\n *         \"category\": \"HARM_CATEGORY_DANGEROUS_CONTENT\",\n *         \"probability\": \"NEGLIGIBLE\"\n *       }\n *     ]\n *   },\n *   \"usage_metadata\": {\n *     \"input_tokens\": 10,\n *     \"output_tokens\": 149,\n *     \"total_tokens\": 159\n *   }\n * }\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Streaming Chunks</strong></summary>\n *\n * ```typescript\n * for await (const chunk of await llm.stream(input)) {\n *   console.log(chunk);\n * }\n * ```\n *\n * ```txt\n * AIMessageChunk {\n *   \"content\": \"There\",\n *   \"response_metadata\": {\n *     \"index\": 0\n *   }\n *   \"usage_metadata\": {\n *     \"input_tokens\": 10,\n *     \"output_tokens\": 1,\n *     \"total_tokens\": 11\n *   }\n * }\n * AIMessageChunk {\n *   \"content\": \" are a few ways to translate \\\"I love programming\\\" into French, depending on\",\n * }\n * AIMessageChunk {\n *   \"content\": \" the level of formality and nuance you want to convey:\\n\\n**Formal:**\\n\\n\",\n * }\n * AIMessageChunk {\n *   \"content\": \"* **J'aime la programmation.** (This is the most literal and formal translation.)\\n\\n**Informal:**\\n\\n* **J'adore programmer.** (This\",\n * }\n * AIMessageChunk {\n *   \"content\": \" is a more enthusiastic and informal translation.)\\n* **J'aime beaucoup programmer.** (This is a slightly less enthusiastic but still informal translation.)\\n\\n**More\",\n * }\n * AIMessageChunk {\n *   \"content\": \" specific:**\\n\\n* **J'aime beaucoup coder.** (This specifically refers to writing code.)\\n* **J'aime beaucoup développer des logiciels.** (This specifically refers to developing software.)\\n\\nThe best translation will depend on the context and\",\n * }\n * AIMessageChunk {\n *   \"content\": \" your intended audience. \\n\",\n * }\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Aggregate Streamed Chunks</strong></summary>\n *\n * ```typescript\n * import { AIMessageChunk } from '@langchain/core/messages';\n * import { concat } from '@langchain/core/utils/stream';\n *\n * const stream = await llm.stream(input);\n * let full: AIMessageChunk | undefined;\n * for await (const chunk of stream) {\n *   full = !full ? chunk : concat(full, chunk);\n * }\n * console.log(full);\n * ```\n *\n * ```txt\n * AIMessageChunk {\n *   \"content\": \"There are a few ways to translate \\\"I love programming\\\" into French, depending on the level of formality and nuance you want to convey:\\n\\n**Formal:**\\n\\n* **J'aime la programmation.** (This is the most literal and formal translation.)\\n\\n**Informal:**\\n\\n* **J'adore programmer.** (This is a more enthusiastic and informal translation.)\\n* **J'aime beaucoup programmer.** (This is a slightly less enthusiastic but still informal translation.)\\n\\n**More specific:**\\n\\n* **J'aime beaucoup coder.** (This specifically refers to writing code.)\\n* **J'aime beaucoup développer des logiciels.** (This specifically refers to developing software.)\\n\\nThe best translation will depend on the context and your intended audience. \\n\",\n *   \"usage_metadata\": {\n *     \"input_tokens\": 10,\n *     \"output_tokens\": 277,\n *     \"total_tokens\": 287\n *   }\n * }\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Bind tools</strong></summary>\n *\n * ```typescript\n * import { z } from 'zod';\n *\n * const GetWeather = {\n *   name: \"GetWeather\",\n *   description: \"Get the current weather in a given location\",\n *   schema: z.object({\n *     location: z.string().describe(\"The city and state, e.g. San Francisco, CA\")\n *   }),\n * }\n *\n * const GetPopulation = {\n *   name: \"GetPopulation\",\n *   description: \"Get the current population in a given location\",\n *   schema: z.object({\n *     location: z.string().describe(\"The city and state, e.g. San Francisco, CA\")\n *   }),\n * }\n *\n * const llmWithTools = llm.bindTools([GetWeather, GetPopulation]);\n * const aiMsg = await llmWithTools.invoke(\n *   \"Which city is hotter today and which is bigger: LA or NY?\"\n * );\n * console.log(aiMsg.tool_calls);\n * ```\n *\n * ```txt\n * [\n *   {\n *     name: 'GetWeather',\n *     args: { location: 'Los Angeles, CA' },\n *     type: 'tool_call'\n *   },\n *   {\n *     name: 'GetWeather',\n *     args: { location: 'New York, NY' },\n *     type: 'tool_call'\n *   },\n *   {\n *     name: 'GetPopulation',\n *     args: { location: 'Los Angeles, CA' },\n *     type: 'tool_call'\n *   },\n *   {\n *     name: 'GetPopulation',\n *     args: { location: 'New York, NY' },\n *     type: 'tool_call'\n *   }\n * ]\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Structured Output</strong></summary>\n *\n * ```typescript\n * const Joke = z.object({\n *   setup: z.string().describe(\"The setup of the joke\"),\n *   punchline: z.string().describe(\"The punchline to the joke\"),\n *   rating: z.number().optional().describe(\"How funny the joke is, from 1 to 10\")\n * }).describe('Joke to tell user.');\n *\n * const structuredLlm = llm.withStructuredOutput(Joke, { name: \"Joke\" });\n * const jokeResult = await structuredLlm.invoke(\"Tell me a joke about cats\");\n * console.log(jokeResult);\n * ```\n *\n * ```txt\n * {\n *   setup: \"Why don\\\\'t cats play poker?\",\n *   punchline: \"Why don\\\\'t cats play poker? Because they always have an ace up their sleeve!\"\n * }\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Multimodal</strong></summary>\n *\n * ```typescript\n * import { HumanMessage } from '@langchain/core/messages';\n *\n * const imageUrl = \"https://example.com/image.jpg\";\n * const imageData = await fetch(imageUrl).then(res => res.arrayBuffer());\n * const base64Image = Buffer.from(imageData).toString('base64');\n *\n * const message = new HumanMessage({\n *   content: [\n *     { type: \"text\", text: \"describe the weather in this image\" },\n *     {\n *       type: \"image_url\",\n *       image_url: { url: `data:image/jpeg;base64,${base64Image}` },\n *     },\n *   ]\n * });\n *\n * const imageDescriptionAiMsg = await llm.invoke([message]);\n * console.log(imageDescriptionAiMsg.content);\n * ```\n *\n * ```txt\n * The weather in the image appears to be clear and sunny. The sky is mostly blue with a few scattered white clouds, indicating fair weather. The bright sunlight is casting shadows on the green, grassy hill, suggesting it is a pleasant day with good visibility. There are no signs of rain or stormy conditions.\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Usage Metadata</strong></summary>\n *\n * ```typescript\n * const aiMsgForMetadata = await llm.invoke(input);\n * console.log(aiMsgForMetadata.usage_metadata);\n * ```\n *\n * ```txt\n * { input_tokens: 10, output_tokens: 149, total_tokens: 159 }\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Response Metadata</strong></summary>\n *\n * ```typescript\n * const aiMsgForResponseMetadata = await llm.invoke(input);\n * console.log(aiMsgForResponseMetadata.response_metadata);\n * ```\n *\n * ```txt\n * {\n *   finishReason: 'STOP',\n *   index: 0,\n *   safetyRatings: [\n *     {\n *       category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',\n *       probability: 'NEGLIGIBLE'\n *     },\n *     {\n *       category: 'HARM_CATEGORY_HATE_SPEECH',\n *       probability: 'NEGLIGIBLE'\n *     },\n *     { category: 'HARM_CATEGORY_HARASSMENT', probability: 'NEGLIGIBLE' },\n *     {\n *       category: 'HARM_CATEGORY_DANGEROUS_CONTENT',\n *       probability: 'NEGLIGIBLE'\n *     }\n *   ]\n * }\n * ```\n * </details>\n *\n * <br />\n *\n * <details>\n * <summary><strong>Document Messages</strong></summary>\n *\n * This example will show you how to pass documents such as PDFs to Google\n * Generative AI through messages.\n *\n * ```typescript\n * const pdfPath = \"/Users/my_user/Downloads/invoice.pdf\";\n * const pdfBase64 = await fs.readFile(pdfPath, \"base64\");\n *\n * const response = await llm.invoke([\n *   [\"system\", \"Use the provided documents to answer the question\"],\n *   [\n *     \"user\",\n *     [\n *       {\n *         type: \"application/pdf\", // If the `type` field includes a single slash (`/`), it will be treated as inline data.\n *         data: pdfBase64,\n *       },\n *       {\n *         type: \"text\",\n *         text: \"Summarize the contents of this PDF\",\n *       },\n *     ],\n *   ],\n * ]);\n *\n * console.log(response.content);\n * ```\n *\n * ```txt\n * This is a billing invoice from Twitter Developers for X API Basic Access. The transaction date is January 7, 2025,\n * and the amount is $194.34, which has been paid. The subscription period is from January 7, 2025 21:02 to February 7, 2025 00:00 (UTC).\n * The tax is $0.00, with a tax rate of 0%. The total amount is $194.34. The payment was made using a Visa card ending in 7022,\n * expiring in 12/2026. The billing address is Brace Sproul, 1234 Main Street, San Francisco, CA, US 94103. The company being billed is\n * X Corp, located at 865 FM 1209 Building 2, Bastrop, TX, US 78602. Terms and conditions apply.\n * ```\n * </details>\n *\n * <br />\n */\nexport class ChatGoogleGenerativeAI\n  extends BaseChatModel<GoogleGenerativeAIChatCallOptions, AIMessageChunk>\n  implements GoogleGenerativeAIChatInput\n{\n  static lc_name() {\n    return \"ChatGoogleGenerativeAI\";\n  }\n\n  lc_serializable = true;\n\n  get lc_secrets(): { [key: string]: string } | undefined {\n    return {\n      apiKey: \"GOOGLE_API_KEY\",\n    };\n  }\n\n  lc_namespace = [\"langchain\", \"chat_models\", \"google_genai\"];\n\n  get lc_aliases() {\n    return {\n      apiKey: \"google_api_key\",\n    };\n  }\n\n  model: string;\n\n  temperature?: number; // default value chosen based on model\n\n  maxOutputTokens?: number;\n\n  topP?: number; // default value chosen based on model\n\n  topK?: number; // default value chosen based on model\n\n  stopSequences: string[] = [];\n\n  safetySettings?: SafetySetting[];\n\n  apiKey?: string;\n\n  streaming = false;\n\n  json?: boolean;\n\n  streamUsage = true;\n\n  convertSystemMessageToHumanContent: boolean | undefined;\n\n  private client: GenerativeModel;\n\n  get _isMultimodalModel() {\n    return (\n      this.model.includes(\"vision\") ||\n      this.model.startsWith(\"gemini-1.5\") ||\n      this.model.startsWith(\"gemini-2\") ||\n      (this.model.startsWith(\"gemma-3-\") &&\n        !this.model.startsWith(\"gemma-3-1b\")) // gemma-3 models are multimodal(but gemma-3n-* and gemma-3-1b are not)\n    );\n  }\n\n  constructor(fields: GoogleGenerativeAIChatInput) {\n    super(fields);\n\n    this.model = fields.model.replace(/^models\\//, \"\");\n\n    this.maxOutputTokens = fields.maxOutputTokens ?? this.maxOutputTokens;\n\n    if (this.maxOutputTokens && this.maxOutputTokens < 0) {\n      throw new Error(\"`maxOutputTokens` must be a positive integer\");\n    }\n\n    this.temperature = fields.temperature ?? this.temperature;\n    if (this.temperature && (this.temperature < 0 || this.temperature > 2)) {\n      throw new Error(\"`temperature` must be in the range of [0.0,2.0]\");\n    }\n\n    this.topP = fields.topP ?? this.topP;\n    if (this.topP && this.topP < 0) {\n      throw new Error(\"`topP` must be a positive integer\");\n    }\n\n    if (this.topP && this.topP > 1) {\n      throw new Error(\"`topP` must be below 1.\");\n    }\n\n    this.topK = fields.topK ?? this.topK;\n    if (this.topK && this.topK < 0) {\n      throw new Error(\"`topK` must be a positive integer\");\n    }\n\n    this.stopSequences = fields.stopSequences ?? this.stopSequences;\n\n    this.apiKey = fields.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n    if (!this.apiKey) {\n      throw new Error(\n        \"Please set an API key for Google GenerativeAI \" +\n          \"in the environment variable GOOGLE_API_KEY \" +\n          \"or in the `apiKey` field of the \" +\n          \"ChatGoogleGenerativeAI constructor\"\n      );\n    }\n\n    this.safetySettings = fields.safetySettings ?? this.safetySettings;\n    if (this.safetySettings && this.safetySettings.length > 0) {\n      const safetySettingsSet = new Set(\n        this.safetySettings.map((s) => s.category)\n      );\n      if (safetySettingsSet.size !== this.safetySettings.length) {\n        throw new Error(\n          \"The categories in `safetySettings` array must be unique\"\n        );\n      }\n    }\n\n    this.streaming = fields.streaming ?? this.streaming;\n    this.json = fields.json;\n\n    this.client = new GenerativeAI(this.apiKey).getGenerativeModel(\n      {\n        model: this.model,\n        safetySettings: this.safetySettings as SafetySetting[],\n        generationConfig: {\n          stopSequences: this.stopSequences,\n          maxOutputTokens: this.maxOutputTokens,\n          temperature: this.temperature,\n          topP: this.topP,\n          topK: this.topK,\n          ...(this.json ? { responseMimeType: \"application/json\" } : {}),\n        },\n      },\n      {\n        apiVersion: fields.apiVersion,\n        baseUrl: fields.baseUrl,\n      }\n    );\n    this.streamUsage = fields.streamUsage ?? this.streamUsage;\n  }\n\n  useCachedContent(\n    cachedContent: CachedContent,\n    modelParams?: ModelParams,\n    requestOptions?: RequestOptions\n  ): void {\n    if (!this.apiKey) return;\n    this.client = new GenerativeAI(\n      this.apiKey\n    ).getGenerativeModelFromCachedContent(\n      cachedContent,\n      modelParams,\n      requestOptions\n    );\n  }\n\n  get useSystemInstruction(): boolean {\n    return typeof this.convertSystemMessageToHumanContent === \"boolean\"\n      ? !this.convertSystemMessageToHumanContent\n      : this.computeUseSystemInstruction;\n  }\n\n  get computeUseSystemInstruction(): boolean {\n    // This works on models from April 2024 and later\n    //   Vertex AI: gemini-1.5-pro and gemini-1.0-002 and later\n    //   AI Studio: gemini-1.5-pro-latest\n    if (this.model === \"gemini-1.0-pro-001\") {\n      return false;\n    } else if (this.model.startsWith(\"gemini-pro-vision\")) {\n      return false;\n    } else if (this.model.startsWith(\"gemini-1.0-pro-vision\")) {\n      return false;\n    } else if (this.model === \"gemini-pro\") {\n      // on AI Studio gemini-pro is still pointing at gemini-1.0-pro-001\n      return false;\n    }\n    return true;\n  }\n\n  getLsParams(options: this[\"ParsedCallOptions\"]): LangSmithParams {\n    return {\n      ls_provider: \"google_genai\",\n      ls_model_name: this.model,\n      ls_model_type: \"chat\",\n      ls_temperature: this.client.generationConfig.temperature,\n      ls_max_tokens: this.client.generationConfig.maxOutputTokens,\n      ls_stop: options.stop,\n    };\n  }\n\n  _combineLLMOutput() {\n    return [];\n  }\n\n  _llmType() {\n    return \"googlegenerativeai\";\n  }\n\n  override bindTools(\n    tools: GoogleGenerativeAIToolType[],\n    kwargs?: Partial<GoogleGenerativeAIChatCallOptions>\n  ): Runnable<\n    BaseLanguageModelInput,\n    AIMessageChunk,\n    GoogleGenerativeAIChatCallOptions\n  > {\n    return this.withConfig({\n      tools: convertToolsToGenAI(tools)?.tools,\n      ...kwargs,\n    });\n  }\n\n  invocationParams(\n    options?: this[\"ParsedCallOptions\"]\n  ): Omit<GenerateContentRequest, \"contents\"> {\n    const toolsAndConfig = options?.tools?.length\n      ? convertToolsToGenAI(options.tools, {\n          toolChoice: options.tool_choice,\n          allowedFunctionNames: options.allowedFunctionNames,\n        })\n      : undefined;\n\n    if (options?.responseSchema) {\n      this.client.generationConfig.responseSchema = options.responseSchema;\n      this.client.generationConfig.responseMimeType = \"application/json\";\n    } else {\n      this.client.generationConfig.responseSchema = undefined;\n      this.client.generationConfig.responseMimeType = this.json\n        ? \"application/json\"\n        : undefined;\n    }\n\n    return {\n      ...(toolsAndConfig?.tools ? { tools: toolsAndConfig.tools } : {}),\n      ...(toolsAndConfig?.toolConfig\n        ? { toolConfig: toolsAndConfig.toolConfig }\n        : {}),\n    };\n  }\n\n  async _generate(\n    messages: BaseMessage[],\n    options: this[\"ParsedCallOptions\"],\n    runManager?: CallbackManagerForLLMRun\n  ): Promise<ChatResult> {\n    const prompt = convertBaseMessagesToContent(\n      messages,\n      this._isMultimodalModel,\n      this.useSystemInstruction\n    );\n    let actualPrompt = prompt;\n    if (prompt[0].role === \"system\") {\n      const [systemInstruction] = prompt;\n      this.client.systemInstruction = systemInstruction;\n      actualPrompt = prompt.slice(1);\n    }\n    const parameters = this.invocationParams(options);\n\n    // Handle streaming\n    if (this.streaming) {\n      const tokenUsage: TokenUsage = {};\n      const stream = this._streamResponseChunks(messages, options, runManager);\n      const finalChunks: Record<number, ChatGenerationChunk> = {};\n\n      for await (const chunk of stream) {\n        const index =\n          (chunk.generationInfo as NewTokenIndices)?.completion ?? 0;\n        if (finalChunks[index] === undefined) {\n          finalChunks[index] = chunk;\n        } else {\n          finalChunks[index] = finalChunks[index].concat(chunk);\n        }\n      }\n      const generations = Object.entries(finalChunks)\n        .sort(([aKey], [bKey]) => parseInt(aKey, 10) - parseInt(bKey, 10))\n        .map(([_, value]) => value);\n\n      return { generations, llmOutput: { estimatedTokenUsage: tokenUsage } };\n    }\n\n    const res = await this.completionWithRetry({\n      ...parameters,\n      contents: actualPrompt,\n    });\n\n    let usageMetadata: UsageMetadata | undefined;\n    if (\"usageMetadata\" in res.response) {\n      const genAIUsageMetadata = res.response.usageMetadata as {\n        promptTokenCount: number | undefined;\n        candidatesTokenCount: number | undefined;\n        totalTokenCount: number | undefined;\n      };\n      usageMetadata = {\n        input_tokens: genAIUsageMetadata.promptTokenCount ?? 0,\n        output_tokens: genAIUsageMetadata.candidatesTokenCount ?? 0,\n        total_tokens: genAIUsageMetadata.totalTokenCount ?? 0,\n      };\n    }\n\n    const generationResult = mapGenerateContentResultToChatResult(\n      res.response,\n      {\n        usageMetadata,\n      }\n    );\n    // may not have generations in output if there was a refusal for safety reasons, malformed function call, etc.\n    if (generationResult.generations?.length > 0) {\n      await runManager?.handleLLMNewToken(\n        generationResult.generations[0]?.text ?? \"\"\n      );\n    }\n    return generationResult;\n  }\n\n  async *_streamResponseChunks(\n    messages: BaseMessage[],\n    options: this[\"ParsedCallOptions\"],\n    runManager?: CallbackManagerForLLMRun\n  ): AsyncGenerator<ChatGenerationChunk> {\n    const prompt = convertBaseMessagesToContent(\n      messages,\n      this._isMultimodalModel,\n      this.useSystemInstruction\n    );\n    let actualPrompt = prompt;\n    if (prompt[0].role === \"system\") {\n      const [systemInstruction] = prompt;\n      this.client.systemInstruction = systemInstruction;\n      actualPrompt = prompt.slice(1);\n    }\n    const parameters = this.invocationParams(options);\n    const request = {\n      ...parameters,\n      contents: actualPrompt,\n    };\n    const stream = await this.caller.callWithOptions(\n      { signal: options?.signal },\n      async () => {\n        const { stream } = await this.client.generateContentStream(request);\n        return stream;\n      }\n    );\n\n    let usageMetadata: UsageMetadata | undefined;\n    // Keep prior cumulative counts for calculating token deltas while streaming\n    let prevPromptTokenCount = 0;\n    let prevCandidatesTokenCount = 0;\n    let prevTotalTokenCount = 0;\n    let index = 0;\n    for await (const response of stream) {\n      if (\n        \"usageMetadata\" in response &&\n        response.usageMetadata !== undefined &&\n        this.streamUsage !== false &&\n        options.streamUsage !== false\n      ) {\n        usageMetadata = {\n          input_tokens: response.usageMetadata.promptTokenCount ?? 0,\n          output_tokens: response.usageMetadata.candidatesTokenCount ?? 0,\n          total_tokens: response.usageMetadata.totalTokenCount ?? 0,\n        };\n\n        // Under the hood, LangChain combines the prompt tokens. Google returns the updated\n        // total each time, so we need to find the difference between the tokens.\n        const newPromptTokenCount =\n          response.usageMetadata.promptTokenCount ?? 0;\n        usageMetadata.input_tokens = Math.max(\n          0,\n          newPromptTokenCount - prevPromptTokenCount\n        );\n        prevPromptTokenCount = newPromptTokenCount;\n\n        const newCandidatesTokenCount =\n          response.usageMetadata.candidatesTokenCount ?? 0;\n        usageMetadata.output_tokens = Math.max(\n          0,\n          newCandidatesTokenCount - prevCandidatesTokenCount\n        );\n        prevCandidatesTokenCount = newCandidatesTokenCount;\n\n        const newTotalTokenCount = response.usageMetadata.totalTokenCount ?? 0;\n        usageMetadata.total_tokens = Math.max(\n          0,\n          newTotalTokenCount - prevTotalTokenCount\n        );\n        prevTotalTokenCount = newTotalTokenCount;\n      }\n\n      const chunk = convertResponseContentToChatGenerationChunk(response, {\n        usageMetadata,\n        index,\n      });\n      index += 1;\n      if (!chunk) {\n        continue;\n      }\n\n      yield chunk;\n      await runManager?.handleLLMNewToken(chunk.text ?? \"\");\n    }\n  }\n\n  async completionWithRetry(\n    request: string | GenerateContentRequest | (string | GenerativeAIPart)[],\n    options?: this[\"ParsedCallOptions\"]\n  ) {\n    return this.caller.callWithOptions(\n      { signal: options?.signal },\n      async () => {\n        try {\n          return await this.client.generateContent(request);\n          // eslint-disable-next-line @typescript-eslint/no-explicit-any\n        } catch (e: any) {\n          // TODO: Improve error handling\n          if (e.message?.includes(\"400 Bad Request\")) {\n            e.status = 400;\n          }\n          throw e;\n        }\n      }\n    );\n  }\n\n  withStructuredOutput<\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    RunOutput extends Record<string, any> = Record<string, any>\n  >(\n    outputSchema:\n      | InteropZodType<RunOutput>\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      | Record<string, any>,\n    config?: StructuredOutputMethodOptions<false>\n  ): Runnable<BaseLanguageModelInput, RunOutput>;\n\n  withStructuredOutput<\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    RunOutput extends Record<string, any> = Record<string, any>\n  >(\n    outputSchema:\n      | InteropZodType<RunOutput>\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      | Record<string, any>,\n    config?: StructuredOutputMethodOptions<true>\n  ): Runnable<BaseLanguageModelInput, { raw: BaseMessage; parsed: RunOutput }>;\n\n  withStructuredOutput<\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    RunOutput extends Record<string, any> = Record<string, any>\n  >(\n    outputSchema:\n      | InteropZodType<RunOutput>\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      | Record<string, any>,\n    config?: StructuredOutputMethodOptions<boolean>\n  ):\n    | Runnable<BaseLanguageModelInput, RunOutput>\n    | Runnable<\n        BaseLanguageModelInput,\n        { raw: BaseMessage; parsed: RunOutput }\n      > {\n    // eslint-disable-next-line @typescript-eslint/no-explicit-any\n    const schema: InteropZodType<RunOutput> | Record<string, any> =\n      outputSchema;\n    const name = config?.name;\n    const method = config?.method;\n    const includeRaw = config?.includeRaw;\n    if (method === \"jsonMode\") {\n      throw new Error(\n        `ChatGoogleGenerativeAI only supports \"jsonSchema\" or \"functionCalling\" as a method.`\n      );\n    }\n\n    let llm;\n    let outputParser: BaseLLMOutputParser<RunOutput>;\n    if (method === \"functionCalling\") {\n      let functionName = name ?? \"extract\";\n      let tools: GoogleGenerativeAIFunctionDeclarationsTool[];\n      if (isInteropZodSchema(schema)) {\n        const jsonSchema = schemaToGenerativeAIParameters(schema);\n        tools = [\n          {\n            functionDeclarations: [\n              {\n                name: functionName,\n                description:\n                  jsonSchema.description ?? \"A function available to call.\",\n                parameters: jsonSchema as GenerativeAIFunctionDeclarationSchema,\n              },\n            ],\n          },\n        ];\n        outputParser = new GoogleGenerativeAIToolsOutputParser<\n          InferInteropZodOutput<typeof schema>\n        >({\n          returnSingle: true,\n          keyName: functionName,\n          zodSchema: schema,\n        });\n      } else {\n        let geminiFunctionDefinition: GenerativeAIFunctionDeclaration;\n        if (\n          typeof schema.name === \"string\" &&\n          typeof schema.parameters === \"object\" &&\n          schema.parameters != null\n        ) {\n          geminiFunctionDefinition = schema as GenerativeAIFunctionDeclaration;\n          geminiFunctionDefinition.parameters = removeAdditionalProperties(\n            schema.parameters\n          ) as GenerativeAIFunctionDeclarationSchema;\n          functionName = schema.name;\n        } else {\n          geminiFunctionDefinition = {\n            name: functionName,\n            description: schema.description ?? \"\",\n            parameters: removeAdditionalProperties(\n              schema\n            ) as GenerativeAIFunctionDeclarationSchema,\n          };\n        }\n        tools = [\n          {\n            functionDeclarations: [geminiFunctionDefinition],\n          },\n        ];\n        outputParser = new GoogleGenerativeAIToolsOutputParser<RunOutput>({\n          returnSingle: true,\n          keyName: functionName,\n        });\n      }\n      llm = this.bindTools(tools).withConfig({\n        allowedFunctionNames: [functionName],\n      });\n    } else {\n      const jsonSchema = schemaToGenerativeAIParameters(schema);\n      llm = this.withConfig({\n        responseSchema: jsonSchema as Schema,\n      });\n      outputParser = new JsonOutputParser();\n    }\n\n    if (!includeRaw) {\n      return llm.pipe(outputParser).withConfig({\n        runName: \"ChatGoogleGenerativeAIStructuredOutput\",\n      }) as Runnable<BaseLanguageModelInput, RunOutput>;\n    }\n\n    const parserAssign = RunnablePassthrough.assign({\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      parsed: (input: any, config) => outputParser.invoke(input.raw, config),\n    });\n    const parserNone = RunnablePassthrough.assign({\n      parsed: () => null,\n    });\n    const parsedWithFallback = parserAssign.withFallbacks({\n      fallbacks: [parserNone],\n    });\n    return RunnableSequence.from<\n      BaseLanguageModelInput,\n      { raw: BaseMessage; parsed: RunOutput }\n    >([\n      {\n        raw: llm,\n      },\n      parsedWithFallback,\n    ]).withConfig({\n      runName: \"StructuredOutputRunnable\",\n    });\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAgkBA,IAAa,yBAAb,cACU,cAEV;CACE,OAAO,UAAU;AACf,SAAO;CACR;CAED,kBAAkB;CAElB,IAAI,aAAoD;AACtD,SAAO,EACL,QAAQ,iBACT;CACF;CAED,eAAe;EAAC;EAAa;EAAe;CAAe;CAE3D,IAAI,aAAa;AACf,SAAO,EACL,QAAQ,iBACT;CACF;CAED;CAEA;CAEA;CAEA;CAEA;CAEA,gBAA0B,CAAE;CAE5B;CAEA;CAEA,YAAY;CAEZ;CAEA,cAAc;CAEd;CAEA,AAAQ;CAER,IAAI,qBAAqB;AACvB,SACE,KAAK,MAAM,SAAS,SAAS,IAC7B,KAAK,MAAM,WAAW,aAAa,IACnC,KAAK,MAAM,WAAW,WAAW,IAChC,KAAK,MAAM,WAAW,WAAW,IAChC,CAAC,KAAK,MAAM,WAAW,aAAa;CAEzC;CAED,YAAYA,QAAqC;EAC/C,MAAM,OAAO;EAEb,KAAK,QAAQ,OAAO,MAAM,QAAQ,aAAa,GAAG;EAElD,KAAK,kBAAkB,OAAO,mBAAmB,KAAK;AAEtD,MAAI,KAAK,mBAAmB,KAAK,kBAAkB,EACjD,OAAM,IAAI,MAAM;EAGlB,KAAK,cAAc,OAAO,eAAe,KAAK;AAC9C,MAAI,KAAK,gBAAgB,KAAK,cAAc,KAAK,KAAK,cAAc,GAClE,OAAM,IAAI,MAAM;EAGlB,KAAK,OAAO,OAAO,QAAQ,KAAK;AAChC,MAAI,KAAK,QAAQ,KAAK,OAAO,EAC3B,OAAM,IAAI,MAAM;AAGlB,MAAI,KAAK,QAAQ,KAAK,OAAO,EAC3B,OAAM,IAAI,MAAM;EAGlB,KAAK,OAAO,OAAO,QAAQ,KAAK;AAChC,MAAI,KAAK,QAAQ,KAAK,OAAO,EAC3B,OAAM,IAAI,MAAM;EAGlB,KAAK,gBAAgB,OAAO,iBAAiB,KAAK;EAElD,KAAK,SAAS,OAAO,UAAU,uBAAuB,iBAAiB;AACvE,MAAI,CAAC,KAAK,OACR,OAAM,IAAI,MACR;EAOJ,KAAK,iBAAiB,OAAO,kBAAkB,KAAK;AACpD,MAAI,KAAK,kBAAkB,KAAK,eAAe,SAAS,GAAG;GACzD,MAAM,oBAAoB,IAAI,IAC5B,KAAK,eAAe,IAAI,CAAC,MAAM,EAAE,SAAS;AAE5C,OAAI,kBAAkB,SAAS,KAAK,eAAe,OACjD,OAAM,IAAI,MACR;EAGL;EAED,KAAK,YAAY,OAAO,aAAa,KAAK;EAC1C,KAAK,OAAO,OAAO;EAEnB,KAAK,SAAS,IAAIC,mBAAa,KAAK,QAAQ,mBAC1C;GACE,OAAO,KAAK;GACZ,gBAAgB,KAAK;GACrB,kBAAkB;IAChB,eAAe,KAAK;IACpB,iBAAiB,KAAK;IACtB,aAAa,KAAK;IAClB,MAAM,KAAK;IACX,MAAM,KAAK;IACX,GAAI,KAAK,OAAO,EAAE,kBAAkB,mBAAoB,IAAG,CAAE;GAC9D;EACF,GACD;GACE,YAAY,OAAO;GACnB,SAAS,OAAO;EACjB,EACF;EACD,KAAK,cAAc,OAAO,eAAe,KAAK;CAC/C;CAED,iBACEC,eACAC,aACAC,gBACM;AACN,MAAI,CAAC,KAAK,OAAQ;EAClB,KAAK,SAAS,IAAIH,mBAChB,KAAK,QACL,oCACA,eACA,aACA,eACD;CACF;CAED,IAAI,uBAAgC;AAClC,SAAO,OAAO,KAAK,uCAAuC,YACtD,CAAC,KAAK,qCACN,KAAK;CACV;CAED,IAAI,8BAAuC;AAIzC,MAAI,KAAK,UAAU,qBACjB,QAAO;WACE,KAAK,MAAM,WAAW,oBAAoB,CACnD,QAAO;WACE,KAAK,MAAM,WAAW,wBAAwB,CACvD,QAAO;WACE,KAAK,UAAU,aAExB,QAAO;AAET,SAAO;CACR;CAED,YAAYI,SAAqD;AAC/D,SAAO;GACL,aAAa;GACb,eAAe,KAAK;GACpB,eAAe;GACf,gBAAgB,KAAK,OAAO,iBAAiB;GAC7C,eAAe,KAAK,OAAO,iBAAiB;GAC5C,SAAS,QAAQ;EAClB;CACF;CAED,oBAAoB;AAClB,SAAO,CAAE;CACV;CAED,WAAW;AACT,SAAO;CACR;CAED,AAAS,UACPC,OACAC,QAKA;AACA,SAAO,KAAK,WAAW;GACrB,OAAO,oBAAoB,MAAM,EAAE;GACnC,GAAG;EACJ,EAAC;CACH;CAED,iBACEC,SAC0C;EAC1C,MAAM,iBAAiB,SAAS,OAAO,SACnC,oBAAoB,QAAQ,OAAO;GACjC,YAAY,QAAQ;GACpB,sBAAsB,QAAQ;EAC/B,EAAC,GACF;AAEJ,MAAI,SAAS,gBAAgB;GAC3B,KAAK,OAAO,iBAAiB,iBAAiB,QAAQ;GACtD,KAAK,OAAO,iBAAiB,mBAAmB;EACjD,OAAM;GACL,KAAK,OAAO,iBAAiB,iBAAiB;GAC9C,KAAK,OAAO,iBAAiB,mBAAmB,KAAK,OACjD,qBACA;EACL;AAED,SAAO;GACL,GAAI,gBAAgB,QAAQ,EAAE,OAAO,eAAe,MAAO,IAAG,CAAE;GAChE,GAAI,gBAAgB,aAChB,EAAE,YAAY,eAAe,WAAY,IACzC,CAAE;EACP;CACF;CAED,MAAM,UACJC,UACAJ,SACAK,YACqB;EACrB,MAAM,SAAS,6BACb,UACA,KAAK,oBACL,KAAK,qBACN;EACD,IAAI,eAAe;AACnB,MAAI,OAAO,GAAG,SAAS,UAAU;GAC/B,MAAM,CAAC,kBAAkB,GAAG;GAC5B,KAAK,OAAO,oBAAoB;GAChC,eAAe,OAAO,MAAM,EAAE;EAC/B;EACD,MAAM,aAAa,KAAK,iBAAiB,QAAQ;AAGjD,MAAI,KAAK,WAAW;GAClB,MAAMC,aAAyB,CAAE;GACjC,MAAM,SAAS,KAAK,sBAAsB,UAAU,SAAS,WAAW;GACxE,MAAMC,cAAmD,CAAE;AAE3D,cAAW,MAAM,SAAS,QAAQ;IAChC,MAAM,QACH,MAAM,gBAAoC,cAAc;AAC3D,QAAI,YAAY,WAAW,QACzB,YAAY,SAAS;SAErB,YAAY,SAAS,YAAY,OAAO,OAAO,MAAM;GAExD;GACD,MAAM,cAAc,OAAO,QAAQ,YAAY,CAC5C,KAAK,CAAC,CAAC,KAAK,EAAE,CAAC,KAAK,KAAK,SAAS,MAAM,GAAG,GAAG,SAAS,MAAM,GAAG,CAAC,CACjE,IAAI,CAAC,CAAC,GAAG,MAAM,KAAK,MAAM;AAE7B,UAAO;IAAE;IAAa,WAAW,EAAE,qBAAqB,WAAY;GAAE;EACvE;EAED,MAAM,MAAM,MAAM,KAAK,oBAAoB;GACzC,GAAG;GACH,UAAU;EACX,EAAC;EAEF,IAAIC;AACJ,MAAI,mBAAmB,IAAI,UAAU;GACnC,MAAM,qBAAqB,IAAI,SAAS;GAKxC,gBAAgB;IACd,cAAc,mBAAmB,oBAAoB;IACrD,eAAe,mBAAmB,wBAAwB;IAC1D,cAAc,mBAAmB,mBAAmB;GACrD;EACF;EAED,MAAM,mBAAmB,qCACvB,IAAI,UACJ,EACE,cACD,EACF;AAED,MAAI,iBAAiB,aAAa,SAAS,GACzC,MAAM,YAAY,kBAChB,iBAAiB,YAAY,IAAI,QAAQ,GAC1C;AAEH,SAAO;CACR;CAED,OAAO,sBACLJ,UACAJ,SACAK,YACqC;EACrC,MAAM,SAAS,6BACb,UACA,KAAK,oBACL,KAAK,qBACN;EACD,IAAI,eAAe;AACnB,MAAI,OAAO,GAAG,SAAS,UAAU;GAC/B,MAAM,CAAC,kBAAkB,GAAG;GAC5B,KAAK,OAAO,oBAAoB;GAChC,eAAe,OAAO,MAAM,EAAE;EAC/B;EACD,MAAM,aAAa,KAAK,iBAAiB,QAAQ;EACjD,MAAM,UAAU;GACd,GAAG;GACH,UAAU;EACX;EACD,MAAM,SAAS,MAAM,KAAK,OAAO,gBAC/B,EAAE,QAAQ,SAAS,OAAQ,GAC3B,YAAY;GACV,MAAM,EAAE,kBAAQ,GAAG,MAAM,KAAK,OAAO,sBAAsB,QAAQ;AACnE,UAAOI;EACR,EACF;EAED,IAAID;EAEJ,IAAI,uBAAuB;EAC3B,IAAI,2BAA2B;EAC/B,IAAI,sBAAsB;EAC1B,IAAI,QAAQ;AACZ,aAAW,MAAM,YAAY,QAAQ;AACnC,OACE,mBAAmB,YACnB,SAAS,kBAAkB,UAC3B,KAAK,gBAAgB,SACrB,QAAQ,gBAAgB,OACxB;IACA,gBAAgB;KACd,cAAc,SAAS,cAAc,oBAAoB;KACzD,eAAe,SAAS,cAAc,wBAAwB;KAC9D,cAAc,SAAS,cAAc,mBAAmB;IACzD;IAID,MAAM,sBACJ,SAAS,cAAc,oBAAoB;IAC7C,cAAc,eAAe,KAAK,IAChC,GACA,sBAAsB,qBACvB;IACD,uBAAuB;IAEvB,MAAM,0BACJ,SAAS,cAAc,wBAAwB;IACjD,cAAc,gBAAgB,KAAK,IACjC,GACA,0BAA0B,yBAC3B;IACD,2BAA2B;IAE3B,MAAM,qBAAqB,SAAS,cAAc,mBAAmB;IACrE,cAAc,eAAe,KAAK,IAChC,GACA,qBAAqB,oBACtB;IACD,sBAAsB;GACvB;GAED,MAAM,QAAQ,4CAA4C,UAAU;IAClE;IACA;GACD,EAAC;GACF,SAAS;AACT,OAAI,CAAC,MACH;GAGF,MAAM;GACN,MAAM,YAAY,kBAAkB,MAAM,QAAQ,GAAG;EACtD;CACF;CAED,MAAM,oBACJE,SACAP,SACA;AACA,SAAO,KAAK,OAAO,gBACjB,EAAE,QAAQ,SAAS,OAAQ,GAC3B,YAAY;AACV,OAAI;AACF,WAAO,MAAM,KAAK,OAAO,gBAAgB,QAAQ;GAElD,SAAQQ,GAAQ;AAEf,QAAI,EAAE,SAAS,SAAS,kBAAkB,EACxC,EAAE,SAAS;AAEb,UAAM;GACP;EACF,EACF;CACF;CAwBD,qBAIEC,cAIAC,QAMI;EAEJ,MAAMC,SACJ;EACF,MAAM,OAAO,QAAQ;EACrB,MAAM,SAAS,QAAQ;EACvB,MAAM,aAAa,QAAQ;AAC3B,MAAI,WAAW,WACb,OAAM,IAAI,MACR,CAAC,mFAAmF,CAAC;EAIzF,IAAI;EACJ,IAAIC;AACJ,MAAI,WAAW,mBAAmB;GAChC,IAAI,eAAe,QAAQ;GAC3B,IAAIC;AACJ,OAAI,mBAAmB,OAAO,EAAE;IAC9B,MAAM,aAAa,+BAA+B,OAAO;IACzD,QAAQ,CACN,EACE,sBAAsB,CACpB;KACE,MAAM;KACN,aACE,WAAW,eAAe;KAC5B,YAAY;IACb,CACF,EACF,CACF;IACD,eAAe,IAAI,oCAEjB;KACA,cAAc;KACd,SAAS;KACT,WAAW;IACZ;GACF,OAAM;IACL,IAAIC;AACJ,QACE,OAAO,OAAO,SAAS,YACvB,OAAO,OAAO,eAAe,YAC7B,OAAO,cAAc,MACrB;KACA,2BAA2B;KAC3B,yBAAyB,aAAa,2BACpC,OAAO,WACR;KACD,eAAe,OAAO;IACvB,OACC,2BAA2B;KACzB,MAAM;KACN,aAAa,OAAO,eAAe;KACnC,YAAY,2BACV,OACD;IACF;IAEH,QAAQ,CACN,EACE,sBAAsB,CAAC,wBAAyB,EACjD,CACF;IACD,eAAe,IAAI,oCAA+C;KAChE,cAAc;KACd,SAAS;IACV;GACF;GACD,MAAM,KAAK,UAAU,MAAM,CAAC,WAAW,EACrC,sBAAsB,CAAC,YAAa,EACrC,EAAC;EACH,OAAM;GACL,MAAM,aAAa,+BAA+B,OAAO;GACzD,MAAM,KAAK,WAAW,EACpB,gBAAgB,WACjB,EAAC;GACF,eAAe,IAAI;EACpB;AAED,MAAI,CAAC,WACH,QAAO,IAAI,KAAK,aAAa,CAAC,WAAW,EACvC,SAAS,yCACV,EAAC;EAGJ,MAAM,eAAe,oBAAoB,OAAO,EAE9C,QAAQ,CAACC,OAAYC,aAAW,aAAa,OAAO,MAAM,KAAKA,SAAO,CACvE,EAAC;EACF,MAAM,aAAa,oBAAoB,OAAO,EAC5C,QAAQ,MAAM,KACf,EAAC;EACF,MAAM,qBAAqB,aAAa,cAAc,EACpD,WAAW,CAAC,UAAW,EACxB,EAAC;AACF,SAAO,iBAAiB,KAGtB,CACA,EACE,KAAK,IACN,GACD,kBACD,EAAC,CAAC,WAAW,EACZ,SAAS,2BACV,EAAC;CACH;AACF"}

package/dist/embeddings.cjs

@@ -1,153 +1,99 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GoogleGenerativeAIEmbeddings = void 0;
-const generative_ai_1 = require("@google/generative-ai");
-const env_1 = require("@langchain/core/utils/env");
-const embeddings_1 = require("@langchain/core/embeddings");
-const chunk_array_1 = require("@langchain/core/utils/chunk_array");
+const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
+const __google_generative_ai = require_rolldown_runtime.__toESM(require("@google/generative-ai"));
+const __langchain_core_utils_env = require_rolldown_runtime.__toESM(require("@langchain/core/utils/env"));
+const __langchain_core_embeddings = require_rolldown_runtime.__toESM(require("@langchain/core/embeddings"));
+const __langchain_core_utils_chunk_array = require_rolldown_runtime.__toESM(require("@langchain/core/utils/chunk_array"));
+
+//#region src/embeddings.ts
 /**
- * Class that extends the Embeddings class and provides methods for
- * generating embeddings using the Google Palm API.
- * @example
- * ```typescript
- * const model = new GoogleGenerativeAIEmbeddings({
- *   apiKey: "<YOUR API KEY>",
- *   modelName: "embedding-001",
- * });
- *
- * // Embed a single query
- * const res = await model.embedQuery(
- *   "What would be a good company name for a company that makes colorful socks?"
- * );
- * console.log({ res });
- *
- * // Embed multiple documents
- * const documentRes = await model.embedDocuments(["Hello world", "Bye bye"]);
- * console.log({ documentRes });
- * ```
- */
-class GoogleGenerativeAIEmbeddings extends embeddings_1.Embeddings {
-    constructor(fields) {
-        super(fields ?? {});
-        Object.defineProperty(this, "apiKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "modelName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "embedding-001"
-        });
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "embedding-001"
-        });
-        Object.defineProperty(this, "taskType", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "title", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "stripNewLines", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: true
-        });
-        Object.defineProperty(this, "maxBatchSize", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 100
-        }); // Max batch size for embedDocuments set by GenerativeModel client's batchEmbedContents call
-        Object.defineProperty(this, "client", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.modelName =
-            fields?.model?.replace(/^models\//, "") ??
-                fields?.modelName?.replace(/^models\//, "") ??
-                this.modelName;
-        this.model = this.modelName;
-        this.taskType = fields?.taskType ?? this.taskType;
-        this.title = fields?.title ?? this.title;
-        if (this.title && this.taskType !== "RETRIEVAL_DOCUMENT") {
-            throw new Error("title can only be sepcified with TaskType.RETRIEVAL_DOCUMENT");
-        }
-        this.apiKey = fields?.apiKey ?? (0, env_1.getEnvironmentVariable)("GOOGLE_API_KEY");
-        if (!this.apiKey) {
-            throw new Error("Please set an API key for Google GenerativeAI " +
-                "in the environmentb variable GOOGLE_API_KEY " +
-                "or in the `apiKey` field of the " +
-                "GoogleGenerativeAIEmbeddings constructor");
-        }
-        this.client = new generative_ai_1.GoogleGenerativeAI(this.apiKey).getGenerativeModel({
-            model: this.model,
-        }, {
-            baseUrl: fields?.baseUrl,
-        });
-    }
-    _convertToContent(text) {
-        const cleanedText = this.stripNewLines ? text.replace(/\n/g, " ") : text;
-        return {
-            content: { role: "user", parts: [{ text: cleanedText }] },
-            taskType: this.taskType,
-            title: this.title,
-        };
-    }
-    async _embedQueryContent(text) {
-        const req = this._convertToContent(text);
-        const res = await this.client.embedContent(req);
-        return res.embedding.values ?? [];
-    }
-    async _embedDocumentsContent(documents) {
-        const batchEmbedChunks = (0, chunk_array_1.chunkArray)(documents, this.maxBatchSize);
-        const batchEmbedRequests = batchEmbedChunks.map((chunk) => ({
-            requests: chunk.map((doc) => this._convertToContent(doc)),
-        }));
-        const responses = await Promise.allSettled(batchEmbedRequests.map((req) => this.client.batchEmbedContents(req)));
-        const embeddings = responses.flatMap((res, idx) => {
-            if (res.status === "fulfilled") {
-                return res.value.embeddings.map((e) => e.values || []);
-            }
-            else {
-                return Array(batchEmbedChunks[idx].length).fill([]);
-            }
-        });
-        return embeddings;
-    }
-    /**
-     * Method that takes a document as input and returns a promise that
-     * resolves to an embedding for the document. It calls the _embedText
-     * method with the document as the input.
-     * @param document Document for which to generate an embedding.
-     * @returns Promise that resolves to an embedding for the input document.
-     */
-    embedQuery(document) {
-        return this.caller.call(this._embedQueryContent.bind(this), document);
-    }
-    /**
-     * Method that takes an array of documents as input and returns a promise
-     * that resolves to a 2D array of embeddings for each document. It calls
-     * the _embedText method for each document in the array.
-     * @param documents Array of documents for which to generate embeddings.
-     * @returns Promise that resolves to a 2D array of embeddings for each input document.
-     */
-    embedDocuments(documents) {
-        return this.caller.call(this._embedDocumentsContent.bind(this), documents);
-    }
-}
+* Class that extends the Embeddings class and provides methods for
+* generating embeddings using the Google Palm API.
+* @example
+* ```typescript
+* const model = new GoogleGenerativeAIEmbeddings({
+*   apiKey: "<YOUR API KEY>",
+*   modelName: "embedding-001",
+* });
+*
+* // Embed a single query
+* const res = await model.embedQuery(
+*   "What would be a good company name for a company that makes colorful socks?"
+* );
+* console.log({ res });
+*
+* // Embed multiple documents
+* const documentRes = await model.embedDocuments(["Hello world", "Bye bye"]);
+* console.log({ documentRes });
+* ```
+*/
+var GoogleGenerativeAIEmbeddings = class extends __langchain_core_embeddings.Embeddings {
+	apiKey;
+	modelName = "embedding-001";
+	model = "embedding-001";
+	taskType;
+	title;
+	stripNewLines = true;
+	maxBatchSize = 100;
+	client;
+	constructor(fields) {
+		super(fields ?? {});
+		this.modelName = fields?.model?.replace(/^models\//, "") ?? fields?.modelName?.replace(/^models\//, "") ?? this.modelName;
+		this.model = this.modelName;
+		this.taskType = fields?.taskType ?? this.taskType;
+		this.title = fields?.title ?? this.title;
+		if (this.title && this.taskType !== "RETRIEVAL_DOCUMENT") throw new Error("title can only be sepcified with TaskType.RETRIEVAL_DOCUMENT");
+		this.apiKey = fields?.apiKey ?? (0, __langchain_core_utils_env.getEnvironmentVariable)("GOOGLE_API_KEY");
+		if (!this.apiKey) throw new Error("Please set an API key for Google GenerativeAI in the environmentb variable GOOGLE_API_KEY or in the `apiKey` field of the GoogleGenerativeAIEmbeddings constructor");
+		this.client = new __google_generative_ai.GoogleGenerativeAI(this.apiKey).getGenerativeModel({ model: this.model }, { baseUrl: fields?.baseUrl });
+	}
+	_convertToContent(text) {
+		const cleanedText = this.stripNewLines ? text.replace(/\n/g, " ") : text;
+		return {
+			content: {
+				role: "user",
+				parts: [{ text: cleanedText }]
+			},
+			taskType: this.taskType,
+			title: this.title
+		};
+	}
+	async _embedQueryContent(text) {
+		const req = this._convertToContent(text);
+		const res = await this.client.embedContent(req);
+		return res.embedding.values ?? [];
+	}
+	async _embedDocumentsContent(documents) {
+		const batchEmbedChunks = (0, __langchain_core_utils_chunk_array.chunkArray)(documents, this.maxBatchSize);
+		const batchEmbedRequests = batchEmbedChunks.map((chunk) => ({ requests: chunk.map((doc) => this._convertToContent(doc)) }));
+		const responses = await Promise.allSettled(batchEmbedRequests.map((req) => this.client.batchEmbedContents(req)));
+		const embeddings = responses.flatMap((res, idx) => {
+			if (res.status === "fulfilled") return res.value.embeddings.map((e) => e.values || []);
+			else return Array(batchEmbedChunks[idx].length).fill([]);
+		});
+		return embeddings;
+	}
+	/**
+	* Method that takes a document as input and returns a promise that
+	* resolves to an embedding for the document. It calls the _embedText
+	* method with the document as the input.
+	* @param document Document for which to generate an embedding.
+	* @returns Promise that resolves to an embedding for the input document.
+	*/
+	embedQuery(document) {
+		return this.caller.call(this._embedQueryContent.bind(this), document);
+	}
+	/**
+	* Method that takes an array of documents as input and returns a promise
+	* that resolves to a 2D array of embeddings for each document. It calls
+	* the _embedText method for each document in the array.
+	* @param documents Array of documents for which to generate embeddings.
+	* @returns Promise that resolves to a 2D array of embeddings for each input document.
+	*/
+	embedDocuments(documents) {
+		return this.caller.call(this._embedDocumentsContent.bind(this), documents);
+	}
+};
+
+//#endregion
 exports.GoogleGenerativeAIEmbeddings = GoogleGenerativeAIEmbeddings;
+//# sourceMappingURL=embeddings.cjs.map

package/dist/embeddings.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.cjs","names":["Embeddings","fields?: GoogleGenerativeAIEmbeddingsParams","GoogleGenerativeAI","text: string","documents: string[]","batchEmbedChunks: string[][]","document: string"],"sources":["../src/embeddings.ts"],"sourcesContent":["import { GoogleGenerativeAI, GenerativeModel } from \"@google/generative-ai\";\nimport type { TaskType, EmbedContentRequest } from \"@google/generative-ai\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\nimport { Embeddings, EmbeddingsParams } from \"@langchain/core/embeddings\";\nimport { chunkArray } from \"@langchain/core/utils/chunk_array\";\n\n/**\n * Interface that extends EmbeddingsParams and defines additional\n * parameters specific to the GoogleGenerativeAIEmbeddings class.\n */\nexport interface GoogleGenerativeAIEmbeddingsParams extends EmbeddingsParams {\n  /**\n   * Model Name to use\n   *\n   * Alias for `model`\n   *\n   * Note: The format must follow the pattern - `{model}`\n   */\n  modelName?: string;\n  /**\n   * Model Name to use\n   *\n   * Note: The format must follow the pattern - `{model}`\n   */\n  model?: string;\n\n  /**\n   * Type of task for which the embedding will be used\n   *\n   * Note: currently only supported by `embedding-001` model\n   */\n  taskType?: TaskType;\n\n  /**\n   * An optional title for the text. Only applicable when TaskType is\n   * `RETRIEVAL_DOCUMENT`\n   *\n   * Note: currently only supported by `embedding-001` model\n   */\n  title?: string;\n\n  /**\n   * Whether to strip new lines from the input text. Default to true\n   */\n  stripNewLines?: boolean;\n\n  /**\n   * Google API key to use\n   */\n  apiKey?: string;\n\n  /**\n   * Google API base URL to use\n   */\n  baseUrl?: string;\n}\n\n/**\n * Class that extends the Embeddings class and provides methods for\n * generating embeddings using the Google Palm API.\n * @example\n * ```typescript\n * const model = new GoogleGenerativeAIEmbeddings({\n *   apiKey: \"<YOUR API KEY>\",\n *   modelName: \"embedding-001\",\n * });\n *\n * // Embed a single query\n * const res = await model.embedQuery(\n *   \"What would be a good company name for a company that makes colorful socks?\"\n * );\n * console.log({ res });\n *\n * // Embed multiple documents\n * const documentRes = await model.embedDocuments([\"Hello world\", \"Bye bye\"]);\n * console.log({ documentRes });\n * ```\n */\nexport class GoogleGenerativeAIEmbeddings\n  extends Embeddings\n  implements GoogleGenerativeAIEmbeddingsParams\n{\n  apiKey?: string;\n\n  modelName = \"embedding-001\";\n\n  model = \"embedding-001\";\n\n  taskType?: TaskType;\n\n  title?: string;\n\n  stripNewLines = true;\n\n  maxBatchSize = 100; // Max batch size for embedDocuments set by GenerativeModel client's batchEmbedContents call\n\n  private client: GenerativeModel;\n\n  constructor(fields?: GoogleGenerativeAIEmbeddingsParams) {\n    super(fields ?? {});\n\n    this.modelName =\n      fields?.model?.replace(/^models\\//, \"\") ??\n      fields?.modelName?.replace(/^models\\//, \"\") ??\n      this.modelName;\n    this.model = this.modelName;\n\n    this.taskType = fields?.taskType ?? this.taskType;\n\n    this.title = fields?.title ?? this.title;\n\n    if (this.title && this.taskType !== \"RETRIEVAL_DOCUMENT\") {\n      throw new Error(\n        \"title can only be sepcified with TaskType.RETRIEVAL_DOCUMENT\"\n      );\n    }\n\n    this.apiKey = fields?.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n    if (!this.apiKey) {\n      throw new Error(\n        \"Please set an API key for Google GenerativeAI \" +\n          \"in the environmentb variable GOOGLE_API_KEY \" +\n          \"or in the `apiKey` field of the \" +\n          \"GoogleGenerativeAIEmbeddings constructor\"\n      );\n    }\n\n    this.client = new GoogleGenerativeAI(this.apiKey).getGenerativeModel(\n      {\n        model: this.model,\n      },\n      {\n        baseUrl: fields?.baseUrl,\n      }\n    );\n  }\n\n  private _convertToContent(text: string): EmbedContentRequest {\n    const cleanedText = this.stripNewLines ? text.replace(/\\n/g, \" \") : text;\n    return {\n      content: { role: \"user\", parts: [{ text: cleanedText }] },\n      taskType: this.taskType,\n      title: this.title,\n    };\n  }\n\n  protected async _embedQueryContent(text: string): Promise<number[]> {\n    const req = this._convertToContent(text);\n    const res = await this.client.embedContent(req);\n    return res.embedding.values ?? [];\n  }\n\n  protected async _embedDocumentsContent(\n    documents: string[]\n  ): Promise<number[][]> {\n    const batchEmbedChunks: string[][] = chunkArray<string>(\n      documents,\n      this.maxBatchSize\n    );\n\n    const batchEmbedRequests = batchEmbedChunks.map((chunk) => ({\n      requests: chunk.map((doc) => this._convertToContent(doc)),\n    }));\n\n    const responses = await Promise.allSettled(\n      batchEmbedRequests.map((req) => this.client.batchEmbedContents(req))\n    );\n\n    const embeddings = responses.flatMap((res, idx) => {\n      if (res.status === \"fulfilled\") {\n        return res.value.embeddings.map((e) => e.values || []);\n      } else {\n        return Array(batchEmbedChunks[idx].length).fill([]);\n      }\n    });\n\n    return embeddings;\n  }\n\n  /**\n   * Method that takes a document as input and returns a promise that\n   * resolves to an embedding for the document. It calls the _embedText\n   * method with the document as the input.\n   * @param document Document for which to generate an embedding.\n   * @returns Promise that resolves to an embedding for the input document.\n   */\n  embedQuery(document: string): Promise<number[]> {\n    return this.caller.call(this._embedQueryContent.bind(this), document);\n  }\n\n  /**\n   * Method that takes an array of documents as input and returns a promise\n   * that resolves to a 2D array of embeddings for each document. It calls\n   * the _embedText method for each document in the array.\n   * @param documents Array of documents for which to generate embeddings.\n   * @returns Promise that resolves to a 2D array of embeddings for each input document.\n   */\n  embedDocuments(documents: string[]): Promise<number[][]> {\n    return this.caller.call(this._embedDocumentsContent.bind(this), documents);\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AA8EA,IAAa,+BAAb,cACUA,uCAEV;CACE;CAEA,YAAY;CAEZ,QAAQ;CAER;CAEA;CAEA,gBAAgB;CAEhB,eAAe;CAEf,AAAQ;CAER,YAAYC,QAA6C;EACvD,MAAM,UAAU,CAAE,EAAC;EAEnB,KAAK,YACH,QAAQ,OAAO,QAAQ,aAAa,GAAG,IACvC,QAAQ,WAAW,QAAQ,aAAa,GAAG,IAC3C,KAAK;EACP,KAAK,QAAQ,KAAK;EAElB,KAAK,WAAW,QAAQ,YAAY,KAAK;EAEzC,KAAK,QAAQ,QAAQ,SAAS,KAAK;AAEnC,MAAI,KAAK,SAAS,KAAK,aAAa,qBAClC,OAAM,IAAI,MACR;EAIJ,KAAK,SAAS,QAAQ,iEAAiC,iBAAiB;AACxE,MAAI,CAAC,KAAK,OACR,OAAM,IAAI,MACR;EAOJ,KAAK,SAAS,IAAIC,0CAAmB,KAAK,QAAQ,mBAChD,EACE,OAAO,KAAK,MACb,GACD,EACE,SAAS,QAAQ,QAClB,EACF;CACF;CAED,AAAQ,kBAAkBC,MAAmC;EAC3D,MAAM,cAAc,KAAK,gBAAgB,KAAK,QAAQ,OAAO,IAAI,GAAG;AACpE,SAAO;GACL,SAAS;IAAE,MAAM;IAAQ,OAAO,CAAC,EAAE,MAAM,YAAa,CAAC;GAAE;GACzD,UAAU,KAAK;GACf,OAAO,KAAK;EACb;CACF;CAED,MAAgB,mBAAmBA,MAAiC;EAClE,MAAM,MAAM,KAAK,kBAAkB,KAAK;EACxC,MAAM,MAAM,MAAM,KAAK,OAAO,aAAa,IAAI;AAC/C,SAAO,IAAI,UAAU,UAAU,CAAE;CAClC;CAED,MAAgB,uBACdC,WACqB;EACrB,MAAMC,sEACJ,WACA,KAAK,aACN;EAED,MAAM,qBAAqB,iBAAiB,IAAI,CAAC,WAAW,EAC1D,UAAU,MAAM,IAAI,CAAC,QAAQ,KAAK,kBAAkB,IAAI,CAAC,CAC1D,GAAE;EAEH,MAAM,YAAY,MAAM,QAAQ,WAC9B,mBAAmB,IAAI,CAAC,QAAQ,KAAK,OAAO,mBAAmB,IAAI,CAAC,CACrE;EAED,MAAM,aAAa,UAAU,QAAQ,CAAC,KAAK,QAAQ;AACjD,OAAI,IAAI,WAAW,YACjB,QAAO,IAAI,MAAM,WAAW,IAAI,CAAC,MAAM,EAAE,UAAU,CAAE,EAAC;OAEtD,QAAO,MAAM,iBAAiB,KAAK,OAAO,CAAC,KAAK,CAAE,EAAC;EAEtD,EAAC;AAEF,SAAO;CACR;;;;;;;;CASD,WAAWC,UAAqC;AAC9C,SAAO,KAAK,OAAO,KAAK,KAAK,mBAAmB,KAAK,KAAK,EAAE,SAAS;CACtE;;;;;;;;CASD,eAAeF,WAA0C;AACvD,SAAO,KAAK,OAAO,KAAK,KAAK,uBAAuB,KAAK,KAAK,EAAE,UAAU;CAC3E;AACF"}

package/dist/embeddings.d.cts

@@ -0,0 +1,104 @@
+import { TaskType } from "@google/generative-ai";
+import { Embeddings, EmbeddingsParams } from "@langchain/core/embeddings";
+
+//#region src/embeddings.d.ts
+
+/**
+ * Interface that extends EmbeddingsParams and defines additional
+ * parameters specific to the GoogleGenerativeAIEmbeddings class.
+ */
+interface GoogleGenerativeAIEmbeddingsParams extends EmbeddingsParams {
+  /**
+   * Model Name to use
+   *
+   * Alias for `model`
+   *
+   * Note: The format must follow the pattern - `{model}`
+   */
+  modelName?: string;
+  /**
+   * Model Name to use
+   *
+   * Note: The format must follow the pattern - `{model}`
+   */
+  model?: string;
+  /**
+   * Type of task for which the embedding will be used
+   *
+   * Note: currently only supported by `embedding-001` model
+   */
+  taskType?: TaskType;
+  /**
+   * An optional title for the text. Only applicable when TaskType is
+   * `RETRIEVAL_DOCUMENT`
+   *
+   * Note: currently only supported by `embedding-001` model
+   */
+  title?: string;
+  /**
+   * Whether to strip new lines from the input text. Default to true
+   */
+  stripNewLines?: boolean;
+  /**
+   * Google API key to use
+   */
+  apiKey?: string;
+  /**
+   * Google API base URL to use
+   */
+  baseUrl?: string;
+}
+/**
+ * Class that extends the Embeddings class and provides methods for
+ * generating embeddings using the Google Palm API.
+ * @example
+ * ```typescript
+ * const model = new GoogleGenerativeAIEmbeddings({
+ *   apiKey: "<YOUR API KEY>",
+ *   modelName: "embedding-001",
+ * });
+ *
+ * // Embed a single query
+ * const res = await model.embedQuery(
+ *   "What would be a good company name for a company that makes colorful socks?"
+ * );
+ * console.log({ res });
+ *
+ * // Embed multiple documents
+ * const documentRes = await model.embedDocuments(["Hello world", "Bye bye"]);
+ * console.log({ documentRes });
+ * ```
+ */
+declare class GoogleGenerativeAIEmbeddings extends Embeddings implements GoogleGenerativeAIEmbeddingsParams {
+  apiKey?: string;
+  modelName: string;
+  model: string;
+  taskType?: TaskType;
+  title?: string;
+  stripNewLines: boolean;
+  maxBatchSize: number; // Max batch size for embedDocuments set by GenerativeModel client's batchEmbedContents call
+  private client;
+  constructor(fields?: GoogleGenerativeAIEmbeddingsParams);
+  private _convertToContent;
+  protected _embedQueryContent(text: string): Promise<number[]>;
+  protected _embedDocumentsContent(documents: string[]): Promise<number[][]>;
+  /**
+   * Method that takes a document as input and returns a promise that
+   * resolves to an embedding for the document. It calls the _embedText
+   * method with the document as the input.
+   * @param document Document for which to generate an embedding.
+   * @returns Promise that resolves to an embedding for the input document.
+   */
+  embedQuery(document: string): Promise<number[]>;
+  /**
+   * Method that takes an array of documents as input and returns a promise
+   * that resolves to a 2D array of embeddings for each document. It calls
+   * the _embedText method for each document in the array.
+   * @param documents Array of documents for which to generate embeddings.
+   * @returns Promise that resolves to a 2D array of embeddings for each input document.
+   */
+  embedDocuments(documents: string[]): Promise<number[][]>;
+}
+//#endregion
+export { GoogleGenerativeAIEmbeddings, GoogleGenerativeAIEmbeddingsParams };
+//# sourceMappingURL=embeddings.d.cts.map

package/dist/embeddings.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.cts","names":["TaskType","Embeddings","EmbeddingsParams","GoogleGenerativeAIEmbeddingsParams","GoogleGenerativeAIEmbeddings","Promise"],"sources":["../src/embeddings.d.ts"],"sourcesContent":["import type { TaskType } from \"@google/generative-ai\";\nimport { Embeddings, EmbeddingsParams } from \"@langchain/core/embeddings\";\n/**\n * Interface that extends EmbeddingsParams and defines additional\n * parameters specific to the GoogleGenerativeAIEmbeddings class.\n */\nexport interface GoogleGenerativeAIEmbeddingsParams extends EmbeddingsParams {\n    /**\n     * Model Name to use\n     *\n     * Alias for `model`\n     *\n     * Note: The format must follow the pattern - `{model}`\n     */\n    modelName?: string;\n    /**\n     * Model Name to use\n     *\n     * Note: The format must follow the pattern - `{model}`\n     */\n    model?: string;\n    /**\n     * Type of task for which the embedding will be used\n     *\n     * Note: currently only supported by `embedding-001` model\n     */\n    taskType?: TaskType;\n    /**\n     * An optional title for the text. Only applicable when TaskType is\n     * `RETRIEVAL_DOCUMENT`\n     *\n     * Note: currently only supported by `embedding-001` model\n     */\n    title?: string;\n    /**\n     * Whether to strip new lines from the input text. Default to true\n     */\n    stripNewLines?: boolean;\n    /**\n     * Google API key to use\n     */\n    apiKey?: string;\n    /**\n     * Google API base URL to use\n     */\n    baseUrl?: string;\n}\n/**\n * Class that extends the Embeddings class and provides methods for\n * generating embeddings using the Google Palm API.\n * @example\n * ```typescript\n * const model = new GoogleGenerativeAIEmbeddings({\n *   apiKey: \"<YOUR API KEY>\",\n *   modelName: \"embedding-001\",\n * });\n *\n * // Embed a single query\n * const res = await model.embedQuery(\n *   \"What would be a good company name for a company that makes colorful socks?\"\n * );\n * console.log({ res });\n *\n * // Embed multiple documents\n * const documentRes = await model.embedDocuments([\"Hello world\", \"Bye bye\"]);\n * console.log({ documentRes });\n * ```\n */\nexport declare class GoogleGenerativeAIEmbeddings extends Embeddings implements GoogleGenerativeAIEmbeddingsParams {\n    apiKey?: string;\n    modelName: string;\n    model: string;\n    taskType?: TaskType;\n    title?: string;\n    stripNewLines: boolean;\n    maxBatchSize: number; // Max batch size for embedDocuments set by GenerativeModel client's batchEmbedContents call\n    private client;\n    constructor(fields?: GoogleGenerativeAIEmbeddingsParams);\n    private _convertToContent;\n    protected _embedQueryContent(text: string): Promise<number[]>;\n    protected _embedDocumentsContent(documents: string[]): Promise<number[][]>;\n    /**\n     * Method that takes a document as input and returns a promise that\n     * resolves to an embedding for the document. It calls the _embedText\n     * method with the document as the input.\n     * @param document Document for which to generate an embedding.\n     * @returns Promise that resolves to an embedding for the input document.\n     */\n    embedQuery(document: string): Promise<number[]>;\n    /**\n     * Method that takes an array of documents as input and returns a promise\n     * that resolves to a 2D array of embeddings for each document. It calls\n     * the _embedText method for each document in the array.\n     * @param documents Array of documents for which to generate embeddings.\n     * @returns Promise that resolves to a 2D array of embeddings for each input document.\n     */\n    embedDocuments(documents: string[]): Promise<number[][]>;\n}\n"],"mappings":";;;;;;;AAMA;;AAoBeA,UApBEG,kCAAAA,SAA2CD,gBAoB7CF,CAAAA;EAAQ;AApBqD;AA8D5E;;;;;EAWuD,SACIK,CAAAA,EAAAA,MAAAA;EAAO;;;;AAZgD;;;;;;;aA1CnGL;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cA0CMI,4BAAAA,SAAqCH,UAAAA,YAAsBE;;;;aAIjEH;;;;;uBAKUG;;8CAEuBE;yDACWA;;;;;;;;gCAQzBA;;;;;;;;uCAQOA"}