feat: Replace runners by default when stopping

projects/fal/src/fal/api/client.py

@@ -192,13 +192,16 @@ def list(self, *, since=None) -> List[RunnerInfo]:
         """
         return runners_api.list_runners(self.client, since=since)
 
-    def stop(self, runner_id: str) -> None:
+    def stop(self, runner_id: str, replace_first: bool = False) -> None:
         """Gracefully stop a runner.
 
         Args:
             runner_id: The ID of the runner to stop.
+            replace_first: Whether to replace the runner before stopping it.
         """
-        return runners_api.stop_runner(self.client, runner_id)
+        return runners_api.stop_runner(
+            self.client, runner_id, replace_first=replace_first
+        )
 
     def kill(self, runner_id: str) -> None:
         """Forcefully kill a runner.

projects/fal/src/fal/api/runners.py

@@ -16,9 +16,11 @@ def list_runners(
         return conn.list_runners(start_time=since)
 
 
-def stop_runner(client: SyncServerlessClient, runner_id: str) -> None:
+def stop_runner(
+    client: SyncServerlessClient, runner_id: str, replace_first: bool = False
+) -> None:
     with FalServerlessClient(client._grpc_host, client._credentials).connect() as conn:
-        conn.stop_runner(runner_id)
+        conn.stop_runner(runner_id, replace_first=replace_first)
 
 
 def kill_runner(client: SyncServerlessClient, runner_id: str) -> None:

projects/fal/src/fal/cli/runners.py

@@ -252,7 +252,7 @@ def stream_inputs():
 
 def _stop(args):
     client = SyncServerlessClient(host=args.host, team=args.team)
-    client.runners.stop(args.id)
+    client.runners.stop(args.id, replace_first=not args.no_replace)
 
 
 def _kill(args):
@@ -356,6 +356,13 @@ def _add_stop_parser(subparsers, parents):
         "id",
         help="Runner ID.",
     )
+    parser.add_argument(
+        "-r",
+        "--replace",
+        action="store_true",
+        default=False,
+        help="Stop the runner after first spawning a replacement.",
+    )
     parser.set_defaults(func=_stop)
 
 

projects/fal/src/fal/sdk.py

@@ -1123,8 +1123,10 @@ def list_secrets(
             for secret in response.secrets
         ]
 
-    def stop_runner(self, runner_id: str) -> None:
-        request = isolate_proto.StopRunnerRequest(runner_id=runner_id)
+    def stop_runner(self, runner_id: str, replace_first: bool = False) -> None:
+        request = isolate_proto.StopRunnerRequest(
+            runner_id=runner_id, replace_first=replace_first
+        )
         self.stub.StopRunner(request)
 
     def kill_runner(self, runner_id: str) -> None: