modelfusion 0.13.0 → 0.14.0

package/prompt/OpenAIChatPromptFormat.d.ts

@@ -0,0 +1,12 @@
+import { OpenAIChatMessage } from "../model-provider/openai/chat/OpenAIChatMessage.js";
+import { ChatPrompt } from "./chat/ChatPrompt.js";
+import { InstructionPrompt } from "./InstructionPrompt.js";
+import { PromptFormat } from "./PromptFormat.js";
+/**
+ * Formats an instruction prompt as an OpenAI chat prompt.
+ */
+export declare const OpenAIChatInstructionPromptFormat: () => PromptFormat<InstructionPrompt, Array<OpenAIChatMessage>>;
+/**
+ * Formats a chat prompt as an OpenAI chat prompt.
+ */
+export declare const OpenAIChatChatPromptFormat: () => PromptFormat<ChatPrompt, Array<OpenAIChatMessage>>;

package/prompt/{OpenAIChatPromptMapping.js → OpenAIChatPromptFormat.js}

@@ -1,6 +1,9 @@
 import { validateChatPrompt } from "./chat/validateChatPrompt.js";
-export const InstructionToOpenAIChatPromptMapping = () => ({
-    map: (instruction) => {
+/**
+ * Formats an instruction prompt as an OpenAI chat prompt.
+ */
+export const OpenAIChatInstructionPromptFormat = () => ({
+    format: (instruction) => {
         const messages = [];
         if (instruction.system != null) {
             messages.push({
@@ -22,8 +25,11 @@ export const InstructionToOpenAIChatPromptMapping = () => ({
     },
     stopTokens: [],
 });
-export const ChatToOpenAIChatPromptMapping = () => ({
-    map: (chatPrompt) => {
+/**
+ * Formats a chat prompt as an OpenAI chat prompt.
+ */
+export const OpenAIChatChatPromptFormat = () => ({
+    format: (chatPrompt) => {
         validateChatPrompt(chatPrompt);
         const messages = [];
         for (let i = 0; i < chatPrompt.length; i++) {

package/prompt/PromptFormat.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Prompt formats format a source prompt into the structure of a target prompt.
+ */
+export interface PromptFormat<SOURCE_PROMPT, TARGET_PROMPT> {
+    /**
+     * Formats the source prompt into the structure of the target prompt.
+     */
+    format(sourcePrompt: SOURCE_PROMPT): TARGET_PROMPT;
+    /**
+     * The tokens that should be used as default stop tokens.
+     * This is e.g. important for chat formats.
+     */
+    stopTokens: string[];
+}

package/prompt/{PromptMappingTextGenerationModel.cjs → PromptFormatTextGenerationModel.cjs}

@@ -1,22 +1,22 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.PromptMappingTextGenerationModel = void 0;
-class PromptMappingTextGenerationModel {
-    constructor({ model, promptMapping, }) {
+exports.PromptFormatTextGenerationModel = void 0;
+class PromptFormatTextGenerationModel {
+    constructor({ model, promptFormat, }) {
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "promptMapping", {
+        Object.defineProperty(this, "promptFormat", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
         this.model = model;
-        this.promptMapping = promptMapping;
+        this.promptFormat = promptFormat;
     }
     get modelInformation() {
         return this.model.modelInformation;
@@ -35,10 +35,10 @@ class PromptMappingTextGenerationModel {
         if (originalCountPromptTokens === undefined) {
             return undefined;
         }
-        return ((prompt) => originalCountPromptTokens(this.promptMapping.map(prompt)));
+        return ((prompt) => originalCountPromptTokens(this.promptFormat.format(prompt)));
     }
     generateTextResponse(prompt, options) {
-        const mappedPrompt = this.promptMapping.map(prompt);
+        const mappedPrompt = this.promptFormat.format(prompt);
         return this.model.generateTextResponse(mappedPrompt, options);
     }
     extractText(response) {
@@ -50,39 +50,39 @@ class PromptMappingTextGenerationModel {
             return undefined;
         }
         return ((prompt, options) => {
-            const mappedPrompt = this.promptMapping.map(prompt);
+            const mappedPrompt = this.promptFormat.format(prompt);
             return originalGenerateDeltaStreamResponse(mappedPrompt, options);
         });
     }
     get extractTextDelta() {
         return this.model.extractTextDelta;
     }
-    mapPrompt(promptMapping) {
-        return new PromptMappingTextGenerationModel({
-            model: this.withStopTokens(promptMapping.stopTokens),
-            promptMapping,
+    withPromptFormat(promptFormat) {
+        return new PromptFormatTextGenerationModel({
+            model: this.withStopTokens(promptFormat.stopTokens),
+            promptFormat,
         });
     }
     withSettings(additionalSettings) {
-        return new PromptMappingTextGenerationModel({
+        return new PromptFormatTextGenerationModel({
             model: this.model.withSettings(additionalSettings),
-            promptMapping: this.promptMapping,
+            promptFormat: this.promptFormat,
         });
     }
     get maxCompletionTokens() {
         return this.model.maxCompletionTokens;
     }
     withMaxCompletionTokens(maxCompletionTokens) {
-        return new PromptMappingTextGenerationModel({
+        return new PromptFormatTextGenerationModel({
             model: this.model.withMaxCompletionTokens(maxCompletionTokens),
-            promptMapping: this.promptMapping,
+            promptFormat: this.promptFormat,
         });
     }
     withStopTokens(stopTokens) {
-        return new PromptMappingTextGenerationModel({
+        return new PromptFormatTextGenerationModel({
             model: this.model.withStopTokens(stopTokens),
-            promptMapping: this.promptMapping,
+            promptFormat: this.promptFormat,
         });
     }
 }
-exports.PromptMappingTextGenerationModel = PromptMappingTextGenerationModel;
+exports.PromptFormatTextGenerationModel = PromptFormatTextGenerationModel;

package/prompt/{PromptMappingTextGenerationModel.d.ts → PromptFormatTextGenerationModel.d.ts}

@@ -1,13 +1,13 @@
 import { FunctionOptions } from "../model-function/FunctionOptions.js";
 import { DeltaEvent } from "../model-function/generate-text/DeltaEvent.js";
 import { TextGenerationModel, TextGenerationModelSettings } from "../model-function/generate-text/TextGenerationModel.js";
-import { PromptMapping } from "./PromptMapping.js";
-export declare class PromptMappingTextGenerationModel<PROMPT, MODEL_PROMPT, RESPONSE, FULL_DELTA, SETTINGS extends TextGenerationModelSettings, MODEL extends TextGenerationModel<MODEL_PROMPT, RESPONSE, FULL_DELTA, SETTINGS>> implements TextGenerationModel<PROMPT, RESPONSE, FULL_DELTA, SETTINGS> {
+import { PromptFormat } from "./PromptFormat.js";
+export declare class PromptFormatTextGenerationModel<PROMPT, MODEL_PROMPT, RESPONSE, FULL_DELTA, SETTINGS extends TextGenerationModelSettings, MODEL extends TextGenerationModel<MODEL_PROMPT, RESPONSE, FULL_DELTA, SETTINGS>> implements TextGenerationModel<PROMPT, RESPONSE, FULL_DELTA, SETTINGS> {
     private readonly model;
-    private readonly promptMapping;
-    constructor({ model, promptMapping, }: {
+    private readonly promptFormat;
+    constructor({ model, promptFormat, }: {
         model: MODEL;
-        promptMapping: PromptMapping<PROMPT, MODEL_PROMPT>;
+        promptFormat: PromptFormat<PROMPT, MODEL_PROMPT>;
     });
     get modelInformation(): import("../index.js").ModelInformation;
     get settings(): SETTINGS;
@@ -18,7 +18,7 @@ export declare class PromptMappingTextGenerationModel<PROMPT, MODEL_PROMPT, RESP
     extractText(response: RESPONSE): string;
     get generateDeltaStreamResponse(): MODEL["generateDeltaStreamResponse"] extends undefined ? undefined : (prompt: PROMPT, options: FunctionOptions<SETTINGS>) => PromiseLike<AsyncIterable<DeltaEvent<FULL_DELTA>>>;
     get extractTextDelta(): MODEL["extractTextDelta"];
-    mapPrompt<INPUT_PROMPT>(promptMapping: PromptMapping<INPUT_PROMPT, PROMPT>): PromptMappingTextGenerationModel<INPUT_PROMPT, PROMPT, RESPONSE, FULL_DELTA, SETTINGS, this>;
+    withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, PROMPT>): PromptFormatTextGenerationModel<INPUT_PROMPT, PROMPT, RESPONSE, FULL_DELTA, SETTINGS, this>;
     withSettings(additionalSettings: Partial<SETTINGS>): this;
     get maxCompletionTokens(): MODEL["maxCompletionTokens"];
     withMaxCompletionTokens(maxCompletionTokens: number): this;

package/prompt/{PromptMappingTextGenerationModel.js → PromptFormatTextGenerationModel.js}

@@ -1,19 +1,19 @@
-export class PromptMappingTextGenerationModel {
-    constructor({ model, promptMapping, }) {
+export class PromptFormatTextGenerationModel {
+    constructor({ model, promptFormat, }) {
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "promptMapping", {
+        Object.defineProperty(this, "promptFormat", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
         this.model = model;
-        this.promptMapping = promptMapping;
+        this.promptFormat = promptFormat;
     }
     get modelInformation() {
         return this.model.modelInformation;
@@ -32,10 +32,10 @@ export class PromptMappingTextGenerationModel {
         if (originalCountPromptTokens === undefined) {
             return undefined;
         }
-        return ((prompt) => originalCountPromptTokens(this.promptMapping.map(prompt)));
+        return ((prompt) => originalCountPromptTokens(this.promptFormat.format(prompt)));
     }
     generateTextResponse(prompt, options) {
-        const mappedPrompt = this.promptMapping.map(prompt);
+        const mappedPrompt = this.promptFormat.format(prompt);
         return this.model.generateTextResponse(mappedPrompt, options);
     }
     extractText(response) {
@@ -47,38 +47,38 @@ export class PromptMappingTextGenerationModel {
             return undefined;
         }
         return ((prompt, options) => {
-            const mappedPrompt = this.promptMapping.map(prompt);
+            const mappedPrompt = this.promptFormat.format(prompt);
             return originalGenerateDeltaStreamResponse(mappedPrompt, options);
         });
     }
     get extractTextDelta() {
         return this.model.extractTextDelta;
     }
-    mapPrompt(promptMapping) {
-        return new PromptMappingTextGenerationModel({
-            model: this.withStopTokens(promptMapping.stopTokens),
-            promptMapping,
+    withPromptFormat(promptFormat) {
+        return new PromptFormatTextGenerationModel({
+            model: this.withStopTokens(promptFormat.stopTokens),
+            promptFormat,
         });
     }
     withSettings(additionalSettings) {
-        return new PromptMappingTextGenerationModel({
+        return new PromptFormatTextGenerationModel({
             model: this.model.withSettings(additionalSettings),
-            promptMapping: this.promptMapping,
+            promptFormat: this.promptFormat,
         });
     }
     get maxCompletionTokens() {
         return this.model.maxCompletionTokens;
     }
     withMaxCompletionTokens(maxCompletionTokens) {
-        return new PromptMappingTextGenerationModel({
+        return new PromptFormatTextGenerationModel({
             model: this.model.withMaxCompletionTokens(maxCompletionTokens),
-            promptMapping: this.promptMapping,
+            promptFormat: this.promptFormat,
         });
     }
     withStopTokens(stopTokens) {
-        return new PromptMappingTextGenerationModel({
+        return new PromptFormatTextGenerationModel({
             model: this.model.withStopTokens(stopTokens),
-            promptMapping: this.promptMapping,
+            promptFormat: this.promptFormat,
         });
     }
 }

package/prompt/{TextPromptMapping.cjs → TextPromptFormat.cjs}

@@ -1,10 +1,13 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChatToTextPromptMapping = exports.InstructionToTextPromptMapping = void 0;
+exports.TextChatPromptFormat = exports.TextInstructionPromptFormat = void 0;
 const validateChatPrompt_js_1 = require("./chat/validateChatPrompt.cjs");
-const InstructionToTextPromptMapping = () => ({
+/**
+ * Formats an instruction prompt as a basic text prompt.
+ */
+const TextInstructionPromptFormat = () => ({
     stopTokens: [],
-    map: (instruction) => {
+    format: (instruction) => {
         let text = "";
         if (instruction.system != null) {
             text += `${instruction.system}\n\n`;
@@ -16,15 +19,15 @@ const InstructionToTextPromptMapping = () => ({
         return text;
     },
 });
-exports.InstructionToTextPromptMapping = InstructionToTextPromptMapping;
+exports.TextInstructionPromptFormat = TextInstructionPromptFormat;
 /**
- * A mapping from a chat prompt to a text prompt.
+ * Formats a chat prompt as a basic text prompt.
  *
  * @param user The label of the user in the chat.
  * @param ai The name of the AI in the chat.
  */
-const ChatToTextPromptMapping = ({ user, ai }) => ({
-    map: (chatPrompt) => {
+const TextChatPromptFormat = ({ user, ai }) => ({
+    format: (chatPrompt) => {
         (0, validateChatPrompt_js_1.validateChatPrompt)(chatPrompt);
         let text = "";
         for (let i = 0; i < chatPrompt.length; i++) {
@@ -55,4 +58,4 @@ const ChatToTextPromptMapping = ({ user, ai }) => ({
     },
     stopTokens: [`\n${user}:`],
 });
-exports.ChatToTextPromptMapping = ChatToTextPromptMapping;
+exports.TextChatPromptFormat = TextChatPromptFormat;

package/prompt/TextPromptFormat.d.ts

@@ -0,0 +1,17 @@
+import { PromptFormat } from "./PromptFormat.js";
+import { InstructionPrompt } from "./InstructionPrompt.js";
+import { ChatPrompt } from "./chat/ChatPrompt.js";
+/**
+ * Formats an instruction prompt as a basic text prompt.
+ */
+export declare const TextInstructionPromptFormat: () => PromptFormat<InstructionPrompt, string>;
+/**
+ * Formats a chat prompt as a basic text prompt.
+ *
+ * @param user The label of the user in the chat.
+ * @param ai The name of the AI in the chat.
+ */
+export declare const TextChatPromptFormat: ({ user, ai, }: {
+    user: string;
+    ai: string;
+}) => PromptFormat<ChatPrompt, string>;

package/prompt/{TextPromptMapping.js → TextPromptFormat.js}

@@ -1,7 +1,10 @@
 import { validateChatPrompt } from "./chat/validateChatPrompt.js";
-export const InstructionToTextPromptMapping = () => ({
+/**
+ * Formats an instruction prompt as a basic text prompt.
+ */
+export const TextInstructionPromptFormat = () => ({
     stopTokens: [],
-    map: (instruction) => {
+    format: (instruction) => {
         let text = "";
         if (instruction.system != null) {
             text += `${instruction.system}\n\n`;
@@ -14,13 +17,13 @@ export const InstructionToTextPromptMapping = () => ({
     },
 });
 /**
- * A mapping from a chat prompt to a text prompt.
+ * Formats a chat prompt as a basic text prompt.
  *
  * @param user The label of the user in the chat.
  * @param ai The name of the AI in the chat.
  */
-export const ChatToTextPromptMapping = ({ user, ai }) => ({
-    map: (chatPrompt) => {
+export const TextChatPromptFormat = ({ user, ai }) => ({
+    format: (chatPrompt) => {
         validateChatPrompt(chatPrompt);
         let text = "";
         for (let i = 0; i < chatPrompt.length; i++) {

package/prompt/{VicunaPromptMapping.cjs → VicunaPromptFormat.cjs}

@@ -1,10 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ChatToVicunaPromptMapping = void 0;
+exports.VicunaChatPromptFormat = void 0;
 const validateChatPrompt_js_1 = require("./chat/validateChatPrompt.cjs");
 const DEFAULT_SYSTEM_PROMPT = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.";
 /**
- * A mapping from a chat prompt to a Vicuna prompt.
+ * Formats a chat prompt as a Vicuna prompt.
  *
  * Overridding the system message in the first chat message can affect model respones.
  *
@@ -16,8 +16,8 @@ const DEFAULT_SYSTEM_PROMPT = "A chat between a curious user and an artificial i
  * ASSISTANT:
  * ```
  */
-const ChatToVicunaPromptMapping = () => ({
-    map: (chatPrompt) => {
+const VicunaChatPromptFormat = () => ({
+    format: (chatPrompt) => {
         (0, validateChatPrompt_js_1.validateChatPrompt)(chatPrompt);
         let text = "";
         for (let i = 0; i < chatPrompt.length; i++) {
@@ -52,4 +52,4 @@ const ChatToVicunaPromptMapping = () => ({
     },
     stopTokens: [`\nUSER:`],
 });
-exports.ChatToVicunaPromptMapping = ChatToVicunaPromptMapping;
+exports.VicunaChatPromptFormat = VicunaChatPromptFormat;

package/prompt/{VicunaPromptMapping.d.ts → VicunaPromptFormat.d.ts}

@@ -1,7 +1,7 @@
-import { PromptMapping } from "./PromptMapping.js";
+import { PromptFormat } from "./PromptFormat.js";
 import { ChatPrompt } from "./chat/ChatPrompt.js";
 /**
- * A mapping from a chat prompt to a Vicuna prompt.
+ * Formats a chat prompt as a Vicuna prompt.
  *
  * Overridding the system message in the first chat message can affect model respones.
  *
@@ -13,4 +13,4 @@ import { ChatPrompt } from "./chat/ChatPrompt.js";
  * ASSISTANT:
  * ```
  */
-export declare const ChatToVicunaPromptMapping: () => PromptMapping<ChatPrompt, string>;
+export declare const VicunaChatPromptFormat: () => PromptFormat<ChatPrompt, string>;

package/prompt/{VicunaPromptMapping.js → VicunaPromptFormat.js}

@@ -1,7 +1,7 @@
 import { validateChatPrompt } from "./chat/validateChatPrompt.js";
 const DEFAULT_SYSTEM_PROMPT = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.";
 /**
- * A mapping from a chat prompt to a Vicuna prompt.
+ * Formats a chat prompt as a Vicuna prompt.
  *
  * Overridding the system message in the first chat message can affect model respones.
  *
@@ -13,8 +13,8 @@ const DEFAULT_SYSTEM_PROMPT = "A chat between a curious user and an artificial i
  * ASSISTANT:
  * ```
  */
-export const ChatToVicunaPromptMapping = () => ({
-    map: (chatPrompt) => {
+export const VicunaChatPromptFormat = () => ({
+    format: (chatPrompt) => {
         validateChatPrompt(chatPrompt);
         let text = "";
         for (let i = 0; i < chatPrompt.length; i++) {

package/prompt/chat/trimChatPrompt.cjs

@@ -10,7 +10,7 @@ const validateChatPrompt_js_1 = require("./validateChatPrompt.cjs");
  * When the minimal chat prompt (system message + last user message) is already too long, it will only
  * return this minimal chat prompt.
  *
- * @see https://modelfusion.dev/guide/function/generate-text/prompt-mapping#limiting-the-chat-length
+ * @see https://modelfusion.dev/guide/function/generate-text/prompt-format#limiting-the-chat-length
  */
 async function trimChatPrompt({ prompt, model, tokenLimit = model.contextWindowSize -
     (model.maxCompletionTokens ?? model.contextWindowSize / 4), }) {

package/prompt/chat/trimChatPrompt.d.ts

@@ -8,7 +8,7 @@ import { ChatPrompt } from "./ChatPrompt.js";
  * When the minimal chat prompt (system message + last user message) is already too long, it will only
  * return this minimal chat prompt.
  *
- * @see https://modelfusion.dev/guide/function/generate-text/prompt-mapping#limiting-the-chat-length
+ * @see https://modelfusion.dev/guide/function/generate-text/prompt-format#limiting-the-chat-length
  */
 export declare function trimChatPrompt({ prompt, model, tokenLimit, }: {
     prompt: ChatPrompt;

package/prompt/chat/trimChatPrompt.js

@@ -7,7 +7,7 @@ import { validateChatPrompt } from "./validateChatPrompt.js";
  * When the minimal chat prompt (system message + last user message) is already too long, it will only
  * return this minimal chat prompt.
  *
- * @see https://modelfusion.dev/guide/function/generate-text/prompt-mapping#limiting-the-chat-length
+ * @see https://modelfusion.dev/guide/function/generate-text/prompt-format#limiting-the-chat-length
  */
 export async function trimChatPrompt({ prompt, model, tokenLimit = model.contextWindowSize -
     (model.maxCompletionTokens ?? model.contextWindowSize / 4), }) {

package/prompt/index.cjs

@@ -14,14 +14,14 @@ 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 });
-__exportStar(require("./AlpacaPromptMapping.cjs"), exports);
+__exportStar(require("./AlpacaPromptFormat.cjs"), exports);
 __exportStar(require("./InstructionPrompt.cjs"), exports);
-__exportStar(require("./Llama2PromptMapping.cjs"), exports);
-__exportStar(require("./OpenAIChatPromptMapping.cjs"), exports);
-__exportStar(require("./PromptMapping.cjs"), exports);
-__exportStar(require("./PromptMappingTextGenerationModel.cjs"), exports);
-__exportStar(require("./TextPromptMapping.cjs"), exports);
-__exportStar(require("./VicunaPromptMapping.cjs"), exports);
+__exportStar(require("./Llama2PromptFormat.cjs"), exports);
+__exportStar(require("./OpenAIChatPromptFormat.cjs"), exports);
+__exportStar(require("./PromptFormat.cjs"), exports);
+__exportStar(require("./PromptFormatTextGenerationModel.cjs"), exports);
+__exportStar(require("./TextPromptFormat.cjs"), exports);
+__exportStar(require("./VicunaPromptFormat.cjs"), exports);
 __exportStar(require("./chat/ChatPrompt.cjs"), exports);
 __exportStar(require("./chat/trimChatPrompt.cjs"), exports);
 __exportStar(require("./chat/validateChatPrompt.cjs"), exports);

package/prompt/index.d.ts

@@ -1,11 +1,11 @@
-export * from "./AlpacaPromptMapping.js";
+export * from "./AlpacaPromptFormat.js";
 export * from "./InstructionPrompt.js";
-export * from "./Llama2PromptMapping.js";
-export * from "./OpenAIChatPromptMapping.js";
-export * from "./PromptMapping.js";
-export * from "./PromptMappingTextGenerationModel.js";
-export * from "./TextPromptMapping.js";
-export * from "./VicunaPromptMapping.js";
+export * from "./Llama2PromptFormat.js";
+export * from "./OpenAIChatPromptFormat.js";
+export * from "./PromptFormat.js";
+export * from "./PromptFormatTextGenerationModel.js";
+export * from "./TextPromptFormat.js";
+export * from "./VicunaPromptFormat.js";
 export * from "./chat/ChatPrompt.js";
 export * from "./chat/trimChatPrompt.js";
 export * from "./chat/validateChatPrompt.js";

package/prompt/index.js

@@ -1,11 +1,11 @@
-export * from "./AlpacaPromptMapping.js";
+export * from "./AlpacaPromptFormat.js";
 export * from "./InstructionPrompt.js";
-export * from "./Llama2PromptMapping.js";
-export * from "./OpenAIChatPromptMapping.js";
-export * from "./PromptMapping.js";
-export * from "./PromptMappingTextGenerationModel.js";
-export * from "./TextPromptMapping.js";
-export * from "./VicunaPromptMapping.js";
+export * from "./Llama2PromptFormat.js";
+export * from "./OpenAIChatPromptFormat.js";
+export * from "./PromptFormat.js";
+export * from "./PromptFormatTextGenerationModel.js";
+export * from "./TextPromptFormat.js";
+export * from "./VicunaPromptFormat.js";
 export * from "./chat/ChatPrompt.js";
 export * from "./chat/trimChatPrompt.js";
 export * from "./chat/validateChatPrompt.js";

package/tool/WebSearchTool.cjs

@@ -3,9 +3,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.WebSearchTool = void 0;
 const zod_1 = require("zod");
 const Tool_js_1 = require("./Tool.cjs");
-const INPUT_SCHEMA = zod_1.z.object({
-    query: zod_1.z.string(),
-});
 const OUTPUT_SCHEMA = zod_1.z.object({
     results: zod_1.z.array(zod_1.z.object({
         title: zod_1.z.string(),
@@ -13,39 +10,21 @@ const OUTPUT_SCHEMA = zod_1.z.object({
         snippet: zod_1.z.string(),
     })),
 });
+// expose the schemas to library consumers:
+const createInputSchema = (description) => 
+// same structure, but with description:
+zod_1.z.object({
+    query: zod_1.z.string().describe(description),
+});
 class WebSearchTool extends Tool_js_1.Tool {
     constructor({ name, description, queryDescription = "Search query", execute, }) {
         super({
             name,
             description,
-            inputSchema: WebSearchTool.createInputSchema(queryDescription),
+            inputSchema: createInputSchema(queryDescription),
             outputSchema: OUTPUT_SCHEMA,
             execute,
         });
-        Object.defineProperty(this, "outputSchema", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.outputSchema = OUTPUT_SCHEMA;
     }
 }
 exports.WebSearchTool = WebSearchTool;
-// expose the schemas to library consumers:
-Object.defineProperty(WebSearchTool, "createInputSchema", {
-    enumerable: true,
-    configurable: true,
-    writable: true,
-    value: (description) => 
-    // same structure, but with description:
-    zod_1.z.object({
-        query: zod_1.z.string().describe(description),
-    })
-});
-Object.defineProperty(WebSearchTool, "createOutputSchema", {
-    enumerable: true,
-    configurable: true,
-    writable: true,
-    value: () => OUTPUT_SCHEMA
-});