From 51f4a8c9d04056ab72c13267f208620679bbc365 Mon Sep 17 00:00:00 2001 From: Dita Aji Pratama Date: Thu, 2 Jul 2026 17:29:46 +0700 Subject: [PATCH] polished --- memories_semantic_search.py | 4 ++-- requirements.txt | 1 + table_modify.py | 4 ++-- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/memories_semantic_search.py b/memories_semantic_search.py index 616d7c4..3930585 100644 --- a/memories_semantic_search.py +++ b/memories_semantic_search.py @@ -2,8 +2,8 @@ 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) +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) diff --git a/requirements.txt b/requirements.txt index 9a3182a..0449730 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,3 @@ +requests lancedb pyarrow diff --git a/table_modify.py b/table_modify.py index 2628367..9a58bfe 100644 --- a/table_modify.py +++ b/table_modify.py @@ -1,5 +1,5 @@ import lancedb -import embedding +import config, embedding db = lancedb.connect(config.memories_db_path) table = db.open_table(config.memories_table) docs = [ @@ -34,7 +34,7 @@ for doc in docs: "id" : doc["id" ], "title" : doc["title" ], "story" : doc["story" ], - "vector_title" : embedding.embed_text(doc["title"]), + "vector_title" : embedding.embed_text(config.model_url, config.model_name, doc["title"]), } table.add([row]) print(f"Menambahkan record: {doc['title']}")