langchain 0.0.153 → 0.0.154

package/dist/chat_models/base.cjs

@@ -94,33 +94,18 @@ class BaseChatModel extends index_js_2.BaseLanguageModel {
             })));
         }
     }
-    /**
-     * Generates chat based on the input messages.
-     * @param messages An array of arrays of BaseMessage instances.
-     * @param options The call options or an array of stop sequences.
-     * @param callbacks The callbacks for the language model.
-     * @returns A Promise that resolves to an LLMResult.
-     */
-    async generate(messages, options, callbacks) {
-        // parse call options
-        let parsedOptions;
-        if (Array.isArray(options)) {
-            parsedOptions = { stop: options };
-        }
-        else {
-            parsedOptions = options;
-        }
+    /** @ignore */
+    async _generateUncached(messages, parsedOptions, handledOptions) {
         const baseMessages = messages.map((messageList) => messageList.map(index_js_1.coerceMessageLikeToMessage));
-        const [runnableConfig, callOptions] = this._separateRunnableConfigFromCallOptions(parsedOptions);
         // create callback manager and start run
-        const callbackManager_ = await manager_js_1.CallbackManager.configure(runnableConfig.callbacks ?? callbacks, this.callbacks, runnableConfig.tags, this.tags, runnableConfig.metadata, this.metadata, { verbose: this.verbose });
+        const callbackManager_ = await manager_js_1.CallbackManager.configure(handledOptions.callbacks, this.callbacks, handledOptions.tags, this.tags, handledOptions.metadata, this.metadata, { verbose: this.verbose });
         const extra = {
-            options: callOptions,
+            options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
         const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), baseMessages, undefined, undefined, extra);
         // generate results
-        const results = await Promise.allSettled(baseMessages.map((messageList, i) => this._generate(messageList, { ...callOptions, promptIndex: i }, runManagers?.[i])));
+        const results = await Promise.allSettled(baseMessages.map((messageList, i) => this._generate(messageList, { ...parsedOptions, promptIndex: i }, runManagers?.[i])));
         // handle results
         const generations = [];
         const llmOutputs = [];
@@ -155,6 +140,54 @@ class BaseChatModel extends index_js_2.BaseLanguageModel {
         });
         return output;
     }
+    /**
+     * Generates chat based on the input messages.
+     * @param messages An array of arrays of BaseMessage instances.
+     * @param options The call options or an array of stop sequences.
+     * @param callbacks The callbacks for the language model.
+     * @returns A Promise that resolves to an LLMResult.
+     */
+    async generate(messages, options, callbacks) {
+        // parse call options
+        let parsedOptions;
+        if (Array.isArray(options)) {
+            parsedOptions = { stop: options };
+        }
+        else {
+            parsedOptions = options;
+        }
+        const baseMessages = messages.map((messageList) => messageList.map(index_js_1.coerceMessageLikeToMessage));
+        const [runnableConfig, callOptions] = this._separateRunnableConfigFromCallOptions(parsedOptions);
+        runnableConfig.callbacks = runnableConfig.callbacks ?? callbacks;
+        if (!this.cache) {
+            return this._generateUncached(baseMessages, callOptions, runnableConfig);
+        }
+        const { cache } = this;
+        const llmStringKey = this._getSerializedCacheKeyParametersForCall(callOptions);
+        const missingPromptIndices = [];
+        const generations = await Promise.all(baseMessages.map(async (baseMessage, index) => {
+            // Join all content into one string for the prompt index
+            const prompt = BaseChatModel._convertInputToPromptValue(baseMessage).toString();
+            const result = await cache.lookup(prompt, llmStringKey);
+            if (!result) {
+                missingPromptIndices.push(index);
+            }
+            return result;
+        }));
+        let llmOutput = {};
+        if (missingPromptIndices.length > 0) {
+            const results = await this._generateUncached(missingPromptIndices.map((i) => baseMessages[i]), callOptions, runnableConfig);
+            await Promise.all(results.generations.map(async (generation, index) => {
+                const promptIndex = missingPromptIndices[index];
+                generations[promptIndex] = generation;
+                // Join all content into one string for the prompt index
+                const prompt = BaseChatModel._convertInputToPromptValue(baseMessages[promptIndex]).toString();
+                return cache.update(prompt, llmStringKey, generation);
+            }));
+            llmOutput = results.llmOutput ?? {};
+        }
+        return { generations, llmOutput };
+    }
     /**
      * Get the parameters used to invoke the model
      */
@@ -165,6 +198,17 @@ class BaseChatModel extends index_js_2.BaseLanguageModel {
     _modelType() {
         return "base_chat_model";
     }
+    /**
+     * @deprecated
+     * Return a json-like object representing this LLM.
+     */
+    serialize() {
+        return {
+            ...this.invocationParams(),
+            _type: this._llmType(),
+            _model: this._modelType(),
+        };
+    }
     /**
      * Generates a prompt based on the input prompt values.
      * @param promptValues An array of BasePromptValue instances.

package/dist/chat_models/base.d.ts

@@ -49,6 +49,8 @@ export declare abstract class BaseChatModel<CallOptions extends BaseChatModelCal
     invoke(input: BaseLanguageModelInput, options?: CallOptions): Promise<BaseMessageChunk>;
     _streamResponseChunks(_messages: BaseMessage[], _options: this["ParsedCallOptions"], _runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
     _streamIterator(input: BaseLanguageModelInput, options?: CallOptions): AsyncGenerator<BaseMessageChunk>;
+    /** @ignore */
+    _generateUncached(messages: BaseMessageLike[][], parsedOptions: this["ParsedCallOptions"], handledOptions: RunnableConfig): Promise<LLMResult>;
     /**
      * Generates chat based on the input messages.
      * @param messages An array of arrays of BaseMessage instances.
@@ -63,6 +65,11 @@ export declare abstract class BaseChatModel<CallOptions extends BaseChatModelCal
     invocationParams(_options?: this["ParsedCallOptions"]): any;
     _modelType(): string;
     abstract _llmType(): string;
+    /**
+     * @deprecated
+     * Return a json-like object representing this LLM.
+     */
+    serialize(): SerializedLLM;
     /**
      * Generates a prompt based on the input prompt values.
      * @param promptValues An array of BasePromptValue instances.

package/dist/chat_models/base.js

@@ -90,33 +90,18 @@ export class BaseChatModel extends BaseLanguageModel {
             })));
         }
     }
-    /**
-     * Generates chat based on the input messages.
-     * @param messages An array of arrays of BaseMessage instances.
-     * @param options The call options or an array of stop sequences.
-     * @param callbacks The callbacks for the language model.
-     * @returns A Promise that resolves to an LLMResult.
-     */
-    async generate(messages, options, callbacks) {
-        // parse call options
-        let parsedOptions;
-        if (Array.isArray(options)) {
-            parsedOptions = { stop: options };
-        }
-        else {
-            parsedOptions = options;
-        }
+    /** @ignore */
+    async _generateUncached(messages, parsedOptions, handledOptions) {
         const baseMessages = messages.map((messageList) => messageList.map(coerceMessageLikeToMessage));
-        const [runnableConfig, callOptions] = this._separateRunnableConfigFromCallOptions(parsedOptions);
         // create callback manager and start run
-        const callbackManager_ = await CallbackManager.configure(runnableConfig.callbacks ?? callbacks, this.callbacks, runnableConfig.tags, this.tags, runnableConfig.metadata, this.metadata, { verbose: this.verbose });
+        const callbackManager_ = await CallbackManager.configure(handledOptions.callbacks, this.callbacks, handledOptions.tags, this.tags, handledOptions.metadata, this.metadata, { verbose: this.verbose });
         const extra = {
-            options: callOptions,
+            options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
         const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), baseMessages, undefined, undefined, extra);
         // generate results
-        const results = await Promise.allSettled(baseMessages.map((messageList, i) => this._generate(messageList, { ...callOptions, promptIndex: i }, runManagers?.[i])));
+        const results = await Promise.allSettled(baseMessages.map((messageList, i) => this._generate(messageList, { ...parsedOptions, promptIndex: i }, runManagers?.[i])));
         // handle results
         const generations = [];
         const llmOutputs = [];
@@ -151,6 +136,54 @@ export class BaseChatModel extends BaseLanguageModel {
         });
         return output;
     }
+    /**
+     * Generates chat based on the input messages.
+     * @param messages An array of arrays of BaseMessage instances.
+     * @param options The call options or an array of stop sequences.
+     * @param callbacks The callbacks for the language model.
+     * @returns A Promise that resolves to an LLMResult.
+     */
+    async generate(messages, options, callbacks) {
+        // parse call options
+        let parsedOptions;
+        if (Array.isArray(options)) {
+            parsedOptions = { stop: options };
+        }
+        else {
+            parsedOptions = options;
+        }
+        const baseMessages = messages.map((messageList) => messageList.map(coerceMessageLikeToMessage));
+        const [runnableConfig, callOptions] = this._separateRunnableConfigFromCallOptions(parsedOptions);
+        runnableConfig.callbacks = runnableConfig.callbacks ?? callbacks;
+        if (!this.cache) {
+            return this._generateUncached(baseMessages, callOptions, runnableConfig);
+        }
+        const { cache } = this;
+        const llmStringKey = this._getSerializedCacheKeyParametersForCall(callOptions);
+        const missingPromptIndices = [];
+        const generations = await Promise.all(baseMessages.map(async (baseMessage, index) => {
+            // Join all content into one string for the prompt index
+            const prompt = BaseChatModel._convertInputToPromptValue(baseMessage).toString();
+            const result = await cache.lookup(prompt, llmStringKey);
+            if (!result) {
+                missingPromptIndices.push(index);
+            }
+            return result;
+        }));
+        let llmOutput = {};
+        if (missingPromptIndices.length > 0) {
+            const results = await this._generateUncached(missingPromptIndices.map((i) => baseMessages[i]), callOptions, runnableConfig);
+            await Promise.all(results.generations.map(async (generation, index) => {
+                const promptIndex = missingPromptIndices[index];
+                generations[promptIndex] = generation;
+                // Join all content into one string for the prompt index
+                const prompt = BaseChatModel._convertInputToPromptValue(baseMessages[promptIndex]).toString();
+                return cache.update(prompt, llmStringKey, generation);
+            }));
+            llmOutput = results.llmOutput ?? {};
+        }
+        return { generations, llmOutput };
+    }
     /**
      * Get the parameters used to invoke the model
      */
@@ -161,6 +194,17 @@ export class BaseChatModel extends BaseLanguageModel {
     _modelType() {
         return "base_chat_model";
     }
+    /**
+     * @deprecated
+     * Return a json-like object representing this LLM.
+     */
+    serialize() {
+        return {
+            ...this.invocationParams(),
+            _type: this._llmType(),
+            _model: this._modelType(),
+        };
+    }
     /**
      * Generates a prompt based on the input prompt values.
      * @param promptValues An array of BasePromptValue instances.

package/dist/llms/base.cjs

@@ -1,16 +1,15 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LLM = exports.BaseLLM = void 0;
-const index_js_1 = require("../cache/index.cjs");
-const index_js_2 = require("../schema/index.cjs");
-const index_js_3 = require("../base_language/index.cjs");
+const index_js_1 = require("../schema/index.cjs");
+const index_js_2 = require("../base_language/index.cjs");
 const manager_js_1 = require("../callbacks/manager.cjs");
 const base_js_1 = require("../memory/base.cjs");
 /**
  * LLM Wrapper. Provides an {@link call} (an {@link generate}) function that takes in a prompt (or prompts) and returns a string.
  */
-class BaseLLM extends index_js_3.BaseLanguageModel {
-    constructor({ cache, concurrency, ...rest }) {
+class BaseLLM extends index_js_2.BaseLanguageModel {
+    constructor({ concurrency, ...rest }) {
         super(concurrency ? { maxConcurrency: concurrency, ...rest } : rest);
         Object.defineProperty(this, "lc_namespace", {
             enumerable: true,
@@ -18,21 +17,6 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
             writable: true,
             value: ["langchain", "llms", this._llmType()]
         });
-        Object.defineProperty(this, "cache", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        if (typeof cache === "object") {
-            this.cache = cache;
-        }
-        else if (cache) {
-            this.cache = index_js_1.InMemoryCache.global();
-        }
-        else {
-            this.cache = undefined;
-        }
     }
     /**
      * This method takes an input and options, and returns a string. It
@@ -72,7 +56,7 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
                 invocation_params: this?.invocationParams(callOptions),
             };
             const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra);
-            let generation = new index_js_2.GenerationChunk({
+            let generation = new index_js_1.GenerationChunk({
                 text: "",
             });
             try {
@@ -160,7 +144,7 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
         // This defines RUN_KEY as a non-enumerable property on the output object
         // so that it is not serialized when the output is stringified, and so that
         // it isnt included when listing the keys of the output object.
-        Object.defineProperty(output, index_js_2.RUN_KEY, {
+        Object.defineProperty(output, index_js_1.RUN_KEY, {
             value: runIds ? { runIds } : undefined,
             configurable: true,
         });
@@ -186,9 +170,7 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
             return this._generateUncached(prompts, callOptions, runnableConfig);
         }
         const { cache } = this;
-        const params = this.serialize();
-        params.stop = callOptions.stop ?? params.stop;
-        const llmStringKey = `${Object.entries(params).sort()}`;
+        const llmStringKey = this._getSerializedCacheKeyParametersForCall(callOptions);
         const missingPromptIndices = [];
         const generations = await Promise.all(prompts.map(async (prompt, index) => {
             const result = await cache.lookup(prompt, llmStringKey);
@@ -238,7 +220,7 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
     async predictMessages(messages, options, callbacks) {
         const text = (0, base_js_1.getBufferString)(messages);
         const prediction = await this.call(text, options, callbacks);
-        return new index_js_2.AIMessage(prediction);
+        return new index_js_1.AIMessage(prediction);
     }
     /**
      * Get the identifying parameters of the LLM.
@@ -248,6 +230,7 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
         return {};
     }
     /**
+     * @deprecated
      * Return a json-like object representing this LLM.
      */
     serialize() {
@@ -261,6 +244,7 @@ class BaseLLM extends index_js_3.BaseLanguageModel {
         return "base_llm";
     }
     /**
+     * @deprecated
      * Load an LLM from a json-like object describing it.
      */
     static async deserialize(data) {

package/dist/llms/base.d.ts

@@ -1,4 +1,4 @@
-import { BaseCache, BaseMessage, BasePromptValue, GenerationChunk, LLMResult } from "../schema/index.js";
+import { 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/config.js";
@@ -11,7 +11,6 @@ export interface BaseLLMParams extends BaseLanguageModelParams {
      * @deprecated Use `maxConcurrency` instead
      */
     concurrency?: number;
-    cache?: BaseCache | boolean;
 }
 export interface BaseLLMCallOptions extends BaseLanguageModelCallOptions {
 }
@@ -21,8 +20,7 @@ export interface BaseLLMCallOptions extends BaseLanguageModelCallOptions {
 export declare abstract class BaseLLM<CallOptions extends BaseLLMCallOptions = BaseLLMCallOptions> extends BaseLanguageModel<string, CallOptions> {
     ParsedCallOptions: Omit<CallOptions, keyof RunnableConfig & "timeout">;
     lc_namespace: string[];
-    cache?: BaseCache;
-    constructor({ cache, concurrency, ...rest }: BaseLLMParams);
+    constructor({ concurrency, ...rest }: BaseLLMParams);
     /**
      * This method takes an input and options, and returns a string. It
      * converts the input to a prompt value and generates a result based on
@@ -90,11 +88,13 @@ export declare abstract class BaseLLM<CallOptions extends BaseLLMCallOptions = B
      */
     abstract _llmType(): string;
     /**
+     * @deprecated
      * Return a json-like object representing this LLM.
      */
     serialize(): SerializedLLM;
     _modelType(): string;
     /**
+     * @deprecated
      * Load an LLM from a json-like object describing it.
      */
     static deserialize(data: SerializedLLM): Promise<BaseLLM>;

package/dist/llms/base.js

@@ -1,4 +1,3 @@
-import { InMemoryCache } from "../cache/index.js";
 import { AIMessage, GenerationChunk, RUN_KEY, } from "../schema/index.js";
 import { BaseLanguageModel, } from "../base_language/index.js";
 import { CallbackManager, } from "../callbacks/manager.js";
@@ -7,7 +6,7 @@ import { getBufferString } from "../memory/base.js";
  * LLM Wrapper. Provides an {@link call} (an {@link generate}) function that takes in a prompt (or prompts) and returns a string.
  */
 export class BaseLLM extends BaseLanguageModel {
-    constructor({ cache, concurrency, ...rest }) {
+    constructor({ concurrency, ...rest }) {
         super(concurrency ? { maxConcurrency: concurrency, ...rest } : rest);
         Object.defineProperty(this, "lc_namespace", {
             enumerable: true,
@@ -15,21 +14,6 @@ export class BaseLLM extends BaseLanguageModel {
             writable: true,
             value: ["langchain", "llms", this._llmType()]
         });
-        Object.defineProperty(this, "cache", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        if (typeof cache === "object") {
-            this.cache = cache;
-        }
-        else if (cache) {
-            this.cache = InMemoryCache.global();
-        }
-        else {
-            this.cache = undefined;
-        }
     }
     /**
      * This method takes an input and options, and returns a string. It
@@ -183,9 +167,7 @@ export class BaseLLM extends BaseLanguageModel {
             return this._generateUncached(prompts, callOptions, runnableConfig);
         }
         const { cache } = this;
-        const params = this.serialize();
-        params.stop = callOptions.stop ?? params.stop;
-        const llmStringKey = `${Object.entries(params).sort()}`;
+        const llmStringKey = this._getSerializedCacheKeyParametersForCall(callOptions);
         const missingPromptIndices = [];
         const generations = await Promise.all(prompts.map(async (prompt, index) => {
             const result = await cache.lookup(prompt, llmStringKey);
@@ -245,6 +227,7 @@ export class BaseLLM extends BaseLanguageModel {
         return {};
     }
     /**
+     * @deprecated
      * Return a json-like object representing this LLM.
      */
     serialize() {
@@ -258,6 +241,7 @@ export class BaseLLM extends BaseLanguageModel {
         return "base_llm";
     }
     /**
+     * @deprecated
      * Load an LLM from a json-like object describing it.
      */
     static async deserialize(data) {

package/dist/schema/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Docstore = exports.BaseEntityStore = exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BasePromptValue = exports.ChatGenerationChunk = exports.ChatMessageChunk = exports.coerceMessageLikeToMessage = exports.isBaseMessage = exports.ChatMessage = exports.FunctionMessageChunk = exports.FunctionMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.SystemMessageChunk = exports.SystemMessage = exports.AIMessageChunk = exports.AIMessage = exports.HumanMessageChunk = exports.HumanMessage = exports.BaseMessageChunk = exports.BaseMessage = exports.GenerationChunk = exports.RUN_KEY = void 0;
+exports.Docstore = exports.BaseEntityStore = exports.BaseFileStore = exports.BaseCache = exports.BaseListChatMessageHistory = exports.BaseChatMessageHistory = exports.BasePromptValue = exports.mapStoredMessageToChatMessage = exports.ChatGenerationChunk = exports.ChatMessageChunk = exports.coerceMessageLikeToMessage = exports.isBaseMessage = exports.ChatMessage = exports.FunctionMessageChunk = exports.FunctionMessage = exports.SystemChatMessage = exports.AIChatMessage = exports.HumanChatMessage = exports.BaseChatMessage = exports.SystemMessageChunk = exports.SystemMessage = exports.AIMessageChunk = exports.AIMessage = exports.HumanMessageChunk = exports.HumanMessage = exports.BaseMessageChunk = exports.BaseMessage = exports.GenerationChunk = exports.RUN_KEY = void 0;
 const serializable_js_1 = require("../load/serializable.cjs");
 exports.RUN_KEY = "__run";
 /**
@@ -405,6 +405,55 @@ class ChatGenerationChunk extends GenerationChunk {
     }
 }
 exports.ChatGenerationChunk = ChatGenerationChunk;
+/**
+ * Maps messages from an older format (V1) to the current `StoredMessage`
+ * format. If the message is already in the `StoredMessage` format, it is
+ * returned as is. Otherwise, it transforms the V1 message into a
+ * `StoredMessage`. This function is important for maintaining
+ * compatibility with older message formats.
+ */
+function mapV1MessageToStoredMessage(message) {
+    // TODO: Remove this mapper when we deprecate the old message format.
+    if (message.data !== undefined) {
+        return message;
+    }
+    else {
+        const v1Message = message;
+        return {
+            type: v1Message.type,
+            data: {
+                content: v1Message.text,
+                role: v1Message.role,
+                name: undefined,
+            },
+        };
+    }
+}
+function mapStoredMessageToChatMessage(message) {
+    const storedMessage = mapV1MessageToStoredMessage(message);
+    switch (storedMessage.type) {
+        case "human":
+            return new HumanMessage(storedMessage.data);
+        case "ai":
+            return new AIMessage(storedMessage.data);
+        case "system":
+            return new SystemMessage(storedMessage.data);
+        case "function":
+            if (storedMessage.data.name === undefined) {
+                throw new Error("Name must be defined for function messages");
+            }
+            return new FunctionMessage(storedMessage.data);
+        case "chat": {
+            if (storedMessage.data.role === undefined) {
+                throw new Error("Role must be defined for chat messages");
+            }
+            return new ChatMessage(storedMessage.data);
+        }
+        default:
+            throw new Error(`Got unexpected type: ${storedMessage.type}`);
+    }
+}
+exports.mapStoredMessageToChatMessage = mapStoredMessageToChatMessage;
 /**
  * Base PromptValue class. All prompt values should extend this class.
  */

package/dist/schema/index.d.ts

@@ -59,6 +59,10 @@ export interface StoredMessage {
     type: string;
     data: StoredMessageData;
 }
+export interface StoredGeneration {
+    text: string;
+    message?: StoredMessage;
+}
 export type MessageType = "human" | "ai" | "generic" | "system" | "function";
 export interface BaseMessageFields {
     content: string;
@@ -239,6 +243,7 @@ export declare class ChatGenerationChunk extends GenerationChunk implements Chat
     constructor(fields: ChatGenerationChunkFields);
     concat(chunk: ChatGenerationChunk): ChatGenerationChunk;
 }
+export declare function mapStoredMessageToChatMessage(message: StoredMessage): HumanMessage | AIMessage | SystemMessage | FunctionMessage | ChatMessage;
 export interface ChatResult {
     generations: ChatGeneration[];
     llmOutput?: Record<string, any>;

package/dist/schema/index.js

@@ -386,6 +386,54 @@ export class ChatGenerationChunk extends GenerationChunk {
         });
     }
 }
+/**
+ * Maps messages from an older format (V1) to the current `StoredMessage`
+ * format. If the message is already in the `StoredMessage` format, it is
+ * returned as is. Otherwise, it transforms the V1 message into a
+ * `StoredMessage`. This function is important for maintaining
+ * compatibility with older message formats.
+ */
+function mapV1MessageToStoredMessage(message) {
+    // TODO: Remove this mapper when we deprecate the old message format.
+    if (message.data !== undefined) {
+        return message;
+    }
+    else {
+        const v1Message = message;
+        return {
+            type: v1Message.type,
+            data: {
+                content: v1Message.text,
+                role: v1Message.role,
+                name: undefined,
+            },
+        };
+    }
+}
+export function mapStoredMessageToChatMessage(message) {
+    const storedMessage = mapV1MessageToStoredMessage(message);
+    switch (storedMessage.type) {
+        case "human":
+            return new HumanMessage(storedMessage.data);
+        case "ai":
+            return new AIMessage(storedMessage.data);
+        case "system":
+            return new SystemMessage(storedMessage.data);
+        case "function":
+            if (storedMessage.data.name === undefined) {
+                throw new Error("Name must be defined for function messages");
+            }
+            return new FunctionMessage(storedMessage.data);
+        case "chat": {
+            if (storedMessage.data.role === undefined) {
+                throw new Error("Role must be defined for chat messages");
+            }
+            return new ChatMessage(storedMessage.data);
+        }
+        default:
+            throw new Error(`Got unexpected type: ${storedMessage.type}`);
+    }
+}
 /**
  * Base PromptValue class. All prompt values should extend this class.
  */

package/dist/stores/message/utils.cjs

@@ -1,32 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.mapChatMessagesToStoredMessages = exports.mapStoredMessagesToChatMessages = exports.mapV1MessageToStoredMessage = void 0;
+exports.mapChatMessagesToStoredMessages = exports.mapStoredMessagesToChatMessages = void 0;
 const index_js_1 = require("../../schema/index.cjs");
-/**
- * Maps messages from an older format (V1) to the current `StoredMessage`
- * format. If the message is already in the `StoredMessage` format, it is
- * returned as is. Otherwise, it transforms the V1 message into a
- * `StoredMessage`. This function is important for maintaining
- * compatibility with older message formats.
- */
-function mapV1MessageToStoredMessage(message) {
-    // TODO: Remove this mapper when we deprecate the old message format.
-    if (message.data !== undefined) {
-        return message;
-    }
-    else {
-        const v1Message = message;
-        return {
-            type: v1Message.type,
-            data: {
-                content: v1Message.text,
-                role: v1Message.role,
-                name: undefined,
-            },
-        };
-    }
-}
-exports.mapV1MessageToStoredMessage = mapV1MessageToStoredMessage;
 /**
  * Transforms an array of `StoredMessage` instances into an array of
  * `BaseMessage` instances. It uses the `mapV1MessageToStoredMessage`
@@ -36,30 +11,7 @@ exports.mapV1MessageToStoredMessage = mapV1MessageToStoredMessage;
  * messages for use in a chat context.
  */
 function mapStoredMessagesToChatMessages(messages) {
-    return messages.map((message) => {
-        const storedMessage = mapV1MessageToStoredMessage(message);
-        switch (storedMessage.type) {
-            case "human":
-                return new index_js_1.HumanMessage(storedMessage.data);
-            case "ai":
-                return new index_js_1.AIMessage(storedMessage.data);
-            case "system":
-                return new index_js_1.SystemMessage(storedMessage.data);
-            case "function":
-                if (storedMessage.data.name === undefined) {
-                    throw new Error("Name must be defined for function messages");
-                }
-                return new index_js_1.FunctionMessage(storedMessage.data);
-            case "chat": {
-                if (storedMessage.data.role === undefined) {
-                    throw new Error("Role must be defined for chat messages");
-                }
-                return new index_js_1.ChatMessage(storedMessage.data);
-            }
-            default:
-                throw new Error(`Got unexpected type: ${storedMessage.type}`);
-        }
-    });
+    return messages.map(index_js_1.mapStoredMessageToChatMessage);
 }
 exports.mapStoredMessagesToChatMessages = mapStoredMessagesToChatMessages;
 /**

package/dist/stores/message/utils.d.ts

@@ -1,17 +1,4 @@
 import { BaseMessage, StoredMessage } from "../../schema/index.js";
-interface StoredMessageV1 {
-    type: string;
-    role: string | undefined;
-    text: string;
-}
-/**
- * Maps messages from an older format (V1) to the current `StoredMessage`
- * format. If the message is already in the `StoredMessage` format, it is
- * returned as is. Otherwise, it transforms the V1 message into a
- * `StoredMessage`. This function is important for maintaining
- * compatibility with older message formats.
- */
-export declare function mapV1MessageToStoredMessage(message: StoredMessage | StoredMessageV1): StoredMessage;
 /**
  * Transforms an array of `StoredMessage` instances into an array of
  * `BaseMessage` instances. It uses the `mapV1MessageToStoredMessage`
@@ -28,4 +15,3 @@ export declare function mapStoredMessagesToChatMessages(messages: StoredMessage[
  * is used to prepare chat messages for storage.
  */
 export declare function mapChatMessagesToStoredMessages(messages: BaseMessage[]): StoredMessage[];
-export {};