From 2db4282666dcd46fdb2ed7543617a65595fa2c3e Mon Sep 17 00:00:00 2001
From: CaIon <1808837298@qq.com>
Date: Wed, 20 Mar 2024 20:15:32 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E4=BF=9D=E7=95=99=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 constant/sensitive.go                  |  3 +++
 model/option.go                        |  3 +++
 web/src/components/OperationSetting.js | 14 ++++++++++++++
 3 files changed, 20 insertions(+)

diff --git a/constant/sensitive.go b/constant/sensitive.go
index 10ecfe6..fabca67 100644
--- a/constant/sensitive.go
+++ b/constant/sensitive.go
@@ -9,6 +9,9 @@ var CheckSensitiveOnCompletionEnabled = true
 // StopOnSensitiveEnabled 如果检测到敏感词，是否立刻停止生成，否则替换敏感词
 var StopOnSensitiveEnabled = true
 
+// StreamCacheQueueLength 流模式缓存队列长度，0表示无缓存
+var StreamCacheQueueLength = 0
+
 // SensitiveWords 敏感词
 // var SensitiveWords []string
 var SensitiveWords = []string{
diff --git a/model/option.go b/model/option.go
index 7422da1..46e41da 100644
--- a/model/option.go
+++ b/model/option.go
@@ -95,6 +95,7 @@ func InitOptionMap() {
 	common.OptionMap["CheckSensitiveOnCompletionEnabled"] = strconv.FormatBool(constant.CheckSensitiveOnCompletionEnabled)
 	common.OptionMap["StopOnSensitiveEnabled"] = strconv.FormatBool(constant.StopOnSensitiveEnabled)
 	common.OptionMap["SensitiveWords"] = constant.SensitiveWordsToString()
+	common.OptionMap["StreamCacheQueueLength"] = strconv.Itoa(constant.StreamCacheQueueLength)
 
 	common.OptionMapRWMutex.Unlock()
 	loadOptionsFromDatabase()
@@ -288,6 +289,8 @@ func updateOptionMap(key string, value string) (err error) {
 		common.QuotaPerUnit, _ = strconv.ParseFloat(value, 64)
 	case "SensitiveWords":
 		constant.SensitiveWordsFromString(value)
+	case "StreamCacheQueueLength":
+		constant.StreamCacheQueueLength, _ = strconv.Atoi(value)
 	}
 	return err
 }
diff --git a/web/src/components/OperationSetting.js b/web/src/components/OperationSetting.js
index c006106..728eab0 100644
--- a/web/src/components/OperationSetting.js
+++ b/web/src/components/OperationSetting.js
@@ -10,6 +10,7 @@ const OperationSetting = () => {
     QuotaForInvitee: 0,
     QuotaRemindThreshold: 0,
     PreConsumedQuota: 0,
+    StreamCacheQueueLength: 0,
     ModelRatio: '',
     ModelPrice: '',
     GroupRatio: '',
@@ -307,6 +308,8 @@ const OperationSetting = () => {
               name="CheckSensitiveOnCompletionEnabled"
               onChange={handleInputChange}
             />
+          </Form.Group>
+          <Form.Group inline>
             <Form.Checkbox
               checked={inputs.StopOnSensitiveEnabled === 'true'}
               label="在检测到屏蔽词时，立刻停止生成，否则替换屏蔽词"
@@ -314,6 +317,17 @@ const OperationSetting = () => {
               onChange={handleInputChange}
             />
           </Form.Group>
+          {/*<Form.Group>*/}
+          {/*  <Form.Input*/}
+          {/*    label="流模式下缓存队列，默认不缓存，设置越大检测越准确，但是回复会有卡顿感"*/}
+          {/*    name="StreamCacheTextLength"*/}
+          {/*    onChange={handleInputChange}*/}
+          {/*    value={inputs.StreamCacheQueueLength}*/}
+          {/*    type="number"*/}
+          {/*    min="0"*/}
+          {/*    placeholder="例如：10"*/}
+          {/*  />*/}
+          {/*</Form.Group>*/}
           <Form.Group widths="equal">
             <Form.TextArea
               label="屏蔽词列表，一行一个屏蔽词，不需要符号分割"