Update agents.py

agents.py

@@ -17,8 +17,6 @@ import time
 
 # --- Poke-Env ---
 from poke_env.player import Player
-from poke_env.environment.move import Move
-from poke_env.environment.pokemon import Pokemon
 from typing import Optional, Dict, Any, Union
 
 class MaxDamagePlayer(Player):
@@ -166,7 +164,7 @@ class LLMAgentBase(Player):
                     f"Opponent Side Conditions: {battle.opponent_side_conditions}"
         return state_str.strip()
 
-    def _find_move_by_name(self, battle, move_name) -> Optional[Move]:
+    def _find_move_by_name(self, battle, move_name):
         normalized_name = normalize_name(move_name)
         # Prioritize exact ID match
         for move in battle.available_moves:
@@ -179,7 +177,7 @@ class LLMAgentBase(Player):
                 return move
         return None
 
-    def _find_pokemon_by_name(self, battle, pokemon_name) -> Optional[Pokemon]:
+    def _find_pokemon_by_name(self, battle, pokemon_name):
         normalized_name = normalize_name(pokemon_name)
         for pkmn in battle.available_switches:
             # Normalize the species name for comparison