From 621578881bb34c0d2ebd6deda49f8a32a1fe8aec Mon Sep 17 00:00:00 2001
From: Spike Lu <luyuanxin1995@gmail.com>
Date: Fri, 27 Dec 2024 08:04:16 -0800
Subject: [PATCH] add cost tracking for o1

---
 internal/provider/openai/cost.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/internal/provider/openai/cost.go b/internal/provider/openai/cost.go
index dfa9040..641a577 100644
--- a/internal/provider/openai/cost.go
+++ b/internal/provider/openai/cost.go
@@ -34,6 +34,8 @@ func parseFinetuneModel(model string) string {
 
 var OpenAiPerThousandTokenCost = map[string]map[string]float64{
 	"prompt": {
+		"o1":                          0.015,
+		"o1-2024-12-17":               0.015,
 		"o1-preview":                  0.015,
 		"o1-preview-2024-09-12":       0.015,
 		"gpt-4o":                      0.0025,
@@ -98,6 +100,8 @@ var OpenAiPerThousandTokenCost = map[string]map[string]float64{
 		"tts-1-hd":  0.03,
 	},
 	"completion": {
+		"o1":                          0.06,
+		"o1-2024-12-17":               0.06,
 		"o1-preview":                  0.06,
 		"o1-preview-2024-09-12":       0.06,
 		"gpt-3.5-turbo-1106":          0.002,