127 lines
4.8 KiB
Python
127 lines
4.8 KiB
Python
"""
|
||
Persistence layer for V2 autonomous system.
|
||
Saves and restores critical context data across bot restarts.
|
||
"""
|
||
|
||
import json
|
||
import time
|
||
from pathlib import Path
|
||
from typing import Dict, Optional
|
||
from datetime import datetime, timezone
|
||
|
||
CONTEXT_FILE = Path("memory/autonomous_context.json")
|
||
|
||
|
||
def save_autonomous_context(server_contexts: dict, server_last_action: dict):
|
||
"""
|
||
Save critical context data to disk.
|
||
Only saves data that makes sense to persist (not ephemeral stats).
|
||
"""
|
||
now = time.time()
|
||
|
||
data = {
|
||
"saved_at": now,
|
||
"saved_at_readable": datetime.now(timezone.utc).isoformat(),
|
||
"servers": {}
|
||
}
|
||
|
||
for guild_id, ctx in server_contexts.items():
|
||
data["servers"][str(guild_id)] = {
|
||
# Critical timing data
|
||
"time_since_last_action": ctx.time_since_last_action,
|
||
"time_since_last_interaction": ctx.time_since_last_interaction,
|
||
"messages_since_last_appearance": ctx.messages_since_last_appearance,
|
||
|
||
# Decay-able activity data (will be aged on restore)
|
||
"conversation_momentum": ctx.conversation_momentum,
|
||
"unique_users_active": ctx.unique_users_active,
|
||
|
||
# Last action timestamp (absolute time)
|
||
"last_action_timestamp": server_last_action.get(guild_id, 0),
|
||
|
||
# Mood state (already persisted in servers_config.json, but include for completeness)
|
||
"current_mood": ctx.current_mood,
|
||
"mood_energy_level": ctx.mood_energy_level
|
||
}
|
||
|
||
try:
|
||
CONTEXT_FILE.parent.mkdir(parents=True, exist_ok=True)
|
||
with open(CONTEXT_FILE, 'w') as f:
|
||
json.dump(data, f, indent=2)
|
||
print(f"💾 [V2] Saved autonomous context for {len(server_contexts)} servers")
|
||
except Exception as e:
|
||
print(f"⚠️ [V2] Failed to save autonomous context: {e}")
|
||
|
||
|
||
def load_autonomous_context() -> tuple[Dict[int, dict], Dict[int, float]]:
|
||
"""
|
||
Load and restore context data from disk.
|
||
Returns (server_context_data, server_last_action).
|
||
|
||
Applies staleness/decay rules based on downtime:
|
||
- conversation_momentum decays over time
|
||
- Timestamps are adjusted for elapsed time
|
||
"""
|
||
if not CONTEXT_FILE.exists():
|
||
print("ℹ️ [V2] No saved context found, starting fresh")
|
||
return {}, {}
|
||
|
||
try:
|
||
with open(CONTEXT_FILE, 'r') as f:
|
||
data = json.load(f)
|
||
|
||
saved_at = data.get("saved_at", 0)
|
||
downtime = time.time() - saved_at
|
||
downtime_minutes = downtime / 60
|
||
|
||
print(f"📂 [V2] Loading context from {downtime_minutes:.1f} minutes ago")
|
||
|
||
context_data = {}
|
||
last_action = {}
|
||
|
||
for guild_id_str, server_data in data.get("servers", {}).items():
|
||
guild_id = int(guild_id_str)
|
||
|
||
# Apply decay/staleness rules
|
||
momentum = server_data.get("conversation_momentum", 0.0)
|
||
|
||
# Momentum decays: half-life of 10 minutes
|
||
if downtime > 0:
|
||
decay_factor = 0.5 ** (downtime_minutes / 10)
|
||
momentum = momentum * decay_factor
|
||
|
||
# Restore data with adjustments
|
||
context_data[guild_id] = {
|
||
"time_since_last_action": server_data.get("time_since_last_action", 0) + downtime,
|
||
"time_since_last_interaction": server_data.get("time_since_last_interaction", 0) + downtime,
|
||
"messages_since_last_appearance": server_data.get("messages_since_last_appearance", 0),
|
||
"conversation_momentum": momentum,
|
||
"unique_users_active": 0, # Reset (stale data)
|
||
"current_mood": server_data.get("current_mood", "neutral"),
|
||
"mood_energy_level": server_data.get("mood_energy_level", 0.5)
|
||
}
|
||
|
||
# Restore last action timestamp
|
||
last_action_timestamp = server_data.get("last_action_timestamp", 0)
|
||
if last_action_timestamp > 0:
|
||
last_action[guild_id] = last_action_timestamp
|
||
|
||
print(f"✅ [V2] Restored context for {len(context_data)} servers")
|
||
print(f" └─ Momentum decay factor: {decay_factor:.3f} (from {downtime_minutes:.1f}min downtime)")
|
||
|
||
return context_data, last_action
|
||
|
||
except Exception as e:
|
||
print(f"⚠️ [V2] Failed to load autonomous context: {e}")
|
||
return {}, {}
|
||
|
||
|
||
def apply_context_to_signals(context_data: dict, context_signals):
|
||
"""
|
||
Apply loaded context data to a ContextSignals object.
|
||
Call this after creating a fresh ContextSignals instance.
|
||
"""
|
||
for key, value in context_data.items():
|
||
if hasattr(context_signals, key):
|
||
setattr(context_signals, key, value)
|