From 5ea8b16f84fe1b6c424c369662da9300b8d552cc Mon Sep 17 00:00:00 2001 From: sabaimran Date: Thu, 4 Jul 2024 12:08:22 +0530 Subject: [PATCH] Fix missing method error --- src/khoj/database/adapters/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 71b089f5..2ea9e9af 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -906,6 +906,10 @@ class ConversationAdapters: async def aget_text_to_image_model_config(): return await TextToImageModelConfig.objects.filter().prefetch_related("openai_config").afirst() + @staticmethod + def get_text_to_image_model_config(): + return TextToImageModelConfig.objects.filter().first() + @staticmethod def get_text_to_image_model_options(): return TextToImageModelConfig.objects.all()