주제가 있는 대화 기능 추가 완료

npc_social_network/npc/npc_manager.py

@@ -52,45 +52,56 @@ class NPCManager:
         """관리 중인 모든 NPC의 리스트를 반환합니다."""
         return self.npcs
     
-    def initiate_npc_to_npc_interaction(self, time_context: str):
+    def initiate_npc_to_npc_interaction(self, initiator: 'NPC', target: 'NPC',
+                                        time_context: str, topic: Optional[str] = None):
         """
         NPC 간의 상호작용을 시작시키는 함수
         - 참여한 NPC를 반환해서 소문을 퍼트릴 '목격자' 생성
+        - 주제가 있을 경우, 목표 지향적 대화를 생성
         """
         from ..models.llm_helper import query_llm_with_prompt
 
         if len(self.npcs) < 2:
             return None, None # 상호작용할 NPC가 최소 2명 필요
         
-        # 상호작용을 시작할 NPC (initiator)와 대상 NPC (target)를 랜덤으로 선택
-        initiator = self.get_random_npc()
-        target = self.get_random_npc(exclude=initiator)
-
-        if not initiator or not target:
-            return
-        
         initiator_postposition = get_korean_postposition(initiator.korean_name, "이", "가")
         target_postposition = get_korean_postposition(target.korean_name, "에게", "에게")
         
         print(f"\n---[NPC 상호작용 이벤트]---\n{initiator.korean_name}{initiator_postposition} {target.korean_name}{target_postposition} 상호작용을 시도합니다.")
-       
-        # 1. Initiator가 Target에 대해 무슨 말을 할지 LLM으로 생성
-        prompt = f"""
-        # 지시사항
-        나는 '{initiator.korean_name}'입니다. 지금 나는 '{target.korean_name}'와 마주쳤습니다.
-        '{target.korean_name}'에 대한 당신의 현재 생각과 감정을 바탕으로, 먼저 건넬 자연스러운 첫 마디를 생성하세요.
-        (예: "앨리스, 어제 시장에서 파는 그 옷 정말 예쁘더라.", "밥, 요즘 허리가 아프다던데 괜찮아?")
+        
+        if topic:
+            # 주제가 있는 경우
+            prompt = f"""
+# 지시사항
+나는 '{initiator.korean_name}'입니다.
+지금 나는 '{target.korean_name}'와 마주쳤습니다.
+나는 상대에게 '{topic}'에 대해 이야기하고 싶습니다.
+이 주제로 대화를 시작할 자연스러운 첫 마디를 생성하세요.
 
-        # 나의 정보
-        - '{target.korean_name}'와의 관계: {initiator.relationships.get_relationship_summary(target.name)}
+# 당신의 정보
+- '{target.korean_name}'와의 관계: {initiator.relationships.get_relationship_summary(target.name)}
 
-        → 첫 마디:
-        """
+→ 첫 마디:
+"""
+        else:
+            # 주제가 없는 경우 (기존의 가벼운 인사)
+            prompt = f"""
+# 지시사항
+나는 '{initiator.korean_name}'입니다.
+지금 나는 '{target.korean_name}'와 마주쳤습니다.
+'{target.korean_name}'에 대한 당신의 현재 생각과 감정을 바탕으로, 먼저 건넬 자연스러운 첫 마디를 생성하세요.
+(예: "앨리스, 어제 시장에서 파는 그 옷 정말 예쁘더라.", "밥, 요즘 허리가 아프다던데 괜찮아?")
+
+# 나의 정보
+- '{target.korean_name}'와의 관계: {initiator.relationships.get_relationship_summary(target.name)}
+
+→ 첫 마디:
+"""
         initial_utterance = query_llm_with_prompt(prompt).replace("'", "").strip()
 
         if "[LLM Error]" in initial_utterance:
             print(f"[{initiator.korean_name}] 대화 시작에 실패했습니다.")
-            return None, None
+            return None, None, None
 
         print(f"[{initiator.korean_name}]: {initial_utterance}")