68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from relay.conversation import RECENT_TURNS_KEPT, Conversation, render_for_log
|
||
|
|
|
||
|
|
|
||
|
|
def test_append_and_read_back(tmp_path: Path) -> None:
|
||
|
|
convo = Conversation(tmp_path / "c.json")
|
||
|
|
convo.append("user", "hello", session_id="session-1")
|
||
|
|
convo.append("assistant", "hi back")
|
||
|
|
assert [t.role for t in convo.turns] == ["user", "assistant"]
|
||
|
|
persisted = json.loads((tmp_path / "c.json").read_text())
|
||
|
|
assert persisted[0]["session_id"] == "session-1"
|
||
|
|
|
||
|
|
|
||
|
|
def test_total_chars_sums_content(tmp_path: Path) -> None:
|
||
|
|
convo = Conversation(tmp_path / "c.json")
|
||
|
|
convo.append("user", "abc")
|
||
|
|
convo.append("assistant", "defg")
|
||
|
|
assert convo.total_chars() == 7
|
||
|
|
|
||
|
|
|
||
|
|
def test_needs_summarization_threshold(tmp_path: Path) -> None:
|
||
|
|
convo = Conversation(tmp_path / "c.json")
|
||
|
|
convo.append("user", "x" * 100)
|
||
|
|
assert not convo.needs_summarization(200)
|
||
|
|
assert convo.needs_summarization(50)
|
||
|
|
|
||
|
|
|
||
|
|
def test_replace_with_summary_keeps_recent_turns(tmp_path: Path) -> None:
|
||
|
|
convo = Conversation(tmp_path / "c.json")
|
||
|
|
for i in range(RECENT_TURNS_KEPT + 5):
|
||
|
|
convo.append("user", f"u{i}")
|
||
|
|
convo.append("assistant", f"a{i}")
|
||
|
|
convo.replace_with_summary("SUMMARY")
|
||
|
|
assert convo.turns[0].role == "assistant"
|
||
|
|
assert convo.turns[0].content == "SUMMARY"
|
||
|
|
assert convo.turns[0].meta == "summary"
|
||
|
|
# 1 summary + RECENT_TURNS_KEPT verbatim
|
||
|
|
assert len(convo.turns) == 1 + RECENT_TURNS_KEPT
|
||
|
|
|
||
|
|
|
||
|
|
def test_replace_with_summary_persists(tmp_path: Path) -> None:
|
||
|
|
convo = Conversation(tmp_path / "c.json")
|
||
|
|
for i in range(20):
|
||
|
|
convo.append("user", f"u{i}")
|
||
|
|
convo.replace_with_summary("S")
|
||
|
|
reloaded = Conversation(tmp_path / "c.json")
|
||
|
|
assert reloaded.turns[0].content == "S"
|
||
|
|
|
||
|
|
|
||
|
|
def test_to_api_messages_strips_metadata(tmp_path: Path) -> None:
|
||
|
|
convo = Conversation(tmp_path / "c.json")
|
||
|
|
convo.append("user", "x", session_id="s1")
|
||
|
|
convo.append("assistant", "y")
|
||
|
|
msgs = convo.to_api_messages()
|
||
|
|
assert msgs == [{"role": "user", "content": "x"}, {"role": "assistant", "content": "y"}]
|
||
|
|
|
||
|
|
|
||
|
|
def test_render_for_log_truncates_long_content(tmp_path: Path) -> None:
|
||
|
|
convo = Conversation(tmp_path / "c.json")
|
||
|
|
convo.append("user", "a" * 1000)
|
||
|
|
rendered = render_for_log(convo.turns[0], max_chars=50)
|
||
|
|
assert len(rendered["content"]) < 100
|
||
|
|
assert "more chars" in rendered["content"]
|