klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-09 21:29:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix retrieving image model by prefetching the openai config in the async method

Browse Source
This commit is contained in:
sabaimran
2024-07-04 11:58:33 +05:30
parent a129b017b9
commit d61bddf56c
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/khoj/database/adapters/__init__.py
View File

@@ -904,7 +904,7 @@ class ConversationAdapters:
@staticmethod @staticmethod
async def aget_text_to_image_model_config(): 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 @staticmethod
def get_text_to_image_model_options(): def get_text_to_image_model_options():
@@ -914,7 +914,10 @@ class ConversationAdapters:
def get_user_text_to_image_model_config(user: KhojUser): def get_user_text_to_image_model_config(user: KhojUser):
config = UserTextToImageModelConfig.objects.filter(user=user).first() config = UserTextToImageModelConfig.objects.filter(user=user).first()
if not config: if not config:
default_config = ConversationAdapters.get_text_to_image_model_config()
if not default_config:
return None return None
return default_config
return config.setting return config.setting
@staticmethod @staticmethod