Update agents.py

agents.py

@@ -17,7 +17,6 @@ import time
 
 # --- Poke-Env ---
 from poke_env.player import Player
-from poke_env.environment.battle import Battle
 from poke_env.environment.move import Move
 from poke_env.environment.pokemon import Pokemon
 from typing import Optional, Dict, Any, Union
@@ -121,7 +120,7 @@ class LLMAgentBase(Player):
         self.battle_history = []
         self.battle_delay = battle_delay
 
-    def _format_battle_state(self, battle: Battle) -> str:
+    def _format_battle_state(self, battle) -> str:
         active_pkmn = battle.active_pokemon
         active_pkmn_info = f"Your active Pokemon: {active_pkmn.species} " \
                            f"(Type: {'/'.join(map(str, active_pkmn.types))}) " \
@@ -167,7 +166,7 @@ class LLMAgentBase(Player):
                     f"Opponent Side Conditions: {battle.opponent_side_conditions}"
         return state_str.strip()
 
-    def _find_move_by_name(self, battle: Battle, move_name: str) -> Optional[Move]:
+    def _find_move_by_name(self, battle, move_name) -> Optional[Move]:
         normalized_name = normalize_name(move_name)
         # Prioritize exact ID match
         for move in battle.available_moves:
@@ -180,7 +179,7 @@ class LLMAgentBase(Player):
                 return move
         return None
 
-    def _find_pokemon_by_name(self, battle: Battle, pokemon_name: str) -> Optional[Pokemon]:
+    def _find_pokemon_by_name(self, battle, pokemon_name) -> Optional[Pokemon]:
         normalized_name = normalize_name(pokemon_name)
         for pkmn in battle.available_switches:
             # Normalize the species name for comparison
@@ -188,7 +187,7 @@ class LLMAgentBase(Player):
                 return pkmn
         return None
 
-    async def choose_move(self, battle: Battle) -> str:
+    async def choose_move(self, battle) -> str:
         #time.sleep(self.battle_delay)
 
         battle_state_str = self._format_battle_state(battle)