Browse Source

完全移除近似估算token功能

CaIon 2 years ago
parent
commit
46930bc890
4 changed files with 3 additions and 20 deletions
  1. 1 1
      common/constants.go
  2. 2 9
      controller/relay-utils.go
  3. 0 3
      model/option.go
  4. 0 7
      web/src/components/OperationSetting.js

+ 1 - 1
common/constants.go

@@ -84,7 +84,7 @@ var ChannelDisableThreshold = 5.0
 var AutomaticDisableChannelEnabled = false
 var QuotaRemindThreshold = 1000
 var PreConsumedQuota = 500
-var ApproximateTokenEnabled = false
+
 var RetryTimes = 0
 
 var RootUserEmail = ""

+ 2 - 9
controller/relay-utils.go

@@ -8,7 +8,6 @@ import (
 	"io"
 	"net/http"
 	"one-api/common"
-	"regexp"
 	"strconv"
 	"strings"
 )
@@ -60,9 +59,6 @@ func getTokenEncoder(model string) *tiktoken.Tiktoken {
 }
 
 func getTokenNum(tokenEncoder *tiktoken.Tiktoken, text string) int {
-	if common.ApproximateTokenEnabled {
-		return int(float64(len(text)) * 0.38)
-	}
 	return len(tokenEncoder.Encode(text, nil, nil))
 }
 
@@ -118,11 +114,8 @@ func countTokenText(text string, model string) int {
 func errorWrapper(err error, code string, statusCode int) *OpenAIErrorWithStatusCode {
 	text := err.Error()
 	// 定义一个正则表达式匹配URL
-	urlPattern := `http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\\(\\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+`
-
-	urlRegexp, reErr := regexp.Compile(urlPattern)
-	if reErr == nil {
-		text = urlRegexp.ReplaceAllString(text, "https://api.openai.com")
+	if strings.Contains(text, "Post") {
+		text = "请求上游地址失败"
 	}
 	//避免暴露内部错误
 

+ 0 - 3
model/option.go

@@ -34,7 +34,6 @@ func InitOptionMap() {
 	common.OptionMap["TurnstileCheckEnabled"] = strconv.FormatBool(common.TurnstileCheckEnabled)
 	common.OptionMap["RegisterEnabled"] = strconv.FormatBool(common.RegisterEnabled)
 	common.OptionMap["AutomaticDisableChannelEnabled"] = strconv.FormatBool(common.AutomaticDisableChannelEnabled)
-	common.OptionMap["ApproximateTokenEnabled"] = strconv.FormatBool(common.ApproximateTokenEnabled)
 	common.OptionMap["LogConsumeEnabled"] = strconv.FormatBool(common.LogConsumeEnabled)
 	common.OptionMap["DisplayInCurrencyEnabled"] = strconv.FormatBool(common.DisplayInCurrencyEnabled)
 	common.OptionMap["DisplayTokenStatEnabled"] = strconv.FormatBool(common.DisplayTokenStatEnabled)
@@ -153,8 +152,6 @@ func updateOptionMap(key string, value string) (err error) {
 			common.EmailDomainRestrictionEnabled = boolValue
 		case "AutomaticDisableChannelEnabled":
 			common.AutomaticDisableChannelEnabled = boolValue
-		case "ApproximateTokenEnabled":
-			common.ApproximateTokenEnabled = boolValue
 		case "LogConsumeEnabled":
 			common.LogConsumeEnabled = boolValue
 		case "DisplayInCurrencyEnabled":

+ 0 - 7
web/src/components/OperationSetting.js

@@ -19,7 +19,6 @@ const OperationSetting = () => {
         LogConsumeEnabled: '',
         DisplayInCurrencyEnabled: '',
         DisplayTokenStatEnabled: '',
-        ApproximateTokenEnabled: '',
         RetryTimes: 0
     });
     const [originInputs, setOriginInputs] = useState({});
@@ -201,12 +200,6 @@ const OperationSetting = () => {
                             name='DisplayTokenStatEnabled'
                             onChange={handleInputChange}
                         />
-                        <Form.Checkbox
-                            checked={inputs.ApproximateTokenEnabled === 'true'}
-                            label='使用近似的方式估算 token 数以减少计算量'
-                            name='ApproximateTokenEnabled'
-                            onChange={handleInputChange}
-                        />
                     </Form.Group>
                     <Form.Button onClick={() => {
                         submitConfig('general').then();