memories/memories_semantic_search.py

21 lines
609 B
Python
Raw Normal View History

2026-07-02 15:20:55 +07:00
import lancedb
import config, embedding
db = lancedb.connect(config.memories_db_path)
table = db.open_table(config.memories_table)
2026-07-02 17:29:46 +07:00
question = "cara mengecek spam di outlook"
query_vector = embedding.embed_text(config.model_url, config.model_name, question)
2026-07-02 15:20:55 +07:00
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"])