ai
Version:
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.
23 lines (21 loc) • 663 B
text/typescript
import { EmbeddingModelV3CallOptions } from '@ai-sdk/provider';
import { EmbeddingModelMiddleware } from '../types';
import { mergeObjects } from '../util/merge-objects';
/**
* Applies default settings for an embedding model.
*/
export function defaultEmbeddingSettingsMiddleware({
settings,
}: {
settings: Partial<{
headers?: EmbeddingModelV3CallOptions['headers'];
providerOptions?: EmbeddingModelV3CallOptions['providerOptions'];
}>;
}): EmbeddingModelMiddleware {
return {
specificationVersion: 'v3',
transformParams: async ({ params }) => {
return mergeObjects(settings, params) as EmbeddingModelV3CallOptions;
},
};
}