feat(fal): support SETUP runner state (#616)

feat(fal): suppor SETUP runner state

Author: efiop

projects/fal/pyproject.toml

@@ -23,7 +23,7 @@ readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [
     "isolate[build]>=0.18.0,<0.20.0",
-    "isolate-proto>=0.12.0,<0.13.0",
+    "isolate-proto>=0.14.1,<0.15.0",
     "grpcio>=1.64.0,<2",
     "dill==0.3.7",
     "cloudpickle==3.0.0",

projects/fal/src/fal/cli/apps.py

@@ -297,8 +297,14 @@ def _runners(args):
     pending_runners = [
         runner for runner in alias_runners if runner.state == RunnerState.PENDING
     ]
-    args.console.print(f"Runners: {len(alias_runners) - len(pending_runners)}")
-    args.console.print(f"Pending Runners: {len(pending_runners)}")
+    setup_runners = [
+        runner for runner in alias_runners if runner.state == RunnerState.SETUP
+    ]
+    args.console.print(
+        f"Runners: {len(alias_runners) - len(pending_runners) - len(setup_runners)}"
+    )
+    args.console.print(f"Runners Pending: {len(pending_runners)}")
+    args.console.print(f"Runners Setting Up: {len(setup_runners)}")
     # Drop the alias column, which is the first column
     runners_table.columns.pop(0)
     args.console.print(runners_table)

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

@@ -81,7 +81,7 @@ def _kill(args):
         connection.kill_runner(args.id)
 
 
-def _list_json(args, runners: list[RunnerInfo], pending_runners: list[RunnerInfo]):
+def _list_json(args, runners: list[RunnerInfo]):
     json_runners = [
         {
             "alias": r.alias,
@@ -108,16 +108,23 @@ def _list(args):
         pending_runners = [
             runner for runner in runners if runner.state == RunnerState.PENDING
         ]
+        setup_runners = [
+            runner for runner in runners if runner.state == RunnerState.SETUP
+        ]
         if args.output == "pretty":
-            args.console.print(f"Runners: {len(runners) - len(pending_runners)}")
-            args.console.print(f"Pending Runners: {len(pending_runners)}")
+            args.console.print(
+                "Runners: "
+                + str(len(runners) - len(pending_runners) - len(setup_runners))
+            )
+            args.console.print(f"Runners Pending: {len(pending_runners)}")
+            args.console.print(f"Runners Setting Up: {len(setup_runners)}")
             args.console.print(runners_table(runners))
 
             requests_table = runners_requests_table(runners)
             args.console.print(f"Requests: {len(requests_table.rows)}")
             args.console.print(requests_table)
         elif args.output == "json":
-            _list_json(args, runners, pending_runners)
+            _list_json(args, runners)
         else:
             raise AssertionError(f"Invalid output format: {args.output}")
 

projects/fal/src/fal/sdk.py

@@ -264,6 +264,7 @@ class AliasInfo:
 class RunnerState(Enum):
     RUNNING = "running"
     PENDING = "pending"
+    SETUP = "setup"
     UNKNOWN = "unknown"
 
     @staticmethod
@@ -272,6 +273,8 @@ def from_proto(proto: isolate_proto.RunnerInfo.State) -> RunnerState:
             return RunnerState.RUNNING
         elif proto is isolate_proto.RunnerInfo.State.PENDING:
             return RunnerState.PENDING
+        elif proto is isolate_proto.RunnerInfo.State.SETUP:
+            return RunnerState.SETUP
         else:
             return RunnerState.UNKNOWN