mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 07:13:17 +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>
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
"""SA housekeeping: delete orphan chat_message / tool_log rows."""
|
|
|
|
from __future__ import annotations
|
|
|
|
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_prune.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_delete_orphan_chat_message_and_tool_log(fresh_sqlite_store: SqliteStore) -> None:
|
|
s = fresh_sqlite_store
|
|
with s._connect() as conn:
|
|
conn.execute("PRAGMA foreign_keys = OFF")
|
|
conn.execute(
|
|
"INSERT INTO chat_message (session_id, role, content, timestamp) VALUES (?, ?, ?, ?)",
|
|
("ghost-sid", "user", "orphan", "2020-01-01T00:00:00+00:00"),
|
|
)
|
|
conn.execute(
|
|
"""
|
|
INSERT INTO tool_log (session_id, tool_name, specialist, args, result, timestamp, duration_ms)
|
|
VALUES (?, ?, '', '{}', '{}', ?, NULL)
|
|
""",
|
|
("ghost-sid", "t", "2020-01-01T00:00:01+00:00"),
|
|
)
|
|
conn.execute("PRAGMA foreign_keys = ON")
|
|
n_msg = s._chat_messages_repo().delete_messages_where_session_missing()
|
|
n_tl = s._tool_log_queries_repo().delete_tool_logs_where_session_missing()
|
|
assert n_msg >= 1
|
|
assert n_tl >= 1
|
|
|
|
|
|
def test_sa_prune_runs_with_sa_deletes(fresh_sqlite_store: SqliteStore) -> None:
|
|
s = fresh_sqlite_store
|
|
s.create_session("keep")
|
|
with s._connect() as conn:
|
|
s._prune_rows_for_missing_chat_session(conn)
|
|
# second call should not raise
|
|
with s._connect() as conn:
|
|
s._prune_rows_for_missing_chat_session(conn)
|