modelfusion 0.121.1 → 0.122.0

package/model-function/generate-text/generateText.cjs

@@ -2,8 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateText = void 0;
 const executeStandardCall_js_1 = require("../executeStandardCall.cjs");
-async function generateText(model, prompt, options) {
-    const fullResponse = await (0, executeStandardCall_js_1.executeStandardCall)({
+async function generateText({ model, prompt, fullResponse, ...options }) {
+    const callResponse = await (0, executeStandardCall_js_1.executeStandardCall)({
         functionType: "generate-text",
         input: prompt,
         model,
@@ -66,16 +66,16 @@ async function generateText(model, prompt, options) {
             };
         },
     });
-    const textGenerationResults = fullResponse.value;
+    const textGenerationResults = callResponse.value;
     const firstResult = textGenerationResults[0];
-    return options?.fullResponse
+    return fullResponse
         ? {
             text: firstResult.text,
             finishReason: firstResult.finishReason,
             texts: textGenerationResults.map((textGeneration) => textGeneration.text),
             textGenerationResults,
-            rawResponse: fullResponse.rawResponse,
-            metadata: fullResponse.metadata,
+            rawResponse: callResponse.rawResponse,
+            metadata: callResponse.metadata,
         }
         : firstResult.text;
 }

package/model-function/generate-text/generateText.d.ts

@@ -12,23 +12,26 @@ import { TextGenerationFinishReason, TextGenerationResult } from "./TextGenerati
  * @see https://modelfusion.dev/guide/function/generate-text
  *
  * @example
- * const text = await generateText(
- *   openai.CompletionTextGenerator(...),
- *   "Write a short story about a robot learning to love:\n\n"
- * );
+ * const text = await generateText({
+ *   model: openai.CompletionTextGenerator(...),
+ *   prompt: "Write a short story about a robot learning to love:\n\n"
+ * });
  *
  * @param {TextGenerationModel<PROMPT, TextGenerationModelSettings>} model - The text generation model to use.
  * @param {PROMPT} prompt - The prompt to use for text generation.
- * @param {FunctionOptions} [options] - Optional parameters for the function.
  *
  * @returns {Promise<string>} - A promise that resolves to the generated text.
  */
-export declare function generateText<PROMPT>(model: TextGenerationModel<PROMPT, TextGenerationModelSettings>, prompt: PROMPT, options?: FunctionOptions & {
+export declare function generateText<PROMPT>(args: {
+    model: TextGenerationModel<PROMPT, TextGenerationModelSettings>;
+    prompt: PROMPT;
     fullResponse?: false;
-}): Promise<string>;
-export declare function generateText<PROMPT>(model: TextGenerationModel<PROMPT, TextGenerationModelSettings>, prompt: PROMPT, options: FunctionOptions & {
+} & FunctionOptions): Promise<string>;
+export declare function generateText<PROMPT>(args: {
+    model: TextGenerationModel<PROMPT, TextGenerationModelSettings>;
+    prompt: PROMPT;
     fullResponse: true;
-}): Promise<{
+} & FunctionOptions): Promise<{
     text: string;
     finishReason: TextGenerationFinishReason;
     texts: string[];

package/model-function/generate-text/generateText.js

@@ -1,6 +1,6 @@
 import { executeStandardCall } from "../executeStandardCall.js";
-export async function generateText(model, prompt, options) {
-    const fullResponse = await executeStandardCall({
+export async function generateText({ model, prompt, fullResponse, ...options }) {
+    const callResponse = await executeStandardCall({
         functionType: "generate-text",
         input: prompt,
         model,
@@ -63,16 +63,16 @@ export async function generateText(model, prompt, options) {
             };
         },
     });
-    const textGenerationResults = fullResponse.value;
+    const textGenerationResults = callResponse.value;
     const firstResult = textGenerationResults[0];
-    return options?.fullResponse
+    return fullResponse
         ? {
             text: firstResult.text,
             finishReason: firstResult.finishReason,
             texts: textGenerationResults.map((textGeneration) => textGeneration.text),
             textGenerationResults,
-            rawResponse: fullResponse.rawResponse,
-            metadata: fullResponse.metadata,
+            rawResponse: callResponse.rawResponse,
+            metadata: callResponse.metadata,
         }
         : firstResult.text;
 }

package/model-function/generate-text/streamText.cjs

@@ -2,7 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.streamText = void 0;
 const executeStreamCall_js_1 = require("../executeStreamCall.cjs");
-async function streamText(model, prompt, options) {
+async function streamText({ model, prompt, fullResponse, ...options }) {
     const shouldTrimWhitespace = model.settings.trimWhitespace ?? true;
     let accumulatedText = "";
     let isFirstDelta = true;
@@ -11,7 +11,7 @@ async function streamText(model, prompt, options) {
     const textPromise = new Promise((resolve) => {
         resolveText = resolve;
     });
-    const fullResponse = await (0, executeStreamCall_js_1.executeStreamCall)({
+    const callResponse = await (0, executeStreamCall_js_1.executeStreamCall)({
         functionType: "stream-text",
         input: prompt,
         model,
@@ -43,12 +43,12 @@ async function streamText(model, prompt, options) {
             resolveText(accumulatedText);
         },
     });
-    return options?.fullResponse
+    return fullResponse
         ? {
-            textStream: fullResponse.value,
+            textStream: callResponse.value,
             text: textPromise,
-            metadata: fullResponse.metadata,
+            metadata: callResponse.metadata,
         }
-        : fullResponse.value;
+        : callResponse.value;
 }
 exports.streamText = streamText;

package/model-function/generate-text/streamText.d.ts

@@ -11,10 +11,10 @@ import { TextStreamingModel } from "./TextGenerationModel.js";
  * @see https://modelfusion.dev/guide/function/generate-text
  *
  * @example
- * const textStream = await streamText(
- *   openai.CompletionTextGenerator(...),
- *   "Write a short story about a robot learning to love:\n\n"
- * );
+ * const textStream = await streamText({
+ *   model: openai.CompletionTextGenerator(...),
+ *   prompt: "Write a short story about a robot learning to love:\n\n"
+ * });
  *
  * for await (const textPart of textStream) {
  *   // ...
@@ -26,12 +26,16 @@ import { TextStreamingModel } from "./TextGenerationModel.js";
  *
  * @returns {AsyncIterableResultPromise<string>} An async iterable promise that yields the generated text.
  */
-export declare function streamText<PROMPT>(model: TextStreamingModel<PROMPT>, prompt: PROMPT, options?: FunctionOptions & {
+export declare function streamText<PROMPT>(args: {
+    model: TextStreamingModel<PROMPT>;
+    prompt: PROMPT;
     fullResponse?: false;
-}): Promise<AsyncIterable<string>>;
-export declare function streamText<PROMPT>(model: TextStreamingModel<PROMPT>, prompt: PROMPT, options: FunctionOptions & {
+} & FunctionOptions): Promise<AsyncIterable<string>>;
+export declare function streamText<PROMPT>(args: {
+    model: TextStreamingModel<PROMPT>;
+    prompt: PROMPT;
     fullResponse: true;
-}): Promise<{
+} & FunctionOptions): Promise<{
     textStream: AsyncIterable<string>;
     text: PromiseLike<string>;
     metadata: Omit<ModelCallMetadata, "durationInMs" | "finishTimestamp">;

package/model-function/generate-text/streamText.js

@@ -1,5 +1,5 @@
 import { executeStreamCall } from "../executeStreamCall.js";
-export async function streamText(model, prompt, options) {
+export async function streamText({ model, prompt, fullResponse, ...options }) {
     const shouldTrimWhitespace = model.settings.trimWhitespace ?? true;
     let accumulatedText = "";
     let isFirstDelta = true;
@@ -8,7 +8,7 @@ export async function streamText(model, prompt, options) {
     const textPromise = new Promise((resolve) => {
         resolveText = resolve;
     });
-    const fullResponse = await executeStreamCall({
+    const callResponse = await executeStreamCall({
         functionType: "stream-text",
         input: prompt,
         model,
@@ -40,11 +40,11 @@ export async function streamText(model, prompt, options) {
             resolveText(accumulatedText);
         },
     });
-    return options?.fullResponse
+    return fullResponse
         ? {
-            textStream: fullResponse.value,
+            textStream: callResponse.value,
             text: textPromise,
-            metadata: fullResponse.metadata,
+            metadata: callResponse.metadata,
         }
-        : fullResponse.value;
+        : callResponse.value;
 }

package/model-function/generate-transcription/generateTranscription.cjs

@@ -2,8 +2,8 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generateTranscription = void 0;
 const executeStandardCall_js_1 = require("../executeStandardCall.cjs");
-async function generateTranscription(model, data, options) {
-    const fullResponse = await (0, executeStandardCall_js_1.executeStandardCall)({
+async function generateTranscription({ model, data, fullResponse, ...options }) {
+    const callResponse = await (0, executeStandardCall_js_1.executeStandardCall)({
         functionType: "generate-transcription",
         input: data,
         model,
@@ -16,6 +16,6 @@ async function generateTranscription(model, data, options) {
             };
         },
     });
-    return options?.fullResponse ? fullResponse : fullResponse.value;
+    return fullResponse ? callResponse : callResponse.value;
 }
 exports.generateTranscription = generateTranscription;

package/model-function/generate-transcription/generateTranscription.d.ts

@@ -9,23 +9,26 @@ import { TranscriptionModel, TranscriptionModelSettings } from "./TranscriptionM
  * @example
  * const data = await fs.promises.readFile("data/test.mp3");
  *
- * const transcription = await generateTranscription(
- *   openai.Transcriber({ model: "whisper-1" }),
- *   { type: "mp3", data }
- * );
+ * const transcription = await generateTranscription({
+ *   model: openai.Transcriber({ model: "whisper-1" }),
+ *   data: { type: "mp3", data }
+ * });
  *
  * @param {TranscriptionModel<DATA, TranscriptionModelSettings>} model - The model to use for transcription.
  * @param {DATA} data - The data to transcribe.
- * @param {FunctionOptions} [options] - Optional parameters for the function.
  *
  * @returns {Promise<string>} A promise that resolves to the transcribed text.
  */
-export declare function generateTranscription<DATA>(model: TranscriptionModel<DATA, TranscriptionModelSettings>, data: DATA, options?: FunctionOptions & {
+export declare function generateTranscription<DATA>(args: {
+    model: TranscriptionModel<DATA, TranscriptionModelSettings>;
+    data: DATA;
     fullResponse?: false;
-}): Promise<string>;
-export declare function generateTranscription<DATA>(model: TranscriptionModel<DATA, TranscriptionModelSettings>, data: DATA, options: FunctionOptions & {
+} & FunctionOptions): Promise<string>;
+export declare function generateTranscription<DATA>(args: {
+    model: TranscriptionModel<DATA, TranscriptionModelSettings>;
+    data: DATA;
     fullResponse: true;
-}): Promise<{
+} & FunctionOptions): Promise<{
     value: string;
     rawResponse: unknown;
     metadata: ModelCallMetadata;

package/model-function/generate-transcription/generateTranscription.js

@@ -1,6 +1,6 @@
 import { executeStandardCall } from "../executeStandardCall.js";
-export async function generateTranscription(model, data, options) {
-    const fullResponse = await executeStandardCall({
+export async function generateTranscription({ model, data, fullResponse, ...options }) {
+    const callResponse = await executeStandardCall({
         functionType: "generate-transcription",
         input: data,
         model,
@@ -13,5 +13,5 @@ export async function generateTranscription(model, data, options) {
             };
         },
     });
-    return options?.fullResponse ? fullResponse : fullResponse.value;
+    return fullResponse ? callResponse : callResponse.value;
 }

package/model-provider/cohere/CohereTextGenerationModel.d.ts

@@ -60,8 +60,8 @@ export declare class CohereTextGenerationModel extends AbstractModel<CohereTextG
     get settingsForEvent(): Partial<CohereTextGenerationModelSettings>;
     doGenerateTexts(prompt: string, options: FunctionCallOptions): Promise<{
         rawResponse: {
-            id: string;
             prompt: string;
+            id: string;
             generations: {
                 text: string;
                 id: string;
@@ -80,8 +80,8 @@ export declare class CohereTextGenerationModel extends AbstractModel<CohereTextG
     }>;
     restoreGeneratedTexts(rawResponse: unknown): {
         rawResponse: {
-            id: string;
             prompt: string;
+            id: string;
             generations: {
                 text: string;
                 id: string;
@@ -100,8 +100,8 @@ export declare class CohereTextGenerationModel extends AbstractModel<CohereTextG
     };
     processTextGenerationResponse(rawResponse: CohereTextGenerationResponse): {
         rawResponse: {
-            id: string;
             prompt: string;
+            id: string;
             generations: {
                 text: string;
                 id: string;
@@ -124,8 +124,8 @@ export declare class CohereTextGenerationModel extends AbstractModel<CohereTextG
         is_finished: false;
     } | {
         response: {
-            id: string;
             prompt: string;
+            id: string;
             generations: {
                 text: string;
                 id: string;
@@ -190,8 +190,8 @@ declare const cohereTextGenerationResponseSchema: z.ZodObject<{
         };
     }>>;
 }, "strip", z.ZodTypeAny, {
-    id: string;
     prompt: string;
+    id: string;
     generations: {
         text: string;
         id: string;
@@ -203,8 +203,8 @@ declare const cohereTextGenerationResponseSchema: z.ZodObject<{
         };
     } | undefined;
 }, {
-    id: string;
     prompt: string;
+    id: string;
     generations: {
         text: string;
         id: string;
@@ -263,8 +263,8 @@ declare const cohereTextStreamChunkSchema: z.ZodDiscriminatedUnion<"is_finished"
             };
         }>>;
     }, "strip", z.ZodTypeAny, {
-        id: string;
         prompt: string;
+        id: string;
         generations: {
             text: string;
             id: string;
@@ -276,8 +276,8 @@ declare const cohereTextStreamChunkSchema: z.ZodDiscriminatedUnion<"is_finished"
             };
         } | undefined;
     }, {
-        id: string;
         prompt: string;
+        id: string;
         generations: {
             text: string;
             id: string;
@@ -291,8 +291,8 @@ declare const cohereTextStreamChunkSchema: z.ZodDiscriminatedUnion<"is_finished"
     }>;
 }, "strip", z.ZodTypeAny, {
     response: {
-        id: string;
         prompt: string;
+        id: string;
         generations: {
             text: string;
             id: string;
@@ -308,8 +308,8 @@ declare const cohereTextStreamChunkSchema: z.ZodDiscriminatedUnion<"is_finished"
     is_finished: true;
 }, {
     response: {
-        id: string;
         prompt: string;
+        id: string;
         generations: {
             text: string;
             id: string;
@@ -336,8 +336,8 @@ export declare const CohereTextGenerationResponseFormat: {
     json: {
         stream: boolean;
         handler: ResponseHandler<{
-            id: string;
             prompt: string;
+            id: string;
             generations: {
                 text: string;
                 id: string;
@@ -363,8 +363,8 @@ export declare const CohereTextGenerationResponseFormat: {
             is_finished: false;
         } | {
             response: {
-                id: string;
                 prompt: string;
+                id: string;
                 generations: {
                     text: string;
                     id: string;

package/model-provider/cohere/CohereTextGenerationModel.test.cjs

@@ -19,10 +19,13 @@ describe("streamText", () => {
                 `"text":"Hello, world!","finish_reason":"COMPLETE"}],` +
                 `"prompt":"hello"}}\n`,
         ];
-        const stream = await (0, streamText_js_1.streamText)(new CohereTextGenerationModel_js_1.CohereTextGenerationModel({
-            api: new CohereApiConfiguration_js_1.CohereApiConfiguration({ apiKey: "test-key" }),
-            model: "command-light",
-        }), "hello");
+        const stream = await (0, streamText_js_1.streamText)({
+            model: new CohereTextGenerationModel_js_1.CohereTextGenerationModel({
+                api: new CohereApiConfiguration_js_1.CohereApiConfiguration({ apiKey: "test-key" }),
+                model: "command-light",
+            }),
+            prompt: "hello",
+        });
         // note: space moved to last chunk bc of trimming
         expect(await (0, arrayFromAsync_js_1.arrayFromAsync)(stream)).toStrictEqual([
             "Hello",

package/model-provider/cohere/CohereTextGenerationModel.test.js

@@ -17,10 +17,13 @@ describe("streamText", () => {
                 `"text":"Hello, world!","finish_reason":"COMPLETE"}],` +
                 `"prompt":"hello"}}\n`,
         ];
-        const stream = await streamText(new CohereTextGenerationModel({
-            api: new CohereApiConfiguration({ apiKey: "test-key" }),
-            model: "command-light",
-        }), "hello");
+        const stream = await streamText({
+            model: new CohereTextGenerationModel({
+                api: new CohereApiConfiguration({ apiKey: "test-key" }),
+                model: "command-light",
+            }),
+            prompt: "hello",
+        });
         // note: space moved to last chunk bc of trimming
         expect(await arrayFromAsync(stream)).toStrictEqual([
             "Hello",

package/model-provider/llamacpp/LlamaCppCompletionModel.d.ts

@@ -157,9 +157,9 @@ export declare class LlamaCppCompletionModel<CONTEXT_WINDOW_SIZE extends number
     doGenerateTexts(prompt: LlamaCppCompletionPrompt, options: FunctionCallOptions): Promise<{
         rawResponse: {
             model: string;
+            prompt: string;
             stop: true;
             content: string;
-            prompt: string;
             generation_settings: {
                 model: string;
                 stream: boolean;
@@ -217,9 +217,9 @@ export declare class LlamaCppCompletionModel<CONTEXT_WINDOW_SIZE extends number
     restoreGeneratedTexts(rawResponse: unknown): {
         rawResponse: {
             model: string;
+            prompt: string;
             stop: true;
             content: string;
-            prompt: string;
             generation_settings: {
                 model: string;
                 stream: boolean;
@@ -277,9 +277,9 @@ export declare class LlamaCppCompletionModel<CONTEXT_WINDOW_SIZE extends number
     processTextGenerationResponse(rawResponse: LlamaCppTextGenerationResponse): {
         rawResponse: {
             model: string;
+            prompt: string;
             stop: true;
             content: string;
-            prompt: string;
             generation_settings: {
                 model: string;
                 stream: boolean;
@@ -336,9 +336,9 @@ export declare class LlamaCppCompletionModel<CONTEXT_WINDOW_SIZE extends number
     };
     doStreamText(prompt: LlamaCppCompletionPrompt, options: FunctionCallOptions): Promise<AsyncIterable<Delta<{
         model: string;
+        prompt: string;
         stop: true;
         content: string;
-        prompt: string;
         generation_settings: {
             model: string;
             stream: boolean;
@@ -515,9 +515,9 @@ declare const llamaCppTextGenerationResponseSchema: z.ZodObject<{
     truncated: z.ZodBoolean;
 }, "strip", z.ZodTypeAny, {
     model: string;
+    prompt: string;
     stop: true;
     content: string;
-    prompt: string;
     generation_settings: {
         model: string;
         stream: boolean;
@@ -563,9 +563,9 @@ declare const llamaCppTextGenerationResponseSchema: z.ZodObject<{
     truncated: boolean;
 }, {
     model: string;
+    prompt: string;
     stop: true;
     content: string;
-    prompt: string;
     generation_settings: {
         model: string;
         stream: boolean;
@@ -736,9 +736,9 @@ declare const llamaCppTextStreamChunkSchema: z.ZodDiscriminatedUnion<"stop", [z.
     truncated: z.ZodBoolean;
 }, "strip", z.ZodTypeAny, {
     model: string;
+    prompt: string;
     stop: true;
     content: string;
-    prompt: string;
     generation_settings: {
         model: string;
         stream: boolean;
@@ -784,9 +784,9 @@ declare const llamaCppTextStreamChunkSchema: z.ZodDiscriminatedUnion<"stop", [z.
     truncated: boolean;
 }, {
     model: string;
+    prompt: string;
     stop: true;
     content: string;
-    prompt: string;
     generation_settings: {
         model: string;
         stream: boolean;
@@ -844,9 +844,9 @@ export declare const LlamaCppCompletionResponseFormat: {
         stream: false;
         handler: ResponseHandler<{
             model: string;
+            prompt: string;
             stop: true;
             content: string;
-            prompt: string;
             generation_settings: {
                 model: string;
                 stream: boolean;
@@ -902,9 +902,9 @@ export declare const LlamaCppCompletionResponseFormat: {
             response: Response;
         }) => Promise<AsyncIterable<Delta<{
             model: string;
+            prompt: string;
             stop: true;
             content: string;
-            prompt: string;
             generation_settings: {
                 model: string;
                 stream: boolean;

package/model-provider/llamacpp/LlamaCppCompletionModel.test.cjs

@@ -26,7 +26,10 @@ describe("streamText", () => {
                 `"prompt_per_token_ms":48.845600000000005},"tokens_cached":74,"tokens_evaluated":5,` +
                 `"tokens_predicted":69,"truncated":false}\n\n`,
         ];
-        const stream = await (0, streamText_js_1.streamText)(new LlamaCppCompletionModel_js_1.LlamaCppCompletionModel().withTextPrompt(), "hello");
+        const stream = await (0, streamText_js_1.streamText)({
+            model: new LlamaCppCompletionModel_js_1.LlamaCppCompletionModel().withTextPrompt(),
+            prompt: "hello",
+        });
         // note: space moved to last chunk bc of trimming
         expect(await (0, arrayFromAsync_js_1.arrayFromAsync)(stream)).toStrictEqual([
             "Hello",

package/model-provider/llamacpp/LlamaCppCompletionModel.test.js

@@ -24,7 +24,10 @@ describe("streamText", () => {
                 `"prompt_per_token_ms":48.845600000000005},"tokens_cached":74,"tokens_evaluated":5,` +
                 `"tokens_predicted":69,"truncated":false}\n\n`,
         ];
-        const stream = await streamText(new LlamaCppCompletionModel().withTextPrompt(), "hello");
+        const stream = await streamText({
+            model: new LlamaCppCompletionModel().withTextPrompt(),
+            prompt: "hello",
+        });
         // note: space moved to last chunk bc of trimming
         expect(await arrayFromAsync(stream)).toStrictEqual([
             "Hello",

package/model-provider/mistral/MistralChatModel.test.cjs

@@ -29,10 +29,13 @@ describe("streamText", () => {
             ];
         });
         it("should return a text stream", async () => {
-            const stream = await (0, streamText_js_1.streamText)(new MistralChatModel_js_1.MistralChatModel({
-                api: new MistralApiConfiguration_js_1.MistralApiConfiguration({ apiKey: "test-key" }),
-                model: "mistral-tiny",
-            }).withTextPrompt(), "hello");
+            const stream = await (0, streamText_js_1.streamText)({
+                model: new MistralChatModel_js_1.MistralChatModel({
+                    api: new MistralApiConfiguration_js_1.MistralApiConfiguration({ apiKey: "test-key" }),
+                    model: "mistral-tiny",
+                }).withTextPrompt(),
+                prompt: "hello",
+            });
             // note: space moved to last chunk bc of trimming
             expect(await (0, arrayFromAsync_js_1.arrayFromAsync)(stream)).toStrictEqual([
                 "Hello",
@@ -41,10 +44,14 @@ describe("streamText", () => {
             ]);
         });
         it("should return text", async () => {
-            const { text } = await (0, streamText_js_1.streamText)(new MistralChatModel_js_1.MistralChatModel({
-                api: new MistralApiConfiguration_js_1.MistralApiConfiguration({ apiKey: "test-key" }),
-                model: "mistral-tiny",
-            }).withTextPrompt(), "hello", { fullResponse: true });
+            const { text } = await (0, streamText_js_1.streamText)({
+                model: new MistralChatModel_js_1.MistralChatModel({
+                    api: new MistralApiConfiguration_js_1.MistralApiConfiguration({ apiKey: "test-key" }),
+                    model: "mistral-tiny",
+                }).withTextPrompt(),
+                prompt: "hello",
+                fullResponse: true,
+            });
             expect(await text).toStrictEqual("Hello, world!");
         });
     });

package/model-provider/mistral/MistralChatModel.test.js

@@ -27,10 +27,13 @@ describe("streamText", () => {
             ];
         });
         it("should return a text stream", async () => {
-            const stream = await streamText(new MistralChatModel({
-                api: new MistralApiConfiguration({ apiKey: "test-key" }),
-                model: "mistral-tiny",
-            }).withTextPrompt(), "hello");
+            const stream = await streamText({
+                model: new MistralChatModel({
+                    api: new MistralApiConfiguration({ apiKey: "test-key" }),
+                    model: "mistral-tiny",
+                }).withTextPrompt(),
+                prompt: "hello",
+            });
             // note: space moved to last chunk bc of trimming
             expect(await arrayFromAsync(stream)).toStrictEqual([
                 "Hello",
@@ -39,10 +42,14 @@ describe("streamText", () => {
             ]);
         });
         it("should return text", async () => {
-            const { text } = await streamText(new MistralChatModel({
-                api: new MistralApiConfiguration({ apiKey: "test-key" }),
-                model: "mistral-tiny",
-            }).withTextPrompt(), "hello", { fullResponse: true });
+            const { text } = await streamText({
+                model: new MistralChatModel({
+                    api: new MistralApiConfiguration({ apiKey: "test-key" }),
+                    model: "mistral-tiny",
+                }).withTextPrompt(),
+                prompt: "hello",
+                fullResponse: true,
+            });
             expect(await text).toStrictEqual("Hello, world!");
         });
     });

package/model-provider/ollama/OllamaChatModel.test.cjs

@@ -16,7 +16,12 @@ describe("streamText", () => {
                 `"done":true,"total_duration":4843619375,"load_duration":1101458,"prompt_eval_count":5,"prompt_eval_duration":199339000,` +
                 `"eval_count":317,"eval_duration":4639772000}\n`,
         ];
-        const stream = await (0, streamText_js_1.streamText)(new OllamaChatModel_js_1.OllamaChatModel({ model: "mistral:text" }).withTextPrompt(), "hello");
+        const stream = await (0, streamText_js_1.streamText)({
+            model: new OllamaChatModel_js_1.OllamaChatModel({
+                model: "mistral:text",
+            }).withTextPrompt(),
+            prompt: "hello",
+        });
         // note: space moved to last chunk bc of trimming
         expect(await (0, arrayFromAsync_js_1.arrayFromAsync)(stream)).toStrictEqual([
             "Hello",