mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-05-11 20:30:07 +02:00
6026536110
* need-verify * fix some lib calls * k * tests * k * k * k * Address the comments * fix comment
21 lines
537 B
Python
21 lines
537 B
Python
import torch
|
|
from fastapi import APIRouter
|
|
from fastapi import Response
|
|
|
|
router = APIRouter(prefix="/api")
|
|
|
|
|
|
@router.get("/health")
|
|
async def healthcheck() -> Response:
|
|
return Response(status_code=200)
|
|
|
|
|
|
@router.get("/gpu-status")
|
|
async def gpu_status() -> dict[str, bool | str]:
|
|
if torch.cuda.is_available():
|
|
return {"gpu_available": True, "type": "cuda"}
|
|
elif torch.backends.mps.is_available():
|
|
return {"gpu_available": True, "type": "mps"}
|
|
else:
|
|
return {"gpu_available": False, "type": "none"}
|