From 5043075135f0a9ca06c44d82af6103b3f3eacf6f Mon Sep 17 00:00:00 2001
From: creamlike1024 <creamlike1024@outlook.com>
Date: Tue, 6 May 2025 23:57:22 +0800
Subject: [PATCH] chore: move file search tool price to operation_setting

---
 relay/relay-text.go                | 4 +---
 setting/operation_setting/tools.go | 7 +++++++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/relay/relay-text.go b/relay/relay-text.go
index 104141ec..d2b2c208 100644
--- a/relay/relay-text.go
+++ b/relay/relay-text.go
@@ -376,9 +376,7 @@ func postConsumeQuota(ctx *gin.Context, relayInfo *relaycommon.RelayInfo,
 	var dFileSearchQuota decimal.Decimal
 	if relayInfo.ResponsesUsageInfo != nil {
 		if fileSearchTool, exists := relayInfo.ResponsesUsageInfo.BuiltInTools[dto.BuildInToolFileSearch]; exists && fileSearchTool.CallCount > 0 {
-			// file search tool 调用价格 $2.50/1k calls
-			// 计算 file search tool 调用的配额 (配额 = 价格 * 调用次数 / 1000)
-			dFileSearchQuota = decimal.NewFromFloat(2.5).
+			dFileSearchQuota = decimal.NewFromFloat(operation_setting.GetFileSearchPricePerThousand()).
 				Mul(decimal.NewFromInt(int64(fileSearchTool.CallCount))).
 				Div(decimal.NewFromInt(1000))
 			extraContent += fmt.Sprintf("File Search 调用 %d 次，调用花费 $%s",
diff --git a/setting/operation_setting/tools.go b/setting/operation_setting/tools.go
index 5c560cb6..974c4ed2 100644
--- a/setting/operation_setting/tools.go
+++ b/setting/operation_setting/tools.go
@@ -3,12 +3,15 @@ package operation_setting
 import "strings"
 
 const (
+	// Web search
 	WebSearchHighTierModelPriceLow    = 30.00
 	WebSearchHighTierModelPriceMedium = 35.00
 	WebSearchHighTierModelPriceHigh   = 50.00
 	WebSearchPriceLow                 = 25.00
 	WebSearchPriceMedium              = 27.50
 	WebSearchPriceHigh                = 30.00
+	// File search
+	FileSearchPrice = 2.5
 )
 
 func GetWebSearchPricePerThousand(modelName string, contextSize string) float64 {
@@ -48,3 +51,7 @@ func GetWebSearchPricePerThousand(modelName string, contextSize string) float64
 	}
 	return priceWebSearchPerThousandCalls
 }
+
+func GetFileSearchPricePerThousand() float64 {
+	return FileSearchPrice
+}