typing

main.py

@@ -16,7 +16,7 @@ from fastapi.staticfiles import StaticFiles
 # --- Imports for poke_env and agents ---
 from poke_env.player import Player
 from poke_env import AccountConfiguration, ServerConfiguration
-from poke_env.environment.battle import Battle
+# from poke_env.environment.battle import Battle
 
 # Import the actual agent classes
 from agents import OpenAIAgent, GeminiAgent, MistralAgent
@@ -50,14 +50,14 @@ if not AVAILABLE_AGENT_NAMES:
 active_agent_name: Optional[str] = None
 active_agent_instance: Optional[Player] = None
 active_agent_task: Optional[asyncio.Task] = None
-current_battle_instance: Optional[Battle] = None
+current_battle_instance = None
 background_task_handle: Optional[asyncio.Task] = None
 
 # --- Create FastAPI app ---
 app = FastAPI(title="Pokemon Battle Livestream")
 
 # --- Helper Functions ---
-def get_active_battle(agent: Player) -> Optional[Battle]:
+def get_active_battle(agent: Player):
     """Returns the first non-finished battle for an agent."""
     if agent and agent._battles:
         active_battles = [b for b in agent._battles.values() if not b.finished]