dmitry.galkin/chatbot-template
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: update providers (#881)

Browse source
This commit is contained in:
Jeremy 2025-03-20 14:10:45 -07:00 • committed by GitHub
parent 9058240e10
commit 7811c0f7fb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 122 additions and 108 deletions
Download patch file Download diff file Expand all files Collapse all files

9
lib/ai/providers.ts
View file

@ -3,9 +3,9 @@ import {
extractReasoningMiddleware,
wrapLanguageModel,
} from 'ai';
import { openai } from '@ai-sdk/openai';
import { fireworks } from '@ai-sdk/fireworks';
import { groq } from '@ai-sdk/groq';
import { xai } from '@ai-sdk/xai';
import { fal } from '@ai-sdk/fal';
import { isTestEnvironment } from '../constants';
import {
artifactModel,
@ -27,14 +27,13 @@ export const myProvider = isTestEnvironment
languageModels: {
'chat-model': xai('grok-2-1212'),
'chat-model-reasoning': wrapLanguageModel({
model: fireworks('accounts/fireworks/models/deepseek-r1'),
model: groq('deepseek-r1-distill-llama-70b'),
middleware: extractReasoningMiddleware({ tagName: 'think' }),
}),
'title-model': xai('grok-2-1212'),
'artifact-model': xai('grok-2-1212'),
},
imageModels: {
'small-model': openai.image('dall-e-2'),
'large-model': openai.image('dall-e-3'),
'small-model': fal.image('fal-ai/fast-sdxl'),
},
});