Add a hard trigger for world-state summarization
update_world_state relied entirely on the DM remembering to call it — no code checked whether it actually happened. Add refresh_if_due(), called at the top of every DM turn: it counts messages since the summary was last touched (by the tool or a prior auto-refresh), and once AUTO_REFRESH_MESSAGE_THRESHOLD (10) is crossed, forces a dedicated summarization call — a separate, tool-free completion whose only job is to fold the new messages into the existing summary — and persists the result directly, without waiting on the main DM turn's discretion. The trigger condition (message count) is deterministic; only the summary text itself still needs an LLM, which is unavoidable for a task that requires understanding, not just counting. context.py gained count_messages_since() and build_transcript_text() (an uncapped, since-filtered plain-text transcript for the summarizer, as opposed to build_context()'s char-budget-capped chat list for the main DM call) — factored out of the same underlying message-labeling logic to avoid duplicating the name-resolution joins. Verified with a fake LLM client: confirmed the trigger fires exactly at the threshold and not before, resets after firing, and that the second refresh's prompt carries the prior summary forward while only including messages since that refresh (not the whole history again). Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy import func, select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.models.character import Character
|
||||
@@ -13,12 +14,19 @@ from app.models.user import User
|
||||
MAX_CONTEXT_CHARS = 16_000
|
||||
|
||||
|
||||
async def build_context(session: AsyncSession, game_id: uuid.UUID) -> list[dict]:
|
||||
messages = (
|
||||
await session.execute(
|
||||
select(Message).where(Message.game_id == game_id).order_by(Message.id.asc())
|
||||
)
|
||||
).scalars().all()
|
||||
async def _load_messages(
|
||||
session: AsyncSession, game_id: uuid.UUID, since: datetime | None = None
|
||||
) -> list[Message]:
|
||||
query = select(Message).where(Message.game_id == game_id)
|
||||
if since is not None:
|
||||
query = query.where(Message.created_at > since)
|
||||
query = query.order_by(Message.id.asc())
|
||||
return list((await session.execute(query)).scalars().all())
|
||||
|
||||
|
||||
async def _label_messages(session: AsyncSession, messages: list[Message]) -> list[dict]:
|
||||
"""Resolves player/character names and renders each message as a (role, content) entry —
|
||||
the shared groundwork for both the chat-style context and the plain-text transcript."""
|
||||
if not messages:
|
||||
return []
|
||||
|
||||
@@ -47,6 +55,11 @@ async def build_context(session: AsyncSession, game_id: uuid.UUID) -> list[dict]
|
||||
label = f"{player_name} ({character_name})" if character_name else player_name
|
||||
entries.append({"role": "user", "content": f"{label}: {m.content}"})
|
||||
# sender_type == "system" messages are not sent to the model in Phase 1
|
||||
return entries
|
||||
|
||||
|
||||
async def build_context(session: AsyncSession, game_id: uuid.UUID) -> list[dict]:
|
||||
entries = await _label_messages(session, await _load_messages(session, game_id))
|
||||
|
||||
# Keep the newest entries within the char budget, dropping oldest first.
|
||||
total = 0
|
||||
@@ -59,3 +72,20 @@ async def build_context(session: AsyncSession, game_id: uuid.UUID) -> list[dict]
|
||||
kept.reverse()
|
||||
|
||||
return kept
|
||||
|
||||
|
||||
async def count_messages_since(session: AsyncSession, game_id: uuid.UUID, since: datetime | None) -> int:
|
||||
query = select(func.count()).select_from(Message).where(Message.game_id == game_id)
|
||||
if since is not None:
|
||||
query = query.where(Message.created_at > since)
|
||||
return (await session.execute(query)).scalar_one()
|
||||
|
||||
|
||||
async def build_transcript_text(
|
||||
session: AsyncSession, game_id: uuid.UUID, since: datetime | None = None
|
||||
) -> str:
|
||||
"""Full, uncapped plain-text transcript (optionally only messages after `since`) — used to
|
||||
feed the world-state summarizer, which needs everything new, not a char-budget window."""
|
||||
entries = await _label_messages(session, await _load_messages(session, game_id, since=since))
|
||||
lines = [f"DM: {e['content']}" if e["role"] == "assistant" else e["content"] for e in entries]
|
||||
return "\n\n".join(lines)
|
||||
|
||||
Reference in New Issue
Block a user