From ecd005cac0ad73f04a67a61c6b51a01374f08743 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Wed, 15 Nov 2023 10:41:35 -0800 Subject: [PATCH] Check if search model is already in DB before creating a new one --- src/database/adapters/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/database/adapters/__init__.py b/src/database/adapters/__init__.py index 39ac290b..a61b3d64 100644 --- a/src/database/adapters/__init__.py +++ b/src/database/adapters/__init__.py @@ -216,7 +216,11 @@ async def set_user_github_config(user: KhojUser, pat_token: str, repos: list): def get_or_create_search_model(): - return SearchModel.objects.filter().get_or_create()[0] + search_model = SearchModel.objects.filter().first() + if not search_model: + search_model = SearchModel.objects.create() + + return search_model class ConversationAdapters: