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

package/dist/node_modules/@langchain/google-genai/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/node_modules/@langchain/google-genai/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/node_modules/@langchain/google-genai/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