nesquena-hermes
6951d10ae5
franksong2702's activity_scene_v1 anchor-scene model: live-stream/settled/refresh/re-entry show the same ordered turn. Rebased clean (no conflicts) onto current master. Co-authored-by: franksong2702 <franksong2702@users.noreply.github.com>
99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
from types import SimpleNamespace
|
|
|
|
import api.routes as routes
|
|
|
|
|
|
def _msg(role, content, ts):
|
|
return {"role": role, "content": content, "timestamp": ts}
|
|
|
|
|
|
def test_webui_lineage_display_merge_includes_parent_only_rows(monkeypatch):
|
|
parent = SimpleNamespace(
|
|
session_id="parent",
|
|
messages=[
|
|
_msg("user", "first prompt", 1),
|
|
_msg("assistant", "first answer", 2),
|
|
_msg("user", "parent only prompt", 3),
|
|
_msg("assistant", "parent only answer", 4),
|
|
],
|
|
)
|
|
tip = SimpleNamespace(
|
|
session_id="tip",
|
|
parent_session_id="parent",
|
|
session_source="webui",
|
|
messages=[
|
|
_msg("user", "first prompt", 1),
|
|
_msg("assistant", "first answer", 2),
|
|
_msg("user", "tip prompt", 5),
|
|
_msg("assistant", "tip answer", 6),
|
|
],
|
|
truncation_watermark=None,
|
|
)
|
|
|
|
monkeypatch.setattr(routes, "get_session", lambda sid, metadata_only=False: parent)
|
|
|
|
merged = routes._merged_webui_lineage_messages_for_display(tip, tip.messages)
|
|
|
|
assert [m["content"] for m in merged] == [
|
|
"first prompt",
|
|
"first answer",
|
|
"parent only prompt",
|
|
"parent only answer",
|
|
"tip prompt",
|
|
"tip answer",
|
|
]
|
|
|
|
|
|
def test_webui_lineage_display_keeps_cumulative_child_tail_without_timestamps(monkeypatch):
|
|
parent = SimpleNamespace(
|
|
session_id="parent",
|
|
messages=[
|
|
_msg("user", "first prompt", 1000.0),
|
|
_msg("assistant", "first answer", 1001.0),
|
|
],
|
|
)
|
|
tip = SimpleNamespace(
|
|
session_id="tip",
|
|
parent_session_id="parent",
|
|
session_source="webui",
|
|
messages=[
|
|
_msg("user", "first prompt", 1000.0),
|
|
_msg("assistant", "first answer", 1001.0),
|
|
{"role": "user", "content": "continue after compression"},
|
|
{"role": "assistant", "content": "final after compression"},
|
|
],
|
|
truncation_watermark=None,
|
|
)
|
|
|
|
monkeypatch.setattr(routes, "get_session", lambda sid, metadata_only=False: parent)
|
|
|
|
merged = routes._merged_webui_lineage_messages_for_display(tip, tip.messages)
|
|
|
|
assert [m["content"] for m in merged] == [
|
|
"first prompt",
|
|
"first answer",
|
|
"continue after compression",
|
|
"final after compression",
|
|
]
|
|
assert merged[-1]["role"] == "assistant"
|
|
|
|
|
|
def test_webui_lineage_display_merge_skips_explicit_forks(monkeypatch):
|
|
parent = SimpleNamespace(
|
|
session_id="parent",
|
|
messages=[_msg("user", "parent-only", 1)],
|
|
)
|
|
fork = SimpleNamespace(
|
|
session_id="fork",
|
|
parent_session_id="parent",
|
|
session_source="fork",
|
|
messages=[_msg("user", "fork starts here", 2)],
|
|
truncation_watermark=None,
|
|
)
|
|
|
|
monkeypatch.setattr(routes, "get_session", lambda sid, metadata_only=False: parent)
|
|
|
|
merged = routes._merged_webui_lineage_messages_for_display(fork, fork.messages)
|
|
|
|
assert [m["content"] for m in merged] == ["fork starts here"]
|