langchain 0.0.4

package/dist/cjs/llms/base.js

@@ -0,0 +1,159 @@
+"use strict";
+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;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LLM = exports.BaseLLM = void 0;
+const index_1 = require("./index");
+const cache_1 = require("../cache");
+const getCallbackManager = () => ({
+    handleStart: (..._args) => {
+        // console.log(args);
+    },
+    handleEnd: (..._args) => {
+        // console.log(args);
+    },
+    handleError: (..._args) => {
+        // console.log(args);
+    },
+});
+const getVerbosity = () => true;
+const cache = new cache_1.InMemoryCache();
+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: index_1.OpenAI,
+            }[_type];
+            if (Cls === undefined) {
+                throw new Error(`Cannot load  LLM with type ${_type}`);
+            }
+            return new Cls(rest);
+        });
+    }
+}
+exports.BaseLLM = BaseLLM;
+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 };
+        });
+    }
+}
+exports.LLM = LLM;
+//# sourceMappingURL=base.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../llms/base.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AAAA,mCAAgE;AAChE,oCAAoD;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,qBAAa,EAAE,CAAC;AAO7C,MAAsB,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,cAAM;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;AA/GD,0BA+GC;AAED,MAAsB,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;AAXD,kBAWC"}

package/dist/cjs/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/cjs/llms/index.js

@@ -0,0 +1,11 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadLLM = exports.OpenAI = exports.LLM = exports.BaseLLM = void 0;
+var base_1 = require("./base");
+Object.defineProperty(exports, "BaseLLM", { enumerable: true, get: function () { return base_1.BaseLLM; } });
+Object.defineProperty(exports, "LLM", { enumerable: true, get: function () { return base_1.LLM; } });
+var openai_1 = require("./openai");
+Object.defineProperty(exports, "OpenAI", { enumerable: true, get: function () { return openai_1.OpenAI; } });
+var load_1 = require("./load");
+Object.defineProperty(exports, "loadLLM", { enumerable: true, get: function () { return load_1.loadLLM; } });
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../llms/index.ts"],"names":[],"mappings":";;;AAAA,+BAAqD;AAA5C,+FAAA,OAAO,OAAA;AAAE,2FAAA,GAAG,OAAA;AACrB,mCAAkC;AAAzB,gGAAA,MAAM,OAAA;AACf,+BAAiC;AAAxB,+FAAA,OAAO,OAAA"}

package/dist/cjs/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/cjs/llms/load.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.loadLLMFromFile = exports.loadLLM = void 0;
+const base_1 = require("./base");
+const util_1 = require("../util");
+exports.loadLLM = base_1.BaseLLM.deserialize;
+const loadLLMFromFile = (file) => (0, exports.loadLLM)((0, util_1.parseFileConfig)(file));
+exports.loadLLMFromFile = loadLLMFromFile;
+//# sourceMappingURL=load.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"load.js","sourceRoot":"","sources":["../../../llms/load.ts"],"names":[],"mappings":";;;AAAA,iCAAiC;AACjC,kCAA0C;AAE7B,QAAA,OAAO,GAAG,cAAO,CAAC,WAAW,CAAC;AACpC,MAAM,eAAe,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,IAAA,eAAO,EAAC,IAAA,sBAAe,EAAC,IAAI,CAAC,CAAC,CAAC;AAAnE,QAAA,eAAe,mBAAoD"}

package/dist/cjs/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 {};

package/dist/cjs/llms/openai.js

@@ -0,0 +1,202 @@
+"use strict";
+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());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAI = void 0;
+const exponential_backoff_1 = require("exponential-backoff");
+const _1 = require(".");
+let Configuration = null;
+let OpenAIApi = null;
+try {
+    // eslint-disable-next-line global-require
+    ({ Configuration, OpenAIApi } = require("openai"));
+}
+catch (_a) {
+    // ignore error
+}
+const chunkArray = (arr, chunkSize) => arr.reduce((chunks, elem, index) => {
+    const chunkIndex = Math.floor(index / chunkSize);
+    const chunk = chunks[chunkIndex] || [];
+    // eslint-disable-next-line no-param-reassign
+    chunks[chunkIndex] = chunk.concat([elem]);
+    return chunks;
+}, []);
+class OpenAI extends _1.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 install -S 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);
+    }
+    invocationParams() {
+        return Object.assign({ 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);
+    }
+    identifyingParams() {
+        return Object.assign({ model_name: this.modelName }, this.invocationParams());
+    }
+    _generate(prompts, stop) {
+        var _a, _b, _c, _d;
+        return __awaiter(this, void 0, void 0, function* () {
+            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 } = yield this.completionWithRetry(Object.assign(Object.assign({}, 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 },
+            };
+        });
+    }
+    completionWithRetry(request) {
+        const makeCompletionRequest = () => this.client.createCompletion(request);
+        return (0, exponential_backoff_1.backOff)(makeCompletionRequest, {
+            startingDelay: 4,
+            maxDelay: 10,
+            numOfAttempts: this.maxRetries,
+            // TODO(sean) pass custom retry function to check error types.
+        });
+    }
+    _llmType() {
+        return "openai";
+    }
+}
+exports.OpenAI = OpenAI;
+//# sourceMappingURL=openai.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../llms/openai.ts"],"names":[],"mappings":";;;;;;;;;;;;AAOA,6DAA8C;AAC9C,wBAA2D;AAE3D,IAAI,aAAa,GAAiC,IAAI,CAAC;AACvD,IAAI,SAAS,GAA6B,IAAI,CAAC;AAE/C,IAAI;IACF,0CAA0C;IAC1C,CAAC,EAAE,aAAa,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;CACpD;AAAC,WAAM;IACN,eAAe;CAChB;AAsBD,MAAM,UAAU,GAAG,CAAI,GAAQ,EAAE,SAAiB,EAAE,EAAE,CACpD,GAAG,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE;IACjC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC;IACjD,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;IACvC,6CAA6C;IAC7C,MAAM,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC1C,OAAO,MAAM,CAAC;AAChB,CAAC,EAAE,EAAW,CAAC,CAAC;AAElB,MAAa,MAAO,SAAQ,UAAO;IA6BjC,YACE,MASC;;QAED,KAAK,CAAC,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,eAAe,EAAE,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,OAAO,CAAC,CAAC;QAxClD;;;;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;QAezB,IAAI,aAAa,KAAK,IAAI,IAAI,SAAS,KAAK,IAAI,EAAE;YAChD,MAAM,IAAI,KAAK,CACb,0EAA0E,CAC3E,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,gBAAgB;QACd,uBACE,KAAK,EAAE,IAAI,CAAC,SAAS,EACrB,WAAW,EAAE,IAAI,CAAC,WAAW,EAC7B,UAAU,EAAE,IAAI,CAAC,SAAS,EAC1B,KAAK,EAAE,IAAI,CAAC,IAAI,EAChB,iBAAiB,EAAE,IAAI,CAAC,gBAAgB,EACxC,gBAAgB,EAAE,IAAI,CAAC,eAAe,EACtC,CAAC,EAAE,IAAI,CAAC,CAAC,EACT,OAAO,EAAE,IAAI,CAAC,MAAM,EACpB,UAAU,EAAE,IAAI,CAAC,SAAS,EAC1B,IAAI,EAAE,IAAI,CAAC,IAAI,IACZ,IAAI,CAAC,WAAW,EACnB;IACJ,CAAC;IAED,iBAAiB;QACf,uBACE,UAAU,EAAE,IAAI,CAAC,SAAS,IACvB,IAAI,CAAC,gBAAgB,EAAE,EAC1B;IACJ,CAAC;IAEK,SAAS,CAAC,OAAiB,EAAE,IAAe;;;YAChD,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;YACvD,MAAM,OAAO,GAA2C,EAAE,CAAC;YAC3D,MAAM,UAAU,GAAe,EAAE,CAAC;YAElC,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,EAAE;gBACrB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;aAC3D;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACvC,MAAM,CAAC,IAAI,GAAG,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,MAAM,CAAC,IAAI,CAAC;YAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE;gBAC7C,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,IAAI,CAAC,mBAAmB,iCAC1C,MAAM,KACT,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,IACrB,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC9B,MAAM,EACJ,iBAAiB,EAAE,gBAAgB,EACnC,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,WAAW,GAC1B,GAAG,MAAA,IAAI,CAAC,KAAK,mCAAI,EAAE,CAAC;gBAErB,IAAI,gBAAgB,EAAE;oBACpB,UAAU,CAAC,gBAAgB;wBACzB,CAAC,MAAA,UAAU,CAAC,gBAAgB,mCAAI,CAAC,CAAC,GAAG,gBAAgB,CAAC;iBACzD;gBAED,IAAI,YAAY,EAAE;oBAChB,UAAU,CAAC,YAAY,GAAG,CAAC,MAAA,UAAU,CAAC,YAAY,mCAAI,CAAC,CAAC,GAAG,YAAY,CAAC;iBACzE;gBAED,IAAI,WAAW,EAAE;oBACf,UAAU,CAAC,WAAW,GAAG,CAAC,MAAA,UAAU,CAAC,WAAW,mCAAI,CAAC,CAAC,GAAG,WAAW,CAAC;iBACtE;aACF;YAED,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;;gBAAC,OAAA,CAAC;oBAC7B,IAAI,EAAE,MAAA,MAAM,CAAC,IAAI,mCAAI,EAAE;oBACvB,cAAc,EAAE;wBACd,YAAY,EAAE,MAAM,CAAC,aAAa;wBAClC,QAAQ,EAAE,MAAM,CAAC,QAAQ;qBAC1B;iBACF,CAAC,CAAA;aAAA,CAAC,CACJ,CAAC;YACF,OAAO;gBACL,WAAW;gBACX,SAAS,EAAE,EAAE,UAAU,EAAE;aAC1B,CAAC;;KACH;IAED,mBAAmB,CAAC,OAAgC;QAClD,MAAM,qBAAqB,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC1E,OAAO,IAAA,6BAAO,EAAC,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;AA9JD,wBA8JC"}

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

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

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

@@ -0,0 +1,19 @@
+"use strict";
+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());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const openai_1 = require("../openai");
+(0, globals_1.test)("Test OpenAI", () => __awaiter(void 0, void 0, void 0, function* () {
+    const model = new openai_1.OpenAI({ maxTokens: 5 });
+    const res = yield model.call("Print hello world");
+    console.log({ res });
+}));
+//# sourceMappingURL=openai.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai.test.js","sourceRoot":"","sources":["../../../../llms/tests/openai.test.ts"],"names":[],"mappings":";;;;;;;;;;;AAAA,2CAAqC;AACrC,sCAAmC;AAEnC,IAAA,cAAI,EAAC,aAAa,EAAE,GAAS,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,eAAM,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,CAAA,CAAC,CAAC"}

package/dist/cjs/prompt/base.d.ts

@@ -0,0 +1,19 @@
+import { BaseOutputParser } from "./parser";
+import { PromptTemplate, FewShotPromptTemplate } from "./index";
+declare const templateClasses: (typeof PromptTemplate | typeof FewShotPromptTemplate)[];
+export type SerializedBasePromptTemplate = ReturnType<InstanceType<(typeof templateClasses)[number]>["serialize"]>;
+export type InputValues = Record<string, any>;
+export interface BasePromptTemplateInput {
+    inputVariables: string[];
+    outputParser?: BaseOutputParser;
+}
+export declare abstract class BasePromptTemplate implements BasePromptTemplateInput {
+    inputVariables: string[];
+    outputParser?: BaseOutputParser;
+    constructor(input: BasePromptTemplateInput);
+    abstract format(values: InputValues): string;
+    abstract _getPromptType(): string;
+    abstract serialize(): SerializedBasePromptTemplate;
+    static deserialize(data: SerializedBasePromptTemplate): Promise<BasePromptTemplate>;
+}
+export {};

package/dist/cjs/prompt/base.js

@@ -0,0 +1,54 @@
+"use strict";
+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());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BasePromptTemplate = void 0;
+const index_1 = require("./index");
+const templateClasses = [index_1.PromptTemplate, index_1.FewShotPromptTemplate];
+class BasePromptTemplate {
+    constructor(input) {
+        Object.defineProperty(this, "inputVariables", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "outputParser", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const { inputVariables } = input;
+        if (inputVariables.includes("stop")) {
+            throw new Error("Cannot have an input variable named 'stop', as it is used internally, please rename.");
+        }
+        Object.assign(this, input);
+    }
+    // Deserializing needs to be async because templates (e.g. few_shot) can
+    // reference remote resources that we read asynchronously with a web
+    // request.
+    static deserialize(data) {
+        return __awaiter(this, void 0, void 0, function* () {
+            switch (data._type) {
+                case "prompt":
+                    return index_1.PromptTemplate.deserialize(data);
+                case undefined:
+                    return index_1.PromptTemplate.deserialize(Object.assign(Object.assign({}, data), { _type: "prompt" }));
+                case "few_shot":
+                    return index_1.FewShotPromptTemplate.deserialize(data);
+                default:
+                    throw new Error(`Invalid prompt type in config: ${data._type}`);
+            }
+        });
+    }
+}
+exports.BasePromptTemplate = BasePromptTemplate;
+//# sourceMappingURL=base.js.map

package/dist/cjs/prompt/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../prompt/base.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,mCAAgE;AAEhE,MAAM,eAAe,GAAG,CAAC,sBAAc,EAAE,6BAAqB,CAAC,CAAC;AAchE,MAAsB,kBAAkB;IAKtC,YAAY,KAA8B;QAJ1C;;;;;WAAyB;QAEzB;;;;;WAAgC;QAG9B,MAAM,EAAE,cAAc,EAAE,GAAG,KAAK,CAAC;QACjC,IAAI,cAAc,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;YACnC,MAAM,IAAI,KAAK,CACb,sFAAsF,CACvF,CAAC;SACH;QACD,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC7B,CAAC;IAQD,wEAAwE;IACxE,oEAAoE;IACpE,WAAW;IACX,MAAM,CAAO,WAAW,CACtB,IAAkC;;YAElC,QAAQ,IAAI,CAAC,KAAK,EAAE;gBAClB,KAAK,QAAQ;oBACX,OAAO,sBAAc,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBAC1C,KAAK,SAAS;oBACZ,OAAO,sBAAc,CAAC,WAAW,iCAAM,IAAI,KAAE,KAAK,EAAE,QAAQ,IAAG,CAAC;gBAClE,KAAK,UAAU;oBACb,OAAO,6BAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBACjD;oBACE,MAAM,IAAI,KAAK,CACb,kCACG,IAAqC,CAAC,KACzC,EAAE,CACH,CAAC;aACL;QACH,CAAC;KAAA;CACF;AA1CD,gDA0CC"}

package/dist/cjs/prompt/few_shot.d.ts

@@ -0,0 +1,47 @@
+import { BasePromptTemplate, InputValues, BasePromptTemplateInput } from "./index";
+import { TemplateFormat } from "./template";
+import { PromptTemplate, SerializedPromptTemplate } from "./prompt";
+import { SerializedOutputParser } from "./parser";
+type ExampleSelector = null;
+type Example = Record<string, any>;
+export type SerializedFewShotTemplate = {
+    _type: "few_shot";
+    input_variables: string[];
+    output_parser?: SerializedOutputParser;
+    examples: string | Example[];
+    example_prompt?: SerializedPromptTemplate;
+    example_prompt_path?: string;
+    example_separator: string;
+    prefix?: string;
+    prefix_path?: string;
+    suffix?: string;
+    suffix_path?: string;
+    template_format: TemplateFormat;
+};
+export interface FewShotPromptTemplateInput extends BasePromptTemplateInput {
+    examples?: Example[];
+    examplePrompt: PromptTemplate;
+    exampleSelector?: ExampleSelector;
+    exampleSeparator: string;
+    prefix: string;
+    suffix: string;
+    templateFormat: TemplateFormat;
+    validateTemplate?: boolean;
+}
+export declare class FewShotPromptTemplate extends BasePromptTemplate implements FewShotPromptTemplateInput {
+    examples?: InputValues[];
+    exampleSelector?: ExampleSelector;
+    examplePrompt: PromptTemplate;
+    suffix: string;
+    exampleSeparator: string;
+    prefix: string;
+    templateFormat: TemplateFormat;
+    validateTemplate: boolean;
+    constructor(input: FewShotPromptTemplateInput);
+    _getPromptType(): "few_shot";
+    private getExamples;
+    format(values: InputValues): string;
+    serialize(): SerializedFewShotTemplate;
+    static deserialize(data: SerializedFewShotTemplate): Promise<FewShotPromptTemplate>;
+}
+export {};