feat(harness): port backend Onda 0 — vendor, lshindex, sqlcheck, execute, rest/exec, ctetest, report, datamart
8 moduli leaf portati verbatim da ChironeWp3 con rename psdwp3→tht: - vendor/thoth_lsh (MinHash/LSH, leaf puro datasketch+tqdm) + VENDORED.md - lshindex/ (build/save/load/query, dipende vendor + LshConfig) - sqlcheck/ (validate_sql, leaf ExecutionConfig+mschema) - execute/ + execute/warnings (run_controlled/explain, leaf sqlglot+sqlalchemy) - rest/execute + rest/explain (REST variants, dipendono execute+rest.client) - ctetest (CTE test records, leaf sqlglot+pydantic) - report (validation report rendering, dipende execute+sqlcheck) - datamart (stub NotImplementedError) Verifica: import smoke catena completa OK, pytest 109 passed. Deps (datasketch, sqlglot, sqlalchemy, pydantic, requests, tqdm) già in pyproject. VENDORED.md neutralizzato (riferimenti PsdWp3→Thoth).
This commit is contained in:
@@ -0,0 +1,77 @@
|
||||
import json
|
||||
import pickle
|
||||
from dataclasses import dataclass
|
||||
from pathlib import Path
|
||||
from typing import Any
|
||||
|
||||
from datasketch import MinHash, MinHashLSH
|
||||
|
||||
from tht.config import LshConfig
|
||||
from tht.vendor.thoth_lsh import create_lsh_index, create_minhash
|
||||
|
||||
|
||||
class LshIndexError(Exception):
|
||||
pass
|
||||
|
||||
|
||||
@dataclass
|
||||
class LshHit:
|
||||
table: str
|
||||
column: str
|
||||
value: str
|
||||
score: float
|
||||
|
||||
|
||||
def build_index(
|
||||
values: dict[str, dict[str, list[str]]], cfg: LshConfig, verbose: bool = False
|
||||
) -> tuple[MinHashLSH, dict[str, tuple[MinHash, str, str, str]]]:
|
||||
return create_lsh_index(
|
||||
values, signature_size=cfg.signature_size, n_gram=cfg.n_gram,
|
||||
threshold=cfg.threshold, verbose=verbose,
|
||||
)
|
||||
|
||||
|
||||
def _paths(directory: Path, name: str) -> tuple[Path, Path, Path]:
|
||||
return (
|
||||
directory / f"{name}_lsh.pkl",
|
||||
directory / f"{name}_minhashes.pkl",
|
||||
directory / f"{name}_meta.json",
|
||||
)
|
||||
|
||||
|
||||
def save_index(lsh, minhashes, cfg: LshConfig, directory: Path, name: str) -> None:
|
||||
directory.mkdir(parents=True, exist_ok=True)
|
||||
lsh_path, mh_path, meta_path = _paths(directory, name)
|
||||
lsh_path.write_bytes(pickle.dumps(lsh))
|
||||
mh_path.write_bytes(pickle.dumps(minhashes))
|
||||
meta_path.write_text(
|
||||
json.dumps(
|
||||
{"signature_size": cfg.signature_size, "n_gram": cfg.n_gram,
|
||||
"threshold": cfg.threshold, "entries": len(minhashes)},
|
||||
indent=2,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
def load_index(directory: Path, name: str) -> tuple[Any, Any, dict]:
|
||||
lsh_path, mh_path, meta_path = _paths(directory, name)
|
||||
if not (lsh_path.exists() and mh_path.exists() and meta_path.exists()):
|
||||
raise LshIndexError(
|
||||
f"Indice LSH non trovato in {directory} (atteso {name}_lsh.pkl). "
|
||||
f"Esegui prima `nsp lsh build`."
|
||||
)
|
||||
lsh = pickle.loads(lsh_path.read_bytes())
|
||||
minhashes = pickle.loads(mh_path.read_bytes())
|
||||
meta = json.loads(meta_path.read_text())
|
||||
return lsh, minhashes, meta
|
||||
|
||||
|
||||
def query_index(lsh, minhashes, keyword: str, meta: dict, top_n: int = 10) -> list[LshHit]:
|
||||
"""Query con score: i parametri MinHash vengono dal meta dell'indice, non dalla config."""
|
||||
qmh = create_minhash(meta["signature_size"], keyword, meta["n_gram"])
|
||||
scored = [(key, qmh.jaccard(minhashes[key][0])) for key in lsh.query(qmh)]
|
||||
scored.sort(key=lambda kv: kv[1], reverse=True)
|
||||
return [
|
||||
LshHit(table=minhashes[k][1], column=minhashes[k][2], value=minhashes[k][3], score=s)
|
||||
for k, s in scored[:top_n]
|
||||
]
|
||||
Reference in New Issue
Block a user