modelfusion 0.46.0 → 0.47.1

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 (143) 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/fixJson.test.cjs +183 -181
  29. package/model-function/generate-structure/fixJson.test.js +181 -182
  30. package/model-function/generate-structure/generateStructure.cjs +3 -2
  31. package/model-function/generate-structure/generateStructure.d.ts +1 -1
  32. package/model-function/generate-structure/generateStructure.js +3 -2
  33. package/model-function/generate-structure/generateStructureOrText.cjs +3 -2
  34. package/model-function/generate-structure/generateStructureOrText.d.ts +1 -1
  35. package/model-function/generate-structure/generateStructureOrText.js +3 -2
  36. package/model-function/generate-structure/streamStructure.cjs +2 -7
  37. package/model-function/generate-structure/streamStructure.js +2 -4
  38. package/{prompt → model-function/generate-text}/AlpacaPromptFormat.d.ts +2 -2
  39. package/{prompt → model-function/generate-text}/Llama2PromptFormat.cjs +1 -1
  40. package/model-function/generate-text/Llama2PromptFormat.d.ts +13 -0
  41. package/{prompt → model-function/generate-text}/Llama2PromptFormat.js +1 -1
  42. package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.d.ts +7 -7
  43. package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.d.ts +6 -6
  44. package/model-function/generate-text/TextGenerationModel.d.ts +3 -3
  45. package/model-function/generate-text/TextGenerationPromptFormat.cjs +2 -0
  46. package/model-function/generate-text/TextGenerationPromptFormat.d.ts +11 -0
  47. package/model-function/generate-text/TextGenerationPromptFormat.js +1 -0
  48. package/{prompt → model-function/generate-text}/TextPromptFormat.cjs +1 -1
  49. package/{prompt → model-function/generate-text}/TextPromptFormat.d.ts +4 -4
  50. package/{prompt → model-function/generate-text}/TextPromptFormat.js +1 -1
  51. package/{prompt → model-function/generate-text}/VicunaPromptFormat.cjs +1 -1
  52. package/{prompt → model-function/generate-text}/VicunaPromptFormat.d.ts +3 -3
  53. package/{prompt → model-function/generate-text}/VicunaPromptFormat.js +1 -1
  54. package/model-function/generate-text/generateText.cjs +6 -3
  55. package/model-function/generate-text/generateText.d.ts +1 -1
  56. package/model-function/generate-text/generateText.js +6 -3
  57. package/{prompt → model-function/generate-text}/index.cjs +9 -4
  58. package/model-function/generate-text/index.d.ts +16 -0
  59. package/model-function/generate-text/index.js +16 -0
  60. package/{prompt/chat → model-function/generate-text}/trimChatPrompt.d.ts +1 -1
  61. package/model-function/index.cjs +3 -5
  62. package/model-function/index.d.ts +3 -5
  63. package/model-function/index.js +3 -5
  64. package/model-function/synthesize-speech/synthesizeSpeech.cjs +3 -2
  65. package/model-function/synthesize-speech/synthesizeSpeech.d.ts +1 -1
  66. package/model-function/synthesize-speech/synthesizeSpeech.js +3 -2
  67. package/model-function/transcribe-speech/transcribe.cjs +3 -2
  68. package/model-function/transcribe-speech/transcribe.d.ts +1 -1
  69. package/model-function/transcribe-speech/transcribe.js +3 -2
  70. package/model-provider/anthropic/AnthropicPromptFormat.cjs +1 -1
  71. package/model-provider/anthropic/AnthropicPromptFormat.d.ts +5 -5
  72. package/model-provider/anthropic/AnthropicPromptFormat.js +1 -1
  73. package/model-provider/anthropic/AnthropicTextGenerationModel.cjs +1 -1
  74. package/model-provider/anthropic/AnthropicTextGenerationModel.d.ts +3 -3
  75. package/model-provider/anthropic/AnthropicTextGenerationModel.js +1 -1
  76. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.cjs +11 -0
  77. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts +12 -12
  78. package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js +11 -0
  79. package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.cjs +12 -0
  80. package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.d.ts +10 -0
  81. package/model-provider/automatic1111/Automatic1111ImageGenerationPrompt.js +8 -0
  82. package/model-provider/automatic1111/index.cjs +1 -0
  83. package/model-provider/automatic1111/index.d.ts +1 -0
  84. package/model-provider/automatic1111/index.js +1 -0
  85. package/model-provider/cohere/CohereTextGenerationModel.cjs +2 -2
  86. package/model-provider/cohere/CohereTextGenerationModel.d.ts +3 -3
  87. package/model-provider/cohere/CohereTextGenerationModel.js +2 -2
  88. package/model-provider/huggingface/HuggingFaceTextGenerationModel.cjs +2 -2
  89. package/model-provider/huggingface/HuggingFaceTextGenerationModel.d.ts +5 -5
  90. package/model-provider/huggingface/HuggingFaceTextGenerationModel.js +2 -2
  91. package/model-provider/llamacpp/LlamaCppTextGenerationModel.cjs +1 -1
  92. package/model-provider/llamacpp/LlamaCppTextGenerationModel.d.ts +3 -3
  93. package/model-provider/llamacpp/LlamaCppTextGenerationModel.js +1 -1
  94. package/model-provider/openai/OpenAIImageGenerationModel.cjs +8 -1
  95. package/model-provider/openai/OpenAIImageGenerationModel.d.ts +6 -3
  96. package/model-provider/openai/OpenAIImageGenerationModel.js +8 -1
  97. package/model-provider/openai/OpenAITextGenerationModel.cjs +2 -2
  98. package/model-provider/openai/OpenAITextGenerationModel.d.ts +3 -3
  99. package/model-provider/openai/OpenAITextGenerationModel.js +2 -2
  100. package/model-provider/openai/chat/OpenAIChatModel.cjs +1 -1
  101. package/model-provider/openai/chat/OpenAIChatModel.d.ts +3 -3
  102. package/model-provider/openai/chat/OpenAIChatModel.js +1 -1
  103. package/model-provider/openai/chat/OpenAIChatPromptFormat.cjs +1 -1
  104. package/model-provider/openai/chat/OpenAIChatPromptFormat.d.ts +5 -5
  105. package/model-provider/openai/chat/OpenAIChatPromptFormat.js +1 -1
  106. package/model-provider/stability/StabilityImageGenerationModel.cjs +11 -0
  107. package/model-provider/stability/StabilityImageGenerationModel.d.ts +15 -14
  108. package/model-provider/stability/StabilityImageGenerationModel.js +11 -0
  109. package/model-provider/stability/StabilityImageGenerationPrompt.cjs +12 -0
  110. package/model-provider/stability/StabilityImageGenerationPrompt.d.ts +9 -0
  111. package/model-provider/stability/StabilityImageGenerationPrompt.js +8 -0
  112. package/model-provider/stability/index.cjs +1 -0
  113. package/model-provider/stability/index.d.ts +1 -0
  114. package/model-provider/stability/index.js +1 -0
  115. package/package.json +4 -4
  116. package/util/isDeepEqualData.cjs +53 -0
  117. package/util/isDeepEqualData.d.ts +8 -0
  118. package/util/isDeepEqualData.js +49 -0
  119. package/util/isDeepEqualData.test.cjs +108 -0
  120. package/util/isDeepEqualData.test.d.ts +1 -0
  121. package/util/isDeepEqualData.test.js +103 -0
  122. package/prompt/Llama2PromptFormat.d.ts +0 -13
  123. package/prompt/index.d.ts +0 -11
  124. package/prompt/index.js +0 -11
  125. /package/{prompt → model-function}/PromptFormat.cjs +0 -0
  126. /package/{prompt → model-function}/PromptFormat.js +0 -0
  127. /package/{prompt → model-function/generate-text}/AlpacaPromptFormat.cjs +0 -0
  128. /package/{prompt → model-function/generate-text}/AlpacaPromptFormat.js +0 -0
  129. /package/{prompt/chat → model-function/generate-text}/ChatPrompt.cjs +0 -0
  130. /package/{prompt/chat → model-function/generate-text}/ChatPrompt.d.ts +0 -0
  131. /package/{prompt/chat → model-function/generate-text}/ChatPrompt.js +0 -0
  132. /package/{prompt → model-function/generate-text}/InstructionPrompt.cjs +0 -0
  133. /package/{prompt → model-function/generate-text}/InstructionPrompt.d.ts +0 -0
  134. /package/{prompt → model-function/generate-text}/InstructionPrompt.js +0 -0
  135. /package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.cjs +0 -0
  136. /package/{prompt → model-function/generate-text}/PromptFormatTextGenerationModel.js +0 -0
  137. /package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.cjs +0 -0
  138. /package/{prompt → model-function/generate-text}/PromptFormatTextStreamingModel.js +0 -0
  139. /package/{prompt/chat → model-function/generate-text}/trimChatPrompt.cjs +0 -0
  140. /package/{prompt/chat → model-function/generate-text}/trimChatPrompt.js +0 -0
  141. /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.cjs +0 -0
  142. /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.d.ts +0 -0
  143. /package/{prompt/chat → model-function/generate-text}/validateChatPrompt.js +0 -0
@@ -1,13 +1,13 @@
1
- import { FunctionOptions } from "../core/FunctionOptions.js";
2
- import { TextGenerationModelSettings, TextStreamingModel } from "../model-function/generate-text/TextGenerationModel.js";
3
- import { PromptFormat } from "./PromptFormat.js";
1
+ import { FunctionOptions } from "../../core/FunctionOptions.js";
4
2
  import { PromptFormatTextGenerationModel } from "./PromptFormatTextGenerationModel.js";
3
+ import { TextGenerationModelSettings, TextStreamingModel } from "./TextGenerationModel.js";
4
+ import { TextGenerationPromptFormat } from "./TextGenerationPromptFormat.js";
5
5
  export declare class PromptFormatTextStreamingModel<PROMPT, MODEL_PROMPT, SETTINGS extends TextGenerationModelSettings, MODEL extends TextStreamingModel<MODEL_PROMPT, SETTINGS>> extends PromptFormatTextGenerationModel<PROMPT, MODEL_PROMPT, SETTINGS, MODEL> implements TextStreamingModel<PROMPT, SETTINGS> {
6
6
  constructor(options: {
7
7
  model: MODEL;
8
- promptFormat: PromptFormat<PROMPT, MODEL_PROMPT>;
8
+ promptFormat: TextGenerationPromptFormat<PROMPT, MODEL_PROMPT>;
9
9
  });
10
- doStreamText(prompt: PROMPT, options?: FunctionOptions): PromiseLike<AsyncIterable<import("../index.js").Delta<string>>>;
11
- withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, PROMPT>): PromptFormatTextStreamingModel<INPUT_PROMPT, PROMPT, SETTINGS, this>;
10
+ doStreamText(prompt: PROMPT, options?: FunctionOptions): PromiseLike<AsyncIterable<import("../Delta.js").Delta<string>>>;
11
+ withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, PROMPT>): PromptFormatTextStreamingModel<INPUT_PROMPT, PROMPT, SETTINGS, this>;
12
12
  withSettings(additionalSettings: Partial<SETTINGS>): this;
13
13
  }
@@ -1,8 +1,8 @@
1
1
  import { FunctionOptions } from "../../core/FunctionOptions.js";
2
- import { PromptFormat } from "../../prompt/PromptFormat.js";
3
2
  import { Delta } from "../Delta.js";
4
3
  import { Model, ModelSettings } from "../Model.js";
5
4
  import { BasicTokenizer, FullTokenizer } from "../tokenize-text/Tokenizer.js";
5
+ import { TextGenerationPromptFormat } from "./TextGenerationPromptFormat.js";
6
6
  export interface TextGenerationModelSettings extends ModelSettings {
7
7
  /**
8
8
  * Maximum number of tokens to generate.
@@ -43,9 +43,9 @@ export interface TextGenerationModel<PROMPT, SETTINGS extends TextGenerationMode
43
43
  totalTokens: number;
44
44
  };
45
45
  }>;
46
- withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, PROMPT>): TextGenerationModel<INPUT_PROMPT, SETTINGS>;
46
+ withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, PROMPT>): TextGenerationModel<INPUT_PROMPT, SETTINGS>;
47
47
  }
48
48
  export interface TextStreamingModel<PROMPT, SETTINGS extends TextGenerationModelSettings = TextGenerationModelSettings> extends TextGenerationModel<PROMPT, SETTINGS> {
49
49
  doStreamText(prompt: PROMPT, options?: FunctionOptions): PromiseLike<AsyncIterable<Delta<string>>>;
50
- withPromptFormat<INPUT_PROMPT>(promptFormat: PromptFormat<INPUT_PROMPT, PROMPT>): TextStreamingModel<INPUT_PROMPT, SETTINGS>;
50
+ withPromptFormat<INPUT_PROMPT>(promptFormat: TextGenerationPromptFormat<INPUT_PROMPT, PROMPT>): TextStreamingModel<INPUT_PROMPT, SETTINGS>;
51
51
  }
@@ -0,0 +1,2 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
@@ -0,0 +1,11 @@
1
+ import { PromptFormat } from "../PromptFormat.js";
2
+ /**
3
+ * Prompt formats format a source prompt into the structure of a target prompt.
4
+ */
5
+ export interface TextGenerationPromptFormat<SOURCE_PROMPT, TARGET_PROMPT> extends PromptFormat<SOURCE_PROMPT, TARGET_PROMPT> {
6
+ /**
7
+ * The texts that should be used as default stop sequences.
8
+ * This is e.g. important for chat formats.
9
+ */
10
+ stopSequences: string[];
11
+ }
@@ -1,7 +1,7 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.mapChatPromptToTextFormat = exports.mapInstructionPromptToTextFormat = void 0;
4
- const validateChatPrompt_js_1 = require("./chat/validateChatPrompt.cjs");
4
+ const validateChatPrompt_js_1 = require("./validateChatPrompt.cjs");
5
5
  /**
6
6
  * Formats an instruction prompt as a basic text prompt.
7
7
  */
@@ -1,10 +1,10 @@
1
- import { PromptFormat } from "./PromptFormat.js";
1
+ import { ChatPrompt } from "./ChatPrompt.js";
2
2
  import { InstructionPrompt } from "./InstructionPrompt.js";
3
- import { ChatPrompt } from "./chat/ChatPrompt.js";
3
+ import { TextGenerationPromptFormat } from "./TextGenerationPromptFormat.js";
4
4
  /**
5
5
  * Formats an instruction prompt as a basic text prompt.
6
6
  */
7
- export declare const mapInstructionPromptToTextFormat: () => PromptFormat<InstructionPrompt, string>;
7
+ export declare const mapInstructionPromptToTextFormat: () => TextGenerationPromptFormat<InstructionPrompt, string>;
8
8
  /**
9
9
  * Formats a chat prompt as a basic text prompt.
10
10
  *
@@ -14,4 +14,4 @@ export declare const mapInstructionPromptToTextFormat: () => PromptFormat<Instru
14
14
  export declare const mapChatPromptToTextFormat: (options?: {
15
15
  user?: string;
16
16
  ai?: string;
17
- }) => PromptFormat<ChatPrompt, string>;
17
+ }) => TextGenerationPromptFormat<ChatPrompt, string>;
@@ -1,4 +1,4 @@
1
- import { validateChatPrompt } from "./chat/validateChatPrompt.js";
1
+ import { validateChatPrompt } from "./validateChatPrompt.js";
2
2
  /**
3
3
  * Formats an instruction prompt as a basic text prompt.
4
4
  */
@@ -1,7 +1,7 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.mapChatPromptToVicunaFormat = void 0;
4
- const validateChatPrompt_js_1 = require("./chat/validateChatPrompt.cjs");
4
+ const validateChatPrompt_js_1 = require("./validateChatPrompt.cjs");
5
5
  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.";
6
6
  /**
7
7
  * Formats a chat prompt as a Vicuna prompt.
@@ -1,5 +1,5 @@
1
- import { PromptFormat } from "./PromptFormat.js";
2
- import { ChatPrompt } from "./chat/ChatPrompt.js";
1
+ import { ChatPrompt } from "./ChatPrompt.js";
2
+ import { TextGenerationPromptFormat } from "./TextGenerationPromptFormat.js";
3
3
  /**
4
4
  * Formats a chat prompt as a Vicuna prompt.
5
5
  *
@@ -13,4 +13,4 @@ import { ChatPrompt } from "./chat/ChatPrompt.js";
13
13
  * ASSISTANT:
14
14
  * ```
15
15
  */
16
- export declare function mapChatPromptToVicunaFormat(): PromptFormat<ChatPrompt, string>;
16
+ export declare function mapChatPromptToVicunaFormat(): TextGenerationPromptFormat<ChatPrompt, string>;
@@ -1,4 +1,4 @@
1
- import { validateChatPrompt } from "./chat/validateChatPrompt.js";
1
+ import { validateChatPrompt } from "./validateChatPrompt.js";
2
2
  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.";
3
3
  /**
4
4
  * Formats a chat prompt as a Vicuna prompt.
@@ -2,6 +2,7 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.generateText = void 0;
4
4
  const executeCall_js_1 = require("../executeCall.cjs");
5
+ const ModelFunctionPromise_js_1 = require("../ModelFunctionPromise.cjs");
5
6
  /**
6
7
  * Generates a text using a prompt.
7
8
  * The prompt format depends on the model.
@@ -16,7 +17,7 @@ const executeCall_js_1 = require("../executeCall.cjs");
16
17
  * );
17
18
  */
18
19
  function generateText(model, prompt, options) {
19
- return (0, executeCall_js_1.executeCall)({
20
+ return new ModelFunctionPromise_js_1.ModelFunctionPromise((0, executeCall_js_1.executeCall)({
20
21
  functionType: "text-generation",
21
22
  input: prompt,
22
23
  model,
@@ -26,10 +27,12 @@ function generateText(model, prompt, options) {
26
27
  const shouldTrimWhitespace = model.settings.trimWhitespace ?? true;
27
28
  return {
28
29
  response: result.response,
29
- extractedValue: shouldTrimWhitespace ? result.text.trim() : result.text,
30
+ extractedValue: shouldTrimWhitespace
31
+ ? result.text.trim()
32
+ : result.text,
30
33
  usage: result.usage,
31
34
  };
32
35
  },
33
- });
36
+ }));
34
37
  }
35
38
  exports.generateText = generateText;
@@ -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 { TextGenerationModel, TextGenerationModelSettings } from "./TextGenerationModel.js";
4
4
  /**
5
5
  * Generates a text using a prompt.
@@ -1,4 +1,5 @@
1
1
  import { executeCall } from "../executeCall.js";
2
+ import { ModelFunctionPromise } from "../ModelFunctionPromise.js";
2
3
  /**
3
4
  * Generates a text using a prompt.
4
5
  * The prompt format depends on the model.
@@ -13,7 +14,7 @@ import { executeCall } from "../executeCall.js";
13
14
  * );
14
15
  */
15
16
  export function generateText(model, prompt, options) {
16
- return executeCall({
17
+ return new ModelFunctionPromise(executeCall({
17
18
  functionType: "text-generation",
18
19
  input: prompt,
19
20
  model,
@@ -23,9 +24,11 @@ export function generateText(model, prompt, options) {
23
24
  const shouldTrimWhitespace = model.settings.trimWhitespace ?? true;
24
25
  return {
25
26
  response: result.response,
26
- extractedValue: shouldTrimWhitespace ? result.text.trim() : result.text,
27
+ extractedValue: shouldTrimWhitespace
28
+ ? result.text.trim()
29
+ : result.text,
27
30
  usage: result.usage,
28
31
  };
29
32
  },
30
- });
33
+ }));
31
34
  }
@@ -15,13 +15,18 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
15
15
  };
16
16
  Object.defineProperty(exports, "__esModule", { value: true });
17
17
  __exportStar(require("./AlpacaPromptFormat.cjs"), exports);
18
+ __exportStar(require("./ChatPrompt.cjs"), exports);
18
19
  __exportStar(require("./InstructionPrompt.cjs"), exports);
19
20
  __exportStar(require("./Llama2PromptFormat.cjs"), exports);
20
- __exportStar(require("./PromptFormat.cjs"), exports);
21
21
  __exportStar(require("./PromptFormatTextGenerationModel.cjs"), exports);
22
22
  __exportStar(require("./PromptFormatTextStreamingModel.cjs"), exports);
23
+ __exportStar(require("./TextGenerationEvent.cjs"), exports);
24
+ __exportStar(require("./TextGenerationModel.cjs"), exports);
25
+ __exportStar(require("./TextGenerationPromptFormat.cjs"), exports);
23
26
  __exportStar(require("./TextPromptFormat.cjs"), exports);
27
+ __exportStar(require("./TextStreamingEvent.cjs"), exports);
24
28
  __exportStar(require("./VicunaPromptFormat.cjs"), exports);
25
- __exportStar(require("./chat/ChatPrompt.cjs"), exports);
26
- __exportStar(require("./chat/trimChatPrompt.cjs"), exports);
27
- __exportStar(require("./chat/validateChatPrompt.cjs"), exports);
29
+ __exportStar(require("./generateText.cjs"), exports);
30
+ __exportStar(require("./streamText.cjs"), exports);
31
+ __exportStar(require("./trimChatPrompt.cjs"), exports);
32
+ __exportStar(require("./validateChatPrompt.cjs"), exports);
@@ -0,0 +1,16 @@
1
+ export * from "./AlpacaPromptFormat.js";
2
+ export * from "./ChatPrompt.js";
3
+ export * from "./InstructionPrompt.js";
4
+ export * from "./Llama2PromptFormat.js";
5
+ export * from "./PromptFormatTextGenerationModel.js";
6
+ export * from "./PromptFormatTextStreamingModel.js";
7
+ export * from "./TextGenerationEvent.js";
8
+ export * from "./TextGenerationModel.js";
9
+ export * from "./TextGenerationPromptFormat.js";
10
+ export * from "./TextPromptFormat.js";
11
+ export * from "./TextStreamingEvent.js";
12
+ export * from "./VicunaPromptFormat.js";
13
+ export * from "./generateText.js";
14
+ export * from "./streamText.js";
15
+ export * from "./trimChatPrompt.js";
16
+ export * from "./validateChatPrompt.js";
@@ -0,0 +1,16 @@
1
+ export * from "./AlpacaPromptFormat.js";
2
+ export * from "./ChatPrompt.js";
3
+ export * from "./InstructionPrompt.js";
4
+ export * from "./Llama2PromptFormat.js";
5
+ export * from "./PromptFormatTextGenerationModel.js";
6
+ export * from "./PromptFormatTextStreamingModel.js";
7
+ export * from "./TextGenerationEvent.js";
8
+ export * from "./TextGenerationModel.js";
9
+ export * from "./TextGenerationPromptFormat.js";
10
+ export * from "./TextPromptFormat.js";
11
+ export * from "./TextStreamingEvent.js";
12
+ export * from "./VicunaPromptFormat.js";
13
+ export * from "./generateText.js";
14
+ export * from "./streamText.js";
15
+ export * from "./trimChatPrompt.js";
16
+ export * from "./validateChatPrompt.js";
@@ -1,5 +1,5 @@
1
- import { HasContextWindowSize, HasTokenizer, TextGenerationModel, TextGenerationModelSettings } from "../../model-function/generate-text/TextGenerationModel.js";
2
1
  import { ChatPrompt } from "./ChatPrompt.js";
2
+ import { HasContextWindowSize, HasTokenizer, TextGenerationModel, TextGenerationModelSettings } from "./TextGenerationModel.js";
3
3
  /**
4
4
  * Keeps only the most recent messages in the prompt, while leaving enough space for the completion.
5
5
  *
@@ -19,6 +19,7 @@ __exportStar(require("./Model.cjs"), exports);
19
19
  __exportStar(require("./ModelCallEvent.cjs"), exports);
20
20
  __exportStar(require("./ModelCallMetadata.cjs"), exports);
21
21
  __exportStar(require("./ModelInformation.cjs"), exports);
22
+ __exportStar(require("./PromptFormat.cjs"), exports);
22
23
  __exportStar(require("./SuccessfulModelCall.cjs"), exports);
23
24
  __exportStar(require("./describe-image/ImageDescriptionEvent.cjs"), exports);
24
25
  __exportStar(require("./describe-image/ImageDescriptionModel.cjs"), exports);
@@ -28,6 +29,7 @@ __exportStar(require("./embed/EmbeddingModel.cjs"), exports);
28
29
  __exportStar(require("./embed/embed.cjs"), exports);
29
30
  __exportStar(require("./generate-image/ImageGenerationEvent.cjs"), exports);
30
31
  __exportStar(require("./generate-image/ImageGenerationModel.cjs"), exports);
32
+ __exportStar(require("./generate-image/PromptFormatImageGenerationModel.cjs"), exports);
31
33
  __exportStar(require("./generate-image/generateImage.cjs"), exports);
32
34
  __exportStar(require("./generate-structure/NoSuchStructureError.cjs"), exports);
33
35
  __exportStar(require("./generate-structure/StructureFromTextGenerationModel.cjs"), exports);
@@ -40,11 +42,7 @@ __exportStar(require("./generate-structure/StructureValidationError.cjs"), expor
40
42
  __exportStar(require("./generate-structure/generateStructure.cjs"), exports);
41
43
  __exportStar(require("./generate-structure/generateStructureOrText.cjs"), exports);
42
44
  __exportStar(require("./generate-structure/streamStructure.cjs"), exports);
43
- __exportStar(require("./generate-text/TextGenerationEvent.cjs"), exports);
44
- __exportStar(require("./generate-text/TextGenerationModel.cjs"), exports);
45
- __exportStar(require("./generate-text/TextStreamingEvent.cjs"), exports);
46
- __exportStar(require("./generate-text/generateText.cjs"), exports);
47
- __exportStar(require("./generate-text/streamText.cjs"), exports);
45
+ __exportStar(require("./generate-text/index.cjs"), exports);
48
46
  __exportStar(require("./synthesize-speech/SpeechSynthesisEvent.cjs"), exports);
49
47
  __exportStar(require("./synthesize-speech/SpeechSynthesisModel.cjs"), exports);
50
48
  __exportStar(require("./synthesize-speech/synthesizeSpeech.cjs"), exports);
@@ -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
  }