@langchain/google-genai 0.2.18 → 1.0.1

package/dist/embeddings.d.ts

@@ -1,49 +1,52 @@
-import type { TaskType } from "@google/generative-ai";
+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.
  */
-export 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;
+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
@@ -66,33 +69,36 @@ export interface GoogleGenerativeAIEmbeddingsParams extends EmbeddingsParams {
  * console.log({ documentRes });
  * ```
  */
-export declare class GoogleGenerativeAIEmbeddings extends Embeddings implements GoogleGenerativeAIEmbeddingsParams {
-    apiKey?: string;
-    modelName: string;
-    model: string;
-    taskType?: TaskType;
-    title?: string;
-    stripNewLines: boolean;
-    maxBatchSize: number;
-    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[][]>;
+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.ts.map

package/dist/embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.ts","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"}

package/dist/embeddings.js

@@ -2,148 +2,97 @@ import { GoogleGenerativeAI } from "@google/generative-ai";
 import { getEnvironmentVariable } from "@langchain/core/utils/env";
 import { Embeddings } from "@langchain/core/embeddings";
 import { chunkArray } from "@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 });
- * ```
- */
-export class GoogleGenerativeAIEmbeddings extends 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 ?? 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 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 = 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 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 ?? 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 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 = 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
+export { GoogleGenerativeAIEmbeddings };
+//# sourceMappingURL=embeddings.js.map

package/dist/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","names":["fields?: GoogleGenerativeAIEmbeddingsParams","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,cACU,WAEV;CACE;CAEA,YAAY;CAEZ,QAAQ;CAER;CAEA;CAEA,gBAAgB;CAEhB,eAAe;CAEf,AAAQ;CAER,YAAYA,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,UAAU,uBAAuB,iBAAiB;AACxE,MAAI,CAAC,KAAK,OACR,OAAM,IAAI,MACR;EAOJ,KAAK,SAAS,IAAI,mBAAmB,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,mBAA+B,WACnC,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/index.cjs

@@ -1,18 +1,5 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./chat_models.cjs"), exports);
-__exportStar(require("./embeddings.cjs"), exports);
+const require_chat_models = require('./chat_models.cjs');
+const require_embeddings = require('./embeddings.cjs');
+
+exports.ChatGoogleGenerativeAI = require_chat_models.ChatGoogleGenerativeAI;
+exports.GoogleGenerativeAIEmbeddings = require_embeddings.GoogleGenerativeAIEmbeddings;

package/dist/index.d.cts

@@ -0,0 +1,3 @@
+import { BaseMessageExamplePair, ChatGoogleGenerativeAI, GoogleGenerativeAIChatCallOptions, GoogleGenerativeAIChatInput } from "./chat_models.cjs";
+import { GoogleGenerativeAIEmbeddings, GoogleGenerativeAIEmbeddingsParams } from "./embeddings.cjs";
+export { BaseMessageExamplePair, ChatGoogleGenerativeAI, GoogleGenerativeAIChatCallOptions, GoogleGenerativeAIChatInput, GoogleGenerativeAIEmbeddings, GoogleGenerativeAIEmbeddingsParams };

package/dist/index.d.ts

@@ -1,2 +1,3 @@
-export * from "./chat_models.js";
-export * from "./embeddings.js";
+import { BaseMessageExamplePair, ChatGoogleGenerativeAI, GoogleGenerativeAIChatCallOptions, GoogleGenerativeAIChatInput } from "./chat_models.js";
+import { GoogleGenerativeAIEmbeddings, GoogleGenerativeAIEmbeddingsParams } from "./embeddings.js";
+export { BaseMessageExamplePair, ChatGoogleGenerativeAI, GoogleGenerativeAIChatCallOptions, GoogleGenerativeAIChatInput, GoogleGenerativeAIEmbeddings, GoogleGenerativeAIEmbeddingsParams };

package/dist/index.js

@@ -1,2 +1,4 @@
-export * from "./chat_models.js";
-export * from "./embeddings.js";
+import { ChatGoogleGenerativeAI } from "./chat_models.js";
+import { GoogleGenerativeAIEmbeddings } from "./embeddings.js";
+
+export { ChatGoogleGenerativeAI, GoogleGenerativeAIEmbeddings };

package/dist/output_parsers.cjs

@@ -1,76 +1,48 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GoogleGenerativeAIToolsOutputParser = void 0;
-const output_parsers_1 = require("@langchain/core/output_parsers");
-const types_1 = require("@langchain/core/utils/types");
-class GoogleGenerativeAIToolsOutputParser extends output_parsers_1.BaseLLMOutputParser {
-    static lc_name() {
-        return "GoogleGenerativeAIToolsOutputParser";
-    }
-    constructor(params) {
-        super(params);
-        Object.defineProperty(this, "lc_namespace", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ["langchain", "google_genai", "output_parsers"]
-        });
-        Object.defineProperty(this, "returnId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        /** The type of tool calls to return. */
-        Object.defineProperty(this, "keyName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        /** Whether to return only the first tool call. */
-        Object.defineProperty(this, "returnSingle", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "zodSchema", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.keyName = params.keyName;
-        this.returnSingle = params.returnSingle ?? this.returnSingle;
-        this.zodSchema = params.zodSchema;
-    }
-    async _validateResult(result) {
-        if (this.zodSchema === undefined) {
-            return result;
-        }
-        const zodParsedResult = await (0, types_1.interopSafeParseAsync)(this.zodSchema, result);
-        if (zodParsedResult.success) {
-            return zodParsedResult.data;
-        }
-        else {
-            throw new output_parsers_1.OutputParserException(`Failed to parse. Text: "${JSON.stringify(result, null, 2)}". Error: ${JSON.stringify(zodParsedResult.error.issues)}`, JSON.stringify(result, null, 2));
-        }
-    }
-    async parseResult(generations) {
-        const tools = generations.flatMap((generation) => {
-            const { message } = generation;
-            if (!("tool_calls" in message) || !Array.isArray(message.tool_calls)) {
-                return [];
-            }
-            return message.tool_calls;
-        });
-        if (tools[0] === undefined) {
-            throw new Error("No parseable tool calls provided to GoogleGenerativeAIToolsOutputParser.");
-        }
-        const [tool] = tools;
-        const validatedResult = await this._validateResult(tool.args);
-        return validatedResult;
-    }
-}
+const require_rolldown_runtime = require('./_virtual/rolldown_runtime.cjs');
+const __langchain_core_utils_types = require_rolldown_runtime.__toESM(require("@langchain/core/utils/types"));
+const __langchain_core_output_parsers = require_rolldown_runtime.__toESM(require("@langchain/core/output_parsers"));
+
+//#region src/output_parsers.ts
+var GoogleGenerativeAIToolsOutputParser = class extends __langchain_core_output_parsers.BaseLLMOutputParser {
+	static lc_name() {
+		return "GoogleGenerativeAIToolsOutputParser";
+	}
+	lc_namespace = [
+		"langchain",
+		"google_genai",
+		"output_parsers"
+	];
+	returnId = false;
+	/** The type of tool calls to return. */
+	keyName;
+	/** Whether to return only the first tool call. */
+	returnSingle = false;
+	zodSchema;
+	constructor(params) {
+		super(params);
+		this.keyName = params.keyName;
+		this.returnSingle = params.returnSingle ?? this.returnSingle;
+		this.zodSchema = params.zodSchema;
+	}
+	async _validateResult(result) {
+		if (this.zodSchema === void 0) return result;
+		const zodParsedResult = await (0, __langchain_core_utils_types.interopSafeParseAsync)(this.zodSchema, result);
+		if (zodParsedResult.success) return zodParsedResult.data;
+		else throw new __langchain_core_output_parsers.OutputParserException(`Failed to parse. Text: "${JSON.stringify(result, null, 2)}". Error: ${JSON.stringify(zodParsedResult.error.issues)}`, JSON.stringify(result, null, 2));
+	}
+	async parseResult(generations) {
+		const tools = generations.flatMap((generation) => {
+			const { message } = generation;
+			if (!("tool_calls" in message) || !Array.isArray(message.tool_calls)) return [];
+			return message.tool_calls;
+		});
+		if (tools[0] === void 0) throw new Error("No parseable tool calls provided to GoogleGenerativeAIToolsOutputParser.");
+		const [tool] = tools;
+		const validatedResult = await this._validateResult(tool.args);
+		return validatedResult;
+	}
+};
+
+//#endregion
 exports.GoogleGenerativeAIToolsOutputParser = GoogleGenerativeAIToolsOutputParser;
+//# sourceMappingURL=output_parsers.cjs.map

package/dist/output_parsers.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"output_parsers.cjs","names":["BaseLLMOutputParser","params: GoogleGenerativeAIToolsOutputParserParams<T>","result: unknown","OutputParserException","generations: ChatGeneration[]"],"sources":["../src/output_parsers.ts"],"sourcesContent":["import {\n  BaseLLMOutputParser,\n  OutputParserException,\n} from \"@langchain/core/output_parsers\";\nimport { ChatGeneration } from \"@langchain/core/outputs\";\nimport { ToolCall } from \"@langchain/core/messages/tool\";\nimport {\n  InteropZodType,\n  interopSafeParseAsync,\n} from \"@langchain/core/utils/types\";\nimport { JsonOutputKeyToolsParserParamsInterop } from \"@langchain/core/output_parsers/openai_tools\";\n\ninterface GoogleGenerativeAIToolsOutputParserParams<\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  T extends Record<string, any>\n> extends JsonOutputKeyToolsParserParamsInterop<T> {}\n\nexport class GoogleGenerativeAIToolsOutputParser<\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  T extends Record<string, any> = Record<string, any>\n> extends BaseLLMOutputParser<T> {\n  static lc_name() {\n    return \"GoogleGenerativeAIToolsOutputParser\";\n  }\n\n  lc_namespace = [\"langchain\", \"google_genai\", \"output_parsers\"];\n\n  returnId = false;\n\n  /** The type of tool calls to return. */\n  keyName: string;\n\n  /** Whether to return only the first tool call. */\n  returnSingle = false;\n\n  zodSchema?: InteropZodType<T>;\n\n  constructor(params: GoogleGenerativeAIToolsOutputParserParams<T>) {\n    super(params);\n    this.keyName = params.keyName;\n    this.returnSingle = params.returnSingle ?? this.returnSingle;\n    this.zodSchema = params.zodSchema;\n  }\n\n  protected async _validateResult(result: unknown): Promise<T> {\n    if (this.zodSchema === undefined) {\n      return result as T;\n    }\n    const zodParsedResult = await interopSafeParseAsync(this.zodSchema, result);\n    if (zodParsedResult.success) {\n      return zodParsedResult.data;\n    } else {\n      throw new OutputParserException(\n        `Failed to parse. Text: \"${JSON.stringify(\n          result,\n          null,\n          2\n        )}\". Error: ${JSON.stringify(zodParsedResult.error.issues)}`,\n        JSON.stringify(result, null, 2)\n      );\n    }\n  }\n\n  async parseResult(generations: ChatGeneration[]): Promise<T> {\n    const tools = generations.flatMap((generation) => {\n      const { message } = generation;\n      if (!(\"tool_calls\" in message) || !Array.isArray(message.tool_calls)) {\n        return [];\n      }\n      return message.tool_calls as ToolCall[];\n    });\n    if (tools[0] === undefined) {\n      throw new Error(\n        \"No parseable tool calls provided to GoogleGenerativeAIToolsOutputParser.\"\n      );\n    }\n    const [tool] = tools;\n    const validatedResult = await this._validateResult(tool.args);\n    return validatedResult;\n  }\n}\n"],"mappings":";;;;;AAiBA,IAAa,sCAAb,cAGUA,oDAAuB;CAC/B,OAAO,UAAU;AACf,SAAO;CACR;CAED,eAAe;EAAC;EAAa;EAAgB;CAAiB;CAE9D,WAAW;;CAGX;;CAGA,eAAe;CAEf;CAEA,YAAYC,QAAsD;EAChE,MAAM,OAAO;EACb,KAAK,UAAU,OAAO;EACtB,KAAK,eAAe,OAAO,gBAAgB,KAAK;EAChD,KAAK,YAAY,OAAO;CACzB;CAED,MAAgB,gBAAgBC,QAA6B;AAC3D,MAAI,KAAK,cAAc,OACrB,QAAO;EAET,MAAM,kBAAkB,8DAA4B,KAAK,WAAW,OAAO;AAC3E,MAAI,gBAAgB,QAClB,QAAO,gBAAgB;MAEvB,OAAM,IAAIC,sDACR,CAAC,wBAAwB,EAAE,KAAK,UAC9B,QACA,MACA,EACD,CAAC,UAAU,EAAE,KAAK,UAAU,gBAAgB,MAAM,OAAO,EAAE,EAC5D,KAAK,UAAU,QAAQ,MAAM,EAAE;CAGpC;CAED,MAAM,YAAYC,aAA2C;EAC3D,MAAM,QAAQ,YAAY,QAAQ,CAAC,eAAe;GAChD,MAAM,EAAE,SAAS,GAAG;AACpB,OAAI,EAAE,gBAAgB,YAAY,CAAC,MAAM,QAAQ,QAAQ,WAAW,CAClE,QAAO,CAAE;AAEX,UAAO,QAAQ;EAChB,EAAC;AACF,MAAI,MAAM,OAAO,OACf,OAAM,IAAI,MACR;EAGJ,MAAM,CAAC,KAAK,GAAG;EACf,MAAM,kBAAkB,MAAM,KAAK,gBAAgB,KAAK,KAAK;AAC7D,SAAO;CACR;AACF"}