import lancedb
import config, embedding
db = lancedb.connect(config.memories_db_path)
table = db.open_table(config.memories_table)
question = "cara mengecek spam di outlook"
query_vector = embedding.embed_text(config.model_url, config.model_name, question)
results = (
    table.search(query_vector, vector_column_name="vector_title")
    .limit(3)
    .to_list()
)
print("Pertanyaan:")
print(question)
print("Hasil paling mirip:")
for row in results:
    print("=" * 60)
    print("ID:", row["id"])
    print("Title:", row["title"])
    print("Story:", row["story"])
    print("Distance:", row["_distance"])