langchain 0.0.5 → 0.0.7

package/dist/esm/llms/base.d.ts

@@ -1,64 +0,0 @@
-import { LLMCallbackManager, LLMResult } from "./index";
-export type SerializedLLM = {
-    _type: string;
-} & Record<string, any>;
-/**
- * LLM Wrapper. Provides an {@link call} (an {@link generate}) function that takes in a prompt (or prompts) and returns a string.
- */
-export declare abstract class BaseLLM {
-    /**
-     * The name of the LLM class
-     */
-    name: string;
-    cache?: boolean;
-    callbackManager: LLMCallbackManager;
-    /**
-     * Whether to print out response text.
-     */
-    verbose?: boolean;
-    constructor(callbackManager?: LLMCallbackManager, verbose?: boolean);
-    /**
-     * Run the LLM on the given prompts and input.
-     */
-    abstract _generate(prompts: string[], stop?: string[]): Promise<LLMResult>;
-    /** @ignore */
-    _generateUncached(prompts: string[], stop?: string[]): Promise<LLMResult>;
-    /**
-     * Run the LLM on the given propmts an input, handling caching.
-     */
-    generate(prompts: string[], stop?: string[]): Promise<LLMResult>;
-    /**
-     * Convenience wrapper for {@link generate} that takes in a single string prompt and returns a single string output.
-     */
-    call(prompt: string, stop?: string[]): Promise<string>;
-    /**
-     * Get the identifying parameters of the LLM.
-     */
-    _identifyingParams(): Record<string, any>;
-    /**
-     * Return the string type key uniquely identifying this class of LLM.
-     */
-    abstract _llmType(): string;
-    /**
-     * Return a json-like object representing this LLM.
-     */
-    serialize(): SerializedLLM;
-    /**
-     * Load an LLM from a json-like object describing it.
-     */
-    static deserialize(data: SerializedLLM): Promise<BaseLLM>;
-}
-/**
- * LLM class that provides a simpler interface to subclass than {@link BaseLLM}.
- *
- * Requires only implementing a simpler {@link _call} method instead of {@link _generate}.
- *
- * @augments BaseLLM
- */
-export declare abstract class LLM extends BaseLLM {
-    /**
-     * Run the LLM on the given prompt and input.
-     */
-    abstract _call(prompt: string, stop?: string[]): Promise<string>;
-    _generate(prompts: string[], stop?: string[]): Promise<LLMResult>;
-}

package/dist/esm/llms/base.js

@@ -1,159 +0,0 @@
-import { OpenAI } from "./index";
-import { InMemoryCache } from "../cache";
-const getCallbackManager = () => ({
-    handleStart: (..._args) => {
-        // console.log(args);
-    },
-    handleEnd: (..._args) => {
-        // console.log(args);
-    },
-    handleError: (..._args) => {
-        // console.log(args);
-    },
-});
-const getVerbosity = () => true;
-const cache = new InMemoryCache();
-/**
- * LLM Wrapper. Provides an {@link call} (an {@link generate}) function that takes in a prompt (or prompts) and returns a string.
- */
-export class BaseLLM {
-    constructor(callbackManager, verbose) {
-        /**
-         * The name of the LLM class
-         */
-        Object.defineProperty(this, "name", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "cache", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "callbackManager", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        /**
-         * Whether to print out response text.
-         */
-        Object.defineProperty(this, "verbose", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        this.callbackManager = callbackManager !== null && callbackManager !== void 0 ? callbackManager : getCallbackManager();
-        this.verbose = verbose !== null && verbose !== void 0 ? verbose : getVerbosity();
-    }
-    /** @ignore */
-    async _generateUncached(prompts, stop) {
-        this.callbackManager.handleStart({ name: this.name }, prompts, this.verbose);
-        let output;
-        try {
-            output = await this._generate(prompts, stop);
-        }
-        catch (err) {
-            this.callbackManager.handleError(`${err}`, this.verbose);
-            throw err;
-        }
-        this.callbackManager.handleEnd(output, this.verbose);
-        return output;
-    }
-    /**
-     * Run the LLM on the given propmts an input, handling caching.
-     */
-    async generate(prompts, stop) {
-        var _a;
-        if (!Array.isArray(prompts)) {
-            throw new Error("Argument 'prompts' is expected to be a string[]");
-        }
-        if (this.cache === true && cache === null) {
-            throw new Error("Requested cache, but no cache found");
-        }
-        if (cache === null || this.cache === false) {
-            return this._generateUncached(prompts, stop);
-        }
-        const params = this.serialize();
-        params.stop = stop;
-        const llmStringKey = `${Object.entries(params).sort()}`;
-        const missingPromptIndices = [];
-        const generations = prompts.map((prompt, index) => {
-            const result = 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) => prompts[i]), stop);
-            results.generations.forEach((generation, index) => {
-                const promptIndex = missingPromptIndices[index];
-                generations[promptIndex] = generation;
-                cache.update(prompts[promptIndex], llmStringKey, generation);
-            });
-            llmOutput = (_a = results.llmOutput) !== null && _a !== void 0 ? _a : {};
-        }
-        return { generations, llmOutput };
-    }
-    /**
-     * Convenience wrapper for {@link generate} that takes in a single string prompt and returns a single string output.
-     */
-    async call(prompt, stop) {
-        const { generations } = await this.generate([prompt], stop);
-        return generations[0][0].text;
-    }
-    /**
-     * Get the identifying parameters of the LLM.
-     */
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    _identifyingParams() {
-        return {};
-    }
-    /**
-     * Return a json-like object representing this LLM.
-     */
-    serialize() {
-        return {
-            ...this._identifyingParams(),
-            _type: this._llmType(),
-        };
-    }
-    /**
-     * Load an LLM from a json-like object describing it.
-     */
-    static async deserialize(data) {
-        const { _type, ...rest } = data;
-        const Cls = {
-            openai: OpenAI,
-        }[_type];
-        if (Cls === undefined) {
-            throw new Error(`Cannot load  LLM with type ${_type}`);
-        }
-        return new Cls(rest);
-    }
-}
-/**
- * LLM class that provides a simpler interface to subclass than {@link BaseLLM}.
- *
- * Requires only implementing a simpler {@link _call} method instead of {@link _generate}.
- *
- * @augments BaseLLM
- */
-export class LLM extends BaseLLM {
-    async _generate(prompts, stop) {
-        const generations = [];
-        for (let i = 0; i < prompts.length; i += 1) {
-            const text = await this._call(prompts[i], stop);
-            generations.push([{ text }]);
-        }
-        return { generations };
-    }
-}
-//# sourceMappingURL=base.js.map

package/dist/esm/llms/base.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../llms/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiC,MAAM,EAAE,MAAM,SAAS,CAAC;AAChE,OAAO,EAAa,aAAa,EAAE,MAAM,UAAU,CAAC;AAEpD,MAAM,kBAAkB,GAAG,GAAuB,EAAE,CAAC,CAAC;IACpD,WAAW,EAAE,CAAC,GAAG,KAAK,EAAE,EAAE;QACxB,qBAAqB;IACvB,CAAC;IACD,SAAS,EAAE,CAAC,GAAG,KAAK,EAAE,EAAE;QACtB,qBAAqB;IACvB,CAAC;IACD,WAAW,EAAE,CAAC,GAAG,KAAK,EAAE,EAAE;QACxB,qBAAqB;IACvB,CAAC;CACF,CAAC,CAAC;AAEH,MAAM,YAAY,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC;AAEhC,MAAM,KAAK,GAAc,IAAI,aAAa,EAAE,CAAC;AAO7C;;GAEG;AACH,MAAM,OAAgB,OAAO;IAe3B,YAAY,eAAoC,EAAE,OAAiB;QAdnE;;WAEG;QACH;;;;;WAAa;QAEb;;;;;WAAgB;QAEhB;;;;;WAAoC;QAEpC;;WAEG;QACH;;;;mBAAoB,KAAK;WAAC;QAGxB,IAAI,CAAC,eAAe,GAAG,eAAe,aAAf,eAAe,cAAf,eAAe,GAAI,kBAAkB,EAAE,CAAC;QAC/D,IAAI,CAAC,OAAO,GAAG,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,YAAY,EAAE,CAAC;IAC3C,CAAC;IAOD,cAAc;IACd,KAAK,CAAC,iBAAiB,CACrB,OAAiB,EACjB,IAAe;QAEf,IAAI,CAAC,eAAe,CAAC,WAAW,CAC9B,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EACnB,OAAO,EACP,IAAI,CAAC,OAAO,CACb,CAAC;QACF,IAAI,MAAM,CAAC;QACX,IAAI;YACF,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;SAC9C;QAAC,OAAO,GAAG,EAAE;YACZ,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,GAAG,GAAG,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACzD,MAAM,GAAG,CAAC;SACX;QAED,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,OAAiB,EAAE,IAAe;;QAC/C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC3B,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;SACpE;QAED,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,IAAI,EAAE;YACzC,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;SACxD;QAED,IAAI,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK,EAAE;YAC1C,OAAO,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;SAC9C;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;QAEnB,MAAM,YAAY,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;QACxD,MAAM,oBAAoB,GAAa,EAAE,CAAC;QAC1C,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;YAChD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YAClD,IAAI,CAAC,MAAM,EAAE;gBACX,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;aAClC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,IAAI,oBAAoB,CAAC,MAAM,GAAG,CAAC,EAAE;YACnC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAC1C,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAC3C,IAAI,CACL,CAAC;YACF,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE;gBAChD,MAAM,WAAW,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBAChD,WAAW,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC;gBACtC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;YAC/D,CAAC,CAAC,CAAC;YACH,SAAS,GAAG,MAAA,OAAO,CAAC,SAAS,mCAAI,EAAE,CAAC;SACrC;QAED,OAAO,EAAE,WAAW,EAAE,SAAS,EAAe,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,MAAc,EAAE,IAAe;QACxC,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QAC5D,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IAChC,CAAC;IAED;;OAEG;IACH,8DAA8D;IAC9D,kBAAkB;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAOD;;OAEG;IACH,SAAS;QACP,OAAO;YACL,GAAG,IAAI,CAAC,kBAAkB,EAAE;YAC5B,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE;SACvB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAmB;QAC1C,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,EAAE,GAAG,IAAI,CAAC;QAChC,MAAM,GAAG,GAAG;YACV,MAAM,EAAE,MAAM;SACf,CAAC,KAAK,CAAC,CAAC;QACT,IAAI,GAAG,KAAK,SAAS,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;SACxD;QACD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;CAGF;AAED;;;;;;GAMG;AACH,MAAM,OAAgB,GAAI,SAAQ,OAAO;IAMvC,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,IAAe;QAChD,MAAM,WAAW,GAAG,EAAE,CAAC;QACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;YAChD,WAAW,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;SAC9B;QACD,OAAO,EAAE,WAAW,EAAE,CAAC;IACzB,CAAC;CACF"}

package/dist/esm/llms/index.d.ts

@@ -1,37 +0,0 @@
-export { BaseLLM, LLM, SerializedLLM } from "./base";
-export { OpenAI } from "./openai";
-export { loadLLM } from "./load";
-export type LLMCallbackManager = {
-    handleStart: (llm: {
-        name: string;
-    }, prompts: string[], verbose?: boolean) => void;
-    handleError: (err: string, verbose?: boolean) => void;
-    handleEnd: (output: LLMResult, verbose?: boolean) => void;
-};
-/**
- * Output of a single generation.
- */
-export type Generation = {
-    /**
-     * Generated text output
-     */
-    text: string;
-    /**
-     * Raw generation info response from the provider.
-     * May include things like reason for finishing (e.g. in {@link OpenAI})
-     */
-    generationInfo?: Record<string, any>;
-};
-/**
- * Contains all relevant information returned by an LLM.
- */
-export type LLMResult = {
-    /**
-     * List of the things generated. Each input could have multiple {@link Generation | generations}, hence this is a list of lists.
-     */
-    generations: Generation[][];
-    /**
-     * Dictionary of arbitrary LLM-provider specific output.
-     */
-    llmOutput?: Record<string, any>;
-};

package/dist/esm/llms/index.js

@@ -1,4 +0,0 @@
-export { BaseLLM, LLM } from "./base";
-export { OpenAI } from "./openai";
-export { loadLLM } from "./load";
-//# sourceMappingURL=index.js.map

package/dist/esm/llms/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../llms/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,GAAG,EAAiB,MAAM,QAAQ,CAAC;AACrD,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,QAAQ,CAAC"}

package/dist/esm/llms/load.d.ts

@@ -1,11 +0,0 @@
-import { BaseLLM } from "./base";
-/**
- * Load an LLM from a local file.
- *
- * @example
- * ```ts
- * import { loadLLM } from "langchain/llms";
- * const model = await loadLLM("/path/to/llm.json");
- * ```
- */
-export declare const loadLLM: (file: string) => Promise<BaseLLM>;

package/dist/esm/llms/load.js

@@ -1,13 +0,0 @@
-import { BaseLLM } from "./base";
-import { parseFileConfig } from "../util";
-/**
- * Load an LLM from a local file.
- *
- * @example
- * ```ts
- * import { loadLLM } from "langchain/llms";
- * const model = await loadLLM("/path/to/llm.json");
- * ```
- */
-export const loadLLM = (file) => BaseLLM.deserialize(parseFileConfig(file));
-//# sourceMappingURL=load.js.map

package/dist/esm/llms/load.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../llms/load.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE1C;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG,CAAC,IAAY,EAAE,EAAE,CACtC,OAAO,CAAC,WAAW,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC"}

package/dist/esm/llms/openai.js

@@ -1,236 +0,0 @@
-import { backOff } from "exponential-backoff";
-import { chunkArray } from "../util";
-import { BaseLLM } from ".";
-let Configuration = null;
-let OpenAIApi = null;
-try {
-    // eslint-disable-next-line global-require,import/no-extraneous-dependencies
-    ({ Configuration, OpenAIApi } = require("openai"));
-}
-catch (_a) {
-    // ignore error
-}
-/**
- * Wrapper around OpenAI large language models.
- *
- * To use you should have the `openai` package installed, with the
- * `OPENAI_API_KEY` environment variable set.
- *
- * @remarks
- * Any parameters that are valid to be passed to {@link
- * https://platform.openai.com/docs/api-reference/completions/create |
- * `openai.createCompletion`} can be passed through {@link modelKwargs}, even
- * if not explicitly available on this class.
- *
- * @augments BaseLLM
- * @augments OpenAIInput
- */
-export class OpenAI extends BaseLLM {
-    constructor(fields) {
-        var _a, _b, _c, _d, _e, _f, _g, _h, _j, _k, _l, _m;
-        super(fields === null || fields === void 0 ? void 0 : fields.callbackManager, fields === null || fields === void 0 ? void 0 : fields.verbose);
-        Object.defineProperty(this, "temperature", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0.7
-        });
-        Object.defineProperty(this, "maxTokens", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 256
-        });
-        Object.defineProperty(this, "topP", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 1
-        });
-        Object.defineProperty(this, "frequencyPenalty", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-        Object.defineProperty(this, "presencePenalty", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-        Object.defineProperty(this, "n", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 1
-        });
-        Object.defineProperty(this, "bestOf", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 1
-        });
-        Object.defineProperty(this, "logitBias", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "modelName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "text-davinci-003"
-        });
-        Object.defineProperty(this, "modelKwargs", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "batchSize", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 20
-        });
-        Object.defineProperty(this, "maxRetries", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 6
-        });
-        Object.defineProperty(this, "stop", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "client", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        if (Configuration === null || OpenAIApi === null) {
-            throw new Error("Please install openai as a dependency with, e.g. `npm i openai`");
-        }
-        this.modelName = (_a = fields === null || fields === void 0 ? void 0 : fields.modelName) !== null && _a !== void 0 ? _a : this.modelName;
-        this.modelKwargs = (_b = fields === null || fields === void 0 ? void 0 : fields.modelKwargs) !== null && _b !== void 0 ? _b : {};
-        this.batchSize = (_c = fields === null || fields === void 0 ? void 0 : fields.batchSize) !== null && _c !== void 0 ? _c : this.batchSize;
-        this.maxRetries = (_d = fields === null || fields === void 0 ? void 0 : fields.maxRetries) !== null && _d !== void 0 ? _d : this.maxRetries;
-        this.temperature = (_e = fields === null || fields === void 0 ? void 0 : fields.temperature) !== null && _e !== void 0 ? _e : this.temperature;
-        this.maxTokens = (_f = fields === null || fields === void 0 ? void 0 : fields.maxTokens) !== null && _f !== void 0 ? _f : this.maxTokens;
-        this.topP = (_g = fields === null || fields === void 0 ? void 0 : fields.topP) !== null && _g !== void 0 ? _g : this.topP;
-        this.frequencyPenalty = (_h = fields === null || fields === void 0 ? void 0 : fields.frequencyPenalty) !== null && _h !== void 0 ? _h : this.frequencyPenalty;
-        this.presencePenalty = (_j = fields === null || fields === void 0 ? void 0 : fields.presencePenalty) !== null && _j !== void 0 ? _j : this.presencePenalty;
-        this.n = (_k = fields === null || fields === void 0 ? void 0 : fields.n) !== null && _k !== void 0 ? _k : this.n;
-        this.bestOf = (_l = fields === null || fields === void 0 ? void 0 : fields.bestOf) !== null && _l !== void 0 ? _l : this.bestOf;
-        this.logitBias = fields === null || fields === void 0 ? void 0 : fields.logitBias;
-        this.stop = fields === null || fields === void 0 ? void 0 : fields.stop;
-        const clientConfig = new Configuration({
-            apiKey: (_m = fields === null || fields === void 0 ? void 0 : fields.openAIApiKey) !== null && _m !== void 0 ? _m : process.env.OPENAI_API_KEY,
-        });
-        this.client = new OpenAIApi(clientConfig);
-    }
-    /**
-     * Get the parameters used to invoke the model
-     */
-    invocationParams() {
-        return {
-            model: this.modelName,
-            temperature: this.temperature,
-            max_tokens: this.maxTokens,
-            top_p: this.topP,
-            frequency_penalty: this.frequencyPenalty,
-            presence_penalty: this.presencePenalty,
-            n: this.n,
-            best_of: this.bestOf,
-            logit_bias: this.logitBias,
-            stop: this.stop,
-            ...this.modelKwargs,
-        };
-    }
-    /**
-     * Get the identifyin parameters for the model
-     */
-    identifyingParams() {
-        return {
-            model_name: this.modelName,
-            ...this.invocationParams(),
-        };
-    }
-    /**
-     * Call out to OpenAI's endpoint with k unique prompts
-     *
-     * @param prompts - The prompts to pass into the model.
-     * @param [stop] - Optional list of stop words to use when generating.
-     *
-     * @returns The full LLM output.
-     *
-     * @example
-     * ```ts
-     * import { OpenAI } from "langchain/llms";
-     * const openai = new OpenAI();
-     * const response = await openai.generate(["Tell me a joke."]);
-     * ```
-     */
-    async _generate(prompts, stop) {
-        var _a, _b, _c, _d;
-        const subPrompts = chunkArray(prompts, this.batchSize);
-        const choices = [];
-        const tokenUsage = {};
-        if (this.stop && stop) {
-            throw new Error("Stop found in input and default params");
-        }
-        const params = this.invocationParams();
-        params.stop = stop !== null && stop !== void 0 ? stop : params.stop;
-        for (let i = 0; i < subPrompts.length; i += 1) {
-            const { data } = await this.completionWithRetry({
-                ...params,
-                prompt: subPrompts[i],
-            });
-            choices.push(...data.choices);
-            const { completion_tokens: completionTokens, prompt_tokens: promptTokens, total_tokens: totalTokens, } = (_a = data.usage) !== null && _a !== void 0 ? _a : {};
-            if (completionTokens) {
-                tokenUsage.completionTokens =
-                    ((_b = tokenUsage.completionTokens) !== null && _b !== void 0 ? _b : 0) + completionTokens;
-            }
-            if (promptTokens) {
-                tokenUsage.promptTokens = ((_c = tokenUsage.promptTokens) !== null && _c !== void 0 ? _c : 0) + promptTokens;
-            }
-            if (totalTokens) {
-                tokenUsage.totalTokens = ((_d = tokenUsage.totalTokens) !== null && _d !== void 0 ? _d : 0) + totalTokens;
-            }
-        }
-        const generations = chunkArray(choices, this.n).map((promptChoices) => promptChoices.map((choice) => {
-            var _a;
-            return ({
-                text: (_a = choice.text) !== null && _a !== void 0 ? _a : "",
-                generationInfo: {
-                    finishReason: choice.finish_reason,
-                    logprobs: choice.logprobs,
-                },
-            });
-        }));
-        return {
-            generations,
-            llmOutput: { tokenUsage },
-        };
-    }
-    /** @ignore */
-    completionWithRetry(request) {
-        const makeCompletionRequest = () => this.client.createCompletion(request);
-        return backOff(makeCompletionRequest, {
-            startingDelay: 4,
-            maxDelay: 10,
-            numOfAttempts: this.maxRetries,
-            // TODO(sean) pass custom retry function to check error types.
-        });
-    }
-    _llmType() {
-        return "openai";
-    }
-}
-//# sourceMappingURL=openai.js.map

package/dist/esm/llms/openai.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../llms/openai.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,OAAO,EAAiC,MAAM,GAAG,CAAC;AAE3D,IAAI,aAAa,GAAiC,IAAI,CAAC;AACvD,IAAI,SAAS,GAA6B,IAAI,CAAC;AAE/C,IAAI;IACF,4EAA4E;IAC5E,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;CACpD;AAAC,WAAM;IACN,eAAe;CAChB;AAgED;;;;;;;;;;;;;;GAcG;AACH,MAAM,OAAO,MAAO,SAAQ,OAAO;IA6BjC,YACE,MAIC;;QAED,KAAK,CAAC,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,eAAe,EAAE,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,CAAC,CAAC;QAnClD;;;;mBAAc,GAAG;WAAC;QAElB;;;;mBAAY,GAAG;WAAC;QAEhB;;;;mBAAO,CAAC;WAAC;QAET;;;;mBAAmB,CAAC;WAAC;QAErB;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAI,CAAC;WAAC;QAEN;;;;mBAAS,CAAC;WAAC;QAEX;;;;;WAAmC;QAEnC;;;;mBAAY,kBAAkB;WAAC;QAE/B;;;;;WAAqB;QAErB;;;;mBAAY,EAAE;WAAC;QAEf;;;;mBAAa,CAAC;WAAC;QAEf;;;;;WAAgB;QAEhB;;;;;WAA2B;QAUzB,IAAI,aAAa,KAAK,IAAI,IAAI,SAAS,KAAK,IAAI,EAAE;YAChD,MAAM,IAAI,KAAK,CACb,iEAAiE,CAClE,CAAC;SACH;QAED,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,WAAW,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,WAAW,mCAAI,EAAE,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,UAAU,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,UAAU,mCAAI,IAAI,CAAC,UAAU,CAAC;QAExD,IAAI,CAAC,WAAW,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,WAAW,mCAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,IAAI,mCAAI,IAAI,CAAC,IAAI,CAAC;QACtC,IAAI,CAAC,gBAAgB,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,gBAAgB,mCAAI,IAAI,CAAC,gBAAgB,CAAC;QAC1E,IAAI,CAAC,eAAe,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,eAAe,mCAAI,IAAI,CAAC,eAAe,CAAC;QACvE,IAAI,CAAC,CAAC,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,CAAC,mCAAI,IAAI,CAAC,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,MAAM,mCAAI,IAAI,CAAC,MAAM,CAAC;QAC5C,IAAI,CAAC,SAAS,GAAG,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,CAAC;QACnC,IAAI,CAAC,IAAI,GAAG,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,IAAI,CAAC;QAEzB,MAAM,YAAY,GAAG,IAAI,aAAa,CAAC;YACrC,MAAM,EAAE,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,YAAY,mCAAI,OAAO,CAAC,GAAG,CAAC,cAAc;SAC3D,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,IAAI,SAAS,CAAC,YAAY,CAAC,CAAC;IAC5C,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,SAAS;YACrB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB;YACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe;YACtC,CAAC,EAAE,IAAI,CAAC,CAAC;YACT,OAAO,EAAE,IAAI,CAAC,MAAM;YACpB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,GAAG,IAAI,CAAC,WAAW;SACpB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,IAAI,CAAC,gBAAgB,EAAE;SAC3B,CAAC;IACJ,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,KAAK,CAAC,SAAS,CAAC,OAAiB,EAAE,IAAe;;QAChD,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QACvD,MAAM,OAAO,GAA2C,EAAE,CAAC;QAC3D,MAAM,UAAU,GAAe,EAAE,CAAC;QAElC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;YACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;SAC3D;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,MAAM,CAAC,IAAI,GAAG,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,MAAM,CAAC,IAAI,CAAC;QAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;YAC7C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC;gBAC9C,GAAG,MAAM;gBACT,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC;aACtB,CAAC,CAAC;YACH,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;YAC9B,MAAM,EACJ,iBAAiB,EAAE,gBAAgB,EACnC,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,WAAW,GAC1B,GAAG,MAAA,IAAI,CAAC,KAAK,mCAAI,EAAE,CAAC;YAErB,IAAI,gBAAgB,EAAE;gBACpB,UAAU,CAAC,gBAAgB;oBACzB,CAAC,MAAA,UAAU,CAAC,gBAAgB,mCAAI,CAAC,CAAC,GAAG,gBAAgB,CAAC;aACzD;YAED,IAAI,YAAY,EAAE;gBAChB,UAAU,CAAC,YAAY,GAAG,CAAC,MAAA,UAAU,CAAC,YAAY,mCAAI,CAAC,CAAC,GAAG,YAAY,CAAC;aACzE;YAED,IAAI,WAAW,EAAE;gBACf,UAAU,CAAC,WAAW,GAAG,CAAC,MAAA,UAAU,CAAC,WAAW,mCAAI,CAAC,CAAC,GAAG,WAAW,CAAC;aACtE;SACF;QAED,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CACpE,aAAa,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;;YAAC,OAAA,CAAC;gBAC7B,IAAI,EAAE,MAAA,MAAM,CAAC,IAAI,mCAAI,EAAE;gBACvB,cAAc,EAAE;oBACd,YAAY,EAAE,MAAM,CAAC,aAAa;oBAClC,QAAQ,EAAE,MAAM,CAAC,QAAQ;iBAC1B;aACF,CAAC,CAAA;SAAA,CAAC,CACJ,CAAC;QACF,OAAO;YACL,WAAW;YACX,SAAS,EAAE,EAAE,UAAU,EAAE;SAC1B,CAAC;IACJ,CAAC;IAED,cAAc;IACd,mBAAmB,CAAC,OAAgC;QAClD,MAAM,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC1E,OAAO,OAAO,CAAC,qBAAqB,EAAE;YACpC,aAAa,EAAE,CAAC;YAChB,QAAQ,EAAE,EAAE;YACZ,aAAa,EAAE,IAAI,CAAC,UAAU;YAC9B,8DAA8D;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,QAAQ;QACN,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/esm/llms/tests/openai.test.d.ts

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

package/dist/esm/llms/tests/openai.test.js

@@ -1,8 +0,0 @@
-import { test } from "@jest/globals";
-import { OpenAI } from "../openai";
-test("Test OpenAI", async () => {
-    const model = new OpenAI({ maxTokens: 5 });
-    const res = await model.call("Print hello world");
-    console.log({ res });
-});
-//# sourceMappingURL=openai.test.js.map

package/dist/esm/llms/tests/openai.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"openai.test.js","sourceRoot":"","sources":["../../../../llms/tests/openai.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAEnC,IAAI,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IAC3C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAClD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/esm/memory/base.d.ts

@@ -1,7 +0,0 @@
-export type InputValues = Record<string, any>;
-export type OutputValues = Record<string, any>;
-export type MemoryVariables = Record<string, any>;
-export declare abstract class BaseMemory {
-    abstract loadMemoryVariables(values: InputValues): Promise<MemoryVariables>;
-    abstract saveContext(inputValues: InputValues, OutputValues: Promise<OutputValues>): Promise<void>;
-}

package/dist/esm/memory/base.js

@@ -1,3 +0,0 @@
-export class BaseMemory {
-}
-//# sourceMappingURL=base.js.map

package/dist/esm/memory/base.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../memory/base.ts"],"names":[],"mappings":"AAOA,MAAM,OAAgB,UAAU;CAO/B"}

package/dist/esm/memory/buffer_memory.d.ts

@@ -1,15 +0,0 @@
-import { BaseMemory, InputValues, MemoryVariables, OutputValues } from "./base";
-export interface BufferMemoryInput {
-    humanPrefix: string;
-    aiPrefix: string;
-    memoryKey: string;
-}
-export declare class BufferMemory extends BaseMemory implements BufferMemoryInput {
-    humanPrefix: string;
-    aiPrefix: string;
-    memoryKey: string;
-    buffer: string;
-    constructor(fields?: Partial<BufferMemoryInput>);
-    loadMemoryVariables(_values: InputValues): Promise<MemoryVariables>;
-    saveContext(inputValues: InputValues, outputValues: Promise<OutputValues>): Promise<void>;
-}

package/dist/esm/memory/buffer_memory.js

@@ -1,53 +0,0 @@
-import { BaseMemory } from "./base";
-const getInputValue = (inputValues) => {
-    const keys = Object.keys(inputValues);
-    if (keys.length === 1) {
-        return inputValues[keys[0]];
-    }
-    throw new Error("input values have multiple keys, memory only supported when one key currently");
-};
-export class BufferMemory extends BaseMemory {
-    constructor(fields) {
-        var _a, _b, _c;
-        super();
-        Object.defineProperty(this, "humanPrefix", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "Human"
-        });
-        Object.defineProperty(this, "aiPrefix", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "AI"
-        });
-        Object.defineProperty(this, "memoryKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "history"
-        });
-        Object.defineProperty(this, "buffer", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: ""
-        });
-        this.humanPrefix = (_a = fields === null || fields === void 0 ? void 0 : fields.humanPrefix) !== null && _a !== void 0 ? _a : this.humanPrefix;
-        this.aiPrefix = (_b = fields === null || fields === void 0 ? void 0 : fields.aiPrefix) !== null && _b !== void 0 ? _b : this.aiPrefix;
-        this.memoryKey = (_c = fields === null || fields === void 0 ? void 0 : fields.memoryKey) !== null && _c !== void 0 ? _c : this.memoryKey;
-    }
-    async loadMemoryVariables(_values) {
-        const result = { [this.memoryKey]: this.buffer };
-        return result;
-    }
-    async saveContext(inputValues, outputValues) {
-        const values = await outputValues;
-        const human = `${this.humanPrefix}: ${getInputValue(inputValues)}`;
-        const ai = `${this.aiPrefix}: ${getInputValue(values)}`;
-        const newlines = [human, ai];
-        this.buffer += `\n${newlines.join("\n")}`;
-    }
-}
-//# sourceMappingURL=buffer_memory.js.map

package/dist/esm/memory/buffer_memory.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"buffer_memory.js","sourceRoot":"","sources":["../../../memory/buffer_memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAA8C,MAAM,QAAQ,CAAC;AAQhF,MAAM,aAAa,GAAG,CAAC,WAAwB,EAAE,EAAE;IACjD,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACtC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;QACrB,OAAO,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;KAC7B;IACD,MAAM,IAAI,KAAK,CACb,+EAA+E,CAChF,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,OAAO,YAAa,SAAQ,UAAU;IAS1C,YAAY,MAAmC;;QAC7C,KAAK,EAAE,CAAC;QATV;;;;mBAAc,OAAO;WAAC;QAEtB;;;;mBAAW,IAAI;WAAC;QAEhB;;;;mBAAY,SAAS;WAAC;QAEtB;;;;mBAAS,EAAE;WAAC;QAIV,IAAI,CAAC,WAAW,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,WAAW,mCAAI,IAAI,CAAC,WAAW,CAAC;QAC3D,IAAI,CAAC,QAAQ,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,QAAQ,mCAAI,IAAI,CAAC,QAAQ,CAAC;QAClD,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;IACvD,CAAC;IAED,KAAK,CAAC,mBAAmB,CAAC,OAAoB;QAC5C,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;QACjD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,WAAW,CACf,WAAwB,EACxB,YAAmC;QAEnC,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;QAClC,MAAM,KAAK,GAAG,GAAG,IAAI,CAAC,WAAW,KAAK,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;QACnE,MAAM,EAAE,GAAG,GAAG,IAAI,CAAC,QAAQ,KAAK,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;QACxD,MAAM,QAAQ,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7B,IAAI,CAAC,MAAM,IAAI,KAAK,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;IAC5C,CAAC;CACF"}