Added healthcheck handler support to RPC framework

Author: m-messer

lf_toolkit/io/handler.py

@@ -57,6 +57,10 @@ async def handle_preview(self, req: dict):
 
         return await self._call_user_handler("preview", response, request_params)
 
+    async def handle_healthcheck(self, req: dict):
+        from .healthcheck import run_healthcheck
+        return await anyio.to_thread.run_sync(run_healthcheck)
+
     async def handle(self, name: Command, req: dict) -> dict:
         handler = getattr(self, f"handle_{name}", None)
 

lf_toolkit/io/rpc_handler.py

@@ -12,7 +12,7 @@ class JsonRpcHandler(Handler):
 
     def __init__(self):
         self._methods = {
-            name: jsonrpc_handler(self, name) for name in ["eval", "preview"]
+            name: jsonrpc_handler(self, name) for name in ["eval", "preview", "healthcheck"]
         }
 
     async def dispatch(self, req: str) -> str: