From 000fc2f6164bb8e75ded7a6f2450fb8bda30d537 Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Thu, 12 Dec 2024 16:25:35 +0800
Subject: [PATCH] new claude model

---
 pkg/tokenizer/tokenizer.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pkg/tokenizer/tokenizer.go b/pkg/tokenizer/tokenizer.go
index bdcc6a6..665ae89 100644
--- 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