langchain 0.0.146 → 0.0.148

package/dist/evaluation/loader.js

@@ -0,0 +1,56 @@
+import { CriteriaEvalChain, LabeledCriteriaEvalChain, } from "./criteria/index.js";
+import { ChatOpenAI } from "../chat_models/openai.js";
+import { LabeledPairwiseStringEvalChain, PairwiseStringEvalChain, } from "./comparison/index.js";
+import { EmbeddingDistanceEvalChain, PairwiseEmbeddingDistanceEvalChain, } from "./embedding_distance/index.js";
+import { TrajectoryEvalChain } from "./agents/index.js";
+import { BaseChatModel } from "../chat_models/base.js";
+/**
+ * Load the requested evaluation chain specified by a string
+ * @param type The type of evaluator to load.
+ * @param options
+ *        - llm The language model to use for the evaluator.
+ *        - criteria The criteria to use for the evaluator.
+ *        - agentTools A list of tools available to the agent,for TrajectoryEvalChain.
+ */
+export async function loadEvaluator(type, options) {
+    const { llm, chainOptions, criteria, agentTools } = options || {};
+    const llm_ = llm ??
+        new ChatOpenAI({
+            modelName: "gpt-4",
+            temperature: 0.0,
+        });
+    let evaluator;
+    switch (type) {
+        case "criteria":
+            evaluator = await CriteriaEvalChain.fromLLM(llm_, criteria, chainOptions);
+            break;
+        case "labeled_criteria":
+            evaluator = await LabeledCriteriaEvalChain.fromLLM(llm_, criteria, chainOptions);
+            break;
+        case "pairwise_string":
+            evaluator = await PairwiseStringEvalChain.fromLLM(llm_, criteria, chainOptions);
+            break;
+        case "labeled_pairwise_string":
+            evaluator = await LabeledPairwiseStringEvalChain.fromLLM(llm_, criteria, chainOptions);
+            break;
+        case "trajectory":
+            // eslint-disable-next-line no-instanceof/no-instanceof
+            if (!(llm_ instanceof BaseChatModel)) {
+                throw new Error("LLM must be an instance of a base chat model.");
+            }
+            evaluator = await TrajectoryEvalChain.fromLLM(llm_, agentTools, chainOptions);
+            break;
+        case "embedding_distance":
+            evaluator = new EmbeddingDistanceEvalChain({
+                embedding: options?.embedding,
+                distanceMetric: options?.distanceMetric,
+            });
+            break;
+        case "pairwise_embedding_distance":
+            evaluator = new PairwiseEmbeddingDistanceEvalChain({});
+            break;
+        default:
+            throw new Error(`Unknown type: ${type}`);
+    }
+    return evaluator;
+}

package/dist/evaluation/types.cjs

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/evaluation/types.d.ts

@@ -0,0 +1,35 @@
+import { AgentTrajectoryEvaluator, LLMPairwiseStringEvaluator, LLMStringEvaluator, PairwiseStringEvaluator, StringEvaluator } from "./base.js";
+export interface EvaluatorType {
+    /**
+     * The criteria evaluator, which evaluates a model based on a
+     * custom set of criteria without any reference labels.
+     * */
+    criteria: LLMStringEvaluator;
+    /**
+     * The labeled criteria evaluator, which evaluates a model based on a
+     * custom set of criteria, with a reference label.
+     * */
+    labeled_criteria: LLMStringEvaluator;
+    /**
+     *  The pairwise string evaluator, which predicts the preferred prediction from
+     *  between two models.
+     */
+    pairwise_string: LLMPairwiseStringEvaluator;
+    /**
+     * The labeled pairwise string evaluator, which predicts the preferred prediction
+     * from between two models based on a ground truth reference label.
+     * */
+    labeled_pairwise_string: LLMPairwiseStringEvaluator;
+    /**
+     * The agent trajectory evaluator, which grades the agent's intermediate steps.
+     */
+    trajectory: AgentTrajectoryEvaluator;
+    /**
+     * Compare a prediction to a reference label using embedding distance.
+     * */
+    embedding_distance: StringEvaluator;
+    /**
+     * Compare two predictions using embedding distance.
+     * */
+    pairwise_embedding_distance: PairwiseStringEvaluator;
+}

package/dist/evaluation/types.js

@@ -0,0 +1 @@
+export {};

package/dist/experimental/llms/bittensor.cjs

@@ -0,0 +1,141 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NIBittensorLLM = void 0;
+const base_js_1 = require("../../llms/base.cjs");
+/**
+ * Class representing the Neural Internet language model powerd by Bittensor, a decentralized network
+ * full of different AI models.
+ * To analyze API_KEYS and logs of you usage visit
+ *      https://api.neuralinternet.ai/api-keys
+ *      https://api.neuralinternet.ai/logs
+ */
+class NIBittensorLLM extends base_js_1.LLM {
+    static lc_name() {
+        return "NIBittensorLLM";
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "systemPrompt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "topResponses", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.systemPrompt =
+            fields?.systemPrompt ??
+                "You are an assistant which is created by Neural Internet(NI) in decentralized network named as a Bittensor. Your task is to provide accurate response based on user prompt";
+        this.topResponses = fields?.topResponses;
+    }
+    _llmType() {
+        return "NIBittensorLLM";
+    }
+    /** Call out to NIBittensorLLM's complete endpoint.
+     Args:
+         prompt: The prompt to pass into the model.
+  
+         Returns: The string generated by the model.
+  
+     Example:
+     let response = niBittensorLLM.call("Tell me a joke.");
+     */
+    async _call(prompt) {
+        try {
+            // Retrieve API KEY
+            const apiKeyResponse = await fetch("https://test.neuralinternet.ai/admin/api-keys/");
+            if (!apiKeyResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const apiKeysData = await apiKeyResponse.json();
+            const apiKey = apiKeysData[0].api_key;
+            const headers = {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${apiKey}`,
+                "Endpoint-Version": "2023-05-19",
+            };
+            if (this.topResponses !== undefined) {
+                this.topResponses = this.topResponses > 100 ? 100 : this.topResponses;
+            }
+            else {
+                this.topResponses = 0;
+            }
+            const minerResponse = await fetch("https://test.neuralinternet.ai/top_miner_uids", { headers });
+            if (!minerResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const uids = await minerResponse.json();
+            if (Array.isArray(uids) && uids.length && this.topResponses === 0) {
+                for (const uid of uids) {
+                    try {
+                        const payload = {
+                            uids: [uid],
+                            messages: [
+                                { role: "system", content: this.systemPrompt },
+                                { role: "user", content: prompt },
+                            ],
+                        };
+                        const response = await fetch("https://test.neuralinternet.ai/chat", {
+                            method: "POST",
+                            headers,
+                            body: JSON.stringify(payload),
+                        });
+                        if (!response.ok) {
+                            throw new Error("Network response was not ok");
+                        }
+                        const chatData = await response.json();
+                        if (chatData.choices) {
+                            return chatData.choices[0].message.content;
+                        }
+                    }
+                    catch (error) {
+                        continue;
+                    }
+                }
+            }
+            // For top miner based on bittensor response
+            if (this.topResponses === 0) {
+                this.topResponses = 10;
+            }
+            const payload = {
+                top_n: this.topResponses,
+                messages: [
+                    { role: "system", content: this.systemPrompt },
+                    { role: "user", content: prompt },
+                ],
+            };
+            const response = await fetch("https://test.neuralinternet.ai/chat", {
+                method: "POST",
+                headers,
+                body: JSON.stringify(payload),
+            });
+            if (!response.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const responseData = await response.json();
+            if (this.topResponses) {
+                return responseData;
+            }
+            else if (responseData.choices) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                const temp = responseData.choices;
+                return temp[0].message.content;
+            }
+        }
+        catch (error) {
+            return "Sorry I am unable to provide response now, Please try again later.";
+        }
+        return "default";
+    }
+    identifyingParams() {
+        return {
+            systemPrompt: this.systemPrompt,
+            topResponses: this.topResponses,
+        };
+    }
+}
+exports.NIBittensorLLM = NIBittensorLLM;

package/dist/experimental/llms/bittensor.d.ts

@@ -0,0 +1,33 @@
+import { BaseLLMParams, LLM } from "../../llms/base.js";
+export interface BittensorInput extends BaseLLMParams {
+    systemPrompt?: string | null | undefined;
+    topResponses?: number | undefined;
+}
+/**
+ * Class representing the Neural Internet language model powerd by Bittensor, a decentralized network
+ * full of different AI models.
+ * To analyze API_KEYS and logs of you usage visit
+ *      https://api.neuralinternet.ai/api-keys
+ *      https://api.neuralinternet.ai/logs
+ */
+export declare class NIBittensorLLM extends LLM implements BittensorInput {
+    static lc_name(): string;
+    systemPrompt: string;
+    topResponses: number | undefined;
+    constructor(fields?: BittensorInput);
+    _llmType(): string;
+    /** Call out to NIBittensorLLM's complete endpoint.
+     Args:
+         prompt: The prompt to pass into the model.
+  
+         Returns: The string generated by the model.
+  
+     Example:
+     let response = niBittensorLLM.call("Tell me a joke.");
+     */
+    _call(prompt: string): Promise<string>;
+    identifyingParams(): {
+        systemPrompt: string | null | undefined;
+        topResponses: number | undefined;
+    };
+}

package/dist/experimental/llms/bittensor.js

@@ -0,0 +1,137 @@
+import { LLM } from "../../llms/base.js";
+/**
+ * Class representing the Neural Internet language model powerd by Bittensor, a decentralized network
+ * full of different AI models.
+ * To analyze API_KEYS and logs of you usage visit
+ *      https://api.neuralinternet.ai/api-keys
+ *      https://api.neuralinternet.ai/logs
+ */
+export class NIBittensorLLM extends LLM {
+    static lc_name() {
+        return "NIBittensorLLM";
+    }
+    constructor(fields) {
+        super(fields ?? {});
+        Object.defineProperty(this, "systemPrompt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "topResponses", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.systemPrompt =
+            fields?.systemPrompt ??
+                "You are an assistant which is created by Neural Internet(NI) in decentralized network named as a Bittensor. Your task is to provide accurate response based on user prompt";
+        this.topResponses = fields?.topResponses;
+    }
+    _llmType() {
+        return "NIBittensorLLM";
+    }
+    /** Call out to NIBittensorLLM's complete endpoint.
+     Args:
+         prompt: The prompt to pass into the model.
+  
+         Returns: The string generated by the model.
+  
+     Example:
+     let response = niBittensorLLM.call("Tell me a joke.");
+     */
+    async _call(prompt) {
+        try {
+            // Retrieve API KEY
+            const apiKeyResponse = await fetch("https://test.neuralinternet.ai/admin/api-keys/");
+            if (!apiKeyResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const apiKeysData = await apiKeyResponse.json();
+            const apiKey = apiKeysData[0].api_key;
+            const headers = {
+                "Content-Type": "application/json",
+                Authorization: `Bearer ${apiKey}`,
+                "Endpoint-Version": "2023-05-19",
+            };
+            if (this.topResponses !== undefined) {
+                this.topResponses = this.topResponses > 100 ? 100 : this.topResponses;
+            }
+            else {
+                this.topResponses = 0;
+            }
+            const minerResponse = await fetch("https://test.neuralinternet.ai/top_miner_uids", { headers });
+            if (!minerResponse.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const uids = await minerResponse.json();
+            if (Array.isArray(uids) && uids.length && this.topResponses === 0) {
+                for (const uid of uids) {
+                    try {
+                        const payload = {
+                            uids: [uid],
+                            messages: [
+                                { role: "system", content: this.systemPrompt },
+                                { role: "user", content: prompt },
+                            ],
+                        };
+                        const response = await fetch("https://test.neuralinternet.ai/chat", {
+                            method: "POST",
+                            headers,
+                            body: JSON.stringify(payload),
+                        });
+                        if (!response.ok) {
+                            throw new Error("Network response was not ok");
+                        }
+                        const chatData = await response.json();
+                        if (chatData.choices) {
+                            return chatData.choices[0].message.content;
+                        }
+                    }
+                    catch (error) {
+                        continue;
+                    }
+                }
+            }
+            // For top miner based on bittensor response
+            if (this.topResponses === 0) {
+                this.topResponses = 10;
+            }
+            const payload = {
+                top_n: this.topResponses,
+                messages: [
+                    { role: "system", content: this.systemPrompt },
+                    { role: "user", content: prompt },
+                ],
+            };
+            const response = await fetch("https://test.neuralinternet.ai/chat", {
+                method: "POST",
+                headers,
+                body: JSON.stringify(payload),
+            });
+            if (!response.ok) {
+                throw new Error("Network response was not ok");
+            }
+            const responseData = await response.json();
+            if (this.topResponses) {
+                return responseData;
+            }
+            else if (responseData.choices) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                const temp = responseData.choices;
+                return temp[0].message.content;
+            }
+        }
+        catch (error) {
+            return "Sorry I am unable to provide response now, Please try again later.";
+        }
+        return "default";
+    }
+    identifyingParams() {
+        return {
+            systemPrompt: this.systemPrompt,
+            topResponses: this.topResponses,
+        };
+    }
+}

package/dist/experimental/multimodal_embeddings/googlevertexai.cjs

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GoogleVertexAIMultimodalEmbeddings = void 0;
+const google_auth_library_1 = require("google-auth-library");
 const base_js_1 = require("../../embeddings/base.cjs");
 const googlevertexai_connection_js_1 = require("../../util/googlevertexai-connection.cjs");
 /**
@@ -24,7 +25,10 @@ class GoogleVertexAIMultimodalEmbeddings extends base_js_1.Embeddings {
             value: void 0
         });
         this.model = fields?.model ?? this.model;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
+        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,
+        }));
     }
     /**
      * Converts media (text or image) to an instance that can be used for

package/dist/experimental/multimodal_embeddings/googlevertexai.d.ts

@@ -1,11 +1,12 @@
 /// <reference types="node" resolution-mode="require"/>
+import { GoogleAuthOptions } from "google-auth-library";
 import { Embeddings, EmbeddingsParams } from "../../embeddings/base.js";
 import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction, GoogleVertexAILLMResponse } from "../../types/googlevertexai-types.js";
 /**
  * Parameters for the GoogleVertexAIMultimodalEmbeddings class, extending
  * both EmbeddingsParams and GoogleVertexAIConnectionParams.
  */
-export interface GoogleVertexAIMultimodalEmbeddingsParams extends EmbeddingsParams, GoogleVertexAIBaseLLMInput {
+export interface GoogleVertexAIMultimodalEmbeddingsParams extends EmbeddingsParams, GoogleVertexAIBaseLLMInput<GoogleAuthOptions> {
 }
 /**
  * An instance of media (text or image) that can be used for generating

package/dist/experimental/multimodal_embeddings/googlevertexai.js

@@ -1,3 +1,4 @@
+import { GoogleAuth } from "google-auth-library";
 import { Embeddings } from "../../embeddings/base.js";
 import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
 /**
@@ -21,7 +22,10 @@ export class GoogleVertexAIMultimodalEmbeddings extends Embeddings {
             value: void 0
         });
         this.model = fields?.model ?? this.model;
-        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
+        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller, new GoogleAuth({
+            scopes: "https://www.googleapis.com/auth/cloud-platform",
+            ...fields?.authOptions,
+        }));
     }
     /**
      * Converts media (text or image) to an instance that can be used for

package/dist/hub.d.ts

@@ -1,5 +1,5 @@
 import { ClientConfiguration, HubPushOptions } from "langchainhub";
-import { Runnable } from "./schema/runnable.js";
+import { Runnable } from "./schema/runnable/index.js";
 /**
  * Push a prompt to the hub.
  * If the specified repo doesn't already exist, it will be created.

package/dist/llms/base.d.ts

@@ -1,7 +1,7 @@
 import { BaseCache, BaseMessage, BasePromptValue, GenerationChunk, LLMResult } from "../schema/index.js";
 import { BaseLanguageModel, BaseLanguageModelCallOptions, BaseLanguageModelInput, BaseLanguageModelParams } from "../base_language/index.js";
 import { BaseCallbackConfig, CallbackManagerForLLMRun, Callbacks } from "../callbacks/manager.js";
-import { RunnableConfig } from "../schema/runnable.js";
+import { RunnableConfig } from "../schema/runnable/config.js";
 export type SerializedLLM = {
     _model: string;
     _type: string;

package/dist/llms/{googlevertexai.js → googlevertexai/common.cjs}

@@ -1,22 +1,26 @@
-import { BaseLLM } from "./base.js";
-import { GoogleVertexAILLMConnection } from "../util/googlevertexai-connection.js";
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseGoogleVertexAI = void 0;
+const base_js_1 = require("../base.cjs");
 /**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * 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.
+ * Base class for Google Vertex AI LLMs.
+ * Implemented subclasses must provide a GoogleVertexAILLMConnection
+ * with an appropriate auth client.
  */
-export class GoogleVertexAI extends BaseLLM {
+class BaseGoogleVertexAI extends base_js_1.BaseLLM {
+    get lc_aliases() {
+        return {
+            model: "model_name",
+        };
+    }
     constructor(fields) {
         super(fields ?? {});
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
@@ -65,10 +69,9 @@ export class GoogleVertexAI extends BaseLLM {
         this.maxOutputTokens = fields?.maxOutputTokens ?? this.maxOutputTokens;
         this.topP = fields?.topP ?? this.topP;
         this.topK = fields?.topK ?? this.topK;
-        this.connection = new GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     _llmType() {
-        return "googlevertexai";
+        return "vertexai";
     }
     async _generate(prompts, options) {
         const generations = await Promise.all(prompts.map((prompt) => this._generatePrompt(prompt, options)));
@@ -129,3 +132,4 @@ export class GoogleVertexAI extends BaseLLM {
         return result?.data?.predictions[0];
     }
 }
+exports.BaseGoogleVertexAI = BaseGoogleVertexAI;

package/dist/llms/{googlevertexai.d.ts → googlevertexai/common.d.ts}

@@ -1,11 +1,8 @@
-import { BaseLLM } from "./base.js";
-import { Generation, LLMResult } from "../schema/index.js";
-import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction, GoogleVertexAILLMResponse } from "../types/googlevertexai-types.js";
-/**
- * Interface representing the input to the Google Vertex AI model.
- */
-export interface GoogleVertexAITextInput extends GoogleVertexAIBaseLLMInput {
-}
+import { BaseLLM } from "../base.js";
+import { Generation, LLMResult } from "../../schema/index.js";
+import { GoogleVertexAILLMConnection } from "../../util/googlevertexai-connection.js";
+import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction, GoogleVertexAILLMResponse } from "../../types/googlevertexai-types.js";
+import { BaseLanguageModelCallOptions } from "../../base_language/index.js";
 /**
  * Interface representing the instance of text input to the Google Vertex
  * AI model.
@@ -32,27 +29,20 @@ interface TextPrediction extends GoogleVertexAIBasePrediction {
     content: string;
 }
 /**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * 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.
+ * Base class for Google Vertex AI LLMs.
+ * Implemented subclasses must provide a GoogleVertexAILLMConnection
+ * with an appropriate auth client.
  */
-export declare class GoogleVertexAI extends BaseLLM implements GoogleVertexAITextInput {
+export declare class BaseGoogleVertexAI<AuthOptions> extends BaseLLM implements GoogleVertexAIBaseLLMInput<AuthOptions> {
+    lc_serializable: boolean;
     model: string;
     temperature: number;
     maxOutputTokens: number;
     topP: number;
     topK: number;
-    private connection;
-    constructor(fields?: GoogleVertexAITextInput);
+    protected connection: GoogleVertexAILLMConnection<BaseLanguageModelCallOptions, GoogleVertexAILLMInstance, TextPrediction, AuthOptions>;
+    get lc_aliases(): Record<string, string>;
+    constructor(fields?: GoogleVertexAIBaseLLMInput<AuthOptions>);
     _llmType(): string;
     _generate(prompts: string[], options: this["ParsedCallOptions"]): Promise<LLMResult>;
     _generatePrompt(prompt: string, options: this["ParsedCallOptions"]): Promise<Generation[]>;

package/dist/llms/{googlevertexai.cjs → googlevertexai/common.js}

@@ -1,25 +1,23 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.GoogleVertexAI = void 0;
-const base_js_1 = require("./base.cjs");
-const googlevertexai_connection_js_1 = require("../util/googlevertexai-connection.cjs");
+import { BaseLLM } from "../base.js";
 /**
- * Enables calls to the Google Cloud's Vertex AI API to access
- * 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.
+ * Base class for Google Vertex AI LLMs.
+ * Implemented subclasses must provide a GoogleVertexAILLMConnection
+ * with an appropriate auth client.
  */
-class GoogleVertexAI extends base_js_1.BaseLLM {
+export class BaseGoogleVertexAI extends BaseLLM {
+    get lc_aliases() {
+        return {
+            model: "model_name",
+        };
+    }
     constructor(fields) {
         super(fields ?? {});
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
@@ -68,10 +66,9 @@ class GoogleVertexAI extends base_js_1.BaseLLM {
         this.maxOutputTokens = fields?.maxOutputTokens ?? this.maxOutputTokens;
         this.topP = fields?.topP ?? this.topP;
         this.topK = fields?.topK ?? this.topK;
-        this.connection = new googlevertexai_connection_js_1.GoogleVertexAILLMConnection({ ...fields, ...this }, this.caller);
     }
     _llmType() {
-        return "googlevertexai";
+        return "vertexai";
     }
     async _generate(prompts, options) {
         const generations = await Promise.all(prompts.map((prompt) => this._generatePrompt(prompt, options)));
@@ -132,4 +129,3 @@ class GoogleVertexAI extends base_js_1.BaseLLM {
         return result?.data?.predictions[0];
     }
 }
-exports.GoogleVertexAI = GoogleVertexAI;