Make model's AddPrices() public

JiawenZou · Mar 2, 2024 · 30cb1a9 · 30cb1a9
1 parent 538386c
commit 30cb1a9
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/model/claude.go b/model/claude.go
@@ -58,7 +58,7 @@ func (p *ClaudeModelProvider) calculatePrice(modelResult *ModelResult) error {
 
 	inputPrice := getPrice(modelResult.PromptTokenCount, inputPricePerThousandTokens)
 	outputPrice := getPrice(modelResult.ResponseTokenCount, outputPricePerThousandTokens)
-	modelResult.TotalPrice = addPrices(inputPrice, outputPrice)
+	modelResult.TotalPrice = AddPrices(inputPrice, outputPrice)
 	modelResult.Currency = "USD"
 	return nil
 }

diff --git a/model/cohere.go b/model/cohere.go
@@ -94,7 +94,7 @@ func (p *CohereModelProvider) calculatePrice(modelResult *ModelResult) error {
 
 	inputPrice := getPrice(modelResult.PromptTokenCount, inputPricePerThousandTokens)
 	outputPrice := getPrice(modelResult.ResponseTokenCount, outputPricePerThousandTokens)
-	modelResult.TotalPrice = addPrices(inputPrice, outputPrice)
+	modelResult.TotalPrice = AddPrices(inputPrice, outputPrice)
 	modelResult.Currency = "USD"
 	return nil
 }

diff --git a/model/gemini.go b/model/gemini.go
@@ -96,7 +96,7 @@ func (p *GeminiModelProvider) calculatePrice(modelResult *ModelResult) error {
 
 	inputPrice := getPrice(modelResult.PromptTokenCount, inputPricePerThousandTokens)
 	outputPrice := getPrice(modelResult.ResponseTokenCount, outputPricePerThousandTokens)
-	modelResult.TotalPrice = addPrices(inputPrice, outputPrice)
+	modelResult.TotalPrice = AddPrices(inputPrice, outputPrice)
 	modelResult.Currency = "USD"
 	return nil
 }

diff --git a/model/local.go b/model/local.go
@@ -117,7 +117,7 @@ func (p *LocalModelProvider) calculatePrice(modelResult *ModelResult) error {
 
 	inputPrice := getPrice(modelResult.PromptTokenCount, inputPricePerThousandTokens)
 	outputPrice := getPrice(modelResult.ResponseTokenCount, outputPricePerThousandTokens)
-	modelResult.TotalPrice = addPrices(inputPrice, outputPrice)
+	modelResult.TotalPrice = AddPrices(inputPrice, outputPrice)
 	modelResult.Currency = "USD"
 	return nil
 }

diff --git a/model/openrouter.go b/model/openrouter.go
@@ -109,7 +109,7 @@ func (p *OpenRouterModelProvider) calculatePrice(modelResult *ModelResult) error
 
 	inputPrice := getPrice(modelResult.PromptTokenCount, inputPricePerThousandTokens)
 	outputPrice := getPrice(modelResult.ResponseTokenCount, outputPricePerThousandTokens)
-	modelResult.TotalPrice = addPrices(inputPrice, outputPrice)
+	modelResult.TotalPrice = AddPrices(inputPrice, outputPrice)
 	modelResult.Currency = "USD"
 	return nil
 }

diff --git a/model/util.go b/model/util.go
@@ -41,7 +41,7 @@ func getPrice(tokenCount int, pricePerThousandTokens float64) float64 {
 	return res
 }
 
-func addPrices(price1 float64, price2 float64) float64 {
+func AddPrices(price1 float64, price2 float64) float64 {
 	res := price1 + price2
 	res = math.Round(res*1e8) / 1e8
 	return res