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.
Files changed (133) hide show
  1. package/README.md +23 -2
  2. package/index.cjs +0 -1
  3. package/index.d.ts +0 -1
  4. package/index.js +0 -1
  5. package/model-function/ModelFunctionPromise.cjs +37 -0
  6. package/model-function/ModelFunctionPromise.d.ts +18 -0
  7. package/model-function/ModelFunctionPromise.js +33 -0
  8. package/{prompt → model-function}/PromptFormat.d.ts +0 -5
  9. package/model-function/describe-image/describeImage.cjs +3 -2
  10. package/model-function/describe-image/describeImage.d.ts +1 -1
  11. package/model-function/describe-image/describeImage.js +3 -2
  12. package/model-function/embed/embed.cjs +5 -4
  13. package/model-function/embed/embed.d.ts +1 -1
  14. package/model-function/embed/embed.js +5 -4
  15. package/model-function/executeCall.cjs +3 -46
  16. package/model-function/executeCall.d.ts +5 -18
  17. package/model-function/executeCall.js +1 -43
  18. package/model-function/generate-image/ImageGenerationModel.d.ts +2 -0
  19. package/model-function/generate-image/ImageGenerationPromise.cjs +50 -0
  20. package/model-function/generate-image/ImageGenerationPromise.d.ts +22 -0
  21. package/model-function/generate-image/ImageGenerationPromise.js +46 -0
  22. package/model-function/generate-image/PromptFormatImageGenerationModel.cjs +44 -0
  23. package/model-function/generate-image/PromptFormatImageGenerationModel.d.ts +20 -0
  24. package/model-function/generate-image/PromptFormatImageGenerationModel.js +40 -0
  25. package/model-function/generate-image/generateImage.cjs +3 -2
  26. package/model-function/generate-image/generateImage.d.ts +2 -2
  27. package/model-function/generate-image/generateImage.js +3 -2
  28. package/model-function/generate-structure/generateStructure.cjs +3 -2
  29. package/model-function/generate-structure/generateStructure.d.ts +1 -1
  30. package/model-function/generate-structure/generateStructure.js +3 -2
  31. package/model-function/generate-structure/generateStructureOrText.cjs +3 -2
  32. package/model-function/generate-structure/generateStructureOrText.d.ts +1 -1
  33. package/model-function/generate-structure/generateStructureOrText.js +3 -2
  34. package/{prompt → model-function/generate-text}/AlpacaPromptFormat.d.ts +2 -2
  35. package/{prompt → model-function/generate-text}/Llama2PromptFormat.cjs +1 -1
  36. package/model-function/generate-text/Llama2PromptFormat.d.ts +13 -0
  37. package/{prompt → model-function/generate-text}/Llama2PromptFormat.js +1 -1
  38. package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.d.ts +7 -7
  39. package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.d.ts +6 -6
  40. package/model-function/generate-text/TextGenerationModel.d.ts +3 -3
  41. package/model-function/generate-text/TextGenerationPromptFormat.cjs +2 -0
  42. package/model-function/generate-text/TextGenerationPromptFormat.d.ts +11 -0
  43. package/model-function/generate-text/TextGenerationPromptFormat.js +1 -0
  44. package/{prompt → model-function/generate-text}/TextPromptFormat.cjs +1 -1
  45. package/{prompt → model-function/generate-text}/TextPromptFormat.d.ts +4 -4
  46. package/{prompt → model-function/generate-text}/TextPromptFormat.js +1 -1
  47. package/{prompt → model-function/generate-text}/VicunaPromptFormat.cjs +1 -1
  48. package/{prompt → model-function/generate-text}/VicunaPromptFormat.d.ts +3 -3
  49. package/{prompt → model-function/generate-text}/VicunaPromptFormat.js +1 -1
  50. package/model-function/generate-text/generateText.cjs +6 -3
  51. package/model-function/generate-text/generateText.d.ts +1 -1
  52. package/model-function/generate-text/generateText.js +6 -3
  53. package/{prompt → model-function/generate-text}/index.cjs +9 -4
  54. package/model-function/generate-text/index.d.ts +16 -0
  55. package/model-function/generate-text/index.js +16 -0
  56. package/{prompt/chat → model-function/generate-text}/trimChatPrompt.d.ts +1 -1
  57. package/model-function/index.cjs +3 -5
  58. package/model-function/index.d.ts +3 -5
  59. package/model-function/index.js +3 -5
  60. package/model-function/synthesize-speech/synthesizeSpeech.cjs +3 -2
  61. package/model-function/synthesize-speech/synthesizeSpeech.d.ts +1 -1
  62. package/model-function/synthesize-speech/synthesizeSpeech.js +3 -2
  63. package/model-function/transcribe-speech/transcribe.cjs +3 -2
  64. package/model-function/transcribe-speech/transcribe.d.ts +1 -1
  65. package/model-function/transcribe-speech/transcribe.js +3 -2
  66. package/model-provider/anthropic/AnthropicPromptFormat.cjs +1 -1
  67. package/model-provider/anthropic/AnthropicPromptFormat.d.ts +5 -5
  68. package/model-provider/anthropic/AnthropicPromptFormat.js +1 -1
  69. package/model-provider/anthropic/AnthropicTextGenerationModel.cjs +1 -1
  70. package/model-provider/anthropic/AnthropicTextGenerationModel.d.ts +3 -3
  71. package/model-provider/anthropic/AnthropicTextGenerationModel.js +1 -1
  72. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.cjs +11 -0
  73. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts +12 -12
  74. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js +11 -0
  75. package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.cjs +12 -0
  76. package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.d.ts +10 -0
  77. package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.js +8 -0
  78. package/model-provider/automatic1111/index.cjs +1 -0
  79. package/model-provider/automatic1111/index.d.ts +1 -0
  80. package/model-provider/automatic1111/index.js +1 -0
  81. package/model-provider/cohere/CohereTextGenerationModel.cjs +2 -2
  82. package/model-provider/cohere/CohereTextGenerationModel.d.ts +3 -3
  83. package/model-provider/cohere/CohereTextGenerationModel.js +2 -2
  84. package/model-provider/huggingface/HuggingFaceTextGenerationModel.cjs +2 -2
  85. package/model-provider/huggingface/HuggingFaceTextGenerationModel.d.ts +5 -5
  86. package/model-provider/huggingface/HuggingFaceTextGenerationModel.js +2 -2
  87. package/model-provider/llamacpp/LlamaCppTextGenerationModel.cjs +1 -1
  88. package/model-provider/llamacpp/LlamaCppTextGenerationModel.d.ts +3 -3
  89. package/model-provider/llamacpp/LlamaCppTextGenerationModel.js +1 -1
  90. package/model-provider/openai/OpenAIImageGenerationModel.cjs +8 -1
  91. package/model-provider/openai/OpenAIImageGenerationModel.d.ts +6 -3
  92. package/model-provider/openai/OpenAIImageGenerationModel.js +8 -1
  93. package/model-provider/openai/OpenAITextGenerationModel.cjs +2 -2
  94. package/model-provider/openai/OpenAITextGenerationModel.d.ts +3 -3
  95. package/model-provider/openai/OpenAITextGenerationModel.js +2 -2
  96. package/model-provider/openai/chat/OpenAIChatModel.cjs +1 -1
  97. package/model-provider/openai/chat/OpenAIChatModel.d.ts +3 -3
  98. package/model-provider/openai/chat/OpenAIChatModel.js +1 -1
  99. package/model-provider/openai/chat/OpenAIChatPromptFormat.cjs +1 -1
  100. package/model-provider/openai/chat/OpenAIChatPromptFormat.d.ts +5 -5
  101. package/model-provider/openai/chat/OpenAIChatPromptFormat.js +1 -1
  102. package/model-provider/stability/StabilityImageGenerationModel.cjs +11 -0
  103. package/model-provider/stability/StabilityImageGenerationModel.d.ts +15 -14
  104. package/model-provider/stability/StabilityImageGenerationModel.js +11 -0
  105. package/model-provider/stability/StabilityImageGenerationPrompt.cjs +12 -0
  106. package/model-provider/stability/StabilityImageGenerationPrompt.d.ts +9 -0
  107. package/model-provider/stability/StabilityImageGenerationPrompt.js +8 -0
  108. package/model-provider/stability/index.cjs +1 -0
  109. package/model-provider/stability/index.d.ts +1 -0
  110. package/model-provider/stability/index.js +1 -0
  111. package/package.json +1 -1
  112. package/prompt/Llama2PromptFormat.d.ts +0 -13
  113. package/prompt/index.d.ts +0 -11
  114. package/prompt/index.js +0 -11
  115. /package/{prompt → model-function}/PromptFormat.cjs +0 -0
  116. /package/{prompt → model-function}/PromptFormat.js +0 -0
  117. /package/{prompt → model-function/generate-text}/AlpacaPromptFormat.cjs +0 -0
  118. /package/{prompt → model-function/generate-text}/AlpacaPromptFormat.js +0 -0
  119. /package/{prompt/chat → model-function/generate-text}/ChatPrompt.cjs +0 -0
  120. /package/{prompt/chat → model-function/generate-text}/ChatPrompt.d.ts +0 -0
  121. /package/{prompt/chat → model-function/generate-text}/ChatPrompt.js +0 -0
  122. /package/{prompt → model-function/generate-text}/InstructionPrompt.cjs +0 -0
  123. /package/{prompt → model-function/generate-text}/InstructionPrompt.d.ts +0 -0
  124. /package/{prompt → model-function/generate-text}/InstructionPrompt.js +0 -0
  125. /package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.cjs +0 -0
  126. /package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.js +0 -0
  127. /package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.cjs +0 -0
  128. /package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.js +0 -0
  129. /package/{prompt/chat → model-function/generate-text}/trimChatPrompt.cjs +0 -0
  130. /package/{prompt/chat → model-function/generate-text}/trimChatPrompt.js +0 -0
  131. /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.cjs +0 -0
  132. /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.d.ts +0 -0
  133. /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.js +0 -0
@@ -3,6 +3,7 @@ export * from "./Model.js";
3
3
  export * from "./ModelCallEvent.js";
4
4
  export * from "./ModelCallMetadata.js";
5
5
  export * from "./ModelInformation.js";
6
+ export * from "./PromptFormat.js";
6
7
  export * from "./SuccessfulModelCall.js";
7
8
  export * from "./describe-image/ImageDescriptionEvent.js";
8
9
  export * from "./describe-image/ImageDescriptionModel.js";
@@ -12,6 +13,7 @@ export * from "./embed/EmbeddingModel.js";
12
13
  export * from "./embed/embed.js";
13
14
  export * from "./generate-image/ImageGenerationEvent.js";
14
15
  export * from "./generate-image/ImageGenerationModel.js";
16
+ export * from "./generate-image/PromptFormatImageGenerationModel.js";
15
17
  export * from "./generate-image/generateImage.js";
16
18
  export * from "./generate-structure/NoSuchStructureError.js";
17
19
  export * from "./generate-structure/StructureFromTextGenerationModel.js";
@@ -24,11 +26,7 @@ export * from "./generate-structure/StructureValidationError.js";
24
26
  export * from "./generate-structure/generateStructure.js";
25
27
  export * from "./generate-structure/generateStructureOrText.js";
26
28
  export * from "./generate-structure/streamStructure.js";
27
- export * from "./generate-text/TextGenerationEvent.js";
28
- export * from "./generate-text/TextGenerationModel.js";
29
- export * from "./generate-text/TextStreamingEvent.js";
30
- export * from "./generate-text/generateText.js";
31
- export * from "./generate-text/streamText.js";
29
+ export * from "./generate-text/index.js";
32
30
  export * from "./synthesize-speech/SpeechSynthesisEvent.js";
33
31
  export * from "./synthesize-speech/SpeechSynthesisModel.js";
34
32
  export * from "./synthesize-speech/synthesizeSpeech.js";
@@ -3,6 +3,7 @@ export * from "./Model.js";
3
3
  export * from "./ModelCallEvent.js";
4
4
  export * from "./ModelCallMetadata.js";
5
5
  export * from "./ModelInformation.js";
6
+ export * from "./PromptFormat.js";
6
7
  export * from "./SuccessfulModelCall.js";
7
8
  export * from "./describe-image/ImageDescriptionEvent.js";
8
9
  export * from "./describe-image/ImageDescriptionModel.js";
@@ -12,6 +13,7 @@ export * from "./embed/EmbeddingModel.js";
12
13
  export * from "./embed/embed.js";
13
14
  export * from "./generate-image/ImageGenerationEvent.js";
14
15
  export * from "./generate-image/ImageGenerationModel.js";
16
+ export * from "./generate-image/PromptFormatImageGenerationModel.js";
15
17
  export * from "./generate-image/generateImage.js";
16
18
  export * from "./generate-structure/NoSuchStructureError.js";
17
19
  export * from "./generate-structure/StructureFromTextGenerationModel.js";
@@ -24,11 +26,7 @@ export * from "./generate-structure/StructureValidationError.js";
24
26
  export * from "./generate-structure/generateStructure.js";
25
27
  export * from "./generate-structure/generateStructureOrText.js";
26
28
  export * from "./generate-structure/streamStructure.js";
27
- export * from "./generate-text/TextGenerationEvent.js";
28
- export * from "./generate-text/TextGenerationModel.js";
29
- export * from "./generate-text/TextStreamingEvent.js";
30
- export * from "./generate-text/generateText.js";
31
- export * from "./generate-text/streamText.js";
29
+ export * from "./generate-text/index.js";
32
30
  export * from "./synthesize-speech/SpeechSynthesisEvent.js";
33
31
  export * from "./synthesize-speech/SpeechSynthesisModel.js";
34
32
  export * from "./synthesize-speech/synthesizeSpeech.js";
@@ -2,11 +2,12 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.synthesizeSpeech = void 0;
4
4
  const executeCall_js_1 = require("../executeCall.cjs");
5
+ const ModelFunctionPromise_js_1 = require("../ModelFunctionPromise.cjs");
5
6
  /**
6
7
  * Synthesizes speech from text.
7
8
  */
8
9
  function synthesizeSpeech(model, text, options) {
9
- return (0, executeCall_js_1.executeCall)({
10
+ return new ModelFunctionPromise_js_1.ModelFunctionPromise((0, executeCall_js_1.executeCall)({
10
11
  functionType: "speech-synthesis",
11
12
  input: text,
12
13
  model,
@@ -18,6 +19,6 @@ function synthesizeSpeech(model, text, options) {
18
19
  extractedValue: response,
19
20
  };
20
21
  },
21
- });
22
+ }));
22
23
  }
23
24
  exports.synthesizeSpeech = synthesizeSpeech;
@@ -1,6 +1,6 @@
1
1
  /// <reference types="node" />
2
2
  import { FunctionOptions } from "../../core/FunctionOptions.js";
3
- import { ModelFunctionPromise } from "../executeCall.js";
3
+ import { ModelFunctionPromise } from "../ModelFunctionPromise.js";
4
4
  import { SpeechSynthesisModel, SpeechSynthesisModelSettings } from "./SpeechSynthesisModel.js";
5
5
  /**
6
6
  * Synthesizes speech from text.
@@ -1,9 +1,10 @@
1
1
  import { executeCall } from "../executeCall.js";
2
+ import { ModelFunctionPromise } from "../ModelFunctionPromise.js";
2
3
  /**
3
4
  * Synthesizes speech from text.
4
5
  */
5
6
  export function synthesizeSpeech(model, text, options) {
6
- return executeCall({
7
+ return new ModelFunctionPromise(executeCall({
7
8
  functionType: "speech-synthesis",
8
9
  input: text,
9
10
  model,
@@ -15,5 +16,5 @@ export function synthesizeSpeech(model, text, options) {
15
16
  extractedValue: response,
16
17
  };
17
18
  },
18
- });
19
+ }));
19
20
  }
@@ -2,6 +2,7 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.transcribe = void 0;
4
4
  const executeCall_js_1 = require("../executeCall.cjs");
5
+ const ModelFunctionPromise_js_1 = require("../ModelFunctionPromise.cjs");
5
6
  /**
6
7
  * Transcribe audio data into text.
7
8
  *
@@ -17,7 +18,7 @@ const executeCall_js_1 = require("../executeCall.cjs");
17
18
  * );
18
19
  */
19
20
  function transcribe(model, data, options) {
20
- return (0, executeCall_js_1.executeCall)({
21
+ return new ModelFunctionPromise_js_1.ModelFunctionPromise((0, executeCall_js_1.executeCall)({
21
22
  functionType: "transcription",
22
23
  input: data,
23
24
  model,
@@ -29,6 +30,6 @@ function transcribe(model, data, options) {
29
30
  extractedValue: result.transcription,
30
31
  };
31
32
  },
32
- });
33
+ }));
33
34
  }
34
35
  exports.transcribe = transcribe;
@@ -1,5 +1,5 @@
1
1
  import { FunctionOptions } from "../../core/FunctionOptions.js";
2
- import { ModelFunctionPromise } from "../executeCall.js";
2
+ import { ModelFunctionPromise } from "../ModelFunctionPromise.js";
3
3
  import { TranscriptionModel, TranscriptionModelSettings } from "./TranscriptionModel.js";
4
4
  /**
5
5
  * Transcribe audio data into text.
@@ -1,4 +1,5 @@
1
1
  import { executeCall } from "../executeCall.js";
2
+ import { ModelFunctionPromise } from "../ModelFunctionPromise.js";
2
3
  /**
3
4
  * Transcribe audio data into text.
4
5
  *
@@ -14,7 +15,7 @@ import { executeCall } from "../executeCall.js";
14
15
  * );
15
16
  */
16
17
  export function transcribe(model, data, options) {
17
- return executeCall({
18
+ return new ModelFunctionPromise(executeCall({
18
19
  functionType: "transcription",
19
20
  input: data,
20
21
  model,
@@ -26,5 +27,5 @@ export function transcribe(model, data, options) {
26
27
  extractedValue: result.transcription,
27
28
  };
28
29
  },
29
- });
30
+ }));
30
31
  }
@@ -1,7 +1,7 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.mapChatPromptToAnthropicFormat = exports.mapInstructionPromptToAnthropicFormat = void 0;
4
- const validateChatPrompt_js_1 = require("../../prompt/chat/validateChatPrompt.cjs");
4
+ const validateChatPrompt_js_1 = require("../../model-function/generate-text/validateChatPrompt.cjs");
5
5
  /**
6
6
  * Formats an instruction prompt as an Anthropic prompt.
7
7
  */
@@ -1,11 +1,11 @@
1
- import { ChatPrompt } from "../../prompt/chat/ChatPrompt.js";
2
- import { InstructionPrompt } from "../../prompt/InstructionPrompt.js";
3
- import { PromptFormat } from "../../prompt/PromptFormat.js";
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";
4
4
  /**
5
5
  * Formats an instruction prompt as an Anthropic prompt.
6
6
  */
7
- export declare function mapInstructionPromptToAnthropicFormat(): PromptFormat<InstructionPrompt, string>;
7
+ export declare function mapInstructionPromptToAnthropicFormat(): TextGenerationPromptFormat<InstructionPrompt, string>;
8
8
  /**
9
9
  * Formats a chat prompt as an Anthropic prompt.
10
10
  */
11
- export declare function mapChatPromptToAnthropicFormat(): PromptFormat<ChatPrompt, string>;
11
+ export declare function mapChatPromptToAnthropicFormat(): TextGenerationPromptFormat<ChatPrompt, string>;
@@ -1,4 +1,4 @@
1
- import { validateChatPrompt } from "../../prompt/chat/validateChatPrompt.js";
1
+ import { validateChatPrompt } from "../../model-function/generate-text/validateChatPrompt.js";
2
2
  /**
3
3
  * Formats an instruction prompt as an Anthropic prompt.
4
4
  */
@@ -11,7 +11,7 @@ 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("../../prompt/PromptFormatTextStreamingModel.cjs");
14
+ const PromptFormatTextStreamingModel_js_1 = require("../../model-function/generate-text/PromptFormatTextStreamingModel.cjs");
15
15
  const AnthropicApiConfiguration_js_1 = require("./AnthropicApiConfiguration.cjs");
16
16
  const AnthropicError_js_1 = require("./AnthropicError.cjs");
17
17
  const AnthropicPromptFormat_js_1 = require("./AnthropicPromptFormat.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 { PromptFormat } from "../../prompt/PromptFormat.js";
9
- import { PromptFormatTextStreamingModel } from "../../prompt/PromptFormatTextStreamingModel.js";
9
+ import { TextGenerationPromptFormat } from "../../model-function/generate-text/TextGenerationPromptFormat.js";
10
10
  export declare const ANTHROPIC_TEXT_GENERATION_MODELS: {
11
11
  "claude-instant-1": {
12
12
  contextWindowSize: number;
@@ -63,7 +63,7 @@ export declare class AnthropicTextGenerationModel extends AbstractModel<Anthropi
63
63
  * Returns this model with a chat prompt format.
64
64
  */
65
65
  withChatPrompt(): PromptFormatTextStreamingModel<import("../../index.js").ChatPrompt, string, AnthropicTextGenerationModelSettings, this>;
66
- withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, string>): PromptFormatTextStreamingModel<INPUT_PROMPT, string, AnthropicTextGenerationModelSettings, this>;
66
+ withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, string>): PromptFormatTextStreamingModel<INPUT_PROMPT, string, AnthropicTextGenerationModelSettings, this>;
67
67
  withSettings(additionalSettings: Partial<AnthropicTextGenerationModelSettings>): this;
68
68
  }
69
69
  declare const anthropicTextGenerationResponseSchema: z.ZodObject<{
@@ -5,7 +5,7 @@ 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 "../../prompt/PromptFormatTextStreamingModel.js";
8
+ import { PromptFormatTextStreamingModel } from "../../model-function/generate-text/PromptFormatTextStreamingModel.js";
9
9
  import { AnthropicApiConfiguration } from "./AnthropicApiConfiguration.js";
10
10
  import { failedAnthropicCallResponseHandler } from "./AnthropicError.js";
11
11
  import { mapChatPromptToAnthropicFormat, mapInstructionPromptToAnthropicFormat, } from "./AnthropicPromptFormat.js";
@@ -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 Automatic1111ApiConfiguration_js_1 = require("./Automatic1111ApiConfiguration.cjs");
9
10
  const Automatic1111Error_js_1 = require("./Automatic1111Error.cjs");
11
+ const Automatic1111ImageGenerationPrompt_js_1 = require("./Automatic1111ImageGenerationPrompt.cjs");
10
12
  /**
11
13
  * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
12
14
  *
@@ -51,6 +53,15 @@ class Automatic1111ImageGenerationModel extends AbstractModel_js_1.AbstractModel
51
53
  base64Image: response.images[0],
52
54
  };
53
55
  }
56
+ withBasicPrompt() {
57
+ return this.withPromptFormat((0, Automatic1111ImageGenerationPrompt_js_1.mapBasicPromptToAutomatic1111Format)());
58
+ }
59
+ withPromptFormat(promptFormat) {
60
+ return new PromptFormatImageGenerationModel_js_1.PromptFormatImageGenerationModel({
61
+ model: this,
62
+ promptFormat,
63
+ });
64
+ }
54
65
  withSettings(additionalSettings) {
55
66
  return new Automatic1111ImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
56
67
  }
@@ -2,19 +2,22 @@ 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 { Automatic1111ImageGenerationPrompt } from "./Automatic1111ImageGenerationPrompt.js";
6
9
  /**
7
10
  * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
8
11
  *
9
12
  * @see https://github.com/AUTOMATIC1111/stable-diffusion-webui
10
13
  */
11
- export declare class Automatic1111ImageGenerationModel extends AbstractModel<Automatic1111ImageGenerationModelSettings> implements ImageGenerationModel<A111ImageGenerationPrompt, Automatic1111ImageGenerationModelSettings> {
12
- constructor(settings: Automatic1111ImageGenerationModelSettings);
14
+ export declare class Automatic1111ImageGenerationModel extends AbstractModel<Automatic1111ImageGenerationSettings> implements ImageGenerationModel<Automatic1111ImageGenerationPrompt, Automatic1111ImageGenerationSettings> {
15
+ constructor(settings: Automatic1111ImageGenerationSettings);
13
16
  readonly provider: "Automatic1111";
14
17
  get modelName(): string;
15
- callAPI(input: A111ImageGenerationPrompt, options?: FunctionOptions): Promise<Automatic1111ImageGenerationResponse>;
16
- get settingsForEvent(): Partial<Automatic1111ImageGenerationModelSettings>;
17
- doGenerateImage(prompt: A111ImageGenerationPrompt, options?: FunctionOptions): Promise<{
18
+ callAPI(input: Automatic1111ImageGenerationPrompt, options?: FunctionOptions): Promise<Automatic1111ImageGenerationResponse>;
19
+ get settingsForEvent(): Partial<Automatic1111ImageGenerationSettings>;
20
+ doGenerateImage(prompt: Automatic1111ImageGenerationPrompt, options?: FunctionOptions): Promise<{
18
21
  response: {
19
22
  images: string[];
20
23
  parameters: {};
@@ -22,9 +25,11 @@ export declare class Automatic1111ImageGenerationModel extends AbstractModel<Aut
22
25
  };
23
26
  base64Image: string;
24
27
  }>;
25
- withSettings(additionalSettings: Automatic1111ImageGenerationModelSettings): this;
28
+ withBasicPrompt(): PromptFormatImageGenerationModel<string, Automatic1111ImageGenerationPrompt, Automatic1111ImageGenerationSettings, this>;
29
+ withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, Automatic1111ImageGenerationPrompt>): PromptFormatImageGenerationModel<INPUT_PROMPT, Automatic1111ImageGenerationPrompt, Automatic1111ImageGenerationSettings, this>;
30
+ withSettings(additionalSettings: Automatic1111ImageGenerationSettings): this;
26
31
  }
27
- export interface Automatic1111ImageGenerationModelSettings extends ImageGenerationModelSettings {
32
+ export interface Automatic1111ImageGenerationSettings extends ImageGenerationModelSettings {
28
33
  api?: ApiConfiguration;
29
34
  model: string;
30
35
  height?: number;
@@ -46,9 +51,4 @@ declare const Automatic1111ImageGenerationResponseSchema: z.ZodObject<{
46
51
  info: string;
47
52
  }>;
48
53
  export type Automatic1111ImageGenerationResponse = z.infer<typeof Automatic1111ImageGenerationResponseSchema>;
49
- export type A111ImageGenerationPrompt = {
50
- prompt: string;
51
- negativePrompt?: string;
52
- seed?: number;
53
- };
54
54
  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 { Automatic1111ApiConfiguration } from "./Automatic1111ApiConfiguration.js";
6
7
  import { failedAutomatic1111CallResponseHandler } from "./Automatic1111Error.js";
8
+ import { mapBasicPromptToAutomatic1111Format, } from "./Automatic1111ImageGenerationPrompt.js";
7
9
  /**
8
10
  * Create an image generation model that calls the AUTOMATIC1111 Stable Diffusion Web UI API.
9
11
  *
@@ -48,6 +50,15 @@ export class Automatic1111ImageGenerationModel extends AbstractModel {
48
50
  base64Image: response.images[0],
49
51
  };
50
52
  }
53
+ withBasicPrompt() {
54
+ return this.withPromptFormat(mapBasicPromptToAutomatic1111Format());
55
+ }
56
+ withPromptFormat(promptFormat) {
57
+ return new PromptFormatImageGenerationModel({
58
+ model: this,
59
+ promptFormat,
60
+ });
61
+ }
51
62
  withSettings(additionalSettings) {
52
63
  return new Automatic1111ImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
53
64
  }
@@ -0,0 +1,12 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.mapBasicPromptToAutomatic1111Format = void 0;
4
+ /**
5
+ * Formats a basic text prompt as an Automatic1111 prompt.
6
+ */
7
+ function mapBasicPromptToAutomatic1111Format() {
8
+ return {
9
+ format: (description) => ({ prompt: description }),
10
+ };
11
+ }
12
+ exports.mapBasicPromptToAutomatic1111Format = mapBasicPromptToAutomatic1111Format;
@@ -0,0 +1,10 @@
1
+ import { PromptFormat } from "../../model-function/PromptFormat.js";
2
+ export type Automatic1111ImageGenerationPrompt = {
3
+ prompt: string;
4
+ negativePrompt?: string;
5
+ seed?: number;
6
+ };
7
+ /**
8
+ * Formats a basic text prompt as an Automatic1111 prompt.
9
+ */
10
+ export declare function mapBasicPromptToAutomatic1111Format(): PromptFormat<string, Automatic1111ImageGenerationPrompt>;
@@ -0,0 +1,8 @@
1
+ /**
2
+ * Formats a basic text prompt as an Automatic1111 prompt.
3
+ */
4
+ export function mapBasicPromptToAutomatic1111Format() {
5
+ return {
6
+ format: (description) => ({ prompt: description }),
7
+ };
8
+ }
@@ -19,3 +19,4 @@ __exportStar(require("./Automatic1111ApiConfiguration.cjs"), exports);
19
19
  var Automatic1111Error_js_1 = require("./Automatic1111Error.cjs");
20
20
  Object.defineProperty(exports, "Automatic1111Error", { enumerable: true, get: function () { return Automatic1111Error_js_1.Automatic1111Error; } });
21
21
  __exportStar(require("./Automatic1111ImageGenerationModel.cjs"), exports);
22
+ __exportStar(require("./Automatic1111ImageGenerationPrompt.cjs"), exports);
@@ -1,3 +1,4 @@
1
1
  export * from "./Automatic1111ApiConfiguration.js";
2
2
  export { Automatic1111Error, Automatic1111ErrorData, } from "./Automatic1111Error.js";
3
3
  export * from "./Automatic1111ImageGenerationModel.js";
4
+ export * from "./Automatic1111ImageGenerationPrompt.js";
@@ -1,3 +1,4 @@
1
1
  export * from "./Automatic1111ApiConfiguration.js";
2
2
  export { Automatic1111Error, } from "./Automatic1111Error.js";
3
3
  export * from "./Automatic1111ImageGenerationModel.js";
4
+ export * from "./Automatic1111ImageGenerationPrompt.js";
@@ -10,9 +10,9 @@ const callWithRetryAndThrottle_js_1 = require("../../core/api/callWithRetryAndTh
10
10
  const postToApi_js_1 = require("../../core/api/postToApi.cjs");
11
11
  const AsyncQueue_js_1 = require("../../event-source/AsyncQueue.cjs");
12
12
  const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
13
+ const PromptFormatTextStreamingModel_js_1 = require("../../model-function/generate-text/PromptFormatTextStreamingModel.cjs");
14
+ const TextPromptFormat_js_1 = require("../../model-function/generate-text/TextPromptFormat.cjs");
13
15
  const countTokens_js_1 = require("../../model-function/tokenize-text/countTokens.cjs");
14
- const PromptFormatTextStreamingModel_js_1 = require("../../prompt/PromptFormatTextStreamingModel.cjs");
15
- const TextPromptFormat_js_1 = require("../../prompt/TextPromptFormat.cjs");
16
16
  const CohereApiConfiguration_js_1 = require("./CohereApiConfiguration.cjs");
17
17
  const CohereError_js_1 = require("./CohereError.cjs");
18
18
  const CohereTokenizer_js_1 = require("./CohereTokenizer.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 { PromptFormat } from "../../prompt/PromptFormat.js";
9
- import { PromptFormatTextStreamingModel } from "../../prompt/PromptFormatTextStreamingModel.js";
9
+ import { TextGenerationPromptFormat } from "../../model-function/generate-text/TextGenerationPromptFormat.js";
10
10
  import { CohereTokenizer } from "./CohereTokenizer.js";
11
11
  export declare const COHERE_TEXT_GENERATION_MODELS: {
12
12
  command: {
@@ -95,7 +95,7 @@ export declare class CohereTextGenerationModel extends AbstractModel<CohereTextG
95
95
  user?: string;
96
96
  ai?: string;
97
97
  }): PromptFormatTextStreamingModel<import("../../index.js").ChatPrompt, string, CohereTextGenerationModelSettings, this>;
98
- withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, string>): PromptFormatTextStreamingModel<INPUT_PROMPT, string, CohereTextGenerationModelSettings, this>;
98
+ withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, string>): PromptFormatTextStreamingModel<INPUT_PROMPT, string, CohereTextGenerationModelSettings, this>;
99
99
  withSettings(additionalSettings: Partial<CohereTextGenerationModelSettings>): this;
100
100
  }
101
101
  declare const cohereTextGenerationResponseSchema: z.ZodObject<{
@@ -4,9 +4,9 @@ import { callWithRetryAndThrottle } from "../../core/api/callWithRetryAndThrottl
4
4
  import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postToApi.js";
5
5
  import { AsyncQueue } from "../../event-source/AsyncQueue.js";
6
6
  import { AbstractModel } from "../../model-function/AbstractModel.js";
7
+ import { PromptFormatTextStreamingModel } from "../../model-function/generate-text/PromptFormatTextStreamingModel.js";
8
+ import { mapChatPromptToTextFormat, mapInstructionPromptToTextFormat, } from "../../model-function/generate-text/TextPromptFormat.js";
7
9
  import { countTokens } from "../../model-function/tokenize-text/countTokens.js";
8
- import { PromptFormatTextStreamingModel } from "../../prompt/PromptFormatTextStreamingModel.js";
9
- import { mapChatPromptToTextFormat, mapInstructionPromptToTextFormat, } from "../../prompt/TextPromptFormat.js";
10
10
  import { CohereApiConfiguration } from "./CohereApiConfiguration.js";
11
11
  import { failedCohereCallResponseHandler } from "./CohereError.js";
12
12
  import { CohereTokenizer } from "./CohereTokenizer.js";
@@ -5,10 +5,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
5
5
  Object.defineProperty(exports, "__esModule", { value: true });
6
6
  exports.HuggingFaceTextGenerationModel = void 0;
7
7
  const zod_1 = __importDefault(require("zod"));
8
- const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
9
- const PromptFormatTextGenerationModel_js_1 = require("../../prompt/PromptFormatTextGenerationModel.cjs");
10
8
  const callWithRetryAndThrottle_js_1 = require("../../core/api/callWithRetryAndThrottle.cjs");
11
9
  const postToApi_js_1 = require("../../core/api/postToApi.cjs");
10
+ const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
11
+ const PromptFormatTextGenerationModel_js_1 = require("../../model-function/generate-text/PromptFormatTextGenerationModel.cjs");
12
12
  const HuggingFaceApiConfiguration_js_1 = require("./HuggingFaceApiConfiguration.cjs");
13
13
  const HuggingFaceError_js_1 = require("./HuggingFaceError.cjs");
14
14
  /**
@@ -1,10 +1,10 @@
1
1
  import z from "zod";
2
- import { AbstractModel } from "../../model-function/AbstractModel.js";
3
- import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
4
2
  import { FunctionOptions } from "../../core/FunctionOptions.js";
3
+ import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
4
+ import { AbstractModel } from "../../model-function/AbstractModel.js";
5
+ import { PromptFormatTextGenerationModel } from "../../model-function/generate-text/PromptFormatTextGenerationModel.js";
5
6
  import { TextGenerationModel, TextGenerationModelSettings } from "../../model-function/generate-text/TextGenerationModel.js";
6
- import { PromptFormat } from "../../prompt/PromptFormat.js";
7
- import { PromptFormatTextGenerationModel } from "../../prompt/PromptFormatTextGenerationModel.js";
7
+ import { TextGenerationPromptFormat } from "../../model-function/generate-text/TextGenerationPromptFormat.js";
8
8
  export interface HuggingFaceTextGenerationModelSettings extends TextGenerationModelSettings {
9
9
  api?: ApiConfiguration;
10
10
  model: string;
@@ -53,7 +53,7 @@ export declare class HuggingFaceTextGenerationModel extends AbstractModel<Huggin
53
53
  }[];
54
54
  text: string;
55
55
  }>;
56
- withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, string>): PromptFormatTextGenerationModel<INPUT_PROMPT, string, HuggingFaceTextGenerationModelSettings, this>;
56
+ withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, string>): PromptFormatTextGenerationModel<INPUT_PROMPT, string, HuggingFaceTextGenerationModelSettings, this>;
57
57
  withSettings(additionalSettings: Partial<HuggingFaceTextGenerationModelSettings>): this;
58
58
  }
59
59
  declare const huggingFaceTextGenerationResponseSchema: z.ZodArray<z.ZodObject<{
@@ -1,8 +1,8 @@
1
1
  import z from "zod";
2
- import { AbstractModel } from "../../model-function/AbstractModel.js";
3
- import { PromptFormatTextGenerationModel } from "../../prompt/PromptFormatTextGenerationModel.js";
4
2
  import { callWithRetryAndThrottle } from "../../core/api/callWithRetryAndThrottle.js";
5
3
  import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postToApi.js";
4
+ import { AbstractModel } from "../../model-function/AbstractModel.js";
5
+ import { PromptFormatTextGenerationModel } from "../../model-function/generate-text/PromptFormatTextGenerationModel.js";
6
6
  import { HuggingFaceApiConfiguration } from "./HuggingFaceApiConfiguration.js";
7
7
  import { failedHuggingFaceCallResponseHandler } from "./HuggingFaceError.js";
8
8
  /**
@@ -11,7 +11,7 @@ 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("../../prompt/PromptFormatTextStreamingModel.cjs");
14
+ const PromptFormatTextStreamingModel_js_1 = require("../../model-function/generate-text/PromptFormatTextStreamingModel.cjs");
15
15
  const LlamaCppApiConfiguration_js_1 = require("./LlamaCppApiConfiguration.cjs");
16
16
  const LlamaCppError_js_1 = require("./LlamaCppError.cjs");
17
17
  const LlamaCppTokenizer_js_1 = require("./LlamaCppTokenizer.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 { PromptFormat } from "../../prompt/PromptFormat.js";
9
- import { PromptFormatTextStreamingModel } from "../../prompt/PromptFormatTextStreamingModel.js";
9
+ import { TextGenerationPromptFormat } from "../../model-function/generate-text/TextGenerationPromptFormat.js";
10
10
  import { LlamaCppTokenizer } from "./LlamaCppTokenizer.js";
11
11
  export interface LlamaCppTextGenerationModelSettings<CONTEXT_WINDOW_SIZE extends number | undefined> extends TextGenerationModelSettings {
12
12
  api?: ApiConfiguration;
@@ -100,7 +100,7 @@ export declare class LlamaCppTextGenerationModel<CONTEXT_WINDOW_SIZE extends num
100
100
  };
101
101
  }>;
102
102
  doStreamText(prompt: string, options?: FunctionOptions): Promise<AsyncIterable<Delta<string>>>;
103
- withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, string>): PromptFormatTextStreamingModel<INPUT_PROMPT, string, LlamaCppTextGenerationModelSettings<CONTEXT_WINDOW_SIZE>, this>;
103
+ withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, string>): PromptFormatTextStreamingModel<INPUT_PROMPT, string, LlamaCppTextGenerationModelSettings<CONTEXT_WINDOW_SIZE>, this>;
104
104
  withSettings(additionalSettings: Partial<LlamaCppTextGenerationModelSettings<CONTEXT_WINDOW_SIZE>>): this;
105
105
  }
106
106
  declare const llamaCppTextGenerationResponseSchema: z.ZodObject<{
@@ -5,7 +5,7 @@ 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 "../../prompt/PromptFormatTextStreamingModel.js";
8
+ import { PromptFormatTextStreamingModel } from "../../model-function/generate-text/PromptFormatTextStreamingModel.js";
9
9
  import { LlamaCppApiConfiguration } from "./LlamaCppApiConfiguration.js";
10
10
  import { failedLlamaCppCallResponseHandler } from "./LlamaCppError.js";
11
11
  import { LlamaCppTokenizer } from "./LlamaCppTokenizer.js";
@@ -2,9 +2,10 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.OpenAIImageGenerationResponseFormat = exports.OpenAIImageGenerationModel = exports.calculateOpenAIImageGenerationCostInMillicents = void 0;
4
4
  const zod_1 = require("zod");
5
- const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
6
5
  const callWithRetryAndThrottle_js_1 = require("../../core/api/callWithRetryAndThrottle.cjs");
7
6
  const postToApi_js_1 = require("../../core/api/postToApi.cjs");
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 OpenAIApiConfiguration_js_1 = require("./OpenAIApiConfiguration.cjs");
9
10
  const OpenAIError_js_1 = require("./OpenAIError.cjs");
10
11
  /**
@@ -77,6 +78,12 @@ class OpenAIImageGenerationModel extends AbstractModel_js_1.AbstractModel {
77
78
  base64Image: response.data[0].b64_json,
78
79
  };
79
80
  }
81
+ withPromptFormat(promptFormat) {
82
+ return new PromptFormatImageGenerationModel_js_1.PromptFormatImageGenerationModel({
83
+ model: this,
84
+ promptFormat,
85
+ });
86
+ }
80
87
  withSettings(additionalSettings) {
81
88
  return new OpenAIImageGenerationModel(Object.assign({}, this.settings, additionalSettings));
82
89
  }
@@ -1,9 +1,11 @@
1
1
  import { z } from "zod";
2
- import { AbstractModel } from "../../model-function/AbstractModel.js";
2
+ import { FunctionOptions } from "../../core/FunctionOptions.js";
3
3
  import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
4
- import { ImageGenerationModel, ImageGenerationModelSettings } from "../../model-function/generate-image/ImageGenerationModel.js";
5
4
  import { ResponseHandler } from "../../core/api/postToApi.js";
6
- import { FunctionOptions } from "../../core/FunctionOptions.js";
5
+ import { AbstractModel } from "../../model-function/AbstractModel.js";
6
+ import { PromptFormat } from "../../model-function/PromptFormat.js";
7
+ import { ImageGenerationModel, ImageGenerationModelSettings } from "../../model-function/generate-image/ImageGenerationModel.js";
8
+ import { PromptFormatImageGenerationModel } from "../../model-function/generate-image/PromptFormatImageGenerationModel.js";
7
9
  export interface OpenAIImageGenerationCallSettings {
8
10
  n?: number;
9
11
  size?: "256x256" | "512x512" | "1024x1024";
@@ -43,6 +45,7 @@ export declare class OpenAIImageGenerationModel extends AbstractModel<OpenAIImag
43
45
  };
44
46
  base64Image: string;
45
47
  }>;
48
+ withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, string>): PromptFormatImageGenerationModel<INPUT_PROMPT, string, OpenAIImageGenerationSettings, this>;
46
49
  withSettings(additionalSettings: Partial<OpenAIImageGenerationSettings>): this;
47
50
  }
48
51
  export type OpenAIImageGenerationResponseFormatType<T> = {