Update agents.py

agents.py

@@ -88,25 +88,46 @@ class LLMAgentBase(Player):
 class TemplateAgent(LLMAgentBase):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
-        openai.api_key = os.getenv("OPENAI_API_KEY")  # Store key in secret later
+        self.api_url = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
+        self.headers = {
+            "Authorization": f"Bearer {os.getenv('HF_TOKEN')}",
+            "Content-Type": "application/json"
+        }
 
     async def _get_llm_decision(self, battle_state: str) -> Dict[str, Any]:
+        prompt = (
+            "You are a Pokémon battle agent. Based on the battle state, "
+            "decide which move to use or which Pokémon to switch to.\n"
+            "Respond ONLY in JSON format like this:\n"
+            '{"decision": {"name": "choose_move", "arguments": {"move_name": "Flamethrower"}}}\n'
+            "OR\n"
+            '{"decision": {"name": "choose_switch", "arguments": {"pokemon_name": "Pikachu"}}}\n\n"
+            f"Battle state:\n{battle_state}"
+        )
+
+        payload = {
+            "inputs": prompt,
+            "parameters": {"temperature": 0.7, "max_new_tokens": 256}
+        }
+
         try:
-            response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo",
-                messages=[
-                    {"role": "system", "content": "You are a Pokémon battle strategist."},
-                    {"role": "user", "content": f"{battle_state}\n\nChoose move using one of these functions: choose_move(move_name='...') or choose_switch(pokemon_name='...')"}
-                ],
-                functions=list(self.standard_tools.values()),
-                function_call="auto"
-            )
-            choice = response["choices"][0]["message"]["function_call"]
-            return {
-                "decision": {
-                    "name": choice["name"],
-                    "arguments": eval(choice["arguments"])  # Simple for now
-                }
-            }
+            import requests
+            response = requests.post(self.api_url, headers=self.headers, json=payload)
+            if response.status_code != 200:
+                return {"error": f"HTTP {response.status_code}: {response.text}"}
+
+            raw = response.json()
+            output = raw[0]["generated_text"]
+
+            # crude parse for now — better to use regex or json.loads w/ validation
+            if "choose_move" in output:
+                move_name = output.split('"move_name":')[1].split('"')[1]
+                return {"decision": {"name": "choose_move", "arguments": {"move_name": move_name}}}
+            elif "choose_switch" in output:
+                poke_name = output.split('"pokemon_name":')[1].split('"')[1]
+                return {"decision": {"name": "choose_switch", "arguments": {"pokemon_name": poke_name}}}
+            else:
+                return {"error": "Unrecognized response"}
+
         except Exception as e:
-            return {"error": str(e)}
+            return {"error": f"Exception: {str(e)}"}