langchain 0.0.152 → 0.0.154

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/llms/fireworks.cjs

@@ -0,0 +1,92 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Fireworks = void 0;
+const env_js_1 = require("../util/env.cjs");
+const openai_js_1 = require("./openai.cjs");
+/**
+ * Wrapper around Fireworks API for large language models
+ *
+ * Fireworks API is compatible to the OpenAI API with some limitations described in
+ * https://readme.fireworks.ai/docs/openai-compatibility.
+ *
+ * To use, you should have the `openai` package installed and
+ * the `FIREWORKS_API_KEY` environment variable set.
+ */
+class Fireworks extends openai_js_1.OpenAI {
+    static lc_name() {
+        return "Fireworks";
+    }
+    _llmType() {
+        return "fireworks";
+    }
+    get lc_secrets() {
+        return {
+            fireworksApiKey: "FIREWORKS_API_KEY",
+        };
+    }
+    constructor(fields) {
+        const fireworksApiKey = fields?.fireworksApiKey || (0, env_js_1.getEnvironmentVariable)("FIREWORKS_API_KEY");
+        if (!fireworksApiKey) {
+            throw new Error(`Fireworks API key not found. Please set the FIREWORKS_API_KEY environment variable or provide the key into "fireworksApiKey"`);
+        }
+        super({
+            ...fields,
+            openAIApiKey: fireworksApiKey,
+            modelName: fields?.modelName || "accounts/fireworks/models/llama-v2-13b",
+            configuration: {
+                baseURL: "https://api.fireworks.ai/inference/v1",
+            },
+        });
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        Object.defineProperty(this, "fireworksApiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.fireworksApiKey = fireworksApiKey;
+    }
+    toJSON() {
+        const result = super.toJSON();
+        if ("kwargs" in result &&
+            typeof result.kwargs === "object" &&
+            result.kwargs != null) {
+            delete result.kwargs.openai_api_key;
+            delete result.kwargs.configuration;
+        }
+        return result;
+    }
+    /**
+     * Calls the Fireworks API with retry logic in case of failures.
+     * @param request The request to send to the Fireworks API.
+     * @param options Optional configuration for the API call.
+     * @returns The response from the Fireworks API.
+     */
+    async completionWithRetry(request, options) {
+        // https://readme.fireworks.ai/docs/openai-compatibility#api-compatibility
+        if (Array.isArray(request.prompt)) {
+            if (request.prompt.length > 1) {
+                throw new Error("Multiple prompts are not supported by Fireworks");
+            }
+            const prompt = request.prompt[0];
+            if (typeof prompt !== "string") {
+                throw new Error("Only string prompts are supported by Fireworks");
+            }
+            request.prompt = prompt;
+        }
+        delete request.frequency_penalty;
+        delete request.presence_penalty;
+        delete request.best_of;
+        delete request.logit_bias;
+        if (request.stream === true) {
+            return super.completionWithRetry(request, options);
+        }
+        return super.completionWithRetry(request, options);
+    }
+}
+exports.Fireworks = Fireworks;

package/dist/llms/fireworks.d.ts

@@ -0,0 +1,33 @@
+import type { OpenAI as OpenAIClient } from "openai";
+import type { BaseLLMParams } from "./base.js";
+import type { OpenAICallOptions, OpenAIInput } from "./openai.js";
+import type { OpenAICoreRequestOptions } from "../types/openai-types.js";
+import { OpenAI } from "./openai.js";
+type FireworksUnsupportedArgs = "frequencyPenalty" | "presencePenalty" | "bestOf" | "logitBias";
+type FireworksUnsupportedCallOptions = "functions" | "function_call" | "tools";
+export type FireworksCallOptions = Partial<Omit<OpenAICallOptions, FireworksUnsupportedCallOptions>>;
+/**
+ * Wrapper around Fireworks API for large language models
+ *
+ * Fireworks API is compatible to the OpenAI API with some limitations described in
+ * https://readme.fireworks.ai/docs/openai-compatibility.
+ *
+ * To use, you should have the `openai` package installed and
+ * the `FIREWORKS_API_KEY` environment variable set.
+ */
+export declare class Fireworks extends OpenAI<FireworksCallOptions> {
+    static lc_name(): string;
+    _llmType(): string;
+    get lc_secrets(): {
+        [key: string]: string;
+    } | undefined;
+    lc_serializable: boolean;
+    fireworksApiKey?: string;
+    constructor(fields?: Partial<Omit<OpenAIInput, "openAIApiKey" | FireworksUnsupportedArgs>> & BaseLLMParams & {
+        fireworksApiKey?: string;
+    });
+    toJSON(): import("../load/serializable.js").Serialized;
+    completionWithRetry(request: OpenAIClient.CompletionCreateParamsStreaming, options?: OpenAICoreRequestOptions): Promise<AsyncIterable<OpenAIClient.Completion>>;
+    completionWithRetry(request: OpenAIClient.CompletionCreateParamsNonStreaming, options?: OpenAICoreRequestOptions): Promise<OpenAIClient.Completions.Completion>;
+}
+export {};

package/dist/llms/fireworks.js

@@ -0,0 +1,88 @@
+import { getEnvironmentVariable } from "../util/env.js";
+import { OpenAI } from "./openai.js";
+/**
+ * Wrapper around Fireworks API for large language models
+ *
+ * Fireworks API is compatible to the OpenAI API with some limitations described in
+ * https://readme.fireworks.ai/docs/openai-compatibility.
+ *
+ * To use, you should have the `openai` package installed and
+ * the `FIREWORKS_API_KEY` environment variable set.
+ */
+export class Fireworks extends OpenAI {
+    static lc_name() {
+        return "Fireworks";
+    }
+    _llmType() {
+        return "fireworks";
+    }
+    get lc_secrets() {
+        return {
+            fireworksApiKey: "FIREWORKS_API_KEY",
+        };
+    }
+    constructor(fields) {
+        const fireworksApiKey = fields?.fireworksApiKey || getEnvironmentVariable("FIREWORKS_API_KEY");
+        if (!fireworksApiKey) {
+            throw new Error(`Fireworks API key not found. Please set the FIREWORKS_API_KEY environment variable or provide the key into "fireworksApiKey"`);
+        }
+        super({
+            ...fields,
+            openAIApiKey: fireworksApiKey,
+            modelName: fields?.modelName || "accounts/fireworks/models/llama-v2-13b",
+            configuration: {
+                baseURL: "https://api.fireworks.ai/inference/v1",
+            },
+        });
+        Object.defineProperty(this, "lc_serializable", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        Object.defineProperty(this, "fireworksApiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.fireworksApiKey = fireworksApiKey;
+    }
+    toJSON() {
+        const result = super.toJSON();
+        if ("kwargs" in result &&
+            typeof result.kwargs === "object" &&
+            result.kwargs != null) {
+            delete result.kwargs.openai_api_key;
+            delete result.kwargs.configuration;
+        }
+        return result;
+    }
+    /**
+     * Calls the Fireworks API with retry logic in case of failures.
+     * @param request The request to send to the Fireworks API.
+     * @param options Optional configuration for the API call.
+     * @returns The response from the Fireworks API.
+     */
+    async completionWithRetry(request, options) {
+        // https://readme.fireworks.ai/docs/openai-compatibility#api-compatibility
+        if (Array.isArray(request.prompt)) {
+            if (request.prompt.length > 1) {
+                throw new Error("Multiple prompts are not supported by Fireworks");
+            }
+            const prompt = request.prompt[0];
+            if (typeof prompt !== "string") {
+                throw new Error("Only string prompts are supported by Fireworks");
+            }
+            request.prompt = prompt;
+        }
+        delete request.frequency_penalty;
+        delete request.presence_penalty;
+        delete request.best_of;
+        delete request.logit_bias;
+        if (request.stream === true) {
+            return super.completionWithRetry(request, options);
+        }
+        return super.completionWithRetry(request, options);
+    }
+}

package/dist/llms/ollama.cjs

@@ -290,14 +290,30 @@ class Ollama extends base_js_1.LLM {
     async *_streamResponseChunks(prompt, options, runManager) {
         const stream = await this.caller.call(async () => (0, ollama_js_1.createOllamaStream)(this.baseUrl, { ...this.invocationParams(options), prompt }, options));
         for await (const chunk of stream) {
-            yield new index_js_1.GenerationChunk({
-                text: chunk.response,
-                generationInfo: {
-                    ...chunk,
-                    response: undefined,
-                },
-            });
-            await runManager?.handleLLMNewToken(chunk.response ?? "");
+            if (!chunk.done) {
+                yield new index_js_1.GenerationChunk({
+                    text: chunk.response,
+                    generationInfo: {
+                        ...chunk,
+                        response: undefined,
+                    },
+                });
+                await runManager?.handleLLMNewToken(chunk.response ?? "");
+            }
+            else {
+                yield new index_js_1.GenerationChunk({
+                    text: "",
+                    generationInfo: {
+                        model: chunk.model,
+                        total_duration: chunk.total_duration,
+                        load_duration: chunk.load_duration,
+                        prompt_eval_count: chunk.prompt_eval_count,
+                        prompt_eval_duration: chunk.prompt_eval_duration,
+                        eval_count: chunk.eval_count,
+                        eval_duration: chunk.eval_duration,
+                    },
+                });
+            }
         }
     }
     /** @ignore */

package/dist/llms/ollama.d.ts

@@ -6,8 +6,7 @@ import { GenerationChunk } from "../schema/index.js";
  * Class that represents the Ollama language model. It extends the base
  * LLM class and implements the OllamaInput interface.
  */
-export declare class Ollama extends LLM implements OllamaInput {
-    CallOptions: OllamaCallOptions;
+export declare class Ollama extends LLM<OllamaCallOptions> implements OllamaInput {
     static lc_name(): string;
     lc_serializable: boolean;
     model: string;

package/dist/llms/ollama.js

@@ -287,14 +287,30 @@ export class Ollama extends LLM {
     async *_streamResponseChunks(prompt, options, runManager) {
         const stream = await this.caller.call(async () => createOllamaStream(this.baseUrl, { ...this.invocationParams(options), prompt }, options));
         for await (const chunk of stream) {
-            yield new GenerationChunk({
-                text: chunk.response,
-                generationInfo: {
-                    ...chunk,
-                    response: undefined,
-                },
-            });
-            await runManager?.handleLLMNewToken(chunk.response ?? "");
+            if (!chunk.done) {
+                yield new GenerationChunk({
+                    text: chunk.response,
+                    generationInfo: {
+                        ...chunk,
+                        response: undefined,
+                    },
+                });
+                await runManager?.handleLLMNewToken(chunk.response ?? "");
+            }
+            else {
+                yield new GenerationChunk({
+                    text: "",
+                    generationInfo: {
+                        model: chunk.model,
+                        total_duration: chunk.total_duration,
+                        load_duration: chunk.load_duration,
+                        prompt_eval_count: chunk.prompt_eval_count,
+                        prompt_eval_duration: chunk.prompt_eval_duration,
+                        eval_count: chunk.eval_count,
+                        eval_duration: chunk.eval_duration,
+                    },
+                });
+            }
         }
     }
     /** @ignore */

package/dist/llms/openai-chat.cjs

@@ -35,11 +35,7 @@ class OpenAIChat extends base_js_1.LLM {
         return "OpenAIChat";
     }
     get callKeys() {
-        return [
-            ...super.callKeys,
-            "options",
-            "promptIndex",
-        ];
+        return [...super.callKeys, "options", "promptIndex"];
     }
     get lc_secrets() {
         return {

package/dist/llms/openai-chat.d.ts

@@ -36,7 +36,7 @@ export interface OpenAIChatCallOptions extends OpenAICallOptions {
  */
 export declare class OpenAIChat extends LLM<OpenAIChatCallOptions> implements OpenAIChatInput, AzureOpenAIInput {
     static lc_name(): string;
-    get callKeys(): (keyof OpenAIChatCallOptions)[];
+    get callKeys(): string[];
     lc_serializable: boolean;
     get lc_secrets(): {
         [key: string]: string;

package/dist/llms/openai-chat.js

@@ -32,11 +32,7 @@ export class OpenAIChat extends LLM {
         return "OpenAIChat";
     }
     get callKeys() {
-        return [
-            ...super.callKeys,
-            "options",
-            "promptIndex",
-        ];
+        return [...super.callKeys, "options", "promptIndex"];
     }
     get lc_secrets() {
         return {

package/dist/llms/openai.cjs

@@ -59,7 +59,7 @@ class OpenAI extends base_js_1.BaseLLM {
         if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
             fields?.modelName?.startsWith("gpt-4")) &&
             !fields?.modelName?.includes("-instruct")) {
-            // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
+            // eslint-disable-next-line no-constructor-return
             return new openai_chat_js_1.OpenAIChat(fields, configuration);
         }
         super(fields ?? {});

package/dist/llms/openai.d.ts

@@ -22,9 +22,9 @@ export { AzureOpenAIInput, OpenAICallOptions, OpenAIInput };
  * `openai.createCompletion`} can be passed through {@link modelKwargs}, even
  * if not explicitly available on this class.
  */
-export declare class OpenAI extends BaseLLM<OpenAICallOptions> implements OpenAIInput, AzureOpenAIInput {
+export declare class OpenAI<CallOptions extends OpenAICallOptions = OpenAICallOptions> extends BaseLLM<CallOptions> implements OpenAIInput, AzureOpenAIInput {
     static lc_name(): string;
-    get callKeys(): (keyof OpenAICallOptions)[];
+    get callKeys(): string[];
     lc_serializable: boolean;
     get lc_secrets(): {
         [key: string]: string;

package/dist/llms/openai.js

@@ -56,7 +56,7 @@ export class OpenAI extends BaseLLM {
         if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
             fields?.modelName?.startsWith("gpt-4")) &&
             !fields?.modelName?.includes("-instruct")) {
-            // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
+            // eslint-disable-next-line no-constructor-return
             return new OpenAIChat(fields, configuration);
         }
         super(fields ?? {});

package/dist/load/import_map.cjs

@@ -24,8 +24,8 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = void 0;
+exports.retrievers__databerry = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__fireworks = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__fireworks = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
+exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = void 0;
 exports.load__serializable = __importStar(require("../load/serializable.cjs"));
 exports.agents = __importStar(require("../agents/index.cjs"));
 exports.agents__toolkits = __importStar(require("../agents/toolkits/index.cjs"));
@@ -44,6 +44,7 @@ exports.llms__openai = __importStar(require("../llms/openai.cjs"));
 exports.llms__ai21 = __importStar(require("../llms/ai21.cjs"));
 exports.llms__aleph_alpha = __importStar(require("../llms/aleph_alpha.cjs"));
 exports.llms__ollama = __importStar(require("../llms/ollama.cjs"));
+exports.llms__fireworks = __importStar(require("../llms/fireworks.cjs"));
 exports.prompts = __importStar(require("../prompts/index.cjs"));
 exports.vectorstores__base = __importStar(require("../vectorstores/base.cjs"));
 exports.vectorstores__memory = __importStar(require("../vectorstores/memory.cjs"));
@@ -60,6 +61,7 @@ exports.document_transformers__openai_functions = __importStar(require("../docum
 exports.chat_models__base = __importStar(require("../chat_models/base.cjs"));
 exports.chat_models__openai = __importStar(require("../chat_models/openai.cjs"));
 exports.chat_models__anthropic = __importStar(require("../chat_models/anthropic.cjs"));
+exports.chat_models__fireworks = __importStar(require("../chat_models/fireworks.cjs"));
 exports.chat_models__baiduwenxin = __importStar(require("../chat_models/baiduwenxin.cjs"));
 exports.chat_models__ollama = __importStar(require("../chat_models/ollama.cjs"));
 exports.chat_models__minimax = __importStar(require("../chat_models/minimax.cjs"));

package/dist/load/import_map.d.ts

@@ -16,6 +16,7 @@ export * as llms__openai from "../llms/openai.js";
 export * as llms__ai21 from "../llms/ai21.js";
 export * as llms__aleph_alpha from "../llms/aleph_alpha.js";
 export * as llms__ollama from "../llms/ollama.js";
+export * as llms__fireworks from "../llms/fireworks.js";
 export * as prompts from "../prompts/index.js";
 export * as vectorstores__base from "../vectorstores/base.js";
 export * as vectorstores__memory from "../vectorstores/memory.js";
@@ -32,6 +33,7 @@ export * as document_transformers__openai_functions from "../document_transforme
 export * as chat_models__base from "../chat_models/base.js";
 export * as chat_models__openai from "../chat_models/openai.js";
 export * as chat_models__anthropic from "../chat_models/anthropic.js";
+export * as chat_models__fireworks from "../chat_models/fireworks.js";
 export * as chat_models__baiduwenxin from "../chat_models/baiduwenxin.js";
 export * as chat_models__ollama from "../chat_models/ollama.js";
 export * as chat_models__minimax from "../chat_models/minimax.js";

package/dist/load/import_map.js

@@ -17,6 +17,7 @@ export * as llms__openai from "../llms/openai.js";
 export * as llms__ai21 from "../llms/ai21.js";
 export * as llms__aleph_alpha from "../llms/aleph_alpha.js";
 export * as llms__ollama from "../llms/ollama.js";
+export * as llms__fireworks from "../llms/fireworks.js";
 export * as prompts from "../prompts/index.js";
 export * as vectorstores__base from "../vectorstores/base.js";
 export * as vectorstores__memory from "../vectorstores/memory.js";
@@ -33,6 +34,7 @@ export * as document_transformers__openai_functions from "../document_transforme
 export * as chat_models__base from "../chat_models/base.js";
 export * as chat_models__openai from "../chat_models/openai.js";
 export * as chat_models__anthropic from "../chat_models/anthropic.js";
+export * as chat_models__fireworks from "../chat_models/fireworks.js";
 export * as chat_models__baiduwenxin from "../chat_models/baiduwenxin.js";
 export * as chat_models__ollama from "../chat_models/ollama.js";
 export * as chat_models__minimax from "../chat_models/minimax.js";

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>;