modelfusion 0.45.3 → 0.47.0

package/model-provider/stability/StabilityImageGenerationModel.js

@@ -2,8 +2,10 @@ import { z } from "zod";
 import { callWithRetryAndThrottle } from "../../core/api/callWithRetryAndThrottle.js";
 import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postToApi.js";
 import { AbstractModel } from "../../model-function/AbstractModel.js";
+import { PromptFormatImageGenerationModel } from "../../model-function/generate-image/PromptFormatImageGenerationModel.js";
 import { StabilityApiConfiguration } from "./StabilityApiConfiguration.js";
 import { failedStabilityCallResponseHandler } from "./StabilityError.js";
+import { mapBasicPromptToStabilityFormat, } from "./StabilityImageGenerationPrompt.js";
 /**
  * Create an image generation model that calls the Stability AI image generation API.
  *
@@ -73,6 +75,15 @@ export class StabilityImageGenerationModel extends AbstractModel {
             base64Image: response.artifacts[0].base64,
         };
     }
+    withBasicPrompt() {
+        return this.withPromptFormat(mapBasicPromptToStabilityFormat());
+    }
+    withPromptFormat(promptFormat) {
+        return new PromptFormatImageGenerationModel({
+            model: this,
+            promptFormat,
+        });
+    }
     withSettings(additionalSettings) {
         return new StabilityImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
     }

package/model-provider/stability/StabilityImageGenerationPrompt.cjs

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mapBasicPromptToStabilityFormat = void 0;
+/**
+ * Formats a basic text prompt as a Stability prompt.
+ */
+function mapBasicPromptToStabilityFormat() {
+    return {
+        format: (description) => [{ text: description }],
+    };
+}
+exports.mapBasicPromptToStabilityFormat = mapBasicPromptToStabilityFormat;

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

@@ -0,0 +1,9 @@
+import { PromptFormat } from "../../model-function/PromptFormat.js";
+export type StabilityImageGenerationPrompt = Array<{
+    text: string;
+    weight?: number;
+}>;
+/**
+ * Formats a basic text prompt as a Stability prompt.
+ */
+export declare function mapBasicPromptToStabilityFormat(): PromptFormat<string, StabilityImageGenerationPrompt>;

package/model-provider/stability/StabilityImageGenerationPrompt.js

@@ -0,0 +1,8 @@
+/**
+ * Formats a basic text prompt as a Stability prompt.
+ */
+export function mapBasicPromptToStabilityFormat() {
+    return {
+        format: (description) => [{ text: description }],
+    };
+}

package/model-provider/stability/index.cjs

@@ -19,3 +19,4 @@ __exportStar(require("./StabilityApiConfiguration.cjs"), exports);
 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);
+__exportStar(require("./StabilityImageGenerationPrompt.cjs"), exports);

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

@@ -1,3 +1,4 @@
 export * from "./StabilityApiConfiguration.js";
 export { StabilityError, StabilityErrorData } from "./StabilityError.js";
 export * from "./StabilityImageGenerationModel.js";
+export * from "./StabilityImageGenerationPrompt.js";

package/model-provider/stability/index.js

@@ -1,3 +1,4 @@
 export * from "./StabilityApiConfiguration.js";
 export { StabilityError } from "./StabilityError.js";
 export * from "./StabilityImageGenerationModel.js";
+export * from "./StabilityImageGenerationPrompt.js";

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "modelfusion",
   "description": "Build multimodal applications, chatbots, and agents with JavaScript and TypeScript.",
-  "version": "0.45.3",
+  "version": "0.47.0",
   "author": "Lars Grammel",
   "license": "MIT",
   "keywords": [

package/prompt/Llama2PromptFormat.d.ts

@@ -1,13 +0,0 @@
-import { PromptFormat } from "./PromptFormat.js";
-import { InstructionPrompt } from "./InstructionPrompt.js";
-import { ChatPrompt } from "./chat/ChatPrompt.js";
-/**
- * Formats an instruction prompt as a Llama 2 prompt.
- *
- * @see https://www.philschmid.de/llama-2#how-to-prompt-llama-2-chat
- */
-export declare function mapInstructionPromptToLlama2Format(): PromptFormat<InstructionPrompt, string>;
-/**
- * Formats a chat prompt as a Llama 2 prompt.
- */
-export declare function mapChatPromptToLlama2Format(): PromptFormat<ChatPrompt, string>;

package/prompt/TextPromptFormat.d.ts

@@ -1,17 +0,0 @@
-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 mapInstructionPromptToTextFormat: () => 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 mapChatPromptToTextFormat: ({ user, ai, }: {
-    user: string;
-    ai: string;
-}) => PromptFormat<ChatPrompt, string>;

package/prompt/index.d.ts

@@ -1,11 +0,0 @@
-export * from "./AlpacaPromptFormat.js";
-export * from "./InstructionPrompt.js";
-export * from "./Llama2PromptFormat.js";
-export * from "./PromptFormat.js";
-export * from "./PromptFormatTextGenerationModel.js";
-export * from "./PromptFormatTextStreamingModel.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 +0,0 @@
-export * from "./AlpacaPromptFormat.js";
-export * from "./InstructionPrompt.js";
-export * from "./Llama2PromptFormat.js";
-export * from "./PromptFormat.js";
-export * from "./PromptFormatTextGenerationModel.js";
-export * from "./PromptFormatTextStreamingModel.js";
-export * from "./TextPromptFormat.js";
-export * from "./VicunaPromptFormat.js";
-export * from "./chat/ChatPrompt.js";
-export * from "./chat/trimChatPrompt.js";
-export * from "./chat/validateChatPrompt.js";