refactor: engine.py

interaction.py

@@ -0,0 +1,89 @@
+import logging
+
+from langchain_core.messages import HumanMessage, SystemMessage
+
+from entities import Entity
+from llm_runtime import _format_prompt, _normalize_llm_output, llm
+from time_utils import WorldClock, describe_relative_time
+
+logger = logging.getLogger(__name__)
+
+
+def ask_entity(
+    entity: Entity,
+    player: Entity,
+    player_query: str,
+    world_clock: WorldClock,
+    location: str,
+):
+    facts = entity.memory.retrieve(
+        player_query,
+        reference_time=world_clock.current_time,
+    )
+
+    recent_context = "\n".join(
+        [f"{m['role_name']}: {m['content']}" for m in entity.chat_buffer[-5:]]
+    )
+
+    world_time_label = describe_relative_time(
+        world_clock.get_time_str(),
+        world_clock.current_time,
+        prefer_day_part_for_today=True,
+    )
+
+    prompt = [
+        SystemMessage(content=f"WORLD TIME: {world_time_label}"),
+        SystemMessage(
+            content=f"""
+### ROLE
+You are {entity.name}. Persona: {", ".join(entity.traits)}.
+Current Mood: {entity.current_mood}.
+Vibe Time: {world_clock.get_vibe()}.
+Location: {location}.
+
+### WRITING STYLE RULES
+1. NO META-TALK. Never mention "memory," "records," "claims," or "narratives." 
+2. ACT, DON'T EXPLAIN. If you don't know something, just say "Never heard of it" or "I wasn't there." Do not explain WHY you don't know.
+
+### KNOWLEDGE
+MEMORIES: {facts}
+RECENT CHAT: {recent_context} 
+        """
+        ),
+        HumanMessage(content=f"{player.name} speaks to you: {player_query}"),
+    ]
+
+    logger.info("LLM prompt (dialogue):\n%s", _format_prompt(prompt))
+    response = _normalize_llm_output(llm.invoke(prompt).content)
+
+    entity.chat_buffer.append(
+        {
+            "role_id": player.entity_id,
+            "role_name": player.name,
+            "content": player_query,
+        }
+    )
+    entity.chat_buffer.append(
+        {
+            "role_id": entity.entity_id,
+            "role_name": entity.name,
+            "content": response,
+        }
+    )
+
+    player.chat_buffer.append(
+        {
+            "role_id": player.entity_id,
+            "role_name": player.name,
+            "content": player_query,
+        }
+    )
+    player.chat_buffer.append(
+        {
+            "role_id": entity.entity_id,
+            "role_name": entity.name,
+            "content": response,
+        }
+    )
+
+    logger.info("[%s]: %s", entity.name.upper(), response)