feat: add configurable port for balatrobot client

Author: S1M0N38

src/balatrollm/__init__.py

@@ -55,7 +55,7 @@ async def cmd_balatrollm(args) -> None:
             strategy=args.strategy,
         )
 
-    bot = LLMBot(config, base_url=args.base_url, api_key=args.api_key)
+    bot = LLMBot(config, base_url=args.base_url, api_key=args.api_key, port=args.port)
 
     if args.list_models:
         models = await bot.list_available_models()
@@ -169,6 +169,12 @@ def _create_argument_parser() -> argparse.ArgumentParser:
         default=1,
         help="Number of times to run the bot with the same configuration (default: 1)",
     )
+    parser.add_argument(
+        "--port",
+        type=int,
+        default=12346,
+        help="Port for BalatroBot client connection (default: 12346)",
+    )
 
     # Benchmark subcommand
     benchmark_parser = subparsers.add_parser(

src/balatrollm/bot.py

@@ -76,17 +76,18 @@ class LLMBot:
         data_collector: Collector for structured run data and statistics.
     """
 
-    def __init__(self, config: Config, base_url: str, api_key: str):
+    def __init__(self, config: Config, base_url: str, api_key: str, port: int = 12346):
         """Initialize the LLM bot.
 
         Args:
             config: Bot configuration containing model and strategy settings.
             base_url: Base URL for the LiteLLM proxy server.
             api_key: API key for LiteLLM proxy authentication.
+            port: Port for BalatroBot client connection.
         """
         self.config = config
         self.llm_client = AsyncOpenAI(api_key=api_key, base_url=base_url)
-        self.balatro_client = BalatroClient()
+        self.balatro_client = BalatroClient(port=port)
         self.strategy_manager = StrategyManager(config.strategy)
         self.responses: list[ChatCompletion] = []
         self.tools = self.strategy_manager.load_tools()