feat: add app version to metrics (#313)

fal-ai · Sep 23, 2024 · b332d9a · b332d9a
1 parent 2ca1786
commit b332d9a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/projects/fal/src/fal/api.py b/projects/fal/src/fal/api.py
@@ -971,6 +971,8 @@ class RouteSignature(NamedTuple):
 
 
 class BaseServable:
+    version: ClassVar[str] = "unknown"
+
     def collect_routes(self) -> dict[RouteSignature, Callable[..., Any]]:
         raise NotImplementedError
 
@@ -1099,9 +1101,14 @@ def openapi(self) -> dict[str, Any]:
     def serve(self) -> None:
         import asyncio
 
+        from prometheus_client import Gauge
         from starlette_exporter import handle_metrics
         from uvicorn import Config
 
+        # NOTE: this uses the global prometheus registry
+        app_info = Gauge("fal_app_info", "Fal application information", ["version"])
+        app_info.labels(version=self.version).set(1)
+
         app = self._build_app()
         server = Server(
             config=Config(app, host="0.0.0.0", port=8080, timeout_keep_alive=300)

diff --git a/projects/fal/src/fal/app.py b/projects/fal/src/fal/app.py
@@ -189,7 +189,6 @@ class App(fal.api.BaseServable):
     }
     app_name: ClassVar[str]
     app_auth: ClassVar[Literal["private", "public", "shared"]] = "private"
-    version: ClassVar[str] = "unknown"
     request_timeout: ClassVar[int | None] = None
 
     def __init_subclass__(cls, **kwargs):