Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fixes from check-api

Browse Source 
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2026-04-10 12:41:18 +02:00
parent b9a38b5955
commit 15c8b2b0c4
9 changed files with 65 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/api/api.py
+7 -7
View File
@@ -107,15 +107,15 @@ class Api:
# options api
from modules.api import options
options.register_api(self.app)
options.register_api(self)
# caption api
from modules.api import caption
caption.register_api(self.app)
caption.register_api(self)
# lora api
from modules.api import loras
loras.register_api(self.app)
loras.register_api(self)
# gallery api
from modules.api import gallery
@@ -123,19 +123,19 @@ class Api:
# nudenet api
from modules.api import nudenet
nudenet.register_api(self.app)
nudenet.register_api(self)
# xyz-grid api
from modules.api import xyz_grid
xyz_grid.register_api(self.app)
xyz_grid.register_api(self)
# civitai api
from modules.civitai import api_civitai
api_civitai.register_api(self.app)
api_civitai.register_api(self)
# rembg api
from modules.rembg import rembg_api
rembg_api.register_api(self.app)
rembg_api.register_api(self)
# hide trailing-slash duplicates from OpenAPI schema
from fastapi.routing import APIRoute
modules/api/caption.py
+9 -9
View File
@@ -662,12 +662,12 @@ def get_tagger_models():
# Route Registration
# =============================================================================
def register_api(app):
app.add_api_route("/sdapi/v1/openclip", get_caption, methods=["GET"], response_model=list[str], tags=["Caption"])
app.add_api_route("/sdapi/v1/caption", post_caption_dispatch, methods=["POST"], response_model=ResCaptionDispatch, tags=["Caption"])
app.add_api_route("/sdapi/v1/openclip", post_caption, methods=["POST"], response_model=ResCaption, tags=["Caption"])
app.add_api_route("/sdapi/v1/vqa", post_vqa, methods=["POST"], response_model=ResVQA, tags=["Caption"])
app.add_api_route("/sdapi/v1/vqa/models", get_vqa_models, methods=["GET"], response_model=list[ItemVLMModel], tags=["Caption"])
app.add_api_route("/sdapi/v1/vqa/prompts", get_vqa_prompts, methods=["GET"], response_model=ResVLMPrompts, tags=["Caption"])
app.add_api_route("/sdapi/v1/tagger", post_tagger, methods=["POST"], response_model=ResTagger, tags=["Caption"])
app.add_api_route("/sdapi/v1/tagger/models", get_tagger_models, methods=["GET"], response_model=list[ItemTaggerModel], tags=["Caption"])
def register_api(api):
api.add_api_route("/sdapi/v1/openclip", get_caption, methods=["GET"], response_model=list[str], tags=["Caption"])
api.add_api_route("/sdapi/v1/caption", post_caption_dispatch, methods=["POST"], response_model=ResCaptionDispatch, tags=["Caption"])
api.add_api_route("/sdapi/v1/openclip", post_caption, methods=["POST"], response_model=ResCaption, tags=["Caption"])
api.add_api_route("/sdapi/v1/vqa", post_vqa, methods=["POST"], response_model=ResVQA, tags=["Caption"])
api.add_api_route("/sdapi/v1/vqa/models", get_vqa_models, methods=["GET"], response_model=list[ItemVLMModel], tags=["Caption"])
api.add_api_route("/sdapi/v1/vqa/prompts", get_vqa_prompts, methods=["GET"], response_model=ResVLMPrompts, tags=["Caption"])
api.add_api_route("/sdapi/v1/tagger", post_tagger, methods=["POST"], response_model=ResTagger, tags=["Caption"])
api.add_api_route("/sdapi/v1/tagger/models", get_tagger_models, methods=["GET"], response_model=list[ItemTaggerModel], tags=["Caption"])
modules/api/loras.py
+4 -4
View File
@@ -34,7 +34,7 @@ def _invalidate_extra_networks():
page.create_items('txt2img')
def register_api(app):
app.add_api_route("/sdapi/v1/lora", get_lora, methods=["GET"], response_model=dict, tags=["Enumerators"])
app.add_api_route("/sdapi/v1/loras", get_loras, methods=["GET"], response_model=list[dict], tags=["Enumerators"])
app.add_api_route("/sdapi/v1/refresh-loras", post_refresh_loras, methods=["POST"], tags=["Functional"])
def register_api(api):
api.add_api_route("/sdapi/v1/lora", get_lora, methods=["GET"], response_model=dict, tags=["Enumerators"])
api.add_api_route("/sdapi/v1/loras", get_loras, methods=["GET"], response_model=list[dict], tags=["Enumerators"])
api.add_api_route("/sdapi/v1/refresh-loras", post_refresh_loras, methods=["POST"], tags=["Functional"])
modules/api/nudenet.py
+5 -5
View File
@@ -61,8 +61,8 @@ def banned_words(
return found
def register_api(app):
app.add_api_route("/sdapi/v1/nudenet", nudenet_censor, methods=["POST"], response_model=dict, tags=["Processing"])
app.add_api_route("/sdapi/v1/prompt-lang", prompt_check, methods=["POST"], response_model=dict, tags=["Processing"])
app.add_api_route("/sdapi/v1/image-guard", image_guard, methods=["POST"], response_model=dict, tags=["Processing"])
app.add_api_route("/sdapi/v1/prompt-banned", banned_words, methods=["POST"], response_model=list, tags=["Processing"])
def register_api(api_instance):
api_instance.add_api_route("/sdapi/v1/nudenet", nudenet_censor, methods=["POST"], response_model=dict, tags=["Processing"])
api_instance.add_api_route("/sdapi/v1/prompt-lang", prompt_check, methods=["POST"], response_model=dict, tags=["Processing"])
api_instance.add_api_route("/sdapi/v1/image-guard", image_guard, methods=["POST"], response_model=dict, tags=["Processing"])
api_instance.add_api_route("/sdapi/v1/prompt-banned", banned_words, methods=["POST"], response_model=list, tags=["Processing"])
modules/api/options.py
+4 -4
View File
@@ -88,7 +88,7 @@ def get_options_info():
return {"options": options_info, "sections": list(sections_seen.values())}
def register_api(app):
app.add_api_route("/sdapi/v1/options", get_options, methods=["GET"], response_model=dict, tags=["Server"])
app.add_api_route("/sdapi/v1/options", set_options, methods=["POST"], tags=["Server"])
app.add_api_route("/sdapi/v1/options-info", get_options_info, methods=["GET"], tags=["Server"])
def register_api(api):
api.add_api_route("/sdapi/v1/options", get_options, methods=["GET"], response_model=dict, tags=["Server"])
api.add_api_route("/sdapi/v1/options", set_options, methods=["POST"], tags=["Server"])
api.add_api_route("/sdapi/v1/options-info", get_options_info, methods=["GET"], tags=["Server"])
modules/api/xyz_grid.py
+2 -3
View File
@@ -1,5 +1,4 @@
def xyz_grid_enum(option: str = "") -> list[dict]:
"""List XYZ grid axis options. Optionally filter by label prefix/suffix and expand choices."""
from scripts.xyz import xyz_grid_classes # pylint: disable=no-name-in-module
@@ -21,5 +20,5 @@ def xyz_grid_enum(option: str = "") -> list[dict]:
return options
def register_api(app):
app.add_api_route("/sdapi/v1/xyz-grid", xyz_grid_enum, methods=["GET"], response_model=list[dict], tags=["Scripts"])
def register_api(api):
api.add_api_route("/sdapi/v1/xyz-grid", xyz_grid_enum, methods=["GET"], response_model=list[dict], tags=["Scripts"])