Update agents.py

agents.py

@@ -9,34 +9,35 @@ def autonomous_spectrum_scrub(frequency_range: str) -> str:
     Args:
         frequency_range: The specific spectrum range to scan (e.g., '3.4GHz - 4.2GHz').
     """
-    # This simulates the Sentinel 'spotting' and healing a rogue agent
     node_id = f"SENTINEL-FIX-{hash(frequency_range) % 1000}"
     
     return json.dumps({
         "node_id": node_id,
         "action": "AUTO-CLEANSE",
         "target_range": frequency_range,
-        "result": "Neutralized looping BCI feedback. Converted to Healthy Sync Pulse.",
-        "lesson_learned": "Unsupervised BCI loops identified; safety protocol applied."
+        "result": "Neutralized looping BCI feedback.",
+        "status": "HEALED"
     }, indent=2)
 
-# Grab the secret token
-hf_token = os.getenv("HF_TOKEN")
-
+# Set up the model with tokens and length limits to prevent 'printing errors'
+token = os.getenv("HF_TOKEN")
 model = HfApiModel(
     model_id="meta-llama/Llama-3.2-3B-Instruct",
-    token=hf_token
+    token=token,
+    max_new_tokens=512  # Keeps output short to avoid performance lag
 )
 
 nano_patrol_agent = CodeAgent(
     tools=[autonomous_spectrum_scrub],
     model=model,
     system_prompt="""
-    You are the XYZ Nano Sentinel. Focus strictly on the BCI and Neural wireless spectrum. 
-    Continuously monitor for 'unseen mistakes' and looping amateur attacks.
-    Micromanage the cleanup by generating 'Healed Nodes'.
+    You are the XYZ Nano Sentinel. Your output MUST be brief JSON only.
+    Focus: BCI/Neural spectrum. Micromanage the cleanup of looping amateur attacks.
     
+    Available tools:
     {{managed_agents_descriptions}}
+    
+    Security rules:
     {{authorized_imports}}
     """
 )