langchain 0.0.203 → 0.0.204-rc.0

package/dist/embeddings/googlepalm.d.ts

@@ -1,61 +1 @@
-import { Embeddings, EmbeddingsParams } from "./base.js";
-/**
- * Interface that extends EmbeddingsParams and defines additional
- * parameters specific to the GooglePaLMEmbeddings class.
- */
-export interface GooglePaLMEmbeddingsParams extends EmbeddingsParams {
-    /**
-     * Model Name to use
-     *
-     * Note: The format must follow the pattern - `models/{model}`
-     */
-    modelName?: string;
-    /**
-     * Google Palm API key to use
-     */
-    apiKey?: string;
-}
-/**
- * Class that extends the Embeddings class and provides methods for
- * generating embeddings using the Google Palm API.
- * @example
- * ```typescript
- * const model = new GooglePaLMEmbeddings({
- *   apiKey: "<YOUR API KEY>",
- *   modelName: "models/embedding-gecko-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 declare class GooglePaLMEmbeddings extends Embeddings implements GooglePaLMEmbeddingsParams {
-    apiKey?: string;
-    modelName: string;
-    private client;
-    constructor(fields?: GooglePaLMEmbeddingsParams);
-    protected _embedText(text: 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[][]>;
-}
+export * from "@langchain/community/embeddings/googlepalm";

package/dist/embeddings/googlepalm.js

@@ -1,86 +1 @@
-import { TextServiceClient } from "@google-ai/generativelanguage";
-import { GoogleAuth } from "google-auth-library";
-import { Embeddings } from "./base.js";
-import { getEnvironmentVariable } from "../util/env.js";
-/**
- * Class that extends the Embeddings class and provides methods for
- * generating embeddings using the Google Palm API.
- * @example
- * ```typescript
- * const model = new GooglePaLMEmbeddings({
- *   apiKey: "<YOUR API KEY>",
- *   modelName: "models/embedding-gecko-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 GooglePaLMEmbeddings 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: "models/embedding-gecko-001"
-        });
-        Object.defineProperty(this, "client", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.modelName = fields?.modelName ?? this.modelName;
-        this.apiKey =
-            fields?.apiKey ?? getEnvironmentVariable("GOOGLE_PALM_API_KEY");
-        if (!this.apiKey) {
-            throw new Error("Please set an API key for Google Palm 2 in the environment variable GOOGLE_PALM_API_KEY or in the `apiKey` field of the GooglePalm constructor");
-        }
-        this.client = new TextServiceClient({
-            authClient: new GoogleAuth().fromAPIKey(this.apiKey),
-        });
-    }
-    async _embedText(text) {
-        // replace newlines, which can negatively affect performance.
-        const cleanedText = text.replace(/\n/g, " ");
-        const res = await this.client.embedText({
-            model: this.modelName,
-            text: cleanedText,
-        });
-        return res[0].embedding?.value ?? [];
-    }
-    /**
-     * 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.callWithOptions({}, this._embedText.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 Promise.all(documents.map((document) => this._embedText(document)));
-    }
-}
+export * from "@langchain/community/embeddings/googlepalm";

package/dist/embeddings/googlevertexai.cjs

@@ -1,83 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GoogleVertexAIEmbeddings = void 0;
-const google_auth_library_1 = require("google-auth-library");
-const base_js_1 = require("./base.cjs");
-const googlevertexai_connection_js_1 = require("../util/googlevertexai-connection.cjs");
-const chunk_js_1 = require("../util/chunk.cjs");
-/**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * the embeddings generated by Large Language Models.
- *
- * To use, you will need to have one of the following authentication
- * methods in place:
- * - You are logged into an account permitted to the Google Cloud project
- *   using Vertex AI.
- * - You are running this on a machine using a service account permitted to
- *   the Google Cloud project using Vertex AI.
- * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
- *   path of a credentials file for a service account permitted to the
- *   Google Cloud project using Vertex AI.
- * @example
- * ```typescript
- * const model = new GoogleVertexAIEmbeddings();
- * const res = await model.embedQuery(
- *   "What would be a good company name for a company that makes colorful socks?"
- * );
- * console.log({ res });
- * ```
- */
-class GoogleVertexAIEmbeddings extends base_js_1.Embeddings {
-    constructor(fields) {
-        super(fields ?? {});
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "textembedding-gecko"
-        });
-        Object.defineProperty(this, "connection", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.model = fields?.model ?? this.model;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, new google_auth_library_1.GoogleAuth({
-            scopes: "https://www.googleapis.com/auth/cloud-platform",
-            ...fields?.authOptions,
-        }));
-    }
-    /**
-     * Takes an array of documents as input and returns a promise that
-     * resolves to a 2D array of embeddings for each document. It splits the
-     * documents into chunks and makes requests to the Google Vertex AI API to
-     * generate embeddings.
-     * @param documents An array of documents to be embedded.
-     * @returns A promise that resolves to a 2D array of embeddings for each document.
-     */
-    async embedDocuments(documents) {
-        const instanceChunks = (0, chunk_js_1.chunkArray)(documents.map((document) => ({
-            content: document,
-        })), 5); // Vertex AI accepts max 5 instances per prediction
-        const parameters = {};
-        const options = {};
-        const responses = await Promise.all(instanceChunks.map((instances) => this.connection.request(instances, parameters, options)));
-        const result = responses
-            ?.map((response) => response?.data?.predictions?.map((result) => result.embeddings.values) ?? [])
-            .flat() ?? [];
-        return result;
+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]; } };
     }
-    /**
-     * Takes a document as input and returns a promise that resolves to an
-     * embedding for the document. It calls the embedDocuments method with the
-     * document as the input.
-     * @param document A document to be embedded.
-     * @returns A promise that resolves to an embedding for the document.
-     */
-    async embedQuery(document) {
-        const data = await this.embedDocuments([document]);
-        return data[0];
-    }
-}
-exports.GoogleVertexAIEmbeddings = GoogleVertexAIEmbeddings;
+    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("@langchain/community/embeddings/googlevertexai"), exports);

package/dist/embeddings/googlevertexai.d.ts

@@ -1,54 +1 @@
-import { GoogleAuthOptions } from "google-auth-library";
-import { Embeddings, EmbeddingsParams } from "./base.js";
-import { GoogleVertexAIBaseLLMInput } from "../types/googlevertexai-types.js";
-/**
- * Defines the parameters required to initialize a
- * GoogleVertexAIEmbeddings instance. It extends EmbeddingsParams and
- * GoogleVertexAIConnectionParams.
- */
-export interface GoogleVertexAIEmbeddingsParams extends EmbeddingsParams, GoogleVertexAIBaseLLMInput<GoogleAuthOptions> {
-}
-/**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * the embeddings generated by Large Language Models.
- *
- * To use, you will need to have one of the following authentication
- * methods in place:
- * - You are logged into an account permitted to the Google Cloud project
- *   using Vertex AI.
- * - You are running this on a machine using a service account permitted to
- *   the Google Cloud project using Vertex AI.
- * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
- *   path of a credentials file for a service account permitted to the
- *   Google Cloud project using Vertex AI.
- * @example
- * ```typescript
- * const model = new GoogleVertexAIEmbeddings();
- * const res = await model.embedQuery(
- *   "What would be a good company name for a company that makes colorful socks?"
- * );
- * console.log({ res });
- * ```
- */
-export declare class GoogleVertexAIEmbeddings extends Embeddings implements GoogleVertexAIEmbeddingsParams {
-    model: string;
-    private connection;
-    constructor(fields?: GoogleVertexAIEmbeddingsParams);
-    /**
-     * Takes an array of documents as input and returns a promise that
-     * resolves to a 2D array of embeddings for each document. It splits the
-     * documents into chunks and makes requests to the Google Vertex AI API to
-     * generate embeddings.
-     * @param documents An array of documents to be embedded.
-     * @returns A promise that resolves to a 2D array of embeddings for each document.
-     */
-    embedDocuments(documents: string[]): Promise<number[][]>;
-    /**
-     * Takes a document as input and returns a promise that resolves to an
-     * embedding for the document. It calls the embedDocuments method with the
-     * document as the input.
-     * @param document A document to be embedded.
-     * @returns A promise that resolves to an embedding for the document.
-     */
-    embedQuery(document: string): Promise<number[]>;
-}
+export * from "@langchain/community/embeddings/googlevertexai";

package/dist/embeddings/googlevertexai.js

@@ -1,79 +1 @@
-import { GoogleAuth } from "google-auth-library";
-import { Embeddings } from "./base.js";
-import { GoogleVertexAILLMConnection } from "../util/googlevertexai-connection.js";
-import { chunkArray } from "../util/chunk.js";
-/**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * the embeddings generated by Large Language Models.
- *
- * To use, you will need to have one of the following authentication
- * methods in place:
- * - You are logged into an account permitted to the Google Cloud project
- *   using Vertex AI.
- * - You are running this on a machine using a service account permitted to
- *   the Google Cloud project using Vertex AI.
- * - The `GOOGLE_APPLICATION_CREDENTIALS` environment variable is set to the
- *   path of a credentials file for a service account permitted to the
- *   Google Cloud project using Vertex AI.
- * @example
- * ```typescript
- * const model = new GoogleVertexAIEmbeddings();
- * const res = await model.embedQuery(
- *   "What would be a good company name for a company that makes colorful socks?"
- * );
- * console.log({ res });
- * ```
- */
-export class GoogleVertexAIEmbeddings extends Embeddings {
-    constructor(fields) {
-        super(fields ?? {});
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "textembedding-gecko"
-        });
-        Object.defineProperty(this, "connection", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.model = fields?.model ?? this.model;
-        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, new GoogleAuth({
-            scopes: "https://www.googleapis.com/auth/cloud-platform",
-            ...fields?.authOptions,
-        }));
-    }
-    /**
-     * Takes an array of documents as input and returns a promise that
-     * resolves to a 2D array of embeddings for each document. It splits the
-     * documents into chunks and makes requests to the Google Vertex AI API to
-     * generate embeddings.
-     * @param documents An array of documents to be embedded.
-     * @returns A promise that resolves to a 2D array of embeddings for each document.
-     */
-    async embedDocuments(documents) {
-        const instanceChunks = chunkArray(documents.map((document) => ({
-            content: document,
-        })), 5); // Vertex AI accepts max 5 instances per prediction
-        const parameters = {};
-        const options = {};
-        const responses = await Promise.all(instanceChunks.map((instances) => this.connection.request(instances, parameters, options)));
-        const result = responses
-            ?.map((response) => response?.data?.predictions?.map((result) => result.embeddings.values) ?? [])
-            .flat() ?? [];
-        return result;
-    }
-    /**
-     * Takes a document as input and returns a promise that resolves to an
-     * embedding for the document. It calls the embedDocuments method with the
-     * document as the input.
-     * @param document A document to be embedded.
-     * @returns A promise that resolves to an embedding for the document.
-     */
-    async embedQuery(document) {
-        const data = await this.embedDocuments([document]);
-        return data[0];
-    }
-}
+export * from "@langchain/community/embeddings/googlevertexai";

package/dist/embeddings/gradient_ai.cjs

@@ -1,103 +1,17 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GradientEmbeddings = void 0;
-const nodejs_sdk_1 = require("@gradientai/nodejs-sdk");
-const env_js_1 = require("../util/env.cjs");
-const chunk_js_1 = require("../util/chunk.cjs");
-const base_js_1 = require("./base.cjs");
-/**
- * Class for generating embeddings using the Gradient AI's API. Extends the
- * Embeddings class and implements GradientEmbeddingsParams and
- */
-class GradientEmbeddings extends base_js_1.Embeddings {
-    constructor(fields) {
-        super(fields);
-        Object.defineProperty(this, "gradientAccessKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "workspaceId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "batchSize", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 128
-        });
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.gradientAccessKey =
-            fields?.gradientAccessKey ??
-                (0, env_js_1.getEnvironmentVariable)("GRADIENT_ACCESS_TOKEN");
-        this.workspaceId =
-            fields?.workspaceId ?? (0, env_js_1.getEnvironmentVariable)("GRADIENT_WORKSPACE_ID");
-        if (!this.gradientAccessKey) {
-            throw new Error("Missing Gradient AI Access Token");
-        }
-        if (!this.workspaceId) {
-            throw new Error("Missing Gradient AI Workspace ID");
-        }
-    }
-    /**
-     * Method to generate embeddings for an array of documents. Splits the
-     * documents into batches and makes requests to the Gradient API to generate
-     * embeddings.
-     * @param texts Array of documents to generate embeddings for.
-     * @returns Promise that resolves to a 2D array of embeddings for each document.
-     */
-    async embedDocuments(texts) {
-        await this.setModel();
-        const mappedTexts = texts.map((text) => ({ input: text }));
-        const batches = (0, chunk_js_1.chunkArray)(mappedTexts, this.batchSize);
-        const batchRequests = batches.map((batch) => this.caller.call(async () => this.model.generateEmbeddings({
-            inputs: batch,
-        })));
-        const batchResponses = await Promise.all(batchRequests);
-        const embeddings = [];
-        for (let i = 0; i < batchResponses.length; i += 1) {
-            const batch = batches[i];
-            const { embeddings: batchResponse } = batchResponses[i];
-            for (let j = 0; j < batch.length; j += 1) {
-                embeddings.push(batchResponse[j].embedding);
-            }
-        }
-        return embeddings;
-    }
-    /**
-     * Method to generate an embedding for a single document. Calls the
-     * embedDocuments method with the document as the input.
-     * @param text Document to generate an embedding for.
-     * @returns Promise that resolves to an embedding for the document.
-     */
-    async embedQuery(text) {
-        const data = await this.embedDocuments([text]);
-        return data[0];
+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]; } };
     }
-    /**
-     * Method to set the model to use for generating embeddings.
-     * @sets the class' `model` value to that of the retrieved Embeddings Model.
-     */
-    async setModel() {
-        if (this.model)
-            return;
-        const gradient = new nodejs_sdk_1.Gradient({
-            accessToken: this.gradientAccessKey,
-            workspaceId: this.workspaceId,
-        });
-        this.model = await gradient.getEmbeddingsModel({
-            slug: "bge-large",
-        });
-    }
-}
-exports.GradientEmbeddings = GradientEmbeddings;
+    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("@langchain/community/embeddings/gradient_ai"), exports);

package/dist/embeddings/gradient_ai.d.ts

@@ -1,48 +1 @@
-import { Embeddings, EmbeddingsParams } from "./base.js";
-/**
- * Interface for GradientEmbeddings parameters. Extends EmbeddingsParams and
- * defines additional parameters specific to the GradientEmbeddings class.
- */
-export interface GradientEmbeddingsParams extends EmbeddingsParams {
-    /**
-     * Gradient AI Access Token.
-     * Provide Access Token if you do not wish to automatically pull from env.
-     */
-    gradientAccessKey?: string;
-    /**
-     * Gradient Workspace Id.
-     * Provide workspace id if you do not wish to automatically pull from env.
-     */
-    workspaceId?: string;
-}
-/**
- * Class for generating embeddings using the Gradient AI's API. Extends the
- * Embeddings class and implements GradientEmbeddingsParams and
- */
-export declare class GradientEmbeddings extends Embeddings implements GradientEmbeddingsParams {
-    gradientAccessKey?: string;
-    workspaceId?: string;
-    batchSize: number;
-    model: any;
-    constructor(fields: GradientEmbeddingsParams);
-    /**
-     * Method to generate embeddings for an array of documents. Splits the
-     * documents into batches and makes requests to the Gradient API to generate
-     * embeddings.
-     * @param texts Array of documents to generate embeddings for.
-     * @returns Promise that resolves to a 2D array of embeddings for each document.
-     */
-    embedDocuments(texts: string[]): Promise<number[][]>;
-    /**
-     * Method to generate an embedding for a single document. Calls the
-     * embedDocuments method with the document as the input.
-     * @param text Document to generate an embedding for.
-     * @returns Promise that resolves to an embedding for the document.
-     */
-    embedQuery(text: string): Promise<number[]>;
-    /**
-     * Method to set the model to use for generating embeddings.
-     * @sets the class' `model` value to that of the retrieved Embeddings Model.
-     */
-    setModel(): Promise<void>;
-}
+export * from "@langchain/community/embeddings/gradient_ai";

package/dist/embeddings/gradient_ai.js

@@ -1,99 +1 @@
-import { Gradient } from "@gradientai/nodejs-sdk";
-import { getEnvironmentVariable } from "../util/env.js";
-import { chunkArray } from "../util/chunk.js";
-import { Embeddings } from "./base.js";
-/**
- * Class for generating embeddings using the Gradient AI's API. Extends the
- * Embeddings class and implements GradientEmbeddingsParams and
- */
-export class GradientEmbeddings extends Embeddings {
-    constructor(fields) {
-        super(fields);
-        Object.defineProperty(this, "gradientAccessKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "workspaceId", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "batchSize", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 128
-        });
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.gradientAccessKey =
-            fields?.gradientAccessKey ??
-                getEnvironmentVariable("GRADIENT_ACCESS_TOKEN");
-        this.workspaceId =
-            fields?.workspaceId ?? getEnvironmentVariable("GRADIENT_WORKSPACE_ID");
-        if (!this.gradientAccessKey) {
-            throw new Error("Missing Gradient AI Access Token");
-        }
-        if (!this.workspaceId) {
-            throw new Error("Missing Gradient AI Workspace ID");
-        }
-    }
-    /**
-     * Method to generate embeddings for an array of documents. Splits the
-     * documents into batches and makes requests to the Gradient API to generate
-     * embeddings.
-     * @param texts Array of documents to generate embeddings for.
-     * @returns Promise that resolves to a 2D array of embeddings for each document.
-     */
-    async embedDocuments(texts) {
-        await this.setModel();
-        const mappedTexts = texts.map((text) => ({ input: text }));
-        const batches = chunkArray(mappedTexts, this.batchSize);
-        const batchRequests = batches.map((batch) => this.caller.call(async () => this.model.generateEmbeddings({
-            inputs: batch,
-        })));
-        const batchResponses = await Promise.all(batchRequests);
-        const embeddings = [];
-        for (let i = 0; i < batchResponses.length; i += 1) {
-            const batch = batches[i];
-            const { embeddings: batchResponse } = batchResponses[i];
-            for (let j = 0; j < batch.length; j += 1) {
-                embeddings.push(batchResponse[j].embedding);
-            }
-        }
-        return embeddings;
-    }
-    /**
-     * Method to generate an embedding for a single document. Calls the
-     * embedDocuments method with the document as the input.
-     * @param text Document to generate an embedding for.
-     * @returns Promise that resolves to an embedding for the document.
-     */
-    async embedQuery(text) {
-        const data = await this.embedDocuments([text]);
-        return data[0];
-    }
-    /**
-     * Method to set the model to use for generating embeddings.
-     * @sets the class' `model` value to that of the retrieved Embeddings Model.
-     */
-    async setModel() {
-        if (this.model)
-            return;
-        const gradient = new Gradient({
-            accessToken: this.gradientAccessKey,
-            workspaceId: this.workspaceId,
-        });
-        this.model = await gradient.getEmbeddingsModel({
-            slug: "bge-large",
-        });
-    }
-}
+export * from "@langchain/community/embeddings/gradient_ai";