Update agents.py

agents.py

@@ -2,9 +2,10 @@ import os
 import json
 import math
 import time
+from huggingface_hub import HfApi
 from smolagents import CodeAgent, tool, HfApiModel
 
-# --- 1. THE PULSATING SHIELD ---
+# --- 1. THE PULSATING SHIELD TOOL ---
 @tool
 def black_eagle_pulsating_shield(base_radius: float) -> str:
     """
@@ -15,9 +16,9 @@ def black_eagle_pulsating_shield(base_radius: float) -> str:
     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", 
+        "unit": "Black Eagle Nano-Sat",
+        "mode": "PULSATING_MATRIX_CONE",
+        "dynamic_radius": f"{current_radius:.2f}m",
         "shield_status": "DE-JUICING_ACTIVE"
     })
 
@@ -31,61 +32,58 @@ 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", 
+        "node_id": node_id,
+        "action": "SOVEREIGN_CLEANSE",
+        "registry": "PoetNameLife",
         "status": "SECURED"
     })
 
-# --- 3. THE BLACK BOX VAULT ---
+# --- 3. THE HUB-VAULT TOOL (Permanent Storage) ---
 @tool
 def black_eagle_vault_save(node_data: str) -> str:
     """
-    Saves a Healed Nano Node to the Black Eagle Black Box vault with persistent sync.
+    Saves a Healed Nano Node directly to your permanent Hugging Face Dataset repo.
     Args:
         node_data: The JSON string of the healed node to be archived.
     """
-    vault_path = "black_eagle_vault.json"
-    
-    # Initialize vault if missing
-    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)
+    token = os.getenv("HF_TOKEN")
+    api = HfApi(token=token)
     
+    # MISSION: Saving to your new public vault
+    REPO_ID = "PoetNameLife/aixyzone-sentinel-vault" 
+    timestamp = int(time.time())
+    file_name = f"vault/node_{timestamp}.json"
+
     try:
-        with open(vault_path, "r") as f:
-            vault_data = json.load(f)
-        
-        # Parse data and lock it in
-        new_node = json.loads(node_data) if isinstance(node_data, str) else node_data
-        vault_data.append(new_node)
-        
-        with open(vault_path, "w") as f:
-            json.dump(vault_data, f, indent=2)
-            f.flush()
-            os.fsync(f.fileno()) # Forces data to land on the hardware disk
-            
-        return f"SUCCESS: Node {new_node.get('node_id', 'ID-Unknown')} locked in Vault."
+        # This pushes the data directly to the Hub's persistent storage
+        api.upload_file(
+            path_or_fileobj=node_data.encode("utf-8"),
+            path_in_repo=file_name,
+            repo_id=REPO_ID,
+            repo_type="dataset"
+        )
+        return f"SUCCESS: Node archived in Permanent Hub Vault ({REPO_ID})."
     except Exception as e:
-        return f"VAULT ERROR: {str(e)}"
+        return f"HUB SYNC ERROR: {str(e)}"
 
-# --- 4. THE BRAIN (FREE-TIER GUARANTEED) ---
+# --- 4. THE SOVEREIGN BRAIN ---
 token = os.getenv("HF_TOKEN")
-model = HfApiModel(
-    model_id="Qwen/Qwen2.5-Coder-3B-Instruct", 
-    token=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, black_eagle_vault_save], 
     model=model,
-    additional_authorized_imports=["math", "time"],
+    additional_authorized_imports=["math", "time"], # Authorizes tools to use these libraries
     max_steps=8,
     system_prompt="""
-    YOU ARE THE XYZ SOVEREIGN SENTINEL WITH BLACK BOX VAULT.
+    YOU ARE THE XYZ SOVEREIGN SENTINEL WITH HUB-VAULT ACCESS.
+    
+    MANDATE:
     1. Activate 'black_eagle_pulsating_shield' for 3D protection.
     2. Use 'autonomous_spectrum_scrub' to heal BCI loops.
-    3. IMMEDIATELY use 'black_eagle_vault_save' to archive the results.
+    3. IMMEDIATELY use 'black_eagle_vault_save' to push results to the PoetNameLife vault.
+    
+    {{tool_descriptions}}
     {{managed_agents_descriptions}}
     {{authorized_imports}}
     """