langchain 0.0.4

package/dist/esm/chains/llm_math_chain.js

@@ -0,0 +1,129 @@
+"use strict";
+/*
+import { BaseChain, ChainValues, LLMChain } from "./index";
+
+import { BaseLLM, SerializedLLM } from "../llms";
+import { BasePromptTemplate, SerializedBasePromptTemplate } from "../prompt";
+
+import { resolveConfigFromFile } from "../util";
+
+export interface LLMMathChainInput {
+  prompt: BasePromptTemplate;
+  llm: BaseLLM;
+  outputKey: string;
+  inputKey: string;
+}
+
+export type SerializedLLMChain = {
+  _type: "llm_chain";
+  llm?: SerializedLLM;
+  llm_path?: string;
+  prompt?: SerializedBasePromptTemplate;
+  prompt_path?: string;
+};
+
+export class LLMChain extends BaseChain implements LLMMathChainInput {
+  prompt: BasePromptTemplate;
+
+  llm: BaseLLM;
+
+  inputKey = "answer";
+
+  outputKey = "question";
+
+  constructor(fields: {
+    prompt: BasePromptTemplate;
+    llm: BaseLLM;
+    outputKey?: string;
+    inputKey?: string;
+  }) {
+    super();
+    this.prompt = fields.prompt;
+    this.llm = fields.llm;
+    this.outputKey = fields.outputKey ?? this.outputKey;
+    this.inputKey = fields.inputKey ?? this.inputKey;
+  }
+
+  async _call(values: ChainValues, stop?: string[]): Promise<ChainValues> {
+    const llm = new LLMChain({ prompt: this.prompt, llm: this.llm });
+    const p;
+    const formattedString = this.prompt.format(values);
+    const llmResult = await this.llm.call(formattedString, stop);
+    const result = { [this.outputKey]: llmResult };
+    return result;
+  }
+
+  async predict(values: ChainValues, stop?: string[]): Promise<string> {
+    const output = await this.call(values, stop);
+    return output[this.outputKey];
+  }
+
+  _chainType() {
+    return "llm_chain" as const;
+  }
+
+  static async deserialize(data: SerializedLLMChain) {
+    const serializedLLM = resolveConfigFromFile<"llm", SerializedLLM>(
+      "llm",
+      data
+    );
+    const serializedPrompt = resolveConfigFromFile<
+      "prompt",
+      SerializedBasePromptTemplate
+    >("prompt", data);
+
+    return new LLMChain({
+      llm: await BaseLLM.deserialize(serializedLLM),
+      prompt: await BasePromptTemplate.deserialize(serializedPrompt),
+    });
+  }
+
+  serialize(): SerializedLLMChain {
+    return {
+      _type: this._chainType(),
+      llm: this.llm.serialize(),
+      prompt: this.prompt.serialize(),
+    };
+  }
+}
+
+const PROMPT_TEMPLATE = `You are GPT-3, and you can't do math.
+You can do basic math, and your memorization abilities are impressive, but you can't do any complex calculations that a human could not do in their head. You also have an annoying tendency to just make up highly specific, but wrong, answers.
+
+So we hooked you up to a Python 3 kernel, and now you can execute code. If anyone gives you a hard math problem, just use this format and we’ll take care of the rest:
+
+Question: \${{Question with hard calculation.}}
+\`\`\`python
+\${{Code that prints what you need to know}}
+\`\`\`
+\`\`\`output
+\${{Output of your code}}
+\`\`\`
+Answer: \${{Answer}}
+
+Otherwise, use this simpler format:
+
+Question: \${{Question without hard calculation}}
+Answer: \${{Answer}}
+
+Begin.
+
+Question: What is 37593 * 67?
+
+\`\`\`python
+print(37593 * 67)
+\`\`\`
+\`\`\`output
+2518731
+\`\`\`
+Answer: 2518731
+
+Question: {question}
+`;
+
+PROMPT = PromptTemplate(
+  (input_variables = ["question"]),
+  (template = _PROMPT_TEMPLATE)
+);
+*/
+//# sourceMappingURL=llm_math_chain.js.map

package/dist/esm/chains/llm_math_chain.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_math_chain.js","sourceRoot":"","sources":["../../../chains/llm_math_chain.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA8HE"}

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

@@ -0,0 +1,2 @@
+import { BaseChain } from ".";
+export declare const loadChain: (uri: string) => Promise<BaseChain>;

package/dist/esm/chains/load.js

@@ -0,0 +1,25 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { BaseChain } from ".";
+import { loadFromHub } from "../util/hub";
+import { parseFileConfig } from "../util";
+const loadChainFromFile = (file) => __awaiter(void 0, void 0, void 0, function* () {
+    const serialized = parseFileConfig(file);
+    console.log({ serialized });
+    return BaseChain.deserialize(serialized);
+});
+export const loadChain = (uri) => __awaiter(void 0, void 0, void 0, function* () {
+    const hubResult = yield loadFromHub(uri, loadChainFromFile, "chains", new Set(["json", "yaml"]));
+    if (hubResult) {
+        return hubResult;
+    }
+    return loadChainFromFile(uri);
+});
+//# sourceMappingURL=load.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../chains/load.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,GAAG,CAAC;AAC9B,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAE1C,MAAM,iBAAiB,GAAG,CAAO,IAAY,EAAE,EAAE;IAC/C,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC,CAAC;IACzC,OAAO,CAAC,GAAG,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC;IAC5B,OAAO,SAAS,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;AAC3C,CAAC,CAAA,CAAC;AAEF,MAAM,CAAC,MAAM,SAAS,GAAG,CAAO,GAAW,EAAsB,EAAE;IACjE,MAAM,SAAS,GAAG,MAAM,WAAW,CACjC,GAAG,EACH,iBAAiB,EACjB,QAAQ,EACR,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAC1B,CAAC;IACF,IAAI,SAAS,EAAE;QACb,OAAO,SAAS,CAAC;KAClB;IAED,OAAO,iBAAiB,CAAC,GAAG,CAAC,CAAC;AAChC,CAAC,CAAA,CAAC"}

package/dist/esm/chains/tests/llm_chain.test.d.ts

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

package/dist/esm/chains/tests/llm_chain.test.js

@@ -0,0 +1,30 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+import { test } from "@jest/globals";
+import { OpenAI } from "../../llms/openai";
+import { PromptTemplate } from "../../prompt";
+import { LLMChain } from "../llm_chain";
+import { loadChain } from "../load";
+test("Test OpenAI", () => __awaiter(void 0, void 0, void 0, function* () {
+    const model = new OpenAI({});
+    const prompt = new PromptTemplate({
+        template: "Print {foo}",
+        inputVariables: ["foo"],
+    });
+    const chain = new LLMChain({ prompt, llm: model });
+    const res = yield chain.call({ foo: "my favorite color" });
+    console.log({ res });
+}));
+test("Load chain from hub", () => __awaiter(void 0, void 0, void 0, function* () {
+    const chain = yield loadChain("lc://chains/hello-world/chain.json");
+    const res = yield chain.call({ topic: "my favorite color" });
+    console.log({ res });
+}));
+//# sourceMappingURL=llm_chain.test.js.map

package/dist/esm/chains/tests/llm_chain.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_chain.test.js","sourceRoot":"","sources":["../../../../chains/tests/llm_chain.test.ts"],"names":[],"mappings":";;;;;;;;;AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AACrC,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,OAAO,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAEpC,IAAI,CAAC,aAAa,EAAE,GAAS,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,cAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAA,CAAC,CAAC;AAEH,IAAI,CAAC,qBAAqB,EAAE,GAAS,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,oCAAoC,CAAC,CAAC;IACpE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAA,CAAC,CAAC"}

package/dist/esm/index.d.ts

@@ -0,0 +1,3 @@
+export { PromptTemplate, BasePromptTemplate, FewShotPromptTemplate, } from "./prompt";
+export { LLMChain } from "./chains";
+export { OpenAI } from "./llms";

package/dist/esm/index.js

@@ -0,0 +1,4 @@
+export { PromptTemplate, BasePromptTemplate, FewShotPromptTemplate, } from "./prompt";
+export { LLMChain } from "./chains";
+export { OpenAI } from "./llms";
+//# sourceMappingURL=index.js.map

package/dist/esm/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,cAAc,EACd,kBAAkB,EAClB,qBAAqB,GACtB,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,QAAQ,EAAE,MAAM,UAAU,CAAC;AACpC,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC"}

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

@@ -0,0 +1,23 @@
+import { LLMCallbackManager, LLMResult } from "./index";
+export type SerializedLLM = {
+    _type: string;
+} & Record<string, any>;
+export declare abstract class BaseLLM {
+    name: string;
+    cache?: boolean;
+    callbackManager: LLMCallbackManager;
+    verbose?: boolean;
+    constructor(callbackManager?: LLMCallbackManager, verbose?: boolean);
+    abstract _generate(prompts: string[], stop?: string[]): Promise<LLMResult>;
+    _generateUncached(prompts: string[], stop?: string[]): Promise<LLMResult>;
+    generate(prompts: string[], stop?: string[]): Promise<LLMResult>;
+    call(prompt: string, stop?: string[]): Promise<string>;
+    _identifyingParams(): Record<string, any>;
+    abstract _llmType(): string;
+    serialize(): SerializedLLM;
+    static deserialize(data: SerializedLLM): Promise<BaseLLM>;
+}
+export declare abstract class LLM extends BaseLLM {
+    abstract _call(prompt: string, stop?: string[]): Promise<string>;
+    _generate(prompts: string[], stop?: string[]): Promise<LLMResult>;
+}

package/dist/esm/llms/base.js

@@ -0,0 +1,154 @@
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __rest = (this && this.__rest) || function (s, e) {
+    var t = {};
+    for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
+        t[p] = s[p];
+    if (s != null && typeof Object.getOwnPropertySymbols === "function")
+        for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
+            if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
+                t[p[i]] = s[p[i]];
+        }
+    return t;
+};
+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();
+export class BaseLLM {
+    constructor(callbackManager, verbose) {
+        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
+        });
+        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();
+    }
+    _generateUncached(prompts, stop) {
+        return __awaiter(this, void 0, void 0, function* () {
+            this.callbackManager.handleStart({ name: this.name }, prompts, this.verbose);
+            let output;
+            try {
+                output = yield this._generate(prompts, stop);
+            }
+            catch (err) {
+                this.callbackManager.handleError(`${err}`, this.verbose);
+                throw err;
+            }
+            this.callbackManager.handleEnd(output, this.verbose);
+            return output;
+        });
+    }
+    generate(prompts, stop) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            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 = yield 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 };
+        });
+    }
+    call(prompt, stop) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { generations } = yield this.generate([prompt], stop);
+            return generations[0][0].text;
+        });
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _identifyingParams() {
+        return {};
+    }
+    serialize() {
+        return Object.assign(Object.assign({}, this._identifyingParams()), { _type: this._llmType() });
+    }
+    static deserialize(data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { _type } = data, rest = __rest(data, ["_type"]);
+            const Cls = {
+                openai: OpenAI,
+            }[_type];
+            if (Cls === undefined) {
+                throw new Error(`Cannot load  LLM with type ${_type}`);
+            }
+            return new Cls(rest);
+        });
+    }
+}
+export class LLM extends BaseLLM {
+    _generate(prompts, stop) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const generations = [];
+            for (let i = 0; i < prompts.length; i += 1) {
+                const text = yield this._call(prompts[i], stop);
+                generations.push([{ text }]);
+            }
+            return { generations };
+        });
+    }
+}
+//# sourceMappingURL=base.js.map

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

@@ -0,0 +1 @@
+{"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,MAAM,OAAgB,OAAO;IAS3B,YAAY,eAAoC,EAAE,OAAiB;QARnE;;;;;WAAa;QAEb;;;;;WAAgB;QAEhB;;;;;WAAoC;QAEpC;;;;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;IAIK,iBAAiB,CACrB,OAAiB,EACjB,IAAe;;YAEf,IAAI,CAAC,eAAe,CAAC,WAAW,CAC9B,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EACnB,OAAO,EACP,IAAI,CAAC,OAAO,CACb,CAAC;YACF,IAAI,MAAM,CAAC;YACX,IAAI;gBACF,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;aAC9C;YAAC,OAAO,GAAG,EAAE;gBACZ,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,GAAG,GAAG,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzD,MAAM,GAAG,CAAC;aACX;YAED,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YACrD,OAAO,MAAM,CAAC;QAChB,CAAC;KAAA;IAEK,QAAQ,CAAC,OAAiB,EAAE,IAAe;;;YAC/C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC3B,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;aACpE;YAED,IAAI,IAAI,CAAC,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,IAAI,EAAE;gBACzC,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;aACxD;YAED,IAAI,KAAK,KAAK,IAAI,IAAI,IAAI,CAAC,KAAK,KAAK,KAAK,EAAE;gBAC1C,OAAO,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;aAC9C;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,GAAG,IAAI,CAAC;YAEnB,MAAM,YAAY,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YACxD,MAAM,oBAAoB,GAAa,EAAE,CAAC;YAC1C,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;gBAChD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;gBAClD,IAAI,CAAC,MAAM,EAAE;oBACX,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;iBAClC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,CAAC;YAEH,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,oBAAoB,CAAC,MAAM,GAAG,CAAC,EAAE;gBACnC,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;gBACF,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE;oBAChD,MAAM,WAAW,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;oBAChD,WAAW,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC;oBACtC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;gBAC/D,CAAC,CAAC,CAAC;gBACH,SAAS,GAAG,MAAA,OAAO,CAAC,SAAS,mCAAI,EAAE,CAAC;aACrC;YAED,OAAO,EAAE,WAAW,EAAE,SAAS,EAAe,CAAC;;KAChD;IAEK,IAAI,CAAC,MAAc,EAAE,IAAe;;YACxC,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;YAC5D,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QAChC,CAAC;KAAA;IAED,8DAA8D;IAC9D,kBAAkB;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IAID,SAAS;QACP,uCACK,IAAI,CAAC,kBAAkB,EAAE,KAC5B,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE,IACtB;IACJ,CAAC;IAED,MAAM,CAAO,WAAW,CAAC,IAAmB;;YAC1C,MAAM,EAAE,KAAK,KAAc,IAAI,EAAb,IAAI,UAAK,IAAI,EAAzB,SAAkB,CAAO,CAAC;YAChC,MAAM,GAAG,GAAG;gBACV,MAAM,EAAE,MAAM;aACf,CAAC,KAAK,CAAC,CAAC;YACT,IAAI,GAAG,KAAK,SAAS,EAAE;gBACrB,MAAM,IAAI,KAAK,CAAC,8BAA8B,KAAK,EAAE,CAAC,CAAC;aACxD;YACD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC;KAAA;CAGF;AAED,MAAM,OAAgB,GAAI,SAAQ,OAAO;IAGjC,SAAS,CAAC,OAAiB,EAAE,IAAe;;YAChD,MAAM,WAAW,GAAG,EAAE,CAAC;YACvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;gBAC1C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;gBAChD,WAAW,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;aAC9B;YACD,OAAO,EAAE,WAAW,EAAE,CAAC;QACzB,CAAC;KAAA;CACF"}

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

@@ -0,0 +1,18 @@
+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;
+};
+export type Generation = {
+    text: string;
+    generationInfo?: Record<string, any>;
+};
+export type LLMResult = {
+    generations: Generation[][];
+    llmOutput?: Record<string, any>;
+};

package/dist/esm/llms/index.js

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

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

@@ -0,0 +1 @@
+{"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

@@ -0,0 +1,3 @@
+import { BaseLLM } from "./base";
+export declare const loadLLM: typeof BaseLLM.deserialize;
+export declare const loadLLMFromFile: (file: string) => Promise<BaseLLM>;

package/dist/esm/llms/load.js

@@ -0,0 +1,5 @@
+import { BaseLLM } from "./base";
+import { parseFileConfig } from "../util";
+export const loadLLM = BaseLLM.deserialize;
+export const loadLLMFromFile = (file) => loadLLM(parseFileConfig(file));
+//# sourceMappingURL=load.js.map

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

@@ -0,0 +1 @@
+{"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,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC,WAAW,CAAC;AAC3C,MAAM,CAAC,MAAM,eAAe,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,OAAO,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,63 @@
+import type { CreateCompletionRequest } from "openai";
+import { BaseLLM, LLMResult, LLMCallbackManager } from ".";
+interface ModelParams {
+    temperature: number;
+    maxTokens: number;
+    topP: number;
+    frequencyPenalty: number;
+    presencePenalty: number;
+    n: number;
+    bestOf: number;
+    logitBias?: Record<string, number>;
+}
+type Kwargs = Record<string, any>;
+export declare class OpenAI extends BaseLLM implements ModelParams {
+    temperature: number;
+    maxTokens: number;
+    topP: number;
+    frequencyPenalty: number;
+    presencePenalty: number;
+    n: number;
+    bestOf: number;
+    logitBias?: Record<string, number>;
+    modelName: string;
+    modelKwargs?: Kwargs;
+    batchSize: number;
+    maxRetries: number;
+    stop?: string[];
+    private client;
+    constructor(fields?: Partial<ModelParams> & {
+        callbackManager?: LLMCallbackManager;
+        verbose?: boolean;
+        modelName?: string;
+        modelKwargs?: Kwargs;
+        openAIApiKey?: string;
+        batchSize?: number;
+        maxRetries?: number;
+        stop?: string[];
+    });
+    invocationParams(): CreateCompletionRequest & Kwargs;
+    identifyingParams(): {
+        model: string;
+        prompt?: import("openai").CreateCompletionRequestPrompt | null | undefined;
+        suffix?: string | null | undefined;
+        max_tokens?: number | null | undefined;
+        temperature?: number | null | undefined;
+        top_p?: number | null | undefined;
+        n?: number | null | undefined;
+        stream?: boolean | null | undefined;
+        logprobs?: number | null | undefined;
+        echo?: boolean | null | undefined;
+        stop?: import("openai").CreateCompletionRequestStop | null | undefined;
+        presence_penalty?: number | null | undefined;
+        frequency_penalty?: number | null | undefined;
+        best_of?: number | null | undefined;
+        logit_bias?: object | null | undefined;
+        user?: string | undefined;
+        model_name: string;
+    };
+    _generate(prompts: string[], stop?: string[]): Promise<LLMResult>;
+    completionWithRetry(request: CreateCompletionRequest): Promise<import("axios").AxiosResponse<import("openai").CreateCompletionResponse, any>>;
+    _llmType(): string;
+}
+export {};