langchain 0.0.151 → 0.0.153

package/dist/chat_models/fireworks.d.ts

@@ -0,0 +1,33 @@
+import type { OpenAI as OpenAIClient } from "openai";
+import type { ChatOpenAICallOptions, OpenAIChatInput } from "./openai.js";
+import type { OpenAICoreRequestOptions } from "../types/openai-types.js";
+import type { BaseChatModelParams } from "./base.js";
+import { ChatOpenAI } from "./openai.js";
+type FireworksUnsupportedArgs = "frequencyPenalty" | "presencePenalty" | "logitBias" | "functions";
+type FireworksUnsupportedCallOptions = "functions" | "function_call" | "tools";
+export type ChatFireworksCallOptions = Partial<Omit<ChatOpenAICallOptions, FireworksUnsupportedCallOptions>>;
+/**
+ * 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.
+ */
+export declare class ChatFireworks extends ChatOpenAI<ChatFireworksCallOptions> {
+    static lc_name(): string;
+    _llmType(): string;
+    get lc_secrets(): {
+        [key: string]: string;
+    } | undefined;
+    lc_serializable: boolean;
+    fireworksApiKey?: string;
+    constructor(fields?: Partial<Omit<OpenAIChatInput, "openAIApiKey" | FireworksUnsupportedArgs>> & BaseChatModelParams & {
+        fireworksApiKey?: string;
+    });
+    toJSON(): import("../load/serializable.js").Serialized;
+    completionWithRetry(request: OpenAIClient.Chat.ChatCompletionCreateParamsStreaming, options?: OpenAICoreRequestOptions): Promise<AsyncIterable<OpenAIClient.Chat.Completions.ChatCompletionChunk>>;
+    completionWithRetry(request: OpenAIClient.Chat.ChatCompletionCreateParamsNonStreaming, options?: OpenAICoreRequestOptions): Promise<OpenAIClient.Chat.Completions.ChatCompletion>;
+}
+export {};

package/dist/chat_models/fireworks.js

@@ -0,0 +1,77 @@
+import { ChatOpenAI } from "./openai.js";
+import { getEnvironmentVariable } from "../util/env.js";
+/**
+ * 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.
+ */
+export class ChatFireworks extends ChatOpenAI {
+    static lc_name() {
+        return "ChatFireworks";
+    }
+    _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,
+            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);
+    }
+}

package/dist/chat_models/ollama.cjs

@@ -303,11 +303,28 @@ class ChatOllama extends base_js_1.SimpleChatModel {
             prompt: this._formatMessagesAsPrompt(input),
         }, options));
         for await (const chunk of stream) {
-            yield new index_js_1.ChatGenerationChunk({
-                text: chunk.response,
-                message: new index_js_1.AIMessageChunk({ content: chunk.response }),
-            });
-            await runManager?.handleLLMNewToken(chunk.response ?? "");
+            if (!chunk.done) {
+                yield new index_js_1.ChatGenerationChunk({
+                    text: chunk.response,
+                    message: new index_js_1.AIMessageChunk({ content: chunk.response }),
+                });
+                await runManager?.handleLLMNewToken(chunk.response ?? "");
+            }
+            else {
+                yield new index_js_1.ChatGenerationChunk({
+                    text: "",
+                    message: new index_js_1.AIMessageChunk({ content: "" }),
+                    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,
+                    },
+                });
+            }
         }
     }
     _formatMessagesAsPrompt(messages) {
@@ -336,14 +353,10 @@ class ChatOllama extends base_js_1.SimpleChatModel {
         return formattedMessages;
     }
     /** @ignore */
-    async _call(messages, options) {
-        const stream = await this.caller.call(async () => (0, ollama_js_1.createOllamaStream)(this.baseUrl, {
-            ...this.invocationParams(options),
-            prompt: this._formatMessagesAsPrompt(messages),
-        }, options));
+    async _call(messages, options, runManager) {
         const chunks = [];
-        for await (const chunk of stream) {
-            chunks.push(chunk.response);
+        for await (const chunk of this._streamResponseChunks(messages, options, runManager)) {
+            chunks.push(chunk.message.content);
         }
         return chunks.join("");
     }

package/dist/chat_models/ollama.d.ts

@@ -14,8 +14,7 @@ export interface OllamaCallOptions extends BaseLanguageModelCallOptions {
  * models in a chat-like fashion. It extends the SimpleChatModel class and
  * implements the OllamaInput interface.
  */
-export declare class ChatOllama extends SimpleChatModel implements OllamaInput {
-    CallOptions: OllamaCallOptions;
+export declare class ChatOllama extends SimpleChatModel<OllamaCallOptions> implements OllamaInput {
     static lc_name(): string;
     lc_serializable: boolean;
     model: string;
@@ -97,5 +96,5 @@ export declare class ChatOllama extends SimpleChatModel implements OllamaInput {
     _streamResponseChunks(input: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
     protected _formatMessagesAsPrompt(messages: BaseMessage[]): string;
     /** @ignore */
-    _call(messages: BaseMessage[], options: this["ParsedCallOptions"]): Promise<string>;
+    _call(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<string>;
 }

package/dist/chat_models/ollama.js

@@ -300,11 +300,28 @@ export class ChatOllama extends SimpleChatModel {
             prompt: this._formatMessagesAsPrompt(input),
         }, options));
         for await (const chunk of stream) {
-            yield new ChatGenerationChunk({
-                text: chunk.response,
-                message: new AIMessageChunk({ content: chunk.response }),
-            });
-            await runManager?.handleLLMNewToken(chunk.response ?? "");
+            if (!chunk.done) {
+                yield new ChatGenerationChunk({
+                    text: chunk.response,
+                    message: new AIMessageChunk({ content: chunk.response }),
+                });
+                await runManager?.handleLLMNewToken(chunk.response ?? "");
+            }
+            else {
+                yield new ChatGenerationChunk({
+                    text: "",
+                    message: new AIMessageChunk({ content: "" }),
+                    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,
+                    },
+                });
+            }
         }
     }
     _formatMessagesAsPrompt(messages) {
@@ -333,14 +350,10 @@ export class ChatOllama extends SimpleChatModel {
         return formattedMessages;
     }
     /** @ignore */
-    async _call(messages, options) {
-        const stream = await this.caller.call(async () => createOllamaStream(this.baseUrl, {
-            ...this.invocationParams(options),
-            prompt: this._formatMessagesAsPrompt(messages),
-        }, options));
+    async _call(messages, options, runManager) {
         const chunks = [];
-        for await (const chunk of stream) {
-            chunks.push(chunk.response);
+        for await (const chunk of this._streamResponseChunks(messages, options, runManager)) {
+            chunks.push(chunk.message.content);
         }
         return chunks.join("");
     }

package/dist/chat_models/openai.d.ts

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

package/dist/document_loaders/web/pdf.cjs

@@ -0,0 +1,87 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebPDFLoader = void 0;
+const pdf_js_1 = require("pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js");
+const document_js_1 = require("../../document.cjs");
+const base_js_1 = require("../base.cjs");
+/**
+ * A document loader for loading data from PDFs.
+ */
+class WebPDFLoader extends base_js_1.BaseDocumentLoader {
+    constructor(blob, { splitPages = true } = {}) {
+        super();
+        Object.defineProperty(this, "blob", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "splitPages", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        this.blob = blob;
+        this.splitPages = splitPages ?? this.splitPages;
+    }
+    /**
+     * Loads the contents of the PDF as documents.
+     * @returns An array of Documents representing the retrieved data.
+     */
+    async load() {
+        const parsedPdf = await (0, pdf_js_1.getDocument)({
+            data: new Uint8Array(await this.blob.arrayBuffer()),
+            useWorkerFetch: false,
+            isEvalSupported: false,
+            useSystemFonts: true,
+        }).promise;
+        const meta = await parsedPdf.getMetadata().catch(() => null);
+        const documents = [];
+        for (let i = 1; i <= parsedPdf.numPages; i += 1) {
+            const page = await parsedPdf.getPage(i);
+            const content = await page.getTextContent();
+            if (content.items.length === 0) {
+                continue;
+            }
+            const text = content.items
+                .map((item) => item.str)
+                .join("\n");
+            documents.push(new document_js_1.Document({
+                pageContent: text,
+                metadata: {
+                    pdf: {
+                        version: pdf_js_1.version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                    loc: {
+                        pageNumber: i,
+                    },
+                },
+            }));
+        }
+        if (this.splitPages) {
+            return documents;
+        }
+        if (documents.length === 0) {
+            return [];
+        }
+        return [
+            new document_js_1.Document({
+                pageContent: documents.map((doc) => doc.pageContent).join("\n\n"),
+                metadata: {
+                    pdf: {
+                        version: pdf_js_1.version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                },
+            }),
+        ];
+        return documents;
+    }
+}
+exports.WebPDFLoader = WebPDFLoader;

package/dist/document_loaders/web/pdf.d.ts

@@ -0,0 +1,17 @@
+import { Document } from "../../document.js";
+import { BaseDocumentLoader } from "../base.js";
+/**
+ * A document loader for loading data from PDFs.
+ */
+export declare class WebPDFLoader extends BaseDocumentLoader {
+    protected blob: Blob;
+    protected splitPages: boolean;
+    constructor(blob: Blob, { splitPages }?: {
+        splitPages?: boolean | undefined;
+    });
+    /**
+     * Loads the contents of the PDF as documents.
+     * @returns An array of Documents representing the retrieved data.
+     */
+    load(): Promise<Document[]>;
+}

package/dist/document_loaders/web/pdf.js

@@ -0,0 +1,83 @@
+import { getDocument, version, } from "pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js";
+import { Document } from "../../document.js";
+import { BaseDocumentLoader } from "../base.js";
+/**
+ * A document loader for loading data from PDFs.
+ */
+export class WebPDFLoader extends BaseDocumentLoader {
+    constructor(blob, { splitPages = true } = {}) {
+        super();
+        Object.defineProperty(this, "blob", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "splitPages", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        this.blob = blob;
+        this.splitPages = splitPages ?? this.splitPages;
+    }
+    /**
+     * Loads the contents of the PDF as documents.
+     * @returns An array of Documents representing the retrieved data.
+     */
+    async load() {
+        const parsedPdf = await getDocument({
+            data: new Uint8Array(await this.blob.arrayBuffer()),
+            useWorkerFetch: false,
+            isEvalSupported: false,
+            useSystemFonts: true,
+        }).promise;
+        const meta = await parsedPdf.getMetadata().catch(() => null);
+        const documents = [];
+        for (let i = 1; i <= parsedPdf.numPages; i += 1) {
+            const page = await parsedPdf.getPage(i);
+            const content = await page.getTextContent();
+            if (content.items.length === 0) {
+                continue;
+            }
+            const text = content.items
+                .map((item) => item.str)
+                .join("\n");
+            documents.push(new Document({
+                pageContent: text,
+                metadata: {
+                    pdf: {
+                        version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                    loc: {
+                        pageNumber: i,
+                    },
+                },
+            }));
+        }
+        if (this.splitPages) {
+            return documents;
+        }
+        if (documents.length === 0) {
+            return [];
+        }
+        return [
+            new Document({
+                pageContent: documents.map((doc) => doc.pageContent).join("\n\n"),
+                metadata: {
+                    pdf: {
+                        version,
+                        info: meta?.info,
+                        metadata: meta?.metadata,
+                        totalPages: parsedPdf.numPages,
+                    },
+                },
+            }),
+        ];
+        return documents;
+    }
+}

package/dist/evaluation/agents/prompt.cjs

@@ -90,8 +90,7 @@ The model did not use the appropriate tools to answer the question.\
 Judgment: Given the good reasoning in the final answer but otherwise poor performance, we give the model a score of 2.
 
 Score: 2`;
-exports.EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ index_js_1.ChatPromptTemplate.fromPromptMessages([
+exports.EVAL_CHAT_PROMPT = index_js_1.ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ index_js_1.SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ index_js_1.HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ index_js_1.AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),
@@ -124,7 +123,7 @@ i. Is the final answer helpful?
     iv. Does the AI language model use too many steps to answer the question?
     v. Are the appropriate tools used to answer the question?`;
 exports.TOOL_FREE_EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ index_js_1.ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ index_js_1.ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ index_js_1.SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ index_js_1.HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ index_js_1.AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),

package/dist/evaluation/agents/prompt.js

@@ -87,8 +87,7 @@ The model did not use the appropriate tools to answer the question.\
 Judgment: Given the good reasoning in the final answer but otherwise poor performance, we give the model a score of 2.
 
 Score: 2`;
-export const EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ ChatPromptTemplate.fromPromptMessages([
+export const EVAL_CHAT_PROMPT = /* #__PURE__ */ ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),
@@ -121,7 +120,7 @@ i. Is the final answer helpful?
     iv. Does the AI language model use too many steps to answer the question?
     v. Are the appropriate tools used to answer the question?`;
 export const TOOL_FREE_EVAL_CHAT_PROMPT = 
-/* #__PURE__ */ ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ SystemMessagePromptTemplate.fromTemplate("You are a helpful assistant that evaluates language models."),
     /* #__PURE__ */ HumanMessagePromptTemplate.fromTemplate(EXAMPLE_INPUT),
     /* #__PURE__ */ AIMessagePromptTemplate.fromTemplate(EXAMPLE_OUTPUT),

package/dist/experimental/plan_and_execute/prompt.cjs

@@ -14,7 +14,7 @@ exports.PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE = [
     `At the end of your plan, say "<END_OF_PLAN>"`,
 ].join(" ");
 exports.PLANNER_CHAT_PROMPT = 
-/* #__PURE__ */ chat_js_1.ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ chat_js_1.ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ chat_js_1.SystemMessagePromptTemplate.fromTemplate(exports.PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE),
     /* #__PURE__ */ chat_js_1.HumanMessagePromptTemplate.fromTemplate(`{input}`),
 ]);

package/dist/experimental/plan_and_execute/prompt.js

@@ -11,7 +11,7 @@ export const PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE = [
     `At the end of your plan, say "<END_OF_PLAN>"`,
 ].join(" ");
 export const PLANNER_CHAT_PROMPT = 
-/* #__PURE__ */ ChatPromptTemplate.fromPromptMessages([
+/* #__PURE__ */ ChatPromptTemplate.fromMessages([
     /* #__PURE__ */ SystemMessagePromptTemplate.fromTemplate(PLANNER_SYSTEM_PROMPT_MESSAGE_TEMPLATE),
     /* #__PURE__ */ HumanMessagePromptTemplate.fromTemplate(`{input}`),
 ]);

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