add return type to db function

startino · Apr 19, 2024 · 68e343d · 68e343d
1 parent 9691485
commit 68e343d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apps/api/src/interfaces/db.py b/apps/api/src/interfaces/db.py
@@ -870,7 +870,7 @@ def delete_profile(profile_id: UUID) -> Profile:
     return Profile(**response.data[0])
 
 
-def get_models():
+def get_models() -> list[LLMModel]:
     supabase: Client = create_client(url, key)
     response = supabase.table("models").select("*").execute()
     return [LLMModel(**data) for data in response.data]