new: allow to set --max-steps and --max-cost to 0 in order to disable those limits

Author: evilsocket

nerve/cli/run.py

@@ -59,12 +59,13 @@ def run(
     ] = False,
     max_steps: t.Annotated[
         int,
-        typer.Option("--max-steps", "-s", help="Maximum number of steps"),
+        typer.Option("--max-steps", "-s", help="Maximum number of steps. Set to 0 to disable."),
     ] = DEFAULT_MAX_STEPS,
     max_cost: t.Annotated[
         float,
         typer.Option(
-            "--max-cost", help="If cost information is available, stop when the cost exceeds this value in USD."
+            "--max-cost",
+            help="If cost information is available, stop when the cost exceeds this value in USD. Set to 0 to disable.",
         ),
     ] = DEFAULT_MAX_COST,
     timeout: t.Annotated[

nerve/runtime/flow.py

@@ -135,12 +135,12 @@ def done(self) -> bool:
         if self.curr_actor_idx >= len(self.actors):
             return True
 
-        if self.max_steps is not None and self.curr_step > self.max_steps:
+        if self.max_steps > 0 and self.curr_step > self.max_steps:
             state.on_max_steps_reached()
             return True
 
         usage = state.get_usage()
-        if self.max_cost is not None and usage.cost is not None and usage.cost > self.max_cost:
+        if self.max_cost > 0 and usage.cost is not None and usage.cost > self.max_cost:
             state.on_max_cost_reached()
             return True
 

nerve/runtime/logging.py

@@ -68,11 +68,16 @@ def log_event_to_terminal(event: Event) -> None:
             data["flow"] = DictWrapper(data["flow"])
 
         max_steps = data["flow"].max_steps
+        steps = f"max steps: {max_steps}" if max_steps > 0 else "unlimited steps"
+
         max_cost = data["flow"].max_cost
+        cost = f"max cost: {max_cost:.2f}$" if max_cost > 0 else "no cost limit"
+
         timeout = data["flow"].timeout
         timeout = f"{timeout}s timeout" if timeout else "no timeout"
+
         conv_window_strategy = data["flow"].actors[0].conv_window_strategy
-        logger.info(f"🚀 max steps: {max_steps} | max cost: {max_cost:.2f}$ | {timeout} | {conv_window_strategy}")
+        logger.info(f"🚀 {steps} | {cost} | {timeout} | {conv_window_strategy}")
 
     elif event.name == "agent_created":
         if isinstance(data["agent"], dict):