modelfusion 0.21.0 → 0.23.0

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

@@ -1,17 +1,17 @@
 import { ModelFunctionOptions } from "../ModelFunctionOptions.js";
-import { CallMetadata } from "../executeCall.js";
+import { ModelCallMetadata } from "../executeCall.js";
 import { DeltaEvent } from "./DeltaEvent.js";
 import { TextGenerationModel, TextGenerationModelSettings } from "./TextGenerationModel.js";
-export declare class StreamTextPromise<PROMPT, FULL_DELTA, SETTINGS extends TextGenerationModelSettings> extends Promise<AsyncIterable<string>> {
+export declare class StreamTextPromise extends Promise<AsyncIterable<string>> {
     private fullPromise;
     private outputPromise;
     constructor(fullPromise: Promise<{
         output: AsyncIterable<string>;
-        metadata: Omit<CallMetadata<TextGenerationModel<PROMPT, unknown, FULL_DELTA, SETTINGS>>, "durationInMs" | "finishTimestamp">;
+        metadata: Omit<ModelCallMetadata, "durationInMs" | "finishTimestamp">;
     }>);
     asFullResponse(): Promise<{
         output: AsyncIterable<string>;
-        metadata: Omit<CallMetadata<TextGenerationModel<PROMPT, unknown, FULL_DELTA, SETTINGS>>, "durationInMs" | "finishTimestamp">;
+        metadata: Omit<ModelCallMetadata, "durationInMs" | "finishTimestamp">;
     }>;
     then<TResult1 = AsyncIterable<string>, TResult2 = never>(onfulfilled?: ((value: AsyncIterable<string>) => TResult1 | PromiseLike<TResult1>) | undefined | null, onrejected?: ((reason: unknown) => TResult2 | PromiseLike<TResult2>) | undefined | null): Promise<TResult1 | TResult2>;
     catch<TResult = never>(onrejected?: ((reason: unknown) => TResult | PromiseLike<TResult>) | undefined | null): Promise<AsyncIterable<string> | TResult>;
@@ -20,4 +20,4 @@ export declare class StreamTextPromise<PROMPT, FULL_DELTA, SETTINGS extends Text
 export declare function streamText<PROMPT, FULL_DELTA, SETTINGS extends TextGenerationModelSettings>(model: TextGenerationModel<PROMPT, unknown, FULL_DELTA, SETTINGS> & {
     generateDeltaStreamResponse: (prompt: PROMPT, options: ModelFunctionOptions<SETTINGS>) => PromiseLike<AsyncIterable<DeltaEvent<FULL_DELTA>>>;
     extractTextDelta: (fullDelta: FULL_DELTA) => string | undefined;
-}, prompt: PROMPT, options?: ModelFunctionOptions<SETTINGS>): StreamTextPromise<PROMPT, FULL_DELTA, SETTINGS>;
+}, prompt: PROMPT, options?: ModelFunctionOptions<SETTINGS>): StreamTextPromise;

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

@@ -1,6 +1,8 @@
 import { nanoid as createId } from "nanoid";
-import { FunctionEventSource } from "../../run/FunctionEventSource.js";
-import { getGlobalFunctionObservers } from "../../run/GlobalFunctionObservers.js";
+import { FunctionEventSource } from "../../core/FunctionEventSource.js";
+import { getGlobalFunctionLogging } from "../../core/GlobalFunctionLogging.js";
+import { getGlobalFunctionObservers } from "../../core/GlobalFunctionObservers.js";
+import { getFunctionCallLogger } from "../../core/getFunctionCallLogger.js";
 import { startDurationMeasurement } from "../../util/DurationMeasurement.js";
 import { AbortError } from "../../util/api/AbortError.js";
 import { runSafe } from "../../util/runSafe.js";
@@ -54,6 +56,7 @@ async function doStreamText(model, prompt, options) {
     const settings = model.settings;
     const eventSource = new FunctionEventSource({
         observers: [
+            ...getFunctionCallLogger(options?.logging ?? getGlobalFunctionLogging()),
             ...getGlobalFunctionObservers(),
             ...(settings.observers ?? []),
             ...(run?.observers ?? []),
@@ -63,21 +66,21 @@ async function doStreamText(model, prompt, options) {
     });
     const durationMeasurement = startDurationMeasurement();
     const startMetadata = {
+        functionType: "text-streaming",
         callId: `call-${createId()}`,
         runId: run?.runId,
         sessionId: run?.sessionId,
         userId: run?.userId,
         functionId: options?.functionId,
         model: model.modelInformation,
+        settings: model.settingsForEvent,
+        input: prompt,
         timestamp: durationMeasurement.startDate,
         startTimestamp: durationMeasurement.startDate,
     };
     eventSource.notify({
-        ...startMetadata,
-        functionType: "text-streaming",
         eventType: "started",
-        settings,
-        prompt,
+        ...startMetadata,
     });
     const result = await runSafe(async () => extractTextDeltas({
         deltaIterable: await model.generateDeltaStreamResponse(prompt, {
@@ -88,70 +91,65 @@ async function doStreamText(model, prompt, options) {
         extractDelta: (fullDelta) => model.extractTextDelta(fullDelta),
         onDone: (fullText, lastFullDelta) => {
             const finishMetadata = {
-                ...startMetadata,
                 eventType: "finished",
+                ...startMetadata,
                 finishTimestamp: new Date(),
                 durationInMs: durationMeasurement.durationInMs,
             };
             eventSource.notify({
                 ...finishMetadata,
-                functionType: "text-streaming",
-                status: "success",
-                settings,
-                prompt,
-                response: lastFullDelta,
-                generatedText: fullText,
+                result: {
+                    status: "success",
+                    response: lastFullDelta,
+                    output: fullText,
+                },
             });
         },
         onError: (error) => {
             const finishMetadata = {
-                ...startMetadata,
                 eventType: "finished",
+                ...startMetadata,
                 finishTimestamp: new Date(),
                 durationInMs: durationMeasurement.durationInMs,
             };
             eventSource.notify(error instanceof AbortError
                 ? {
                     ...finishMetadata,
-                    functionType: "text-streaming",
-                    status: "abort",
-                    settings,
-                    prompt,
+                    result: {
+                        status: "abort",
+                    },
                 }
                 : {
                     ...finishMetadata,
-                    functionType: "text-streaming",
-                    status: "error",
-                    settings,
-                    prompt,
-                    error,
+                    result: {
+                        status: "error",
+                        error,
+                    },
                 });
         },
     }));
     if (!result.ok) {
         const finishMetadata = {
-            ...startMetadata,
             eventType: "finished",
+            ...startMetadata,
             finishTimestamp: new Date(),
             durationInMs: durationMeasurement.durationInMs,
         };
         if (result.isAborted) {
             eventSource.notify({
                 ...finishMetadata,
-                functionType: "text-streaming",
-                status: "abort",
-                settings,
-                prompt,
+                result: {
+                    status: "abort",
+                },
             });
             throw new AbortError();
         }
         eventSource.notify({
             ...finishMetadata,
-            functionType: "text-streaming",
-            status: "error",
-            settings,
-            prompt,
-            error: result.error,
+            result: {
+                status: "error",
+                error: result.error,
+            },
         });
         throw result.error;
     }

package/model-function/index.cjs

@@ -25,10 +25,10 @@ __exportStar(require("./embed-text/embedText.cjs"), exports);
 __exportStar(require("./generate-image/ImageGenerationEvent.cjs"), exports);
 __exportStar(require("./generate-image/ImageGenerationModel.cjs"), exports);
 __exportStar(require("./generate-image/generateImage.cjs"), exports);
-__exportStar(require("./generate-json/GenerateJsonModel.cjs"), exports);
-__exportStar(require("./generate-json/GenerateJsonOrTextModel.cjs"), exports);
 __exportStar(require("./generate-json/InstructionWithSchemaPrompt.cjs"), exports);
 __exportStar(require("./generate-json/JsonGenerationEvent.cjs"), exports);
+__exportStar(require("./generate-json/JsonGenerationModel.cjs"), exports);
+__exportStar(require("./generate-json/JsonOrTextGenerationModel.cjs"), exports);
 __exportStar(require("./generate-json/JsonTextGenerationModel.cjs"), exports);
 __exportStar(require("./generate-json/NoSuchSchemaError.cjs"), exports);
 __exportStar(require("./generate-json/SchemaDefinition.cjs"), exports);
@@ -39,6 +39,7 @@ __exportStar(require("./generate-text/DeltaEvent.cjs"), exports);
 __exportStar(require("./generate-text/TextDeltaEventSource.cjs"), exports);
 __exportStar(require("./generate-text/TextGenerationEvent.cjs"), exports);
 __exportStar(require("./generate-text/TextGenerationModel.cjs"), exports);
+__exportStar(require("./generate-text/TextStreamingEvent.cjs"), exports);
 __exportStar(require("./generate-text/generateText.cjs"), exports);
 __exportStar(require("./generate-text/streamText.cjs"), exports);
 __exportStar(require("./synthesize-speech/SpeechSynthesisEvent.cjs"), exports);

package/model-function/index.d.ts

@@ -9,10 +9,10 @@ export * from "./embed-text/embedText.js";
 export * from "./generate-image/ImageGenerationEvent.js";
 export * from "./generate-image/ImageGenerationModel.js";
 export * from "./generate-image/generateImage.js";
-export * from "./generate-json/GenerateJsonModel.js";
-export * from "./generate-json/GenerateJsonOrTextModel.js";
 export * from "./generate-json/InstructionWithSchemaPrompt.js";
 export * from "./generate-json/JsonGenerationEvent.js";
+export * from "./generate-json/JsonGenerationModel.js";
+export * from "./generate-json/JsonOrTextGenerationModel.js";
 export * from "./generate-json/JsonTextGenerationModel.js";
 export * from "./generate-json/NoSuchSchemaError.js";
 export * from "./generate-json/SchemaDefinition.js";
@@ -23,6 +23,7 @@ export * from "./generate-text/DeltaEvent.js";
 export * from "./generate-text/TextDeltaEventSource.js";
 export * from "./generate-text/TextGenerationEvent.js";
 export * from "./generate-text/TextGenerationModel.js";
+export * from "./generate-text/TextStreamingEvent.js";
 export * from "./generate-text/generateText.js";
 export * from "./generate-text/streamText.js";
 export * from "./synthesize-speech/SpeechSynthesisEvent.js";

package/model-function/index.js

@@ -9,10 +9,10 @@ export * from "./embed-text/embedText.js";
 export * from "./generate-image/ImageGenerationEvent.js";
 export * from "./generate-image/ImageGenerationModel.js";
 export * from "./generate-image/generateImage.js";
-export * from "./generate-json/GenerateJsonModel.js";
-export * from "./generate-json/GenerateJsonOrTextModel.js";
 export * from "./generate-json/InstructionWithSchemaPrompt.js";
 export * from "./generate-json/JsonGenerationEvent.js";
+export * from "./generate-json/JsonGenerationModel.js";
+export * from "./generate-json/JsonOrTextGenerationModel.js";
 export * from "./generate-json/JsonTextGenerationModel.js";
 export * from "./generate-json/NoSuchSchemaError.js";
 export * from "./generate-json/SchemaDefinition.js";
@@ -23,6 +23,7 @@ export * from "./generate-text/DeltaEvent.js";
 export * from "./generate-text/TextDeltaEventSource.js";
 export * from "./generate-text/TextGenerationEvent.js";
 export * from "./generate-text/TextGenerationModel.js";
+export * from "./generate-text/TextStreamingEvent.js";
 export * from "./generate-text/generateText.js";
 export * from "./generate-text/streamText.js";
 export * from "./synthesize-speech/SpeechSynthesisEvent.js";

package/model-function/synthesize-speech/SpeechSynthesisEvent.d.ts

@@ -1,20 +1,21 @@
 /// <reference types="node" resolution-mode="require"/>
 import { BaseModelCallFinishedEvent, BaseModelCallStartedEvent } from "../ModelCallEvent.js";
-export type SpeechSynthesisStartedEvent = BaseModelCallStartedEvent & {
+export interface SpeechSynthesisStartedEvent extends BaseModelCallStartedEvent {
     functionType: "speech-synthesis";
-    settings: unknown;
-    text: string;
-};
-export type SpeechSynthesisFinishedEvent = BaseModelCallFinishedEvent & {
-    functionType: "speech-synthesis";
-    settings: unknown;
-    text: string;
-} & ({
+    input: string;
+}
+export type SpeechSynthesisFinishedEventResult = {
     status: "success";
-    response: Buffer;
+    response: unknown;
+    output: Buffer;
 } | {
     status: "error";
     error: unknown;
 } | {
     status: "abort";
-});
+};
+export interface SpeechSynthesisFinishedEvent extends BaseModelCallFinishedEvent {
+    functionType: "speech-synthesis";
+    input: string;
+    result: SpeechSynthesisFinishedEventResult;
+}

package/model-function/synthesize-speech/SpeechSynthesisModel.d.ts

@@ -3,7 +3,7 @@ import { ModelFunctionOptions } from "../ModelFunctionOptions.js";
 import { Model, ModelSettings } from "../Model.js";
 export interface SpeechSynthesisModelSettings extends ModelSettings {
 }
-export interface SpeechSynthesisModel<SETTINGS> extends Model<SETTINGS> {
+export interface SpeechSynthesisModel<SETTINGS extends SpeechSynthesisModelSettings> extends Model<SETTINGS> {
     /**
      * Generates an mp3 audio buffer that contains the speech for the given text.
      */

package/model-function/synthesize-speech/synthesizeSpeech.cjs

@@ -7,38 +7,12 @@ const executeCall_js_1 = require("../executeCall.cjs");
  */
 function synthesizeSpeech(model, text, options) {
     return (0, executeCall_js_1.executeCall)({
+        functionType: "speech-synthesis",
+        input: text,
         model,
         options,
         generateResponse: (options) => model.generateSpeechResponse(text, options),
         extractOutputValue: (buffer) => buffer,
-        getStartEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            settings,
-            text,
-        }),
-        getAbortEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            status: "abort",
-            settings,
-            text,
-        }),
-        getFailureEvent: (metadata, settings, error) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            settings,
-            text,
-            error,
-        }),
-        getSuccessEvent: (metadata, settings, response, output) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            settings,
-            text,
-            response,
-            speech: output,
-        }),
     });
 }
 exports.synthesizeSpeech = synthesizeSpeech;

package/model-function/synthesize-speech/synthesizeSpeech.d.ts

@@ -5,4 +5,4 @@ import { SpeechSynthesisModel, SpeechSynthesisModelSettings } from "./SpeechSynt
 /**
  * Synthesizes speech from text.
  */
-export declare function synthesizeSpeech<SETTINGS extends SpeechSynthesisModelSettings>(model: SpeechSynthesisModel<SETTINGS>, text: string, options?: ModelFunctionOptions<SETTINGS>): ModelFunctionPromise<SpeechSynthesisModel<SETTINGS>, Buffer, Buffer>;
+export declare function synthesizeSpeech<SETTINGS extends SpeechSynthesisModelSettings>(model: SpeechSynthesisModel<SETTINGS>, text: string, options?: ModelFunctionOptions<SETTINGS>): ModelFunctionPromise<Buffer, Buffer>;

package/model-function/synthesize-speech/synthesizeSpeech.js

@@ -4,37 +4,11 @@ import { executeCall } from "../executeCall.js";
  */
 export function synthesizeSpeech(model, text, options) {
     return executeCall({
+        functionType: "speech-synthesis",
+        input: text,
         model,
         options,
         generateResponse: (options) => model.generateSpeechResponse(text, options),
         extractOutputValue: (buffer) => buffer,
-        getStartEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            settings,
-            text,
-        }),
-        getAbortEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            status: "abort",
-            settings,
-            text,
-        }),
-        getFailureEvent: (metadata, settings, error) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            settings,
-            text,
-            error,
-        }),
-        getSuccessEvent: (metadata, settings, response, output) => ({
-            ...metadata,
-            functionType: "speech-synthesis",
-            settings,
-            text,
-            response,
-            speech: output,
-        }),
     });
 }

package/model-function/transcribe-speech/TranscriptionEvent.d.ts

@@ -1,20 +1,18 @@
 import { BaseModelCallFinishedEvent, BaseModelCallStartedEvent } from "../ModelCallEvent.js";
-export type TranscriptionStartedEvent = BaseModelCallStartedEvent & {
+export interface TranscriptionStartedEvent extends BaseModelCallStartedEvent {
     functionType: "transcription";
-    settings: unknown;
-    data: unknown;
-};
-export type TranscriptionFinishedEvent = BaseModelCallFinishedEvent & {
-    functionType: "transcription";
-    settings: unknown;
-    data: unknown;
-} & ({
+}
+export type TranscriptionFinishedEventResult = {
     status: "success";
     response: unknown;
-    transcription: string;
+    output: string;
 } | {
     status: "error";
     error: unknown;
 } | {
     status: "abort";
-});
+};
+export interface TranscriptionFinishedEvent extends BaseModelCallFinishedEvent {
+    functionType: "transcription";
+    result: TranscriptionFinishedEventResult;
+}

package/model-function/transcribe-speech/TranscriptionModel.d.ts

@@ -2,7 +2,7 @@ import { ModelFunctionOptions } from "../ModelFunctionOptions.js";
 import { Model, ModelSettings } from "../Model.js";
 export interface TranscriptionModelSettings extends ModelSettings {
 }
-export interface TranscriptionModel<DATA, RESPONSE, SETTINGS> extends Model<SETTINGS> {
+export interface TranscriptionModel<DATA, RESPONSE, SETTINGS extends TranscriptionModelSettings> extends Model<SETTINGS> {
     generateTranscriptionResponse: (data: DATA, options?: ModelFunctionOptions<SETTINGS>) => PromiseLike<RESPONSE>;
     extractTranscriptionText: (response: RESPONSE) => string;
 }

package/model-function/transcribe-speech/transcribe.cjs

@@ -18,37 +18,12 @@ const executeCall_js_1 = require("../executeCall.cjs");
  */
 function transcribe(model, data, options) {
     return (0, executeCall_js_1.executeCall)({
+        functionType: "transcription",
+        input: data,
         model,
         options,
         generateResponse: (options) => model.generateTranscriptionResponse(data, options),
         extractOutputValue: model.extractTranscriptionText,
-        getStartEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-        }),
-        getAbortEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-        }),
-        getFailureEvent: (metadata, settings, error) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-            error,
-        }),
-        getSuccessEvent: (metadata, settings, response, output) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-            response,
-            transcription: output,
-        }),
     });
 }
 exports.transcribe = transcribe;

package/model-function/transcribe-speech/transcribe.d.ts

@@ -15,4 +15,4 @@ import { TranscriptionModel, TranscriptionModelSettings } from "./TranscriptionM
  *   }
  * );
  */
-export declare function transcribe<DATA, RESPONSE, SETTINGS extends TranscriptionModelSettings>(model: TranscriptionModel<DATA, RESPONSE, SETTINGS>, data: DATA, options?: ModelFunctionOptions<SETTINGS>): ModelFunctionPromise<TranscriptionModel<DATA, RESPONSE, SETTINGS>, string, RESPONSE>;
+export declare function transcribe<DATA, RESPONSE, SETTINGS extends TranscriptionModelSettings>(model: TranscriptionModel<DATA, RESPONSE, SETTINGS>, data: DATA, options?: ModelFunctionOptions<SETTINGS>): ModelFunctionPromise<string, RESPONSE>;

package/model-function/transcribe-speech/transcribe.js

@@ -15,36 +15,11 @@ import { executeCall } from "../executeCall.js";
  */
 export function transcribe(model, data, options) {
     return executeCall({
+        functionType: "transcription",
+        input: data,
         model,
         options,
         generateResponse: (options) => model.generateTranscriptionResponse(data, options),
         extractOutputValue: model.extractTranscriptionText,
-        getStartEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-        }),
-        getAbortEvent: (metadata, settings) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-        }),
-        getFailureEvent: (metadata, settings, error) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-            error,
-        }),
-        getSuccessEvent: (metadata, settings, response, output) => ({
-            ...metadata,
-            functionType: "transcription",
-            settings,
-            data,
-            response,
-            transcription: output,
-        }),
     });
 }

package/model-provider/automatic1111/Automatic1111ImageGenerationModel.cjs

@@ -38,6 +38,15 @@ class Automatic1111ImageGenerationModel extends AbstractModel_js_1.AbstractModel
             call: async () => callAutomatic1111ImageGenerationAPI(callSettings),
         });
     }
+    get settingsForEvent() {
+        return {
+            baseUrl: this.settings.baseUrl,
+            height: this.settings.height,
+            width: this.settings.width,
+            sampler: this.settings.sampler,
+            steps: this.settings.steps,
+        };
+    }
     generateImageResponse(prompt, options) {
         return this.callAPI(prompt, options);
     }

package/model-provider/automatic1111/Automatic1111ImageGenerationModel.d.ts

@@ -14,6 +14,7 @@ export declare class Automatic1111ImageGenerationModel extends AbstractModel<Aut
     readonly provider: "Automatic1111";
     get modelName(): string;
     callAPI(input: A111ImageGenerationPrompt, options?: ModelFunctionOptions<Automatic1111ImageGenerationModelSettings>): Promise<Automatic1111ImageGenerationResponse>;
+    get settingsForEvent(): Partial<Automatic1111ImageGenerationModelSettings>;
     generateImageResponse(prompt: A111ImageGenerationPrompt, options?: ModelFunctionOptions<Automatic1111ImageGenerationModelSettings>): Promise<{
         images: string[];
         parameters: {};

package/model-provider/automatic1111/Automatic1111ImageGenerationModel.js

@@ -35,6 +35,15 @@ export class Automatic1111ImageGenerationModel extends AbstractModel {
             call: async () => callAutomatic1111ImageGenerationAPI(callSettings),
         });
     }
+    get settingsForEvent() {
+        return {
+            baseUrl: this.settings.baseUrl,
+            height: this.settings.height,
+            width: this.settings.width,
+            sampler: this.settings.sampler,
+            steps: this.settings.steps,
+        };
+    }
     generateImageResponse(prompt, options) {
         return this.callAPI(prompt, options);
     }

package/model-provider/cohere/CohereTextEmbeddingModel.cjs

@@ -120,6 +120,12 @@ class CohereTextEmbeddingModel extends AbstractModel_js_1.AbstractModel {
             call: async () => callCohereEmbeddingAPI(callSettings),
         });
     }
+    get settingsForEvent() {
+        return {
+            baseUrl: this.settings.baseUrl,
+            truncate: this.settings.truncate,
+        };
+    }
     generateEmbeddingResponse(texts, options) {
         return this.callAPI(texts, options);
     }

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

@@ -62,14 +62,15 @@ export declare class CohereTextEmbeddingModel extends AbstractModel<CohereTextEm
     detokenize(tokens: number[]): Promise<string>;
     private get apiKey();
     callAPI(texts: Array<string>, options?: ModelFunctionOptions<CohereTextEmbeddingModelSettings>): Promise<CohereTextEmbeddingResponse>;
+    get settingsForEvent(): Partial<CohereTextEmbeddingModelSettings>;
     generateEmbeddingResponse(texts: string[], options?: ModelFunctionOptions<CohereTextEmbeddingModelSettings>): Promise<{
-        texts: string[];
         id: string;
         meta: {
             api_version: {
                 version: string;
             };
         };
+        texts: string[];
         embeddings: number[][];
     }>;
     extractEmbeddings(response: CohereTextEmbeddingResponse): number[][];
@@ -97,22 +98,22 @@ declare const cohereTextEmbeddingResponseSchema: z.ZodObject<{
         };
     }>;
 }, "strip", z.ZodTypeAny, {
-    texts: string[];
     id: string;
     meta: {
         api_version: {
             version: string;
         };
     };
+    texts: string[];
     embeddings: number[][];
 }, {
-    texts: string[];
     id: string;
     meta: {
         api_version: {
             version: string;
         };
     };
+    texts: string[];
     embeddings: number[][];
 }>;
 export type CohereTextEmbeddingResponse = z.infer<typeof cohereTextEmbeddingResponseSchema>;

package/model-provider/cohere/CohereTextEmbeddingModel.js

@@ -114,6 +114,12 @@ export class CohereTextEmbeddingModel extends AbstractModel {
             call: async () => callCohereEmbeddingAPI(callSettings),
         });
     }
+    get settingsForEvent() {
+        return {
+            baseUrl: this.settings.baseUrl,
+            truncate: this.settings.truncate,
+        };
+    }
     generateEmbeddingResponse(texts, options) {
         return this.callAPI(texts, options);
     }

package/model-provider/cohere/CohereTextGenerationModel.cjs

@@ -102,6 +102,8 @@ class CohereTextGenerationModel extends AbstractModel_js_1.AbstractModel {
             // to exclude stop tokens from the generated text
             endSequences: combinedSettings.stopSequences,
             maxTokens: combinedSettings.maxCompletionTokens,
+            // mapped name because of conflict with stopSequences:
+            stopSequences: combinedSettings.cohereStopSequences,
             abortSignal: run?.abortSignal,
             prompt,
             responseFormat,
@@ -112,6 +114,24 @@ class CohereTextGenerationModel extends AbstractModel_js_1.AbstractModel {
             call: async () => callCohereTextGenerationAPI(callSettings),
         });
     }
+    get settingsForEvent() {
+        const eventSettingProperties = [
+            "maxCompletionTokens",
+            "stopSequences",
+            "baseUrl",
+            "numGenerations",
+            "temperature",
+            "k",
+            "p",
+            "frequencyPenalty",
+            "presencePenalty",
+            "returnLikelihoods",
+            "logitBias",
+            "truncate",
+            "cohereStopSequences",
+        ];
+        return Object.fromEntries(Object.entries(this.settings).filter(([key]) => eventSettingProperties.includes(key)));
+    }
     generateTextResponse(prompt, options) {
         return this.callAPI(prompt, {
             ...options,