ai/src/model/get-embedding-model-max-input-bytes-per-call.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 { EmbeddingModelV4 } from '@ai-sdk/provider';
import { EXPERIMENTAL_EMBEDDING_MODEL_MAX_INPUT_BYTES_PER_CALL } from '@ai-sdk/provider-utils';

export type EmbeddingModelWithMaxInputBytesPerCall = EmbeddingModelV4 & {
  readonly [EXPERIMENTAL_EMBEDDING_MODEL_MAX_INPUT_BYTES_PER_CALL]?:
    | PromiseLike<number | undefined>
    | number
    | undefined;
};

export function getEmbeddingModelMaxInputBytesPerCall(model: EmbeddingModelV4) {
  return (model as EmbeddingModelWithMaxInputBytesPerCall)[
    EXPERIMENTAL_EMBEDDING_MODEL_MAX_INPUT_BYTES_PER_CALL
  ];
}