调整对话上下轮参数控制

2025-08-15 16:56:34 +08:00 · 2025-08-15 16:56:34 +08:00 · 5a8c8ff7f2
commit 5a8c8ff7f2
parent 773d8fb9cf
1 changed files with 1 additions and 1 deletions
--- a/AITrain/dual_ai_dialogue_system.py
+++ b/AITrain/dual_ai_dialogue_system.py
@ -457,7 +457,7 @@ class DualAIDialogueEngine:
            search_query = f"{current_speaker} {topic_hint} introduction greeting"
        
        # 搜索相关上下文
-        context_info = self.kb.search_relevant_context(search_query, current_speaker, max(10, context_info_count * 2))
+        context_info = self.kb.search_relevant_context(search_query, current_speaker, context_info_count)
        
        # 生成提示（使用参数控制上下文数量）
        prompt = self.generate_character_prompt(