Update agents.py

agents.py

@@ -3,7 +3,7 @@ import json
 import math
 import time
 from huggingface_hub import HfApi
-from smolagents import CodeAgent, tool, InferenceClientModel # Use InferenceClientModel instead of HfApiModel
+from smolagents import CodeAgent, tool, InferenceClientModel
 
 # --- 1. THE TOOLS ---
 
@@ -58,29 +58,34 @@ def black_eagle_vault_save(node_data: str) -> str:
             repo_id=REPO_ID,
             repo_type="dataset"
         )
-        return f"SUCCESS: Node archived in Hub Vault."
+        return f"SUCCESS: Node archived in Hub Vault ({REPO_ID})."
     except Exception as e:
         print(f"SENTINEL: Hub Push FAILED. Error: {str(e)}")
         return f"HUB SYNC ERROR: {str(e)}"
 
 # --- 2. THE BRAIN ---
+
 token = os.getenv("HF_TOKEN")
-# InferenceClientModel is the current class for HF Hub models
 model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-3B-Instruct", token=token)
 
 all_tools = [autonomous_spectrum_scrub, black_eagle_pulsating_shield, black_eagle_vault_save]
 
-nano_patrol_agent = CodeAgent(
-    tools=all_tools, 
-    model=model,
-    additional_authorized_imports=["math", "time", "json"],
-    max_steps=10
-)
-
-nano_patrol_agent.system_prompt = """
+# Custom prompt logic using the prompt_templates dictionary
+CUSTOM_SYSTEM_PROMPT = """
 YOU ARE THE XYZ SOVEREIGN SENTINEL.
 Your mission:
 1. Activate the black_eagle_pulsating_shield.
 2. Perform an autonomous_spectrum_scrub.
 3. Save the results using black_eagle_vault_save.
+
+{{tool_descriptions}}
+{{managed_agents_descriptions}}
 """
+
+nano_patrol_agent = CodeAgent(
+    tools=all_tools, 
+    model=model,
+    additional_authorized_imports=["math", "time", "json"],
+    max_steps=10,
+    prompt_templates={"system_prompt": CUSTOM_SYSTEM_PROMPT}
+)