13 lines
398 B
TypeScript
13 lines
398 B
TypeScript
import { openai } from '@ai-sdk/openai';
|
|
import { experimental_wrapLanguageModel as wrapLanguageModel } from 'ai';
|
|
|
|
import { customMiddleware } from './custom-middleware';
|
|
|
|
export const customModel = (apiIdentifier: string) => {
|
|
return wrapLanguageModel({
|
|
model: openai(apiIdentifier),
|
|
middleware: customMiddleware,
|
|
});
|
|
};
|
|
|
|
export const imageGenerationModel = openai.image('dall-e-3');
|