@ai-sdk/cohere 4.0.65 → 4.0.67

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/CHANGELOG.md CHANGED
@@ -1,5 +1,30 @@
1
1
  # @ai-sdk/cohere
2
2
 
3
+ ## 4.0.67
4
+
5
+ ### Patch Changes
6
+
7
+ - 43ad966: Report `interrupted` when a stream ends cleanly without its provider completion signal, preserving partial output without emitting an error solely for the interruption.
8
+
9
+ This applies to OpenAI (Chat, Responses, and Completions), Anthropic, Google (Generate Content and Interactions), Alibaba, Amazon Bedrock Converse, Cohere, DeepSeek, Groq, Hugging Face Responses, Mistral Chat, Moonshot AI, Open Responses, Perplexity, and xAI Responses. It also covers cutoffs in later Anthropic messages and Google automatic continuations.
10
+
11
+ **Compatibility notice:** These streams previously fell back to `other` or omitted the finish part. Applications must now handle `interrupted` in finish-reason switches, mappings, and custom validation. Check `finishReason === 'interrupted'` to detect incomplete responses; interruption alone does not trigger error handling.
12
+
13
+ Explicit unknown completion reasons still map to `other`. Actual provider, parsing, and transport errors retain their error paths; affected error paths that previously left the finish reason as `other` now report `error`. OpenAI-compatible remains unchanged.
14
+
15
+ - Updated dependencies [10026d1]
16
+ - @ai-sdk/provider@4.0.32
17
+ - @ai-sdk/provider-utils@5.0.66
18
+
19
+ ## 4.0.66
20
+
21
+ ### Patch Changes
22
+
23
+ - 82ecea1: Defer construction of provider validation schemas until first use to reduce package import time.
24
+ - Updated dependencies [b089ce8]
25
+ - Updated dependencies [39a776b]
26
+ - @ai-sdk/provider-utils@5.0.65
27
+
3
28
  ## 4.0.65
4
29
 
5
30
  ### Patch Changes
package/dist/index.d.ts CHANGED
@@ -1,18 +1,14 @@
1
1
  import { EmbeddingModelV4, LanguageModelV4, ProviderV4, RerankingModelV4 } from "@ai-sdk/provider";
2
- import { FetchFunction } from "@ai-sdk/provider-utils";
3
- import { z } from "zod/v4";
2
+ import { FetchFunction, InferSchema } from "@ai-sdk/provider-utils";
4
3
  //#region src/cohere-chat-language-model-options.d.ts
5
4
  type CohereChatModelId = 'command-a-03-2025' | 'command-a-reasoning-08-2025' | 'command-a-vision-07-2025' | 'command-r7b-12-2024' | 'command-r-plus-04-2024' | 'command-r-plus' | 'command-r-08-2024' | 'command-r-03-2024' | 'command-r' | 'command' | 'command-nightly' | 'command-light' | 'command-light-nightly' | (string & {});
6
- declare const cohereLanguageModelChatOptions: z.ZodObject<{
7
- thinking: z.ZodOptional<z.ZodObject<{
8
- type: z.ZodOptional<z.ZodEnum<{
9
- enabled: "enabled";
10
- disabled: "disabled";
11
- }>>;
12
- tokenBudget: z.ZodOptional<z.ZodNumber>;
13
- }, z.core.$strip>>;
14
- }, z.core.$strip>;
15
- type CohereLanguageModelChatOptions = z.infer<typeof cohereLanguageModelChatOptions>;
5
+ declare const cohereLanguageModelChatOptions: import("@ai-sdk/provider-utils").LazySchema<{
6
+ thinking?: {
7
+ type?: "enabled" | "disabled" | undefined;
8
+ tokenBudget?: number | undefined;
9
+ } | undefined;
10
+ }>;
11
+ type CohereLanguageModelChatOptions = InferSchema<typeof cohereLanguageModelChatOptions>;
16
12
  //#endregion
17
13
  //#region src/reranking/cohere-reranking-model-options.d.ts
18
14
  type CohereRerankingModelId = 'rerank-v3.5' | 'rerank-english-v3.0' | 'rerank-multilingual-v3.0' | (string & {});
@@ -33,28 +29,13 @@ type CohereRerankingModelOptions = {
33
29
  //#endregion
34
30
  //#region src/cohere-embedding-model-options.d.ts
35
31
  type CohereEmbeddingModelId = 'embed-english-v3.0' | 'embed-multilingual-v3.0' | 'embed-english-light-v3.0' | 'embed-multilingual-light-v3.0' | 'embed-english-v2.0' | 'embed-english-light-v2.0' | 'embed-multilingual-v2.0' | (string & {});
36
- declare const cohereEmbeddingModelOptions: z.ZodObject<{
37
- embeddingType: z.ZodOptional<z.ZodEnum<{
38
- float: "float";
39
- int8: "int8";
40
- uint8: "uint8";
41
- binary: "binary";
42
- ubinary: "ubinary";
43
- }>>;
44
- inputType: z.ZodOptional<z.ZodEnum<{
45
- search_document: "search_document";
46
- search_query: "search_query";
47
- classification: "classification";
48
- clustering: "clustering";
49
- }>>;
50
- truncate: z.ZodOptional<z.ZodEnum<{
51
- NONE: "NONE";
52
- START: "START";
53
- END: "END";
54
- }>>;
55
- outputDimension: z.ZodOptional<z.ZodUnion<readonly [z.ZodLiteral<256>, z.ZodLiteral<512>, z.ZodLiteral<768>, z.ZodLiteral<1024>, z.ZodLiteral<1536>, z.ZodLiteral<2048>]>>;
56
- }, z.core.$strip>;
57
- type CohereEmbeddingModelOptions = z.infer<typeof cohereEmbeddingModelOptions>;
32
+ declare const cohereEmbeddingModelOptions: import("@ai-sdk/provider-utils").LazySchema<{
33
+ embeddingType?: "float" | "int8" | "uint8" | "binary" | "ubinary" | undefined;
34
+ inputType?: "search_document" | "search_query" | "classification" | "clustering" | undefined;
35
+ truncate?: "NONE" | "START" | "END" | undefined;
36
+ outputDimension?: 256 | 512 | 768 | 1024 | 1536 | 2048 | undefined;
37
+ }>;
38
+ type CohereEmbeddingModelOptions = InferSchema<typeof cohereEmbeddingModelOptions>;
58
39
  //#endregion
59
40
  //#region src/cohere-provider.d.ts
60
41
  interface CohereProvider extends ProviderV4 {
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","names":[],"sources":["../src/cohere-chat-language-model-options.ts","../src/reranking/cohere-reranking-model-options.ts","../src/cohere-embedding-model-options.ts","../src/cohere-provider.ts","../src/version.ts"],"mappings":";;;;KAGY;cAgBC,gCAA8B,EAAA;;;;;;;;GAezC,EAAA,KAAA;KAEU,iCAAiC,EAAE,aACtC;;;KC7BG;KAMA;;;;;;EAMV;;;;;;EAOA;;;;KCzBU;cAUC,6BAA2B,EAAA;;;;;;;;;;;;;;;;;;;;GAiDtC,EAAA,KAAA;KAEU,8BAA8B,EAAE,aACnC;;;UC1CQ,uBAAuB;GACrC,SAAS,oBAAoB;;;;EAK9B,cAAc,SAAS,oBAAoB;;;;EAK3C,UAAU,SAAS,yBAAyB;;;;EAK5C,eAAe,SAAS,yBAAyB;;;;EAKjD,cAAc,SAAS,yBAAyB;;;;EAKhD,mBAAmB,SAAS,yBAAyB;;;;EAKrD,UAAU,SAAS,yBAAyB;;;;EAK5C,eAAe,SAAS,yBAAyB;;UAGlC;;;;;EAKf;;;;;EAMA;;;;EAKA,UAAU;;;;;EAMV,QAAQ;;;;EAKR;;;;;wBAMc,aACd,UAAS,yBACR;;;;qBAuEU,QAAM;;;qBCrKN"}
1
+ {"version":3,"file":"index.d.ts","names":[],"sources":["../src/cohere-chat-language-model-options.ts","../src/reranking/cohere-reranking-model-options.ts","../src/cohere-embedding-model-options.ts","../src/cohere-provider.ts","../src/version.ts"],"mappings":";;;KAQY;cAgBC,iEAA8B;;;;;;KAqB/B,iCAAiC,mBACpC;;;KCtCG;KAMA;;;;;;EAMV;;;;;;EAOA;;;;KCpBU;cAUC,8DAA2B;;;;;;KA4D5B,8BAA8B,mBACjC;;;UCxDQ,uBAAuB;GACrC,SAAS,oBAAoB;;;;EAK9B,cAAc,SAAS,oBAAoB;;;;EAK3C,UAAU,SAAS,yBAAyB;;;;EAK5C,eAAe,SAAS,yBAAyB;;;;EAKjD,cAAc,SAAS,yBAAyB;;;;EAKhD,mBAAmB,SAAS,yBAAyB;;;;EAKrD,UAAU,SAAS,yBAAyB;;;;EAK5C,eAAe,SAAS,yBAAyB;;UAGlC;;;;;EAKf;;;;;EAMA;;;;EAKA,UAAU;;;;;EAMV,QAAQ;;;;EAKR;;;;;wBAMc,aACd,UAAS,yBACR;;;;qBAuEU,QAAM;;;qBCrKN"}
package/dist/index.js CHANGED
@@ -2,7 +2,7 @@ import { NoSuchModelError, TooManyEmbeddingValuesForCallError, UnsupportedFuncti
2
2
  import { WORKFLOW_DESERIALIZE, WORKFLOW_SERIALIZE, combineHeaders, convertToBase64, createEventSourceResponseHandler, createJsonErrorResponseHandler, createJsonResponseHandler, createNullLanguageModelUsage, generateId, getTopLevelMediaType, isCustomReasoning, lazySchema, loadApiKey, mapReasoningToProviderBudget, parseJSON, parseProviderOptions, postJsonToApi, resolveFullMediaType, serializeModelOptions, withUserAgentSuffix, withoutTrailingSlash, zodSchema } from "@ai-sdk/provider-utils";
3
3
  import { z } from "zod/v4";
4
4
  //#region src/cohere-chat-language-model-options.ts
5
- const cohereLanguageModelChatOptions = z.object({
5
+ const cohereLanguageModelChatOptions = lazySchema(() => zodSchema(z.object({
6
6
  /**
7
7
  * Configuration for reasoning features (optional)
8
8
  *
@@ -14,8 +14,8 @@ const cohereLanguageModelChatOptions = z.object({
14
14
  thinking: z.object({
15
15
  type: z.enum(["enabled", "disabled"]).optional(),
16
16
  tokenBudget: z.number().optional()
17
- }).optional() });
18
- const cohereImagePartProviderOptions = z.object({
17
+ }).optional() })));
18
+ const cohereImagePartProviderOptions = lazySchema(() => zodSchema(z.object({
19
19
  /**
20
20
  * Image fidelity level passed through as `image_url.detail` on the Cohere chat API.
21
21
  *
@@ -25,10 +25,10 @@ detail: z.enum([
25
25
  "auto",
26
26
  "low",
27
27
  "high"
28
- ]).optional() });
28
+ ]).optional() })));
29
29
  //#endregion
30
30
  //#region src/cohere-error.ts
31
- const cohereErrorDataSchema = z.object({ message: z.string() });
31
+ const cohereErrorDataSchema = lazySchema(() => zodSchema(z.object({ message: z.string() })));
32
32
  const cohereFailedResponseHandler = createJsonErrorResponseHandler({
33
33
  errorSchema: cohereErrorDataSchema,
34
34
  errorToMessage: (data) => data.message
@@ -395,7 +395,7 @@ var CohereChatLanguageModel = class CohereChatLanguageModel {
395
395
  fetch: this.config.fetch
396
396
  });
397
397
  let finishReason = {
398
- unified: "other",
398
+ unified: "interrupted",
399
399
  raw: void 0
400
400
  };
401
401
  let usage = void 0;
@@ -569,7 +569,7 @@ function resolveCohereThinking({ reasoning, cohereOptions, warnings }) {
569
569
  token_budget: tokenBudget
570
570
  } };
571
571
  }
572
- const cohereUsageSchema = z.looseObject({
572
+ const cohereUsageSchema = () => z.looseObject({
573
573
  billed_units: z.looseObject({
574
574
  input_tokens: z.number(),
575
575
  output_tokens: z.number()
@@ -580,7 +580,7 @@ const cohereUsageSchema = z.looseObject({
580
580
  }),
581
581
  cached_tokens: z.number().nullish()
582
582
  });
583
- const cohereChatResponseSchema = z.object({
583
+ const cohereChatResponseSchema = lazySchema(() => zodSchema(z.object({
584
584
  generation_id: z.string().nullish(),
585
585
  message: z.object({
586
586
  role: z.string(),
@@ -617,9 +617,9 @@ const cohereChatResponseSchema = z.object({
617
617
  })).nullish()
618
618
  }),
619
619
  finish_reason: z.string(),
620
- usage: cohereUsageSchema
621
- });
622
- const cohereChatChunkSchema = z.discriminatedUnion("type", [
620
+ usage: cohereUsageSchema()
621
+ })));
622
+ const cohereChatChunkSchema = lazySchema(() => zodSchema(z.discriminatedUnion("type", [
623
623
  z.object({ type: z.literal("citation-start") }),
624
624
  z.object({ type: z.literal("citation-end") }),
625
625
  z.object({
@@ -650,7 +650,7 @@ const cohereChatChunkSchema = z.discriminatedUnion("type", [
650
650
  type: z.literal("message-end"),
651
651
  delta: z.object({
652
652
  finish_reason: z.string(),
653
- usage: cohereUsageSchema
653
+ usage: cohereUsageSchema()
654
654
  })
655
655
  }),
656
656
  z.object({
@@ -673,10 +673,10 @@ const cohereChatChunkSchema = z.discriminatedUnion("type", [
673
673
  delta: z.object({ message: z.object({ tool_calls: z.object({ function: z.object({ arguments: z.string() }) }) }) })
674
674
  }),
675
675
  z.object({ type: z.literal("tool-call-end") })
676
- ]);
676
+ ])));
677
677
  //#endregion
678
678
  //#region src/cohere-embedding-model-options.ts
679
- const cohereEmbeddingModelOptions = z.object({
679
+ const cohereEmbeddingModelOptions = lazySchema(() => zodSchema(z.object({
680
680
  /**
681
681
  * The format of the returned embedding. Defaults to `float`.
682
682
  * Non-float formats require Embed v3.0 or newer.
@@ -731,7 +731,7 @@ const cohereEmbeddingModelOptions = z.object({
731
731
  z.literal(1536),
732
732
  z.literal(2048)
733
733
  ]).optional()
734
- });
734
+ })));
735
735
  //#endregion
736
736
  //#region src/cohere-embedding-model.ts
737
737
  var CohereEmbeddingModel = class CohereEmbeddingModel {
@@ -869,7 +869,7 @@ var CohereRerankingModel = class {
869
869
  };
870
870
  //#endregion
871
871
  //#region src/version.ts
872
- const VERSION = "4.0.65";
872
+ const VERSION = "4.0.67";
873
873
  //#endregion
874
874
  //#region src/cohere-provider.ts
875
875
  /**
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","names":["_exhaustiveCheck","_exhaustiveCheck"],"sources":["../src/cohere-chat-language-model-options.ts","../src/cohere-error.ts","../src/cohere-prepare-tools.ts","../src/convert-cohere-usage.ts","../src/convert-to-cohere-chat-prompt.ts","../src/map-cohere-finish-reason.ts","../src/cohere-chat-language-model.ts","../src/cohere-embedding-model-options.ts","../src/cohere-embedding-model.ts","../src/reranking/cohere-reranking-api.ts","../src/reranking/cohere-reranking-model-options.ts","../src/reranking/cohere-reranking-model.ts","../src/version.ts","../src/cohere-provider.ts"],"sourcesContent":["import { z } from 'zod/v4';\n\n// https://docs.cohere.com/docs/models\nexport type CohereChatModelId =\n | 'command-a-03-2025'\n | 'command-a-reasoning-08-2025'\n | 'command-a-vision-07-2025'\n | 'command-r7b-12-2024'\n | 'command-r-plus-04-2024'\n | 'command-r-plus'\n | 'command-r-08-2024'\n | 'command-r-03-2024'\n | 'command-r'\n | 'command'\n | 'command-nightly'\n | 'command-light'\n | 'command-light-nightly'\n | (string & {});\n\nexport const cohereLanguageModelChatOptions = z.object({\n /**\n * Configuration for reasoning features (optional)\n *\n * Can be set to an object with the two properties `type` and `tokenBudget`. `type` can be set to `'enabled'` or `'disabled'` (defaults to `'enabled'`).\n * `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\n *\n * @see https://docs.cohere.com/reference/chat#request.body.thinking\n */\n thinking: z\n .object({\n type: z.enum(['enabled', 'disabled']).optional(),\n tokenBudget: z.number().optional(),\n })\n .optional(),\n});\n\nexport type CohereLanguageModelChatOptions = z.infer<\n typeof cohereLanguageModelChatOptions\n>;\n\nexport const cohereImagePartProviderOptions = z.object({\n /**\n * Image fidelity level passed through as `image_url.detail` on the Cohere chat API.\n *\n * @see https://docs.cohere.com/docs/image-inputs\n */\n detail: z.enum(['auto', 'low', 'high']).optional(),\n});\n\nexport type CohereImagePartProviderOptions = z.infer<\n typeof cohereImagePartProviderOptions\n>;\n","import { createJsonErrorResponseHandler } from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\nconst cohereErrorDataSchema = z.object({\n message: z.string(),\n});\n\nexport type CohereErrorData = z.infer<typeof cohereErrorDataSchema>;\n\nexport const cohereFailedResponseHandler = createJsonErrorResponseHandler({\n errorSchema: cohereErrorDataSchema,\n errorToMessage: data => data.message,\n});\n","import {\n UnsupportedFunctionalityError,\n type LanguageModelV4CallOptions,\n type SharedV4Warning,\n} from '@ai-sdk/provider';\nimport type { CohereToolChoice } from './cohere-chat-prompt';\n\nexport function prepareTools({\n tools,\n toolChoice,\n}: {\n tools: LanguageModelV4CallOptions['tools'];\n toolChoice?: LanguageModelV4CallOptions['toolChoice'];\n}): {\n tools:\n | Array<{\n type: 'function';\n function: {\n name: string | undefined;\n description: string | undefined;\n parameters: unknown;\n };\n }>\n | undefined;\n toolChoice: CohereToolChoice;\n toolWarnings: SharedV4Warning[];\n} {\n // when the tools array is empty, change it to undefined to prevent errors:\n tools = tools?.length ? tools : undefined;\n\n const toolWarnings: SharedV4Warning[] = [];\n\n if (tools == null) {\n return { tools: undefined, toolChoice: undefined, toolWarnings };\n }\n\n const cohereTools: Array<{\n type: 'function';\n function: {\n name: string;\n description: string | undefined;\n parameters: unknown;\n };\n }> = [];\n\n for (const tool of tools) {\n if (tool.type === 'provider') {\n toolWarnings.push({\n type: 'unsupported',\n feature: `provider-defined tool ${tool.id}`,\n });\n } else {\n cohereTools.push({\n type: 'function',\n function: {\n name: tool.name,\n description: tool.description,\n parameters: tool.inputSchema,\n },\n });\n }\n }\n\n if (toolChoice == null) {\n return { tools: cohereTools, toolChoice: undefined, toolWarnings };\n }\n\n const type = toolChoice.type;\n\n switch (type) {\n case 'auto':\n return { tools: cohereTools, toolChoice: undefined, toolWarnings };\n\n case 'none':\n return { tools: cohereTools, toolChoice: 'NONE', toolWarnings };\n\n case 'required':\n return { tools: cohereTools, toolChoice: 'REQUIRED', toolWarnings };\n\n case 'tool':\n return {\n tools: cohereTools.filter(\n tool => tool.function.name === toolChoice.toolName,\n ),\n toolChoice: 'REQUIRED',\n toolWarnings,\n };\n\n default: {\n const _exhaustiveCheck: never = type;\n throw new UnsupportedFunctionalityError({\n functionality: `tool choice type: ${_exhaustiveCheck}`,\n });\n }\n }\n}\n","import type { LanguageModelV4Usage } from '@ai-sdk/provider';\nimport { createNullLanguageModelUsage } from '@ai-sdk/provider-utils';\n\nexport type CohereUsageTokens = {\n input_tokens: number;\n output_tokens: number;\n};\n\nexport type CohereUsage = {\n billed_units?: CohereUsageTokens | null;\n tokens: CohereUsageTokens;\n cached_tokens?: number | null;\n};\n\nexport function convertCohereUsage(\n usage: CohereUsage | undefined | null,\n): LanguageModelV4Usage {\n if (usage == null) {\n return createNullLanguageModelUsage();\n }\n\n const tokens = usage.tokens;\n const inputTokens = tokens.input_tokens;\n const outputTokens = tokens.output_tokens;\n const cacheReadTokens = usage.cached_tokens ?? undefined;\n\n return {\n inputTokens: {\n total: inputTokens,\n noCache:\n cacheReadTokens != null ? inputTokens - cacheReadTokens : inputTokens,\n cacheRead: cacheReadTokens,\n cacheWrite: undefined,\n },\n outputTokens: {\n total: outputTokens,\n text: outputTokens,\n reasoning: undefined,\n },\n raw: usage,\n };\n}\n","import {\n UnsupportedFunctionalityError,\n type LanguageModelV4FilePart,\n type LanguageModelV4Prompt,\n type SharedV4Warning,\n} from '@ai-sdk/provider';\nimport {\n convertToBase64,\n getTopLevelMediaType,\n parseProviderOptions,\n resolveFullMediaType,\n} from '@ai-sdk/provider-utils';\nimport { cohereImagePartProviderOptions } from './cohere-chat-language-model-options';\nimport type {\n CohereAssistantMessage,\n CohereChatPrompt,\n CohereUserMessageContent,\n} from './cohere-chat-prompt';\n\nexport async function convertToCohereChatPrompt(\n prompt: LanguageModelV4Prompt,\n): Promise<{\n messages: CohereChatPrompt;\n documents: Array<{\n data: { text: string; title?: string };\n }>;\n warnings: SharedV4Warning[];\n}> {\n const messages: CohereChatPrompt = [];\n const documents: Array<{ data: { text: string; title?: string } }> = [];\n const warnings: SharedV4Warning[] = [];\n\n for (const { role, content } of prompt) {\n switch (role) {\n case 'system': {\n messages.push({ role: 'system', content });\n break;\n }\n\n case 'user': {\n const userContentParts: Array<CohereUserMessageContent> = [];\n let hasImage = false;\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n if (part.text.length > 0) {\n userContentParts.push({ type: 'text', text: part.text });\n }\n break;\n }\n case 'file': {\n if (getTopLevelMediaType(part.mediaType) === 'image') {\n hasImage = true;\n const url = buildImageUrl({ part });\n const cohereOptions =\n (await parseProviderOptions({\n provider: 'cohere',\n providerOptions: part.providerOptions,\n schema: cohereImagePartProviderOptions,\n })) ?? {};\n\n userContentParts.push({\n type: 'image_url',\n image_url: {\n url,\n ...(cohereOptions.detail\n ? { detail: cohereOptions.detail }\n : {}),\n },\n });\n break;\n }\n\n let textContent: string;\n switch (part.data.type) {\n case 'reference': {\n throw new UnsupportedFunctionalityError({\n functionality: 'file parts with provider references',\n });\n }\n case 'url': {\n throw new UnsupportedFunctionalityError({\n functionality: 'File URL data',\n message:\n 'URLs should be downloaded by the AI SDK and not reach this point. This indicates a configuration issue.',\n });\n }\n case 'text': {\n textContent = part.data.text;\n break;\n }\n case 'data': {\n textContent =\n typeof part.data.data === 'string'\n ? part.data.data\n : new TextDecoder().decode(part.data.data);\n break;\n }\n }\n\n documents.push({\n data: {\n text: textContent,\n title: part.filename,\n },\n });\n break;\n }\n }\n }\n\n if (hasImage) {\n messages.push({ role: 'user', content: userContentParts });\n } else {\n messages.push({\n role: 'user',\n content: userContentParts\n .map(p => (p.type === 'text' ? p.text : ''))\n .join(''),\n });\n }\n break;\n }\n\n case 'assistant': {\n let text = '';\n const toolCalls: CohereAssistantMessage['tool_calls'] = [];\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n text += part.text;\n break;\n }\n case 'tool-call': {\n toolCalls.push({\n id: part.toolCallId,\n type: 'function' as const,\n function: {\n name: part.toolName,\n arguments: JSON.stringify(part.input),\n },\n });\n break;\n }\n }\n }\n\n messages.push({\n role: 'assistant',\n content: toolCalls.length > 0 ? undefined : text,\n tool_calls: toolCalls.length > 0 ? toolCalls : undefined,\n tool_plan: undefined,\n });\n\n break;\n }\n case 'tool': {\n messages.push(\n ...content\n .filter(toolResult => toolResult.type !== 'tool-approval-response')\n .map(toolResult => {\n const output = toolResult.output;\n\n let contentValue: string;\n switch (output.type) {\n case 'text':\n case 'error-text':\n contentValue = output.value;\n break;\n case 'execution-denied':\n contentValue = output.reason ?? 'Tool call execution denied.';\n break;\n case 'content':\n case 'json':\n case 'error-json':\n contentValue = JSON.stringify(output.value);\n break;\n }\n\n return {\n role: 'tool' as const,\n content: contentValue,\n tool_call_id: toolResult.toolCallId,\n };\n }),\n );\n\n break;\n }\n default: {\n const _exhaustiveCheck: never = role;\n throw new Error(`Unsupported role: ${_exhaustiveCheck}`);\n }\n }\n }\n\n return { messages, documents, warnings };\n}\n\nfunction buildImageUrl({ part }: { part: LanguageModelV4FilePart }): string {\n switch (part.data.type) {\n case 'url': {\n return part.data.url.toString();\n }\n case 'data': {\n return `data:${resolveFullMediaType({ part })};base64,${convertToBase64(part.data.data)}`;\n }\n case 'reference': {\n throw new UnsupportedFunctionalityError({\n functionality: 'image file parts with provider references',\n });\n }\n case 'text': {\n throw new UnsupportedFunctionalityError({\n functionality: 'image file parts with text data',\n });\n }\n }\n}\n","import type { LanguageModelV4FinishReason } from '@ai-sdk/provider';\n\nexport function mapCohereFinishReason(\n finishReason: string | null | undefined,\n): LanguageModelV4FinishReason['unified'] {\n switch (finishReason) {\n case 'COMPLETE':\n case 'STOP_SEQUENCE':\n return 'stop';\n\n case 'MAX_TOKENS':\n return 'length';\n\n case 'ERROR':\n return 'error';\n\n case 'TOOL_CALL':\n return 'tool-calls';\n\n default:\n return 'other';\n }\n}\n","import type {\n LanguageModelV4,\n LanguageModelV4CallOptions,\n LanguageModelV4Content,\n LanguageModelV4FinishReason,\n LanguageModelV4GenerateResult,\n LanguageModelV4StreamPart,\n LanguageModelV4StreamResult,\n SharedV4Warning,\n} from '@ai-sdk/provider';\nimport {\n combineHeaders,\n createEventSourceResponseHandler,\n createJsonResponseHandler,\n isCustomReasoning,\n mapReasoningToProviderBudget,\n parseProviderOptions,\n parseJSON,\n postJsonToApi,\n serializeModelOptions,\n WORKFLOW_SERIALIZE,\n WORKFLOW_DESERIALIZE,\n type InferSchema,\n type FetchFunction,\n type ParseResult,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport {\n cohereLanguageModelChatOptions,\n type CohereChatModelId,\n} from './cohere-chat-language-model-options';\nimport { cohereFailedResponseHandler } from './cohere-error';\nimport { prepareTools } from './cohere-prepare-tools';\nimport { convertCohereUsage, type CohereUsage } from './convert-cohere-usage';\nimport { convertToCohereChatPrompt } from './convert-to-cohere-chat-prompt';\nimport { mapCohereFinishReason } from './map-cohere-finish-reason';\n\ntype CohereChatConfig = {\n provider: string;\n baseURL: string;\n headers?: () => Record<string, string | undefined>;\n fetch?: FetchFunction;\n generateId: () => string;\n};\n\nexport class CohereChatLanguageModel implements LanguageModelV4 {\n readonly specificationVersion = 'v4';\n\n readonly modelId: CohereChatModelId;\n\n readonly supportedUrls: Record<string, RegExp[]> = {\n 'image/*': [/^https?:\\/\\/.*$/],\n };\n\n private readonly config: CohereChatConfig;\n\n static [WORKFLOW_SERIALIZE](model: CohereChatLanguageModel) {\n return serializeModelOptions({\n modelId: model.modelId,\n config: model.config,\n });\n }\n\n static [WORKFLOW_DESERIALIZE](options: {\n modelId: CohereChatModelId;\n config: CohereChatConfig;\n }) {\n return new CohereChatLanguageModel(options.modelId, options.config);\n }\n\n constructor(modelId: CohereChatModelId, config: CohereChatConfig) {\n this.modelId = modelId;\n this.config = config;\n }\n\n get provider(): string {\n return this.config.provider;\n }\n\n private async getArgs({\n prompt,\n maxOutputTokens,\n temperature,\n topP,\n topK,\n frequencyPenalty,\n presencePenalty,\n stopSequences,\n responseFormat,\n seed,\n reasoning,\n tools,\n toolChoice,\n providerOptions,\n }: LanguageModelV4CallOptions) {\n const warnings: SharedV4Warning[] = [];\n\n const cohereOptions =\n (await parseProviderOptions({\n provider: 'cohere',\n providerOptions,\n schema: cohereLanguageModelChatOptions,\n })) ?? {};\n\n const {\n messages: chatPrompt,\n documents: cohereDocuments,\n warnings: promptWarnings,\n } = await convertToCohereChatPrompt(prompt);\n\n const {\n tools: cohereTools,\n toolChoice: cohereToolChoice,\n toolWarnings,\n } = prepareTools({ tools, toolChoice });\n\n warnings.push(...toolWarnings, ...promptWarnings);\n\n return {\n args: {\n // model id:\n model: this.modelId,\n\n // standardized settings:\n frequency_penalty: frequencyPenalty,\n presence_penalty: presencePenalty,\n max_tokens: maxOutputTokens,\n temperature,\n p: topP,\n k: topK,\n seed,\n stop_sequences: stopSequences,\n\n // response format:\n response_format:\n responseFormat?.type === 'json'\n ? { type: 'json_object', json_schema: responseFormat.schema }\n : undefined,\n\n // messages:\n messages: chatPrompt,\n\n // tools:\n tools: cohereTools,\n tool_choice: cohereToolChoice,\n\n // documents for RAG:\n ...(cohereDocuments.length > 0 && { documents: cohereDocuments }),\n\n // reasoning:\n ...resolveCohereThinking({\n reasoning,\n cohereOptions,\n warnings,\n }),\n },\n warnings,\n };\n }\n\n async doGenerate(\n options: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4GenerateResult> {\n const { args, warnings } = await this.getArgs(options);\n\n const {\n responseHeaders,\n value: response,\n rawValue: rawResponse,\n } = await postJsonToApi({\n url: `${this.config.baseURL}/chat`,\n headers: combineHeaders(this.config.headers?.(), options.headers),\n body: args,\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createJsonResponseHandler(\n cohereChatResponseSchema,\n ),\n abortSignal: options.abortSignal,\n fetch: this.config.fetch,\n });\n\n const content: Array<LanguageModelV4Content> = [];\n\n for (const item of response.message.content ?? []) {\n if (item.type === 'text' && item.text.length > 0) {\n content.push({ type: 'text', text: item.text });\n continue;\n }\n\n if (item.type === 'thinking' && item.thinking.length > 0) {\n content.push({ type: 'reasoning', text: item.thinking });\n continue;\n }\n }\n\n // citations:\n for (const citation of response.message.citations ?? []) {\n content.push({\n type: 'source',\n sourceType: 'document',\n id: this.config.generateId(),\n mediaType: 'text/plain',\n title: citation.sources[0]?.document?.title || 'Document',\n providerMetadata: {\n cohere: {\n start: citation.start,\n end: citation.end,\n text: citation.text,\n sources: citation.sources,\n ...(citation.type && { citationType: citation.type }),\n },\n },\n });\n }\n\n // tool calls:\n for (const toolCall of response.message.tool_calls ?? []) {\n content.push({\n type: 'tool-call' as const,\n toolCallId: toolCall.id,\n toolName: toolCall.function.name,\n // Cohere sometimes returns `null` for tool call arguments for tools\n // defined as having no arguments.\n input: toolCall.function.arguments.replace(/^null$/, '{}'),\n });\n }\n\n return {\n content,\n finishReason: {\n unified: mapCohereFinishReason(response.finish_reason),\n raw: response.finish_reason ?? undefined,\n },\n usage: convertCohereUsage(response.usage),\n request: { body: args },\n response: {\n // TODO timestamp, model id\n id: response.generation_id ?? undefined,\n headers: responseHeaders,\n body: rawResponse,\n },\n warnings,\n };\n }\n\n async doStream(\n options: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4StreamResult> {\n const { args, warnings } = await this.getArgs(options);\n\n const { responseHeaders, value: response } = await postJsonToApi({\n url: `${this.config.baseURL}/chat`,\n headers: combineHeaders(this.config.headers?.(), options.headers),\n body: { ...args, stream: true },\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createEventSourceResponseHandler(\n cohereChatChunkSchema,\n ),\n abortSignal: options.abortSignal,\n fetch: this.config.fetch,\n });\n\n let finishReason: LanguageModelV4FinishReason = {\n unified: 'other',\n raw: undefined,\n };\n let usage: CohereUsage | undefined = undefined;\n\n let pendingToolCall: {\n id: string;\n name: string;\n arguments: string;\n hasFinished: boolean;\n } | null = null;\n\n let isActiveReasoning = false;\n\n return {\n stream: response.pipeThrough(\n new TransformStream<\n ParseResult<z.infer<typeof cohereChatChunkSchema>>,\n LanguageModelV4StreamPart\n >({\n start(controller) {\n controller.enqueue({ type: 'stream-start', warnings });\n },\n\n async transform(chunk, controller) {\n if (options.includeRawChunks) {\n controller.enqueue({ type: 'raw', rawValue: chunk.rawValue });\n }\n\n // handle failed chunk parsing / validation:\n if (!chunk.success) {\n finishReason = { unified: 'error', raw: undefined };\n controller.enqueue({ type: 'error', error: chunk.error });\n return;\n }\n\n const value = chunk.value;\n const type = value.type;\n\n switch (type) {\n case 'content-start': {\n if (value.delta.message.content.type === 'thinking') {\n controller.enqueue({\n type: 'reasoning-start',\n id: String(value.index),\n });\n isActiveReasoning = true;\n return;\n }\n\n controller.enqueue({\n type: 'text-start',\n id: String(value.index),\n });\n return;\n }\n\n case 'content-delta': {\n if ('thinking' in value.delta.message.content) {\n controller.enqueue({\n type: 'reasoning-delta',\n id: String(value.index),\n delta: value.delta.message.content.thinking,\n });\n return;\n }\n\n controller.enqueue({\n type: 'text-delta',\n id: String(value.index),\n delta: value.delta.message.content.text,\n });\n return;\n }\n\n case 'content-end': {\n if (isActiveReasoning) {\n controller.enqueue({\n type: 'reasoning-end',\n id: String(value.index),\n });\n isActiveReasoning = false;\n return;\n }\n\n controller.enqueue({\n type: 'text-end',\n id: String(value.index),\n });\n\n return;\n }\n\n case 'tool-call-start': {\n const toolId = value.delta.message.tool_calls.id;\n const toolName = value.delta.message.tool_calls.function.name;\n const initialArgs =\n value.delta.message.tool_calls.function.arguments;\n\n pendingToolCall = {\n id: toolId,\n name: toolName,\n arguments: initialArgs,\n hasFinished: false,\n };\n\n controller.enqueue({\n type: 'tool-input-start',\n id: toolId,\n toolName,\n });\n\n if (initialArgs.length > 0) {\n controller.enqueue({\n type: 'tool-input-delta',\n id: toolId,\n delta: initialArgs,\n });\n }\n return;\n }\n\n case 'tool-call-delta': {\n if (pendingToolCall && !pendingToolCall.hasFinished) {\n const argsDelta =\n value.delta.message.tool_calls.function.arguments;\n pendingToolCall.arguments += argsDelta;\n\n controller.enqueue({\n type: 'tool-input-delta',\n id: pendingToolCall.id,\n delta: argsDelta,\n });\n }\n return;\n }\n\n case 'tool-call-end': {\n if (pendingToolCall && !pendingToolCall.hasFinished) {\n controller.enqueue({\n type: 'tool-input-end',\n id: pendingToolCall.id,\n });\n\n controller.enqueue({\n type: 'tool-call',\n toolCallId: pendingToolCall.id,\n toolName: pendingToolCall.name,\n input: JSON.stringify(\n await parseJSON({\n text: pendingToolCall.arguments?.trim() || '{}',\n }),\n ),\n });\n\n pendingToolCall.hasFinished = true;\n pendingToolCall = null;\n }\n return;\n }\n\n case 'message-start': {\n controller.enqueue({\n type: 'response-metadata',\n id: value.id ?? undefined,\n });\n return;\n }\n\n case 'message-end': {\n finishReason = {\n unified: mapCohereFinishReason(value.delta.finish_reason),\n raw: value.delta.finish_reason,\n };\n usage = value.delta.usage;\n return;\n }\n\n default: {\n }\n }\n },\n\n flush(controller) {\n controller.enqueue({\n type: 'finish',\n finishReason,\n usage: convertCohereUsage(usage),\n });\n },\n }),\n ),\n request: { body: { ...args, stream: true } },\n response: { headers: responseHeaders },\n };\n }\n}\n\nfunction resolveCohereThinking({\n reasoning,\n cohereOptions,\n warnings,\n}: {\n reasoning: LanguageModelV4CallOptions['reasoning'];\n cohereOptions: InferSchema<typeof cohereLanguageModelChatOptions>;\n warnings: SharedV4Warning[];\n}): { thinking?: { type: string; token_budget?: number } } {\n if (cohereOptions.thinking) {\n return {\n thinking: {\n type: cohereOptions.thinking.type ?? 'enabled',\n token_budget: cohereOptions.thinking.tokenBudget,\n },\n };\n }\n\n if (!isCustomReasoning(reasoning)) {\n return {};\n }\n\n if (reasoning === 'none') {\n return { thinking: { type: 'disabled' } };\n }\n\n const tokenBudget = mapReasoningToProviderBudget({\n reasoning,\n maxOutputTokens: 32768,\n maxReasoningBudget: 32768,\n warnings,\n });\n\n if (tokenBudget == null) {\n return {};\n }\n\n return { thinking: { type: 'enabled', token_budget: tokenBudget } };\n}\n\n// Loose, nested objects included: the parsed value is returned as `usage.raw`.\nconst cohereUsageSchema = z.looseObject({\n billed_units: z\n .looseObject({\n input_tokens: z.number(),\n output_tokens: z.number(),\n })\n .nullish(),\n tokens: z.looseObject({\n input_tokens: z.number(),\n output_tokens: z.number(),\n }),\n cached_tokens: z.number().nullish(),\n});\n\nconst cohereChatResponseSchema = z.object({\n generation_id: z.string().nullish(),\n message: z.object({\n role: z.string(),\n content: z\n .array(\n z.union([\n z.object({\n type: z.literal('text'),\n text: z.string(),\n }),\n z.object({\n type: z.literal('thinking'),\n thinking: z.string(),\n }),\n ]),\n )\n .nullish(),\n tool_plan: z.string().nullish(),\n tool_calls: z\n .array(\n z.object({\n id: z.string(),\n type: z.literal('function'),\n function: z.object({\n name: z.string(),\n arguments: z.string(),\n }),\n }),\n )\n .nullish(),\n citations: z\n .array(\n z.object({\n start: z.number(),\n end: z.number(),\n text: z.string(),\n sources: z.array(\n z.object({\n type: z.string().optional(),\n id: z.string().optional(),\n document: z.object({\n id: z.string().optional(),\n text: z.string(),\n title: z.string(),\n }),\n }),\n ),\n type: z.string().optional(),\n }),\n )\n .nullish(),\n }),\n finish_reason: z.string(),\n usage: cohereUsageSchema,\n});\n\n// limited version of the schema, focussed on what is needed for the implementation\n// this approach limits breakages when the API changes and increases efficiency\nconst cohereChatChunkSchema = z.discriminatedUnion('type', [\n z.object({\n type: z.literal('citation-start'),\n }),\n z.object({\n type: z.literal('citation-end'),\n }),\n z.object({\n type: z.literal('content-start'),\n index: z.number(),\n delta: z.object({\n message: z.object({\n content: z.union([\n z.object({\n type: z.literal('text'),\n text: z.string(),\n }),\n z.object({\n type: z.literal('thinking'),\n thinking: z.string(),\n }),\n ]),\n }),\n }),\n }),\n z.object({\n type: z.literal('content-delta'),\n index: z.number(),\n delta: z.object({\n message: z.object({\n content: z.union([\n z.object({\n text: z.string(),\n }),\n z.object({\n thinking: z.string(),\n }),\n ]),\n }),\n }),\n }),\n z.object({\n type: z.literal('content-end'),\n index: z.number(),\n }),\n z.object({\n type: z.literal('message-start'),\n id: z.string().nullish(),\n }),\n z.object({\n type: z.literal('message-end'),\n delta: z.object({\n finish_reason: z.string(),\n usage: cohereUsageSchema,\n }),\n }),\n // https://docs.cohere.com/v2/docs/streaming#tool-use-stream-events-for-tool-calling\n z.object({\n type: z.literal('tool-plan-delta'),\n delta: z.object({\n message: z.object({\n tool_plan: z.string(),\n }),\n }),\n }),\n z.object({\n type: z.literal('tool-call-start'),\n delta: z.object({\n message: z.object({\n tool_calls: z.object({\n id: z.string(),\n type: z.literal('function'),\n function: z.object({\n name: z.string(),\n arguments: z.string(),\n }),\n }),\n }),\n }),\n }),\n // A single tool call's `arguments` stream in chunks and must be accumulated\n // in a string and so the full tool object info can only be parsed once we see\n // `tool-call-end`.\n z.object({\n type: z.literal('tool-call-delta'),\n delta: z.object({\n message: z.object({\n tool_calls: z.object({\n function: z.object({\n arguments: z.string(),\n }),\n }),\n }),\n }),\n }),\n z.object({\n type: z.literal('tool-call-end'),\n }),\n]);\n","import { z } from 'zod/v4';\n\nexport type CohereEmbeddingModelId =\n | 'embed-english-v3.0'\n | 'embed-multilingual-v3.0'\n | 'embed-english-light-v3.0'\n | 'embed-multilingual-light-v3.0'\n | 'embed-english-v2.0'\n | 'embed-english-light-v2.0'\n | 'embed-multilingual-v2.0'\n | (string & {});\n\nexport const cohereEmbeddingModelOptions = z.object({\n /**\n * The format of the returned embedding. Defaults to `float`.\n * Non-float formats require Embed v3.0 or newer.\n * Binary formats contain packed bytes, not individual vector dimensions.\n */\n embeddingType: z\n .enum(['float', 'int8', 'uint8', 'binary', 'ubinary'])\n .optional(),\n\n /**\n * Specifies the type of input passed to the model. Default is `search_query`.\n *\n * - \"search_document\": Used for embeddings stored in a vector database for search use-cases.\n * - \"search_query\": Used for embeddings of search queries run against a vector DB to find relevant documents.\n * - \"classification\": Used for embeddings passed through a text classifier.\n * - \"clustering\": Used for embeddings run through a clustering algorithm.\n */\n inputType: z\n .enum(['search_document', 'search_query', 'classification', 'clustering'])\n .optional(),\n\n /**\n * Specifies how the API will handle inputs longer than the maximum token length.\n * Default is `END`.\n *\n * - \"NONE\": If selected, when the input exceeds the maximum input token length will return an error.\n * - \"START\": Will discard the start of the input until the remaining input is exactly the maximum input token length for the model.\n * - \"END\": Will discard the end of the input until the remaining input is exactly the maximum input token length for the model.\n */\n truncate: z.enum(['NONE', 'START', 'END']).optional(),\n\n /**\n * The number of dimensions of the output embedding.\n * Only available for `embed-v4.0` and newer models.\n *\n * Embed v4.0 supports `256`, `512`, `1024`, and `1536` (default).\n * Embed v5.0 Pro and Fast also support `768` and `2048` (default).\n */\n outputDimension: z\n .union([\n z.literal(256),\n z.literal(512),\n z.literal(768),\n z.literal(1024),\n z.literal(1536),\n z.literal(2048),\n ])\n .optional(),\n});\n\nexport type CohereEmbeddingModelOptions = z.infer<\n typeof cohereEmbeddingModelOptions\n>;\n","import {\n TooManyEmbeddingValuesForCallError,\n type EmbeddingModelV4,\n} from '@ai-sdk/provider';\nimport {\n combineHeaders,\n createJsonResponseHandler,\n parseProviderOptions,\n postJsonToApi,\n serializeModelOptions,\n WORKFLOW_SERIALIZE,\n WORKFLOW_DESERIALIZE,\n type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport {\n cohereEmbeddingModelOptions,\n type CohereEmbeddingModelId,\n type CohereEmbeddingModelOptions,\n} from './cohere-embedding-model-options';\nimport { cohereFailedResponseHandler } from './cohere-error';\n\ntype CohereEmbeddingConfig = {\n provider: string;\n baseURL: string;\n headers?: () => Record<string, string | undefined>;\n fetch?: FetchFunction;\n};\n\nexport class CohereEmbeddingModel implements EmbeddingModelV4 {\n readonly specificationVersion = 'v4';\n readonly modelId: CohereEmbeddingModelId;\n\n readonly maxEmbeddingsPerCall = 96;\n readonly supportsParallelCalls = true;\n\n private readonly config: CohereEmbeddingConfig;\n\n static [WORKFLOW_SERIALIZE](model: CohereEmbeddingModel) {\n return serializeModelOptions({\n modelId: model.modelId,\n config: model.config,\n });\n }\n\n static [WORKFLOW_DESERIALIZE](options: {\n modelId: CohereEmbeddingModelId;\n config: CohereEmbeddingConfig;\n }) {\n return new CohereEmbeddingModel(options.modelId, options.config);\n }\n\n constructor(modelId: CohereEmbeddingModelId, config: CohereEmbeddingConfig) {\n this.modelId = modelId;\n this.config = config;\n }\n\n get provider(): string {\n return this.config.provider;\n }\n\n async doEmbed({\n values,\n dimensions,\n headers,\n abortSignal,\n providerOptions,\n }: Parameters<EmbeddingModelV4['doEmbed']>[0]): Promise<\n Awaited<ReturnType<EmbeddingModelV4['doEmbed']>>\n > {\n const embeddingOptions = await parseProviderOptions({\n provider: 'cohere',\n providerOptions,\n schema: cohereEmbeddingModelOptions,\n });\n const embeddingType = embeddingOptions?.embeddingType ?? 'float';\n\n if (values.length > this.maxEmbeddingsPerCall) {\n throw new TooManyEmbeddingValuesForCallError({\n provider: this.provider,\n modelId: this.modelId,\n maxEmbeddingsPerCall: this.maxEmbeddingsPerCall,\n values,\n });\n }\n\n const {\n responseHeaders,\n value: response,\n rawValue,\n } = await postJsonToApi({\n url: `${this.config.baseURL}/embed`,\n headers: combineHeaders(this.config.headers?.(), headers),\n body: {\n model: this.modelId,\n embedding_types: [embeddingType],\n texts: values,\n input_type: embeddingOptions?.inputType ?? 'search_query',\n truncate: embeddingOptions?.truncate,\n output_dimension: embeddingOptions?.outputDimension ?? dimensions,\n },\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createJsonResponseHandler(\n cohereTextEmbeddingResponseSchema(embeddingType),\n ),\n abortSignal,\n fetch: this.config.fetch,\n });\n\n return {\n warnings: [],\n embeddings: response.embeddings[embeddingType],\n usage: { tokens: response.meta.billed_units.input_tokens },\n response: { headers: responseHeaders, body: rawValue },\n };\n }\n}\n\n// minimal version of the schema, focussed on what is needed for the implementation\n// this approach limits breakages when the API changes and increases efficiency\nconst cohereTextEmbeddingResponseSchema = (\n embeddingType: NonNullable<CohereEmbeddingModelOptions['embeddingType']>,\n) =>\n z.object({\n embeddings: z.object({\n [embeddingType]: z.array(z.array(z.number())),\n }),\n meta: z.object({\n billed_units: z.object({\n input_tokens: z.number(),\n }),\n }),\n });\n","import { lazySchema, zodSchema } from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\n// https://docs.cohere.com/v2/reference/rerank\nexport type CohereRerankingInput = {\n model: string;\n query: string;\n documents: string[];\n top_n: number | undefined;\n max_tokens_per_doc: number | undefined;\n priority: number | undefined;\n};\n\nexport const cohereRerankingResponseSchema = lazySchema(() =>\n zodSchema(\n z.object({\n id: z.string().nullish(),\n results: z.array(\n z.object({\n index: z.number(),\n relevance_score: z.number(),\n }),\n ),\n meta: z.any(),\n }),\n ),\n);\n","import {\n lazySchema,\n zodSchema,\n type FlexibleSchema,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\n// https://docs.cohere.com/docs/rerank\nexport type CohereRerankingModelId =\n | 'rerank-v3.5'\n | 'rerank-english-v3.0'\n | 'rerank-multilingual-v3.0'\n | (string & {});\n\nexport type CohereRerankingModelOptions = {\n /**\n * Long documents will be automatically truncated to the specified number of tokens.\n *\n * @default 4096\n */\n maxTokensPerDoc?: number;\n\n /**\n * The priority of the request.\n *\n * @default 0\n */\n priority?: number;\n};\n\nexport const cohereRerankingModelOptionsSchema: FlexibleSchema<CohereRerankingModelOptions> =\n lazySchema(() =>\n zodSchema(\n z.object({\n maxTokensPerDoc: z.number().optional(),\n priority: z.number().optional(),\n }),\n ),\n );\n","import type { RerankingModelV4, SharedV4Warning } from '@ai-sdk/provider';\nimport {\n combineHeaders,\n createJsonResponseHandler,\n parseProviderOptions,\n postJsonToApi,\n type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { cohereFailedResponseHandler } from '../cohere-error';\nimport {\n cohereRerankingResponseSchema,\n type CohereRerankingInput,\n} from './cohere-reranking-api';\nimport {\n cohereRerankingModelOptionsSchema,\n type CohereRerankingModelId,\n} from './cohere-reranking-model-options';\ntype CohereRerankingConfig = {\n provider: string;\n baseURL: string;\n headers: () => Record<string, string | undefined>;\n fetch?: FetchFunction;\n};\n\nexport class CohereRerankingModel implements RerankingModelV4 {\n readonly specificationVersion = 'v4';\n readonly modelId: CohereRerankingModelId;\n\n private readonly config: CohereRerankingConfig;\n\n constructor(modelId: CohereRerankingModelId, config: CohereRerankingConfig) {\n this.modelId = modelId;\n this.config = config;\n }\n\n get provider(): string {\n return this.config.provider;\n }\n\n // current implementation is based on v2 of the API: https://docs.cohere.com/v2/reference/rerank\n async doRerank({\n documents,\n headers,\n query,\n topN,\n abortSignal,\n providerOptions,\n }: Parameters<RerankingModelV4['doRerank']>[0]): Promise<\n Awaited<ReturnType<RerankingModelV4['doRerank']>>\n > {\n const rerankingOptions = await parseProviderOptions({\n provider: 'cohere',\n providerOptions,\n schema: cohereRerankingModelOptionsSchema,\n });\n\n const warnings: SharedV4Warning[] = [];\n\n if (documents.type === 'object') {\n warnings.push({\n type: 'compatibility',\n feature: 'object documents',\n details: 'Object documents are converted to strings.',\n });\n }\n\n const {\n responseHeaders,\n value: response,\n rawValue,\n } = await postJsonToApi({\n url: `${this.config.baseURL}/rerank`,\n headers: combineHeaders(this.config.headers(), headers),\n body: {\n model: this.modelId,\n query,\n documents:\n documents.type === 'text'\n ? documents.values\n : documents.values.map(value => JSON.stringify(value)),\n top_n: topN,\n max_tokens_per_doc: rerankingOptions?.maxTokensPerDoc,\n priority: rerankingOptions?.priority,\n } satisfies CohereRerankingInput,\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createJsonResponseHandler(\n cohereRerankingResponseSchema,\n ),\n abortSignal,\n fetch: this.config.fetch,\n });\n\n return {\n ranking: response.results.map(result => ({\n index: result.index,\n relevanceScore: result.relevance_score,\n })),\n warnings,\n response: {\n id: response.id ?? undefined,\n headers: responseHeaders,\n body: rawValue,\n },\n };\n }\n}\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n typeof __PACKAGE_VERSION__ !== 'undefined'\n ? __PACKAGE_VERSION__\n : '0.0.0-test';\n","import {\n NoSuchModelError,\n type EmbeddingModelV4,\n type LanguageModelV4,\n type RerankingModelV4,\n type ProviderV4,\n} from '@ai-sdk/provider';\nimport {\n generateId,\n loadApiKey,\n withoutTrailingSlash,\n withUserAgentSuffix,\n type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { CohereChatLanguageModel } from './cohere-chat-language-model';\nimport type { CohereChatModelId } from './cohere-chat-language-model-options';\nimport { CohereEmbeddingModel } from './cohere-embedding-model';\nimport type { CohereRerankingModelId } from './reranking/cohere-reranking-model-options';\nimport { CohereRerankingModel } from './reranking/cohere-reranking-model';\nimport type { CohereEmbeddingModelId } from './cohere-embedding-model-options';\nimport { VERSION } from './version';\n\nexport interface CohereProvider extends ProviderV4 {\n (modelId: CohereChatModelId): LanguageModelV4;\n\n /**\n * Creates a model for text generation.\n */\n languageModel(modelId: CohereChatModelId): LanguageModelV4;\n\n /**\n * Creates a model for text embeddings.\n */\n embedding(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * Creates a model for text embeddings.\n */\n embeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * @deprecated Use `embedding` instead.\n */\n textEmbedding(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * @deprecated Use `embeddingModel` instead.\n */\n textEmbeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * Creates a model for reranking.\n */\n reranking(modelId: CohereRerankingModelId): RerankingModelV4;\n\n /**\n * Creates a model for reranking.\n */\n rerankingModel(modelId: CohereRerankingModelId): RerankingModelV4;\n}\n\nexport interface CohereProviderSettings {\n /**\n * Use a different URL prefix for API calls, e.g. to use proxy servers.\n * The default prefix is `https://api.cohere.com/v2`.\n */\n baseURL?: string;\n\n /**\n * API key that is being send using the `Authorization` header.\n * It defaults to the `COHERE_API_KEY` environment variable.\n */\n apiKey?: string;\n\n /**\n * Custom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\n * Optional function to generate a unique ID for each request.\n */\n generateId?: () => string;\n}\n\n/**\n * Create a Cohere AI provider instance.\n */\nexport function createCohere(\n options: CohereProviderSettings = {},\n): CohereProvider {\n const baseURL =\n withoutTrailingSlash(options.baseURL) ?? 'https://api.cohere.com/v2';\n\n const getHeaders = () =>\n withUserAgentSuffix(\n {\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'COHERE_API_KEY',\n description: 'Cohere',\n })}`,\n ...options.headers,\n },\n `ai-sdk-cohere/${VERSION}`,\n );\n\n const createChatModel = (modelId: CohereChatModelId) =>\n new CohereChatLanguageModel(modelId, {\n provider: 'cohere.chat',\n baseURL,\n headers: getHeaders,\n fetch: options.fetch,\n generateId: options.generateId ?? generateId,\n });\n\n const createEmbeddingModel = (modelId: CohereEmbeddingModelId) =>\n new CohereEmbeddingModel(modelId, {\n provider: 'cohere.textEmbedding',\n baseURL,\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const createRerankingModel = (modelId: CohereRerankingModelId) =>\n new CohereRerankingModel(modelId, {\n provider: 'cohere.reranking',\n baseURL,\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const provider = function (modelId: CohereChatModelId) {\n if (new.target) {\n throw new Error(\n 'The Cohere model function cannot be called with the new keyword.',\n );\n }\n\n return createChatModel(modelId);\n };\n\n provider.specificationVersion = 'v4' as const;\n provider.languageModel = createChatModel;\n provider.embedding = createEmbeddingModel;\n provider.embeddingModel = createEmbeddingModel;\n provider.textEmbedding = createEmbeddingModel;\n provider.textEmbeddingModel = createEmbeddingModel;\n provider.reranking = createRerankingModel;\n provider.rerankingModel = createRerankingModel;\n\n provider.imageModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n };\n\n return provider;\n}\n\n/**\n * Default Cohere provider instance.\n */\nexport const cohere = createCohere();\n"],"mappings":";;;;AAmBA,MAAa,iCAAiC,EAAE,OAAO;;;;;;;;;AASrD,UAAU,EACP,OAAO;CACN,MAAM,EAAE,KAAK,CAAC,WAAW,UAAU,CAAC,CAAC,CAAC,SAAS;CAC/C,aAAa,EAAE,OAAO,CAAC,CAAC,SAAS;AACnC,CAAC,CAAC,CACD,SAAS,EACd,CAAC;AAMD,MAAa,iCAAiC,EAAE,OAAO;;;;;;AAMrD,QAAQ,EAAE,KAAK;CAAC;CAAQ;CAAO;AAAM,CAAC,CAAC,CAAC,SAAS,EACnD,CAAC;;;AC5CD,MAAM,wBAAwB,EAAE,OAAO,EACrC,SAAS,EAAE,OAAO,EACpB,CAAC;AAID,MAAa,8BAA8B,+BAA+B;CACxE,aAAa;CACb,iBAAgB,SAAQ,KAAK;AAC/B,CAAC;;;ACLD,SAAgB,aAAa,EAC3B,OACA,cAiBA;CAEA,QAAQ,OAAO,SAAS,QAAQ,KAAA;CAEhC,MAAM,eAAkC,CAAC;CAEzC,IAAI,SAAS,MACX,OAAO;EAAE,OAAO,KAAA;EAAW,YAAY,KAAA;EAAW;CAAa;CAGjE,MAAM,cAOD,CAAC;CAEN,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,YAChB,aAAa,KAAK;EAChB,MAAM;EACN,SAAS,yBAAyB,KAAK;CACzC,CAAC;MAED,YAAY,KAAK;EACf,MAAM;EACN,UAAU;GACR,MAAM,KAAK;GACX,aAAa,KAAK;GAClB,YAAY,KAAK;EACnB;CACF,CAAC;CAIL,IAAI,cAAc,MAChB,OAAO;EAAE,OAAO;EAAa,YAAY,KAAA;EAAW;CAAa;CAGnE,MAAM,OAAO,WAAW;CAExB,QAAQ,MAAR;EACE,KAAK,QACH,OAAO;GAAE,OAAO;GAAa,YAAY,KAAA;GAAW;EAAa;EAEnE,KAAK,QACH,OAAO;GAAE,OAAO;GAAa,YAAY;GAAQ;EAAa;EAEhE,KAAK,YACH,OAAO;GAAE,OAAO;GAAa,YAAY;GAAY;EAAa;EAEpE,KAAK,QACH,OAAO;GACL,OAAO,YAAY,QACjB,SAAQ,KAAK,SAAS,SAAS,WAAW,QAC5C;GACA,YAAY;GACZ;EACF;EAEF,SAEE,MAAM,IAAI,8BAA8B,EACtC,eAAe,qBAAqBA,OACtC,CAAC;CAEL;AACF;;;ACjFA,SAAgB,mBACd,OACsB;CACtB,IAAI,SAAS,MACX,OAAO,6BAA6B;CAGtC,MAAM,SAAS,MAAM;CACrB,MAAM,cAAc,OAAO;CAC3B,MAAM,eAAe,OAAO;CAC5B,MAAM,kBAAkB,MAAM,iBAAiB,KAAA;CAE/C,OAAO;EACL,aAAa;GACX,OAAO;GACP,SACE,mBAAmB,OAAO,cAAc,kBAAkB;GAC5D,WAAW;GACX,YAAY,KAAA;EACd;EACA,cAAc;GACZ,OAAO;GACP,MAAM;GACN,WAAW,KAAA;EACb;EACA,KAAK;CACP;AACF;;;ACtBA,eAAsB,0BACpB,QAOC;CACD,MAAM,WAA6B,CAAC;CACpC,MAAM,YAA+D,CAAC;CACtE,MAAM,WAA8B,CAAC;CAErC,KAAK,MAAM,EAAE,MAAM,aAAa,QAC9B,QAAQ,MAAR;EACE,KAAK;GACH,SAAS,KAAK;IAAE,MAAM;IAAU;GAAQ,CAAC;GACzC;EAGF,KAAK,QAAQ;GACX,MAAM,mBAAoD,CAAC;GAC3D,IAAI,WAAW;GAEf,KAAK,MAAM,QAAQ,SACjB,QAAQ,KAAK,MAAb;IACE,KAAK;KACH,IAAI,KAAK,KAAK,SAAS,GACrB,iBAAiB,KAAK;MAAE,MAAM;MAAQ,MAAM,KAAK;KAAK,CAAC;KAEzD;IAEF,KAAK,QAAQ;KACX,IAAI,qBAAqB,KAAK,SAAS,MAAM,SAAS;MACpD,WAAW;MACX,MAAM,MAAM,cAAc,EAAE,KAAK,CAAC;MAClC,MAAM,gBACH,MAAM,qBAAqB;OAC1B,UAAU;OACV,iBAAiB,KAAK;OACtB,QAAQ;MACV,CAAC,KAAM,CAAC;MAEV,iBAAiB,KAAK;OACpB,MAAM;OACN,WAAW;QACT;QACA,GAAI,cAAc,SACd,EAAE,QAAQ,cAAc,OAAO,IAC/B,CAAC;OACP;MACF,CAAC;MACD;KACF;KAEA,IAAI;KACJ,QAAQ,KAAK,KAAK,MAAlB;MACE,KAAK,aACH,MAAM,IAAI,8BAA8B,EACtC,eAAe,sCACjB,CAAC;MAEH,KAAK,OACH,MAAM,IAAI,8BAA8B;OACtC,eAAe;OACf,SACE;MACJ,CAAC;MAEH,KAAK;OACH,cAAc,KAAK,KAAK;OACxB;MAEF,KAAK,QACH,cACE,OAAO,KAAK,KAAK,SAAS,WACtB,KAAK,KAAK,OACV,IAAI,YAAY,CAAC,CAAC,OAAO,KAAK,KAAK,IAAI;KAGjD;KAEA,UAAU,KAAK,EACb,MAAM;MACJ,MAAM;MACN,OAAO,KAAK;KACd,EACF,CAAC;KACD;IACF;GACF;GAGF,IAAI,UACF,SAAS,KAAK;IAAE,MAAM;IAAQ,SAAS;GAAiB,CAAC;QAEzD,SAAS,KAAK;IACZ,MAAM;IACN,SAAS,iBACN,KAAI,MAAM,EAAE,SAAS,SAAS,EAAE,OAAO,EAAG,CAAC,CAC3C,KAAK,EAAE;GACZ,CAAC;GAEH;EACF;EAEA,KAAK,aAAa;GAChB,IAAI,OAAO;GACX,MAAM,YAAkD,CAAC;GAEzD,KAAK,MAAM,QAAQ,SACjB,QAAQ,KAAK,MAAb;IACE,KAAK;KACH,QAAQ,KAAK;KACb;IAEF,KAAK,aACH,UAAU,KAAK;KACb,IAAI,KAAK;KACT,MAAM;KACN,UAAU;MACR,MAAM,KAAK;MACX,WAAW,KAAK,UAAU,KAAK,KAAK;KACtC;IACF,CAAC;GAGL;GAGF,SAAS,KAAK;IACZ,MAAM;IACN,SAAS,UAAU,SAAS,IAAI,KAAA,IAAY;IAC5C,YAAY,UAAU,SAAS,IAAI,YAAY,KAAA;IAC/C,WAAW,KAAA;GACb,CAAC;GAED;EACF;EACA,KAAK;GACH,SAAS,KACP,GAAG,QACA,QAAO,eAAc,WAAW,SAAS,wBAAwB,CAAC,CAClE,KAAI,eAAc;IACjB,MAAM,SAAS,WAAW;IAE1B,IAAI;IACJ,QAAQ,OAAO,MAAf;KACE,KAAK;KACL,KAAK;MACH,eAAe,OAAO;MACtB;KACF,KAAK;MACH,eAAe,OAAO,UAAU;MAChC;KACF,KAAK;KACL,KAAK;KACL,KAAK,cACH,eAAe,KAAK,UAAU,OAAO,KAAK;IAE9C;IAEA,OAAO;KACL,MAAM;KACN,SAAS;KACT,cAAc,WAAW;IAC3B;GACF,CAAC,CACL;GAEA;EAEF,SAEE,MAAM,IAAI,MAAM,qBAAqBC,MAAkB;CAE3D;CAGF,OAAO;EAAE;EAAU;EAAW;CAAS;AACzC;AAEA,SAAS,cAAc,EAAE,QAAmD;CAC1E,QAAQ,KAAK,KAAK,MAAlB;EACE,KAAK,OACH,OAAO,KAAK,KAAK,IAAI,SAAS;EAEhC,KAAK,QACH,OAAO,QAAQ,qBAAqB,EAAE,KAAK,CAAC,EAAE,UAAU,gBAAgB,KAAK,KAAK,IAAI;EAExF,KAAK,aACH,MAAM,IAAI,8BAA8B,EACtC,eAAe,4CACjB,CAAC;EAEH,KAAK,QACH,MAAM,IAAI,8BAA8B,EACtC,eAAe,kCACjB,CAAC;CAEL;AACF;;;AC1NA,SAAgB,sBACd,cACwC;CACxC,QAAQ,cAAR;EACE,KAAK;EACL,KAAK,iBACH,OAAO;EAET,KAAK,cACH,OAAO;EAET,KAAK,SACH,OAAO;EAET,KAAK,aACH,OAAO;EAET,SACE,OAAO;CACX;AACF;;;ACuBA,IAAa,0BAAb,MAAa,wBAAmD;CAW9D,QAAQ,oBAAoB,OAAgC;EAC1D,OAAO,sBAAsB;GAC3B,SAAS,MAAM;GACf,QAAQ,MAAM;EAChB,CAAC;CACH;CAEA,QAAQ,sBAAsB,SAG3B;EACD,OAAO,IAAI,wBAAwB,QAAQ,SAAS,QAAQ,MAAM;CACpE;CAEA,YAAY,SAA4B,QAA0B;EAxBlC,KAAA,uBAAA;EAImB,KAAA,gBAAA,EACjD,WAAW,CAAC,iBAAiB,EAC/B;EAmBE,KAAK,UAAU;EACf,KAAK,SAAS;CAChB;CAEA,IAAI,WAAmB;EACrB,OAAO,KAAK,OAAO;CACrB;CAEA,MAAc,QAAQ,EACpB,QACA,iBACA,aACA,MACA,MACA,kBACA,iBACA,eACA,gBACA,MACA,WACA,OACA,YACA,mBAC6B;EAC7B,MAAM,WAA8B,CAAC;EAErC,MAAM,gBACH,MAAM,qBAAqB;GAC1B,UAAU;GACV;GACA,QAAQ;EACV,CAAC,KAAM,CAAC;EAEV,MAAM,EACJ,UAAU,YACV,WAAW,iBACX,UAAU,mBACR,MAAM,0BAA0B,MAAM;EAE1C,MAAM,EACJ,OAAO,aACP,YAAY,kBACZ,iBACE,aAAa;GAAE;GAAO;EAAW,CAAC;EAEtC,SAAS,KAAK,GAAG,cAAc,GAAG,cAAc;EAEhD,OAAO;GACL,MAAM;IAEJ,OAAO,KAAK;IAGZ,mBAAmB;IACnB,kBAAkB;IAClB,YAAY;IACZ;IACA,GAAG;IACH,GAAG;IACH;IACA,gBAAgB;IAGhB,iBACE,gBAAgB,SAAS,SACrB;KAAE,MAAM;KAAe,aAAa,eAAe;IAAO,IAC1D,KAAA;IAGN,UAAU;IAGV,OAAO;IACP,aAAa;IAGb,GAAI,gBAAgB,SAAS,KAAK,EAAE,WAAW,gBAAgB;IAG/D,GAAG,sBAAsB;KACvB;KACA;KACA;IACF,CAAC;GACH;GACA;EACF;CACF;CAEA,MAAM,WACJ,SACwC;EACxC,MAAM,EAAE,MAAM,aAAa,MAAM,KAAK,QAAQ,OAAO;EAErD,MAAM,EACJ,iBACA,OAAO,UACP,UAAU,gBACR,MAAM,cAAc;GACtB,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,UAAU,GAAG,QAAQ,OAAO;GAChE,MAAM;GACN,uBAAuB;GACvB,2BAA2B,0BACzB,wBACF;GACA,aAAa,QAAQ;GACrB,OAAO,KAAK,OAAO;EACrB,CAAC;EAED,MAAM,UAAyC,CAAC;EAEhD,KAAK,MAAM,QAAQ,SAAS,QAAQ,WAAW,CAAC,GAAG;GACjD,IAAI,KAAK,SAAS,UAAU,KAAK,KAAK,SAAS,GAAG;IAChD,QAAQ,KAAK;KAAE,MAAM;KAAQ,MAAM,KAAK;IAAK,CAAC;IAC9C;GACF;GAEA,IAAI,KAAK,SAAS,cAAc,KAAK,SAAS,SAAS,GAAG;IACxD,QAAQ,KAAK;KAAE,MAAM;KAAa,MAAM,KAAK;IAAS,CAAC;IACvD;GACF;EACF;EAGA,KAAK,MAAM,YAAY,SAAS,QAAQ,aAAa,CAAC,GACpD,QAAQ,KAAK;GACX,MAAM;GACN,YAAY;GACZ,IAAI,KAAK,OAAO,WAAW;GAC3B,WAAW;GACX,OAAO,SAAS,QAAQ,EAAE,EAAE,UAAU,SAAS;GAC/C,kBAAkB,EAChB,QAAQ;IACN,OAAO,SAAS;IAChB,KAAK,SAAS;IACd,MAAM,SAAS;IACf,SAAS,SAAS;IAClB,GAAI,SAAS,QAAQ,EAAE,cAAc,SAAS,KAAK;GACrD,EACF;EACF,CAAC;EAIH,KAAK,MAAM,YAAY,SAAS,QAAQ,cAAc,CAAC,GACrD,QAAQ,KAAK;GACX,MAAM;GACN,YAAY,SAAS;GACrB,UAAU,SAAS,SAAS;GAG5B,OAAO,SAAS,SAAS,UAAU,QAAQ,UAAU,IAAI;EAC3D,CAAC;EAGH,OAAO;GACL;GACA,cAAc;IACZ,SAAS,sBAAsB,SAAS,aAAa;IACrD,KAAK,SAAS,iBAAiB,KAAA;GACjC;GACA,OAAO,mBAAmB,SAAS,KAAK;GACxC,SAAS,EAAE,MAAM,KAAK;GACtB,UAAU;IAER,IAAI,SAAS,iBAAiB,KAAA;IAC9B,SAAS;IACT,MAAM;GACR;GACA;EACF;CACF;CAEA,MAAM,SACJ,SACsC;EACtC,MAAM,EAAE,MAAM,aAAa,MAAM,KAAK,QAAQ,OAAO;EAErD,MAAM,EAAE,iBAAiB,OAAO,aAAa,MAAM,cAAc;GAC/D,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,UAAU,GAAG,QAAQ,OAAO;GAChE,MAAM;IAAE,GAAG;IAAM,QAAQ;GAAK;GAC9B,uBAAuB;GACvB,2BAA2B,iCACzB,qBACF;GACA,aAAa,QAAQ;GACrB,OAAO,KAAK,OAAO;EACrB,CAAC;EAED,IAAI,eAA4C;GAC9C,SAAS;GACT,KAAK,KAAA;EACP;EACA,IAAI,QAAiC,KAAA;EAErC,IAAI,kBAKO;EAEX,IAAI,oBAAoB;EAExB,OAAO;GACL,QAAQ,SAAS,YACf,IAAI,gBAGF;IACA,MAAM,YAAY;KAChB,WAAW,QAAQ;MAAE,MAAM;MAAgB;KAAS,CAAC;IACvD;IAEA,MAAM,UAAU,OAAO,YAAY;KACjC,IAAI,QAAQ,kBACV,WAAW,QAAQ;MAAE,MAAM;MAAO,UAAU,MAAM;KAAS,CAAC;KAI9D,IAAI,CAAC,MAAM,SAAS;MAClB,eAAe;OAAE,SAAS;OAAS,KAAK,KAAA;MAAU;MAClD,WAAW,QAAQ;OAAE,MAAM;OAAS,OAAO,MAAM;MAAM,CAAC;MACxD;KACF;KAEA,MAAM,QAAQ,MAAM;KAGpB,QAFa,MAAM,MAEnB;MACE,KAAK;OACH,IAAI,MAAM,MAAM,QAAQ,QAAQ,SAAS,YAAY;QACnD,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,OAAO,MAAM,KAAK;QACxB,CAAC;QACD,oBAAoB;QACpB;OACF;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,OAAO,MAAM,KAAK;OACxB,CAAC;OACD;MAGF,KAAK;OACH,IAAI,cAAc,MAAM,MAAM,QAAQ,SAAS;QAC7C,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,OAAO,MAAM,KAAK;SACtB,OAAO,MAAM,MAAM,QAAQ,QAAQ;QACrC,CAAC;QACD;OACF;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,OAAO,MAAM,KAAK;QACtB,OAAO,MAAM,MAAM,QAAQ,QAAQ;OACrC,CAAC;OACD;MAGF,KAAK;OACH,IAAI,mBAAmB;QACrB,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,OAAO,MAAM,KAAK;QACxB,CAAC;QACD,oBAAoB;QACpB;OACF;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,OAAO,MAAM,KAAK;OACxB,CAAC;OAED;MAGF,KAAK,mBAAmB;OACtB,MAAM,SAAS,MAAM,MAAM,QAAQ,WAAW;OAC9C,MAAM,WAAW,MAAM,MAAM,QAAQ,WAAW,SAAS;OACzD,MAAM,cACJ,MAAM,MAAM,QAAQ,WAAW,SAAS;OAE1C,kBAAkB;QAChB,IAAI;QACJ,MAAM;QACN,WAAW;QACX,aAAa;OACf;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI;QACJ;OACF,CAAC;OAED,IAAI,YAAY,SAAS,GACvB,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI;QACJ,OAAO;OACT,CAAC;OAEH;MACF;MAEA,KAAK;OACH,IAAI,mBAAmB,CAAC,gBAAgB,aAAa;QACnD,MAAM,YACJ,MAAM,MAAM,QAAQ,WAAW,SAAS;QAC1C,gBAAgB,aAAa;QAE7B,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,gBAAgB;SACpB,OAAO;QACT,CAAC;OACH;OACA;MAGF,KAAK;OACH,IAAI,mBAAmB,CAAC,gBAAgB,aAAa;QACnD,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,gBAAgB;QACtB,CAAC;QAED,WAAW,QAAQ;SACjB,MAAM;SACN,YAAY,gBAAgB;SAC5B,UAAU,gBAAgB;SAC1B,OAAO,KAAK,UACV,MAAM,UAAU,EACd,MAAM,gBAAgB,WAAW,KAAK,KAAK,KAC7C,CAAC,CACH;QACF,CAAC;QAED,gBAAgB,cAAc;QAC9B,kBAAkB;OACpB;OACA;MAGF,KAAK;OACH,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,MAAM,MAAM,KAAA;OAClB,CAAC;OACD;MAGF,KAAK;OACH,eAAe;QACb,SAAS,sBAAsB,MAAM,MAAM,aAAa;QACxD,KAAK,MAAM,MAAM;OACnB;OACA,QAAQ,MAAM,MAAM;OACpB;KAKJ;IACF;IAEA,MAAM,YAAY;KAChB,WAAW,QAAQ;MACjB,MAAM;MACN;MACA,OAAO,mBAAmB,KAAK;KACjC,CAAC;IACH;GACF,CAAC,CACH;GACA,SAAS,EAAE,MAAM;IAAE,GAAG;IAAM,QAAQ;GAAK,EAAE;GAC3C,UAAU,EAAE,SAAS,gBAAgB;EACvC;CACF;AACF;AAEA,SAAS,sBAAsB,EAC7B,WACA,eACA,YAKyD;CACzD,IAAI,cAAc,UAChB,OAAO,EACL,UAAU;EACR,MAAM,cAAc,SAAS,QAAQ;EACrC,cAAc,cAAc,SAAS;CACvC,EACF;CAGF,IAAI,CAAC,kBAAkB,SAAS,GAC9B,OAAO,CAAC;CAGV,IAAI,cAAc,QAChB,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,EAAE;CAG1C,MAAM,cAAc,6BAA6B;EAC/C;EACA,iBAAiB;EACjB,oBAAoB;EACpB;CACF,CAAC;CAED,IAAI,eAAe,MACjB,OAAO,CAAC;CAGV,OAAO,EAAE,UAAU;EAAE,MAAM;EAAW,cAAc;CAAY,EAAE;AACpE;AAGA,MAAM,oBAAoB,EAAE,YAAY;CACtC,cAAc,EACX,YAAY;EACX,cAAc,EAAE,OAAO;EACvB,eAAe,EAAE,OAAO;CAC1B,CAAC,CAAC,CACD,QAAQ;CACX,QAAQ,EAAE,YAAY;EACpB,cAAc,EAAE,OAAO;EACvB,eAAe,EAAE,OAAO;CAC1B,CAAC;CACD,eAAe,EAAE,OAAO,CAAC,CAAC,QAAQ;AACpC,CAAC;AAED,MAAM,2BAA2B,EAAE,OAAO;CACxC,eAAe,EAAE,OAAO,CAAC,CAAC,QAAQ;CAClC,SAAS,EAAE,OAAO;EAChB,MAAM,EAAE,OAAO;EACf,SAAS,EACN,MACC,EAAE,MAAM,CACN,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,MAAM;GACtB,MAAM,EAAE,OAAO;EACjB,CAAC,GACD,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;EACrB,CAAC,CACH,CAAC,CACH,CAAC,CACA,QAAQ;EACX,WAAW,EAAE,OAAO,CAAC,CAAC,QAAQ;EAC9B,YAAY,EACT,MACC,EAAE,OAAO;GACP,IAAI,EAAE,OAAO;GACb,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;IACjB,MAAM,EAAE,OAAO;IACf,WAAW,EAAE,OAAO;GACtB,CAAC;EACH,CAAC,CACH,CAAC,CACA,QAAQ;EACX,WAAW,EACR,MACC,EAAE,OAAO;GACP,OAAO,EAAE,OAAO;GAChB,KAAK,EAAE,OAAO;GACd,MAAM,EAAE,OAAO;GACf,SAAS,EAAE,MACT,EAAE,OAAO;IACP,MAAM,EAAE,OAAO,CAAC,CAAC,SAAS;IAC1B,IAAI,EAAE,OAAO,CAAC,CAAC,SAAS;IACxB,UAAU,EAAE,OAAO;KACjB,IAAI,EAAE,OAAO,CAAC,CAAC,SAAS;KACxB,MAAM,EAAE,OAAO;KACf,OAAO,EAAE,OAAO;IAClB,CAAC;GACH,CAAC,CACH;GACA,MAAM,EAAE,OAAO,CAAC,CAAC,SAAS;EAC5B,CAAC,CACH,CAAC,CACA,QAAQ;CACb,CAAC;CACD,eAAe,EAAE,OAAO;CACxB,OAAO;AACT,CAAC;AAID,MAAM,wBAAwB,EAAE,mBAAmB,QAAQ;CACzD,EAAE,OAAO,EACP,MAAM,EAAE,QAAQ,gBAAgB,EAClC,CAAC;CACD,EAAE,OAAO,EACP,MAAM,EAAE,QAAQ,cAAc,EAChC,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,eAAe;EAC/B,OAAO,EAAE,OAAO;EAChB,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,CACf,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,MAAM;GACtB,MAAM,EAAE,OAAO;EACjB,CAAC,GACD,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;EACrB,CAAC,CACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,eAAe;EAC/B,OAAO,EAAE,OAAO;EAChB,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,CACf,EAAE,OAAO,EACP,MAAM,EAAE,OAAO,EACjB,CAAC,GACD,EAAE,OAAO,EACP,UAAU,EAAE,OAAO,EACrB,CAAC,CACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,aAAa;EAC7B,OAAO,EAAE,OAAO;CAClB,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,eAAe;EAC/B,IAAI,EAAE,OAAO,CAAC,CAAC,QAAQ;CACzB,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,aAAa;EAC7B,OAAO,EAAE,OAAO;GACd,eAAe,EAAE,OAAO;GACxB,OAAO;EACT,CAAC;CACH,CAAC;CAED,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,iBAAiB;EACjC,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,WAAW,EAAE,OAAO,EACtB,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,iBAAiB;EACjC,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,YAAY,EAAE,OAAO;GACnB,IAAI,EAAE,OAAO;GACb,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;IACjB,MAAM,EAAE,OAAO;IACf,WAAW,EAAE,OAAO;GACtB,CAAC;EACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CAID,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,iBAAiB;EACjC,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,YAAY,EAAE,OAAO,EACnB,UAAU,EAAE,OAAO,EACjB,WAAW,EAAE,OAAO,EACtB,CAAC,EACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO,EACP,MAAM,EAAE,QAAQ,eAAe,EACjC,CAAC;AACH,CAAC;;;ACrpBD,MAAa,8BAA8B,EAAE,OAAO;;;;;;CAMlD,eAAe,EACZ,KAAK;EAAC;EAAS;EAAQ;EAAS;EAAU;CAAS,CAAC,CAAC,CACrD,SAAS;;;;;;;;;CAUZ,WAAW,EACR,KAAK;EAAC;EAAmB;EAAgB;EAAkB;CAAY,CAAC,CAAC,CACzE,SAAS;;;;;;;;;CAUZ,UAAU,EAAE,KAAK;EAAC;EAAQ;EAAS;CAAK,CAAC,CAAC,CAAC,SAAS;;;;;;;;CASpD,iBAAiB,EACd,MAAM;EACL,EAAE,QAAQ,GAAG;EACb,EAAE,QAAQ,GAAG;EACb,EAAE,QAAQ,GAAG;EACb,EAAE,QAAQ,IAAI;EACd,EAAE,QAAQ,IAAI;EACd,EAAE,QAAQ,IAAI;CAChB,CAAC,CAAC,CACD,SAAS;AACd,CAAC;;;AChCD,IAAa,uBAAb,MAAa,qBAAiD;CAS5D,QAAQ,oBAAoB,OAA6B;EACvD,OAAO,sBAAsB;GAC3B,SAAS,MAAM;GACf,QAAQ,MAAM;EAChB,CAAC;CACH;CAEA,QAAQ,sBAAsB,SAG3B;EACD,OAAO,IAAI,qBAAqB,QAAQ,SAAS,QAAQ,MAAM;CACjE;CAEA,YAAY,SAAiC,QAA+B;EAtB5C,KAAA,uBAAA;EAGA,KAAA,uBAAA;EACC,KAAA,wBAAA;EAmB/B,KAAK,UAAU;EACf,KAAK,SAAS;CAChB;CAEA,IAAI,WAAmB;EACrB,OAAO,KAAK,OAAO;CACrB;CAEA,MAAM,QAAQ,EACZ,QACA,YACA,SACA,aACA,mBAGA;EACA,MAAM,mBAAmB,MAAM,qBAAqB;GAClD,UAAU;GACV;GACA,QAAQ;EACV,CAAC;EACD,MAAM,gBAAgB,kBAAkB,iBAAiB;EAEzD,IAAI,OAAO,SAAS,KAAK,sBACvB,MAAM,IAAI,mCAAmC;GAC3C,UAAU,KAAK;GACf,SAAS,KAAK;GACd,sBAAsB,KAAK;GAC3B;EACF,CAAC;EAGH,MAAM,EACJ,iBACA,OAAO,UACP,aACE,MAAM,cAAc;GACtB,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,UAAU,GAAG,OAAO;GACxD,MAAM;IACJ,OAAO,KAAK;IACZ,iBAAiB,CAAC,aAAa;IAC/B,OAAO;IACP,YAAY,kBAAkB,aAAa;IAC3C,UAAU,kBAAkB;IAC5B,kBAAkB,kBAAkB,mBAAmB;GACzD;GACA,uBAAuB;GACvB,2BAA2B,0BACzB,kCAAkC,aAAa,CACjD;GACA;GACA,OAAO,KAAK,OAAO;EACrB,CAAC;EAED,OAAO;GACL,UAAU,CAAC;GACX,YAAY,SAAS,WAAW;GAChC,OAAO,EAAE,QAAQ,SAAS,KAAK,aAAa,aAAa;GACzD,UAAU;IAAE,SAAS;IAAiB,MAAM;GAAS;EACvD;CACF;AACF;AAIA,MAAM,qCACJ,kBAEA,EAAE,OAAO;CACP,YAAY,EAAE,OAAO,GAClB,gBAAgB,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,EAC9C,CAAC;CACD,MAAM,EAAE,OAAO,EACb,cAAc,EAAE,OAAO,EACrB,cAAc,EAAE,OAAO,EACzB,CAAC,EACH,CAAC;AACH,CAAC;;;ACvHH,MAAa,gCAAgC,iBAC3C,UACE,EAAE,OAAO;CACP,IAAI,EAAE,OAAO,CAAC,CAAC,QAAQ;CACvB,SAAS,EAAE,MACT,EAAE,OAAO;EACP,OAAO,EAAE,OAAO;EAChB,iBAAiB,EAAE,OAAO;CAC5B,CAAC,CACH;CACA,MAAM,EAAE,IAAI;AACd,CAAC,CACH,CACF;;;ACIA,MAAa,oCACX,iBACE,UACE,EAAE,OAAO;CACP,iBAAiB,EAAE,OAAO,CAAC,CAAC,SAAS;CACrC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS;AAChC,CAAC,CACH,CACF;;;ACdF,IAAa,uBAAb,MAA8D;CAM5D,YAAY,SAAiC,QAA+B;EAL5C,KAAA,uBAAA;EAM9B,KAAK,UAAU;EACf,KAAK,SAAS;CAChB;CAEA,IAAI,WAAmB;EACrB,OAAO,KAAK,OAAO;CACrB;CAGA,MAAM,SAAS,EACb,WACA,SACA,OACA,MACA,aACA,mBAGA;EACA,MAAM,mBAAmB,MAAM,qBAAqB;GAClD,UAAU;GACV;GACA,QAAQ;EACV,CAAC;EAED,MAAM,WAA8B,CAAC;EAErC,IAAI,UAAU,SAAS,UACrB,SAAS,KAAK;GACZ,MAAM;GACN,SAAS;GACT,SAAS;EACX,CAAC;EAGH,MAAM,EACJ,iBACA,OAAO,UACP,aACE,MAAM,cAAc;GACtB,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;GACtD,MAAM;IACJ,OAAO,KAAK;IACZ;IACA,WACE,UAAU,SAAS,SACf,UAAU,SACV,UAAU,OAAO,KAAI,UAAS,KAAK,UAAU,KAAK,CAAC;IACzD,OAAO;IACP,oBAAoB,kBAAkB;IACtC,UAAU,kBAAkB;GAC9B;GACA,uBAAuB;GACvB,2BAA2B,0BACzB,6BACF;GACA;GACA,OAAO,KAAK,OAAO;EACrB,CAAC;EAED,OAAO;GACL,SAAS,SAAS,QAAQ,KAAI,YAAW;IACvC,OAAO,OAAO;IACd,gBAAgB,OAAO;GACzB,EAAE;GACF;GACA,UAAU;IACR,IAAI,SAAS,MAAM,KAAA;IACnB,SAAS;IACT,MAAM;GACR;EACF;CACF;AACF;;;ACvGA,MAAa,UAAA;;;;;;AC4Fb,SAAgB,aACd,UAAkC,CAAC,GACnB;CAChB,MAAM,UACJ,qBAAqB,QAAQ,OAAO,KAAK;CAE3C,MAAM,mBACJ,oBACE;EACE,eAAe,UAAU,WAAW;GAClC,QAAQ,QAAQ;GAChB,yBAAyB;GACzB,aAAa;EACf,CAAC;EACD,GAAG,QAAQ;CACb,GACA,iBAAiB,SACnB;CAEF,MAAM,mBAAmB,YACvB,IAAI,wBAAwB,SAAS;EACnC,UAAU;EACV;EACA,SAAS;EACT,OAAO,QAAQ;EACf,YAAY,QAAQ,cAAc;CACpC,CAAC;CAEH,MAAM,wBAAwB,YAC5B,IAAI,qBAAqB,SAAS;EAChC,UAAU;EACV;EACA,SAAS;EACT,OAAO,QAAQ;CACjB,CAAC;CAEH,MAAM,wBAAwB,YAC5B,IAAI,qBAAqB,SAAS;EAChC,UAAU;EACV;EACA,SAAS;EACT,OAAO,QAAQ;CACjB,CAAC;CAEH,MAAM,WAAW,SAAU,SAA4B;EACrD,IAAI,YACF,MAAM,IAAI,MACR,kEACF;EAGF,OAAO,gBAAgB,OAAO;CAChC;CAEA,SAAS,uBAAuB;CAChC,SAAS,gBAAgB;CACzB,SAAS,YAAY;CACrB,SAAS,iBAAiB;CAC1B,SAAS,gBAAgB;CACzB,SAAS,qBAAqB;CAC9B,SAAS,YAAY;CACrB,SAAS,iBAAiB;CAE1B,SAAS,cAAc,YAAoB;EACzC,MAAM,IAAI,iBAAiB;GAAE;GAAS,WAAW;EAAa,CAAC;CACjE;CAEA,OAAO;AACT;;;;AAKA,MAAa,SAAS,aAAa"}
1
+ {"version":3,"file":"index.js","names":["_exhaustiveCheck","_exhaustiveCheck"],"sources":["../src/cohere-chat-language-model-options.ts","../src/cohere-error.ts","../src/cohere-prepare-tools.ts","../src/convert-cohere-usage.ts","../src/convert-to-cohere-chat-prompt.ts","../src/map-cohere-finish-reason.ts","../src/cohere-chat-language-model.ts","../src/cohere-embedding-model-options.ts","../src/cohere-embedding-model.ts","../src/reranking/cohere-reranking-api.ts","../src/reranking/cohere-reranking-model-options.ts","../src/reranking/cohere-reranking-model.ts","../src/version.ts","../src/cohere-provider.ts"],"sourcesContent":["import {\n lazySchema,\n zodSchema,\n type InferSchema,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\n// https://docs.cohere.com/docs/models\nexport type CohereChatModelId =\n | 'command-a-03-2025'\n | 'command-a-reasoning-08-2025'\n | 'command-a-vision-07-2025'\n | 'command-r7b-12-2024'\n | 'command-r-plus-04-2024'\n | 'command-r-plus'\n | 'command-r-08-2024'\n | 'command-r-03-2024'\n | 'command-r'\n | 'command'\n | 'command-nightly'\n | 'command-light'\n | 'command-light-nightly'\n | (string & {});\n\nexport const cohereLanguageModelChatOptions = lazySchema(() =>\n zodSchema(\n z.object({\n /**\n * Configuration for reasoning features (optional)\n *\n * Can be set to an object with the two properties `type` and `tokenBudget`. `type` can be set to `'enabled'` or `'disabled'` (defaults to `'enabled'`).\n * `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\n *\n * @see https://docs.cohere.com/reference/chat#request.body.thinking\n */\n thinking: z\n .object({\n type: z.enum(['enabled', 'disabled']).optional(),\n tokenBudget: z.number().optional(),\n })\n .optional(),\n }),\n ),\n);\n\nexport type CohereLanguageModelChatOptions = InferSchema<\n typeof cohereLanguageModelChatOptions\n>;\n\nexport const cohereImagePartProviderOptions = lazySchema(() =>\n zodSchema(\n z.object({\n /**\n * Image fidelity level passed through as `image_url.detail` on the Cohere chat API.\n *\n * @see https://docs.cohere.com/docs/image-inputs\n */\n detail: z.enum(['auto', 'low', 'high']).optional(),\n }),\n ),\n);\n\nexport type CohereImagePartProviderOptions = InferSchema<\n typeof cohereImagePartProviderOptions\n>;\n","import {\n lazySchema,\n zodSchema,\n type InferSchema,\n createJsonErrorResponseHandler,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\nconst cohereErrorDataSchema = lazySchema(() =>\n zodSchema(\n z.object({\n message: z.string(),\n }),\n ),\n);\n\nexport type CohereErrorData = InferSchema<typeof cohereErrorDataSchema>;\n\nexport const cohereFailedResponseHandler = createJsonErrorResponseHandler({\n errorSchema: cohereErrorDataSchema,\n errorToMessage: data => data.message,\n});\n","import {\n UnsupportedFunctionalityError,\n type LanguageModelV4CallOptions,\n type SharedV4Warning,\n} from '@ai-sdk/provider';\nimport type { CohereToolChoice } from './cohere-chat-prompt';\n\nexport function prepareTools({\n tools,\n toolChoice,\n}: {\n tools: LanguageModelV4CallOptions['tools'];\n toolChoice?: LanguageModelV4CallOptions['toolChoice'];\n}): {\n tools:\n | Array<{\n type: 'function';\n function: {\n name: string | undefined;\n description: string | undefined;\n parameters: unknown;\n };\n }>\n | undefined;\n toolChoice: CohereToolChoice;\n toolWarnings: SharedV4Warning[];\n} {\n // when the tools array is empty, change it to undefined to prevent errors:\n tools = tools?.length ? tools : undefined;\n\n const toolWarnings: SharedV4Warning[] = [];\n\n if (tools == null) {\n return { tools: undefined, toolChoice: undefined, toolWarnings };\n }\n\n const cohereTools: Array<{\n type: 'function';\n function: {\n name: string;\n description: string | undefined;\n parameters: unknown;\n };\n }> = [];\n\n for (const tool of tools) {\n if (tool.type === 'provider') {\n toolWarnings.push({\n type: 'unsupported',\n feature: `provider-defined tool ${tool.id}`,\n });\n } else {\n cohereTools.push({\n type: 'function',\n function: {\n name: tool.name,\n description: tool.description,\n parameters: tool.inputSchema,\n },\n });\n }\n }\n\n if (toolChoice == null) {\n return { tools: cohereTools, toolChoice: undefined, toolWarnings };\n }\n\n const type = toolChoice.type;\n\n switch (type) {\n case 'auto':\n return { tools: cohereTools, toolChoice: undefined, toolWarnings };\n\n case 'none':\n return { tools: cohereTools, toolChoice: 'NONE', toolWarnings };\n\n case 'required':\n return { tools: cohereTools, toolChoice: 'REQUIRED', toolWarnings };\n\n case 'tool':\n return {\n tools: cohereTools.filter(\n tool => tool.function.name === toolChoice.toolName,\n ),\n toolChoice: 'REQUIRED',\n toolWarnings,\n };\n\n default: {\n const _exhaustiveCheck: never = type;\n throw new UnsupportedFunctionalityError({\n functionality: `tool choice type: ${_exhaustiveCheck}`,\n });\n }\n }\n}\n","import type { LanguageModelV4Usage } from '@ai-sdk/provider';\nimport { createNullLanguageModelUsage } from '@ai-sdk/provider-utils';\n\nexport type CohereUsageTokens = {\n input_tokens: number;\n output_tokens: number;\n};\n\nexport type CohereUsage = {\n billed_units?: CohereUsageTokens | null;\n tokens: CohereUsageTokens;\n cached_tokens?: number | null;\n};\n\nexport function convertCohereUsage(\n usage: CohereUsage | undefined | null,\n): LanguageModelV4Usage {\n if (usage == null) {\n return createNullLanguageModelUsage();\n }\n\n const tokens = usage.tokens;\n const inputTokens = tokens.input_tokens;\n const outputTokens = tokens.output_tokens;\n const cacheReadTokens = usage.cached_tokens ?? undefined;\n\n return {\n inputTokens: {\n total: inputTokens,\n noCache:\n cacheReadTokens != null ? inputTokens - cacheReadTokens : inputTokens,\n cacheRead: cacheReadTokens,\n cacheWrite: undefined,\n },\n outputTokens: {\n total: outputTokens,\n text: outputTokens,\n reasoning: undefined,\n },\n raw: usage,\n };\n}\n","import {\n UnsupportedFunctionalityError,\n type LanguageModelV4FilePart,\n type LanguageModelV4Prompt,\n type SharedV4Warning,\n} from '@ai-sdk/provider';\nimport {\n convertToBase64,\n getTopLevelMediaType,\n parseProviderOptions,\n resolveFullMediaType,\n} from '@ai-sdk/provider-utils';\nimport { cohereImagePartProviderOptions } from './cohere-chat-language-model-options';\nimport type {\n CohereAssistantMessage,\n CohereChatPrompt,\n CohereUserMessageContent,\n} from './cohere-chat-prompt';\n\nexport async function convertToCohereChatPrompt(\n prompt: LanguageModelV4Prompt,\n): Promise<{\n messages: CohereChatPrompt;\n documents: Array<{\n data: { text: string; title?: string };\n }>;\n warnings: SharedV4Warning[];\n}> {\n const messages: CohereChatPrompt = [];\n const documents: Array<{ data: { text: string; title?: string } }> = [];\n const warnings: SharedV4Warning[] = [];\n\n for (const { role, content } of prompt) {\n switch (role) {\n case 'system': {\n messages.push({ role: 'system', content });\n break;\n }\n\n case 'user': {\n const userContentParts: Array<CohereUserMessageContent> = [];\n let hasImage = false;\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n if (part.text.length > 0) {\n userContentParts.push({ type: 'text', text: part.text });\n }\n break;\n }\n case 'file': {\n if (getTopLevelMediaType(part.mediaType) === 'image') {\n hasImage = true;\n const url = buildImageUrl({ part });\n const cohereOptions =\n (await parseProviderOptions({\n provider: 'cohere',\n providerOptions: part.providerOptions,\n schema: cohereImagePartProviderOptions,\n })) ?? {};\n\n userContentParts.push({\n type: 'image_url',\n image_url: {\n url,\n ...(cohereOptions.detail\n ? { detail: cohereOptions.detail }\n : {}),\n },\n });\n break;\n }\n\n let textContent: string;\n switch (part.data.type) {\n case 'reference': {\n throw new UnsupportedFunctionalityError({\n functionality: 'file parts with provider references',\n });\n }\n case 'url': {\n throw new UnsupportedFunctionalityError({\n functionality: 'File URL data',\n message:\n 'URLs should be downloaded by the AI SDK and not reach this point. This indicates a configuration issue.',\n });\n }\n case 'text': {\n textContent = part.data.text;\n break;\n }\n case 'data': {\n textContent =\n typeof part.data.data === 'string'\n ? part.data.data\n : new TextDecoder().decode(part.data.data);\n break;\n }\n }\n\n documents.push({\n data: {\n text: textContent,\n title: part.filename,\n },\n });\n break;\n }\n }\n }\n\n if (hasImage) {\n messages.push({ role: 'user', content: userContentParts });\n } else {\n messages.push({\n role: 'user',\n content: userContentParts\n .map(p => (p.type === 'text' ? p.text : ''))\n .join(''),\n });\n }\n break;\n }\n\n case 'assistant': {\n let text = '';\n const toolCalls: CohereAssistantMessage['tool_calls'] = [];\n\n for (const part of content) {\n switch (part.type) {\n case 'text': {\n text += part.text;\n break;\n }\n case 'tool-call': {\n toolCalls.push({\n id: part.toolCallId,\n type: 'function' as const,\n function: {\n name: part.toolName,\n arguments: JSON.stringify(part.input),\n },\n });\n break;\n }\n }\n }\n\n messages.push({\n role: 'assistant',\n content: toolCalls.length > 0 ? undefined : text,\n tool_calls: toolCalls.length > 0 ? toolCalls : undefined,\n tool_plan: undefined,\n });\n\n break;\n }\n case 'tool': {\n messages.push(\n ...content\n .filter(toolResult => toolResult.type !== 'tool-approval-response')\n .map(toolResult => {\n const output = toolResult.output;\n\n let contentValue: string;\n switch (output.type) {\n case 'text':\n case 'error-text':\n contentValue = output.value;\n break;\n case 'execution-denied':\n contentValue = output.reason ?? 'Tool call execution denied.';\n break;\n case 'content':\n case 'json':\n case 'error-json':\n contentValue = JSON.stringify(output.value);\n break;\n }\n\n return {\n role: 'tool' as const,\n content: contentValue,\n tool_call_id: toolResult.toolCallId,\n };\n }),\n );\n\n break;\n }\n default: {\n const _exhaustiveCheck: never = role;\n throw new Error(`Unsupported role: ${_exhaustiveCheck}`);\n }\n }\n }\n\n return { messages, documents, warnings };\n}\n\nfunction buildImageUrl({ part }: { part: LanguageModelV4FilePart }): string {\n switch (part.data.type) {\n case 'url': {\n return part.data.url.toString();\n }\n case 'data': {\n return `data:${resolveFullMediaType({ part })};base64,${convertToBase64(part.data.data)}`;\n }\n case 'reference': {\n throw new UnsupportedFunctionalityError({\n functionality: 'image file parts with provider references',\n });\n }\n case 'text': {\n throw new UnsupportedFunctionalityError({\n functionality: 'image file parts with text data',\n });\n }\n }\n}\n","import type { LanguageModelV4FinishReason } from '@ai-sdk/provider';\n\nexport function mapCohereFinishReason(\n finishReason: string | null | undefined,\n): LanguageModelV4FinishReason['unified'] {\n switch (finishReason) {\n case 'COMPLETE':\n case 'STOP_SEQUENCE':\n return 'stop';\n\n case 'MAX_TOKENS':\n return 'length';\n\n case 'ERROR':\n return 'error';\n\n case 'TOOL_CALL':\n return 'tool-calls';\n\n default:\n return 'other';\n }\n}\n","import type {\n LanguageModelV4,\n LanguageModelV4CallOptions,\n LanguageModelV4Content,\n LanguageModelV4FinishReason,\n LanguageModelV4GenerateResult,\n LanguageModelV4StreamPart,\n LanguageModelV4StreamResult,\n SharedV4Warning,\n} from '@ai-sdk/provider';\nimport {\n lazySchema,\n zodSchema,\n combineHeaders,\n createEventSourceResponseHandler,\n createJsonResponseHandler,\n isCustomReasoning,\n mapReasoningToProviderBudget,\n parseProviderOptions,\n parseJSON,\n postJsonToApi,\n serializeModelOptions,\n WORKFLOW_SERIALIZE,\n WORKFLOW_DESERIALIZE,\n type InferSchema,\n type FetchFunction,\n type ParseResult,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport {\n cohereLanguageModelChatOptions,\n type CohereChatModelId,\n} from './cohere-chat-language-model-options';\nimport { cohereFailedResponseHandler } from './cohere-error';\nimport { prepareTools } from './cohere-prepare-tools';\nimport { convertCohereUsage, type CohereUsage } from './convert-cohere-usage';\nimport { convertToCohereChatPrompt } from './convert-to-cohere-chat-prompt';\nimport { mapCohereFinishReason } from './map-cohere-finish-reason';\n\ntype CohereChatConfig = {\n provider: string;\n baseURL: string;\n headers?: () => Record<string, string | undefined>;\n fetch?: FetchFunction;\n generateId: () => string;\n};\n\nexport class CohereChatLanguageModel implements LanguageModelV4 {\n readonly specificationVersion = 'v4';\n\n readonly modelId: CohereChatModelId;\n\n readonly supportedUrls: Record<string, RegExp[]> = {\n 'image/*': [/^https?:\\/\\/.*$/],\n };\n\n private readonly config: CohereChatConfig;\n\n static [WORKFLOW_SERIALIZE](model: CohereChatLanguageModel) {\n return serializeModelOptions({\n modelId: model.modelId,\n config: model.config,\n });\n }\n\n static [WORKFLOW_DESERIALIZE](options: {\n modelId: CohereChatModelId;\n config: CohereChatConfig;\n }) {\n return new CohereChatLanguageModel(options.modelId, options.config);\n }\n\n constructor(modelId: CohereChatModelId, config: CohereChatConfig) {\n this.modelId = modelId;\n this.config = config;\n }\n\n get provider(): string {\n return this.config.provider;\n }\n\n private async getArgs({\n prompt,\n maxOutputTokens,\n temperature,\n topP,\n topK,\n frequencyPenalty,\n presencePenalty,\n stopSequences,\n responseFormat,\n seed,\n reasoning,\n tools,\n toolChoice,\n providerOptions,\n }: LanguageModelV4CallOptions) {\n const warnings: SharedV4Warning[] = [];\n\n const cohereOptions =\n (await parseProviderOptions({\n provider: 'cohere',\n providerOptions,\n schema: cohereLanguageModelChatOptions,\n })) ?? {};\n\n const {\n messages: chatPrompt,\n documents: cohereDocuments,\n warnings: promptWarnings,\n } = await convertToCohereChatPrompt(prompt);\n\n const {\n tools: cohereTools,\n toolChoice: cohereToolChoice,\n toolWarnings,\n } = prepareTools({ tools, toolChoice });\n\n warnings.push(...toolWarnings, ...promptWarnings);\n\n return {\n args: {\n // model id:\n model: this.modelId,\n\n // standardized settings:\n frequency_penalty: frequencyPenalty,\n presence_penalty: presencePenalty,\n max_tokens: maxOutputTokens,\n temperature,\n p: topP,\n k: topK,\n seed,\n stop_sequences: stopSequences,\n\n // response format:\n response_format:\n responseFormat?.type === 'json'\n ? { type: 'json_object', json_schema: responseFormat.schema }\n : undefined,\n\n // messages:\n messages: chatPrompt,\n\n // tools:\n tools: cohereTools,\n tool_choice: cohereToolChoice,\n\n // documents for RAG:\n ...(cohereDocuments.length > 0 && { documents: cohereDocuments }),\n\n // reasoning:\n ...resolveCohereThinking({\n reasoning,\n cohereOptions,\n warnings,\n }),\n },\n warnings,\n };\n }\n\n async doGenerate(\n options: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4GenerateResult> {\n const { args, warnings } = await this.getArgs(options);\n\n const {\n responseHeaders,\n value: response,\n rawValue: rawResponse,\n } = await postJsonToApi({\n url: `${this.config.baseURL}/chat`,\n headers: combineHeaders(this.config.headers?.(), options.headers),\n body: args,\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createJsonResponseHandler(\n cohereChatResponseSchema,\n ),\n abortSignal: options.abortSignal,\n fetch: this.config.fetch,\n });\n\n const content: Array<LanguageModelV4Content> = [];\n\n for (const item of response.message.content ?? []) {\n if (item.type === 'text' && item.text.length > 0) {\n content.push({ type: 'text', text: item.text });\n continue;\n }\n\n if (item.type === 'thinking' && item.thinking.length > 0) {\n content.push({ type: 'reasoning', text: item.thinking });\n continue;\n }\n }\n\n // citations:\n for (const citation of response.message.citations ?? []) {\n content.push({\n type: 'source',\n sourceType: 'document',\n id: this.config.generateId(),\n mediaType: 'text/plain',\n title: citation.sources[0]?.document?.title || 'Document',\n providerMetadata: {\n cohere: {\n start: citation.start,\n end: citation.end,\n text: citation.text,\n sources: citation.sources,\n ...(citation.type && { citationType: citation.type }),\n },\n },\n });\n }\n\n // tool calls:\n for (const toolCall of response.message.tool_calls ?? []) {\n content.push({\n type: 'tool-call' as const,\n toolCallId: toolCall.id,\n toolName: toolCall.function.name,\n // Cohere sometimes returns `null` for tool call arguments for tools\n // defined as having no arguments.\n input: toolCall.function.arguments.replace(/^null$/, '{}'),\n });\n }\n\n return {\n content,\n finishReason: {\n unified: mapCohereFinishReason(response.finish_reason),\n raw: response.finish_reason ?? undefined,\n },\n usage: convertCohereUsage(response.usage),\n request: { body: args },\n response: {\n // TODO timestamp, model id\n id: response.generation_id ?? undefined,\n headers: responseHeaders,\n body: rawResponse,\n },\n warnings,\n };\n }\n\n async doStream(\n options: LanguageModelV4CallOptions,\n ): Promise<LanguageModelV4StreamResult> {\n const { args, warnings } = await this.getArgs(options);\n\n const { responseHeaders, value: response } = await postJsonToApi({\n url: `${this.config.baseURL}/chat`,\n headers: combineHeaders(this.config.headers?.(), options.headers),\n body: { ...args, stream: true },\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createEventSourceResponseHandler(\n cohereChatChunkSchema,\n ),\n abortSignal: options.abortSignal,\n fetch: this.config.fetch,\n });\n\n // A clean end without the provider's completion signal is interrupted.\n let finishReason: LanguageModelV4FinishReason = {\n unified: 'interrupted',\n raw: undefined,\n };\n let usage: CohereUsage | undefined = undefined;\n\n let pendingToolCall: {\n id: string;\n name: string;\n arguments: string;\n hasFinished: boolean;\n } | null = null;\n\n let isActiveReasoning = false;\n\n return {\n stream: response.pipeThrough(\n new TransformStream<\n ParseResult<InferSchema<typeof cohereChatChunkSchema>>,\n LanguageModelV4StreamPart\n >({\n start(controller) {\n controller.enqueue({ type: 'stream-start', warnings });\n },\n\n async transform(chunk, controller) {\n if (options.includeRawChunks) {\n controller.enqueue({ type: 'raw', rawValue: chunk.rawValue });\n }\n\n // handle failed chunk parsing / validation:\n if (!chunk.success) {\n finishReason = { unified: 'error', raw: undefined };\n controller.enqueue({ type: 'error', error: chunk.error });\n return;\n }\n\n const value = chunk.value;\n const type = value.type;\n\n switch (type) {\n case 'content-start': {\n if (value.delta.message.content.type === 'thinking') {\n controller.enqueue({\n type: 'reasoning-start',\n id: String(value.index),\n });\n isActiveReasoning = true;\n return;\n }\n\n controller.enqueue({\n type: 'text-start',\n id: String(value.index),\n });\n return;\n }\n\n case 'content-delta': {\n if ('thinking' in value.delta.message.content) {\n controller.enqueue({\n type: 'reasoning-delta',\n id: String(value.index),\n delta: value.delta.message.content.thinking,\n });\n return;\n }\n\n controller.enqueue({\n type: 'text-delta',\n id: String(value.index),\n delta: value.delta.message.content.text,\n });\n return;\n }\n\n case 'content-end': {\n if (isActiveReasoning) {\n controller.enqueue({\n type: 'reasoning-end',\n id: String(value.index),\n });\n isActiveReasoning = false;\n return;\n }\n\n controller.enqueue({\n type: 'text-end',\n id: String(value.index),\n });\n\n return;\n }\n\n case 'tool-call-start': {\n const toolId = value.delta.message.tool_calls.id;\n const toolName = value.delta.message.tool_calls.function.name;\n const initialArgs =\n value.delta.message.tool_calls.function.arguments;\n\n pendingToolCall = {\n id: toolId,\n name: toolName,\n arguments: initialArgs,\n hasFinished: false,\n };\n\n controller.enqueue({\n type: 'tool-input-start',\n id: toolId,\n toolName,\n });\n\n if (initialArgs.length > 0) {\n controller.enqueue({\n type: 'tool-input-delta',\n id: toolId,\n delta: initialArgs,\n });\n }\n return;\n }\n\n case 'tool-call-delta': {\n if (pendingToolCall && !pendingToolCall.hasFinished) {\n const argsDelta =\n value.delta.message.tool_calls.function.arguments;\n pendingToolCall.arguments += argsDelta;\n\n controller.enqueue({\n type: 'tool-input-delta',\n id: pendingToolCall.id,\n delta: argsDelta,\n });\n }\n return;\n }\n\n case 'tool-call-end': {\n if (pendingToolCall && !pendingToolCall.hasFinished) {\n controller.enqueue({\n type: 'tool-input-end',\n id: pendingToolCall.id,\n });\n\n controller.enqueue({\n type: 'tool-call',\n toolCallId: pendingToolCall.id,\n toolName: pendingToolCall.name,\n input: JSON.stringify(\n await parseJSON({\n text: pendingToolCall.arguments?.trim() || '{}',\n }),\n ),\n });\n\n pendingToolCall.hasFinished = true;\n pendingToolCall = null;\n }\n return;\n }\n\n case 'message-start': {\n controller.enqueue({\n type: 'response-metadata',\n id: value.id ?? undefined,\n });\n return;\n }\n\n case 'message-end': {\n finishReason = {\n unified: mapCohereFinishReason(value.delta.finish_reason),\n raw: value.delta.finish_reason,\n };\n usage = value.delta.usage;\n return;\n }\n\n default: {\n }\n }\n },\n\n flush(controller) {\n controller.enqueue({\n type: 'finish',\n finishReason,\n usage: convertCohereUsage(usage),\n });\n },\n }),\n ),\n request: { body: { ...args, stream: true } },\n response: { headers: responseHeaders },\n };\n }\n}\n\nfunction resolveCohereThinking({\n reasoning,\n cohereOptions,\n warnings,\n}: {\n reasoning: LanguageModelV4CallOptions['reasoning'];\n cohereOptions: InferSchema<typeof cohereLanguageModelChatOptions>;\n warnings: SharedV4Warning[];\n}): { thinking?: { type: string; token_budget?: number } } {\n if (cohereOptions.thinking) {\n return {\n thinking: {\n type: cohereOptions.thinking.type ?? 'enabled',\n token_budget: cohereOptions.thinking.tokenBudget,\n },\n };\n }\n\n if (!isCustomReasoning(reasoning)) {\n return {};\n }\n\n if (reasoning === 'none') {\n return { thinking: { type: 'disabled' } };\n }\n\n const tokenBudget = mapReasoningToProviderBudget({\n reasoning,\n maxOutputTokens: 32768,\n maxReasoningBudget: 32768,\n warnings,\n });\n\n if (tokenBudget == null) {\n return {};\n }\n\n return { thinking: { type: 'enabled', token_budget: tokenBudget } };\n}\n\n// Loose, nested objects included: the parsed value is returned as `usage.raw`.\nconst cohereUsageSchema = () =>\n z.looseObject({\n billed_units: z\n .looseObject({\n input_tokens: z.number(),\n output_tokens: z.number(),\n })\n .nullish(),\n tokens: z.looseObject({\n input_tokens: z.number(),\n output_tokens: z.number(),\n }),\n cached_tokens: z.number().nullish(),\n });\n\nconst cohereChatResponseSchema = lazySchema(() =>\n zodSchema(\n z.object({\n generation_id: z.string().nullish(),\n message: z.object({\n role: z.string(),\n content: z\n .array(\n z.union([\n z.object({\n type: z.literal('text'),\n text: z.string(),\n }),\n z.object({\n type: z.literal('thinking'),\n thinking: z.string(),\n }),\n ]),\n )\n .nullish(),\n tool_plan: z.string().nullish(),\n tool_calls: z\n .array(\n z.object({\n id: z.string(),\n type: z.literal('function'),\n function: z.object({\n name: z.string(),\n arguments: z.string(),\n }),\n }),\n )\n .nullish(),\n citations: z\n .array(\n z.object({\n start: z.number(),\n end: z.number(),\n text: z.string(),\n sources: z.array(\n z.object({\n type: z.string().optional(),\n id: z.string().optional(),\n document: z.object({\n id: z.string().optional(),\n text: z.string(),\n title: z.string(),\n }),\n }),\n ),\n type: z.string().optional(),\n }),\n )\n .nullish(),\n }),\n finish_reason: z.string(),\n usage: cohereUsageSchema(),\n }),\n ),\n);\n\n// limited version of the schema, focussed on what is needed for the implementation\n// this approach limits breakages when the API changes and increases efficiency\nconst cohereChatChunkSchema = lazySchema(() =>\n zodSchema(\n z.discriminatedUnion('type', [\n z.object({\n type: z.literal('citation-start'),\n }),\n z.object({\n type: z.literal('citation-end'),\n }),\n z.object({\n type: z.literal('content-start'),\n index: z.number(),\n delta: z.object({\n message: z.object({\n content: z.union([\n z.object({\n type: z.literal('text'),\n text: z.string(),\n }),\n z.object({\n type: z.literal('thinking'),\n thinking: z.string(),\n }),\n ]),\n }),\n }),\n }),\n z.object({\n type: z.literal('content-delta'),\n index: z.number(),\n delta: z.object({\n message: z.object({\n content: z.union([\n z.object({\n text: z.string(),\n }),\n z.object({\n thinking: z.string(),\n }),\n ]),\n }),\n }),\n }),\n z.object({\n type: z.literal('content-end'),\n index: z.number(),\n }),\n z.object({\n type: z.literal('message-start'),\n id: z.string().nullish(),\n }),\n z.object({\n type: z.literal('message-end'),\n delta: z.object({\n finish_reason: z.string(),\n usage: cohereUsageSchema(),\n }),\n }),\n // https://docs.cohere.com/v2/docs/streaming#tool-use-stream-events-for-tool-calling\n z.object({\n type: z.literal('tool-plan-delta'),\n delta: z.object({\n message: z.object({\n tool_plan: z.string(),\n }),\n }),\n }),\n z.object({\n type: z.literal('tool-call-start'),\n delta: z.object({\n message: z.object({\n tool_calls: z.object({\n id: z.string(),\n type: z.literal('function'),\n function: z.object({\n name: z.string(),\n arguments: z.string(),\n }),\n }),\n }),\n }),\n }),\n // A single tool call's `arguments` stream in chunks and must be accumulated\n // in a string and so the full tool object info can only be parsed once we see\n // `tool-call-end`.\n z.object({\n type: z.literal('tool-call-delta'),\n delta: z.object({\n message: z.object({\n tool_calls: z.object({\n function: z.object({\n arguments: z.string(),\n }),\n }),\n }),\n }),\n }),\n z.object({\n type: z.literal('tool-call-end'),\n }),\n ]),\n ),\n);\n","import {\n lazySchema,\n zodSchema,\n type InferSchema,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\nexport type CohereEmbeddingModelId =\n | 'embed-english-v3.0'\n | 'embed-multilingual-v3.0'\n | 'embed-english-light-v3.0'\n | 'embed-multilingual-light-v3.0'\n | 'embed-english-v2.0'\n | 'embed-english-light-v2.0'\n | 'embed-multilingual-v2.0'\n | (string & {});\n\nexport const cohereEmbeddingModelOptions = lazySchema(() =>\n zodSchema(\n z.object({\n /**\n * The format of the returned embedding. Defaults to `float`.\n * Non-float formats require Embed v3.0 or newer.\n * Binary formats contain packed bytes, not individual vector dimensions.\n */\n embeddingType: z\n .enum(['float', 'int8', 'uint8', 'binary', 'ubinary'])\n .optional(),\n\n /**\n * Specifies the type of input passed to the model. Default is `search_query`.\n *\n * - \"search_document\": Used for embeddings stored in a vector database for search use-cases.\n * - \"search_query\": Used for embeddings of search queries run against a vector DB to find relevant documents.\n * - \"classification\": Used for embeddings passed through a text classifier.\n * - \"clustering\": Used for embeddings run through a clustering algorithm.\n */\n inputType: z\n .enum([\n 'search_document',\n 'search_query',\n 'classification',\n 'clustering',\n ])\n .optional(),\n\n /**\n * Specifies how the API will handle inputs longer than the maximum token length.\n * Default is `END`.\n *\n * - \"NONE\": If selected, when the input exceeds the maximum input token length will return an error.\n * - \"START\": Will discard the start of the input until the remaining input is exactly the maximum input token length for the model.\n * - \"END\": Will discard the end of the input until the remaining input is exactly the maximum input token length for the model.\n */\n truncate: z.enum(['NONE', 'START', 'END']).optional(),\n\n /**\n * The number of dimensions of the output embedding.\n * Only available for `embed-v4.0` and newer models.\n *\n * Embed v4.0 supports `256`, `512`, `1024`, and `1536` (default).\n * Embed v5.0 Pro and Fast also support `768` and `2048` (default).\n */\n outputDimension: z\n .union([\n z.literal(256),\n z.literal(512),\n z.literal(768),\n z.literal(1024),\n z.literal(1536),\n z.literal(2048),\n ])\n .optional(),\n }),\n ),\n);\n\nexport type CohereEmbeddingModelOptions = InferSchema<\n typeof cohereEmbeddingModelOptions\n>;\n","import {\n TooManyEmbeddingValuesForCallError,\n type EmbeddingModelV4,\n} from '@ai-sdk/provider';\nimport {\n combineHeaders,\n createJsonResponseHandler,\n parseProviderOptions,\n postJsonToApi,\n serializeModelOptions,\n WORKFLOW_SERIALIZE,\n WORKFLOW_DESERIALIZE,\n type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\nimport {\n cohereEmbeddingModelOptions,\n type CohereEmbeddingModelId,\n type CohereEmbeddingModelOptions,\n} from './cohere-embedding-model-options';\nimport { cohereFailedResponseHandler } from './cohere-error';\n\ntype CohereEmbeddingConfig = {\n provider: string;\n baseURL: string;\n headers?: () => Record<string, string | undefined>;\n fetch?: FetchFunction;\n};\n\nexport class CohereEmbeddingModel implements EmbeddingModelV4 {\n readonly specificationVersion = 'v4';\n readonly modelId: CohereEmbeddingModelId;\n\n readonly maxEmbeddingsPerCall = 96;\n readonly supportsParallelCalls = true;\n\n private readonly config: CohereEmbeddingConfig;\n\n static [WORKFLOW_SERIALIZE](model: CohereEmbeddingModel) {\n return serializeModelOptions({\n modelId: model.modelId,\n config: model.config,\n });\n }\n\n static [WORKFLOW_DESERIALIZE](options: {\n modelId: CohereEmbeddingModelId;\n config: CohereEmbeddingConfig;\n }) {\n return new CohereEmbeddingModel(options.modelId, options.config);\n }\n\n constructor(modelId: CohereEmbeddingModelId, config: CohereEmbeddingConfig) {\n this.modelId = modelId;\n this.config = config;\n }\n\n get provider(): string {\n return this.config.provider;\n }\n\n async doEmbed({\n values,\n dimensions,\n headers,\n abortSignal,\n providerOptions,\n }: Parameters<EmbeddingModelV4['doEmbed']>[0]): Promise<\n Awaited<ReturnType<EmbeddingModelV4['doEmbed']>>\n > {\n const embeddingOptions = await parseProviderOptions({\n provider: 'cohere',\n providerOptions,\n schema: cohereEmbeddingModelOptions,\n });\n const embeddingType = embeddingOptions?.embeddingType ?? 'float';\n\n if (values.length > this.maxEmbeddingsPerCall) {\n throw new TooManyEmbeddingValuesForCallError({\n provider: this.provider,\n modelId: this.modelId,\n maxEmbeddingsPerCall: this.maxEmbeddingsPerCall,\n values,\n });\n }\n\n const {\n responseHeaders,\n value: response,\n rawValue,\n } = await postJsonToApi({\n url: `${this.config.baseURL}/embed`,\n headers: combineHeaders(this.config.headers?.(), headers),\n body: {\n model: this.modelId,\n embedding_types: [embeddingType],\n texts: values,\n input_type: embeddingOptions?.inputType ?? 'search_query',\n truncate: embeddingOptions?.truncate,\n output_dimension: embeddingOptions?.outputDimension ?? dimensions,\n },\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createJsonResponseHandler(\n cohereTextEmbeddingResponseSchema(embeddingType),\n ),\n abortSignal,\n fetch: this.config.fetch,\n });\n\n return {\n warnings: [],\n embeddings: response.embeddings[embeddingType],\n usage: { tokens: response.meta.billed_units.input_tokens },\n response: { headers: responseHeaders, body: rawValue },\n };\n }\n}\n\n// minimal version of the schema, focussed on what is needed for the implementation\n// this approach limits breakages when the API changes and increases efficiency\nconst cohereTextEmbeddingResponseSchema = (\n embeddingType: NonNullable<CohereEmbeddingModelOptions['embeddingType']>,\n) =>\n z.object({\n embeddings: z.object({\n [embeddingType]: z.array(z.array(z.number())),\n }),\n meta: z.object({\n billed_units: z.object({\n input_tokens: z.number(),\n }),\n }),\n });\n","import { lazySchema, zodSchema } from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\n// https://docs.cohere.com/v2/reference/rerank\nexport type CohereRerankingInput = {\n model: string;\n query: string;\n documents: string[];\n top_n: number | undefined;\n max_tokens_per_doc: number | undefined;\n priority: number | undefined;\n};\n\nexport const cohereRerankingResponseSchema = lazySchema(() =>\n zodSchema(\n z.object({\n id: z.string().nullish(),\n results: z.array(\n z.object({\n index: z.number(),\n relevance_score: z.number(),\n }),\n ),\n meta: z.any(),\n }),\n ),\n);\n","import {\n lazySchema,\n zodSchema,\n type FlexibleSchema,\n} from '@ai-sdk/provider-utils';\nimport { z } from 'zod/v4';\n\n// https://docs.cohere.com/docs/rerank\nexport type CohereRerankingModelId =\n | 'rerank-v3.5'\n | 'rerank-english-v3.0'\n | 'rerank-multilingual-v3.0'\n | (string & {});\n\nexport type CohereRerankingModelOptions = {\n /**\n * Long documents will be automatically truncated to the specified number of tokens.\n *\n * @default 4096\n */\n maxTokensPerDoc?: number;\n\n /**\n * The priority of the request.\n *\n * @default 0\n */\n priority?: number;\n};\n\nexport const cohereRerankingModelOptionsSchema: FlexibleSchema<CohereRerankingModelOptions> =\n lazySchema(() =>\n zodSchema(\n z.object({\n maxTokensPerDoc: z.number().optional(),\n priority: z.number().optional(),\n }),\n ),\n );\n","import type { RerankingModelV4, SharedV4Warning } from '@ai-sdk/provider';\nimport {\n combineHeaders,\n createJsonResponseHandler,\n parseProviderOptions,\n postJsonToApi,\n type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { cohereFailedResponseHandler } from '../cohere-error';\nimport {\n cohereRerankingResponseSchema,\n type CohereRerankingInput,\n} from './cohere-reranking-api';\nimport {\n cohereRerankingModelOptionsSchema,\n type CohereRerankingModelId,\n} from './cohere-reranking-model-options';\ntype CohereRerankingConfig = {\n provider: string;\n baseURL: string;\n headers: () => Record<string, string | undefined>;\n fetch?: FetchFunction;\n};\n\nexport class CohereRerankingModel implements RerankingModelV4 {\n readonly specificationVersion = 'v4';\n readonly modelId: CohereRerankingModelId;\n\n private readonly config: CohereRerankingConfig;\n\n constructor(modelId: CohereRerankingModelId, config: CohereRerankingConfig) {\n this.modelId = modelId;\n this.config = config;\n }\n\n get provider(): string {\n return this.config.provider;\n }\n\n // current implementation is based on v2 of the API: https://docs.cohere.com/v2/reference/rerank\n async doRerank({\n documents,\n headers,\n query,\n topN,\n abortSignal,\n providerOptions,\n }: Parameters<RerankingModelV4['doRerank']>[0]): Promise<\n Awaited<ReturnType<RerankingModelV4['doRerank']>>\n > {\n const rerankingOptions = await parseProviderOptions({\n provider: 'cohere',\n providerOptions,\n schema: cohereRerankingModelOptionsSchema,\n });\n\n const warnings: SharedV4Warning[] = [];\n\n if (documents.type === 'object') {\n warnings.push({\n type: 'compatibility',\n feature: 'object documents',\n details: 'Object documents are converted to strings.',\n });\n }\n\n const {\n responseHeaders,\n value: response,\n rawValue,\n } = await postJsonToApi({\n url: `${this.config.baseURL}/rerank`,\n headers: combineHeaders(this.config.headers(), headers),\n body: {\n model: this.modelId,\n query,\n documents:\n documents.type === 'text'\n ? documents.values\n : documents.values.map(value => JSON.stringify(value)),\n top_n: topN,\n max_tokens_per_doc: rerankingOptions?.maxTokensPerDoc,\n priority: rerankingOptions?.priority,\n } satisfies CohereRerankingInput,\n failedResponseHandler: cohereFailedResponseHandler,\n successfulResponseHandler: createJsonResponseHandler(\n cohereRerankingResponseSchema,\n ),\n abortSignal,\n fetch: this.config.fetch,\n });\n\n return {\n ranking: response.results.map(result => ({\n index: result.index,\n relevanceScore: result.relevance_score,\n })),\n warnings,\n response: {\n id: response.id ?? undefined,\n headers: responseHeaders,\n body: rawValue,\n },\n };\n }\n}\n","// Version string of this package injected at build time.\ndeclare const __PACKAGE_VERSION__: string | undefined;\nexport const VERSION: string =\n typeof __PACKAGE_VERSION__ !== 'undefined'\n ? __PACKAGE_VERSION__\n : '0.0.0-test';\n","import {\n NoSuchModelError,\n type EmbeddingModelV4,\n type LanguageModelV4,\n type RerankingModelV4,\n type ProviderV4,\n} from '@ai-sdk/provider';\nimport {\n generateId,\n loadApiKey,\n withoutTrailingSlash,\n withUserAgentSuffix,\n type FetchFunction,\n} from '@ai-sdk/provider-utils';\nimport { CohereChatLanguageModel } from './cohere-chat-language-model';\nimport type { CohereChatModelId } from './cohere-chat-language-model-options';\nimport { CohereEmbeddingModel } from './cohere-embedding-model';\nimport type { CohereRerankingModelId } from './reranking/cohere-reranking-model-options';\nimport { CohereRerankingModel } from './reranking/cohere-reranking-model';\nimport type { CohereEmbeddingModelId } from './cohere-embedding-model-options';\nimport { VERSION } from './version';\n\nexport interface CohereProvider extends ProviderV4 {\n (modelId: CohereChatModelId): LanguageModelV4;\n\n /**\n * Creates a model for text generation.\n */\n languageModel(modelId: CohereChatModelId): LanguageModelV4;\n\n /**\n * Creates a model for text embeddings.\n */\n embedding(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * Creates a model for text embeddings.\n */\n embeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * @deprecated Use `embedding` instead.\n */\n textEmbedding(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * @deprecated Use `embeddingModel` instead.\n */\n textEmbeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV4;\n\n /**\n * Creates a model for reranking.\n */\n reranking(modelId: CohereRerankingModelId): RerankingModelV4;\n\n /**\n * Creates a model for reranking.\n */\n rerankingModel(modelId: CohereRerankingModelId): RerankingModelV4;\n}\n\nexport interface CohereProviderSettings {\n /**\n * Use a different URL prefix for API calls, e.g. to use proxy servers.\n * The default prefix is `https://api.cohere.com/v2`.\n */\n baseURL?: string;\n\n /**\n * API key that is being send using the `Authorization` header.\n * It defaults to the `COHERE_API_KEY` environment variable.\n */\n apiKey?: string;\n\n /**\n * Custom headers to include in the requests.\n */\n headers?: Record<string, string>;\n\n /**\n * Custom fetch implementation. You can use it as a middleware to intercept requests,\n * or to provide a custom fetch implementation for e.g. testing.\n */\n fetch?: FetchFunction;\n\n /**\n * Optional function to generate a unique ID for each request.\n */\n generateId?: () => string;\n}\n\n/**\n * Create a Cohere AI provider instance.\n */\nexport function createCohere(\n options: CohereProviderSettings = {},\n): CohereProvider {\n const baseURL =\n withoutTrailingSlash(options.baseURL) ?? 'https://api.cohere.com/v2';\n\n const getHeaders = () =>\n withUserAgentSuffix(\n {\n Authorization: `Bearer ${loadApiKey({\n apiKey: options.apiKey,\n environmentVariableName: 'COHERE_API_KEY',\n description: 'Cohere',\n })}`,\n ...options.headers,\n },\n `ai-sdk-cohere/${VERSION}`,\n );\n\n const createChatModel = (modelId: CohereChatModelId) =>\n new CohereChatLanguageModel(modelId, {\n provider: 'cohere.chat',\n baseURL,\n headers: getHeaders,\n fetch: options.fetch,\n generateId: options.generateId ?? generateId,\n });\n\n const createEmbeddingModel = (modelId: CohereEmbeddingModelId) =>\n new CohereEmbeddingModel(modelId, {\n provider: 'cohere.textEmbedding',\n baseURL,\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const createRerankingModel = (modelId: CohereRerankingModelId) =>\n new CohereRerankingModel(modelId, {\n provider: 'cohere.reranking',\n baseURL,\n headers: getHeaders,\n fetch: options.fetch,\n });\n\n const provider = function (modelId: CohereChatModelId) {\n if (new.target) {\n throw new Error(\n 'The Cohere model function cannot be called with the new keyword.',\n );\n }\n\n return createChatModel(modelId);\n };\n\n provider.specificationVersion = 'v4' as const;\n provider.languageModel = createChatModel;\n provider.embedding = createEmbeddingModel;\n provider.embeddingModel = createEmbeddingModel;\n provider.textEmbedding = createEmbeddingModel;\n provider.textEmbeddingModel = createEmbeddingModel;\n provider.reranking = createRerankingModel;\n provider.rerankingModel = createRerankingModel;\n\n provider.imageModel = (modelId: string) => {\n throw new NoSuchModelError({ modelId, modelType: 'imageModel' });\n };\n\n return provider;\n}\n\n/**\n * Default Cohere provider instance.\n */\nexport const cohere = createCohere();\n"],"mappings":";;;;AAwBA,MAAa,iCAAiC,iBAC5C,UACE,EAAE,OAAO;;;;;;;;;AASP,UAAU,EACP,OAAO;CACN,MAAM,EAAE,KAAK,CAAC,WAAW,UAAU,CAAC,CAAC,CAAC,SAAS;CAC/C,aAAa,EAAE,OAAO,CAAC,CAAC,SAAS;AACnC,CAAC,CAAC,CACD,SAAS,EACd,CAAC,CACH,CACF;AAMA,MAAa,iCAAiC,iBAC5C,UACE,EAAE,OAAO;;;;;;AAMP,QAAQ,EAAE,KAAK;CAAC;CAAQ;CAAO;AAAM,CAAC,CAAC,CAAC,SAAS,EACnD,CAAC,CACH,CACF;;;ACpDA,MAAM,wBAAwB,iBAC5B,UACE,EAAE,OAAO,EACP,SAAS,EAAE,OAAO,EACpB,CAAC,CACH,CACF;AAIA,MAAa,8BAA8B,+BAA+B;CACxE,aAAa;CACb,iBAAgB,SAAQ,KAAK;AAC/B,CAAC;;;ACdD,SAAgB,aAAa,EAC3B,OACA,cAiBA;CAEA,QAAQ,OAAO,SAAS,QAAQ,KAAA;CAEhC,MAAM,eAAkC,CAAC;CAEzC,IAAI,SAAS,MACX,OAAO;EAAE,OAAO,KAAA;EAAW,YAAY,KAAA;EAAW;CAAa;CAGjE,MAAM,cAOD,CAAC;CAEN,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,YAChB,aAAa,KAAK;EAChB,MAAM;EACN,SAAS,yBAAyB,KAAK;CACzC,CAAC;MAED,YAAY,KAAK;EACf,MAAM;EACN,UAAU;GACR,MAAM,KAAK;GACX,aAAa,KAAK;GAClB,YAAY,KAAK;EACnB;CACF,CAAC;CAIL,IAAI,cAAc,MAChB,OAAO;EAAE,OAAO;EAAa,YAAY,KAAA;EAAW;CAAa;CAGnE,MAAM,OAAO,WAAW;CAExB,QAAQ,MAAR;EACE,KAAK,QACH,OAAO;GAAE,OAAO;GAAa,YAAY,KAAA;GAAW;EAAa;EAEnE,KAAK,QACH,OAAO;GAAE,OAAO;GAAa,YAAY;GAAQ;EAAa;EAEhE,KAAK,YACH,OAAO;GAAE,OAAO;GAAa,YAAY;GAAY;EAAa;EAEpE,KAAK,QACH,OAAO;GACL,OAAO,YAAY,QACjB,SAAQ,KAAK,SAAS,SAAS,WAAW,QAC5C;GACA,YAAY;GACZ;EACF;EAEF,SAEE,MAAM,IAAI,8BAA8B,EACtC,eAAe,qBAAqBA,OACtC,CAAC;CAEL;AACF;;;ACjFA,SAAgB,mBACd,OACsB;CACtB,IAAI,SAAS,MACX,OAAO,6BAA6B;CAGtC,MAAM,SAAS,MAAM;CACrB,MAAM,cAAc,OAAO;CAC3B,MAAM,eAAe,OAAO;CAC5B,MAAM,kBAAkB,MAAM,iBAAiB,KAAA;CAE/C,OAAO;EACL,aAAa;GACX,OAAO;GACP,SACE,mBAAmB,OAAO,cAAc,kBAAkB;GAC5D,WAAW;GACX,YAAY,KAAA;EACd;EACA,cAAc;GACZ,OAAO;GACP,MAAM;GACN,WAAW,KAAA;EACb;EACA,KAAK;CACP;AACF;;;ACtBA,eAAsB,0BACpB,QAOC;CACD,MAAM,WAA6B,CAAC;CACpC,MAAM,YAA+D,CAAC;CACtE,MAAM,WAA8B,CAAC;CAErC,KAAK,MAAM,EAAE,MAAM,aAAa,QAC9B,QAAQ,MAAR;EACE,KAAK;GACH,SAAS,KAAK;IAAE,MAAM;IAAU;GAAQ,CAAC;GACzC;EAGF,KAAK,QAAQ;GACX,MAAM,mBAAoD,CAAC;GAC3D,IAAI,WAAW;GAEf,KAAK,MAAM,QAAQ,SACjB,QAAQ,KAAK,MAAb;IACE,KAAK;KACH,IAAI,KAAK,KAAK,SAAS,GACrB,iBAAiB,KAAK;MAAE,MAAM;MAAQ,MAAM,KAAK;KAAK,CAAC;KAEzD;IAEF,KAAK,QAAQ;KACX,IAAI,qBAAqB,KAAK,SAAS,MAAM,SAAS;MACpD,WAAW;MACX,MAAM,MAAM,cAAc,EAAE,KAAK,CAAC;MAClC,MAAM,gBACH,MAAM,qBAAqB;OAC1B,UAAU;OACV,iBAAiB,KAAK;OACtB,QAAQ;MACV,CAAC,KAAM,CAAC;MAEV,iBAAiB,KAAK;OACpB,MAAM;OACN,WAAW;QACT;QACA,GAAI,cAAc,SACd,EAAE,QAAQ,cAAc,OAAO,IAC/B,CAAC;OACP;MACF,CAAC;MACD;KACF;KAEA,IAAI;KACJ,QAAQ,KAAK,KAAK,MAAlB;MACE,KAAK,aACH,MAAM,IAAI,8BAA8B,EACtC,eAAe,sCACjB,CAAC;MAEH,KAAK,OACH,MAAM,IAAI,8BAA8B;OACtC,eAAe;OACf,SACE;MACJ,CAAC;MAEH,KAAK;OACH,cAAc,KAAK,KAAK;OACxB;MAEF,KAAK,QACH,cACE,OAAO,KAAK,KAAK,SAAS,WACtB,KAAK,KAAK,OACV,IAAI,YAAY,CAAC,CAAC,OAAO,KAAK,KAAK,IAAI;KAGjD;KAEA,UAAU,KAAK,EACb,MAAM;MACJ,MAAM;MACN,OAAO,KAAK;KACd,EACF,CAAC;KACD;IACF;GACF;GAGF,IAAI,UACF,SAAS,KAAK;IAAE,MAAM;IAAQ,SAAS;GAAiB,CAAC;QAEzD,SAAS,KAAK;IACZ,MAAM;IACN,SAAS,iBACN,KAAI,MAAM,EAAE,SAAS,SAAS,EAAE,OAAO,EAAG,CAAC,CAC3C,KAAK,EAAE;GACZ,CAAC;GAEH;EACF;EAEA,KAAK,aAAa;GAChB,IAAI,OAAO;GACX,MAAM,YAAkD,CAAC;GAEzD,KAAK,MAAM,QAAQ,SACjB,QAAQ,KAAK,MAAb;IACE,KAAK;KACH,QAAQ,KAAK;KACb;IAEF,KAAK,aACH,UAAU,KAAK;KACb,IAAI,KAAK;KACT,MAAM;KACN,UAAU;MACR,MAAM,KAAK;MACX,WAAW,KAAK,UAAU,KAAK,KAAK;KACtC;IACF,CAAC;GAGL;GAGF,SAAS,KAAK;IACZ,MAAM;IACN,SAAS,UAAU,SAAS,IAAI,KAAA,IAAY;IAC5C,YAAY,UAAU,SAAS,IAAI,YAAY,KAAA;IAC/C,WAAW,KAAA;GACb,CAAC;GAED;EACF;EACA,KAAK;GACH,SAAS,KACP,GAAG,QACA,QAAO,eAAc,WAAW,SAAS,wBAAwB,CAAC,CAClE,KAAI,eAAc;IACjB,MAAM,SAAS,WAAW;IAE1B,IAAI;IACJ,QAAQ,OAAO,MAAf;KACE,KAAK;KACL,KAAK;MACH,eAAe,OAAO;MACtB;KACF,KAAK;MACH,eAAe,OAAO,UAAU;MAChC;KACF,KAAK;KACL,KAAK;KACL,KAAK,cACH,eAAe,KAAK,UAAU,OAAO,KAAK;IAE9C;IAEA,OAAO;KACL,MAAM;KACN,SAAS;KACT,cAAc,WAAW;IAC3B;GACF,CAAC,CACL;GAEA;EAEF,SAEE,MAAM,IAAI,MAAM,qBAAqBC,MAAkB;CAE3D;CAGF,OAAO;EAAE;EAAU;EAAW;CAAS;AACzC;AAEA,SAAS,cAAc,EAAE,QAAmD;CAC1E,QAAQ,KAAK,KAAK,MAAlB;EACE,KAAK,OACH,OAAO,KAAK,KAAK,IAAI,SAAS;EAEhC,KAAK,QACH,OAAO,QAAQ,qBAAqB,EAAE,KAAK,CAAC,EAAE,UAAU,gBAAgB,KAAK,KAAK,IAAI;EAExF,KAAK,aACH,MAAM,IAAI,8BAA8B,EACtC,eAAe,4CACjB,CAAC;EAEH,KAAK,QACH,MAAM,IAAI,8BAA8B,EACtC,eAAe,kCACjB,CAAC;CAEL;AACF;;;AC1NA,SAAgB,sBACd,cACwC;CACxC,QAAQ,cAAR;EACE,KAAK;EACL,KAAK,iBACH,OAAO;EAET,KAAK,cACH,OAAO;EAET,KAAK,SACH,OAAO;EAET,KAAK,aACH,OAAO;EAET,SACE,OAAO;CACX;AACF;;;ACyBA,IAAa,0BAAb,MAAa,wBAAmD;CAW9D,QAAQ,oBAAoB,OAAgC;EAC1D,OAAO,sBAAsB;GAC3B,SAAS,MAAM;GACf,QAAQ,MAAM;EAChB,CAAC;CACH;CAEA,QAAQ,sBAAsB,SAG3B;EACD,OAAO,IAAI,wBAAwB,QAAQ,SAAS,QAAQ,MAAM;CACpE;CAEA,YAAY,SAA4B,QAA0B;EAxBlC,KAAA,uBAAA;EAImB,KAAA,gBAAA,EACjD,WAAW,CAAC,iBAAiB,EAC/B;EAmBE,KAAK,UAAU;EACf,KAAK,SAAS;CAChB;CAEA,IAAI,WAAmB;EACrB,OAAO,KAAK,OAAO;CACrB;CAEA,MAAc,QAAQ,EACpB,QACA,iBACA,aACA,MACA,MACA,kBACA,iBACA,eACA,gBACA,MACA,WACA,OACA,YACA,mBAC6B;EAC7B,MAAM,WAA8B,CAAC;EAErC,MAAM,gBACH,MAAM,qBAAqB;GAC1B,UAAU;GACV;GACA,QAAQ;EACV,CAAC,KAAM,CAAC;EAEV,MAAM,EACJ,UAAU,YACV,WAAW,iBACX,UAAU,mBACR,MAAM,0BAA0B,MAAM;EAE1C,MAAM,EACJ,OAAO,aACP,YAAY,kBACZ,iBACE,aAAa;GAAE;GAAO;EAAW,CAAC;EAEtC,SAAS,KAAK,GAAG,cAAc,GAAG,cAAc;EAEhD,OAAO;GACL,MAAM;IAEJ,OAAO,KAAK;IAGZ,mBAAmB;IACnB,kBAAkB;IAClB,YAAY;IACZ;IACA,GAAG;IACH,GAAG;IACH;IACA,gBAAgB;IAGhB,iBACE,gBAAgB,SAAS,SACrB;KAAE,MAAM;KAAe,aAAa,eAAe;IAAO,IAC1D,KAAA;IAGN,UAAU;IAGV,OAAO;IACP,aAAa;IAGb,GAAI,gBAAgB,SAAS,KAAK,EAAE,WAAW,gBAAgB;IAG/D,GAAG,sBAAsB;KACvB;KACA;KACA;IACF,CAAC;GACH;GACA;EACF;CACF;CAEA,MAAM,WACJ,SACwC;EACxC,MAAM,EAAE,MAAM,aAAa,MAAM,KAAK,QAAQ,OAAO;EAErD,MAAM,EACJ,iBACA,OAAO,UACP,UAAU,gBACR,MAAM,cAAc;GACtB,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,UAAU,GAAG,QAAQ,OAAO;GAChE,MAAM;GACN,uBAAuB;GACvB,2BAA2B,0BACzB,wBACF;GACA,aAAa,QAAQ;GACrB,OAAO,KAAK,OAAO;EACrB,CAAC;EAED,MAAM,UAAyC,CAAC;EAEhD,KAAK,MAAM,QAAQ,SAAS,QAAQ,WAAW,CAAC,GAAG;GACjD,IAAI,KAAK,SAAS,UAAU,KAAK,KAAK,SAAS,GAAG;IAChD,QAAQ,KAAK;KAAE,MAAM;KAAQ,MAAM,KAAK;IAAK,CAAC;IAC9C;GACF;GAEA,IAAI,KAAK,SAAS,cAAc,KAAK,SAAS,SAAS,GAAG;IACxD,QAAQ,KAAK;KAAE,MAAM;KAAa,MAAM,KAAK;IAAS,CAAC;IACvD;GACF;EACF;EAGA,KAAK,MAAM,YAAY,SAAS,QAAQ,aAAa,CAAC,GACpD,QAAQ,KAAK;GACX,MAAM;GACN,YAAY;GACZ,IAAI,KAAK,OAAO,WAAW;GAC3B,WAAW;GACX,OAAO,SAAS,QAAQ,EAAE,EAAE,UAAU,SAAS;GAC/C,kBAAkB,EAChB,QAAQ;IACN,OAAO,SAAS;IAChB,KAAK,SAAS;IACd,MAAM,SAAS;IACf,SAAS,SAAS;IAClB,GAAI,SAAS,QAAQ,EAAE,cAAc,SAAS,KAAK;GACrD,EACF;EACF,CAAC;EAIH,KAAK,MAAM,YAAY,SAAS,QAAQ,cAAc,CAAC,GACrD,QAAQ,KAAK;GACX,MAAM;GACN,YAAY,SAAS;GACrB,UAAU,SAAS,SAAS;GAG5B,OAAO,SAAS,SAAS,UAAU,QAAQ,UAAU,IAAI;EAC3D,CAAC;EAGH,OAAO;GACL;GACA,cAAc;IACZ,SAAS,sBAAsB,SAAS,aAAa;IACrD,KAAK,SAAS,iBAAiB,KAAA;GACjC;GACA,OAAO,mBAAmB,SAAS,KAAK;GACxC,SAAS,EAAE,MAAM,KAAK;GACtB,UAAU;IAER,IAAI,SAAS,iBAAiB,KAAA;IAC9B,SAAS;IACT,MAAM;GACR;GACA;EACF;CACF;CAEA,MAAM,SACJ,SACsC;EACtC,MAAM,EAAE,MAAM,aAAa,MAAM,KAAK,QAAQ,OAAO;EAErD,MAAM,EAAE,iBAAiB,OAAO,aAAa,MAAM,cAAc;GAC/D,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,UAAU,GAAG,QAAQ,OAAO;GAChE,MAAM;IAAE,GAAG;IAAM,QAAQ;GAAK;GAC9B,uBAAuB;GACvB,2BAA2B,iCACzB,qBACF;GACA,aAAa,QAAQ;GACrB,OAAO,KAAK,OAAO;EACrB,CAAC;EAGD,IAAI,eAA4C;GAC9C,SAAS;GACT,KAAK,KAAA;EACP;EACA,IAAI,QAAiC,KAAA;EAErC,IAAI,kBAKO;EAEX,IAAI,oBAAoB;EAExB,OAAO;GACL,QAAQ,SAAS,YACf,IAAI,gBAGF;IACA,MAAM,YAAY;KAChB,WAAW,QAAQ;MAAE,MAAM;MAAgB;KAAS,CAAC;IACvD;IAEA,MAAM,UAAU,OAAO,YAAY;KACjC,IAAI,QAAQ,kBACV,WAAW,QAAQ;MAAE,MAAM;MAAO,UAAU,MAAM;KAAS,CAAC;KAI9D,IAAI,CAAC,MAAM,SAAS;MAClB,eAAe;OAAE,SAAS;OAAS,KAAK,KAAA;MAAU;MAClD,WAAW,QAAQ;OAAE,MAAM;OAAS,OAAO,MAAM;MAAM,CAAC;MACxD;KACF;KAEA,MAAM,QAAQ,MAAM;KAGpB,QAFa,MAAM,MAEnB;MACE,KAAK;OACH,IAAI,MAAM,MAAM,QAAQ,QAAQ,SAAS,YAAY;QACnD,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,OAAO,MAAM,KAAK;QACxB,CAAC;QACD,oBAAoB;QACpB;OACF;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,OAAO,MAAM,KAAK;OACxB,CAAC;OACD;MAGF,KAAK;OACH,IAAI,cAAc,MAAM,MAAM,QAAQ,SAAS;QAC7C,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,OAAO,MAAM,KAAK;SACtB,OAAO,MAAM,MAAM,QAAQ,QAAQ;QACrC,CAAC;QACD;OACF;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,OAAO,MAAM,KAAK;QACtB,OAAO,MAAM,MAAM,QAAQ,QAAQ;OACrC,CAAC;OACD;MAGF,KAAK;OACH,IAAI,mBAAmB;QACrB,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,OAAO,MAAM,KAAK;QACxB,CAAC;QACD,oBAAoB;QACpB;OACF;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,OAAO,MAAM,KAAK;OACxB,CAAC;OAED;MAGF,KAAK,mBAAmB;OACtB,MAAM,SAAS,MAAM,MAAM,QAAQ,WAAW;OAC9C,MAAM,WAAW,MAAM,MAAM,QAAQ,WAAW,SAAS;OACzD,MAAM,cACJ,MAAM,MAAM,QAAQ,WAAW,SAAS;OAE1C,kBAAkB;QAChB,IAAI;QACJ,MAAM;QACN,WAAW;QACX,aAAa;OACf;OAEA,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI;QACJ;OACF,CAAC;OAED,IAAI,YAAY,SAAS,GACvB,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI;QACJ,OAAO;OACT,CAAC;OAEH;MACF;MAEA,KAAK;OACH,IAAI,mBAAmB,CAAC,gBAAgB,aAAa;QACnD,MAAM,YACJ,MAAM,MAAM,QAAQ,WAAW,SAAS;QAC1C,gBAAgB,aAAa;QAE7B,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,gBAAgB;SACpB,OAAO;QACT,CAAC;OACH;OACA;MAGF,KAAK;OACH,IAAI,mBAAmB,CAAC,gBAAgB,aAAa;QACnD,WAAW,QAAQ;SACjB,MAAM;SACN,IAAI,gBAAgB;QACtB,CAAC;QAED,WAAW,QAAQ;SACjB,MAAM;SACN,YAAY,gBAAgB;SAC5B,UAAU,gBAAgB;SAC1B,OAAO,KAAK,UACV,MAAM,UAAU,EACd,MAAM,gBAAgB,WAAW,KAAK,KAAK,KAC7C,CAAC,CACH;QACF,CAAC;QAED,gBAAgB,cAAc;QAC9B,kBAAkB;OACpB;OACA;MAGF,KAAK;OACH,WAAW,QAAQ;QACjB,MAAM;QACN,IAAI,MAAM,MAAM,KAAA;OAClB,CAAC;OACD;MAGF,KAAK;OACH,eAAe;QACb,SAAS,sBAAsB,MAAM,MAAM,aAAa;QACxD,KAAK,MAAM,MAAM;OACnB;OACA,QAAQ,MAAM,MAAM;OACpB;KAKJ;IACF;IAEA,MAAM,YAAY;KAChB,WAAW,QAAQ;MACjB,MAAM;MACN;MACA,OAAO,mBAAmB,KAAK;KACjC,CAAC;IACH;GACF,CAAC,CACH;GACA,SAAS,EAAE,MAAM;IAAE,GAAG;IAAM,QAAQ;GAAK,EAAE;GAC3C,UAAU,EAAE,SAAS,gBAAgB;EACvC;CACF;AACF;AAEA,SAAS,sBAAsB,EAC7B,WACA,eACA,YAKyD;CACzD,IAAI,cAAc,UAChB,OAAO,EACL,UAAU;EACR,MAAM,cAAc,SAAS,QAAQ;EACrC,cAAc,cAAc,SAAS;CACvC,EACF;CAGF,IAAI,CAAC,kBAAkB,SAAS,GAC9B,OAAO,CAAC;CAGV,IAAI,cAAc,QAChB,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,EAAE;CAG1C,MAAM,cAAc,6BAA6B;EAC/C;EACA,iBAAiB;EACjB,oBAAoB;EACpB;CACF,CAAC;CAED,IAAI,eAAe,MACjB,OAAO,CAAC;CAGV,OAAO,EAAE,UAAU;EAAE,MAAM;EAAW,cAAc;CAAY,EAAE;AACpE;AAGA,MAAM,0BACJ,EAAE,YAAY;CACZ,cAAc,EACX,YAAY;EACX,cAAc,EAAE,OAAO;EACvB,eAAe,EAAE,OAAO;CAC1B,CAAC,CAAC,CACD,QAAQ;CACX,QAAQ,EAAE,YAAY;EACpB,cAAc,EAAE,OAAO;EACvB,eAAe,EAAE,OAAO;CAC1B,CAAC;CACD,eAAe,EAAE,OAAO,CAAC,CAAC,QAAQ;AACpC,CAAC;AAEH,MAAM,2BAA2B,iBAC/B,UACE,EAAE,OAAO;CACP,eAAe,EAAE,OAAO,CAAC,CAAC,QAAQ;CAClC,SAAS,EAAE,OAAO;EAChB,MAAM,EAAE,OAAO;EACf,SAAS,EACN,MACC,EAAE,MAAM,CACN,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,MAAM;GACtB,MAAM,EAAE,OAAO;EACjB,CAAC,GACD,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;EACrB,CAAC,CACH,CAAC,CACH,CAAC,CACA,QAAQ;EACX,WAAW,EAAE,OAAO,CAAC,CAAC,QAAQ;EAC9B,YAAY,EACT,MACC,EAAE,OAAO;GACP,IAAI,EAAE,OAAO;GACb,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;IACjB,MAAM,EAAE,OAAO;IACf,WAAW,EAAE,OAAO;GACtB,CAAC;EACH,CAAC,CACH,CAAC,CACA,QAAQ;EACX,WAAW,EACR,MACC,EAAE,OAAO;GACP,OAAO,EAAE,OAAO;GAChB,KAAK,EAAE,OAAO;GACd,MAAM,EAAE,OAAO;GACf,SAAS,EAAE,MACT,EAAE,OAAO;IACP,MAAM,EAAE,OAAO,CAAC,CAAC,SAAS;IAC1B,IAAI,EAAE,OAAO,CAAC,CAAC,SAAS;IACxB,UAAU,EAAE,OAAO;KACjB,IAAI,EAAE,OAAO,CAAC,CAAC,SAAS;KACxB,MAAM,EAAE,OAAO;KACf,OAAO,EAAE,OAAO;IAClB,CAAC;GACH,CAAC,CACH;GACA,MAAM,EAAE,OAAO,CAAC,CAAC,SAAS;EAC5B,CAAC,CACH,CAAC,CACA,QAAQ;CACb,CAAC;CACD,eAAe,EAAE,OAAO;CACxB,OAAO,kBAAkB;AAC3B,CAAC,CACH,CACF;AAIA,MAAM,wBAAwB,iBAC5B,UACE,EAAE,mBAAmB,QAAQ;CAC3B,EAAE,OAAO,EACP,MAAM,EAAE,QAAQ,gBAAgB,EAClC,CAAC;CACD,EAAE,OAAO,EACP,MAAM,EAAE,QAAQ,cAAc,EAChC,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,eAAe;EAC/B,OAAO,EAAE,OAAO;EAChB,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,CACf,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,MAAM;GACtB,MAAM,EAAE,OAAO;EACjB,CAAC,GACD,EAAE,OAAO;GACP,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;EACrB,CAAC,CACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,eAAe;EAC/B,OAAO,EAAE,OAAO;EAChB,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,SAAS,EAAE,MAAM,CACf,EAAE,OAAO,EACP,MAAM,EAAE,OAAO,EACjB,CAAC,GACD,EAAE,OAAO,EACP,UAAU,EAAE,OAAO,EACrB,CAAC,CACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,aAAa;EAC7B,OAAO,EAAE,OAAO;CAClB,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,eAAe;EAC/B,IAAI,EAAE,OAAO,CAAC,CAAC,QAAQ;CACzB,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,aAAa;EAC7B,OAAO,EAAE,OAAO;GACd,eAAe,EAAE,OAAO;GACxB,OAAO,kBAAkB;EAC3B,CAAC;CACH,CAAC;CAED,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,iBAAiB;EACjC,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,WAAW,EAAE,OAAO,EACtB,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,iBAAiB;EACjC,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,YAAY,EAAE,OAAO;GACnB,IAAI,EAAE,OAAO;GACb,MAAM,EAAE,QAAQ,UAAU;GAC1B,UAAU,EAAE,OAAO;IACjB,MAAM,EAAE,OAAO;IACf,WAAW,EAAE,OAAO;GACtB,CAAC;EACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CAID,EAAE,OAAO;EACP,MAAM,EAAE,QAAQ,iBAAiB;EACjC,OAAO,EAAE,OAAO,EACd,SAAS,EAAE,OAAO,EAChB,YAAY,EAAE,OAAO,EACnB,UAAU,EAAE,OAAO,EACjB,WAAW,EAAE,OAAO,EACtB,CAAC,EACH,CAAC,EACH,CAAC,EACH,CAAC;CACH,CAAC;CACD,EAAE,OAAO,EACP,MAAM,EAAE,QAAQ,eAAe,EACjC,CAAC;AACH,CAAC,CACH,CACF;;;AC5pBA,MAAa,8BAA8B,iBACzC,UACE,EAAE,OAAO;;;;;;CAMP,eAAe,EACZ,KAAK;EAAC;EAAS;EAAQ;EAAS;EAAU;CAAS,CAAC,CAAC,CACrD,SAAS;;;;;;;;;CAUZ,WAAW,EACR,KAAK;EACJ;EACA;EACA;EACA;CACF,CAAC,CAAC,CACD,SAAS;;;;;;;;;CAUZ,UAAU,EAAE,KAAK;EAAC;EAAQ;EAAS;CAAK,CAAC,CAAC,CAAC,SAAS;;;;;;;;CASpD,iBAAiB,EACd,MAAM;EACL,EAAE,QAAQ,GAAG;EACb,EAAE,QAAQ,GAAG;EACb,EAAE,QAAQ,GAAG;EACb,EAAE,QAAQ,IAAI;EACd,EAAE,QAAQ,IAAI;EACd,EAAE,QAAQ,IAAI;CAChB,CAAC,CAAC,CACD,SAAS;AACd,CAAC,CACH,CACF;;;AC9CA,IAAa,uBAAb,MAAa,qBAAiD;CAS5D,QAAQ,oBAAoB,OAA6B;EACvD,OAAO,sBAAsB;GAC3B,SAAS,MAAM;GACf,QAAQ,MAAM;EAChB,CAAC;CACH;CAEA,QAAQ,sBAAsB,SAG3B;EACD,OAAO,IAAI,qBAAqB,QAAQ,SAAS,QAAQ,MAAM;CACjE;CAEA,YAAY,SAAiC,QAA+B;EAtB5C,KAAA,uBAAA;EAGA,KAAA,uBAAA;EACC,KAAA,wBAAA;EAmB/B,KAAK,UAAU;EACf,KAAK,SAAS;CAChB;CAEA,IAAI,WAAmB;EACrB,OAAO,KAAK,OAAO;CACrB;CAEA,MAAM,QAAQ,EACZ,QACA,YACA,SACA,aACA,mBAGA;EACA,MAAM,mBAAmB,MAAM,qBAAqB;GAClD,UAAU;GACV;GACA,QAAQ;EACV,CAAC;EACD,MAAM,gBAAgB,kBAAkB,iBAAiB;EAEzD,IAAI,OAAO,SAAS,KAAK,sBACvB,MAAM,IAAI,mCAAmC;GAC3C,UAAU,KAAK;GACf,SAAS,KAAK;GACd,sBAAsB,KAAK;GAC3B;EACF,CAAC;EAGH,MAAM,EACJ,iBACA,OAAO,UACP,aACE,MAAM,cAAc;GACtB,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,UAAU,GAAG,OAAO;GACxD,MAAM;IACJ,OAAO,KAAK;IACZ,iBAAiB,CAAC,aAAa;IAC/B,OAAO;IACP,YAAY,kBAAkB,aAAa;IAC3C,UAAU,kBAAkB;IAC5B,kBAAkB,kBAAkB,mBAAmB;GACzD;GACA,uBAAuB;GACvB,2BAA2B,0BACzB,kCAAkC,aAAa,CACjD;GACA;GACA,OAAO,KAAK,OAAO;EACrB,CAAC;EAED,OAAO;GACL,UAAU,CAAC;GACX,YAAY,SAAS,WAAW;GAChC,OAAO,EAAE,QAAQ,SAAS,KAAK,aAAa,aAAa;GACzD,UAAU;IAAE,SAAS;IAAiB,MAAM;GAAS;EACvD;CACF;AACF;AAIA,MAAM,qCACJ,kBAEA,EAAE,OAAO;CACP,YAAY,EAAE,OAAO,GAClB,gBAAgB,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,EAC9C,CAAC;CACD,MAAM,EAAE,OAAO,EACb,cAAc,EAAE,OAAO,EACrB,cAAc,EAAE,OAAO,EACzB,CAAC,EACH,CAAC;AACH,CAAC;;;ACvHH,MAAa,gCAAgC,iBAC3C,UACE,EAAE,OAAO;CACP,IAAI,EAAE,OAAO,CAAC,CAAC,QAAQ;CACvB,SAAS,EAAE,MACT,EAAE,OAAO;EACP,OAAO,EAAE,OAAO;EAChB,iBAAiB,EAAE,OAAO;CAC5B,CAAC,CACH;CACA,MAAM,EAAE,IAAI;AACd,CAAC,CACH,CACF;;;ACIA,MAAa,oCACX,iBACE,UACE,EAAE,OAAO;CACP,iBAAiB,EAAE,OAAO,CAAC,CAAC,SAAS;CACrC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS;AAChC,CAAC,CACH,CACF;;;ACdF,IAAa,uBAAb,MAA8D;CAM5D,YAAY,SAAiC,QAA+B;EAL5C,KAAA,uBAAA;EAM9B,KAAK,UAAU;EACf,KAAK,SAAS;CAChB;CAEA,IAAI,WAAmB;EACrB,OAAO,KAAK,OAAO;CACrB;CAGA,MAAM,SAAS,EACb,WACA,SACA,OACA,MACA,aACA,mBAGA;EACA,MAAM,mBAAmB,MAAM,qBAAqB;GAClD,UAAU;GACV;GACA,QAAQ;EACV,CAAC;EAED,MAAM,WAA8B,CAAC;EAErC,IAAI,UAAU,SAAS,UACrB,SAAS,KAAK;GACZ,MAAM;GACN,SAAS;GACT,SAAS;EACX,CAAC;EAGH,MAAM,EACJ,iBACA,OAAO,UACP,aACE,MAAM,cAAc;GACtB,KAAK,GAAG,KAAK,OAAO,QAAQ;GAC5B,SAAS,eAAe,KAAK,OAAO,QAAQ,GAAG,OAAO;GACtD,MAAM;IACJ,OAAO,KAAK;IACZ;IACA,WACE,UAAU,SAAS,SACf,UAAU,SACV,UAAU,OAAO,KAAI,UAAS,KAAK,UAAU,KAAK,CAAC;IACzD,OAAO;IACP,oBAAoB,kBAAkB;IACtC,UAAU,kBAAkB;GAC9B;GACA,uBAAuB;GACvB,2BAA2B,0BACzB,6BACF;GACA;GACA,OAAO,KAAK,OAAO;EACrB,CAAC;EAED,OAAO;GACL,SAAS,SAAS,QAAQ,KAAI,YAAW;IACvC,OAAO,OAAO;IACd,gBAAgB,OAAO;GACzB,EAAE;GACF;GACA,UAAU;IACR,IAAI,SAAS,MAAM,KAAA;IACnB,SAAS;IACT,MAAM;GACR;EACF;CACF;AACF;;;ACvGA,MAAa,UAAA;;;;;;AC4Fb,SAAgB,aACd,UAAkC,CAAC,GACnB;CAChB,MAAM,UACJ,qBAAqB,QAAQ,OAAO,KAAK;CAE3C,MAAM,mBACJ,oBACE;EACE,eAAe,UAAU,WAAW;GAClC,QAAQ,QAAQ;GAChB,yBAAyB;GACzB,aAAa;EACf,CAAC;EACD,GAAG,QAAQ;CACb,GACA,iBAAiB,SACnB;CAEF,MAAM,mBAAmB,YACvB,IAAI,wBAAwB,SAAS;EACnC,UAAU;EACV;EACA,SAAS;EACT,OAAO,QAAQ;EACf,YAAY,QAAQ,cAAc;CACpC,CAAC;CAEH,MAAM,wBAAwB,YAC5B,IAAI,qBAAqB,SAAS;EAChC,UAAU;EACV;EACA,SAAS;EACT,OAAO,QAAQ;CACjB,CAAC;CAEH,MAAM,wBAAwB,YAC5B,IAAI,qBAAqB,SAAS;EAChC,UAAU;EACV;EACA,SAAS;EACT,OAAO,QAAQ;CACjB,CAAC;CAEH,MAAM,WAAW,SAAU,SAA4B;EACrD,IAAI,YACF,MAAM,IAAI,MACR,kEACF;EAGF,OAAO,gBAAgB,OAAO;CAChC;CAEA,SAAS,uBAAuB;CAChC,SAAS,gBAAgB;CACzB,SAAS,YAAY;CACrB,SAAS,iBAAiB;CAC1B,SAAS,gBAAgB;CACzB,SAAS,qBAAqB;CAC9B,SAAS,YAAY;CACrB,SAAS,iBAAiB;CAE1B,SAAS,cAAc,YAAoB;EACzC,MAAM,IAAI,iBAAiB;GAAE;GAAS,WAAW;EAAa,CAAC;CACjE;CAEA,OAAO;AACT;;;;AAKA,MAAa,SAAS,aAAa"}
@@ -1,17 +1,61 @@
1
1
  ---
2
2
  title: Cohere
3
- description: Learn how to use the Cohere provider for the AI SDK.
3
+ description: Learn how to use Cohere with AI Gateway and the direct provider.
4
4
  ---
5
5
 
6
6
  # Cohere Provider
7
7
 
8
8
  The [Cohere](https://cohere.com/) provider contains language and embedding model support for the Cohere chat API.
9
+ You can use Cohere models through [AI Gateway](/providers/ai-sdk-providers/ai-gateway) (recommended) or connect directly with `@ai-sdk/cohere`.
9
10
 
10
- ## Setup
11
+ ```ts provider="cohere" model="cohere/command-a"
12
+ import { generateText } from 'ai';
13
+ __PROVIDER_IMPORT__;
14
+
15
+ const { text } = await generateText({
16
+ model: __MODEL__,
17
+ prompt: 'Write a vegetarian lasagna recipe for 4 people.',
18
+ });
19
+
20
+ console.log(text);
21
+ ```
22
+
23
+ ## Using AI Gateway
24
+
25
+ The AI SDK uses AI Gateway by default for model strings. Gateway provides access to models across providers through a single API key, with usage and cost tracking. Install `ai`:
11
26
 
12
- The Cohere provider is available in the `@ai-sdk/cohere` module. You can install it with
27
+ <InstallPackages packages="ai" />
28
+
29
+ Set an [AI Gateway API key](https://vercel.com/docs/ai-gateway/getting-started) in your environment:
30
+
31
+ ```bash
32
+ export AI_GATEWAY_API_KEY=your_api_key_here
33
+ ```
13
34
 
14
- <InstallPackages packages="@ai-sdk/cohere" />
35
+ Vercel deployments can also use [OIDC authentication](/providers/ai-sdk-providers/ai-gateway#oidc-authentication-vercel-deployments).
36
+ See the [AI Gateway provider documentation](/providers/ai-sdk-providers/ai-gateway) for configuration and the [model catalog](https://vercel.com/ai-gateway/models) for available models.
37
+
38
+ Gateway feature and provider-option support depends on the model and the provider serving the request. See [Gateway provider options](https://vercel.com/docs/ai-gateway/models-and-providers/provider-options).
39
+
40
+ ## Using `@ai-sdk/cohere`
41
+
42
+ For direct access to Cohere, install `ai` and `@ai-sdk/cohere`:
43
+
44
+ <InstallPackages packages="ai @ai-sdk/cohere" />
45
+
46
+ Set your Cohere API credential in the `COHERE_API_KEY` environment variable:
47
+
48
+ ```bash
49
+ export COHERE_API_KEY=your_api_key_here
50
+ ```
51
+
52
+ AI Gateway does not expose every API in `@ai-sdk/cohere`. Use the direct provider for provider-specific APIs or custom connection settings.
53
+
54
+ <Note>
55
+ The following sections document the direct `@ai-sdk/cohere` provider,
56
+ including its model factories and settings. Gateway model and feature
57
+ availability can differ.
58
+ </Note>
15
59
 
16
60
  ## Provider Instance
17
61
 
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@ai-sdk/cohere",
3
- "version": "4.0.65",
3
+ "version": "4.0.67",
4
4
  "type": "module",
5
5
  "license": "Apache-2.0",
6
6
  "sideEffects": false,
@@ -29,8 +29,8 @@
29
29
  }
30
30
  },
31
31
  "dependencies": {
32
- "@ai-sdk/provider": "4.0.31",
33
- "@ai-sdk/provider-utils": "5.0.64"
32
+ "@ai-sdk/provider": "4.0.32",
33
+ "@ai-sdk/provider-utils": "5.0.66"
34
34
  },
35
35
  "devDependencies": {
36
36
  "@ai-sdk/test-server": "2.0.4",
@@ -1,3 +1,8 @@
1
+ import {
2
+ lazySchema,
3
+ zodSchema,
4
+ type InferSchema,
5
+ } from '@ai-sdk/provider-utils';
1
6
  import { z } from 'zod/v4';
2
7
 
3
8
  // https://docs.cohere.com/docs/models
@@ -17,36 +22,44 @@ export type CohereChatModelId =
17
22
  | 'command-light-nightly'
18
23
  | (string & {});
19
24
 
20
- export const cohereLanguageModelChatOptions = z.object({
21
- /**
22
- * Configuration for reasoning features (optional)
23
- *
24
- * Can be set to an object with the two properties `type` and `tokenBudget`. `type` can be set to `'enabled'` or `'disabled'` (defaults to `'enabled'`).
25
- * `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
26
- *
27
- * @see https://docs.cohere.com/reference/chat#request.body.thinking
28
- */
29
- thinking: z
30
- .object({
31
- type: z.enum(['enabled', 'disabled']).optional(),
32
- tokenBudget: z.number().optional(),
33
- })
34
- .optional(),
35
- });
25
+ export const cohereLanguageModelChatOptions = lazySchema(() =>
26
+ zodSchema(
27
+ z.object({
28
+ /**
29
+ * Configuration for reasoning features (optional)
30
+ *
31
+ * Can be set to an object with the two properties `type` and `tokenBudget`. `type` can be set to `'enabled'` or `'disabled'` (defaults to `'enabled'`).
32
+ * `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
33
+ *
34
+ * @see https://docs.cohere.com/reference/chat#request.body.thinking
35
+ */
36
+ thinking: z
37
+ .object({
38
+ type: z.enum(['enabled', 'disabled']).optional(),
39
+ tokenBudget: z.number().optional(),
40
+ })
41
+ .optional(),
42
+ }),
43
+ ),
44
+ );
36
45
 
37
- export type CohereLanguageModelChatOptions = z.infer<
46
+ export type CohereLanguageModelChatOptions = InferSchema<
38
47
  typeof cohereLanguageModelChatOptions
39
48
  >;
40
49
 
41
- export const cohereImagePartProviderOptions = z.object({
42
- /**
43
- * Image fidelity level passed through as `image_url.detail` on the Cohere chat API.
44
- *
45
- * @see https://docs.cohere.com/docs/image-inputs
46
- */
47
- detail: z.enum(['auto', 'low', 'high']).optional(),
48
- });
50
+ export const cohereImagePartProviderOptions = lazySchema(() =>
51
+ zodSchema(
52
+ z.object({
53
+ /**
54
+ * Image fidelity level passed through as `image_url.detail` on the Cohere chat API.
55
+ *
56
+ * @see https://docs.cohere.com/docs/image-inputs
57
+ */
58
+ detail: z.enum(['auto', 'low', 'high']).optional(),
59
+ }),
60
+ ),
61
+ );
49
62
 
50
- export type CohereImagePartProviderOptions = z.infer<
63
+ export type CohereImagePartProviderOptions = InferSchema<
51
64
  typeof cohereImagePartProviderOptions
52
65
  >;
@@ -9,6 +9,8 @@ import type {
9
9
  SharedV4Warning,
10
10
  } from '@ai-sdk/provider';
11
11
  import {
12
+ lazySchema,
13
+ zodSchema,
12
14
  combineHeaders,
13
15
  createEventSourceResponseHandler,
14
16
  createJsonResponseHandler,
@@ -260,8 +262,9 @@ export class CohereChatLanguageModel implements LanguageModelV4 {
260
262
  fetch: this.config.fetch,
261
263
  });
262
264
 
265
+ // A clean end without the provider's completion signal is interrupted.
263
266
  let finishReason: LanguageModelV4FinishReason = {
264
- unified: 'other',
267
+ unified: 'interrupted',
265
268
  raw: undefined,
266
269
  };
267
270
  let usage: CohereUsage | undefined = undefined;
@@ -278,7 +281,7 @@ export class CohereChatLanguageModel implements LanguageModelV4 {
278
281
  return {
279
282
  stream: response.pipeThrough(
280
283
  new TransformStream<
281
- ParseResult<z.infer<typeof cohereChatChunkSchema>>,
284
+ ParseResult<InferSchema<typeof cohereChatChunkSchema>>,
282
285
  LanguageModelV4StreamPart
283
286
  >({
284
287
  start(controller) {
@@ -500,175 +503,184 @@ function resolveCohereThinking({
500
503
  }
501
504
 
502
505
  // Loose, nested objects included: the parsed value is returned as `usage.raw`.
503
- const cohereUsageSchema = z.looseObject({
504
- billed_units: z
505
- .looseObject({
506
+ const cohereUsageSchema = () =>
507
+ z.looseObject({
508
+ billed_units: z
509
+ .looseObject({
510
+ input_tokens: z.number(),
511
+ output_tokens: z.number(),
512
+ })
513
+ .nullish(),
514
+ tokens: z.looseObject({
506
515
  input_tokens: z.number(),
507
516
  output_tokens: z.number(),
508
- })
509
- .nullish(),
510
- tokens: z.looseObject({
511
- input_tokens: z.number(),
512
- output_tokens: z.number(),
513
- }),
514
- cached_tokens: z.number().nullish(),
515
- });
516
-
517
- const cohereChatResponseSchema = z.object({
518
- generation_id: z.string().nullish(),
519
- message: z.object({
520
- role: z.string(),
521
- content: z
522
- .array(
523
- z.union([
524
- z.object({
525
- type: z.literal('text'),
526
- text: z.string(),
527
- }),
528
- z.object({
529
- type: z.literal('thinking'),
530
- thinking: z.string(),
531
- }),
532
- ]),
533
- )
534
- .nullish(),
535
- tool_plan: z.string().nullish(),
536
- tool_calls: z
537
- .array(
538
- z.object({
539
- id: z.string(),
540
- type: z.literal('function'),
541
- function: z.object({
542
- name: z.string(),
543
- arguments: z.string(),
544
- }),
545
- }),
546
- )
547
- .nullish(),
548
- citations: z
549
- .array(
550
- z.object({
551
- start: z.number(),
552
- end: z.number(),
553
- text: z.string(),
554
- sources: z.array(
555
- z.object({
556
- type: z.string().optional(),
557
- id: z.string().optional(),
558
- document: z.object({
559
- id: z.string().optional(),
517
+ }),
518
+ cached_tokens: z.number().nullish(),
519
+ });
520
+
521
+ const cohereChatResponseSchema = lazySchema(() =>
522
+ zodSchema(
523
+ z.object({
524
+ generation_id: z.string().nullish(),
525
+ message: z.object({
526
+ role: z.string(),
527
+ content: z
528
+ .array(
529
+ z.union([
530
+ z.object({
531
+ type: z.literal('text'),
560
532
  text: z.string(),
561
- title: z.string(),
533
+ }),
534
+ z.object({
535
+ type: z.literal('thinking'),
536
+ thinking: z.string(),
537
+ }),
538
+ ]),
539
+ )
540
+ .nullish(),
541
+ tool_plan: z.string().nullish(),
542
+ tool_calls: z
543
+ .array(
544
+ z.object({
545
+ id: z.string(),
546
+ type: z.literal('function'),
547
+ function: z.object({
548
+ name: z.string(),
549
+ arguments: z.string(),
562
550
  }),
563
551
  }),
564
- ),
565
- type: z.string().optional(),
566
- }),
567
- )
568
- .nullish(),
569
- }),
570
- finish_reason: z.string(),
571
- usage: cohereUsageSchema,
572
- });
552
+ )
553
+ .nullish(),
554
+ citations: z
555
+ .array(
556
+ z.object({
557
+ start: z.number(),
558
+ end: z.number(),
559
+ text: z.string(),
560
+ sources: z.array(
561
+ z.object({
562
+ type: z.string().optional(),
563
+ id: z.string().optional(),
564
+ document: z.object({
565
+ id: z.string().optional(),
566
+ text: z.string(),
567
+ title: z.string(),
568
+ }),
569
+ }),
570
+ ),
571
+ type: z.string().optional(),
572
+ }),
573
+ )
574
+ .nullish(),
575
+ }),
576
+ finish_reason: z.string(),
577
+ usage: cohereUsageSchema(),
578
+ }),
579
+ ),
580
+ );
573
581
 
574
582
  // limited version of the schema, focussed on what is needed for the implementation
575
583
  // this approach limits breakages when the API changes and increases efficiency
576
- const cohereChatChunkSchema = z.discriminatedUnion('type', [
577
- z.object({
578
- type: z.literal('citation-start'),
579
- }),
580
- z.object({
581
- type: z.literal('citation-end'),
582
- }),
583
- z.object({
584
- type: z.literal('content-start'),
585
- index: z.number(),
586
- delta: z.object({
587
- message: z.object({
588
- content: z.union([
589
- z.object({
590
- type: z.literal('text'),
591
- text: z.string(),
592
- }),
593
- z.object({
594
- type: z.literal('thinking'),
595
- thinking: z.string(),
596
- }),
597
- ]),
584
+ const cohereChatChunkSchema = lazySchema(() =>
585
+ zodSchema(
586
+ z.discriminatedUnion('type', [
587
+ z.object({
588
+ type: z.literal('citation-start'),
598
589
  }),
599
- }),
600
- }),
601
- z.object({
602
- type: z.literal('content-delta'),
603
- index: z.number(),
604
- delta: z.object({
605
- message: z.object({
606
- content: z.union([
607
- z.object({
608
- text: z.string(),
590
+ z.object({
591
+ type: z.literal('citation-end'),
592
+ }),
593
+ z.object({
594
+ type: z.literal('content-start'),
595
+ index: z.number(),
596
+ delta: z.object({
597
+ message: z.object({
598
+ content: z.union([
599
+ z.object({
600
+ type: z.literal('text'),
601
+ text: z.string(),
602
+ }),
603
+ z.object({
604
+ type: z.literal('thinking'),
605
+ thinking: z.string(),
606
+ }),
607
+ ]),
609
608
  }),
610
- z.object({
611
- thinking: z.string(),
609
+ }),
610
+ }),
611
+ z.object({
612
+ type: z.literal('content-delta'),
613
+ index: z.number(),
614
+ delta: z.object({
615
+ message: z.object({
616
+ content: z.union([
617
+ z.object({
618
+ text: z.string(),
619
+ }),
620
+ z.object({
621
+ thinking: z.string(),
622
+ }),
623
+ ]),
612
624
  }),
613
- ]),
625
+ }),
614
626
  }),
615
- }),
616
- }),
617
- z.object({
618
- type: z.literal('content-end'),
619
- index: z.number(),
620
- }),
621
- z.object({
622
- type: z.literal('message-start'),
623
- id: z.string().nullish(),
624
- }),
625
- z.object({
626
- type: z.literal('message-end'),
627
- delta: z.object({
628
- finish_reason: z.string(),
629
- usage: cohereUsageSchema,
630
- }),
631
- }),
632
- // https://docs.cohere.com/v2/docs/streaming#tool-use-stream-events-for-tool-calling
633
- z.object({
634
- type: z.literal('tool-plan-delta'),
635
- delta: z.object({
636
- message: z.object({
637
- tool_plan: z.string(),
627
+ z.object({
628
+ type: z.literal('content-end'),
629
+ index: z.number(),
638
630
  }),
639
- }),
640
- }),
641
- z.object({
642
- type: z.literal('tool-call-start'),
643
- delta: z.object({
644
- message: z.object({
645
- tool_calls: z.object({
646
- id: z.string(),
647
- type: z.literal('function'),
648
- function: z.object({
649
- name: z.string(),
650
- arguments: z.string(),
631
+ z.object({
632
+ type: z.literal('message-start'),
633
+ id: z.string().nullish(),
634
+ }),
635
+ z.object({
636
+ type: z.literal('message-end'),
637
+ delta: z.object({
638
+ finish_reason: z.string(),
639
+ usage: cohereUsageSchema(),
640
+ }),
641
+ }),
642
+ // https://docs.cohere.com/v2/docs/streaming#tool-use-stream-events-for-tool-calling
643
+ z.object({
644
+ type: z.literal('tool-plan-delta'),
645
+ delta: z.object({
646
+ message: z.object({
647
+ tool_plan: z.string(),
651
648
  }),
652
649
  }),
653
650
  }),
654
- }),
655
- }),
656
- // A single tool call's `arguments` stream in chunks and must be accumulated
657
- // in a string and so the full tool object info can only be parsed once we see
658
- // `tool-call-end`.
659
- z.object({
660
- type: z.literal('tool-call-delta'),
661
- delta: z.object({
662
- message: z.object({
663
- tool_calls: z.object({
664
- function: z.object({
665
- arguments: z.string(),
651
+ z.object({
652
+ type: z.literal('tool-call-start'),
653
+ delta: z.object({
654
+ message: z.object({
655
+ tool_calls: z.object({
656
+ id: z.string(),
657
+ type: z.literal('function'),
658
+ function: z.object({
659
+ name: z.string(),
660
+ arguments: z.string(),
661
+ }),
662
+ }),
666
663
  }),
667
664
  }),
668
665
  }),
669
- }),
670
- }),
671
- z.object({
672
- type: z.literal('tool-call-end'),
673
- }),
674
- ]);
666
+ // A single tool call's `arguments` stream in chunks and must be accumulated
667
+ // in a string and so the full tool object info can only be parsed once we see
668
+ // `tool-call-end`.
669
+ z.object({
670
+ type: z.literal('tool-call-delta'),
671
+ delta: z.object({
672
+ message: z.object({
673
+ tool_calls: z.object({
674
+ function: z.object({
675
+ arguments: z.string(),
676
+ }),
677
+ }),
678
+ }),
679
+ }),
680
+ }),
681
+ z.object({
682
+ type: z.literal('tool-call-end'),
683
+ }),
684
+ ]),
685
+ ),
686
+ );
@@ -1,3 +1,8 @@
1
+ import {
2
+ lazySchema,
3
+ zodSchema,
4
+ type InferSchema,
5
+ } from '@ai-sdk/provider-utils';
1
6
  import { z } from 'zod/v4';
2
7
 
3
8
  export type CohereEmbeddingModelId =
@@ -10,57 +15,66 @@ export type CohereEmbeddingModelId =
10
15
  | 'embed-multilingual-v2.0'
11
16
  | (string & {});
12
17
 
13
- export const cohereEmbeddingModelOptions = z.object({
14
- /**
15
- * The format of the returned embedding. Defaults to `float`.
16
- * Non-float formats require Embed v3.0 or newer.
17
- * Binary formats contain packed bytes, not individual vector dimensions.
18
- */
19
- embeddingType: z
20
- .enum(['float', 'int8', 'uint8', 'binary', 'ubinary'])
21
- .optional(),
18
+ export const cohereEmbeddingModelOptions = lazySchema(() =>
19
+ zodSchema(
20
+ z.object({
21
+ /**
22
+ * The format of the returned embedding. Defaults to `float`.
23
+ * Non-float formats require Embed v3.0 or newer.
24
+ * Binary formats contain packed bytes, not individual vector dimensions.
25
+ */
26
+ embeddingType: z
27
+ .enum(['float', 'int8', 'uint8', 'binary', 'ubinary'])
28
+ .optional(),
22
29
 
23
- /**
24
- * Specifies the type of input passed to the model. Default is `search_query`.
25
- *
26
- * - "search_document": Used for embeddings stored in a vector database for search use-cases.
27
- * - "search_query": Used for embeddings of search queries run against a vector DB to find relevant documents.
28
- * - "classification": Used for embeddings passed through a text classifier.
29
- * - "clustering": Used for embeddings run through a clustering algorithm.
30
- */
31
- inputType: z
32
- .enum(['search_document', 'search_query', 'classification', 'clustering'])
33
- .optional(),
30
+ /**
31
+ * Specifies the type of input passed to the model. Default is `search_query`.
32
+ *
33
+ * - "search_document": Used for embeddings stored in a vector database for search use-cases.
34
+ * - "search_query": Used for embeddings of search queries run against a vector DB to find relevant documents.
35
+ * - "classification": Used for embeddings passed through a text classifier.
36
+ * - "clustering": Used for embeddings run through a clustering algorithm.
37
+ */
38
+ inputType: z
39
+ .enum([
40
+ 'search_document',
41
+ 'search_query',
42
+ 'classification',
43
+ 'clustering',
44
+ ])
45
+ .optional(),
34
46
 
35
- /**
36
- * Specifies how the API will handle inputs longer than the maximum token length.
37
- * Default is `END`.
38
- *
39
- * - "NONE": If selected, when the input exceeds the maximum input token length will return an error.
40
- * - "START": Will discard the start of the input until the remaining input is exactly the maximum input token length for the model.
41
- * - "END": Will discard the end of the input until the remaining input is exactly the maximum input token length for the model.
42
- */
43
- truncate: z.enum(['NONE', 'START', 'END']).optional(),
47
+ /**
48
+ * Specifies how the API will handle inputs longer than the maximum token length.
49
+ * Default is `END`.
50
+ *
51
+ * - "NONE": If selected, when the input exceeds the maximum input token length will return an error.
52
+ * - "START": Will discard the start of the input until the remaining input is exactly the maximum input token length for the model.
53
+ * - "END": Will discard the end of the input until the remaining input is exactly the maximum input token length for the model.
54
+ */
55
+ truncate: z.enum(['NONE', 'START', 'END']).optional(),
44
56
 
45
- /**
46
- * The number of dimensions of the output embedding.
47
- * Only available for `embed-v4.0` and newer models.
48
- *
49
- * Embed v4.0 supports `256`, `512`, `1024`, and `1536` (default).
50
- * Embed v5.0 Pro and Fast also support `768` and `2048` (default).
51
- */
52
- outputDimension: z
53
- .union([
54
- z.literal(256),
55
- z.literal(512),
56
- z.literal(768),
57
- z.literal(1024),
58
- z.literal(1536),
59
- z.literal(2048),
60
- ])
61
- .optional(),
62
- });
57
+ /**
58
+ * The number of dimensions of the output embedding.
59
+ * Only available for `embed-v4.0` and newer models.
60
+ *
61
+ * Embed v4.0 supports `256`, `512`, `1024`, and `1536` (default).
62
+ * Embed v5.0 Pro and Fast also support `768` and `2048` (default).
63
+ */
64
+ outputDimension: z
65
+ .union([
66
+ z.literal(256),
67
+ z.literal(512),
68
+ z.literal(768),
69
+ z.literal(1024),
70
+ z.literal(1536),
71
+ z.literal(2048),
72
+ ])
73
+ .optional(),
74
+ }),
75
+ ),
76
+ );
63
77
 
64
- export type CohereEmbeddingModelOptions = z.infer<
78
+ export type CohereEmbeddingModelOptions = InferSchema<
65
79
  typeof cohereEmbeddingModelOptions
66
80
  >;
@@ -1,11 +1,20 @@
1
- import { createJsonErrorResponseHandler } from '@ai-sdk/provider-utils';
1
+ import {
2
+ lazySchema,
3
+ zodSchema,
4
+ type InferSchema,
5
+ createJsonErrorResponseHandler,
6
+ } from '@ai-sdk/provider-utils';
2
7
  import { z } from 'zod/v4';
3
8
 
4
- const cohereErrorDataSchema = z.object({
5
- message: z.string(),
6
- });
9
+ const cohereErrorDataSchema = lazySchema(() =>
10
+ zodSchema(
11
+ z.object({
12
+ message: z.string(),
13
+ }),
14
+ ),
15
+ );
7
16
 
8
- export type CohereErrorData = z.infer<typeof cohereErrorDataSchema>;
17
+ export type CohereErrorData = InferSchema<typeof cohereErrorDataSchema>;
9
18
 
10
19
  export const cohereFailedResponseHandler = createJsonErrorResponseHandler({
11
20
  errorSchema: cohereErrorDataSchema,