From f3c04ab0c3c0609605aca3d939a7f650e66b6b5f Mon Sep 17 00:00:00 2001 From: Dita Aji Pratama Date: Sun, 19 Jul 2026 10:52:36 +0700 Subject: [PATCH] Cleanup slob --- lib/ragroleplay.py | 6 +++--- tools/ragroleplay.py | 8 +++----- 2 files changed, 6 insertions(+), 8 deletions(-) diff --git a/lib/ragroleplay.py b/lib/ragroleplay.py index d9771ce..5f26d8e 100644 --- a/lib/ragroleplay.py +++ b/lib/ragroleplay.py @@ -5,7 +5,7 @@ def _schema_user(vector_size): return pyarrow.schema([ pyarrow.field('id', pyarrow.uuid(), metadata={'description': 'Unique identifier (UUID)'}), pyarrow.field('timestamp', pyarrow.timestamp('ms'), metadata={'description': 'When record created'}), - pyarrow.field('character', pyarrow.string(), nullable=True, metadata={'description': 'Active Character'}), # Seharusnya WAJIB, GAK BOLEH NULL + pyarrow.field('character', pyarrow.string(), metadata={'description': 'Active Character'}), pyarrow.field('fullname', pyarrow.string(), metadata={'description': 'Fullname'}), pyarrow.field('nickname', pyarrow.string(), metadata={'description': 'Nickname'}), pyarrow.field('familyname', pyarrow.string(), nullable=True, metadata={'description': 'Family/clan name'}), @@ -14,7 +14,7 @@ def _schema_user(vector_size): pyarrow.field('persona', pyarrow.string(), nullable=True, metadata={'description': 'Persona'}), pyarrow.field('telegram_id', pyarrow.string(), nullable=True, metadata={'description': 'Telegram ID'}), pyarrow.field('telegram_username', pyarrow.string(), nullable=True, metadata={'description': 'Telegram username'}), - pyarrow.field('xampp_username', pyarrow.string(), nullable=True, metadata={'description': 'XAMPP username'}), + pyarrow.field('xampp_username', pyarrow.string(), nullable=True, metadata={'description': 'XAMPP username'}), # XAMPP??? pyarrow.field('vector_fullname', pyarrow.list_(pyarrow.float32(), vector_size), metadata={'description': 'Vector data of fullname'}), ]) @@ -126,7 +126,7 @@ 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, familyname=None, character=None): +def users_store(db_path, model_url, model_name, fullname, nickname, alias, salutation, persona, telegram_id, telegram_username, xampp_username, character, familyname=None): try: db = lancedb.connect(db_path) table = db.open_table("knowledge_user") diff --git a/tools/ragroleplay.py b/tools/ragroleplay.py index 8f033ef..5b0a847 100644 --- a/tools/ragroleplay.py +++ b/tools/ragroleplay.py @@ -31,9 +31,7 @@ schema_users_store = { } } -def users_store(fullname, nickname, alias=None, salutation=None, persona=None, telegram_id=None, telegram_username=None, xampp_username=None, familyname=None, character=None): - if not character: - character = personality.PERSONALITY.name +def users_store(fullname, nickname, character, 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, @@ -47,8 +45,8 @@ def users_store(fullname, nickname, alias=None, salutation=None, persona=None, t telegram_id, telegram_username, xampp_username, - familyname, - character + character, + familyname ) if success: return "Berhasil menyimpan profil user baru."