diff --git a/scripts/tag_autocomplete_helper.py b/scripts/tag_autocomplete_helper.py index 7d91371..a86e2d4 100644 --- a/scripts/tag_autocomplete_helper.py +++ b/scripts/tag_autocomplete_helper.py @@ -419,16 +419,23 @@ def api_tac(_: gr.Blocks, app: FastAPI): return json.dumps({"error": e}) @app.get("/tacapi/v1/lora-info/{folder}/{lora_name}") - async def get_lora_info(folder, lora_name): + async def get_lora_info_subfolder(folder, lora_name): if LORA_PATH is None: return json.dumps({}) return await get_json_info(LORA_PATH.joinpath(folder).joinpath(lora_name)) @app.get("/tacapi/v1/lyco-info/{folder}/{lyco_name}") - async def get_lyco_info(folder, lyco_name): + async def get_lyco_info_subfolder(folder, lyco_name): if LYCO_PATH is None: return json.dumps({}) return await get_json_info(LYCO_PATH.joinpath(folder).joinpath(lyco_name)) + @app.get("/tacapi/v1/lora-info/{lora_name}") + async def get_lora_info(lora_name): + return await get_lora_info_subfolder(".", lora_name) + + @app.get("/tacapi/v1/lyco-info/{lyco_name}") + async def get_lyco_info(lyco_name): + return await get_lyco_info_subfolder(".", lyco_name) script_callbacks.on_app_started(api_tac) \ No newline at end of file