mirror of
https://github.com/lllin000/PaperForge.git
synced 2026-07-22 06:50:53 +00:00
75 lines
2.9 KiB
Python
75 lines
2.9 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from paperforge.commands import retrieve as retrieve_command
|
||
|
|
from paperforge.commands import search as search_command
|
||
|
|
|
||
|
|
|
||
|
|
class _Args:
|
||
|
|
def __init__(self, vault_path: Path) -> None:
|
||
|
|
self.vault_path = vault_path
|
||
|
|
self.query = "Lin 2024 Electrical Stimulation"
|
||
|
|
self.json = True
|
||
|
|
self.limit = 10
|
||
|
|
self.domain = None
|
||
|
|
self.year_from = None
|
||
|
|
self.year_to = None
|
||
|
|
self.ocr = None
|
||
|
|
self.deep = None
|
||
|
|
self.lifecycle = None
|
||
|
|
self.next_step = None
|
||
|
|
self.expand = True
|
||
|
|
|
||
|
|
|
||
|
|
class _Conn:
|
||
|
|
def close(self) -> None:
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
def test_search_zero_results_emits_query_diagnostic(monkeypatch, tmp_path: Path, capsys) -> None:
|
||
|
|
db_path = tmp_path / "paperforge.db"
|
||
|
|
db_path.write_text("", encoding="utf-8")
|
||
|
|
monkeypatch.setattr(search_command, "get_memory_db_path", lambda vault: db_path)
|
||
|
|
monkeypatch.setattr(search_command, "get_connection", lambda db_path, read_only=True: _Conn())
|
||
|
|
monkeypatch.setattr(search_command, "search_papers", lambda *args, **kwargs: [])
|
||
|
|
|
||
|
|
args = _Args(tmp_path)
|
||
|
|
assert search_command.run(args) == 0
|
||
|
|
payload = json.loads(capsys.readouterr().out)
|
||
|
|
assert payload["ok"] is True
|
||
|
|
assert payload["data"]["count"] == 0
|
||
|
|
assert "query_diagnostic" in payload["data"]
|
||
|
|
assert payload["warnings"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_retrieve_unavailable_emits_suggested_modes(monkeypatch, tmp_path: Path, capsys) -> None:
|
||
|
|
monkeypatch.setattr("paperforge.embedding.get_embed_status", lambda vault: {"healthy": True, "chunk_count": 0, "db_exists": False, "error": ""})
|
||
|
|
|
||
|
|
args = _Args(tmp_path)
|
||
|
|
args.query = "galvanotaxis"
|
||
|
|
assert retrieve_command.run(args) == 1
|
||
|
|
payload = json.loads(capsys.readouterr().out)
|
||
|
|
assert payload["ok"] is False
|
||
|
|
assert payload["data"]["interactive_fallback_required"] is True
|
||
|
|
assert payload["data"]["suggested_modes"]
|
||
|
|
|
||
|
|
|
||
|
|
def test_retrieve_low_confidence_hits_emit_warning(monkeypatch, tmp_path: Path, capsys) -> None:
|
||
|
|
monkeypatch.setattr("paperforge.embedding.get_embed_status", lambda vault: {"healthy": True, "chunk_count": 5, "db_exists": True, "error": ""})
|
||
|
|
monkeypatch.setattr(retrieve_command, "retrieve_chunks", lambda vault, query, limit=5, expand=True: [
|
||
|
|
{"paper_id": "AAA11111", "chunk_text": "[Figure]", "score": 0.58},
|
||
|
|
{"paper_id": "AAA11111", "chunk_text": "### Keywords", "score": 0.57},
|
||
|
|
{"paper_id": "AAA11111", "chunk_text": "None.", "score": 0.56},
|
||
|
|
])
|
||
|
|
monkeypatch.setattr(retrieve_command, "get_memory_db_path", lambda vault: tmp_path / "missing.db")
|
||
|
|
|
||
|
|
args = _Args(tmp_path)
|
||
|
|
args.query = "unlikelynonexistenttermxyz"
|
||
|
|
assert retrieve_command.run(args) == 0
|
||
|
|
payload = json.loads(capsys.readouterr().out)
|
||
|
|
assert payload["ok"] is True
|
||
|
|
assert payload["warnings"]
|
||
|
|
assert payload["data"]["query_diagnostic"]["interactive_fallback_required"] is True
|