From 68e343dcd90207ed7eb98f6d596cabe2704155f6 Mon Sep 17 00:00:00 2001 From: Leon Lam Nilsson Date: Fri, 19 Apr 2024 14:46:16 +0200 Subject: [PATCH] add return type to db function --- apps/api/src/interfaces/db.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/apps/api/src/interfaces/db.py b/apps/api/src/interfaces/db.py index 5fe90601..4f688b0e 100644 --- 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]