ai/src/model/as-reranking-model-v4.ts

AI SDK by Vercel - build apps like ChatGPT, Claude, Gemini, and more with a single interface for any model using the Vercel AI Gateway or go direct to OpenAI, Anthropic, Google, or any other model provider.

import type { RerankingModelV3, RerankingModelV4 } from '@ai-sdk/provider';

export function asRerankingModelV4(
  model: RerankingModelV3 | RerankingModelV4,
): RerankingModelV4 {
  if (model.specificationVersion === 'v4') {
    return model;
  }

  return new Proxy(model, {
    get(target, prop: keyof RerankingModelV3) {
      if (prop === 'specificationVersion') return 'v4';
      return target[prop];
    },
  }) as unknown as RerankingModelV4;
}