fix: include ActionSchema in __init__.py exporting

Author: S1M0N38

bots/example.py

@@ -7,8 +7,7 @@
 import itertools
 from typing import Any, Iterator
 
-from balatrobot import Actions, Bot, Decks, Stakes
-from balatrobot.base import ActionSchema
+from balatrobot import Actions, ActionSchema, Bot, Decks, Stakes
 
 # Predefined sequence of actions using the ActionSchema format
 plays: Iterator[ActionSchema] = itertools.cycle(

docs/bot-development.md

@@ -117,8 +117,7 @@ a predefined sequence of actions.
 import itertools
 from typing import Any, Iterator
 
-from balatrobot import Actions, Bot, Decks, Stakes
-from balatrobot.base import ActionSchema
+from balatrobot import ActionSchema, Actions, Bot, Decks, Stakes
 
 # Predefined sequence of actions using the ActionSchema format
 plays: Iterator[ActionSchema] = itertools.cycle(
@@ -545,7 +544,7 @@ Always use proper type hints for better code quality:
 
 ```python
 from typing import Any
-from balatrobot.base import ActionSchema
+from balatrobot import ActionSchema
 
 def select_shop_action(self, env: dict[str, Any]) -> ActionSchema:
     """Type hints help catch errors early."""

src/balatrobot/__init__.py

@@ -1,7 +1,8 @@
-from .base import Bot
+from .base import ActionSchema, Bot
 from .enums import Actions, Decks, Stakes, State
 
 __all__ = [
+    "ActionSchema",
     "Actions",
     "Bot",
     "Decks",