TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: missing parameter (#93)

Browse Source
This commit is contained in:
Hayden
2025-01-13 15:58:11 +08:00
committed by GitHub
parent 4132b2d8c4
commit c2406a1fd1
5 changed files with 19 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
__init__.py
View File

@@ -95,7 +95,7 @@ async def get_model_paths(request):
"""
Returns the base folders for models.
"""
model_base_paths = utils.resolve_model_base_paths()
model_base_paths = utils.resolve_model_base_paths(request)
return web.json_response({"success": True, "data": model_base_paths})
@@ -160,7 +160,7 @@ async def read_model_info(request):
filename = request.match_info.get("filename", None)
try:
model_path = utils.get_valid_full_path(model_type, index, filename)
model_path = utils.get_valid_full_path(model_type, index, filename, request)
result = services.get_model_info(model_path)
return web.json_response({"success": True, "data": result})
except Exception as e:
@@ -189,10 +189,10 @@ async def update_model(request):
model_data: dict = await request.json()
try:
model_path = utils.get_valid_full_path(model_type, index, filename)
model_path = utils.get_valid_full_path(model_type, index, filename, request)
if model_path is None:
raise RuntimeError(f"File {filename} not found")
services.update_model(model_path, model_data)
services.update_model(model_path, model_data, request)
return web.json_response({"success": True})
except Exception as e:
error_msg = f"Update model failed: {str(e)}"
@@ -210,7 +210,7 @@ async def delete_model(request):
filename = request.match_info.get("filename", None)
try:
model_path = utils.get_valid_full_path(model_type, index, filename)
model_path = utils.get_valid_full_path(model_type, index, filename, request)
if model_path is None:
raise RuntimeError(f"File {filename} not found")
services.remove_model(model_path)