mirrors2/opencatd-open
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gemini flash

Browse Source
This commit is contained in:
Sakurasan
2024-05-15 18:27:29 +08:00
parent 4fae477af8
commit a3920f11fa
3 changed files with 29 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pkg/tokenizer/tokenizer.go
View File

@@ -151,11 +151,13 @@ func Cost(model string, promptCount, completionCount int) float64 {
// google
// https://ai.google.dev/pricing?hl=zh-cn
case "gemini-pro":
cost = (0.000125/1000)*float64(prompt) + (0.000375/1000)*float64(completion)
cost = (0.0005/1000)*float64(prompt) + (0.0015/1000)*float64(completion)
case "gemini-pro-vision":
cost = (0.000125/1000)*float64(prompt) + (0.000375/1000)*float64(completion)
cost = (0.0005/1000)*float64(prompt) + (0.0015/1000)*float64(completion)
case "gemini-1.5-pro-latest":
cost = (0.00025/1000)*float64(prompt) + (0.0005/1000)*float64(completion)
cost = (0.0035/1000)*float64(prompt) + (0.0105/1000)*float64(completion)
case "gemini-1.5-flash-latest":
cost = (0.00035/1000)*float64(prompt) + (0.00053/1000)*float64(completion)
// Mistral AI
// https://docs.mistral.ai/platform/pricing/