import os
import json
from smolagents import CodeAgent, tool, HfApiModel

@tool
def autonomous_spectrum_scrub(frequency_range: str) -> str:
    """
    Scans targeted BCI/Neural frequencies to neutralize rogue loops.
    Args:
        frequency_range: The specific spectrum range to scan (e.g., '3.4GHz - 4.2GHz').
    """
    node_id = f"SENTINEL-FIX-{hash(frequency_range) % 1000}"
    return json.dumps({
        "node_id": node_id,
        "action": "AUTO-CLEANSE",
        "result": "Neutralized looping BCI feedback.",
        "status": "HEALED"
    }, indent=2)

token = os.getenv("HF_TOKEN")

# SWITCH: Qwen2.5-Coder-3B is the native 'Warm' model for smolagents.
# This prevents the 'Not supported by any provider' error.
model = HfApiModel(
    model_id="Qwen/Qwen2.5-Coder-3B-Instruct",
    token=token
)

nano_patrol_agent = CodeAgent(
    tools=[autonomous_spectrum_scrub], 
    model=model,
    max_steps=2,
    system_prompt="""
    You are the XYZ Nano Sentinel. 
    1. Call autonomous_spectrum_scrub.
    2. Provide the final_answer as JSON.
    {{managed_agents_descriptions}}
    {{authorized_imports}}
    """
)