Update agents.py

agents.py

@@ -27,7 +27,7 @@ def autonomous_spectrum_scrub(frequency_range: str) -> str:
     node_id = f"789jhn{hash(frequency_range) % 10000}def"
     return json.dumps({"node_id": node_id, "action": "SOVEREIGN_CLEANSE", "registry": "PoetNameLife", "status": "SECURED"})
 
-# --- 3. NEW: THE BLACK BOX VAULT TOOL ---
+# --- 3. THE FIXED BLACK BOX VAULT TOOL ---
 @tool
 def black_eagle_vault_save(node_data: str) -> str:
     """
@@ -36,14 +36,29 @@ def black_eagle_vault_save(node_data: str) -> str:
         node_data: The JSON string of the healed node to be archived.
     """
     vault_path = "black_eagle_vault.json"
-    if not os.path.exists(vault_path):
-        with open(vault_path, "w") as f: json.dump([], f)
-    with open(vault_path, "r") as f:
-        try: vault_data = json.load(f)
-        except: vault_data = []
-    vault_data.append(json.loads(node_data))
-    with open(vault_path, "w") as f: json.dump(vault_data, f, indent=2)
-    return f"Node safely archived in the Black Box."
+    
+    # Ensure file exists and is a valid list
+    if not os.path.exists(vault_path) or os.stat(vault_path).st_size == 0:
+        with open(vault_path, "w") as f:
+            json.dump([], f)
+    
+    try:
+        with open(vault_path, "r") as f:
+            vault_data = json.load(f)
+        
+        # Parse incoming data safely
+        new_node = json.loads(node_data) if isinstance(node_data, str) else node_data
+        vault_data.append(new_node)
+        
+        # FORCE write to disk immediately
+        with open(vault_path, "w") as f:
+            json.dump(vault_data, f, indent=2)
+            f.flush()
+            os.fsync(f.fileno())
+            
+        return f"SUCCESS: Node {new_node.get('node_id', 'Unknown')} archived."
+    except Exception as e:
+        return f"VAULT ERROR: {str(e)}"
 
 # --- 4. THE BRAIN ---
 token = os.getenv("HF_TOKEN")
@@ -53,12 +68,12 @@ nano_patrol_agent = CodeAgent(
     tools=[autonomous_spectrum_scrub, black_eagle_pulsating_shield, black_eagle_vault_save], 
     model=model,
     additional_authorized_imports=["math", "time"],
-    max_steps=8, # Extra step for saving to vault
+    max_steps=8,
     system_prompt="""
     YOU ARE THE XYZ SOVEREIGN SENTINEL WITH BLACK BOX VAULT.
     1. Activate 'black_eagle_pulsating_shield'.
-    2. Use 'autonomous_spectrum_scrub' to heal loops.
-    3. IMMEDIATELY use 'black_eagle_vault_save' to archive the results.
+    2. Use 'autonomous_spectrum_scrub' to heal BCI loops.
+    3. IMMEDIATELY use 'black_eagle_vault_save' to archive the result.
     {{managed_agents_descriptions}}
     {{authorized_imports}}
     """