Dani
48585acb6f
Added life log, persona, and plugins manager. changed it so that any new .json files aren't uploaded
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
import json
|
||
import os
|
||
import re
|
||
|
||
class Persona:
|
||
"""
|
||
Learns Ruby’s persona entirely from her model by asking for JSON,
|
||
with no hard-coded examples or defaults.
|
||
"""
|
||
def __init__(self, path="persona.json"):
|
||
self.path = path
|
||
self.traits = {}
|
||
self.load()
|
||
|
||
def load(self):
|
||
if os.path.isfile(self.path):
|
||
with open(self.path, "r", encoding="utf-8") as f:
|
||
self.traits = json.load(f)
|
||
|
||
def save(self):
|
||
with open(self.path, "w", encoding="utf-8") as f:
|
||
json.dump(self.traits, f, ensure_ascii=False, indent=2)
|
||
|
||
def summary(self) -> str:
|
||
if not self.traits:
|
||
return ""
|
||
name = self.traits.get("name", "")
|
||
age = self.traits.get("age", "")
|
||
hobbies = self.traits.get("hobbies", [])
|
||
tone = self.traits.get("tone", "")
|
||
hlist = ", ".join(hobbies) if isinstance(hobbies, list) else str(hobbies)
|
||
return f"I'm {name}, {age} years old, I love {hlist}, and speak in a {tone} tone."
|
||
|
||
def bootstrap(self, system):
|
||
"""
|
||
Ask Ruby to introspect and output a JSON object with keys:
|
||
name, age, hobbies (array), and tone (string). No examples given.
|
||
"""
|
||
prompt = (
|
||
"Based on all the text you have absorbed, introduce yourself by OUTPUTTING "
|
||
"ONLY a JSON object with these keys exactly:\n"
|
||
' "name": string,\n'
|
||
' "age": number,\n'
|
||
' "hobbies": array of strings,\n'
|
||
' "tone": string describing your speaking style\n'
|
||
"Do not output anything else."
|
||
)
|
||
raw = system.generate(prompt, max_len=150, temperature=0.7, top_p=0.9)
|
||
|
||
# extract the first JSON object block
|
||
m = re.search(r"\{.*?\}", raw, flags=re.DOTALL)
|
||
if not m:
|
||
return
|
||
try:
|
||
data = json.loads(m.group(0))
|
||
except json.JSONDecodeError:
|
||
return
|
||
|
||
# keep only expected keys
|
||
updated = {}
|
||
for key in ("name", "age", "hobbies", "tone"):
|
||
if key in data:
|
||
updated[key] = data[key]
|
||
|
||
if updated:
|
||
self.traits = updated
|
||
self.save()
|