From d57a3b459c15434a58c782a7eed6f13998b7eac6 Mon Sep 17 00:00:00 2001
From: TerminalMan <84923604+SecretiveShell@users.noreply.github.com>
Date: Sat, 7 Sep 2024 18:27:10 +0100
Subject: [PATCH] fix function arguments for get_model_list

---
 endpoints/core/router.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/endpoints/core/router.py b/endpoints/core/router.py
index 1f9d1948..cc9af247 100644
--- 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