update: claude version

mirrors2 · Oct 28, 2024 · af4f8db · af4f8db
1 parent ae3b72c
commit af4f8db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/tokenizer/tokenizer.go b/pkg/tokenizer/tokenizer.go
@@ -167,7 +167,7 @@ func Cost(model string, promptCount, completionCount int) float64 {
 		cost = (0.003/1000)*float64(prompt) + (0.015/1000)*float64(completion)
 	case "claude-3-opus-20240229":
 		cost = (0.015/1000)*float64(prompt) + (0.075/1000)*float64(completion)
-	case "claude-3-5-sonnet", "claude-3-5-sonnet-20240620":
+	case "claude-3-5-sonnet", "claude-3-5-sonnet-20240620", "claude-3-5-sonnet-20241022", "claude-3-5-sonnet-latest":
 		cost = (0.003/1000)*float64(prompt) + (0.015/1000)*float64(completion)
 	// google
 	// https://ai.google.dev/pricing?hl=zh-cn