diff --git a/AITrain/dual_ai_dialogue_system.py b/AITrain/dual_ai_dialogue_system.py index 959f069..f156e05 100644 --- 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(