Add --cost-limit parameter to constrain model costs

Author: Your Name

cecli/args.py

@@ -260,6 +260,12 @@ def get_parser(default_config_files, git_root):
         default=3,
         help="Maximum number of retries a model gets on malformed outputs (default: 3)",
     )
+    group.add_argument(
+        "--cost-limit",
+        type=float,
+        default=None,
+        help="Cost limit per session, exceeding this forces prompt confirmation (default: None)",
+    )
     group.add_argument(
         "--file-diffs",
         action=argparse.BooleanOptionalAction,

cecli/coders/base_coder.py

@@ -157,6 +157,7 @@ class Coder:
     last_user_message = ""
     uuid = ""
     model_kwargs = {}
+    cost_multiplier = 1
 
     # Task coordination state variables
     input_running = False
@@ -1580,6 +1581,18 @@ async def run_one(self, user_message, preproc):
             self.reflected_message = None
             self.tool_reflection = False
 
+            if float(self.total_cost) > self.cost_multiplier * nested.getter(
+                self.args, "cost_limit", float("inf")
+            ):
+                if await self.io.confirm_ask(
+                    "You have reached your configured cost limit. Continue?",
+                    group_response="Cost Limit",
+                    explicit_yes_required=True,
+                ):
+                    Coder.cost_multiplier += 1
+                else:
+                    return
+
             async for _ in self.send_message(message):
                 pass