U j' @s\ddlZddlmZddlmZmZedZddZddZdd d Zd d Z dddZ dS)N)Path)datetimetimezonez/var/lib/hermes-nexus/memory.dbcCs>tjttdd}|d|d|d|d|S)N )ZtimeoutzPRAGMA busy_timeout=10000aJ CREATE TABLE IF NOT EXISTS hermes_conversaciones ( id INTEGER PRIMARY KEY AUTOINCREMENT, usuario TEXT NOT NULL, conversacion TEXT NOT NULL, rol TEXT NOT NULL CHECK(rol IN ('user','model')), contenido TEXT NOT NULL, fecha TEXT NOT NULL ) zr CREATE INDEX IF NOT EXISTS idx_historial ON hermes_conversaciones(usuario, conversacion, id) a CREATE TABLE IF NOT EXISTS hermes_preferencias ( id INTEGER PRIMARY KEY AUTOINCREMENT, usuario TEXT NOT NULL, instruccion TEXT NOT NULL, fecha TEXT NOT NULL, UNIQUE(usuario, instruccion) ) )sqlite3ZconnectstrDBexecute)Zconexionr hermes_memory.pyconectars    r c CsH|dkrtdt(}|d||||ttjfW5QRXdS)N)userZmodelzRol no permitidoz INSERT INTO hermes_conversaciones (usuario, conversacion, rol, contenido, fecha) VALUES (?, ?, ?, ?, ?) ) ValueErrorr r rnowrutc isoformat)usuario conversacionrol contenidodbr r r guardar#sr c CsNtdtt|d}t}|d|||f}W5QRXddt|DS)Nz SELECT rol, contenido FROM hermes_conversaciones WHERE usuario=? AND conversacion=? ORDER BY id DESC LIMIT ? cSs g|]\}}|d|igdqS)text)Zrolepartsr ).0rrr r r @szhistorial..)maxminintr r fetchallreversed)rrlimiterfilasr r r historial4sr&c Cshd|}dt|kr*dks4ntdt$}|d||tt j  fW5QRX|S)N iu3La instrucción debe tener entre 3 y 400 caractereszy INSERT OR IGNORE INTO hermes_preferencias(usuario, instruccion, fecha) VALUES (?, ?, ?) ) joinstripsplitlenrr r rrrrr)rZ instruccionrr r r recordarFsr-dc CsHt*}|d|tdtt|df}W5QRXddt|DS)Nzv SELECT instruccion FROM hermes_preferencias WHERE usuario=? ORDER BY id DESC LIMIT ? rr.cSsg|] }|dqS)rr )rZfilar r r rXsz preferencias..)r r rr r!r"r#)rr$rr%r r r preferenciasRs r/)r)r.) rpathlibrrrrr rr&r-r/r r r r s