From 77c4de60922ce6805a3317372c6b30a7e2a1ecc7 Mon Sep 17 00:00:00 2001
From: Dita Aji Pratama <ditaajipratama@protonmail.com>
Date: Sun, 19 Jul 2026 11:26:54 +0700
Subject: [PATCH] Untuk perbaikin data loss

---
 lib/ragroleplay.py   | 8 ++++++--
 tools/ragroleplay.py | 5 +++--
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/lib/ragroleplay.py b/lib/ragroleplay.py
index 5f26d8e..e5a483c 100644
--- a/lib/ragroleplay.py
+++ b/lib/ragroleplay.py
@@ -126,13 +126,17 @@ def embed_text(url, model, text):
     data      = response.json()
     return data["embeddings"][0]
 
-def users_store(db_path, model_url, model_name, fullname, nickname, alias, salutation, persona, telegram_id, telegram_username, xampp_username, character, familyname=None):
+def users_store(db_path, model_url, model_name, fullname, nickname, alias, salutation, persona, telegram_id, telegram_username, xampp_username, character, familyname=None, id=None):
     try:
         db      = lancedb.connect(db_path)
         table   = db.open_table("knowledge_user")
         vector  = embed_text(model_url, model_name, fullname)
+        
+        # Gunakan id yang disediakan atau generate baru jika None
+        user_id = _uuid(id) if id is not None else uuid.uuid4()
+        
         record  = {
-            "id": uuid.uuid4(),
+            "id": user_id,
             "timestamp": datetime.now(),
             "character": character,
             "fullname": fullname,
diff --git a/tools/ragroleplay.py b/tools/ragroleplay.py
index 5b0a847..f35b878 100644
--- a/tools/ragroleplay.py
+++ b/tools/ragroleplay.py
@@ -31,7 +31,7 @@ schema_users_store = {
     }
 }
 
-def users_store(fullname, nickname, character, alias=None, salutation=None, persona=None, telegram_id=None, telegram_username=None, xampp_username=None, familyname=None):
+def users_store(fullname, nickname, character, id=None, alias=None, salutation=None, persona=None, telegram_id=None, telegram_username=None, xampp_username=None, familyname=None):
     try:
         success = ragroleplay_lib.users_store(
             config.ragroleplay_db_path,
@@ -46,7 +46,8 @@ def users_store(fullname, nickname, character, alias=None, salutation=None, pers
             telegram_username,
             xampp_username,
             character,
-            familyname
+            familyname,
+            id
         )
         if success:
             return "Berhasil menyimpan profil user baru."