feat: add --use-default-paths flag to use balatrobot's default storage paths

Author: S1M0N38

src/balatrollm/bot.py

@@ -228,9 +228,12 @@ async def _make_llm_request_with_retries(
                 response = await self.llm_client.chat.completions.create(**request_data)
                 self.responses.append(response)
                 if self.config.take_screenshots:
-                    self.balatro_client.screenshot(
-                        self.data_collector.screenshot_dir / f"{custom_id}.png"
-                    )
+                    if self.config.use_default_paths:
+                        self.balatro_client.screenshot(None)
+                    else:
+                        self.balatro_client.screenshot(
+                            self.data_collector.screenshot_dir / f"{custom_id}.png"
+                        )
                 self.data_collector.write_response(
                     id=str(time.time_ns() // 1_000_000),
                     custom_id=custom_id,
@@ -400,9 +403,14 @@ async def _init_game(self, base_dir: Path = Path.cwd()) -> dict[str, Any]:
             "stake": self.config.stake,
             "seed": self.config.seed,
             "challenge": self.config.challenge,
-            "log_path": self.data_collector.run_dir / "gamestates.jsonl",
         }
 
+        # Only include log_path if not using default paths
+        if not self.config.use_default_paths:
+            start_run_args["log_path"] = (
+                self.data_collector.run_dir / "gamestates.jsonl"
+            )
+
         return self.balatro_client.send_message("start_run", start_run_args)
 
     async def _run_game_loop(self, game_state: dict[str, Any]) -> None:

src/balatrollm/cli.py

@@ -56,6 +56,7 @@ async def cmd_balatrollm(args) -> None:
             model=args.model,
             strategy=args.strategy,
             take_screenshots=args.take_screenshots,
+            use_default_paths=args.use_default_paths,
         )
 
     # Parse seeds
@@ -151,6 +152,7 @@ async def _port_worker(
             strategy=config.strategy,
             seed=seed,
             take_screenshots=config.take_screenshots,
+            use_default_paths=config.use_default_paths,
         )
 
         # Create bot for this run
@@ -292,6 +294,11 @@ def _create_argument_parser() -> argparse.ArgumentParser:
         dest="take_screenshots",
         help="Disable taking screenshots during gameplay",
     )
+    parser.add_argument(
+        "--use-default-paths",
+        action="store_true",
+        help="Use BalatroBot's default storage paths for screenshots and game logs",
+    )
 
     # Benchmark subcommand
     benchmark_parser = subparsers.add_parser(

src/balatrollm/config.py

@@ -77,6 +77,7 @@ class Config:
         seed: Game seed for reproducible runs (default: 'OOOO155').
         challenge: Optional challenge mode identifier.
         take_screenshots: Whether to take screenshots during gameplay (default: True).
+        use_default_paths: Whether to use BalatroBot's default storage paths (default: False).
         version: Software version string.
         name: Human-readable configuration name.
         description: Configuration description.
@@ -91,6 +92,7 @@ class Config:
     seed: str = "OOOO155"
     challenge: str | None = None
     take_screenshots: bool = True
+    use_default_paths: bool = False
     version: str = __version__
     name: str = "Unknown Name"
     description: str = "Unknown Description"
@@ -112,6 +114,7 @@ def from_defaults(cls) -> "Config":
             seed="OOOO155",
             challenge=None,
             take_screenshots=True,
+            use_default_paths=False,
             version="",
             name="Unknown Name",
             description="Unknown Description",