alessandro trinca tornidor commited on
Commit
cd399d1
·
1 Parent(s): 1c71bac

feat: add log version (gradio and samgis-core) to /health endpoint

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. lisa_on_cuda/routes.py +5 -2
app.py CHANGED
@@ -15,6 +15,7 @@ from fastapi.templating import Jinja2Templates
15
  from samgis_core.utilities import create_folders_if_not_exists
16
  from samgis_core.utilities import frontend_builder
17
  from samgis_core.utilities.session_logger import setup_logging
 
18
  from spaces import GPU as SPACES_GPU
19
 
20
  from lisa_on_cuda.utils import app_helpers
 
15
  from samgis_core.utilities import create_folders_if_not_exists
16
  from samgis_core.utilities import frontend_builder
17
  from samgis_core.utilities.session_logger import setup_logging
18
+ # for some reason this do
19
  from spaces import GPU as SPACES_GPU
20
 
21
  from lisa_on_cuda.utils import app_helpers
lisa_on_cuda/routes.py CHANGED
@@ -19,8 +19,11 @@ router = APIRouter()
19
  @router.get("/health")
20
  def health() -> str:
21
  try:
22
- app_logger.info("health check")
23
- return json.dumps({"msg": "ok"})
 
 
 
24
  except Exception as e:
25
  app_logger.error(f"exception:{e}.")
26
  return json.dumps({"msg": "request failed"})
 
19
  @router.get("/health")
20
  def health() -> str:
21
  try:
22
+ from samgis_core.__version__ import __version__ as version_core
23
+ from gradio import __version__ as gradio_version
24
+
25
+ app_logger.info(f"still alive, gradio_version:{gradio_version}, version_core:{version_core}.")
26
+ return json.dumps({"msg": "lisa on cuda: still alive..."})
27
  except Exception as e:
28
  app_logger.error(f"exception:{e}.")
29
  return json.dumps({"msg": "request failed"})