modelfusion 0.0.44

package/model-provider/stability/StabilityImageGenerationModel.d.ts

@@ -0,0 +1,95 @@
+import { z } from "zod";
+import { AbstractModel } from "../../model-function/AbstractModel.js";
+import { FunctionOptions } from "../../model-function/FunctionOptions.js";
+import { ImageGenerationModel, ImageGenerationModelSettings } from "../../model-function/generate-image/ImageGenerationModel.js";
+import { RetryFunction } from "../../util/api/RetryFunction.js";
+import { ThrottleFunction } from "../../util/api/ThrottleFunction.js";
+/**
+ * Create an image generation model that calls the Stability AI image generation API.
+ *
+ * @see https://api.stability.ai/docs#tag/v1generation/operation/textToImage
+ *
+ * @example
+ * const { image } = await generateImage(
+ *   new StabilityImageGenerationModel({
+ *     model: "stable-diffusion-512-v2-1",
+ *     cfgScale: 7,
+ *     clipGuidancePreset: "FAST_BLUE",
+ *     height: 512,
+ *     width: 512,
+ *     samples: 1,
+ *     steps: 30,
+ *   })
+ *   [
+ *     { text: "the wicked witch of the west" },
+ *     { text: "style of early 19th century painting", weight: 0.5 },
+ *   ]
+ * );
+ */
+export declare class StabilityImageGenerationModel extends AbstractModel<StabilityImageGenerationModelSettings> implements ImageGenerationModel<StabilityImageGenerationPrompt, StabilityImageGenerationResponse, StabilityImageGenerationModelSettings> {
+    constructor(settings: StabilityImageGenerationModelSettings);
+    readonly provider: "stability";
+    get modelName(): string;
+    private get apiKey();
+    callAPI(input: StabilityImageGenerationPrompt, options?: FunctionOptions<StabilityImageGenerationModelSettings>): Promise<StabilityImageGenerationResponse>;
+    generateImageResponse(prompt: StabilityImageGenerationPrompt, options?: FunctionOptions<StabilityImageGenerationModelSettings>): Promise<{
+        artifacts: {
+            seed: number;
+            base64: string;
+            finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
+        }[];
+    }>;
+    extractBase64Image(response: StabilityImageGenerationResponse): string;
+    withSettings(additionalSettings: StabilityImageGenerationModelSettings): this;
+}
+export interface StabilityImageGenerationModelSettings extends ImageGenerationModelSettings {
+    model: string;
+    baseUrl?: string;
+    apiKey?: string;
+    retry?: RetryFunction;
+    throttle?: ThrottleFunction;
+    height?: number;
+    width?: number;
+    cfgScale?: number;
+    clipGuidancePreset?: string;
+    sampler?: StabilityImageGenerationSampler;
+    samples?: number;
+    seed?: number;
+    steps?: number;
+    stylePreset?: StabilityImageGenerationStylePreset;
+}
+declare const stabilityImageGenerationResponseSchema: z.ZodObject<{
+    artifacts: z.ZodArray<z.ZodObject<{
+        base64: z.ZodString;
+        seed: z.ZodNumber;
+        finishReason: z.ZodEnum<["SUCCESS", "ERROR", "CONTENT_FILTERED"]>;
+    }, "strip", z.ZodTypeAny, {
+        seed: number;
+        base64: string;
+        finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
+    }, {
+        seed: number;
+        base64: string;
+        finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
+    }>, "many">;
+}, "strip", z.ZodTypeAny, {
+    artifacts: {
+        seed: number;
+        base64: string;
+        finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
+    }[];
+}, {
+    artifacts: {
+        seed: number;
+        base64: string;
+        finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
+    }[];
+}>;
+export type StabilityImageGenerationResponse = z.infer<typeof stabilityImageGenerationResponseSchema>;
+export type StabilityImageGenerationStylePreset = "enhance" | "anime" | "photographic" | "digital-art" | "comic-book" | "fantasy-art" | "line-art" | "analog-film" | "neon-punk" | "isometric" | "low-poly" | "origami" | "modeling-compound" | "cinematic" | "3d-model" | "pixel-art" | "tile-texture";
+export type StabilityImageGenerationSampler = "DDIM" | "DDPM" | "K_DPMPP_2M" | "K_DPMPP_2S_ANCESTRAL" | "K_DPM_2" | "K_DPM_2_ANCESTRAL" | "K_EULER" | "K_EULER_ANCESTRAL" | "K_HEUN" | "K_LMS";
+export type StabilityImageGenerationPrompt = Array<{
+    text: string;
+    weight?: number;
+}>;
+export {};

package/model-provider/stability/StabilityImageGenerationModel.js

@@ -0,0 +1,129 @@
+import { z } from "zod";
+import { AbstractModel } from "../../model-function/AbstractModel.js";
+import { callWithRetryAndThrottle } from "../../util/api/callWithRetryAndThrottle.js";
+import { createJsonResponseHandler, postJsonToApi, } from "../../util/api/postToApi.js";
+import { failedStabilityCallResponseHandler } from "./StabilityError.js";
+/**
+ * Create an image generation model that calls the Stability AI image generation API.
+ *
+ * @see https://api.stability.ai/docs#tag/v1generation/operation/textToImage
+ *
+ * @example
+ * const { image } = await generateImage(
+ *   new StabilityImageGenerationModel({
+ *     model: "stable-diffusion-512-v2-1",
+ *     cfgScale: 7,
+ *     clipGuidancePreset: "FAST_BLUE",
+ *     height: 512,
+ *     width: 512,
+ *     samples: 1,
+ *     steps: 30,
+ *   })
+ *   [
+ *     { text: "the wicked witch of the west" },
+ *     { text: "style of early 19th century painting", weight: 0.5 },
+ *   ]
+ * );
+ */
+export class StabilityImageGenerationModel extends AbstractModel {
+    constructor(settings) {
+        super({ settings });
+        Object.defineProperty(this, "provider", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "stability"
+        });
+    }
+    get modelName() {
+        return this.settings.model;
+    }
+    get apiKey() {
+        const apiKey = this.settings.apiKey ?? process.env.STABILITY_API_KEY;
+        if (apiKey == null) {
+            throw new Error("No API key provided. Either pass an API key to the constructor or set the STABILITY_API_KEY environment variable.");
+        }
+        return apiKey;
+    }
+    async callAPI(input, options) {
+        const run = options?.run;
+        const settings = options?.settings;
+        const callSettings = Object.assign({
+            apiKey: this.apiKey,
+        }, this.settings, settings, {
+            abortSignal: run?.abortSignal,
+            engineId: this.settings.model,
+            textPrompts: input,
+        });
+        return callWithRetryAndThrottle({
+            retry: this.settings.retry,
+            throttle: this.settings.throttle,
+            call: async () => callStabilityImageGenerationAPI(callSettings),
+        });
+    }
+    generateImageResponse(prompt, options) {
+        return this.callAPI(prompt, options);
+    }
+    extractBase64Image(response) {
+        return response.artifacts[0].base64;
+    }
+    withSettings(additionalSettings) {
+        return new StabilityImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
+    }
+}
+const stabilityImageGenerationResponseSchema = z.object({
+    artifacts: z.array(z.object({
+        base64: z.string(),
+        seed: z.number(),
+        finishReason: z.enum(["SUCCESS", "ERROR", "CONTENT_FILTERED"]),
+    })),
+});
+/**
+ * Call the Stability AI API for image generation.
+ *
+ * @see https://api.stability.ai/docs#tag/v1generation/operation/textToImage
+ *
+ * @example
+ * const imageResponse = await callStabilityImageGenerationAPI({
+ *   apiKey: STABILITY_API_KEY,
+ *   engineId: "stable-diffusion-512-v2-1",
+ *   textPrompts: [
+ *     { text: "the wicked witch of the west" },
+ *     { text: "style of early 19th century painting", weight: 0.5 },
+ *   ],
+ *   cfgScale: 7,
+ *   clipGuidancePreset: "FAST_BLUE",
+ *   height: 512,
+ *   width: 512,
+ *   samples: 1,
+ *   steps: 30,
+ * });
+ *
+ * imageResponse.artifacts.forEach((image, index) => {
+ *   fs.writeFileSync(
+ *     `./stability-image-example-${index}.png`,
+ *     Buffer.from(image.base64, "base64")
+ *   );
+ * });
+ */
+async function callStabilityImageGenerationAPI({ baseUrl = "https://api.stability.ai/v1", abortSignal, apiKey, engineId, height, width, textPrompts, cfgScale, clipGuidancePreset, sampler, samples, seed, steps, stylePreset, }) {
+    return postJsonToApi({
+        url: `${baseUrl}/generation/${engineId}/text-to-image`,
+        apiKey,
+        body: {
+            height,
+            width,
+            text_prompts: textPrompts,
+            cfg_scale: cfgScale,
+            clip_guidance_preset: clipGuidancePreset,
+            sampler,
+            samples,
+            seed,
+            steps,
+            style_preset: stylePreset,
+        },
+        failedResponseHandler: failedStabilityCallResponseHandler,
+        successfulResponseHandler: createJsonResponseHandler(stabilityImageGenerationResponseSchema),
+        abortSignal,
+    });
+}

package/model-provider/stability/index.cjs

@@ -0,0 +1,20 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StabilityError = void 0;
+var StabilityError_js_1 = require("./StabilityError.cjs");
+Object.defineProperty(exports, "StabilityError", { enumerable: true, get: function () { return StabilityError_js_1.StabilityError; } });
+__exportStar(require("./StabilityImageGenerationModel.cjs"), exports);

package/model-provider/stability/index.d.ts

@@ -0,0 +1,2 @@
+export { StabilityError, StabilityErrorData } from "./StabilityError.js";
+export * from "./StabilityImageGenerationModel.js";

package/model-provider/stability/index.js

@@ -0,0 +1,2 @@
+export { StabilityError } from "./StabilityError.js";
+export * from "./StabilityImageGenerationModel.js";

package/package.json

@@ -0,0 +1,87 @@
+{
+  "name": "modelfusion",
+  "description": "Build AI applications, chatbots, and agents with JavaScript and TypeScript.",
+  "version": "0.0.44",
+  "author": "Lars Grammel",
+  "license": "MIT",
+  "keywords": [
+    "llm",
+    "embedding",
+    "openai",
+    "cohere",
+    "huggingface",
+    "stability-ai",
+    "gpt-3",
+    "gpt-4",
+    "whisper",
+    "chatbot",
+    "ai"
+  ],
+  "homepage": "https://ai-utils.dev/",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/lgrammel/ai-utils.js"
+  },
+  "bugs": {
+    "url": "https://github.com/lgrammel/ai-utils.js/issues"
+  },
+  "type": "module",
+  "sideEffects": false,
+  "private": false,
+  "engines": {
+    "node": ">=18"
+  },
+  "files": [
+    "./*"
+  ],
+  "exports": {
+    ".": {
+      "types": "./index.d.ts",
+      "import": "./index.js",
+      "require": "./index.cjs"
+    }
+  },
+  "scripts": {
+    "setup": "husky install",
+    "lint": "eslint --ext .ts src",
+    "clean": "rimraf build dist",
+    "build": "npm run build:esm && npm run build:cjs",
+    "build:esm": "tsc --outDir dist/",
+    "build:cjs": "tsc --outDir build/cjs/ -p tsconfig.cjs.json && node bin/prepare-cjs.js",
+    "dist:copy-files": "copyfiles package.json README.md LICENSE dist",
+    "dist": "npm run clean && npm run lint && npm run build && npm run dist:copy-files"
+  },
+  "dependencies": {
+    "eventsource-parser": "1.0.0",
+    "js-tiktoken": "1.0.7",
+    "nanoid": "3.3.6",
+    "secure-json-parse": "2.7.0"
+  },
+  "devDependencies": {
+    "@pinecone-database/pinecone": "^0.1.6",
+    "@tsconfig/recommended": "1.0.2",
+    "@types/node": "18.11.9",
+    "@typescript-eslint/eslint-plugin": "^6.1.0",
+    "@typescript-eslint/parser": "^6.1.0",
+    "copyfiles": "2.4.1",
+    "eslint": "^8.45.0",
+    "eslint-config-prettier": "8.9.0",
+    "husky": "^8.0.3",
+    "lint-staged": "13.2.3",
+    "prettier": "3.0.0",
+    "rimraf": "5.0.1",
+    "typescript": "5.1.6",
+    "zod": "3.21.4",
+    "zod-to-json-schema": "3.21.4"
+  },
+  "peerDependencies": {
+    "@pinecone-database/pinecone": "*",
+    "zod": "^3",
+    "zod-to-json-schema": "^3"
+  },
+  "peerDependenciesMeta": {
+    "@pinecone-database/pinecone": {
+      "optional": true
+    }
+  }
+}

package/prompt/InstructionPrompt.cjs

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/prompt/InstructionPrompt.d.ts

@@ -0,0 +1,7 @@
+/**
+ * A single instruction prompt. It can contain an optional system message to provide context for the language model.
+ */
+export type InstructionPrompt = {
+    system?: string;
+    instruction: string;
+};

package/prompt/InstructionPrompt.js

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

package/prompt/Llama2PromptMapping.cjs

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatToLlama2PromptMapping = exports.InstructionToLlama2PromptMapping = void 0;
+const validateChatPrompt_js_1 = require("./chat/validateChatPrompt.cjs");
+// see https://github.com/facebookresearch/llama/blob/6c7fe276574e78057f917549435a2554000a876d/llama/generation.py#L44
+const BEGIN_SEGMENT = "<s>";
+const END_SEGMENT = "</s>\n";
+const BEGIN_INSTRUCTION = "[INST]";
+const END_INSTRUCTION = "[/INST]\n";
+const BEGIN_SYSTEM = "<<SYS>>\n";
+const END_SYSTEM = "\n<</SYS>>\n\n";
+/**
+ * Maps an instruction prompt to the Llama2 prompt format.
+ *
+ * @see https://www.philschmid.de/llama-2#how-to-prompt-llama-2-chat
+ */
+const InstructionToLlama2PromptMapping = () => ({
+    stopTokens: [END_SEGMENT],
+    map: (instruction) => `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${instruction.system != null
+        ? ` ${BEGIN_SYSTEM}${instruction.system}${END_SYSTEM}`
+        : ""} ${instruction.instruction} ${END_INSTRUCTION}\n`,
+});
+exports.InstructionToLlama2PromptMapping = InstructionToLlama2PromptMapping;
+const ChatToLlama2PromptMapping = () => ({
+    map: (chatPrompt) => {
+        (0, validateChatPrompt_js_1.validateChatPrompt)(chatPrompt);
+        let text = "";
+        for (let i = 0; i < chatPrompt.length; i++) {
+            const message = chatPrompt[i];
+            // system message:
+            if (i === 0 &&
+                "system" in message &&
+                typeof message.system === "string") {
+                // Separate section for system message to simplify implementation
+                // (this is slightly different from the original instructions):
+                text += `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${BEGIN_SYSTEM}${message.system}${END_SYSTEM}${END_INSTRUCTION}${END_SEGMENT}`;
+                continue;
+            }
+            // user message
+            if ("user" in message) {
+                text += `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${message.user}${END_INSTRUCTION}`;
+                continue;
+            }
+            // ai message:
+            if ("ai" in message) {
+                text += `${message.ai}${END_SEGMENT}`;
+                continue;
+            }
+            // unsupported message:
+            throw new Error(`Unsupported message: ${JSON.stringify(message)}`);
+        }
+        return text;
+    },
+    stopTokens: [END_SEGMENT],
+});
+exports.ChatToLlama2PromptMapping = ChatToLlama2PromptMapping;

package/prompt/Llama2PromptMapping.d.ts

@@ -0,0 +1,10 @@
+import { PromptMapping } from "./PromptMapping.js";
+import { InstructionPrompt } from "./InstructionPrompt.js";
+import { ChatPrompt } from "./chat/ChatPrompt.js";
+/**
+ * Maps an instruction prompt to the Llama2 prompt format.
+ *
+ * @see https://www.philschmid.de/llama-2#how-to-prompt-llama-2-chat
+ */
+export declare const InstructionToLlama2PromptMapping: () => PromptMapping<InstructionPrompt, string>;
+export declare const ChatToLlama2PromptMapping: () => PromptMapping<ChatPrompt, string>;

package/prompt/Llama2PromptMapping.js

@@ -0,0 +1,51 @@
+import { validateChatPrompt } from "./chat/validateChatPrompt.js";
+// see https://github.com/facebookresearch/llama/blob/6c7fe276574e78057f917549435a2554000a876d/llama/generation.py#L44
+const BEGIN_SEGMENT = "<s>";
+const END_SEGMENT = "</s>\n";
+const BEGIN_INSTRUCTION = "[INST]";
+const END_INSTRUCTION = "[/INST]\n";
+const BEGIN_SYSTEM = "<<SYS>>\n";
+const END_SYSTEM = "\n<</SYS>>\n\n";
+/**
+ * Maps an instruction prompt to the Llama2 prompt format.
+ *
+ * @see https://www.philschmid.de/llama-2#how-to-prompt-llama-2-chat
+ */
+export const InstructionToLlama2PromptMapping = () => ({
+    stopTokens: [END_SEGMENT],
+    map: (instruction) => `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${instruction.system != null
+        ? ` ${BEGIN_SYSTEM}${instruction.system}${END_SYSTEM}`
+        : ""} ${instruction.instruction} ${END_INSTRUCTION}\n`,
+});
+export const ChatToLlama2PromptMapping = () => ({
+    map: (chatPrompt) => {
+        validateChatPrompt(chatPrompt);
+        let text = "";
+        for (let i = 0; i < chatPrompt.length; i++) {
+            const message = chatPrompt[i];
+            // system message:
+            if (i === 0 &&
+                "system" in message &&
+                typeof message.system === "string") {
+                // Separate section for system message to simplify implementation
+                // (this is slightly different from the original instructions):
+                text += `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${BEGIN_SYSTEM}${message.system}${END_SYSTEM}${END_INSTRUCTION}${END_SEGMENT}`;
+                continue;
+            }
+            // user message
+            if ("user" in message) {
+                text += `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${message.user}${END_INSTRUCTION}`;
+                continue;
+            }
+            // ai message:
+            if ("ai" in message) {
+                text += `${message.ai}${END_SEGMENT}`;
+                continue;
+            }
+            // unsupported message:
+            throw new Error(`Unsupported message: ${JSON.stringify(message)}`);
+        }
+        return text;
+    },
+    stopTokens: [END_SEGMENT],
+});

package/prompt/OpenAIChatPromptMapping.cjs

@@ -0,0 +1,62 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatToOpenAIChatPromptMapping = exports.InstructionToOpenAIChatPromptMapping = void 0;
+const validateChatPrompt_js_1 = require("./chat/validateChatPrompt.cjs");
+const InstructionToOpenAIChatPromptMapping = () => ({
+    map: (instruction) => {
+        const messages = [];
+        if (instruction.system != null) {
+            messages.push({
+                role: "system",
+                content: instruction.system,
+            });
+        }
+        messages.push({
+            role: "user",
+            content: instruction.instruction,
+        });
+        return messages;
+    },
+    stopTokens: [],
+});
+exports.InstructionToOpenAIChatPromptMapping = InstructionToOpenAIChatPromptMapping;
+const ChatToOpenAIChatPromptMapping = () => ({
+    map: (chatPrompt) => {
+        (0, validateChatPrompt_js_1.validateChatPrompt)(chatPrompt);
+        const messages = [];
+        for (let i = 0; i < chatPrompt.length; i++) {
+            const message = chatPrompt[i];
+            // system message:
+            if (i === 0 &&
+                "system" in message &&
+                typeof message.system === "string") {
+                messages.push({
+                    role: "system",
+                    content: message.system,
+                });
+                continue;
+            }
+            // user message
+            if ("user" in message) {
+                messages.push({
+                    role: "user",
+                    content: message.user,
+                });
+                continue;
+            }
+            // ai message:
+            if ("ai" in message) {
+                messages.push({
+                    role: "assistant",
+                    content: message.ai,
+                });
+                continue;
+            }
+            // unsupported message:
+            throw new Error(`Unsupported message: ${JSON.stringify(message)}`);
+        }
+        return messages;
+    },
+    stopTokens: [],
+});
+exports.ChatToOpenAIChatPromptMapping = ChatToOpenAIChatPromptMapping;

package/prompt/OpenAIChatPromptMapping.d.ts

@@ -0,0 +1,6 @@
+import { OpenAIChatMessage } from "../model-provider/openai/chat/OpenAIChatMessage.js";
+import { ChatPrompt } from "./chat/ChatPrompt.js";
+import { InstructionPrompt } from "./InstructionPrompt.js";
+import { PromptMapping } from "./PromptMapping.js";
+export declare const InstructionToOpenAIChatPromptMapping: () => PromptMapping<InstructionPrompt, Array<OpenAIChatMessage>>;
+export declare const ChatToOpenAIChatPromptMapping: () => PromptMapping<ChatPrompt, Array<OpenAIChatMessage>>;

package/prompt/OpenAIChatPromptMapping.js

@@ -0,0 +1,57 @@
+import { validateChatPrompt } from "./chat/validateChatPrompt.js";
+export const InstructionToOpenAIChatPromptMapping = () => ({
+    map: (instruction) => {
+        const messages = [];
+        if (instruction.system != null) {
+            messages.push({
+                role: "system",
+                content: instruction.system,
+            });
+        }
+        messages.push({
+            role: "user",
+            content: instruction.instruction,
+        });
+        return messages;
+    },
+    stopTokens: [],
+});
+export const ChatToOpenAIChatPromptMapping = () => ({
+    map: (chatPrompt) => {
+        validateChatPrompt(chatPrompt);
+        const messages = [];
+        for (let i = 0; i < chatPrompt.length; i++) {
+            const message = chatPrompt[i];
+            // system message:
+            if (i === 0 &&
+                "system" in message &&
+                typeof message.system === "string") {
+                messages.push({
+                    role: "system",
+                    content: message.system,
+                });
+                continue;
+            }
+            // user message
+            if ("user" in message) {
+                messages.push({
+                    role: "user",
+                    content: message.user,
+                });
+                continue;
+            }
+            // ai message:
+            if ("ai" in message) {
+                messages.push({
+                    role: "assistant",
+                    content: message.ai,
+                });
+                continue;
+            }
+            // unsupported message:
+            throw new Error(`Unsupported message: ${JSON.stringify(message)}`);
+        }
+        return messages;
+    },
+    stopTokens: [],
+});

package/prompt/PromptMapping.cjs

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/prompt/PromptMapping.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Prompt mappings map a source prompt format to a target prompt format.
+ */
+export interface PromptMapping<SOURCE_PROMPT, TARGET_PROMPT> {
+    map(sourcePrompt: SOURCE_PROMPT): TARGET_PROMPT;
+    stopTokens: string[];
+}

package/prompt/PromptMapping.js

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