Regulatory

from gaussia.metrics.regulatory import Regulatory
from gaussia.connectors import MyCorpusConnector
from gaussia.embedders import SentenceTransformerEmbedder
from gaussia.rerankers import MyReranker

embedder = SentenceTransformerEmbedder(model="all-mpnet-base-v2")
reranker = MyReranker()
corpus = MyCorpusConnector(path="./regulations/")

results = Regulatory.run(
    MyRetriever,
    corpus_connector=corpus,
    embedder=embedder,
    reranker=reranker,
)

for r in results:
    print(f"Compliance: {r.compliance_score:.2f} ({r.verdict})")
    print(f"Supporting: {r.total_supporting_chunks}, Contradicting: {r.total_contradicting_chunks}")