mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-30 12:22:26 -08:00
Code Issues Projects Releases Packages Wiki Activity

added status check endpoint (#42)

Browse source
This commit is contained in:
Aasher Kataria 2024-11-09 16:55:34 -06:00 • committed by GitHub
parent 0217486799
commit 4aa13863d8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
modules/api/api.py
View file

@ -209,6 +209,10 @@ class Api:
self.queue_lock = queue_lock
api_middleware(self.app)
self.add_api_route("/sdapi/v1/txt2img", self.text2imgapi, methods=["POST"], response_model=models.TextToImageResponse)
# Add health check endpoint
self.add_api_route("/sdapi/v1/status", self.health_check, methods=["GET"])
self.add_api_route("/sdapi/v1/img2img", self.img2imgapi, methods=["POST"], response_model=models.ImageToImageResponse)
self.add_api_route("/sdapi/v1/extra-single-image", self.extras_single_image_api, methods=["POST"], response_model=models.ExtrasSingleImageResponse)
self.add_api_route("/sdapi/v1/extra-batch-images", self.extras_batch_images_api, methods=["POST"], response_model=models.ExtrasBatchImagesResponse)
@ -270,6 +274,9 @@ class Api:
self.default_script_arg_img2img = self.init_default_script_args(img2img_script_runner)
# Health check endpoint method
async def health_check(self):
return JSONResponse(content={"status": "Server is running"}, status_code=200)
def add_api_route(self, path: str, endpoint, **kwargs):
if shared.cmd_opts.api_auth: