Say which memories the model can actually see

Retrieval has been path-scoped since SP3: a memory on a branch this story
never travelled is never sent to the AI. The drawer listed every memory
alike, so on a fork you read "Fell down the cellar stairs; badly hurt" and
reasonably concluded the model knew it. It does not. That is worse than
either hiding the row or retrieving it — it is the screen claiming
something the engine contradicts.

Hiding them is not the answer either; that was the original comment's
point, and it stands. A memory nobody can list is a memory nobody can
delete, in a phase whose rule is that nothing is removed automatically.

So the whole bank still lists, and the rows off the current path are set
back, dashed, and labelled "another branch". MemoryOut.on_path carries it,
computed from the predicate retrieval itself uses rather than a second
spelling of the same idea — two spellings drift, and the failure mode here
is a badge that says the opposite of what the model gets. Pinning does not
override it: the path clause runs before pinning is considered.

The relationship is asymmetric and there is now a test that says so. A fork
borrows its ancestors, so a memory written on the parent is on the fork's
path too; the reverse never is. Worth pinning before somebody makes it
symmetric on the grounds that it looks wrong.

399 tests, three new, egress ceilings intact — the flag costs one id-only
query. Verified in a browser on tools/branch_fixture.py.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_015H5qiyiR7gtFQaoDphHZ3g
This commit is contained in:
parththakkar106
2026-08-18 19:14:07 +05:30
committed by Parth
co-authored by Claude Opus 5
parent 811368d048
commit 1d1367ce3a
7 changed files with 171 additions and 9 deletions
+22 -3
View File
@@ -4,7 +4,7 @@ import threading
from fastapi import APIRouter, Body, Depends, HTTPException, Request
from fastapi.responses import StreamingResponse
from sqlalchemy import func
from sqlalchemy import func, select
from sqlalchemy.orm import Session, load_only, undefer
from sqlalchemy.orm.attributes import set_committed_value
@@ -1874,7 +1874,7 @@ def action_context(
def list_memories(
adventure_id: int, db: Session = Depends(get_db), user: models.User = CurrentUser
):
get_adventure_or_404(adventure_id, db, user)
adventure = get_adventure_or_404(adventure_id, db, user)
# A query naming its columns, not a walk of `adventure.memories`. The walk
# is what retrieval used to do, and it is the reason a turn cost megabytes:
# a relationship load takes whole entities, so it picks up whatever the
@@ -1885,13 +1885,32 @@ def list_memories(
# the story being played, the drawer manages the bank. Hiding a branch's
# memories from the drawer would mean memories nobody can find to delete,
# in a phase whose rule is that nothing is ever removed automatically.
return (
rows = (
db.query(models.Memory)
.options(load_only(*MEMORY_LIST_COLUMNS))
.filter(models.Memory.adventure_id == adventure_id)
.order_by(models.Memory.id)
.all()
)
# ...but listing them alike would be its own lie. A memory on a branch this
# story never travelled is never retrieved, so showing it beside one that is
# tells the player the model remembers something it cannot see. The ids come
# from **the predicate retrieval itself uses**, deliberately: two spellings
# of "on this path" would eventually disagree, and the failure would be a
# badge that says the opposite of what the model gets.
on_path = {
row[0] for row in db.execute(
select(models.Memory.id).where(
models.Memory.adventure_id == adventure_id,
lineage.path_of(db, adventure).clause(
models.Memory, unanchored=True
),
)
)
}
for row in rows:
row.on_path = row.id in on_path
return rows
@router.post("/{adventure_id}/memories", response_model=schemas.MemoryOut, status_code=201)
+6
View File
@@ -289,6 +289,12 @@ class MemoryOut(ORMModel):
last_used_at: datetime | None
source_start: int | None
source_end: int | None
# Whether this memory is on the story currently being read (Phase 14, SP7).
# The drawer lists the whole bank so nothing becomes impossible to find and
# delete, but a memory belonging to another branch will never be retrieved
# into context — and a list that showed the two alike would be telling the
# player the model knows something it cannot see.
on_path: bool = True
created_at: datetime
+84 -1
View File
@@ -31,9 +31,10 @@ os.environ.pop("DATABASE_URL", None)
import pytest
from fastapi import Depends
from fastapi.testclient import TestClient
from sqlalchemy import select
from app import auth, limits, models, schemas
from app.context import cursors
from app.context import cursors, lineage
from app.database import Base, SessionLocal, engine, get_db
from app.main import app
from app.providers import PromptParts
@@ -346,6 +347,88 @@ def test_deleting_an_unknown_branch_is_a_404(client):
assert _delete(client, forked + 9999).status_code == 404
# --------------------------------------------------------- the bank vs a path
def _memories(client) -> list[dict]:
r = client.get(f"/api/adventures/{client.adv_id}/memories")
assert r.status_code == 200, r.text
return r.json()
def _add_memory(client, text):
r = client.post(f"/api/adventures/{client.adv_id}/memories", json={"text": text})
assert r.status_code == 201, r.text
return r.json()["id"]
def test_the_drawer_keeps_every_memory_but_says_which_are_off_the_path(client):
"""Both halves of the split, in one test, because either alone is a bug.
Listing only the path's memories would leave the rest impossible to find
and delete, in a phase whose rule is that nothing is removed automatically.
Listing them all *alike* would tell the player the model remembers
something that is never retrieved on this branch.
"""
root, forked = _forked(client)
on_the_fork = _add_memory(client, "Took the other door.")
_switch(client, root)
on_the_root = _add_memory(client, "Went the long way instead.")
listed = {m["id"]: m for m in _memories(client)}
assert set(listed) == {on_the_fork, on_the_root}, "the whole bank, always"
assert listed[on_the_root]["on_path"] is True
assert listed[on_the_fork]["on_path"] is False, "written on a branch we left"
# And it follows the reader rather than being a property of the memory —
# but **asymmetrically**, which is the part worth pinning. A fork borrows
# its parent's story, so a memory written on the parent is on the fork's
# path too. The reverse is not true: the parent never went down the fork.
_switch(client, forked)
listed = {m["id"]: m for m in _memories(client)}
assert listed[on_the_fork]["on_path"] is True
assert listed[on_the_root]["on_path"] is True, "an ancestor's memory is shared"
def test_the_off_path_flag_agrees_with_what_retrieval_can_see(client):
"""The flag has to come from retrieval's own predicate, not a second
spelling of it. Two spellings would drift, and the failure mode is a badge
claiming the opposite of what the model is actually given."""
root, forked = _forked(client)
_add_memory(client, "Took the other door.")
_switch(client, root)
_add_memory(client, "Went the long way instead.")
db = SessionLocal()
try:
adventure = db.get(models.Adventure, client.adv_id)
visible = {
row[0] for row in db.execute(
select(models.Memory.id).where(
models.Memory.adventure_id == adventure.id,
lineage.path_of(db, adventure).clause(
models.Memory, unanchored=True
),
)
)
}
finally:
db.close()
flagged = {m["id"] for m in _memories(client) if m["on_path"]}
assert flagged == visible
def test_a_memory_from_a_deleted_branch_is_gone_from_the_drawer(client):
"""Not merely off-path — the row goes with the branch, through the cascade."""
root, forked = _forked(client)
doomed = _add_memory(client, "Took the other door.")
_switch(client, root)
assert doomed in {m["id"] for m in _memories(client)}
assert _delete(client, forked).status_code == 204
assert doomed not in {m["id"] for m in _memories(client)}
# ------------------------------------------------------------------- backup
def test_a_bundle_carries_the_name_a_player_chose(client):