21 lines
597 B
Python
21 lines
597 B
Python
|
import lancedb
|
||
|
|
import config, embedding
|
||
|
|
db = lancedb.connect(config.memories_db_path)
|
||
|
|
table = db.open_table(config.memories_table)
|
||
|
|
question = "Email group Microsoft tidak muncul, apakah masuk spam?"
|
||
|
|
query_vector = embedding.embed_text(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"])
|