modelfusion 0.110.0 → 0.112.0

package/model-provider/cohere/CohereFacade.cjs

@@ -1,9 +1,18 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Tokenizer = exports.TextEmbedder = exports.TextGenerator = void 0;
+exports.Tokenizer = exports.TextEmbedder = exports.TextGenerator = exports.Api = void 0;
+const CohereApiConfiguration_js_1 = require("./CohereApiConfiguration.cjs");
 const CohereTextEmbeddingModel_js_1 = require("./CohereTextEmbeddingModel.cjs");
 const CohereTextGenerationModel_js_1 = require("./CohereTextGenerationModel.cjs");
 const CohereTokenizer_js_1 = require("./CohereTokenizer.cjs");
+/**
+ * Creates an API configuration for the Cohere API.
+ * It calls the API at https://api.cohere.ai/v1 and uses the `COHERE_API_KEY` env variable by default.
+ */
+function Api(settings) {
+    return new CohereApiConfiguration_js_1.CohereApiConfiguration(settings);
+}
+exports.Api = Api;
 /**
  * Create a text generation model that calls the Cohere Co.Generate API.
  *
@@ -11,7 +20,7 @@ const CohereTokenizer_js_1 = require("./CohereTokenizer.cjs");
  *
  * @example
  * const model = cohere.TextGenerator({
- *   model: "command-nightly",
+ *   model: "command",
  *   temperature: 0.7,
  *   maxGenerationTokens: 500,
  * });
@@ -54,7 +63,7 @@ exports.TextEmbedder = TextEmbedder;
  * @see https://docs.cohere.com/reference/detokenize-1
  *
  * @example
- * const tokenizer = cohere.Tokenizer({ model: "command-nightly" });
+ * const tokenizer = cohere.Tokenizer({ model: "command" });
  *
  * const text = "At first, Nox didn't know what to do with the pup.";
  *

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

@@ -1,6 +1,15 @@
+import { PartialBaseUrlPartsApiConfigurationOptions } from "../../core/api/BaseUrlApiConfiguration.js";
+import { CohereApiConfiguration } from "./CohereApiConfiguration.js";
 import { CohereTextEmbeddingModel, CohereTextEmbeddingModelSettings } from "./CohereTextEmbeddingModel.js";
 import { CohereTextGenerationModel, CohereTextGenerationModelSettings } from "./CohereTextGenerationModel.js";
 import { CohereTokenizer, CohereTokenizerSettings } from "./CohereTokenizer.js";
+/**
+ * Creates an API configuration for the Cohere API.
+ * It calls the API at https://api.cohere.ai/v1 and uses the `COHERE_API_KEY` env variable by default.
+ */
+export declare function Api(settings: PartialBaseUrlPartsApiConfigurationOptions & {
+    apiKey?: string;
+}): CohereApiConfiguration;
 /**
  * Create a text generation model that calls the Cohere Co.Generate API.
  *
@@ -8,7 +17,7 @@ import { CohereTokenizer, CohereTokenizerSettings } from "./CohereTokenizer.js";
  *
  * @example
  * const model = cohere.TextGenerator({
- *   model: "command-nightly",
+ *   model: "command",
  *   temperature: 0.7,
  *   maxGenerationTokens: 500,
  * });
@@ -45,7 +54,7 @@ export declare function TextEmbedder(settings: CohereTextEmbeddingModelSettings)
  * @see https://docs.cohere.com/reference/detokenize-1
  *
  * @example
- * const tokenizer = cohere.Tokenizer({ model: "command-nightly" });
+ * const tokenizer = cohere.Tokenizer({ model: "command" });
  *
  * const text = "At first, Nox didn't know what to do with the pup.";
  *

package/model-provider/cohere/CohereFacade.js

@@ -1,6 +1,14 @@
+import { CohereApiConfiguration } from "./CohereApiConfiguration.js";
 import { CohereTextEmbeddingModel, } from "./CohereTextEmbeddingModel.js";
 import { CohereTextGenerationModel, } from "./CohereTextGenerationModel.js";
 import { CohereTokenizer } from "./CohereTokenizer.js";
+/**
+ * Creates an API configuration for the Cohere API.
+ * It calls the API at https://api.cohere.ai/v1 and uses the `COHERE_API_KEY` env variable by default.
+ */
+export function Api(settings) {
+    return new CohereApiConfiguration(settings);
+}
 /**
  * Create a text generation model that calls the Cohere Co.Generate API.
  *
@@ -8,7 +16,7 @@ import { CohereTokenizer } from "./CohereTokenizer.js";
  *
  * @example
  * const model = cohere.TextGenerator({
- *   model: "command-nightly",
+ *   model: "command",
  *   temperature: 0.7,
  *   maxGenerationTokens: 500,
  * });
@@ -49,7 +57,7 @@ export function TextEmbedder(settings) {
  * @see https://docs.cohere.com/reference/detokenize-1
  *
  * @example
- * const tokenizer = cohere.Tokenizer({ model: "command-nightly" });
+ * const tokenizer = cohere.Tokenizer({ model: "command" });
  *
  * const text = "At first, Nox didn't know what to do with the pup.";
  *

package/model-provider/cohere/CohereTextEmbeddingModel.cjs

@@ -4,6 +4,7 @@ exports.CohereTextEmbeddingModel = exports.COHERE_TEXT_EMBEDDING_MODELS = void 0
 const zod_1 = require("zod");
 const callWithRetryAndThrottle_js_1 = require("../../core/api/callWithRetryAndThrottle.cjs");
 const postToApi_js_1 = require("../../core/api/postToApi.cjs");
+const ZodSchema_js_1 = require("../../core/schema/ZodSchema.cjs");
 const AbstractModel_js_1 = require("../../model-function/AbstractModel.cjs");
 const CohereApiConfiguration_js_1 = require("./CohereApiConfiguration.cjs");
 const CohereError_js_1 = require("./CohereError.cjs");
@@ -18,7 +19,7 @@ exports.COHERE_TEXT_EMBEDDING_MODELS = {
         embeddingDimensions: 4096,
     },
     "embed-multilingual-v2.0": {
-        contextWindowSize: 512,
+        contextWindowSize: 256,
         embeddingDimensions: 768,
     },
     "embed-english-v3.0": {
@@ -116,13 +117,23 @@ class CohereTextEmbeddingModel extends AbstractModel_js_1.AbstractModel {
         if (texts.length > this.maxValuesPerCall) {
             throw new Error(`The Cohere embedding API only supports ${this.maxValuesPerCall} texts per API call.`);
         }
+        const api = this.settings.api ?? new CohereApiConfiguration_js_1.CohereApiConfiguration();
+        const abortSignal = options?.run?.abortSignal;
         return (0, callWithRetryAndThrottle_js_1.callWithRetryAndThrottle)({
-            retry: this.settings.api?.retry,
-            throttle: this.settings.api?.throttle,
-            call: async () => callCohereEmbeddingAPI({
-                ...this.settings,
-                abortSignal: options?.run?.abortSignal,
-                texts,
+            retry: api.retry,
+            throttle: api.throttle,
+            call: async () => (0, postToApi_js_1.postJsonToApi)({
+                url: api.assembleUrl(`/embed`),
+                headers: api.headers,
+                body: {
+                    model: this.settings.model,
+                    texts,
+                    input_type: this.settings.inputType,
+                    truncate: this.settings.truncate,
+                },
+                failedResponseHandler: CohereError_js_1.failedCohereCallResponseHandler,
+                successfulResponseHandler: (0, postToApi_js_1.createJsonResponseHandler)((0, ZodSchema_js_1.zodSchema)(cohereTextEmbeddingResponseSchema)),
+                abortSignal,
             }),
         });
     }
@@ -153,18 +164,3 @@ const cohereTextEmbeddingResponseSchema = zod_1.z.object({
         }),
     }),
 });
-async function callCohereEmbeddingAPI({ api = new CohereApiConfiguration_js_1.CohereApiConfiguration(), abortSignal, model, texts, inputType, truncate, }) {
-    return (0, postToApi_js_1.postJsonToApi)({
-        url: api.assembleUrl(`/embed`),
-        headers: api.headers,
-        body: {
-            model,
-            texts,
-            input_type: inputType,
-            truncate,
-        },
-        failedResponseHandler: CohereError_js_1.failedCohereCallResponseHandler,
-        successfulResponseHandler: (0, postToApi_js_1.createJsonResponseHandler)(cohereTextEmbeddingResponseSchema),
-        abortSignal,
-    });
-}

package/model-provider/cohere/CohereTextEmbeddingModel.js

@@ -1,6 +1,7 @@
 import { z } from "zod";
 import { callWithRetryAndThrottle } from "../../core/api/callWithRetryAndThrottle.js";
 import { createJsonResponseHandler, postJsonToApi, } from "../../core/api/postToApi.js";
+import { zodSchema } from "../../core/schema/ZodSchema.js";
 import { AbstractModel } from "../../model-function/AbstractModel.js";
 import { CohereApiConfiguration } from "./CohereApiConfiguration.js";
 import { failedCohereCallResponseHandler } from "./CohereError.js";
@@ -15,7 +16,7 @@ export const COHERE_TEXT_EMBEDDING_MODELS = {
         embeddingDimensions: 4096,
     },
     "embed-multilingual-v2.0": {
-        contextWindowSize: 512,
+        contextWindowSize: 256,
         embeddingDimensions: 768,
     },
     "embed-english-v3.0": {
@@ -113,13 +114,23 @@ export class CohereTextEmbeddingModel extends AbstractModel {
         if (texts.length > this.maxValuesPerCall) {
             throw new Error(`The Cohere embedding API only supports ${this.maxValuesPerCall} texts per API call.`);
         }
+        const api = this.settings.api ?? new CohereApiConfiguration();
+        const abortSignal = options?.run?.abortSignal;
         return callWithRetryAndThrottle({
-            retry: this.settings.api?.retry,
-            throttle: this.settings.api?.throttle,
-            call: async () => callCohereEmbeddingAPI({
-                ...this.settings,
-                abortSignal: options?.run?.abortSignal,
-                texts,
+            retry: api.retry,
+            throttle: api.throttle,
+            call: async () => postJsonToApi({
+                url: api.assembleUrl(`/embed`),
+                headers: api.headers,
+                body: {
+                    model: this.settings.model,
+                    texts,
+                    input_type: this.settings.inputType,
+                    truncate: this.settings.truncate,
+                },
+                failedResponseHandler: failedCohereCallResponseHandler,
+                successfulResponseHandler: createJsonResponseHandler(zodSchema(cohereTextEmbeddingResponseSchema)),
+                abortSignal,
             }),
         });
     }
@@ -149,18 +160,3 @@ const cohereTextEmbeddingResponseSchema = z.object({
         }),
     }),
 });
-async function callCohereEmbeddingAPI({ api = new CohereApiConfiguration(), abortSignal, model, texts, inputType, truncate, }) {
-    return postJsonToApi({
-        url: api.assembleUrl(`/embed`),
-        headers: api.headers,
-        body: {
-            model,
-            texts,
-            input_type: inputType,
-            truncate,
-        },
-        failedResponseHandler: failedCohereCallResponseHandler,
-        successfulResponseHandler: createJsonResponseHandler(cohereTextEmbeddingResponseSchema),
-        abortSignal,
-    });
-}

package/model-provider/cohere/CohereTextGenerationModel.cjs

@@ -16,16 +16,10 @@ const CohereError_js_1 = require("./CohereError.cjs");
 const CohereTokenizer_js_1 = require("./CohereTokenizer.cjs");
 exports.COHERE_TEXT_GENERATION_MODELS = {
     command: {
-        contextWindowSize: 2048,
-    },
-    "command-nightly": {
-        contextWindowSize: 2048,
+        contextWindowSize: 4096,
     },
     "command-light": {
-        contextWindowSize: 2048,
-    },
-    "command-light-nightly": {
-        contextWindowSize: 2048,
+        contextWindowSize: 4096,
     },
 };
 /**
@@ -35,7 +29,7 @@ exports.COHERE_TEXT_GENERATION_MODELS = {
  *
  * @example
  * const model = new CohereTextGenerationModel({
- *   model: "command-nightly",
+ *   model: "command",
  *   temperature: 0.7,
  *   maxGenerationTokens: 500,
  * });
@@ -86,32 +80,30 @@ class CohereTextGenerationModel extends AbstractModel_js_1.AbstractModel {
         return (0, callWithRetryAndThrottle_js_1.callWithRetryAndThrottle)({
             retry: api.retry,
             throttle: api.throttle,
-            call: async () => {
-                return (0, postToApi_js_1.postJsonToApi)({
-                    url: api.assembleUrl(`/generate`),
-                    headers: api.headers,
-                    body: {
-                        stream: responseFormat.stream,
-                        model: this.settings.model,
-                        prompt,
-                        num_generations: this.settings.numberOfGenerations,
-                        max_tokens: this.settings.maxGenerationTokens,
-                        temperature: this.settings.temperature,
-                        k: this.settings.k,
-                        p: this.settings.p,
-                        frequency_penalty: this.settings.frequencyPenalty,
-                        presence_penalty: this.settings.presencePenalty,
-                        end_sequences: this.settings.stopSequences,
-                        stop_sequences: this.settings.cohereStopSequences,
-                        return_likelihoods: this.settings.returnLikelihoods,
-                        logit_bias: this.settings.logitBias,
-                        truncate: this.settings.truncate,
-                    },
-                    failedResponseHandler: CohereError_js_1.failedCohereCallResponseHandler,
-                    successfulResponseHandler: responseFormat.handler,
-                    abortSignal,
-                });
-            },
+            call: async () => (0, postToApi_js_1.postJsonToApi)({
+                url: api.assembleUrl(`/generate`),
+                headers: api.headers,
+                body: {
+                    stream: responseFormat.stream,
+                    model: this.settings.model,
+                    prompt,
+                    num_generations: this.settings.numberOfGenerations,
+                    max_tokens: this.settings.maxGenerationTokens,
+                    temperature: this.settings.temperature,
+                    k: this.settings.k,
+                    p: this.settings.p,
+                    frequency_penalty: this.settings.frequencyPenalty,
+                    presence_penalty: this.settings.presencePenalty,
+                    end_sequences: this.settings.stopSequences,
+                    stop_sequences: this.settings.cohereStopSequences,
+                    return_likelihoods: this.settings.returnLikelihoods,
+                    logit_bias: this.settings.logitBias,
+                    truncate: this.settings.truncate,
+                },
+                failedResponseHandler: CohereError_js_1.failedCohereCallResponseHandler,
+                successfulResponseHandler: responseFormat.handler,
+                abortSignal,
+            }),
         });
     }
     get settingsForEvent() {
@@ -210,7 +202,7 @@ const cohereTextGenerationResponseSchema = zod_1.z.object({
     })
         .optional(),
 });
-const cohereTextStreamChunkSchema = (0, ZodSchema_js_1.zodSchema)(zod_1.z.discriminatedUnion("is_finished", [
+const cohereTextStreamChunkSchema = zod_1.z.discriminatedUnion("is_finished", [
     zod_1.z.object({
         text: zod_1.z.string(),
         is_finished: zod_1.z.literal(false),
@@ -220,14 +212,14 @@ const cohereTextStreamChunkSchema = (0, ZodSchema_js_1.zodSchema)(zod_1.z.discri
         finish_reason: zod_1.z.string(),
         response: cohereTextGenerationResponseSchema,
     }),
-]));
+]);
 exports.CohereTextGenerationResponseFormat = {
     /**
      * Returns the response as a JSON object.
      */
     json: {
         stream: false,
-        handler: (0, postToApi_js_1.createJsonResponseHandler)(cohereTextGenerationResponseSchema),
+        handler: (0, postToApi_js_1.createJsonResponseHandler)((0, ZodSchema_js_1.zodSchema)(cohereTextGenerationResponseSchema)),
     },
     /**
      * Returns an async iterable over the full deltas (all choices, including full current state at time of event)
@@ -235,6 +227,6 @@ exports.CohereTextGenerationResponseFormat = {
      */
     deltaIterable: {
         stream: true,
-        handler: (0, createJsonStreamResponseHandler_js_1.createJsonStreamResponseHandler)(cohereTextStreamChunkSchema),
+        handler: (0, createJsonStreamResponseHandler_js_1.createJsonStreamResponseHandler)((0, ZodSchema_js_1.zodSchema)(cohereTextStreamChunkSchema)),
     },
 };

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

@@ -12,15 +12,9 @@ export declare const COHERE_TEXT_GENERATION_MODELS: {
     command: {
         contextWindowSize: number;
     };
-    "command-nightly": {
-        contextWindowSize: number;
-    };
     "command-light": {
         contextWindowSize: number;
     };
-    "command-light-nightly": {
-        contextWindowSize: number;
-    };
 };
 export type CohereTextGenerationModelType = keyof typeof COHERE_TEXT_GENERATION_MODELS;
 export interface CohereTextGenerationModelSettings extends TextGenerationModelSettings {
@@ -43,7 +37,7 @@ export interface CohereTextGenerationModelSettings extends TextGenerationModelSe
  *
  * @example
  * const model = new CohereTextGenerationModel({
- *   model: "command-nightly",
+ *   model: "command",
  *   temperature: 0.7,
  *   maxGenerationTokens: 500,
  * });
@@ -56,7 +50,7 @@ export interface CohereTextGenerationModelSettings extends TextGenerationModelSe
 export declare class CohereTextGenerationModel extends AbstractModel<CohereTextGenerationModelSettings> implements TextStreamingModel<string, CohereTextGenerationModelSettings> {
     constructor(settings: CohereTextGenerationModelSettings);
     readonly provider: "cohere";
-    get modelName(): "command" | "command-nightly" | "command-light" | "command-light-nightly";
+    get modelName(): "command" | "command-light";
     readonly contextWindowSize: number;
     readonly tokenizer: CohereTokenizer;
     countPromptTokens(input: string): Promise<number>;
@@ -182,6 +176,114 @@ declare const cohereTextGenerationResponseSchema: z.ZodObject<{
     } | undefined;
 }>;
 export type CohereTextGenerationResponse = z.infer<typeof cohereTextGenerationResponseSchema>;
+declare const cohereTextStreamChunkSchema: z.ZodDiscriminatedUnion<"is_finished", [z.ZodObject<{
+    text: z.ZodString;
+    is_finished: z.ZodLiteral<false>;
+}, "strip", z.ZodTypeAny, {
+    text: string;
+    is_finished: false;
+}, {
+    text: string;
+    is_finished: false;
+}>, z.ZodObject<{
+    is_finished: z.ZodLiteral<true>;
+    finish_reason: z.ZodString;
+    response: z.ZodObject<{
+        id: z.ZodString;
+        generations: z.ZodArray<z.ZodObject<{
+            id: z.ZodString;
+            text: z.ZodString;
+            finish_reason: z.ZodOptional<z.ZodString>;
+        }, "strip", z.ZodTypeAny, {
+            text: string;
+            id: string;
+            finish_reason?: string | undefined;
+        }, {
+            text: string;
+            id: string;
+            finish_reason?: string | undefined;
+        }>, "many">;
+        prompt: z.ZodString;
+        meta: z.ZodOptional<z.ZodObject<{
+            api_version: z.ZodObject<{
+                version: z.ZodString;
+            }, "strip", z.ZodTypeAny, {
+                version: string;
+            }, {
+                version: string;
+            }>;
+        }, "strip", z.ZodTypeAny, {
+            api_version: {
+                version: string;
+            };
+        }, {
+            api_version: {
+                version: string;
+            };
+        }>>;
+    }, "strip", z.ZodTypeAny, {
+        id: string;
+        prompt: string;
+        generations: {
+            text: string;
+            id: string;
+            finish_reason?: string | undefined;
+        }[];
+        meta?: {
+            api_version: {
+                version: string;
+            };
+        } | undefined;
+    }, {
+        id: string;
+        prompt: string;
+        generations: {
+            text: string;
+            id: string;
+            finish_reason?: string | undefined;
+        }[];
+        meta?: {
+            api_version: {
+                version: string;
+            };
+        } | undefined;
+    }>;
+}, "strip", z.ZodTypeAny, {
+    response: {
+        id: string;
+        prompt: string;
+        generations: {
+            text: string;
+            id: string;
+            finish_reason?: string | undefined;
+        }[];
+        meta?: {
+            api_version: {
+                version: string;
+            };
+        } | undefined;
+    };
+    finish_reason: string;
+    is_finished: true;
+}, {
+    response: {
+        id: string;
+        prompt: string;
+        generations: {
+            text: string;
+            id: string;
+            finish_reason?: string | undefined;
+        }[];
+        meta?: {
+            api_version: {
+                version: string;
+            };
+        } | undefined;
+    };
+    finish_reason: string;
+    is_finished: true;
+}>]>;
+export type CohereTextStreamChunk = z.infer<typeof cohereTextStreamChunkSchema>;
 export type CohereTextGenerationResponseFormatType<T> = {
     stream: boolean;
     handler: ResponseHandler<T>;

package/model-provider/cohere/CohereTextGenerationModel.js

@@ -13,16 +13,10 @@ import { failedCohereCallResponseHandler } from "./CohereError.js";
 import { CohereTokenizer } from "./CohereTokenizer.js";
 export const COHERE_TEXT_GENERATION_MODELS = {
     command: {
-        contextWindowSize: 2048,
-    },
-    "command-nightly": {
-        contextWindowSize: 2048,
+        contextWindowSize: 4096,
     },
     "command-light": {
-        contextWindowSize: 2048,
-    },
-    "command-light-nightly": {
-        contextWindowSize: 2048,
+        contextWindowSize: 4096,
     },
 };
 /**
@@ -32,7 +26,7 @@ export const COHERE_TEXT_GENERATION_MODELS = {
  *
  * @example
  * const model = new CohereTextGenerationModel({
- *   model: "command-nightly",
+ *   model: "command",
  *   temperature: 0.7,
  *   maxGenerationTokens: 500,
  * });
@@ -83,32 +77,30 @@ export class CohereTextGenerationModel extends AbstractModel {
         return callWithRetryAndThrottle({
             retry: api.retry,
             throttle: api.throttle,
-            call: async () => {
-                return postJsonToApi({
-                    url: api.assembleUrl(`/generate`),
-                    headers: api.headers,
-                    body: {
-                        stream: responseFormat.stream,
-                        model: this.settings.model,
-                        prompt,
-                        num_generations: this.settings.numberOfGenerations,
-                        max_tokens: this.settings.maxGenerationTokens,
-                        temperature: this.settings.temperature,
-                        k: this.settings.k,
-                        p: this.settings.p,
-                        frequency_penalty: this.settings.frequencyPenalty,
-                        presence_penalty: this.settings.presencePenalty,
-                        end_sequences: this.settings.stopSequences,
-                        stop_sequences: this.settings.cohereStopSequences,
-                        return_likelihoods: this.settings.returnLikelihoods,
-                        logit_bias: this.settings.logitBias,
-                        truncate: this.settings.truncate,
-                    },
-                    failedResponseHandler: failedCohereCallResponseHandler,
-                    successfulResponseHandler: responseFormat.handler,
-                    abortSignal,
-                });
-            },
+            call: async () => postJsonToApi({
+                url: api.assembleUrl(`/generate`),
+                headers: api.headers,
+                body: {
+                    stream: responseFormat.stream,
+                    model: this.settings.model,
+                    prompt,
+                    num_generations: this.settings.numberOfGenerations,
+                    max_tokens: this.settings.maxGenerationTokens,
+                    temperature: this.settings.temperature,
+                    k: this.settings.k,
+                    p: this.settings.p,
+                    frequency_penalty: this.settings.frequencyPenalty,
+                    presence_penalty: this.settings.presencePenalty,
+                    end_sequences: this.settings.stopSequences,
+                    stop_sequences: this.settings.cohereStopSequences,
+                    return_likelihoods: this.settings.returnLikelihoods,
+                    logit_bias: this.settings.logitBias,
+                    truncate: this.settings.truncate,
+                },
+                failedResponseHandler: failedCohereCallResponseHandler,
+                successfulResponseHandler: responseFormat.handler,
+                abortSignal,
+            }),
         });
     }
     get settingsForEvent() {
@@ -206,7 +198,7 @@ const cohereTextGenerationResponseSchema = z.object({
     })
         .optional(),
 });
-const cohereTextStreamChunkSchema = zodSchema(z.discriminatedUnion("is_finished", [
+const cohereTextStreamChunkSchema = z.discriminatedUnion("is_finished", [
     z.object({
         text: z.string(),
         is_finished: z.literal(false),
@@ -216,14 +208,14 @@ const cohereTextStreamChunkSchema = zodSchema(z.discriminatedUnion("is_finished"
         finish_reason: z.string(),
         response: cohereTextGenerationResponseSchema,
     }),
-]));
+]);
 export const CohereTextGenerationResponseFormat = {
     /**
      * Returns the response as a JSON object.
      */
     json: {
         stream: false,
-        handler: createJsonResponseHandler(cohereTextGenerationResponseSchema),
+        handler: createJsonResponseHandler(zodSchema(cohereTextGenerationResponseSchema)),
     },
     /**
      * Returns an async iterable over the full deltas (all choices, including full current state at time of event)
@@ -231,6 +223,6 @@ export const CohereTextGenerationResponseFormat = {
      */
     deltaIterable: {
         stream: true,
-        handler: createJsonStreamResponseHandler(cohereTextStreamChunkSchema),
+        handler: createJsonStreamResponseHandler(zodSchema(cohereTextStreamChunkSchema)),
     },
 };