Merge pull request #191 from SecretiveShell/list-draft-models

fix function arguments for get_model_list
2024-09-07 22:29:05 -04:00 · 2024-09-07 22:29:05 -04:00 · 0c74cd80ea
commit 0c74cd80ea
parent acd3eb1140 d57a3b459c
1 changed files with 1 additions and 1 deletions
--- a/endpoints/core/router.py
+++ b/endpoints/core/router.py
@ -103,7 +103,7 @@ async def list_draft_models(request: Request) -> ModelList:

        models = get_model_list(draft_model_path.resolve())
    else:
-        models = await get_current_model_list(is_draft=True)
+        models = await get_current_model_list(model_type="draft")

    return models