@ai-sdk/cohere 4.0.0-beta.2 → 4.0.0-beta.20

package/src/cohere-provider.ts

@@ -1,9 +1,9 @@
 import {
-  EmbeddingModelV3,
-  LanguageModelV3,
+  EmbeddingModelV4,
+  LanguageModelV4,
   NoSuchModelError,
-  RerankingModelV3,
-  ProviderV3,
+  RerankingModelV4,
+  ProviderV4,
 } from '@ai-sdk/provider';
 
 import {
@@ -21,43 +21,43 @@ import { CohereRerankingModel } from './reranking/cohere-reranking-model';
 import { CohereEmbeddingModelId } from './cohere-embedding-options';
 import { VERSION } from './version';
 
-export interface CohereProvider extends ProviderV3 {
-  (modelId: CohereChatModelId): LanguageModelV3;
+export interface CohereProvider extends ProviderV4 {
+  (modelId: CohereChatModelId): LanguageModelV4;
 
   /**
    * Creates a model for text generation.
    */
-  languageModel(modelId: CohereChatModelId): LanguageModelV3;
+  languageModel(modelId: CohereChatModelId): LanguageModelV4;
 
   /**
    * Creates a model for text embeddings.
    */
-  embedding(modelId: CohereEmbeddingModelId): EmbeddingModelV3;
+  embedding(modelId: CohereEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * Creates a model for text embeddings.
    */
-  embeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV3;
+  embeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * @deprecated Use `embedding` instead.
    */
-  textEmbedding(modelId: CohereEmbeddingModelId): EmbeddingModelV3;
+  textEmbedding(modelId: CohereEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * @deprecated Use `embeddingModel` instead.
    */
-  textEmbeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV3;
+  textEmbeddingModel(modelId: CohereEmbeddingModelId): EmbeddingModelV4;
 
   /**
    * Creates a model for reranking.
    */
-  reranking(modelId: CohereRerankingModelId): RerankingModelV3;
+  reranking(modelId: CohereRerankingModelId): RerankingModelV4;
 
   /**
    * Creates a model for reranking.
    */
-  rerankingModel(modelId: CohereRerankingModelId): RerankingModelV3;
+  rerankingModel(modelId: CohereRerankingModelId): RerankingModelV4;
 }
 
 export interface CohereProviderSettings {
@@ -147,7 +147,7 @@ export function createCohere(
     return createChatModel(modelId);
   };
 
-  provider.specificationVersion = 'v3' as const;
+  provider.specificationVersion = 'v4' as const;
   provider.languageModel = createChatModel;
   provider.embedding = createEmbeddingModel;
   provider.embeddingModel = createEmbeddingModel;

package/src/convert-cohere-usage.ts

@@ -1,4 +1,4 @@
-import { LanguageModelV3Usage } from '@ai-sdk/provider';
+import { LanguageModelV4Usage } from '@ai-sdk/provider';
 
 export type CohereUsageTokens = {
   input_tokens: number;
@@ -7,7 +7,7 @@ export type CohereUsageTokens = {
 
 export function convertCohereUsage(
   tokens: CohereUsageTokens | undefined | null,
-): LanguageModelV3Usage {
+): LanguageModelV4Usage {
   if (tokens == null) {
     return {
       inputTokens: {

package/src/convert-to-cohere-chat-prompt.ts

@@ -1,20 +1,21 @@
 import {
-  SharedV3Warning,
-  LanguageModelV3Prompt,
+  SharedV4Warning,
+  LanguageModelV4Prompt,
   UnsupportedFunctionalityError,
 } from '@ai-sdk/provider';
+import { isProviderReference } from '@ai-sdk/provider-utils';
 import { CohereAssistantMessage, CohereChatPrompt } from './cohere-chat-prompt';
 
-export function convertToCohereChatPrompt(prompt: LanguageModelV3Prompt): {
+export function convertToCohereChatPrompt(prompt: LanguageModelV4Prompt): {
   messages: CohereChatPrompt;
   documents: Array<{
     data: { text: string; title?: string };
   }>;
-  warnings: SharedV3Warning[];
+  warnings: SharedV4Warning[];
 } {
   const messages: CohereChatPrompt = [];
   const documents: Array<{ data: { text: string; title?: string } }> = [];
-  const warnings: SharedV3Warning[] = [];
+  const warnings: SharedV4Warning[] = [];
 
   for (const { role, content } of prompt) {
     switch (role) {
@@ -33,6 +34,12 @@ export function convertToCohereChatPrompt(prompt: LanguageModelV3Prompt): {
                   return part.text;
                 }
                 case 'file': {
+                  if (isProviderReference(part.data)) {
+                    throw new UnsupportedFunctionalityError({
+                      functionality: 'file parts with provider references',
+                    });
+                  }
+
                   // Extract documents for RAG
                   let textContent: string;
 

package/src/map-cohere-finish-reason.ts

@@ -1,8 +1,8 @@
-import { LanguageModelV3FinishReason } from '@ai-sdk/provider';
+import { LanguageModelV4FinishReason } from '@ai-sdk/provider';
 
 export function mapCohereFinishReason(
   finishReason: string | null | undefined,
-): LanguageModelV3FinishReason['unified'] {
+): LanguageModelV4FinishReason['unified'] {
   switch (finishReason) {
     case 'COMPLETE':
     case 'STOP_SEQUENCE':

package/src/reranking/cohere-reranking-model.ts

@@ -1,4 +1,4 @@
-import { RerankingModelV3, SharedV3Warning } from '@ai-sdk/provider';
+import { RerankingModelV4, SharedV4Warning } from '@ai-sdk/provider';
 import {
   combineHeaders,
   createJsonResponseHandler,
@@ -23,8 +23,8 @@ type CohereRerankingConfig = {
   fetch?: FetchFunction;
 };
 
-export class CohereRerankingModel implements RerankingModelV3 {
-  readonly specificationVersion = 'v3';
+export class CohereRerankingModel implements RerankingModelV4 {
+  readonly specificationVersion = 'v4';
   readonly modelId: CohereRerankingModelId;
 
   private readonly config: CohereRerankingConfig;
@@ -46,8 +46,8 @@ export class CohereRerankingModel implements RerankingModelV3 {
     topN,
     abortSignal,
     providerOptions,
-  }: Parameters<RerankingModelV3['doRerank']>[0]): Promise<
-    Awaited<ReturnType<RerankingModelV3['doRerank']>>
+  }: Parameters<RerankingModelV4['doRerank']>[0]): Promise<
+    Awaited<ReturnType<RerankingModelV4['doRerank']>>
   > {
     const rerankingOptions = await parseProviderOptions({
       provider: 'cohere',
@@ -55,7 +55,7 @@ export class CohereRerankingModel implements RerankingModelV3 {
       schema: cohereRerankingModelOptionsSchema,
     });
 
-    const warnings: SharedV3Warning[] = [];
+    const warnings: SharedV4Warning[] = [];
 
     if (documents.type === 'object') {
       warnings.push({