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

integrate ai gateway and improve ui components (#1145)

Browse source
This commit is contained in:
josh 2025-09-01 11:07:07 +01:00 • committed by GitHub
parent 7315ced92f
commit dff2191611
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
29 changed files with 235 additions and 217 deletions
Download patch file Download diff file Expand all files Collapse all files

8
lib/ai/models.ts
View file

@ -9,12 +9,12 @@ export interface ChatModel {
export const chatModels: Array<ChatModel> = [
{
id: 'chat-model',
name: 'Chat model',
description: 'Primary model for all-purpose chat',
name: 'Grok Vision',
description: 'Advanced multimodal model with vision and text capabilities',
},
{
id: 'chat-model-reasoning',
name: 'Reasoning model',
description: 'Uses advanced reasoning',
name: 'Grok Reasoning',
description: 'Uses advanced chain-of-thought reasoning for complex problems',
},
];

13
lib/ai/providers.ts
View file

@ -3,7 +3,7 @@ import {
extractReasoningMiddleware,
wrapLanguageModel,
} from 'ai';
import { xai } from '@ai-sdk/xai';
import { gateway } from '@ai-sdk/gateway';
import {
artifactModel,
chatModel,
@ -23,15 +23,12 @@ export const myProvider = isTestEnvironment
})
: customProvider({
languageModels: {
'chat-model': xai('grok-2-vision-1212'),
'chat-model': gateway.languageModel('xai/grok-2-vision-1212'),
'chat-model-reasoning': wrapLanguageModel({
model: xai('grok-3-mini-beta'),
model: gateway.languageModel('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'),
'title-model': gateway.languageModel('xai/grok-2-1212'),
'artifact-model': gateway.languageModel('xai/grok-2-1212'),
},
});