feat(server): add workbench API, DSH planner worker and RAG pipeline

This commit is contained in:
yi
2026-09-16 19:06:20 +08:00
parent 72fc5b47c7
commit d4b5b3f43a
122 changed files with 31906 additions and 0 deletions
+47
View File
@@ -0,0 +1,47 @@
"""Bounded ordinary system-log writer tests."""
from pathlib import Path
import tempfile
from sqlmodel import Session, SQLModel, create_engine, select
from app import system_log
from app.models import SystemLog
def _engine():
path = Path(tempfile.mktemp(suffix=".db"))
return create_engine(f"sqlite:///{path}"), path
def test_emit_batches_ordinary_logs_and_redacts():
engine, path = _engine()
try:
SQLModel.metadata.create_all(engine)
with Session(engine) as session:
assert system_log.emit(session, level="INFO", source="test",
message="token=secret-value") is True
# emit must not commit the request session: the row is only visible
# after the writer creates its own Session and flushes the batch.
assert session.exec(select(SystemLog)).all() == []
assert system_log.flush(3) is True
with Session(engine) as session:
rows = session.exec(select(SystemLog)).all()
assert len(rows) == 1
assert "[REDACTED]" in rows[0].message
finally:
system_log.shutdown()
path.unlink(missing_ok=True)
def test_critical_emit_remains_synchronous_and_returns_id():
engine, path = _engine()
try:
SQLModel.metadata.create_all(engine)
with Session(engine) as session:
assert system_log.emit(session, level="ERROR", source="test",
message="failure") is True
row = session.exec(select(SystemLog)).one()
assert row.id and row.level == "ERROR"
finally:
system_log.shutdown()
path.unlink(missing_ok=True)