Prechádzať zdrojové kódy

Update response_type_detector: add random_rate

StrayWarrior 3 dní pred
rodič
commit
48b5a5ee7b
1 zmenil súbory, kde vykonal 3 pridanie a 2 odobranie
  1. 3 2
      pqai_agent/response_type_detector.py

+ 3 - 2
pqai_agent/response_type_detector.py

@@ -38,7 +38,8 @@ class ResponseTypeDetector:
         )
         self.model_name = chat_service.VOLCENGINE_MODEL_DOUBAO_PRO_1_5
 
-    def detect_type(self, dialogue_history: List[Dict], next_message: Dict, enable_random=False):
+    def detect_type(self, dialogue_history: List[Dict], next_message: Dict, enable_random=False,
+                    random_rate=0.25):
         if configs.get().get('debug_flags', {}).get('disable_llm_api_call', False):
             return MessageType.TEXT
         composed_dialogue = self.compose_dialogue(dialogue_history)
@@ -62,7 +63,7 @@ class ResponseTypeDetector:
             suitable_for_voice = self.if_message_suitable_for_voice(next_message_content)
             logger.debug(f"voice suitable[{suitable_for_voice}], message: {next_message_content}")
             if suitable_for_voice:
-                if random.random() < 0.6:
+                if random.random() < random_rate:
                     logger.info(f"enable voice response randomly for message: {next_message_content}")
                     return MessageType.VOICE
         return MessageType.TEXT