@ai-sdk/cohere 2.0.44 → 2.0.46

package/dist/index.js

@@ -28,51 +28,64 @@ module.exports = __toCommonJS(index_exports);
 
 // src/cohere-provider.ts
 var import_provider4 = require("@ai-sdk/provider");
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_provider_utils7 = require("@ai-sdk/provider-utils");
 
 // src/cohere-chat-language-model.ts
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var import_provider_utils4 = require("@ai-sdk/provider-utils");
 var import_v43 = require("zod/v4");
 
 // src/cohere-chat-options.ts
+var import_provider_utils = require("@ai-sdk/provider-utils");
 var import_v4 = require("zod/v4");
-var cohereChatModelOptions = import_v4.z.object({
-  /**
-   * Configuration for reasoning features (optional)
-   *
-   * Can be set to an object with the two properties `type` and `tokenBudget`. `type` can be set to `'enabled'` or `'disabled'` (defaults to `'enabled'`).
-   * `tokenBudget` is the maximum number of tokens the model can use for thinking, which must be set to a positive integer. The model will stop thinking if it reaches the thinking token budget and will proceed with the response
-   *
-   * @see https://docs.cohere.com/reference/chat#request.body.thinking
-   */
-  thinking: import_v4.z.object({
-    type: import_v4.z.enum(["enabled", "disabled"]).optional(),
-    tokenBudget: import_v4.z.number().optional()
-  }).optional()
-});
-var cohereImagePartProviderOptions = import_v4.z.object({
-  /**
-   * Image fidelity level passed through as `image_url.detail` on the Cohere chat API.
-   *
-   * @see https://docs.cohere.com/docs/image-inputs
-   */
-  detail: import_v4.z.enum(["auto", "low", "high"]).optional()
-});
+var cohereChatModelOptions = (0, import_provider_utils.lazySchema)(
+  () => (0, import_provider_utils.zodSchema)(
+    import_v4.z.object({
+      /**
+       * Configuration for reasoning features (optional)
+       *
+       * Can be set to an object with the two properties `type` and `tokenBudget`. `type` can be set to `'enabled'` or `'disabled'` (defaults to `'enabled'`).
+       * `tokenBudget` is the maximum number of tokens the model can use for thinking, which must be set to a positive integer. The model will stop thinking if it reaches the thinking token budget and will proceed with the response
+       *
+       * @see https://docs.cohere.com/reference/chat#request.body.thinking
+       */
+      thinking: import_v4.z.object({
+        type: import_v4.z.enum(["enabled", "disabled"]).optional(),
+        tokenBudget: import_v4.z.number().optional()
+      }).optional()
+    })
+  )
+);
+var cohereImagePartProviderOptions = (0, import_provider_utils.lazySchema)(
+  () => (0, import_provider_utils.zodSchema)(
+    import_v4.z.object({
+      /**
+       * Image fidelity level passed through as `image_url.detail` on the Cohere chat API.
+       *
+       * @see https://docs.cohere.com/docs/image-inputs
+       */
+      detail: import_v4.z.enum(["auto", "low", "high"]).optional()
+    })
+  )
+);
 
 // src/cohere-error.ts
-var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
 var import_v42 = require("zod/v4");
-var cohereErrorDataSchema = import_v42.z.object({
-  message: import_v42.z.string()
-});
-var cohereFailedResponseHandler = (0, import_provider_utils.createJsonErrorResponseHandler)({
+var cohereErrorDataSchema = (0, import_provider_utils2.lazySchema)(
+  () => (0, import_provider_utils2.zodSchema)(
+    import_v42.z.object({
+      message: import_v42.z.string()
+    })
+  )
+);
+var cohereFailedResponseHandler = (0, import_provider_utils2.createJsonErrorResponseHandler)({
   errorSchema: cohereErrorDataSchema,
   errorToMessage: (data) => data.message
 });
 
 // src/convert-to-cohere-chat-prompt.ts
 var import_provider = require("@ai-sdk/provider");
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_provider_utils3 = require("@ai-sdk/provider-utils");
 async function convertToCohereChatPrompt(prompt) {
   var _a, _b;
   const messages = [];
@@ -99,7 +112,7 @@ async function convertToCohereChatPrompt(prompt) {
               if (isImageMediaType(part.mediaType)) {
                 hasImage = true;
                 const url = buildImageUrl({ part });
-                const cohereOptions = (_a = await (0, import_provider_utils2.parseProviderOptions)({
+                const cohereOptions = (_a = await (0, import_provider_utils3.parseProviderOptions)({
                   provider: "cohere",
                   providerOptions: part.providerOptions,
                   schema: cohereImagePartProviderOptions
@@ -221,7 +234,7 @@ function buildImageUrl({ part }) {
     return part.data.toString();
   }
   const mediaType = part.mediaType === "image" || part.mediaType === "image/*" ? "image/jpeg" : part.mediaType;
-  return `data:${mediaType};base64,${(0, import_provider_utils2.convertToBase64)(part.data)}`;
+  return `data:${mediaType};base64,${(0, import_provider_utils3.convertToBase64)(part.data)}`;
 }
 
 // src/map-cohere-finish-reason.ts
@@ -324,7 +337,7 @@ var CohereChatLanguageModel = class {
     providerOptions
   }) {
     var _a, _b;
-    const cohereOptions = (_a = await (0, import_provider_utils3.parseProviderOptions)({
+    const cohereOptions = (_a = await (0, import_provider_utils4.parseProviderOptions)({
       provider: "cohere",
       providerOptions,
       schema: cohereChatModelOptions
@@ -379,12 +392,12 @@ var CohereChatLanguageModel = class {
       responseHeaders,
       value: response,
       rawValue: rawResponse
-    } = await (0, import_provider_utils3.postJsonToApi)({
+    } = await (0, import_provider_utils4.postJsonToApi)({
       url: `${this.config.baseURL}/chat`,
-      headers: (0, import_provider_utils3.combineHeaders)(this.config.headers(), options.headers),
+      headers: (0, import_provider_utils4.combineHeaders)(this.config.headers(), options.headers),
       body: args,
       failedResponseHandler: cohereFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils3.createJsonResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils4.createJsonResponseHandler)(
         cohereChatResponseSchema
       ),
       abortSignal: options.abortSignal,
@@ -449,12 +462,12 @@ var CohereChatLanguageModel = class {
   }
   async doStream(options) {
     const { args, warnings } = await this.getArgs(options);
-    const { responseHeaders, value: response } = await (0, import_provider_utils3.postJsonToApi)({
+    const { responseHeaders, value: response } = await (0, import_provider_utils4.postJsonToApi)({
       url: `${this.config.baseURL}/chat`,
-      headers: (0, import_provider_utils3.combineHeaders)(this.config.headers(), options.headers),
+      headers: (0, import_provider_utils4.combineHeaders)(this.config.headers(), options.headers),
       body: { ...args, stream: true },
       failedResponseHandler: cohereFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils3.createEventSourceResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils4.createEventSourceResponseHandler)(
         cohereChatChunkSchema
       ),
       abortSignal: options.abortSignal,
@@ -621,197 +634,215 @@ var CohereChatLanguageModel = class {
     };
   }
 };
-var cohereChatResponseSchema = import_v43.z.object({
-  generation_id: import_v43.z.string().nullish(),
-  message: import_v43.z.object({
-    role: import_v43.z.string(),
-    content: import_v43.z.array(
-      import_v43.z.union([
-        import_v43.z.object({
-          type: import_v43.z.literal("text"),
-          text: import_v43.z.string()
-        }),
-        import_v43.z.object({
-          type: import_v43.z.literal("thinking"),
-          thinking: import_v43.z.string()
-        })
-      ])
-    ).nullish(),
-    tool_plan: import_v43.z.string().nullish(),
-    tool_calls: import_v43.z.array(
-      import_v43.z.object({
-        id: import_v43.z.string(),
-        type: import_v43.z.literal("function"),
-        function: import_v43.z.object({
-          name: import_v43.z.string(),
-          arguments: import_v43.z.string()
-        })
-      })
-    ).nullish(),
-    citations: import_v43.z.array(
-      import_v43.z.object({
-        start: import_v43.z.number(),
-        end: import_v43.z.number(),
-        text: import_v43.z.string(),
-        sources: import_v43.z.array(
-          import_v43.z.object({
-            type: import_v43.z.string().optional(),
-            id: import_v43.z.string().optional(),
-            document: import_v43.z.object({
-              id: import_v43.z.string().optional(),
-              text: import_v43.z.string(),
-              title: import_v43.z.string()
-            })
-          })
-        ),
-        type: import_v43.z.string().optional()
-      })
-    ).nullish()
-  }),
-  finish_reason: import_v43.z.string(),
-  usage: import_v43.z.object({
-    billed_units: import_v43.z.object({
-      input_tokens: import_v43.z.number(),
-      output_tokens: import_v43.z.number()
-    }),
-    tokens: import_v43.z.object({
-      input_tokens: import_v43.z.number(),
-      output_tokens: import_v43.z.number()
-    })
-  })
-});
-var cohereChatChunkSchema = import_v43.z.discriminatedUnion("type", [
-  import_v43.z.object({
-    type: import_v43.z.literal("citation-start")
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("citation-end")
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("content-start"),
-    index: import_v43.z.number(),
-    delta: import_v43.z.object({
+var cohereChatResponseSchema = (0, import_provider_utils4.lazySchema)(
+  () => (0, import_provider_utils4.zodSchema)(
+    import_v43.z.object({
+      generation_id: import_v43.z.string().nullish(),
       message: import_v43.z.object({
-        content: import_v43.z.union([
-          import_v43.z.object({
-            type: import_v43.z.literal("text"),
-            text: import_v43.z.string()
-          }),
+        role: import_v43.z.string(),
+        content: import_v43.z.array(
+          import_v43.z.union([
+            import_v43.z.object({
+              type: import_v43.z.literal("text"),
+              text: import_v43.z.string()
+            }),
+            import_v43.z.object({
+              type: import_v43.z.literal("thinking"),
+              thinking: import_v43.z.string()
+            })
+          ])
+        ).nullish(),
+        tool_plan: import_v43.z.string().nullish(),
+        tool_calls: import_v43.z.array(
           import_v43.z.object({
-            type: import_v43.z.literal("thinking"),
-            thinking: import_v43.z.string()
+            id: import_v43.z.string(),
+            type: import_v43.z.literal("function"),
+            function: import_v43.z.object({
+              name: import_v43.z.string(),
+              arguments: import_v43.z.string()
+            })
           })
-        ])
-      })
-    })
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("content-delta"),
-    index: import_v43.z.number(),
-    delta: import_v43.z.object({
-      message: import_v43.z.object({
-        content: import_v43.z.union([
+        ).nullish(),
+        citations: import_v43.z.array(
           import_v43.z.object({
-            text: import_v43.z.string()
-          }),
-          import_v43.z.object({
-            thinking: import_v43.z.string()
+            start: import_v43.z.number(),
+            end: import_v43.z.number(),
+            text: import_v43.z.string(),
+            sources: import_v43.z.array(
+              import_v43.z.object({
+                type: import_v43.z.string().optional(),
+                id: import_v43.z.string().optional(),
+                document: import_v43.z.object({
+                  id: import_v43.z.string().optional(),
+                  text: import_v43.z.string(),
+                  title: import_v43.z.string()
+                })
+              })
+            ),
+            type: import_v43.z.string().optional()
           })
-        ])
-      })
-    })
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("content-end"),
-    index: import_v43.z.number()
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("message-start"),
-    id: import_v43.z.string().nullish()
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("message-end"),
-    delta: import_v43.z.object({
+        ).nullish()
+      }),
       finish_reason: import_v43.z.string(),
       usage: import_v43.z.object({
+        billed_units: import_v43.z.object({
+          input_tokens: import_v43.z.number(),
+          output_tokens: import_v43.z.number()
+        }),
         tokens: import_v43.z.object({
           input_tokens: import_v43.z.number(),
           output_tokens: import_v43.z.number()
         })
       })
     })
-  }),
-  // https://docs.cohere.com/v2/docs/streaming#tool-use-stream-events-for-tool-calling
-  import_v43.z.object({
-    type: import_v43.z.literal("tool-plan-delta"),
-    delta: import_v43.z.object({
-      message: import_v43.z.object({
-        tool_plan: import_v43.z.string()
-      })
-    })
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("tool-call-start"),
-    delta: import_v43.z.object({
-      message: import_v43.z.object({
-        tool_calls: import_v43.z.object({
-          id: import_v43.z.string(),
-          type: import_v43.z.literal("function"),
-          function: import_v43.z.object({
-            name: import_v43.z.string(),
-            arguments: import_v43.z.string()
+  )
+);
+var cohereChatChunkSchema = (0, import_provider_utils4.lazySchema)(
+  () => (0, import_provider_utils4.zodSchema)(
+    import_v43.z.discriminatedUnion("type", [
+      import_v43.z.object({
+        type: import_v43.z.literal("citation-start")
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("citation-end")
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("content-start"),
+        index: import_v43.z.number(),
+        delta: import_v43.z.object({
+          message: import_v43.z.object({
+            content: import_v43.z.union([
+              import_v43.z.object({
+                type: import_v43.z.literal("text"),
+                text: import_v43.z.string()
+              }),
+              import_v43.z.object({
+                type: import_v43.z.literal("thinking"),
+                thinking: import_v43.z.string()
+              })
+            ])
           })
         })
-      })
-    })
-  }),
-  // A single tool call's `arguments` stream in chunks and must be accumulated
-  // in a string and so the full tool object info can only be parsed once we see
-  // `tool-call-end`.
-  import_v43.z.object({
-    type: import_v43.z.literal("tool-call-delta"),
-    delta: import_v43.z.object({
-      message: import_v43.z.object({
-        tool_calls: import_v43.z.object({
-          function: import_v43.z.object({
-            arguments: import_v43.z.string()
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("content-delta"),
+        index: import_v43.z.number(),
+        delta: import_v43.z.object({
+          message: import_v43.z.object({
+            content: import_v43.z.union([
+              import_v43.z.object({
+                text: import_v43.z.string()
+              }),
+              import_v43.z.object({
+                thinking: import_v43.z.string()
+              })
+            ])
+          })
+        })
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("content-end"),
+        index: import_v43.z.number()
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("message-start"),
+        id: import_v43.z.string().nullish()
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("message-end"),
+        delta: import_v43.z.object({
+          finish_reason: import_v43.z.string(),
+          usage: import_v43.z.object({
+            tokens: import_v43.z.object({
+              input_tokens: import_v43.z.number(),
+              output_tokens: import_v43.z.number()
+            })
+          })
+        })
+      }),
+      // https://docs.cohere.com/v2/docs/streaming#tool-use-stream-events-for-tool-calling
+      import_v43.z.object({
+        type: import_v43.z.literal("tool-plan-delta"),
+        delta: import_v43.z.object({
+          message: import_v43.z.object({
+            tool_plan: import_v43.z.string()
+          })
+        })
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("tool-call-start"),
+        delta: import_v43.z.object({
+          message: import_v43.z.object({
+            tool_calls: import_v43.z.object({
+              id: import_v43.z.string(),
+              type: import_v43.z.literal("function"),
+              function: import_v43.z.object({
+                name: import_v43.z.string(),
+                arguments: import_v43.z.string()
+              })
+            })
           })
         })
+      }),
+      // A single tool call's `arguments` stream in chunks and must be accumulated
+      // in a string and so the full tool object info can only be parsed once we see
+      // `tool-call-end`.
+      import_v43.z.object({
+        type: import_v43.z.literal("tool-call-delta"),
+        delta: import_v43.z.object({
+          message: import_v43.z.object({
+            tool_calls: import_v43.z.object({
+              function: import_v43.z.object({
+                arguments: import_v43.z.string()
+              })
+            })
+          })
+        })
+      }),
+      import_v43.z.object({
+        type: import_v43.z.literal("tool-call-end")
       })
-    })
-  }),
-  import_v43.z.object({
-    type: import_v43.z.literal("tool-call-end")
-  })
-]);
+    ])
+  )
+);
 
 // src/cohere-embedding-model.ts
 var import_provider3 = require("@ai-sdk/provider");
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
+var import_provider_utils6 = require("@ai-sdk/provider-utils");
 var import_v45 = require("zod/v4");
 
 // src/cohere-embedding-options.ts
+var import_provider_utils5 = require("@ai-sdk/provider-utils");
 var import_v44 = require("zod/v4");
-var cohereEmbeddingOptions = import_v44.z.object({
-  /**
-   * Specifies the type of input passed to the model. Default is `search_query`.
-   *
-   * - "search_document": Used for embeddings stored in a vector database for search use-cases.
-   * - "search_query": Used for embeddings of search queries run against a vector DB to find relevant documents.
-   * - "classification": Used for embeddings passed through a text classifier.
-   * - "clustering": Used for embeddings run through a clustering algorithm.
-   */
-  inputType: import_v44.z.enum(["search_document", "search_query", "classification", "clustering"]).optional(),
-  /**
-   * Specifies how the API will handle inputs longer than the maximum token length.
-   * Default is `END`.
-   *
-   * - "NONE": If selected, when the input exceeds the maximum input token length will return an error.
-   * - "START": Will discard the start of the input until the remaining input is exactly the maximum input token length for the model.
-   * - "END": Will discard the end of the input until the remaining input is exactly the maximum input token length for the model.
-   */
-  truncate: import_v44.z.enum(["NONE", "START", "END"]).optional()
-});
+var cohereEmbeddingOptions = (0, import_provider_utils5.lazySchema)(
+  () => (0, import_provider_utils5.zodSchema)(
+    import_v44.z.object({
+      /**
+       * Specifies the type of input passed to the model. Default is `search_query`.
+       *
+       * - "search_document": Used for embeddings stored in a vector database for search use-cases.
+       * - "search_query": Used for embeddings of search queries run against a vector DB to find relevant documents.
+       * - "classification": Used for embeddings passed through a text classifier.
+       * - "clustering": Used for embeddings run through a clustering algorithm.
+       */
+      inputType: import_v44.z.enum([
+        "search_document",
+        "search_query",
+        "classification",
+        "clustering"
+      ]).optional(),
+      /**
+       * Specifies how the API will handle inputs longer than the maximum token length.
+       * Default is `END`.
+       *
+       * - "NONE": If selected, when the input exceeds the maximum input token length will return an error.
+       * - "START": Will discard the start of the input until the remaining input is exactly the maximum input token length for the model.
+       * - "END": Will discard the end of the input until the remaining input is exactly the maximum input token length for the model.
+       */
+      truncate: import_v44.z.enum(["NONE", "START", "END"]).optional()
+    })
+  )
+);
 
 // src/cohere-embedding-model.ts
 var CohereEmbeddingModel = class {
@@ -832,7 +863,7 @@ var CohereEmbeddingModel = class {
     providerOptions
   }) {
     var _a;
-    const embeddingOptions = await (0, import_provider_utils4.parseProviderOptions)({
+    const embeddingOptions = await (0, import_provider_utils6.parseProviderOptions)({
       provider: "cohere",
       providerOptions,
       schema: cohereEmbeddingOptions
@@ -849,9 +880,9 @@ var CohereEmbeddingModel = class {
       responseHeaders,
       value: response,
       rawValue
-    } = await (0, import_provider_utils4.postJsonToApi)({
+    } = await (0, import_provider_utils6.postJsonToApi)({
       url: `${this.config.baseURL}/embed`,
-      headers: (0, import_provider_utils4.combineHeaders)(this.config.headers(), headers),
+      headers: (0, import_provider_utils6.combineHeaders)(this.config.headers(), headers),
       body: {
         model: this.modelId,
         // The AI SDK only supports 'float' embeddings which are also the only ones
@@ -863,7 +894,7 @@ var CohereEmbeddingModel = class {
         truncate: embeddingOptions == null ? void 0 : embeddingOptions.truncate
       },
       failedResponseHandler: cohereFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils4.createJsonResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils6.createJsonResponseHandler)(
         cohereTextEmbeddingResponseSchema
       ),
       abortSignal,
@@ -876,27 +907,31 @@ var CohereEmbeddingModel = class {
     };
   }
 };
-var cohereTextEmbeddingResponseSchema = import_v45.z.object({
-  embeddings: import_v45.z.object({
-    float: import_v45.z.array(import_v45.z.array(import_v45.z.number()))
-  }),
-  meta: import_v45.z.object({
-    billed_units: import_v45.z.object({
-      input_tokens: import_v45.z.number()
+var cohereTextEmbeddingResponseSchema = (0, import_provider_utils6.lazySchema)(
+  () => (0, import_provider_utils6.zodSchema)(
+    import_v45.z.object({
+      embeddings: import_v45.z.object({
+        float: import_v45.z.array(import_v45.z.array(import_v45.z.number()))
+      }),
+      meta: import_v45.z.object({
+        billed_units: import_v45.z.object({
+          input_tokens: import_v45.z.number()
+        })
+      })
     })
-  })
-});
+  )
+);
 
 // src/version.ts
-var VERSION = true ? "2.0.44" : "0.0.0-test";
+var VERSION = true ? "2.0.46" : "0.0.0-test";
 
 // src/cohere-provider.ts
 function createCohere(options = {}) {
   var _a;
-  const baseURL = (_a = (0, import_provider_utils5.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://api.cohere.com/v2";
-  const getHeaders = () => (0, import_provider_utils5.withUserAgentSuffix)(
+  const baseURL = (_a = (0, import_provider_utils7.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://api.cohere.com/v2";
+  const getHeaders = () => (0, import_provider_utils7.withUserAgentSuffix)(
     {
-      Authorization: `Bearer ${(0, import_provider_utils5.loadApiKey)({
+      Authorization: `Bearer ${(0, import_provider_utils7.loadApiKey)({
         apiKey: options.apiKey,
         environmentVariableName: "COHERE_API_KEY",
         description: "Cohere"
@@ -912,7 +947,7 @@ function createCohere(options = {}) {
       baseURL,
       headers: getHeaders,
       fetch: options.fetch,
-      generateId: (_a2 = options.generateId) != null ? _a2 : import_provider_utils5.generateId
+      generateId: (_a2 = options.generateId) != null ? _a2 : import_provider_utils7.generateId
     });
   };
   const createTextEmbeddingModel = (modelId) => new CohereEmbeddingModel(modelId, {