refactor(examples): standardize imports and improve type annotations

- Update example and replay bots to use balatrobot imports
- Add proper return type annotation to connect() method
- Improve send_message() parameter handling with None default

Co-Authored-By: Claude <[email protected]>

Author: S1M0N38

bots/example.py

@@ -2,8 +2,8 @@
 
 import logging
 
-from src.balatrobot.client import BalatroClient
-from src.balatrobot.exceptions import BalatroError
+from balatrobot.client import BalatroClient
+from balatrobot.exceptions import BalatroError
 
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)

bots/replay.py

@@ -5,8 +5,8 @@
 import sys
 from pathlib import Path
 
-from src.balatrobot.client import BalatroClient
-from src.balatrobot.exceptions import BalatroError, ConnectionFailedError
+from balatrobot.client import BalatroClient
+from balatrobot.exceptions import BalatroError, ConnectionFailedError
 
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.INFO)

src/balatrobot/client.py

@@ -45,7 +45,7 @@ def __exit__(self, exc_type, exc_val, exc_tb) -> None:
         """Exit context manager and disconnect from the game."""
         self.disconnect()
 
-    def connect(self):
+    def connect(self) -> None:
         """Connect to Balatro TCP server
 
         Raises:
@@ -82,7 +82,7 @@ def disconnect(self) -> None:
             self._socket = None
         self._connected = False
 
-    def send_message(self, name: str, arguments: dict = {}) -> dict:
+    def send_message(self, name: str, arguments: dict | None = None) -> dict:
         """Send JSON message to Balatro and receive response
 
         Args:
@@ -96,6 +96,9 @@ def send_message(self, name: str, arguments: dict = {}) -> dict:
             ConnectionFailedError: If not connected to the game
             BalatroError: If the API returns an error
         """
+        if arguments is None:
+            arguments = {}
+
         if not self._connected or not self._socket:
             raise ConnectionFailedError(
                 "Not connected to the game API",