oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-31 12:54:33 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

API: Add an /v1/internal/health endpoint

Browse source
This commit is contained in:
oobabooga 2025-04-26 15:42:27 -07:00
parent bc55feaf3e
commit 35717a088c
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
extensions/openai/script.py
View file

@ -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)