new claude model

mirrors2 · Dec 12, 2024 · 000fc2f · 000fc2f
1 parent 9788703
commit 000fc2f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/tokenizer/tokenizer.go b/pkg/tokenizer/tokenizer.go
@@ -163,11 +163,13 @@ func Cost(model string, promptCount, completionCount int) float64 {
 		cost = (0.015/1000)*float64(prompt) + (0.075/1000)*float64(completion)
 	case "claude-3-haiku-20240307":
 		cost = (0.00025/1000)*float64(prompt) + (0.00125/1000)*float64(completion)
+	case "claude-3-5-haiku-latest", "claude-3-5-haiku-20241022":
+		cost = (0.001/1000)*float64(prompt) + (0.005/1000)*float64(completion)
 	case "claude-3-sonnet-20240229":
 		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", "claude-3-5-sonnet-20241022", "claude-3-5-sonnet-latest":
+	case "claude-3-5-sonnet", "claude-3-5-sonnet-latest", "claude-3-5-sonnet-20240620", "claude-3-5-sonnet-20241022":
 		cost = (0.003/1000)*float64(prompt) + (0.015/1000)*float64(completion)
 	// google
 	// https://ai.google.dev/pricing?hl=zh-cn