example.py

"""
example.py

Minimal usage example for the HV Context selector.
"""

from hv_context import HVContext


def main():
    model = HVContext.load("context_model")

    queries = [
        "what is the forecast for rome",
        "how do i debug a python function",
        "how much did i spend on groceries",
        "how do i cook soup",
    ]

    print("Top-3 relevant turns for each query:")
    print()
    for q in queries:
        idx, scores = model.retrieve(q, k=3)
        print(f"Query: {q}")
        for i, (j, s) in enumerate(zip(idx, scores)):
            topic = model.topics[j]
            text = model.texts[j][:60]
            print(f"  top{i+1} [{topic:<10}] sim={s:.3f}")
            print(f"         {text}...")
        print()


if __name__ == "__main__":
    main()