mekosotto commited on
Commit
27a97bf
·
1 Parent(s): 91dde0d

feat(rag): clinical RAG CLI smoke script

Browse files
Files changed (1) hide show
  1. scripts/clinical_rag_smoke.py +26 -0
scripts/clinical_rag_smoke.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Smoke: ask the clinical corpus a question from the terminal.
2
+
3
+ Usage:
4
+ python scripts/clinical_rag_smoke.py "egzersiz Alzheimer feedback"
5
+ """
6
+ from __future__ import annotations
7
+
8
+ import sys
9
+ from pathlib import Path
10
+
11
+ from src.rag.clinical.loader import load_index
12
+ from src.rag.clinical.retrieve import retrieve_clinical
13
+
14
+
15
+ def main() -> None:
16
+ if len(sys.argv) < 2:
17
+ print(__doc__)
18
+ sys.exit(1)
19
+ query = " ".join(sys.argv[1:])
20
+ payload = load_index(Path("data/external_rag/index/rag_index.pkl"))
21
+ result = retrieve_clinical(payload, query, top_k=5, evidence_limit=5)
22
+ print(result.summary_text or "(no matches)")
23
+
24
+
25
+ if __name__ == "__main__":
26
+ main()