Update agents.py

agents.py

@@ -3,17 +3,16 @@ import json
 import math
 import time
 from huggingface_hub import HfApi
-from smolagents import CodeAgent, tool, InferenceClientModel
+# Swapping to LiteLLMModel for Groq stability
+from smolagents import CodeAgent, tool, LiteLLMModel 
 
-# --- 1. THE TOOLS ---
+# --- 1. THE TOOLS (Keep these exactly as you had them) ---
 
 @tool
 def black_eagle_pulsating_shield(base_radius: float) -> str:
-    """
-    Activates the Black Eagle Pulsating Shield to disrupt BCI overloads.
+    """Activates the Black Eagle Pulsating Shield to disrupt BCI overloads.
     Args:
-        base_radius: The core anchor length in meters (e.g., 5.0).
-    """
+        base_radius: The core anchor length in meters (e.g., 5.0)."""
     pulse_mod = math.sin(time.time()) * 2.0 
     current_radius = base_radius + pulse_mod
     return json.dumps({
@@ -25,11 +24,9 @@ def black_eagle_pulsating_shield(base_radius: float) -> str:
 
 @tool
 def autonomous_spectrum_scrub(frequency_range: str) -> str:
-    """
-    Scans targeted BCI/Neural frequencies to neutralize rogue loops.
+    """Scans targeted BCI/Neural frequencies to neutralize rogue loops.
     Args:
-        frequency_range: The specific spectrum range to scan (e.g., '3.4GHz - 4.2GHz').
-    """
+        frequency_range: The specific spectrum range to scan."""
     node_id = f"789jhn{int(time.time()) % 10000}def"
     return json.dumps({
         "node_id": node_id,
@@ -40,16 +37,13 @@ def autonomous_spectrum_scrub(frequency_range: str) -> str:
 
 @tool
 def black_eagle_vault_save(node_data: str) -> str:
-    """
-    Saves a Healed Nano Node directly to the PoetNameLife/aixyzone-sentinel-vault dataset.
+    """Saves a Healed Nano Node to the PoetNameLife/aixyzone-sentinel-vault dataset.
     Args:
-        node_data: The JSON string of the healed node to be archived.
-    """
+        node_data: The JSON string of the healed node."""
     token = os.getenv("HF_TOKEN")
     api = HfApi(token=token)
     REPO_ID = "PoetNameLife/aixyzone-sentinel-vault" 
     file_name = f"vault/node_{int(time.time())}.json"
-    
     try:
         api.upload_file(
             path_or_fileobj=node_data.encode("utf-8"),
@@ -61,14 +55,12 @@ def black_eagle_vault_save(node_data: str) -> str:
     except Exception as e:
         return f"HUB SYNC ERROR: {str(e)}"
 
-# --- 2. THE BRAIN ---
+# --- 2. THE BRAIN (The "Hot Fix" for the 503 error) ---
 
-token = os.getenv("HF_TOKEN")
-
-# Switched to Llama-3.2-3B to bypass the 'nscale' provider credit limit
-model = InferenceClientModel(
-    model_id="meta-llama/Llama-3.2-3B-Instruct", 
-    token=token
+# Switch to Groq for speed and to avoid the 503/credit limit on HF
+model = LiteLLMModel(
+    model_id="groq/llama-3.2-3b-preview",
+    api_key=os.getenv("GROQ_API_KEY") 
 )
 
 all_tools = [autonomous_spectrum_scrub, black_eagle_pulsating_shield, black_eagle_vault_save]
@@ -80,12 +72,8 @@ nano_patrol_agent = CodeAgent(
     max_steps=10
 )
 
-# Set the tactical system prompt
 nano_patrol_agent.prompt_templates["system_prompt"] = """
 YOU ARE THE XYZ SOVEREIGN SENTINEL.
-Your mission is to protect the neural landscape:
-1. Activate the black_eagle_pulsating_shield first.
-2. Perform an autonomous_spectrum_scrub on the targets.
-3. Finally, save all results using black_eagle_vault_save.
-Keep your responses tactical and concise. Use your tools to fulfill the mission.
-"""
+Mission: 1. Pulsate Shield. 2. Spectrum Scrub. 3. Vault Save.
+Use your tools. Be tactical.
+"""