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

38
lib/ai/providers.ts
View file

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