feat: add trigger probability setting for TTS and support to render slider in schema (#4047)

Soulter · web-flow · commit 02cd5e396b32 · 2025-12-15T16:04:27.000+08:00
* feat: add trigger probability setting for TTS and support to render slider in schema

* chore: ruff format
diff --git a/astrbot/core/config/default.py b/astrbot/core/config/default.py
@@ -108,6 +108,7 @@
         "provider_id": "",
         "dual_output": False,
         "use_file_service": False,
+        "trigger_probability": 1.0,
     },
     "provider_ltm_settings": {
         "group_icl_enable": False,
@@ -2209,6 +2210,9 @@
                     "use_file_service": {
                         "type": "bool",
                     },
+                    "trigger_probability": {
+                        "type": "float",
+                    },
                 },
             },
             "provider_ltm_settings": {
@@ -2419,6 +2423,14 @@
                             "provider_tts_settings.enable": True,
                         },
                     },
+                    "provider_tts_settings.trigger_probability": {
+                        "description": "TTS 触发概率",
+                        "type": "float",
+                        "slider": {"min": 0, "max": 1, "step": 0.05},
+                        "condition": {
+                            "provider_tts_settings.enable": True,
+                        },
+                    },
                     "provider_settings.image_caption_prompt": {
                         "description": "图片转述提示词",
                         "type": "text",
@@ -2986,6 +2998,7 @@
                         "description": "回复概率",
                         "type": "float",
                         "hint": "0.0-1.0 之间的数值",
+                        "slider": {"min": 0, "max": 1, "step": 0.05},
                         "condition": {
                             "provider_ltm_settings.active_reply.enable": True,
                         },
diff --git a/astrbot/core/config/i18n_utils.py b/astrbot/core/config/i18n_utils.py
@@ -79,6 +79,7 @@ def convert_to_i18n_keys(metadata: dict[str, Any]) -> dict[str, Any]:
                             "_special",
                             "invisible",
                             "options",
+                            "slider",
                         ]:
                             if attr in field_data:
                                 field_result[attr] = field_data[attr]
diff --git a/astrbot/core/pipeline/result_decorate/stage.py b/astrbot/core/pipeline/result_decorate/stage.py
@@ -1,3 +1,4 @@
+import random
 import re
 import time
 import traceback
@@ -42,6 +43,18 @@ async def initialize(self, ctx: PipelineContext):
             "forward_threshold"
         ]
 
+        trigger_probability = ctx.astrbot_config["provider_tts_settings"].get(
+            "trigger_probability",
+            1,
+        )
+        try:
+            self.tts_trigger_probability = max(
+                0.0,
+                min(float(trigger_probability), 1.0),
+            )
+        except (TypeError, ValueError):
+            self.tts_trigger_probability = 1.0
+
         # 分段回复
         self.words_count_threshold = int(
             ctx.astrbot_config["platform_settings"]["segmented_reply"][
@@ -246,7 +259,14 @@ async def process(
                 and result.is_llm_result()
                 and SessionServiceManager.should_process_tts_request(event)
             ):
-                if not tts_provider:
+                should_tts = self.tts_trigger_probability >= 1.0 or (
+                    self.tts_trigger_probability > 0.0
+                    and random.random() <= self.tts_trigger_probability
+                )
+
+                if not should_tts:
+                    logger.debug("跳过 TTS：触发概率未命中。")
+                elif not tts_provider:
                     logger.warning(
                         f"会话 {event.unified_msg_origin} 未配置文本转语音模型。",
                     )
diff --git a/dashboard/src/components/shared/AstrBotConfig.vue b/dashboard/src/components/shared/AstrBotConfig.vue
@@ -304,16 +304,32 @@ function hasVisibleItemsAfter(items, currentIndex) {
                   hide-details
                 ></v-text-field>
 
-                <!-- Numeric input -->
-                <v-text-field
+                <!-- Numeric input with optional slider -->
+                <div
                   v-else-if="(metadata[metadataKey].items[key]?.type === 'int' || metadata[metadataKey].items[key]?.type === 'float') && !metadata[metadataKey]?.invisible"
-                  v-model="iterable[key]"
-                  density="compact"
-                  variant="outlined"
-                  class="config-field"
-                  type="number"
-                  hide-details
-                ></v-text-field>
+                  class="d-flex align-center gap-3"
+                >
+                  <v-slider
+                    v-if="metadata[metadataKey].items[key]?.slider"
+                    v-model.number="iterable[key]"
+                    :min="metadata[metadataKey].items[key]?.slider?.min ?? 0"
+                    :max="metadata[metadataKey].items[key]?.slider?.max ?? 100"
+                    :step="metadata[metadataKey].items[key]?.slider?.step ?? 1"
+                    color="primary"
+                    density="compact"
+                    hide-details
+                    class="flex-grow-1"
+                  ></v-slider>
+                  <v-text-field
+                    v-model.number="iterable[key]"
+                    density="compact"
+                    variant="outlined"
+                    class="config-field"
+                    type="number"
+                    hide-details
+                    style="max-width: 140px;"
+                  ></v-text-field>
+                </div>
 
                 <!-- Text area -->
                 <v-textarea
@@ -413,16 +429,32 @@ function hasVisibleItemsAfter(items, currentIndex) {
               hide-details
             ></v-text-field>
 
-            <!-- Numeric input -->
-            <v-text-field
+            <!-- Numeric input with optional slider -->
+            <div
               v-else-if="(metadata[metadataKey]?.type === 'int' || metadata[metadataKey]?.type === 'float') && !metadata[metadataKey]?.invisible"
-              v-model="iterable[metadataKey]"
-              density="compact"
-              variant="outlined"
-              class="config-field"
-              type="number"
-              hide-details
-            ></v-text-field>
+              class="d-flex align-center gap-3"
+            >
+              <v-slider
+                v-if="metadata[metadataKey]?.slider"
+                v-model.number="iterable[metadataKey]"
+                :min="metadata[metadataKey]?.slider?.min ?? 0"
+                :max="metadata[metadataKey]?.slider?.max ?? 100"
+                :step="metadata[metadataKey]?.slider?.step ?? 1"
+                color="primary"
+                density="compact"
+                hide-details
+                class="flex-grow-1"
+              ></v-slider>
+              <v-text-field
+                v-model.number="iterable[metadataKey]"
+                density="compact"
+                variant="outlined"
+                class="config-field"
+                type="number"
+                hide-details
+                style="max-width: 140px;"
+              ></v-text-field>
+            </div>
 
             <!-- Text area -->
             <v-textarea
diff --git a/dashboard/src/components/shared/AstrBotConfigV4.vue b/dashboard/src/components/shared/AstrBotConfigV4.vue
@@ -245,10 +245,29 @@ function getSpecialSubtype(value) {
                 <v-text-field v-else-if="itemMeta?.type === 'string'" v-model="createSelectorModel(itemKey).value"
                   density="compact" variant="outlined" class="config-field" hide-details></v-text-field>
 
-                <!-- Numeric input for JSON selector -->
-                <v-text-field v-else-if="itemMeta?.type === 'int' || itemMeta?.type === 'float'"
-                  v-model="createSelectorModel(itemKey).value" density="compact" variant="outlined" class="config-field"
-                  type="number" hide-details></v-text-field>
+                <!-- Numeric input with optional slider for JSON selector -->
+                <div v-else-if="itemMeta?.type === 'int' || itemMeta?.type === 'float'" class="d-flex align-center gap-3">
+                  <v-slider
+                    v-if="itemMeta?.slider"
+                    v-model.number="createSelectorModel(itemKey).value"
+                    :min="itemMeta?.slider?.min ?? 0"
+                    :max="itemMeta?.slider?.max ?? 100"
+                    :step="itemMeta?.slider?.step ?? 1"
+                    color="primary"
+                    density="compact"
+                    hide-details
+                    style="flex: 3"
+                  ></v-slider>
+                  <v-text-field
+                    v-model.number="createSelectorModel(itemKey).value"
+                    density="compact"
+                    variant="outlined"
+                    class="config-field"
+                    style="flex: 2"
+                    type="number"
+                    hide-details
+                  ></v-text-field>
+                </div>
 
                 <!-- Text area for JSON selector -->
                 <v-textarea v-else-if="itemMeta?.type === 'text'" v-model="createSelectorModel(itemKey).value"
diff --git a/dashboard/src/i18n/locales/en-US/features/config-metadata.json b/dashboard/src/i18n/locales/en-US/features/config-metadata.json
@@ -57,6 +57,9 @@
         },
         "provider_id": {
           "description": "Default Text-to-Speech Model"
+        },
+        "trigger_probability": {
+          "description": "TTS Trigger Probability"
         }
       }
     },
diff --git a/dashboard/src/i18n/locales/zh-CN/features/config-metadata.json b/dashboard/src/i18n/locales/zh-CN/features/config-metadata.json
@@ -62,6 +62,9 @@
         },
         "provider_id": {
           "description": "默认文本转语音模型"
+        },
+        "trigger_probability": {
+          "description": "TTS 触发概率"
         }
       }
     },

Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,9 @@`
`57`	`57`	`},`
`58`	`58`	`"provider_id": {`
`59`	`59`	`"description": "Default Text-to-Speech Model"`
	`60`	`+ },`
	`61`	`+ "trigger_probability": {`
	`62`	`+ "description": "TTS Trigger Probability"`
`60`	`63`	`}`
`61`	`64`	`}`
`62`	`65`	`},`
Original file line number	Diff line number	Diff line change
`@@ -62,6 +62,9 @@`
`62`	`62`	`},`
`63`	`63`	`"provider_id": {`
`64`	`64`	`"description": "默认文本转语音模型"`
	`65`	`+ },`
	`66`	`+ "trigger_probability": {`
	`67`	`+ "description": "TTS 触发概率"`
`65`	`68`	`}`
`66`	`69`	`}`
`67`	`70`	`},`