modelfusion 0.46.0 → 0.47.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +23 -2
- package/index.cjs +0 -1
- package/index.d.ts +0 -1
- package/index.js +0 -1
- package/model-function/ModelFunctionPromise.cjs +37 -0
- package/model-function/ModelFunctionPromise.d.ts +18 -0
- package/model-function/ModelFunctionPromise.js +33 -0
- package/{prompt → model-function}/PromptFormat.d.ts +0 -5
- package/model-function/describe-image/describeImage.cjs +3 -2
- package/model-function/describe-image/describeImage.d.ts +1 -1
- package/model-function/describe-image/describeImage.js +3 -2
- package/model-function/embed/embed.cjs +5 -4
- package/model-function/embed/embed.d.ts +1 -1
- package/model-function/embed/embed.js +5 -4
- package/model-function/executeCall.cjs +3 -46
- package/model-function/executeCall.d.ts +5 -18
- package/model-function/executeCall.js +1 -43
- package/model-function/generate-image/ImageGenerationModel.d.ts +2 -0
- package/model-function/generate-image/ImageGenerationPromise.cjs +50 -0
- package/model-function/generate-image/ImageGenerationPromise.d.ts +22 -0
- package/model-function/generate-image/ImageGenerationPromise.js +46 -0
- package/model-function/generate-image/PromptFormatImageGenerationModel.cjs +44 -0
- package/model-function/generate-image/PromptFormatImageGenerationModel.d.ts +20 -0
- package/model-function/generate-image/PromptFormatImageGenerationModel.js +40 -0
- package/model-function/generate-image/generateImage.cjs +3 -2
- package/model-function/generate-image/generateImage.d.ts +2 -2
- package/model-function/generate-image/generateImage.js +3 -2
- package/model-function/generate-structure/generateStructure.cjs +3 -2
- package/model-function/generate-structure/generateStructure.d.ts +1 -1
- package/model-function/generate-structure/generateStructure.js +3 -2
- package/model-function/generate-structure/generateStructureOrText.cjs +3 -2
- package/model-function/generate-structure/generateStructureOrText.d.ts +1 -1
- package/model-function/generate-structure/generateStructureOrText.js +3 -2
- package/{prompt → model-function/generate-text}/AlpacaPromptFormat.d.ts +2 -2
- package/{prompt → model-function/generate-text}/Llama2PromptFormat.cjs +1 -1
- package/model-function/generate-text/Llama2PromptFormat.d.ts +13 -0
- package/{prompt → model-function/generate-text}/Llama2PromptFormat.js +1 -1
- package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.d.ts +7 -7
- package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.d.ts +6 -6
- package/model-function/generate-text/TextGenerationModel.d.ts +3 -3
- package/model-function/generate-text/TextGenerationPromptFormat.cjs +2 -0
- package/model-function/generate-text/TextGenerationPromptFormat.d.ts +11 -0
- package/model-function/generate-text/TextGenerationPromptFormat.js +1 -0
- package/{prompt → model-function/generate-text}/TextPromptFormat.cjs +1 -1
- package/{prompt → model-function/generate-text}/TextPromptFormat.d.ts +4 -4
- package/{prompt → model-function/generate-text}/TextPromptFormat.js +1 -1
- package/{prompt → model-function/generate-text}/VicunaPromptFormat.cjs +1 -1
- package/{prompt → model-function/generate-text}/VicunaPromptFormat.d.ts +3 -3
- package/{prompt → model-function/generate-text}/VicunaPromptFormat.js +1 -1
- package/model-function/generate-text/generateText.cjs +6 -3
- package/model-function/generate-text/generateText.d.ts +1 -1
- package/model-function/generate-text/generateText.js +6 -3
- package/{prompt → model-function/generate-text}/index.cjs +9 -4
- package/model-function/generate-text/index.d.ts +16 -0
- package/model-function/generate-text/index.js +16 -0
- package/{prompt/chat → model-function/generate-text}/trimChatPrompt.d.ts +1 -1
- package/model-function/index.cjs +3 -5
- package/model-function/index.d.ts +3 -5
- package/model-function/index.js +3 -5
- package/model-function/synthesize-speech/synthesizeSpeech.cjs +3 -2
- package/model-function/synthesize-speech/synthesizeSpeech.d.ts +1 -1
- package/model-function/synthesize-speech/synthesizeSpeech.js +3 -2
- package/model-function/transcribe-speech/transcribe.cjs +3 -2
- package/model-function/transcribe-speech/transcribe.d.ts +1 -1
- package/model-function/transcribe-speech/transcribe.js +3 -2
- package/model-provider/anthropic/AnthropicPromptFormat.cjs +1 -1
- package/model-provider/anthropic/AnthropicPromptFormat.d.ts +5 -5
- package/model-provider/anthropic/AnthropicPromptFormat.js +1 -1
- package/model-provider/anthropic/AnthropicTextGenerationModel.cjs +1 -1
- package/model-provider/anthropic/AnthropicTextGenerationModel.d.ts +3 -3
- package/model-provider/anthropic/AnthropicTextGenerationModel.js +1 -1
- package/model-provider/automatic1111/Automatic1111ImageGenerationModel.cjs +11 -0
- package/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts +12 -12
- package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js +11 -0
- package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.cjs +12 -0
- package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.d.ts +10 -0
- package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.js +8 -0
- package/model-provider/automatic1111/index.cjs +1 -0
- package/model-provider/automatic1111/index.d.ts +1 -0
- package/model-provider/automatic1111/index.js +1 -0
- package/model-provider/cohere/CohereTextGenerationModel.cjs +2 -2
- package/model-provider/cohere/CohereTextGenerationModel.d.ts +3 -3
- package/model-provider/cohere/CohereTextGenerationModel.js +2 -2
- package/model-provider/huggingface/HuggingFaceTextGenerationModel.cjs +2 -2
- package/model-provider/huggingface/HuggingFaceTextGenerationModel.d.ts +5 -5
- package/model-provider/huggingface/HuggingFaceTextGenerationModel.js +2 -2
- package/model-provider/llamacpp/LlamaCppTextGenerationModel.cjs +1 -1
- package/model-provider/llamacpp/LlamaCppTextGenerationModel.d.ts +3 -3
- package/model-provider/llamacpp/LlamaCppTextGenerationModel.js +1 -1
- package/model-provider/openai/OpenAIImageGenerationModel.cjs +8 -1
- package/model-provider/openai/OpenAIImageGenerationModel.d.ts +6 -3
- package/model-provider/openai/OpenAIImageGenerationModel.js +8 -1
- package/model-provider/openai/OpenAITextGenerationModel.cjs +2 -2
- package/model-provider/openai/OpenAITextGenerationModel.d.ts +3 -3
- package/model-provider/openai/OpenAITextGenerationModel.js +2 -2
- package/model-provider/openai/chat/OpenAIChatModel.cjs +1 -1
- package/model-provider/openai/chat/OpenAIChatModel.d.ts +3 -3
- package/model-provider/openai/chat/OpenAIChatModel.js +1 -1
- package/model-provider/openai/chat/OpenAIChatPromptFormat.cjs +1 -1
- package/model-provider/openai/chat/OpenAIChatPromptFormat.d.ts +5 -5
- package/model-provider/openai/chat/OpenAIChatPromptFormat.js +1 -1
- package/model-provider/stability/StabilityImageGenerationModel.cjs +11 -0
- package/model-provider/stability/StabilityImageGenerationModel.d.ts +15 -14
- package/model-provider/stability/StabilityImageGenerationModel.js +11 -0
- package/model-provider/stability/StabilityImageGenerationPrompt.cjs +12 -0
- package/model-provider/stability/StabilityImageGenerationPrompt.d.ts +9 -0
- package/model-provider/stability/StabilityImageGenerationPrompt.js +8 -0
- package/model-provider/stability/index.cjs +1 -0
- package/model-provider/stability/index.d.ts +1 -0
- package/model-provider/stability/index.js +1 -0
- package/package.json +1 -1
- package/prompt/Llama2PromptFormat.d.ts +0 -13
- package/prompt/index.d.ts +0 -11
- package/prompt/index.js +0 -11
- /package/{prompt → model-function}/PromptFormat.cjs +0 -0
- /package/{prompt → model-function}/PromptFormat.js +0 -0
- /package/{prompt → model-function/generate-text}/AlpacaPromptFormat.cjs +0 -0
- /package/{prompt → model-function/generate-text}/AlpacaPromptFormat.js +0 -0
- /package/{prompt/chat → model-function/generate-text}/ChatPrompt.cjs +0 -0
- /package/{prompt/chat → model-function/generate-text}/ChatPrompt.d.ts +0 -0
- /package/{prompt/chat → model-function/generate-text}/ChatPrompt.js +0 -0
- /package/{prompt → model-function/generate-text}/InstructionPrompt.cjs +0 -0
- /package/{prompt → model-function/generate-text}/InstructionPrompt.d.ts +0 -0
- /package/{prompt → model-function/generate-text}/InstructionPrompt.js +0 -0
- /package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.cjs +0 -0
- /package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.js +0 -0
- /package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.cjs +0 -0
- /package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.js +0 -0
- /package/{prompt/chat → model-function/generate-text}/trimChatPrompt.cjs +0 -0
- /package/{prompt/chat → model-function/generate-text}/trimChatPrompt.js +0 -0
- /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.cjs +0 -0
- /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.d.ts +0 -0
- /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.js +0 -0
@@ -1,7 +1,8 @@
|
|
1
1
|
import { z } from "zod";
|
2
|
-
import { AbstractModel } from "../../model-function/AbstractModel.js";
|
3
2
|
import { callWithRetryAndThrottle } from "../../core/api/callWithRetryAndThrottle.js";
|
4
3
|
import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postToApi.js";
|
4
|
+
import { AbstractModel } from "../../model-function/AbstractModel.js";
|
5
|
+
import { PromptFormatImageGenerationModel } from "../../model-function/generate-image/PromptFormatImageGenerationModel.js";
|
5
6
|
import { OpenAIApiConfiguration } from "./OpenAIApiConfiguration.js";
|
6
7
|
import { failedOpenAICallResponseHandler } from "./OpenAIError.js";
|
7
8
|
/**
|
@@ -73,6 +74,12 @@ export class OpenAIImageGenerationModel extends AbstractModel {
|
|
73
74
|
base64Image: response.data[0].b64_json,
|
74
75
|
};
|
75
76
|
}
|
77
|
+
withPromptFormat(promptFormat) {
|
78
|
+
return new PromptFormatImageGenerationModel({
|
79
|
+
model: this,
|
80
|
+
promptFormat,
|
81
|
+
});
|
82
|
+
}
|
76
83
|
withSettings(additionalSettings) {
|
77
84
|
return new OpenAIImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
|
78
85
|
}
|
@@ -11,9 +11,9 @@ const postToApi_js_1 = require("../../core/api/postToApi.cjs");
|
|
11
11
|
const AsyncQueue_js_1 = require("../../event-source/AsyncQueue.cjs");
|
12
12
|
const parseEventSourceStream_js_1 = require("../../event-source/parseEventSourceStream.cjs");
|
13
13
|
const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
|
14
|
+
const PromptFormatTextStreamingModel_js_1 = require("../../model-function/generate-text/PromptFormatTextStreamingModel.cjs");
|
15
|
+
const TextPromptFormat_js_1 = require("../../model-function/generate-text/TextPromptFormat.cjs");
|
14
16
|
const countTokens_js_1 = require("../../model-function/tokenize-text/countTokens.cjs");
|
15
|
-
const PromptFormatTextStreamingModel_js_1 = require("../../prompt/PromptFormatTextStreamingModel.cjs");
|
16
|
-
const TextPromptFormat_js_1 = require("../../prompt/TextPromptFormat.cjs");
|
17
17
|
const OpenAIApiConfiguration_js_1 = require("./OpenAIApiConfiguration.cjs");
|
18
18
|
const OpenAIError_js_1 = require("./OpenAIError.cjs");
|
19
19
|
const TikTokenTokenizer_js_1 = require("./TikTokenTokenizer.cjs");
|
@@ -4,9 +4,9 @@ import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
|
|
4
4
|
import { ResponseHandler } from "../../core/api/postToApi.js";
|
5
5
|
import { AbstractModel } from "../../model-function/AbstractModel.js";
|
6
6
|
import { Delta } from "../../model-function/Delta.js";
|
7
|
+
import { PromptFormatTextStreamingModel } from "../../model-function/generate-text/PromptFormatTextStreamingModel.js";
|
7
8
|
import { TextGenerationModelSettings, TextStreamingModel } from "../../model-function/generate-text/TextGenerationModel.js";
|
8
|
-
import {
|
9
|
-
import { PromptFormatTextStreamingModel } from "../../prompt/PromptFormatTextStreamingModel.js";
|
9
|
+
import { TextGenerationPromptFormat } from "../../model-function/generate-text/TextGenerationPromptFormat.js";
|
10
10
|
import { TikTokenTokenizer } from "./TikTokenTokenizer.js";
|
11
11
|
/**
|
12
12
|
* @see https://platform.openai.com/docs/models/
|
@@ -182,7 +182,7 @@ export declare class OpenAITextGenerationModel extends AbstractModel<OpenAITextG
|
|
182
182
|
user?: string;
|
183
183
|
ai?: string;
|
184
184
|
}): PromptFormatTextStreamingModel<import("../../index.js").ChatPrompt, string, OpenAITextGenerationModelSettings, this>;
|
185
|
-
withPromptFormat<INPUT_PROMPT>(promptFormat:
|
185
|
+
withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, string>): PromptFormatTextStreamingModel<INPUT_PROMPT, string, OpenAITextGenerationModelSettings, this>;
|
186
186
|
withSettings(additionalSettings: Partial<OpenAITextGenerationModelSettings>): this;
|
187
187
|
}
|
188
188
|
declare const openAITextGenerationResponseSchema: z.ZodObject<{
|
@@ -5,9 +5,9 @@ import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postTo
|
|
5
5
|
import { AsyncQueue } from "../../event-source/AsyncQueue.js";
|
6
6
|
import { parseEventSourceStream } from "../../event-source/parseEventSourceStream.js";
|
7
7
|
import { AbstractModel } from "../../model-function/AbstractModel.js";
|
8
|
+
import { PromptFormatTextStreamingModel } from "../../model-function/generate-text/PromptFormatTextStreamingModel.js";
|
9
|
+
import { mapChatPromptToTextFormat, mapInstructionPromptToTextFormat, } from "../../model-function/generate-text/TextPromptFormat.js";
|
8
10
|
import { countTokens } from "../../model-function/tokenize-text/countTokens.js";
|
9
|
-
import { PromptFormatTextStreamingModel } from "../../prompt/PromptFormatTextStreamingModel.js";
|
10
|
-
import { mapChatPromptToTextFormat, mapInstructionPromptToTextFormat, } from "../../prompt/TextPromptFormat.js";
|
11
11
|
import { OpenAIApiConfiguration } from "./OpenAIApiConfiguration.js";
|
12
12
|
import { failedOpenAICallResponseHandler } from "./OpenAIError.js";
|
13
13
|
import { TikTokenTokenizer } from "./TikTokenTokenizer.js";
|
@@ -11,7 +11,7 @@ const postToApi_js_1 = require("../../../core/api/postToApi.cjs");
|
|
11
11
|
const AbstractModel_js_1 = require("../../../model-function/AbstractModel.cjs");
|
12
12
|
const StructureParseError_js_1 = require("../../../model-function/generate-structure/StructureParseError.cjs");
|
13
13
|
const parsePartialJson_js_1 = require("../../../model-function/generate-structure/parsePartialJson.cjs");
|
14
|
-
const PromptFormatTextStreamingModel_js_1 = require("../../../
|
14
|
+
const PromptFormatTextStreamingModel_js_1 = require("../../../model-function/generate-text/PromptFormatTextStreamingModel.cjs");
|
15
15
|
const OpenAIApiConfiguration_js_1 = require("../OpenAIApiConfiguration.cjs");
|
16
16
|
const OpenAIError_js_1 = require("../OpenAIError.cjs");
|
17
17
|
const TikTokenTokenizer_js_1 = require("../TikTokenTokenizer.cjs");
|
@@ -7,9 +7,9 @@ import { AbstractModel } from "../../../model-function/AbstractModel.js";
|
|
7
7
|
import { Delta } from "../../../model-function/Delta.js";
|
8
8
|
import { StructureGenerationModel } from "../../../model-function/generate-structure/StructureGenerationModel.js";
|
9
9
|
import { StructureOrTextGenerationModel } from "../../../model-function/generate-structure/StructureOrTextGenerationModel.js";
|
10
|
+
import { PromptFormatTextStreamingModel } from "../../../model-function/generate-text/PromptFormatTextStreamingModel.js";
|
10
11
|
import { TextGenerationModelSettings, TextStreamingModel } from "../../../model-function/generate-text/TextGenerationModel.js";
|
11
|
-
import {
|
12
|
-
import { PromptFormatTextStreamingModel } from "../../../prompt/PromptFormatTextStreamingModel.js";
|
12
|
+
import { TextGenerationPromptFormat } from "../../../model-function/generate-text/TextGenerationPromptFormat.js";
|
13
13
|
import { TikTokenTokenizer } from "../TikTokenTokenizer.js";
|
14
14
|
import { OpenAIChatMessage } from "./OpenAIChatMessage.js";
|
15
15
|
export declare const OPENAI_CHAT_MODELS: {
|
@@ -315,7 +315,7 @@ export declare class OpenAIChatModel extends AbstractModel<OpenAIChatSettings> i
|
|
315
315
|
* Returns this model with a chat prompt format.
|
316
316
|
*/
|
317
317
|
withChatPrompt(): PromptFormatTextStreamingModel<import("../../../index.js").ChatPrompt, OpenAIChatMessage[], OpenAIChatSettings, this>;
|
318
|
-
withPromptFormat<INPUT_PROMPT>(promptFormat:
|
318
|
+
withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, OpenAIChatMessage[]>): PromptFormatTextStreamingModel<INPUT_PROMPT, OpenAIChatMessage[], OpenAIChatSettings, this>;
|
319
319
|
withSettings(additionalSettings: Partial<OpenAIChatSettings>): this;
|
320
320
|
}
|
321
321
|
declare const openAIChatResponseSchema: z.ZodObject<{
|
@@ -5,7 +5,7 @@ import { createJsonResponseHandler, postJsonToApi, } from "../../../core/api/pos
|
|
5
5
|
import { AbstractModel } from "../../../model-function/AbstractModel.js";
|
6
6
|
import { StructureParseError } from "../../../model-function/generate-structure/StructureParseError.js";
|
7
7
|
import { parsePartialJson } from "../../../model-function/generate-structure/parsePartialJson.js";
|
8
|
-
import { PromptFormatTextStreamingModel } from "../../../
|
8
|
+
import { PromptFormatTextStreamingModel } from "../../../model-function/generate-text/PromptFormatTextStreamingModel.js";
|
9
9
|
import { OpenAIApiConfiguration } from "../OpenAIApiConfiguration.js";
|
10
10
|
import { failedOpenAICallResponseHandler } from "../OpenAIError.js";
|
11
11
|
import { TikTokenTokenizer } from "../TikTokenTokenizer.js";
|
@@ -1,7 +1,7 @@
|
|
1
1
|
"use strict";
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
3
3
|
exports.mapChatPromptToOpenAIChatFormat = exports.mapInstructionPromptToOpenAIChatFormat = void 0;
|
4
|
-
const validateChatPrompt_js_1 = require("../../../
|
4
|
+
const validateChatPrompt_js_1 = require("../../../model-function/generate-text/validateChatPrompt.cjs");
|
5
5
|
/**
|
6
6
|
* Formats an instruction prompt as an OpenAI chat prompt.
|
7
7
|
*/
|
@@ -1,12 +1,12 @@
|
|
1
|
+
import { ChatPrompt } from "../../../model-function/generate-text/ChatPrompt.js";
|
2
|
+
import { InstructionPrompt } from "../../../model-function/generate-text/InstructionPrompt.js";
|
3
|
+
import { TextGenerationPromptFormat } from "../../../model-function/generate-text/TextGenerationPromptFormat.js";
|
1
4
|
import { OpenAIChatMessage } from "./OpenAIChatMessage.js";
|
2
|
-
import { ChatPrompt } from "../../../prompt/chat/ChatPrompt.js";
|
3
|
-
import { InstructionPrompt } from "../../../prompt/InstructionPrompt.js";
|
4
|
-
import { PromptFormat } from "../../../prompt/PromptFormat.js";
|
5
5
|
/**
|
6
6
|
* Formats an instruction prompt as an OpenAI chat prompt.
|
7
7
|
*/
|
8
|
-
export declare function mapInstructionPromptToOpenAIChatFormat():
|
8
|
+
export declare function mapInstructionPromptToOpenAIChatFormat(): TextGenerationPromptFormat<InstructionPrompt, Array<OpenAIChatMessage>>;
|
9
9
|
/**
|
10
10
|
* Formats a chat prompt as an OpenAI chat prompt.
|
11
11
|
*/
|
12
|
-
export declare function mapChatPromptToOpenAIChatFormat():
|
12
|
+
export declare function mapChatPromptToOpenAIChatFormat(): TextGenerationPromptFormat<ChatPrompt, Array<OpenAIChatMessage>>;
|
@@ -5,8 +5,10 @@ const zod_1 = require("zod");
|
|
5
5
|
const callWithRetryAndThrottle_js_1 = require("../../core/api/callWithRetryAndThrottle.cjs");
|
6
6
|
const postToApi_js_1 = require("../../core/api/postToApi.cjs");
|
7
7
|
const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
|
8
|
+
const PromptFormatImageGenerationModel_js_1 = require("../../model-function/generate-image/PromptFormatImageGenerationModel.cjs");
|
8
9
|
const StabilityApiConfiguration_js_1 = require("./StabilityApiConfiguration.cjs");
|
9
10
|
const StabilityError_js_1 = require("./StabilityError.cjs");
|
11
|
+
const StabilityImageGenerationPrompt_js_1 = require("./StabilityImageGenerationPrompt.cjs");
|
10
12
|
/**
|
11
13
|
* Create an image generation model that calls the Stability AI image generation API.
|
12
14
|
*
|
@@ -76,6 +78,15 @@ class StabilityImageGenerationModel extends AbstractModel_js_1.AbstractModel {
|
|
76
78
|
base64Image: response.artifacts[0].base64,
|
77
79
|
};
|
78
80
|
}
|
81
|
+
withBasicPrompt() {
|
82
|
+
return this.withPromptFormat((0, StabilityImageGenerationPrompt_js_1.mapBasicPromptToStabilityFormat)());
|
83
|
+
}
|
84
|
+
withPromptFormat(promptFormat) {
|
85
|
+
return new PromptFormatImageGenerationModel_js_1.PromptFormatImageGenerationModel({
|
86
|
+
model: this,
|
87
|
+
promptFormat,
|
88
|
+
});
|
89
|
+
}
|
79
90
|
withSettings(additionalSettings) {
|
80
91
|
return new StabilityImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
|
81
92
|
}
|
@@ -2,7 +2,10 @@ import { z } from "zod";
|
|
2
2
|
import { FunctionOptions } from "../../core/FunctionOptions.js";
|
3
3
|
import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
|
4
4
|
import { AbstractModel } from "../../model-function/AbstractModel.js";
|
5
|
+
import { PromptFormat } from "../../model-function/PromptFormat.js";
|
5
6
|
import { ImageGenerationModel, ImageGenerationModelSettings } from "../../model-function/generate-image/ImageGenerationModel.js";
|
7
|
+
import { PromptFormatImageGenerationModel } from "../../model-function/generate-image/PromptFormatImageGenerationModel.js";
|
8
|
+
import { StabilityImageGenerationPrompt } from "./StabilityImageGenerationPrompt.js";
|
6
9
|
/**
|
7
10
|
* Create an image generation model that calls the Stability AI image generation API.
|
8
11
|
*
|
@@ -25,27 +28,29 @@ import { ImageGenerationModel, ImageGenerationModelSettings } from "../../model-
|
|
25
28
|
* ]
|
26
29
|
* );
|
27
30
|
*/
|
28
|
-
export declare class StabilityImageGenerationModel extends AbstractModel<
|
29
|
-
constructor(settings:
|
31
|
+
export declare class StabilityImageGenerationModel extends AbstractModel<StabilityImageGenerationSettings> implements ImageGenerationModel<StabilityImageGenerationPrompt, StabilityImageGenerationSettings> {
|
32
|
+
constructor(settings: StabilityImageGenerationSettings);
|
30
33
|
readonly provider: "stability";
|
31
34
|
get modelName(): StabilityImageGenerationModelType;
|
32
35
|
callAPI(input: StabilityImageGenerationPrompt, options?: FunctionOptions): Promise<StabilityImageGenerationResponse>;
|
33
|
-
get settingsForEvent(): Partial<
|
36
|
+
get settingsForEvent(): Partial<StabilityImageGenerationSettings>;
|
34
37
|
doGenerateImage(prompt: StabilityImageGenerationPrompt, options?: FunctionOptions): Promise<{
|
35
38
|
response: {
|
36
39
|
artifacts: {
|
37
|
-
seed: number;
|
38
40
|
base64: string;
|
41
|
+
seed: number;
|
39
42
|
finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
|
40
43
|
}[];
|
41
44
|
};
|
42
45
|
base64Image: string;
|
43
46
|
}>;
|
44
|
-
|
47
|
+
withBasicPrompt(): PromptFormatImageGenerationModel<string, StabilityImageGenerationPrompt, StabilityImageGenerationSettings, this>;
|
48
|
+
withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, StabilityImageGenerationPrompt>): PromptFormatImageGenerationModel<INPUT_PROMPT, StabilityImageGenerationPrompt, StabilityImageGenerationSettings, this>;
|
49
|
+
withSettings(additionalSettings: StabilityImageGenerationSettings): this;
|
45
50
|
}
|
46
51
|
declare const stabilityImageGenerationModels: readonly ["stable-diffusion-v1-5", "stable-diffusion-512-v2-1", "stable-diffusion-xl-1024-v0-9", "stable-diffusion-xl-1024-v1-0"];
|
47
52
|
export type StabilityImageGenerationModelType = (typeof stabilityImageGenerationModels)[number] | (string & {});
|
48
|
-
export interface
|
53
|
+
export interface StabilityImageGenerationSettings extends ImageGenerationModelSettings {
|
49
54
|
api?: ApiConfiguration;
|
50
55
|
model: StabilityImageGenerationModelType;
|
51
56
|
height?: number;
|
@@ -64,32 +69,28 @@ declare const stabilityImageGenerationResponseSchema: z.ZodObject<{
|
|
64
69
|
seed: z.ZodNumber;
|
65
70
|
finishReason: z.ZodEnum<["SUCCESS", "ERROR", "CONTENT_FILTERED"]>;
|
66
71
|
}, "strip", z.ZodTypeAny, {
|
67
|
-
seed: number;
|
68
72
|
base64: string;
|
73
|
+
seed: number;
|
69
74
|
finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
|
70
75
|
}, {
|
71
|
-
seed: number;
|
72
76
|
base64: string;
|
77
|
+
seed: number;
|
73
78
|
finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
|
74
79
|
}>, "many">;
|
75
80
|
}, "strip", z.ZodTypeAny, {
|
76
81
|
artifacts: {
|
77
|
-
seed: number;
|
78
82
|
base64: string;
|
83
|
+
seed: number;
|
79
84
|
finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
|
80
85
|
}[];
|
81
86
|
}, {
|
82
87
|
artifacts: {
|
83
|
-
seed: number;
|
84
88
|
base64: string;
|
89
|
+
seed: number;
|
85
90
|
finishReason: "SUCCESS" | "ERROR" | "CONTENT_FILTERED";
|
86
91
|
}[];
|
87
92
|
}>;
|
88
93
|
export type StabilityImageGenerationResponse = z.infer<typeof stabilityImageGenerationResponseSchema>;
|
89
94
|
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";
|
90
95
|
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";
|
91
|
-
export type StabilityImageGenerationPrompt = Array<{
|
92
|
-
text: string;
|
93
|
-
weight?: number;
|
94
|
-
}>;
|
95
96
|
export {};
|
@@ -2,8 +2,10 @@ import { z } from "zod";
|
|
2
2
|
import { callWithRetryAndThrottle } from "../../core/api/callWithRetryAndThrottle.js";
|
3
3
|
import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postToApi.js";
|
4
4
|
import { AbstractModel } from "../../model-function/AbstractModel.js";
|
5
|
+
import { PromptFormatImageGenerationModel } from "../../model-function/generate-image/PromptFormatImageGenerationModel.js";
|
5
6
|
import { StabilityApiConfiguration } from "./StabilityApiConfiguration.js";
|
6
7
|
import { failedStabilityCallResponseHandler } from "./StabilityError.js";
|
8
|
+
import { mapBasicPromptToStabilityFormat, } from "./StabilityImageGenerationPrompt.js";
|
7
9
|
/**
|
8
10
|
* Create an image generation model that calls the Stability AI image generation API.
|
9
11
|
*
|
@@ -73,6 +75,15 @@ export class StabilityImageGenerationModel extends AbstractModel {
|
|
73
75
|
base64Image: response.artifacts[0].base64,
|
74
76
|
};
|
75
77
|
}
|
78
|
+
withBasicPrompt() {
|
79
|
+
return this.withPromptFormat(mapBasicPromptToStabilityFormat());
|
80
|
+
}
|
81
|
+
withPromptFormat(promptFormat) {
|
82
|
+
return new PromptFormatImageGenerationModel({
|
83
|
+
model: this,
|
84
|
+
promptFormat,
|
85
|
+
});
|
86
|
+
}
|
76
87
|
withSettings(additionalSettings) {
|
77
88
|
return new StabilityImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
|
78
89
|
}
|
@@ -0,0 +1,12 @@
|
|
1
|
+
"use strict";
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
3
|
+
exports.mapBasicPromptToStabilityFormat = void 0;
|
4
|
+
/**
|
5
|
+
* Formats a basic text prompt as a Stability prompt.
|
6
|
+
*/
|
7
|
+
function mapBasicPromptToStabilityFormat() {
|
8
|
+
return {
|
9
|
+
format: (description) => [{ text: description }],
|
10
|
+
};
|
11
|
+
}
|
12
|
+
exports.mapBasicPromptToStabilityFormat = mapBasicPromptToStabilityFormat;
|
@@ -0,0 +1,9 @@
|
|
1
|
+
import { PromptFormat } from "../../model-function/PromptFormat.js";
|
2
|
+
export type StabilityImageGenerationPrompt = Array<{
|
3
|
+
text: string;
|
4
|
+
weight?: number;
|
5
|
+
}>;
|
6
|
+
/**
|
7
|
+
* Formats a basic text prompt as a Stability prompt.
|
8
|
+
*/
|
9
|
+
export declare function mapBasicPromptToStabilityFormat(): PromptFormat<string, StabilityImageGenerationPrompt>;
|
@@ -19,3 +19,4 @@ __exportStar(require("./StabilityApiConfiguration.cjs"), exports);
|
|
19
19
|
var StabilityError_js_1 = require("./StabilityError.cjs");
|
20
20
|
Object.defineProperty(exports, "StabilityError", { enumerable: true, get: function () { return StabilityError_js_1.StabilityError; } });
|
21
21
|
__exportStar(require("./StabilityImageGenerationModel.cjs"), exports);
|
22
|
+
__exportStar(require("./StabilityImageGenerationPrompt.cjs"), exports);
|
package/package.json
CHANGED
@@ -1,13 +0,0 @@
|
|
1
|
-
import { PromptFormat } from "./PromptFormat.js";
|
2
|
-
import { InstructionPrompt } from "./InstructionPrompt.js";
|
3
|
-
import { ChatPrompt } from "./chat/ChatPrompt.js";
|
4
|
-
/**
|
5
|
-
* Formats an instruction prompt as a Llama 2 prompt.
|
6
|
-
*
|
7
|
-
* @see https://www.philschmid.de/llama-2#how-to-prompt-llama-2-chat
|
8
|
-
*/
|
9
|
-
export declare function mapInstructionPromptToLlama2Format(): PromptFormat<InstructionPrompt, string>;
|
10
|
-
/**
|
11
|
-
* Formats a chat prompt as a Llama 2 prompt.
|
12
|
-
*/
|
13
|
-
export declare function mapChatPromptToLlama2Format(): PromptFormat<ChatPrompt, string>;
|
package/prompt/index.d.ts
DELETED
@@ -1,11 +0,0 @@
|
|
1
|
-
export * from "./AlpacaPromptFormat.js";
|
2
|
-
export * from "./InstructionPrompt.js";
|
3
|
-
export * from "./Llama2PromptFormat.js";
|
4
|
-
export * from "./PromptFormat.js";
|
5
|
-
export * from "./PromptFormatTextGenerationModel.js";
|
6
|
-
export * from "./PromptFormatTextStreamingModel.js";
|
7
|
-
export * from "./TextPromptFormat.js";
|
8
|
-
export * from "./VicunaPromptFormat.js";
|
9
|
-
export * from "./chat/ChatPrompt.js";
|
10
|
-
export * from "./chat/trimChatPrompt.js";
|
11
|
-
export * from "./chat/validateChatPrompt.js";
|
package/prompt/index.js
DELETED
@@ -1,11 +0,0 @@
|
|
1
|
-
export * from "./AlpacaPromptFormat.js";
|
2
|
-
export * from "./InstructionPrompt.js";
|
3
|
-
export * from "./Llama2PromptFormat.js";
|
4
|
-
export * from "./PromptFormat.js";
|
5
|
-
export * from "./PromptFormatTextGenerationModel.js";
|
6
|
-
export * from "./PromptFormatTextStreamingModel.js";
|
7
|
-
export * from "./TextPromptFormat.js";
|
8
|
-
export * from "./VicunaPromptFormat.js";
|
9
|
-
export * from "./chat/ChatPrompt.js";
|
10
|
-
export * from "./chat/trimChatPrompt.js";
|
11
|
-
export * from "./chat/validateChatPrompt.js";
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|