Skip to content

Commit 22d0ccb

Browse files
committed
chore: added gemini models
1 parent 1a18f04 commit 22d0ccb

File tree

1 file changed

+37
-1
lines changed
  • packages/core/src/services/ai/estimateCost

1 file changed

+37
-1
lines changed

packages/core/src/services/ai/estimateCost/google.ts

Lines changed: 37 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,20 @@ import { createModelSpec } from './helpers'
77
export const GOOGLE_MODELS = createModelSpec({
88
defaultModel: 'gemini-1.5-flash',
99
models: {
10+
'gemini-2.5-pro': {
11+
// Note: Pricing includes thinking tokens
12+
cost: [
13+
{ input: 1.25, output: 10.0 },
14+
{ input: 2.5, output: 15.0, tokensRangeStart: 200_000 },
15+
],
16+
},
17+
'gemini-2.5-pro-preview-05-06': {
18+
// Note: Pricing includes thinking tokens
19+
cost: [
20+
{ input: 1.25, output: 10.0 },
21+
{ input: 2.5, output: 15.0, tokensRangeStart: 200_000 },
22+
],
23+
},
1024
// --- Gemini 2.5 Preview Models ---
1125
'gemini-2.5-pro-preview-03-25': {
1226
// Note: Pricing includes thinking tokens
@@ -15,6 +29,18 @@ export const GOOGLE_MODELS = createModelSpec({
1529
{ input: 2.5, output: 15.0, tokensRangeStart: 200_000 },
1630
],
1731
},
32+
'gemini-2.5-flash-lite-preview-06-17': {
33+
// Note: Using non-thinking output cost
34+
cost: { input: 0.1, output: 0.4 }, // Input assumes text/image/video cost
35+
},
36+
'gemini-2.5-flash': {
37+
// Note: Using non-thinking output cost
38+
cost: { input: 0.3, output: 2.5 }, // Input assumes text/image/video cost
39+
},
40+
'gemini-2.5-flash-preview-05-20': {
41+
// Note: Using non-thinking output cost
42+
cost: { input: 0.15, output: 0.6 }, // Input assumes text/image/video cost
43+
},
1844
'gemini-2.5-flash-preview-04-17': {
1945
// Note: Using non-thinking output cost
2046
cost: { input: 0.15, output: 0.6 }, // Input assumes text/image/video cost
@@ -59,11 +85,21 @@ export const GOOGLE_MODELS = createModelSpec({
5985
},
6086
modelName: (model: string) => {
6187
// Exact matches first for specific/preview models
88+
if (model === 'gemini-2.5-pro') return 'gemini-2.5-pro'
89+
if (model === 'gemini-2.5-pro-preview-05-06')
90+
return 'gemini-2.5-pro-preview-05-06'
6291
if (model === 'gemini-2.5-pro-preview-03-25')
6392
return 'gemini-2.5-pro-preview-03-25'
93+
if (model === 'gemini-2.5-pro-exp-03-25') return 'gemini-2.5-pro-exp-03-25'
94+
if (model === 'gemini-2.5-pro-preview-03-25')
95+
return 'gemini-2.5-pro-preview-03-25'
96+
if (model === 'gemini-2.5-flash-lite-preview-06-17')
97+
return 'gemini-2.5-flash-lite-preview-06-17'
98+
if (model === 'gemini-2.5-flash') return 'gemini-2.5-flash'
99+
if (model === 'gemini-2.5-flash-preview-05-20')
100+
return 'gemini-2.5-flash-preview-05-20'
64101
if (model === 'gemini-2.5-flash-preview-04-17')
65102
return 'gemini-2.5-flash-preview-04-17'
66-
if (model === 'gemini-2.5-pro-exp-03-25') return 'gemini-2.5-pro-exp-03-25'
67103
if (model === 'imagen-3') return 'imagen-3'
68104
if (model === 'veo-2') return 'veo-2'
69105
if (model === 'gemma-3') return 'gemma-3'

0 commit comments

Comments
 (0)