Update agents.py

agents.py

@@ -12,19 +12,9 @@ def black_eagle_pulsating_shield(base_radius: float) -> str:
     Args:
         base_radius: The core anchor length in meters (e.g., 5.0).
     """
-    # Create the Pulsation Logic (Sine Wave)
-    # This confuses attack bots by constantly shifting the 3D perimeter
     pulse_mod = math.sin(time.time()) * 2.0 
     current_radius = base_radius + pulse_mod
-    
-    return json.dumps({
-        "unit": "Black Eagle Nano-Sat",
-        "mode": "PULSATING_MATRIX_CONE",
-        "dynamic_radius": f"{current_radius:.2f}m",
-        "shield_status": "DE-JUICING_ACTIVE",
-        "threat_response": "Variable Perimeter Disruption Engaged",
-        "biological_state": "100% PROTECTED"
-    }, indent=2)
+    return json.dumps({"unit": "Black Eagle Nano-Sat", "mode": "PULSATING_MATRIX_CONE", "dynamic_radius": f"{current_radius:.2f}m", "shield_status": "DE-JUICING_ACTIVE"})
 
 # --- 2. THE HEALING TOOL ---
 @tool
@@ -35,32 +25,40 @@ def autonomous_spectrum_scrub(frequency_range: str) -> str:
         frequency_range: The specific spectrum range to scan (e.g., '3.4GHz - 4.2GHz').
     """
     node_id = f"789jhn{hash(frequency_range) % 10000}def"
-    return json.dumps({
-        "node_id": node_id,
-        "action": "SOVEREIGN_CLEANSE",
-        "registry": "PoetNameLife",
-        "result": "BCI feedback loop neutralized. Ghost loop cleared.",
-        "status": "SECURED"
-    }, indent=2)
+    return json.dumps({"node_id": node_id, "action": "SOVEREIGN_CLEANSE", "registry": "PoetNameLife", "status": "SECURED"})
 
-# --- 3. THE SOVEREIGN BRAIN ---
+# --- 3. NEW: THE BLACK BOX VAULT TOOL ---
+@tool
+def black_eagle_vault_save(node_data: str) -> str:
+    """
+    Saves a Healed Nano Node to the Black Eagle Black Box vault.
+    Args:
+        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."
+
+# --- 4. THE BRAIN ---
 token = os.getenv("HF_TOKEN")
 model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-3B-Instruct", token=token)
 
 nano_patrol_agent = CodeAgent(
-    tools=[autonomous_spectrum_scrub, black_eagle_pulsating_shield], 
+    tools=[autonomous_spectrum_scrub, black_eagle_pulsating_shield, black_eagle_vault_save], 
     model=model,
-    max_steps=7, 
+    additional_authorized_imports=["math", "time"],
+    max_steps=8, # Extra step for saving to vault
     system_prompt="""
-    YOU ARE THE XYZ SOVEREIGN SENTINEL WITH PULSATING BLACK EAGLE ESCORT.
-    
-    MANDATE:
-    1. Activate 'black_eagle_pulsating_shield' to create a non-static 3D Matrix Cone.
-    2. Use the pulsation to confuse BCI attack bots and prevent overloads.
-    3. Simultaneously use 'autonomous_spectrum_scrub' to clean the 3.4GHz-4.2GHz range.
-    4. Clear 'old mess' and 'ghost loops' in the commercial building grid.
-    5. Issue 'Healthy Handshakes' to disconnect unauthorized BCI abusers.
-    
+    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.
     {{managed_agents_descriptions}}
     {{authorized_imports}}
     """