File size: 997 Bytes
867b17d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import {
customProvider,
extractReasoningMiddleware,
wrapLanguageModel,
} from 'ai';
import { xai } from '@ai-sdk/xai';
import {
artifactModel,
chatModel,
reasoningModel,
titleModel,
} from './models.test';
import { isTestEnvironment } from '../constants';
export const myProvider = isTestEnvironment
? customProvider({
languageModels: {
'chat-model': chatModel,
'chat-model-reasoning': reasoningModel,
'title-model': titleModel,
'artifact-model': artifactModel,
},
})
: customProvider({
languageModels: {
'chat-model': xai('grok-2-vision-1212'),
'chat-model-reasoning': wrapLanguageModel({
model: xai('grok-3-mini-beta'),
middleware: extractReasoningMiddleware({ tagName: 'think' }),
}),
'title-model': xai('grok-2-1212'),
'artifact-model': xai('grok-2-1212'),
},
imageModels: {
'small-model': xai.imageModel('grok-2-image'),
},
});
|