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

fix quotes

Browse source
This commit is contained in:
Walter Korman 2025-03-17 16:54:40 -07:00
parent fb3e01981e
commit 6e6cf7830a
2 changed files with 26 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

14
lib/ai/models.ts
View file

@ -1,4 +1,4 @@
export const DEFAULT_CHAT_MODEL: string = "chat-model-small";
export const DEFAULT_CHAT_MODEL: string = 'chat-model-small';
interface ChatModel {
id: string;
@ -8,13 +8,13 @@ interface ChatModel {
export const chatModels: Array<ChatModel> = [
{
id: "chat-model-small",
name: "Small model",
description: "Small model for fast, lightweight tasks",
id: 'chat-model-small',
name: 'Small model',
description: 'Small model for fast, lightweight tasks',
},
{
id: "chat-model-reasoning",
name: "Reasoning model",
description: "Uses advanced reasoning",
id: 'chat-model-reasoning',
name: 'Reasoning model',
description: 'Uses advanced reasoning',
},
];

38
lib/ai/providers.ts
View file

@ -2,40 +2,40 @@ import {
customProvider,
extractReasoningMiddleware,
wrapLanguageModel,
} from "ai";
import { openai } from "@ai-sdk/openai";
import { fireworks } from "@ai-sdk/fireworks";
import { xai } from "@ai-sdk/xai";
import { isTestEnvironment } from "../constants";
} from 'ai';
import { openai } from '@ai-sdk/openai';
import { fireworks } from '@ai-sdk/fireworks';
import { xai } from '@ai-sdk/xai';
import { isTestEnvironment } from '../constants';
import {
artifactModel,
chatModel,
reasoningModel,
titleModel,
} from "./models.test";
} from './models.test';
export const myProvider = isTestEnvironment
? customProvider({
languageModels: {
"chat-model-small": chatModel,
"chat-model-large": chatModel,
"chat-model-reasoning": reasoningModel,
"title-model": titleModel,
"artifact-model": artifactModel,
'chat-model-small': chatModel,
'chat-model-large': chatModel,
'chat-model-reasoning': reasoningModel,
'title-model': titleModel,
'artifact-model': artifactModel,
},
})
: customProvider({
languageModels: {
"chat-model-small": xai("grok-2-1212"),
"chat-model-reasoning": wrapLanguageModel({
model: fireworks("accounts/fireworks/models/deepseek-r1"),
middleware: extractReasoningMiddleware({ tagName: "think" }),
'chat-model-small': xai('grok-2-1212'),
'chat-model-reasoning': wrapLanguageModel({
model: fireworks('accounts/fireworks/models/deepseek-r1'),
middleware: extractReasoningMiddleware({ tagName: 'think' }),
}),
"title-model": xai("grok-2-1212"),
"artifact-model": xai("grok-2-1212"),
'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': openai.image('dall-e-2'),
'large-model': openai.image('dall-e-3'),
},
});