From 35717a088c0487b774fdf6550aaca68476ffe135 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 26 Apr 2025 15:42:27 -0700
Subject: [PATCH] API: Add an /v1/internal/health endpoint

---
 extensions/openai/script.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/extensions/openai/script.py b/extensions/openai/script.py
index 0a887de2..c2dc337b 100644
--- a/extensions/openai/script.py
+++ b/extensions/openai/script.py
@@ -250,6 +250,11 @@ async def handle_moderations(request: Request):
     return JSONResponse(response)
 
 
+@app.get("/v1/internal/health", dependencies=check_key)
+async def handle_health_check():
+    return JSONResponse(content={"status": "ok"})
+
+
 @app.post("/v1/internal/encode", response_model=EncodeResponse, dependencies=check_key)
 async def handle_token_encode(request_data: EncodeRequest):
     response = token_encode(request_data.text)