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

Compare commits

base: aji:77c4de60922ce6805a3317372c6b30a7e2a1ecc7
Branches Tags
aji:master
aji:XMPP
aji:carrack
aji:ollama
..
compare: aji:562935ea1bbd778f2c31e7fe51cbb813b65bc261
Branches Tags
aji:master
aji:XMPP
aji:carrack
aji:ollama

No commits in common. "77c4de60922ce6805a3317372c6b30a7e2a1ecc7" and "562935ea1bbd778f2c31e7fe51cbb813b65bc261" have entirely different histories.
77c4de6092 ... 562935ea1b

2 changed files with 8 additions and 11 deletions
Show Stats Expand all files Collapse all files

12
lib/ragroleplay.py
View File

@ -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(), metadata={'description': 'Active Character'}),
pyarrow.field('character', pyarrow.string(), nullable=True, metadata={'description': 'Active Character'}), # Seharusnya WAJIB, GAK BOLEH NULL
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'}), # XAMPP???
pyarrow.field('xampp_username', pyarrow.string(), nullable=True, metadata={'description': 'XAMPP username'}),
pyarrow.field('vector_fullname', pyarrow.list_(pyarrow.float32(), vector_size), metadata={'description': 'Vector data of fullname'}),
])
@ -126,17 +126,13 @@ 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, id=None):
def users_store(db_path, model_url, model_name, fullname, nickname, alias, salutation, persona, telegram_id, telegram_username, xampp_username, familyname=None, character=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": user_id,
"id": uuid.uuid4(),
"timestamp": datetime.now(),
"character": character,
"fullname": fullname,

7
tools/ragroleplay.py
View File

@ -31,7 +31,9 @@ schema_users_store = {
}
}
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):
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
try:
success = ragroleplay_lib.users_store(
config.ragroleplay_db_path,
@ -45,9 +47,8 @@ def users_store(fullname, nickname, character, id=None, alias=None, salutation=N
telegram_id,
telegram_username,
xampp_username,
character,
familyname,
id
character
)
if success:
return "Berhasil menyimpan profil user baru."