Update
This commit is contained in:
@@ -33,7 +33,6 @@ def _seed_done_session(owner_id="ds-session-owner"):
|
||||
"findings": [
|
||||
{"title": "Finding one", "detail": "Detail.", "confidence": 0.8, "citations": [1]}
|
||||
],
|
||||
"gaps": ["An open gap."],
|
||||
"sources": [{"url": "https://example.com", "title": "Example", "source": "httpx"}],
|
||||
"score": 70,
|
||||
"confidence": 0.7,
|
||||
@@ -113,6 +112,64 @@ def test_session_html_renders_without_jinja_global_collision(app_server):
|
||||
_clear()
|
||||
|
||||
|
||||
def test_session_reads_disk_report_before_result_commit(app_server):
|
||||
from pathlib import Path
|
||||
import shutil
|
||||
|
||||
from devplacepy.config import DEEPSEARCH_DIR
|
||||
|
||||
owner_id = "ds-race-owner"
|
||||
uid = queue.enqueue(
|
||||
"deepsearch",
|
||||
{"query": "race question", "depth": 2, "max_pages": 10},
|
||||
"user",
|
||||
owner_id,
|
||||
"DeepSearch: race question",
|
||||
)
|
||||
create_deepsearch_session(
|
||||
uid, "user", owner_id, "race question", 2, 10, f"ds_{uid.replace('-', '')}"
|
||||
)
|
||||
report = {
|
||||
"query": "race question",
|
||||
"summary": "A grounded summary from disk.",
|
||||
"findings": [
|
||||
{"title": "Disk finding", "detail": "D.", "confidence": 0.8, "citations": [1]}
|
||||
],
|
||||
"sources": [{"url": "https://example.com", "title": "Example", "source": "httpx"}],
|
||||
"score": 84,
|
||||
"confidence": 0.85,
|
||||
"source_diversity": 0.75,
|
||||
"synthesis": "agents",
|
||||
"page_count": 12,
|
||||
"chunk_count": 61,
|
||||
}
|
||||
session_dir = DEEPSEARCH_DIR / uid
|
||||
session_dir.mkdir(parents=True, exist_ok=True)
|
||||
(session_dir / "report.json").write_text(json.dumps(report), encoding="utf-8")
|
||||
get_table("deepsearch_sessions").update(
|
||||
{"uid": uid, "status": "done"}, ["uid"]
|
||||
)
|
||||
refresh_snapshot()
|
||||
try:
|
||||
r = requests.get(
|
||||
f"{BASE_URL}/tools/deepsearch/{uid}/session", headers=_json_headers()
|
||||
)
|
||||
assert r.status_code == 200, r.text
|
||||
body = r.json()
|
||||
assert body["status"] == "done"
|
||||
assert body["score"] == 84
|
||||
assert body["chunk_count"] == 61
|
||||
assert body["findings"]
|
||||
assert body["sources"]
|
||||
assert body["chat_ws_url"] == f"/tools/deepsearch/{uid}/chat"
|
||||
md = requests.get(f"{BASE_URL}/tools/deepsearch/{uid}/export.md")
|
||||
assert md.status_code == 200, md.text
|
||||
assert "Disk finding" in md.text
|
||||
finally:
|
||||
shutil.rmtree(session_dir, ignore_errors=True)
|
||||
_clear()
|
||||
|
||||
|
||||
def test_session_unknown_uid_404(app_server):
|
||||
r = requests.get(
|
||||
f"{BASE_URL}/tools/deepsearch/nope/session", headers=_json_headers()
|
||||
|
||||
Reference in New Issue
Block a user