modelfusion 0.133.0 → 0.135.0

package/model-function/generate-text/prompt-template/Llama2PromptTemplate.js

@@ -1,128 +0,0 @@
-import { validateContentIsString } from "./ContentPart.js";
-import { InvalidPromptError } from "./InvalidPromptError.js";
-// see https://github.com/facebookresearch/llama/blob/6c7fe276574e78057f917549435a2554000a876d/llama/generation.py#L44
-const BEGIN_SEGMENT = "<s>";
-const END_SEGMENT = " </s>";
-const BEGIN_INSTRUCTION = "[INST] ";
-const END_INSTRUCTION = " [/INST] ";
-const BEGIN_SYSTEM = "<<SYS>>\n";
-const END_SYSTEM = "\n<</SYS>>\n\n";
-/**
- * Formats a text prompt as a Llama 2 prompt.
- *
- * Llama 2 prompt template:
- * ```
- * <s>[INST] { instruction } [/INST]
- * ```
- *
- * @see https://www.philschmid.de/llama-2#how-to-prompt-llama-2-chat
- */
-export function text() {
-    return {
-        stopSequences: [END_SEGMENT],
-        format(prompt) {
-            return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt}${END_INSTRUCTION}`;
-        },
-    };
-}
-/**
- * Formats an instruction prompt as a Llama 2 prompt.
- *
- * Llama 2 prompt template:
- * ```
- * <s>[INST] <<SYS>>
- * ${ system prompt }
- * <</SYS>>
- * ${ instruction }
- * [/INST]
- * ${ response prefix }
- * ```
- *
- * @see https://www.philschmid.de/llama-2#how-to-prompt-llama-2-chat
- */
-export function instruction() {
-    return {
-        stopSequences: [END_SEGMENT],
-        format(prompt) {
-            const instruction = validateContentIsString(prompt.instruction, prompt);
-            return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.system != null
-                ? `${BEGIN_SYSTEM}${prompt.system}${END_SYSTEM}`
-                : ""}${instruction}${END_INSTRUCTION}${prompt.responsePrefix ?? ""}`;
-        },
-    };
-}
-/**
- * Formats a chat prompt as a Llama 2 prompt.
- *
- * Llama 2 prompt template:
- * ```
- * <s>[INST] <<SYS>>
- * ${ system prompt }
- * <</SYS>>
- *
- * ${ user msg 1 } [/INST] ${ model response 1 } </s><s>[INST] ${ user msg 2 } [/INST] ${ model response 2 } </s><s>[INST] ${ user msg 3 } [/INST]
- * ```
- */
-export function chat() {
-    return {
-        format(prompt) {
-            validateLlama2Prompt(prompt);
-            // get content of the first message (validated to be a user message)
-            const content = prompt.messages[0].content;
-            let text = `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.system != null
-                ? `${BEGIN_SYSTEM}${prompt.system}${END_SYSTEM}`
-                : ""}${content}${END_INSTRUCTION}`;
-            // process remaining messages
-            for (let i = 1; i < prompt.messages.length; i++) {
-                const { role, content } = prompt.messages[i];
-                switch (role) {
-                    case "user": {
-                        const textContent = validateContentIsString(content, prompt);
-                        text += `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${textContent}${END_INSTRUCTION}`;
-                        break;
-                    }
-                    case "assistant": {
-                        text += `${validateContentIsString(content, prompt)}${END_SEGMENT}`;
-                        break;
-                    }
-                    case "tool": {
-                        throw new InvalidPromptError("Tool messages are not supported.", prompt);
-                    }
-                    default: {
-                        const _exhaustiveCheck = role;
-                        throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
-                    }
-                }
-            }
-            return text;
-        },
-        stopSequences: [END_SEGMENT],
-    };
-}
-/**
- * Checks if a Llama2 chat prompt is valid. Throws a {@link ChatPromptValidationError} if it's not.
- *
- * - The first message of the chat must be a user message.
- * - Then it must be alternating between an assistant message and a user message.
- * - The last message must always be a user message (when submitting to a model).
- *
- * The type checking is done at runtime when you submit a chat prompt to a model with a prompt template.
- *
- * @throws {@link ChatPromptValidationError}
- */
-export function validateLlama2Prompt(chatPrompt) {
-    const messages = chatPrompt.messages;
-    if (messages.length < 1) {
-        throw new InvalidPromptError("ChatPrompt should have at least one message.", chatPrompt);
-    }
-    for (let i = 0; i < messages.length; i++) {
-        const expectedRole = i % 2 === 0 ? "user" : "assistant";
-        const role = messages[i].role;
-        if (role !== expectedRole) {
-            throw new InvalidPromptError(`Message at index ${i} should have role '${expectedRole}', but has role '${role}'.`, chatPrompt);
-        }
-    }
-    if (messages.length % 2 === 0) {
-        throw new InvalidPromptError("The last message must be a user message.", chatPrompt);
-    }
-}

package/model-function/generate-text/prompt-template/Llama2PromptTemplate.test.cjs

@@ -1,60 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const Llama2PromptTemplate_js_1 = require("./Llama2PromptTemplate.cjs");
-describe("text prompt", () => {
-    it("should format prompt", () => {
-        const prompt = (0, Llama2PromptTemplate_js_1.text)().format("prompt");
-        expect(prompt).toMatchSnapshot();
-    });
-});
-describe("instruction prompt", () => {
-    it("should format prompt with instruction", () => {
-        const prompt = (0, Llama2PromptTemplate_js_1.instruction)().format({
-            instruction: "instruction",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with system and instruction", () => {
-        const prompt = (0, Llama2PromptTemplate_js_1.instruction)().format({
-            system: "system",
-            instruction: "instruction",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with instruction and response prefix", () => {
-        const prompt = (0, Llama2PromptTemplate_js_1.instruction)().format({
-            instruction: "instruction",
-            responsePrefix: "response prefix",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-});
-describe("chat prompt", () => {
-    it("should format prompt with user message", () => {
-        const prompt = (0, Llama2PromptTemplate_js_1.chat)().format({
-            messages: [{ role: "user", content: "user message" }],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with user-assistant-user messages", () => {
-        const prompt = (0, Llama2PromptTemplate_js_1.chat)().format({
-            messages: [
-                { role: "user", content: "1st user message" },
-                { role: "assistant", content: "assistant message" },
-                { role: "user", content: "2nd user message" },
-            ],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with system message and user-assistant-user messages", () => {
-        const prompt = (0, Llama2PromptTemplate_js_1.chat)().format({
-            system: "you are a chatbot",
-            messages: [
-                { role: "user", content: "1st user message" },
-                { role: "assistant", content: "assistant message" },
-                { role: "user", content: "2nd user message" },
-            ],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-});

package/model-function/generate-text/prompt-template/Llama2PromptTemplate.test.d.ts

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

package/model-function/generate-text/prompt-template/Llama2PromptTemplate.test.js

@@ -1,58 +0,0 @@
-import { chat, instruction, text } from "./Llama2PromptTemplate.js";
-describe("text prompt", () => {
-    it("should format prompt", () => {
-        const prompt = text().format("prompt");
-        expect(prompt).toMatchSnapshot();
-    });
-});
-describe("instruction prompt", () => {
-    it("should format prompt with instruction", () => {
-        const prompt = instruction().format({
-            instruction: "instruction",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with system and instruction", () => {
-        const prompt = instruction().format({
-            system: "system",
-            instruction: "instruction",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with instruction and response prefix", () => {
-        const prompt = instruction().format({
-            instruction: "instruction",
-            responsePrefix: "response prefix",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-});
-describe("chat prompt", () => {
-    it("should format prompt with user message", () => {
-        const prompt = chat().format({
-            messages: [{ role: "user", content: "user message" }],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with user-assistant-user messages", () => {
-        const prompt = chat().format({
-            messages: [
-                { role: "user", content: "1st user message" },
-                { role: "assistant", content: "assistant message" },
-                { role: "user", content: "2nd user message" },
-            ],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with system message and user-assistant-user messages", () => {
-        const prompt = chat().format({
-            system: "you are a chatbot",
-            messages: [
-                { role: "user", content: "1st user message" },
-                { role: "assistant", content: "assistant message" },
-                { role: "user", content: "2nd user message" },
-            ],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-});

package/model-function/generate-text/prompt-template/MistralInstructPromptTemplate.cjs

@@ -1,150 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.validateMistralPrompt = exports.chat = exports.instruction = exports.text = void 0;
-const ContentPart_js_1 = require("./ContentPart.cjs");
-const InvalidPromptError_js_1 = require("./InvalidPromptError.cjs");
-const BEGIN_SEGMENT = "<s>";
-const END_SEGMENT = "</s>";
-const BEGIN_INSTRUCTION = "[INST] ";
-const END_INSTRUCTION = " [/INST] ";
-/**
- * Formats a text prompt as a Mistral instruct prompt.
- *
- * Mistral prompt template:
- * ```
- * <s>[INST] { instruction } [/INST]
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-function text() {
-    return {
-        stopSequences: [END_SEGMENT],
-        format(prompt) {
-            return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt}${END_INSTRUCTION}`;
-        },
-    };
-}
-exports.text = text;
-/**
- * Formats an instruction prompt as a Mistral instruct prompt.
- *
- * Note that Mistral does not support system prompts. We emulate them.
- *
- * Mistral prompt template when system prompt is set:
- * ```
- * <s>[INST] ${ system prompt } [/INST] </s>[INST] ${instruction} [/INST] ${ response prefix }
- * ```
- *
- * Mistral prompt template when there is no system prompt:
- * ```
- * <s>[INST] ${ instruction } [/INST] ${ response prefix }
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-function instruction() {
-    return {
-        stopSequences: [END_SEGMENT],
-        format(prompt) {
-            const instruction = (0, ContentPart_js_1.validateContentIsString)(prompt.instruction, prompt);
-            if (prompt.system != null) {
-                return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.system}${END_INSTRUCTION}${END_SEGMENT}${BEGIN_INSTRUCTION}${instruction}${END_INSTRUCTION}${prompt.responsePrefix ?? ""}`;
-            }
-            return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${instruction}${END_INSTRUCTION}${prompt.responsePrefix ?? ""}`;
-        },
-    };
-}
-exports.instruction = instruction;
-/**
- * Formats a chat prompt as a Mistral instruct prompt.
- *
- * Note that Mistral does not support system prompts. We emulate them.
- *
- * Mistral prompt template when system prompt is set:
- * ```
- * <s>[INST] ${ system prompt } [/INST] </s> [INST] ${ user msg 1 } [/INST] ${ model response 1 } [INST] ${ user msg 2 } [/INST] ${ model response 2 } [INST] ${ user msg 3 } [/INST]
- * ```
- *
- * Mistral prompt template when there is no system prompt:
- * ```
- * <s>[INST] ${ user msg 1 } [/INST] ${ model response 1 } </s>[INST] ${ user msg 2 } [/INST] ${ model response 2 } [INST] ${ user msg 3 } [/INST]
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-function chat() {
-    return {
-        format(prompt) {
-            validateMistralPrompt(prompt);
-            let text = "";
-            let i = 0;
-            // handle the special first segment
-            if (prompt.system != null) {
-                text += `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.system}${END_INSTRUCTION}${END_SEGMENT}`;
-            }
-            else {
-                // get content of the first message (validated to be a user message)
-                text = `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.messages[0].content}${END_INSTRUCTION}`;
-                // process 2nd message (validated to be an assistant message)
-                if (prompt.messages.length > 1) {
-                    text += `${prompt.messages[1].content}${END_SEGMENT}`;
-                }
-                i = 2;
-            }
-            // process remaining messages
-            for (; i < prompt.messages.length; i++) {
-                const { role, content } = prompt.messages[i];
-                switch (role) {
-                    case "user": {
-                        const textContent = (0, ContentPart_js_1.validateContentIsString)(content, prompt);
-                        text += `${BEGIN_INSTRUCTION}${textContent}${END_INSTRUCTION}`;
-                        break;
-                    }
-                    case "assistant": {
-                        text += (0, ContentPart_js_1.validateContentIsString)(content, prompt);
-                        break;
-                    }
-                    case "tool": {
-                        throw new InvalidPromptError_js_1.InvalidPromptError("Tool messages are not supported.", prompt);
-                    }
-                    default: {
-                        const _exhaustiveCheck = role;
-                        throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
-                    }
-                }
-            }
-            return text;
-        },
-        stopSequences: [END_SEGMENT],
-    };
-}
-exports.chat = chat;
-/**
- * Checks if a Mistral chat prompt is valid. Throws a {@link ChatPromptValidationError} if it's not.
- *
- * - The first message of the chat must be a user message.
- * - Then it must be alternating between an assistant message and a user message.
- * - The last message must always be a user message (when submitting to a model).
- *
- * The type checking is done at runtime when you submit a chat prompt to a model with a prompt template.
- *
- * @throws {@link ChatPromptValidationError}
- */
-function validateMistralPrompt(chatPrompt) {
-    const messages = chatPrompt.messages;
-    if (messages.length < 1) {
-        throw new InvalidPromptError_js_1.InvalidPromptError("ChatPrompt should have at least one message.", chatPrompt);
-    }
-    for (let i = 0; i < messages.length; i++) {
-        const expectedRole = i % 2 === 0 ? "user" : "assistant";
-        const role = messages[i].role;
-        if (role !== expectedRole) {
-            throw new InvalidPromptError_js_1.InvalidPromptError(`Message at index ${i} should have role '${expectedRole}', but has role '${role}'.`, chatPrompt);
-        }
-    }
-    if (messages.length % 2 === 0) {
-        throw new InvalidPromptError_js_1.InvalidPromptError("The last message must be a user message.", chatPrompt);
-    }
-}
-exports.validateMistralPrompt = validateMistralPrompt;

package/model-function/generate-text/prompt-template/MistralInstructPromptTemplate.d.ts

@@ -1,62 +0,0 @@
-import { TextGenerationPromptTemplate } from "../TextGenerationPromptTemplate.js";
-import { ChatPrompt } from "./ChatPrompt.js";
-import { InstructionPrompt } from "./InstructionPrompt.js";
-/**
- * Formats a text prompt as a Mistral instruct prompt.
- *
- * Mistral prompt template:
- * ```
- * <s>[INST] { instruction } [/INST]
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-export declare function text(): TextGenerationPromptTemplate<string, string>;
-/**
- * Formats an instruction prompt as a Mistral instruct prompt.
- *
- * Note that Mistral does not support system prompts. We emulate them.
- *
- * Mistral prompt template when system prompt is set:
- * ```
- * <s>[INST] ${ system prompt } [/INST] </s>[INST] ${instruction} [/INST] ${ response prefix }
- * ```
- *
- * Mistral prompt template when there is no system prompt:
- * ```
- * <s>[INST] ${ instruction } [/INST] ${ response prefix }
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-export declare function instruction(): TextGenerationPromptTemplate<InstructionPrompt, string>;
-/**
- * Formats a chat prompt as a Mistral instruct prompt.
- *
- * Note that Mistral does not support system prompts. We emulate them.
- *
- * Mistral prompt template when system prompt is set:
- * ```
- * <s>[INST] ${ system prompt } [/INST] </s> [INST] ${ user msg 1 } [/INST] ${ model response 1 } [INST] ${ user msg 2 } [/INST] ${ model response 2 } [INST] ${ user msg 3 } [/INST]
- * ```
- *
- * Mistral prompt template when there is no system prompt:
- * ```
- * <s>[INST] ${ user msg 1 } [/INST] ${ model response 1 } </s>[INST] ${ user msg 2 } [/INST] ${ model response 2 } [INST] ${ user msg 3 } [/INST]
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-export declare function chat(): TextGenerationPromptTemplate<ChatPrompt, string>;
-/**
- * Checks if a Mistral chat prompt is valid. Throws a {@link ChatPromptValidationError} if it's not.
- *
- * - The first message of the chat must be a user message.
- * - Then it must be alternating between an assistant message and a user message.
- * - The last message must always be a user message (when submitting to a model).
- *
- * The type checking is done at runtime when you submit a chat prompt to a model with a prompt template.
- *
- * @throws {@link ChatPromptValidationError}
- */
-export declare function validateMistralPrompt(chatPrompt: ChatPrompt): void;

package/model-function/generate-text/prompt-template/MistralInstructPromptTemplate.js

@@ -1,143 +0,0 @@
-import { validateContentIsString } from "./ContentPart.js";
-import { InvalidPromptError } from "./InvalidPromptError.js";
-const BEGIN_SEGMENT = "<s>";
-const END_SEGMENT = "</s>";
-const BEGIN_INSTRUCTION = "[INST] ";
-const END_INSTRUCTION = " [/INST] ";
-/**
- * Formats a text prompt as a Mistral instruct prompt.
- *
- * Mistral prompt template:
- * ```
- * <s>[INST] { instruction } [/INST]
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-export function text() {
-    return {
-        stopSequences: [END_SEGMENT],
-        format(prompt) {
-            return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt}${END_INSTRUCTION}`;
-        },
-    };
-}
-/**
- * Formats an instruction prompt as a Mistral instruct prompt.
- *
- * Note that Mistral does not support system prompts. We emulate them.
- *
- * Mistral prompt template when system prompt is set:
- * ```
- * <s>[INST] ${ system prompt } [/INST] </s>[INST] ${instruction} [/INST] ${ response prefix }
- * ```
- *
- * Mistral prompt template when there is no system prompt:
- * ```
- * <s>[INST] ${ instruction } [/INST] ${ response prefix }
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-export function instruction() {
-    return {
-        stopSequences: [END_SEGMENT],
-        format(prompt) {
-            const instruction = validateContentIsString(prompt.instruction, prompt);
-            if (prompt.system != null) {
-                return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.system}${END_INSTRUCTION}${END_SEGMENT}${BEGIN_INSTRUCTION}${instruction}${END_INSTRUCTION}${prompt.responsePrefix ?? ""}`;
-            }
-            return `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${instruction}${END_INSTRUCTION}${prompt.responsePrefix ?? ""}`;
-        },
-    };
-}
-/**
- * Formats a chat prompt as a Mistral instruct prompt.
- *
- * Note that Mistral does not support system prompts. We emulate them.
- *
- * Mistral prompt template when system prompt is set:
- * ```
- * <s>[INST] ${ system prompt } [/INST] </s> [INST] ${ user msg 1 } [/INST] ${ model response 1 } [INST] ${ user msg 2 } [/INST] ${ model response 2 } [INST] ${ user msg 3 } [/INST]
- * ```
- *
- * Mistral prompt template when there is no system prompt:
- * ```
- * <s>[INST] ${ user msg 1 } [/INST] ${ model response 1 } </s>[INST] ${ user msg 2 } [/INST] ${ model response 2 } [INST] ${ user msg 3 } [/INST]
- * ```
- *
- * @see https://docs.mistral.ai/models/#chat-template
- */
-export function chat() {
-    return {
-        format(prompt) {
-            validateMistralPrompt(prompt);
-            let text = "";
-            let i = 0;
-            // handle the special first segment
-            if (prompt.system != null) {
-                text += `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.system}${END_INSTRUCTION}${END_SEGMENT}`;
-            }
-            else {
-                // get content of the first message (validated to be a user message)
-                text = `${BEGIN_SEGMENT}${BEGIN_INSTRUCTION}${prompt.messages[0].content}${END_INSTRUCTION}`;
-                // process 2nd message (validated to be an assistant message)
-                if (prompt.messages.length > 1) {
-                    text += `${prompt.messages[1].content}${END_SEGMENT}`;
-                }
-                i = 2;
-            }
-            // process remaining messages
-            for (; i < prompt.messages.length; i++) {
-                const { role, content } = prompt.messages[i];
-                switch (role) {
-                    case "user": {
-                        const textContent = validateContentIsString(content, prompt);
-                        text += `${BEGIN_INSTRUCTION}${textContent}${END_INSTRUCTION}`;
-                        break;
-                    }
-                    case "assistant": {
-                        text += validateContentIsString(content, prompt);
-                        break;
-                    }
-                    case "tool": {
-                        throw new InvalidPromptError("Tool messages are not supported.", prompt);
-                    }
-                    default: {
-                        const _exhaustiveCheck = role;
-                        throw new Error(`Unsupported role: ${_exhaustiveCheck}`);
-                    }
-                }
-            }
-            return text;
-        },
-        stopSequences: [END_SEGMENT],
-    };
-}
-/**
- * Checks if a Mistral chat prompt is valid. Throws a {@link ChatPromptValidationError} if it's not.
- *
- * - The first message of the chat must be a user message.
- * - Then it must be alternating between an assistant message and a user message.
- * - The last message must always be a user message (when submitting to a model).
- *
- * The type checking is done at runtime when you submit a chat prompt to a model with a prompt template.
- *
- * @throws {@link ChatPromptValidationError}
- */
-export function validateMistralPrompt(chatPrompt) {
-    const messages = chatPrompt.messages;
-    if (messages.length < 1) {
-        throw new InvalidPromptError("ChatPrompt should have at least one message.", chatPrompt);
-    }
-    for (let i = 0; i < messages.length; i++) {
-        const expectedRole = i % 2 === 0 ? "user" : "assistant";
-        const role = messages[i].role;
-        if (role !== expectedRole) {
-            throw new InvalidPromptError(`Message at index ${i} should have role '${expectedRole}', but has role '${role}'.`, chatPrompt);
-        }
-    }
-    if (messages.length % 2 === 0) {
-        throw new InvalidPromptError("The last message must be a user message.", chatPrompt);
-    }
-}

package/model-function/generate-text/prompt-template/MistralInstructPromptTemplate.test.cjs

@@ -1,60 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const MistralInstructPromptTemplate_js_1 = require("./MistralInstructPromptTemplate.cjs");
-describe("text prompt", () => {
-    it("should format prompt", () => {
-        const prompt = (0, MistralInstructPromptTemplate_js_1.text)().format("prompt");
-        expect(prompt).toMatchSnapshot();
-    });
-});
-describe("instruction prompt", () => {
-    it("should format prompt with instruction", () => {
-        const prompt = (0, MistralInstructPromptTemplate_js_1.instruction)().format({
-            instruction: "instruction",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with system and instruction", () => {
-        const prompt = (0, MistralInstructPromptTemplate_js_1.instruction)().format({
-            system: "system",
-            instruction: "instruction",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with instruction and response prefix", () => {
-        const prompt = (0, MistralInstructPromptTemplate_js_1.instruction)().format({
-            instruction: "instruction",
-            responsePrefix: "response prefix",
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-});
-describe("chat prompt", () => {
-    it("should format prompt with user message", () => {
-        const prompt = (0, MistralInstructPromptTemplate_js_1.chat)().format({
-            messages: [{ role: "user", content: "user message" }],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with user-assistant-user messages", () => {
-        const prompt = (0, MistralInstructPromptTemplate_js_1.chat)().format({
-            messages: [
-                { role: "user", content: "1st user message" },
-                { role: "assistant", content: "assistant message" },
-                { role: "user", content: "2nd user message" },
-            ],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-    it("should format prompt with system message and user-assistant-user messages", () => {
-        const prompt = (0, MistralInstructPromptTemplate_js_1.chat)().format({
-            system: "you are a chatbot",
-            messages: [
-                { role: "user", content: "1st user message" },
-                { role: "assistant", content: "assistant message" },
-                { role: "user", content: "2nd user message" },
-            ],
-        });
-        expect(prompt).toMatchSnapshot();
-    });
-});

package/model-function/generate-text/prompt-template/MistralInstructPromptTemplate.test.d.ts

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