diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 9f8ba124..71b089f5 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -904,7 +904,7 @@ class ConversationAdapters: @staticmethod async def aget_text_to_image_model_config(): - return await TextToImageModelConfig.objects.filter().afirst() + return await TextToImageModelConfig.objects.filter().prefetch_related("openai_config").afirst() @staticmethod def get_text_to_image_model_options(): @@ -914,7 +914,10 @@ class ConversationAdapters: def get_user_text_to_image_model_config(user: KhojUser): config = UserTextToImageModelConfig.objects.filter(user=user).first() if not config: - return None + default_config = ConversationAdapters.get_text_to_image_model_config() + if not default_config: + return None + return default_config return config.setting @staticmethod