remove typings

agents.py

@@ -15,9 +15,6 @@ from mistralai import Mistral
 
 # --- 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
 
 # --- Helper Function & Base Class ---
@@ -101,7 +98,7 @@ class LLMAgentBase(Player):
         self.standard_tools = STANDARD_TOOL_SCHEMA
         self.battle_history = []
 
-    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))}) " \
@@ -147,7 +144,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: str) -> Optional[Move]:
         normalized_name = normalize_name(move_name)
         # Prioritize exact ID match
         for move in battle.available_moves:
@@ -160,7 +157,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: str) -> Optional[Pokemon]:
         normalized_name = normalize_name(pokemon_name)
         for pkmn in battle.available_switches:
             # Normalize the species name for comparison
@@ -168,7 +165,7 @@ class LLMAgentBase(Player):
                 return pkmn
         return None
 
-    async def choose_move(self, battle: Battle) -> str:
+    async def choose_move(self, battle) -> str:
         battle_state_str = self._format_battle_state(battle)
         decision_result = await self._get_llm_decision(battle_state_str)
         print(decision_result)