Skip to content

Commit 01a6e2a

Browse files
committed
Add Gemini 2.5 Pro Exp model and standardize quote style
1 parent bb9899f commit 01a6e2a

File tree

2 files changed

+52
-44
lines changed

2 files changed

+52
-44
lines changed

lib/ai/models.ts

Lines changed: 27 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
export const DEFAULT_CHAT_MODEL: string = 'gemini-2.0-flash';
1+
export const DEFAULT_CHAT_MODEL: string = "gemini-2.0-flash";
22

33
interface ChatModel {
44
id: string;
@@ -8,34 +8,41 @@ interface ChatModel {
88

99
export const chatModels: Array<ChatModel> = [
1010
{
11-
id: 'gemini-2.0-pro-exp-02-05',
12-
name: 'Gemini 2.0 Pro Exp 02-05',
13-
description: 'Improved quality, especially for world knowledge, code, and long context',
11+
id: "gemini-2.5-pro-exp-03-25",
12+
name: "Gemini 2.5 Pro Exp 03-25",
13+
description:
14+
"Gemini 2.5 Pro Experimental 是我们最先进的思考模型,能够推理代码、数学和 STEM 领域的复杂问题,以及使用长上下文分析大型数据集、代码库和文档。",
1415
},
1516
{
16-
id: 'gemini-2.0-flash-thinking-exp',
17-
name: 'Gemini 2.0 Flash Thinking Exp',
18-
description: 'Reasoning for complex problems, features a new Thinking mode',
17+
id: "gemini-2.0-pro-exp-02-05",
18+
name: "Gemini 2.0 Pro Exp 02-05",
19+
description:
20+
"Improved quality, especially for world knowledge, code, and long context",
1921
},
2022
{
21-
id: 'gemini-2.0-flash',
22-
name: 'Gemini 2.0 Flash',
23+
id: "gemini-2.0-flash-thinking-exp",
24+
name: "Gemini 2.0 Flash Thinking Exp",
25+
description: "Reasoning for complex problems, features a new Thinking mode",
26+
},
27+
{
28+
id: "gemini-2.0-flash",
29+
name: "Gemini 2.0 Flash",
2330
description:
24-
'Next generation features, speed, and multimodal generation for a diverse variety of tasks',
31+
"Next generation features, speed, and multimodal generation for a diverse variety of tasks",
2532
},
2633
{
27-
id: 'gemini-exp-1206',
28-
name: 'Gemini Exp 1206',
29-
description: 'Quality improvements, celebrate 1 year of Gemini'
34+
id: "gemini-exp-1206",
35+
name: "Gemini Exp 1206",
36+
description: "Quality improvements, celebrate 1 year of Gemini",
3037
},
3138
{
32-
id: 'deepseek-chat',
33-
name: 'DeepSeek Chat',
34-
description: 'DeepSeek-V3',
39+
id: "deepseek-chat",
40+
name: "DeepSeek Chat",
41+
description: "DeepSeek-V3",
3542
},
3643
{
37-
id: 'chat-model-reasoning',
38-
name: 'Reasoning model',
39-
description: 'Uses advanced reasoning',
40-
}
44+
id: "chat-model-reasoning",
45+
name: "Reasoning model",
46+
description: "Uses advanced reasoning",
47+
},
4148
];

lib/ai/providers.ts

Lines changed: 25 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -2,48 +2,49 @@ import {
22
customProvider,
33
extractReasoningMiddleware,
44
wrapLanguageModel,
5-
} from 'ai';
6-
import { google } from '@ai-sdk/google';
7-
import { fireworks } from '@ai-sdk/fireworks';
8-
import { isTestEnvironment } from '../constants';
5+
} from "ai";
6+
import { google } from "@ai-sdk/google";
7+
import { fireworks } from "@ai-sdk/fireworks";
8+
import { isTestEnvironment } from "../constants";
99
import {
1010
artifactModel,
1111
chatModel,
1212
reasoningModel,
1313
titleModel,
14-
} from './models.test';
14+
} from "./models.test";
1515

1616
export const myProvider = isTestEnvironment
1717
? customProvider({
1818
languageModels: {
19-
'chat-model': chatModel,
20-
'chat-model-reasoning': reasoningModel,
21-
'title-model': titleModel,
22-
'artifact-model': artifactModel,
19+
"chat-model": chatModel,
20+
"chat-model-reasoning": reasoningModel,
21+
"title-model": titleModel,
22+
"artifact-model": artifactModel,
2323
},
2424
})
2525
: customProvider({
2626
languageModels: {
27-
'gemini-2.0-pro-exp-02-05': google('gemini-2.0-pro-exp-02-05'),
28-
'gemini-2.0-flash-thinking-exp': google(
29-
'gemini-2.0-flash-thinking-exp'
27+
"gemini-2.5-pro-exp-03-25": google("gemini-2.5-pro-exp-03-25"),
28+
"gemini-2.0-pro-exp-02-05": google("gemini-2.0-pro-exp-02-05"),
29+
"gemini-2.0-flash-thinking-exp": google(
30+
"gemini-2.0-flash-thinking-exp",
3031
),
31-
'gemini-2.0-flash': google('gemini-2.0-flash'),
32-
'gemini-exp-1206': google('gemini-exp-1206'),
33-
'deepseek-chat': fireworks('accounts/fireworks/models/deepseek-v3'),
34-
'chat-model-reasoning': wrapLanguageModel({
35-
model: fireworks('accounts/fireworks/models/deepseek-r1'),
36-
middleware: extractReasoningMiddleware({ tagName: 'think' }),
32+
"gemini-2.0-flash": google("gemini-2.0-flash"),
33+
"gemini-exp-1206": google("gemini-exp-1206"),
34+
"deepseek-chat": fireworks("accounts/fireworks/models/deepseek-v3"),
35+
"chat-model-reasoning": wrapLanguageModel({
36+
model: fireworks("accounts/fireworks/models/deepseek-r1"),
37+
middleware: extractReasoningMiddleware({ tagName: "think" }),
3738
}),
38-
'title-model': google('gemini-2.0-flash'),
39-
'artifact-model': google('gemini-2.0-flash'),
39+
"title-model": google("gemini-2.0-flash"),
40+
"artifact-model": google("gemini-2.0-flash"),
4041
},
4142
imageModels: {
42-
'small-model': fireworks.image(
43-
'accounts/fireworks/models/stable-diffusion-3p5-medium'
43+
"small-model": fireworks.image(
44+
"accounts/fireworks/models/stable-diffusion-3p5-medium",
4445
),
45-
'large-model': fireworks.image(
46-
'accounts/fireworks/models/stable-diffusion-3p5-large-turbo'
46+
"large-model": fireworks.image(
47+
"accounts/fireworks/models/stable-diffusion-3p5-large-turbo",
4748
),
4849
},
4950
});

0 commit comments

Comments
 (0)