aji/hendrik
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Untuk perbaikin data loss

Browse Source
This commit is contained in:
Dita Aji Pratama 2026-07-19 11:26:54 +07:00
parent f3c04ab0c3
commit 77c4de6092
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/ragroleplay.py
View File

@ -126,13 +126,17 @@ def embed_text(url, model, text):
data = response.json() data = response.json()
return data["embeddings"][0] 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: try:
db = lancedb.connect(db_path) db = lancedb.connect(db_path)
table = db.open_table("knowledge_user") table = db.open_table("knowledge_user")
vector = embed_text(model_url, model_name, fullname) 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 = { record = {
"id": uuid.uuid4(), "id": user_id,
"timestamp": datetime.now(), "timestamp": datetime.now(),
"character": character, "character": character,
"fullname": fullname, "fullname": fullname,

5
tools/ragroleplay.py
View File

@ -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: try:
success = ragroleplay_lib.users_store( success = ragroleplay_lib.users_store(
config.ragroleplay_db_path, config.ragroleplay_db_path,
@ -46,7 +46,8 @@ def users_store(fullname, nickname, character, alias=None, salutation=None, pers
telegram_username, telegram_username,
xampp_username, xampp_username,
character, character,
familyname familyname,
id
) )
if success: if success:
return "Berhasil menyimpan profil user baru." return "Berhasil menyimpan profil user baru."