langchain 0.0.151 → 0.0.153

package/cache/cloudflare_kv.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/cache/cloudflare_kv.cjs');

package/cache/cloudflare_kv.d.ts

@@ -0,0 +1 @@
+export * from '../dist/cache/cloudflare_kv.js'

package/cache/cloudflare_kv.js

@@ -0,0 +1 @@
+export * from '../dist/cache/cloudflare_kv.js'

package/chat_models/fireworks.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/chat_models/fireworks.cjs');

package/chat_models/fireworks.d.ts

@@ -0,0 +1 @@
+export * from '../dist/chat_models/fireworks.js'

package/chat_models/fireworks.js

@@ -0,0 +1 @@
+export * from '../dist/chat_models/fireworks.js'

package/dist/agents/chat/index.cjs

@@ -91,7 +91,7 @@ class ChatAgent extends agent_js_1.Agent {
             chat_js_1.SystemMessagePromptTemplate.fromTemplate(template),
             chat_js_1.HumanMessagePromptTemplate.fromTemplate(humanMessageTemplate),
         ];
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+        return chat_js_1.ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a ChatAgent instance using a language model, tools, and

package/dist/agents/chat/index.js

@@ -88,7 +88,7 @@ export class ChatAgent extends Agent {
             SystemMessagePromptTemplate.fromTemplate(template),
             HumanMessagePromptTemplate.fromTemplate(humanMessageTemplate),
         ];
-        return ChatPromptTemplate.fromPromptMessages(messages);
+        return ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a ChatAgent instance using a language model, tools, and

package/dist/agents/chat_convo/index.cjs

@@ -110,7 +110,7 @@ class ChatConversationalAgent extends agent_js_1.Agent {
             chat_js_1.HumanMessagePromptTemplate.fromTemplate(renderedHumanMessage),
             new chat_js_1.MessagesPlaceholder("agent_scratchpad"),
         ];
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+        return chat_js_1.ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates an instance of the ChatConversationalAgent class from a

package/dist/agents/chat_convo/index.js

@@ -107,7 +107,7 @@ export class ChatConversationalAgent extends Agent {
             HumanMessagePromptTemplate.fromTemplate(renderedHumanMessage),
             new MessagesPlaceholder("agent_scratchpad"),
         ];
-        return ChatPromptTemplate.fromPromptMessages(messages);
+        return ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates an instance of the ChatConversationalAgent class from a

package/dist/agents/executor.cjs

@@ -147,8 +147,15 @@ class AgentExecutor extends base_js_1.BaseChain {
                 // eslint-disable-next-line no-instanceof/no-instanceof
                 if (e instanceof output_parser_js_1.OutputParserException) {
                     let observation;
+                    let text = e.message;
                     if (this.handleParsingErrors === true) {
-                        observation = "Invalid or incomplete response";
+                        if (e.sendToLLM) {
+                            observation = e.observation;
+                            text = e.llmOutput ?? "";
+                        }
+                        else {
+                            observation = "Invalid or incomplete response";
+                        }
                     }
                     else if (typeof this.handleParsingErrors === "string") {
                         observation = this.handleParsingErrors;
@@ -162,7 +169,7 @@ class AgentExecutor extends base_js_1.BaseChain {
                     output = {
                         tool: "_Exception",
                         toolInput: observation,
-                        log: e.message,
+                        log: text,
                     };
                 }
                 else {

package/dist/agents/executor.js

@@ -143,8 +143,15 @@ export class AgentExecutor extends BaseChain {
                 // eslint-disable-next-line no-instanceof/no-instanceof
                 if (e instanceof OutputParserException) {
                     let observation;
+                    let text = e.message;
                     if (this.handleParsingErrors === true) {
-                        observation = "Invalid or incomplete response";
+                        if (e.sendToLLM) {
+                            observation = e.observation;
+                            text = e.llmOutput ?? "";
+                        }
+                        else {
+                            observation = "Invalid or incomplete response";
+                        }
                     }
                     else if (typeof this.handleParsingErrors === "string") {
                         observation = this.handleParsingErrors;
@@ -158,7 +165,7 @@ export class AgentExecutor extends BaseChain {
                     output = {
                         tool: "_Exception",
                         toolInput: observation,
-                        log: e.message,
+                        log: text,
                     };
                 }
                 else {

package/dist/agents/openai/index.cjs

@@ -102,7 +102,7 @@ class OpenAIAgent extends agent_js_1.Agent {
      */
     static createPrompt(_tools, fields) {
         const { prefix = prompt_js_1.PREFIX } = fields || {};
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages([
+        return chat_js_1.ChatPromptTemplate.fromMessages([
             chat_js_1.SystemMessagePromptTemplate.fromTemplate(prefix),
             new chat_js_1.MessagesPlaceholder("chat_history"),
             chat_js_1.HumanMessagePromptTemplate.fromTemplate("{input}"),

package/dist/agents/openai/index.js

@@ -98,7 +98,7 @@ export class OpenAIAgent extends Agent {
      */
     static createPrompt(_tools, fields) {
         const { prefix = PREFIX } = fields || {};
-        return ChatPromptTemplate.fromPromptMessages([
+        return ChatPromptTemplate.fromMessages([
             SystemMessagePromptTemplate.fromTemplate(prefix),
             new MessagesPlaceholder("chat_history"),
             HumanMessagePromptTemplate.fromTemplate("{input}"),

package/dist/agents/structured_chat/index.cjs

@@ -120,7 +120,7 @@ class StructuredChatAgent extends agent_js_1.Agent {
                 inputVariables,
             })),
         ];
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+        return chat_js_1.ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a StructuredChatAgent from an LLM and a list of tools.

package/dist/agents/structured_chat/index.js

@@ -117,7 +117,7 @@ export class StructuredChatAgent extends Agent {
                 inputVariables,
             })),
         ];
-        return ChatPromptTemplate.fromPromptMessages(messages);
+        return ChatPromptTemplate.fromMessages(messages);
     }
     /**
      * Creates a StructuredChatAgent from an LLM and a list of tools.

package/dist/agents/xml/index.cjs

@@ -65,7 +65,7 @@ class XMLAgent extends agent_js_1.BaseSingleActionAgent {
         return ["input"];
     }
     static createPrompt() {
-        return chat_js_1.ChatPromptTemplate.fromPromptMessages([
+        return chat_js_1.ChatPromptTemplate.fromMessages([
             chat_js_1.HumanMessagePromptTemplate.fromTemplate(prompt_js_1.AGENT_INSTRUCTIONS),
             chat_js_1.AIMessagePromptTemplate.fromTemplate("{intermediate_steps}"),
         ]);

package/dist/agents/xml/index.js

@@ -61,7 +61,7 @@ export class XMLAgent extends BaseSingleActionAgent {
         return ["input"];
     }
     static createPrompt() {
-        return ChatPromptTemplate.fromPromptMessages([
+        return ChatPromptTemplate.fromMessages([
             HumanMessagePromptTemplate.fromTemplate(AGENT_INSTRUCTIONS),
             AIMessagePromptTemplate.fromTemplate("{intermediate_steps}"),
         ]);

package/dist/base_language/count_tokens.cjs

@@ -58,11 +58,12 @@ exports.getModelContextSize = getModelContextSize;
 const calculateMaxTokens = async ({ prompt, modelName, }) => {
     let numTokens;
     try {
-        numTokens = (await (0, tiktoken_js_1.encodingForModel)(modelName)).encode(prompt).length;
+        numTokens = (await (0, tiktoken_js_1.encodingForModel)((0, exports.getModelNameForTiktoken)(modelName))).encode(prompt).length;
     }
     catch (error) {
         console.warn("Failed to calculate number of tokens, falling back to approximate count");
         // fallback to approximate calculation if tiktoken is not available
+        // each token is ~4 characters: https://help.openai.com/en/articles/4936856-what-are-tokens-and-how-to-count-them#
         numTokens = Math.ceil(prompt.length / 4);
     }
     const maxTokens = (0, exports.getModelContextSize)(modelName);

package/dist/base_language/count_tokens.js

@@ -52,11 +52,12 @@ export const getModelContextSize = (modelName) => {
 export const calculateMaxTokens = async ({ prompt, modelName, }) => {
     let numTokens;
     try {
-        numTokens = (await encodingForModel(modelName)).encode(prompt).length;
+        numTokens = (await encodingForModel(getModelNameForTiktoken(modelName))).encode(prompt).length;
     }
     catch (error) {
         console.warn("Failed to calculate number of tokens, falling back to approximate count");
         // fallback to approximate calculation if tiktoken is not available
+        // each token is ~4 characters: https://help.openai.com/en/articles/4936856-what-are-tokens-and-how-to-count-them#
         numTokens = Math.ceil(prompt.length / 4);
     }
     const maxTokens = getModelContextSize(modelName);

package/dist/base_language/index.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.calculateMaxTokens = exports.BaseLanguageModel = exports.BaseLangChain = void 0;
+exports.getModelContextSize = exports.calculateMaxTokens = exports.BaseLanguageModel = exports.BaseLangChain = void 0;
 const index_js_1 = require("../schema/index.cjs");
 const async_caller_js_1 = require("../util/async_caller.cjs");
 const count_tokens_js_1 = require("./count_tokens.cjs");
@@ -155,8 +155,10 @@ class BaseLanguageModel extends BaseLangChain {
 }
 exports.BaseLanguageModel = BaseLanguageModel;
 /*
- * Calculate max tokens for given model and prompt.
- * That is the model size - number of tokens in prompt.
+ * Export utility functions for token calculations:
+ * - calculateMaxTokens: Calculate max tokens for a given model and prompt (the model context size - tokens in prompt).
+ * - getModelContextSize: Get the context size for a specific model.
  */
 var count_tokens_js_2 = require("./count_tokens.cjs");
 Object.defineProperty(exports, "calculateMaxTokens", { enumerable: true, get: function () { return count_tokens_js_2.calculateMaxTokens; } });
+Object.defineProperty(exports, "getModelContextSize", { enumerable: true, get: function () { return count_tokens_js_2.getModelContextSize; } });

package/dist/base_language/index.d.ts

@@ -101,4 +101,4 @@ export declare abstract class BaseLanguageModel<RunOutput = any, CallOptions ext
      */
     static deserialize(data: SerializedLLM): Promise<BaseLanguageModel>;
 }
-export { calculateMaxTokens } from "./count_tokens.js";
+export { calculateMaxTokens, getModelContextSize } from "./count_tokens.js";

package/dist/base_language/index.js

@@ -150,7 +150,8 @@ export class BaseLanguageModel extends BaseLangChain {
     }
 }
 /*
- * Calculate max tokens for given model and prompt.
- * That is the model size - number of tokens in prompt.
+ * Export utility functions for token calculations:
+ * - calculateMaxTokens: Calculate max tokens for a given model and prompt (the model context size - tokens in prompt).
+ * - getModelContextSize: Get the context size for a specific model.
  */
-export { calculateMaxTokens } from "./count_tokens.js";
+export { calculateMaxTokens, getModelContextSize } from "./count_tokens.js";

package/dist/cache/cloudflare_kv.cjs

@@ -0,0 +1,61 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CloudflareKVCache = void 0;
+const index_js_1 = require("../schema/index.cjs");
+const base_js_1 = require("./base.cjs");
+/**
+ * Represents a specific implementation of a caching mechanism using Cloudflare KV
+ * as the underlying storage system. It extends the `BaseCache` class and
+ * overrides its methods to provide the Cloudflare KV-specific logic.
+ */
+class CloudflareKVCache extends index_js_1.BaseCache {
+    constructor(binding) {
+        super();
+        Object.defineProperty(this, "binding", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.binding = binding;
+    }
+    /**
+     * Retrieves data from the cache. It constructs a cache key from the given
+     * `prompt` and `llmKey`, and retrieves the corresponding value from the
+     * Cloudflare KV namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @returns An array of Generations if found, null otherwise.
+     */
+    async lookup(prompt, llmKey) {
+        let idx = 0;
+        let key = (0, base_js_1.getCacheKey)(prompt, llmKey, String(idx));
+        let value = await this.binding.get(key);
+        const generations = [];
+        while (value) {
+            if (!value) {
+                break;
+            }
+            generations.push({ text: value });
+            idx += 1;
+            key = (0, base_js_1.getCacheKey)(prompt, llmKey, String(idx));
+            value = await this.binding.get(key);
+        }
+        return generations.length > 0 ? generations : null;
+    }
+    /**
+     * Updates the cache with new data. It constructs a cache key from the
+     * given `prompt` and `llmKey`, and stores the `value` in the Cloudflare KV
+     * namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @param value The value to be stored in the cache.
+     */
+    async update(prompt, llmKey, value) {
+        for (let i = 0; i < value.length; i += 1) {
+            const key = (0, base_js_1.getCacheKey)(prompt, llmKey, String(i));
+            await this.binding.put(key, value[i].text);
+        }
+    }
+}
+exports.CloudflareKVCache = CloudflareKVCache;

package/dist/cache/cloudflare_kv.d.ts

@@ -0,0 +1,29 @@
+import type { KVNamespace } from "@cloudflare/workers-types";
+import { BaseCache, Generation } from "../schema/index.js";
+/**
+ * Represents a specific implementation of a caching mechanism using Cloudflare KV
+ * as the underlying storage system. It extends the `BaseCache` class and
+ * overrides its methods to provide the Cloudflare KV-specific logic.
+ */
+export declare class CloudflareKVCache extends BaseCache {
+    private binding;
+    constructor(binding: KVNamespace);
+    /**
+     * Retrieves data from the cache. It constructs a cache key from the given
+     * `prompt` and `llmKey`, and retrieves the corresponding value from the
+     * Cloudflare KV namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @returns An array of Generations if found, null otherwise.
+     */
+    lookup(prompt: string, llmKey: string): Promise<Generation[] | null>;
+    /**
+     * Updates the cache with new data. It constructs a cache key from the
+     * given `prompt` and `llmKey`, and stores the `value` in the Cloudflare KV
+     * namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @param value The value to be stored in the cache.
+     */
+    update(prompt: string, llmKey: string, value: Generation[]): Promise<void>;
+}

package/dist/cache/cloudflare_kv.js

@@ -0,0 +1,57 @@
+import { BaseCache } from "../schema/index.js";
+import { getCacheKey } from "./base.js";
+/**
+ * Represents a specific implementation of a caching mechanism using Cloudflare KV
+ * as the underlying storage system. It extends the `BaseCache` class and
+ * overrides its methods to provide the Cloudflare KV-specific logic.
+ */
+export class CloudflareKVCache extends BaseCache {
+    constructor(binding) {
+        super();
+        Object.defineProperty(this, "binding", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.binding = binding;
+    }
+    /**
+     * Retrieves data from the cache. It constructs a cache key from the given
+     * `prompt` and `llmKey`, and retrieves the corresponding value from the
+     * Cloudflare KV namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @returns An array of Generations if found, null otherwise.
+     */
+    async lookup(prompt, llmKey) {
+        let idx = 0;
+        let key = getCacheKey(prompt, llmKey, String(idx));
+        let value = await this.binding.get(key);
+        const generations = [];
+        while (value) {
+            if (!value) {
+                break;
+            }
+            generations.push({ text: value });
+            idx += 1;
+            key = getCacheKey(prompt, llmKey, String(idx));
+            value = await this.binding.get(key);
+        }
+        return generations.length > 0 ? generations : null;
+    }
+    /**
+     * Updates the cache with new data. It constructs a cache key from the
+     * given `prompt` and `llmKey`, and stores the `value` in the Cloudflare KV
+     * namespace.
+     * @param prompt The prompt used to construct the cache key.
+     * @param llmKey The LLM key used to construct the cache key.
+     * @param value The value to be stored in the cache.
+     */
+    async update(prompt, llmKey, value) {
+        for (let i = 0; i < value.length; i += 1) {
+            const key = getCacheKey(prompt, llmKey, String(i));
+            await this.binding.put(key, value[i].text);
+        }
+    }
+}

package/dist/chains/openai_functions/openapi.cjs

@@ -375,7 +375,7 @@ async function createOpenAPIChain(spec, options = {}) {
     if (defaultExecutionMethod === undefined) {
         throw new Error(`Could not parse any valid operations from the provided spec.`);
     }
-    const { llm = new openai_js_1.ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = chat_js_1.ChatPromptTemplate.fromPromptMessages([
+    const { llm = new openai_js_1.ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = chat_js_1.ChatPromptTemplate.fromMessages([
         chat_js_1.HumanMessagePromptTemplate.fromTemplate("Use the provided API's to respond to this user query:\n\n{query}"),
     ]), requestChain = new SimpleRequestChain({
         requestMethod: async (name, args) => defaultExecutionMethod(name, args, {

package/dist/chains/openai_functions/openapi.js

@@ -371,7 +371,7 @@ export async function createOpenAPIChain(spec, options = {}) {
     if (defaultExecutionMethod === undefined) {
         throw new Error(`Could not parse any valid operations from the provided spec.`);
     }
-    const { llm = new ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = ChatPromptTemplate.fromPromptMessages([
+    const { llm = new ChatOpenAI({ modelName: "gpt-3.5-turbo-0613" }), prompt = ChatPromptTemplate.fromMessages([
         HumanMessagePromptTemplate.fromTemplate("Use the provided API's to respond to this user query:\n\n{query}"),
     ]), requestChain = new SimpleRequestChain({
         requestMethod: async (name, args) => defaultExecutionMethod(name, args, {

package/dist/chains/openai_functions/structured_output.d.ts

@@ -46,5 +46,5 @@ export declare class FunctionCallStructuredOutputParser<T extends z.AnyZodObject
  * as well as an additional required "outputSchema" JSON Schema object.
  * @returns OpenAPIChain
  */
-export declare function createStructuredOutputChain<T extends z.AnyZodObject = z.AnyZodObject>(input: StructuredOutputChainInput): LLMChain<any, ChatOpenAI | BaseChatModel<BaseFunctionCallOptions>>;
-export declare function createStructuredOutputChainFromZod<T extends z.AnyZodObject>(zodSchema: T, input: Omit<StructuredOutputChainInput, "outputSchema">): LLMChain<any, ChatOpenAI | BaseChatModel<BaseFunctionCallOptions>>;
+export declare function createStructuredOutputChain<T extends z.AnyZodObject = z.AnyZodObject>(input: StructuredOutputChainInput): LLMChain<any, BaseChatModel<BaseFunctionCallOptions> | ChatOpenAI<BaseFunctionCallOptions>>;
+export declare function createStructuredOutputChainFromZod<T extends z.AnyZodObject>(zodSchema: T, input: Omit<StructuredOutputChainInput, "outputSchema">): LLMChain<any, BaseChatModel<BaseFunctionCallOptions> | ChatOpenAI<BaseFunctionCallOptions>>;

package/dist/chains/question_answering/map_reduce_prompts.cjs

@@ -21,8 +21,7 @@ const messages = [
     /*#__PURE__*/ chat_js_1.SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ chat_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_QA_PROMPT = 
-/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_QA_PROMPT = /*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromMessages(messages);
 exports.COMBINE_QA_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_COMBINE_QA_PROMPT, [
     [conditional_js_1.isChatModel, CHAT_QA_PROMPT],
@@ -68,7 +67,7 @@ const combine_messages = [
     /*#__PURE__*/ chat_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 const CHAT_COMBINE_PROMPT = 
-/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(combine_messages);
+/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromMessages(combine_messages);
 exports.COMBINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.COMBINE_PROMPT, [
     [conditional_js_1.isChatModel, CHAT_COMBINE_PROMPT],

package/dist/chains/question_answering/map_reduce_prompts.js

@@ -18,8 +18,7 @@ const messages = [
     /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_QA_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_QA_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);
 export const COMBINE_QA_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_COMBINE_QA_PROMPT, [
     [isChatModel, CHAT_QA_PROMPT],
@@ -65,7 +64,7 @@ const combine_messages = [
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 const CHAT_COMBINE_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(combine_messages);
+/*#__PURE__*/ ChatPromptTemplate.fromMessages(combine_messages);
 export const COMBINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(COMBINE_PROMPT, [
     [isChatModel, CHAT_COMBINE_PROMPT],

package/dist/chains/question_answering/refine_prompts.cjs

@@ -32,7 +32,7 @@ const messages = [
     /*#__PURE__*/ index_js_1.HumanMessagePromptTemplate.fromTemplate(refineTemplate),
 ];
 exports.CHAT_REFINE_PROMPT = 
-/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromMessages(messages);
 exports.REFINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_REFINE_PROMPT, [
     [conditional_js_1.isChatModel, exports.CHAT_REFINE_PROMPT],
@@ -56,7 +56,7 @@ const chat_messages = [
     /*#__PURE__*/ index_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 exports.CHAT_QUESTION_PROMPT = 
-/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromPromptMessages(chat_messages);
+/*#__PURE__*/ index_js_1.ChatPromptTemplate.fromMessages(chat_messages);
 exports.QUESTION_PROMPT_SELECTOR = 
 /*#__PURE__*/ new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_TEXT_QA_PROMPT, [
     [conditional_js_1.isChatModel, exports.CHAT_QUESTION_PROMPT],

package/dist/chains/question_answering/refine_prompts.js

@@ -29,7 +29,7 @@ const messages = [
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate(refineTemplate),
 ];
 export const CHAT_REFINE_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(messages);
+/*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);
 export const REFINE_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_REFINE_PROMPT, [
     [isChatModel, CHAT_REFINE_PROMPT],
@@ -53,7 +53,7 @@ const chat_messages = [
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
 export const CHAT_QUESTION_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(chat_messages);
+/*#__PURE__*/ ChatPromptTemplate.fromMessages(chat_messages);
 export const QUESTION_PROMPT_SELECTOR = 
 /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_TEXT_QA_PROMPT, [
     [isChatModel, CHAT_QUESTION_PROMPT],

package/dist/chains/question_answering/stuff_prompts.cjs

@@ -17,6 +17,5 @@ const messages = [
     /*#__PURE__*/ chat_js_1.SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ chat_js_1.HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_PROMPT = 
-/*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_PROMPT = /*#__PURE__*/ chat_js_1.ChatPromptTemplate.fromMessages(messages);
 exports.QA_PROMPT_SELECTOR = new conditional_js_1.ConditionalPromptSelector(exports.DEFAULT_QA_PROMPT, [[conditional_js_1.isChatModel, CHAT_PROMPT]]);

package/dist/chains/question_answering/stuff_prompts.js

@@ -14,6 +14,5 @@ const messages = [
     /*#__PURE__*/ SystemMessagePromptTemplate.fromTemplate(system_template),
     /*#__PURE__*/ HumanMessagePromptTemplate.fromTemplate("{question}"),
 ];
-const CHAT_PROMPT = 
-/*#__PURE__*/ ChatPromptTemplate.fromPromptMessages(messages);
+const CHAT_PROMPT = /*#__PURE__*/ ChatPromptTemplate.fromMessages(messages);
 export const QA_PROMPT_SELECTOR = /*#__PURE__*/ new ConditionalPromptSelector(DEFAULT_QA_PROMPT, [[isChatModel, CHAT_PROMPT]]);

package/dist/chat_models/base.d.ts

@@ -109,7 +109,7 @@ export declare abstract class BaseChatModel<CallOptions extends BaseChatModelCal
  * An abstract class that extends BaseChatModel and provides a simple
  * implementation of _generate.
  */
-export declare abstract class SimpleChatModel extends BaseChatModel {
+export declare abstract class SimpleChatModel<CallOptions extends BaseChatModelCallOptions = BaseChatModelCallOptions> extends BaseChatModel<CallOptions> {
     abstract _call(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<string>;
     _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
 }

package/dist/chat_models/fireworks.cjs

@@ -0,0 +1,81 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatFireworks = void 0;
+const openai_js_1 = require("./openai.cjs");
+const env_js_1 = require("../util/env.cjs");
+/**
+ * Wrapper around Fireworks API for large language models fine-tuned for chat
+ *
+ * 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 ChatFireworks extends openai_js_1.ChatOpenAI {
+    static lc_name() {
+        return "ChatFireworks";
+    }
+    _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,
+            modelName: fields?.modelName || "accounts/fireworks/models/llama-v2-13b-chat",
+            openAIApiKey: fireworksApiKey,
+            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) {
+        delete request.frequency_penalty;
+        delete request.presence_penalty;
+        delete request.logit_bias;
+        delete request.functions;
+        if (request.stream === true) {
+            return super.completionWithRetry(request, options);
+        }
+        return super.completionWithRetry(request, options);
+    }
+}
+exports.ChatFireworks = ChatFireworks;