From a129b017b9a7abef376a31b398796bc6f84958c3 Mon Sep 17 00:00:00 2001 From: sabaimran Date: Thu, 4 Jul 2024 09:13:23 +0530 Subject: [PATCH] Fix image generation on server -- use default config when not set by user --- src/khoj/database/adapters/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 358edf2a..9f8ba124 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -918,10 +918,13 @@ class ConversationAdapters: return config.setting @staticmethod - async def aget_user_text_to_image_model(user: KhojUser): + async def aget_user_text_to_image_model(user: KhojUser) -> Optional[TextToImageModelConfig]: config = await UserTextToImageModelConfig.objects.filter(user=user).prefetch_related("setting").afirst() if not config: - return None + default_config = await ConversationAdapters.aget_text_to_image_model_config() + if not default_config: + return None + return default_config return config.setting @staticmethod