up

2024-04-21 00:54:57 +08:00
parent 151757c2fe
commit 752053f7c4
3 changed files with 5 additions and 3 deletions
--- a/pkg/tokenizer/tokenizer.go
+++ b/pkg/tokenizer/tokenizer.go
@@ -155,7 +155,8 @@ func Cost(model string, promptCount, completionCount int) float64 {
 	case "gemini-1.5-pro-latest":
 		cost = (0.00025/1000)*float64(prompt) + (0.0005/1000)*float64(completion)

-		// Mistral AI
+	// Mistral AI
+	// https://docs.mistral.ai/platform/pricing/
 	case "mistral-small-latest":
 		cost = (0.002/1000)*float64(prompt) + (0.006/1000)*float64(completion)
 	case "mistral-medium-latest":