mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 01:50:44 +08:00
oliver
d14e9d3596
- Add SQLAlchemy Core repos, pg adapter/compat, assistant_store factory, Alembic bootstrap and migration/cutover scripts. - Harden chat_message writes (NUL scrub for PG), turn_uuid on attempt failure, WS turn_runner fallbacks and gateway executed_turn_uuid init. - start_gateway: log paths, PS7 stderr handling via cmd, background stdout/stderr redirect; runtime assistant_runtime_log_dir export. - Ops: clear_all_chat_sessions with PG-only --postgresql and env-gated wipe; clear_postgres_chat_sessions.ps1. - Tests: SA repos, pg compat, persist fallback, smoke env isolation; CI and docs touch-ups. Co-authored-by: Cursor <cursoragent@cursor.com>
85 lines
3.2 KiB
Python
85 lines
3.2 KiB
Python
"""SA migration: fork_session, trim_messages, list_admin_sessions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
import pytest
|
|
|
|
from svc.persistence.db.engine import clear_assistant_engine_cache
|
|
from svc.persistence.sqlite_store import SqliteStore
|
|
|
|
|
|
@pytest.fixture
|
|
def fresh_sqlite_store(monkeypatch: pytest.MonkeyPatch, tmp_path):
|
|
monkeypatch.delenv("AIA_ASSISTANT_DB_BACKEND", raising=False)
|
|
monkeypatch.setenv("OPS_WORKSPACE_ROOT", str(tmp_path))
|
|
dbfile = tmp_path / "sa_fork.sqlite"
|
|
monkeypatch.setenv("AIA_ASSISTANT_DB_PATH", str(dbfile))
|
|
clear_assistant_engine_cache()
|
|
s = SqliteStore(str(dbfile))
|
|
try:
|
|
yield s
|
|
finally:
|
|
clear_assistant_engine_cache()
|
|
|
|
|
|
def test_sa_fork_session_remaps_tool_assistant_id(fresh_sqlite_store: SqliteStore) -> None:
|
|
s = fresh_sqlite_store
|
|
src = s.create_session("Src")
|
|
asst = s.add_message(src.id, "assistant", "a", tool_calls="{}")
|
|
tc = json.dumps({"assistant_message_id": asst.id}, ensure_ascii=False)
|
|
s.add_message(src.id, "tool", "t", tool_calls=tc)
|
|
forked = s.fork_session(src.id, up_to_message_id=int(asst.id) + 1, title="Forked")
|
|
assert forked.id != src.id
|
|
msgs = s.get_messages(forked.id, limit=10)
|
|
assert len(msgs) == 2
|
|
assert msgs[0].role == "assistant"
|
|
assert msgs[1].role == "tool"
|
|
meta = json.loads(msgs[1].tool_calls or "{}")
|
|
assert int(meta.get("assistant_message_id") or 0) == msgs[0].id
|
|
|
|
|
|
def test_sa_fork_session_bad_anchor(fresh_sqlite_store: SqliteStore) -> None:
|
|
s = fresh_sqlite_store
|
|
src = s.create_session("S2")
|
|
m = s.add_message(src.id, "user", "x")
|
|
n_before = s.count_sessions()
|
|
with pytest.raises(ValueError, match="message not in session"):
|
|
s.fork_session(src.id, up_to_message_id=int(m.id) + 99, title="Bad")
|
|
assert s.count_sessions() == n_before
|
|
|
|
|
|
def test_sa_trim_messages_keep_last(fresh_sqlite_store: SqliteStore) -> None:
|
|
s = fresh_sqlite_store
|
|
sess = s.create_session("Trim")
|
|
for i in range(5):
|
|
s.add_message(sess.id, "user", f"m{i}")
|
|
s.trim_messages(sess.id, keep_last=2)
|
|
assert s.count_messages(sess.id) == 2
|
|
rows = s.get_messages(sess.id, limit=10)
|
|
assert [x.content for x in rows] == ["m3", "m4"]
|
|
|
|
|
|
def test_sa_trim_messages_zero_deletes_session(fresh_sqlite_store: SqliteStore) -> None:
|
|
s = fresh_sqlite_store
|
|
sess = s.create_session("Z")
|
|
s.add_message(sess.id, "user", "u")
|
|
s.trim_messages(sess.id, keep_last=0)
|
|
assert s.get_session(sess.id) is None
|
|
|
|
|
|
def test_sa_list_admin_sessions(fresh_sqlite_store: SqliteStore) -> None:
|
|
s = fresh_sqlite_store
|
|
t = s.create_tenant("AdmT")
|
|
u = s.create_user(tenant_id=t["id"], display_name="Alice Admin", role="member")
|
|
sess = s.create_session_for_user(title="S1", tenant_id=t["id"], user_id=u["id"])
|
|
s.add_message(sess.id, "user", "hello")
|
|
total, rows = s.list_admin_sessions(tenant_id=t["id"], limit=10, offset=0)
|
|
assert total >= 1
|
|
hit = next(x for x in rows if x["session_id"] == sess.id)
|
|
assert hit["message_count"] == 1
|
|
assert "alice" in hit["display_name"].lower() or "alice" in hit["username"].lower()
|
|
t2, rows2 = s.list_admin_sessions(tenant_id=t["id"], q="S1", limit=10, offset=0)
|
|
assert t2 >= 1
|
|
assert any(x["session_id"] == sess.id for x in rows2)
|