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

Fix image generation on server -- use default config when not set by user

Browse Source
This commit is contained in:
sabaimran
2024-07-04 09:13:23 +05:30
parent 6fa2dbc042
commit a129b017b9
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -918,10 +918,13 @@ class ConversationAdapters:
return config.setting return config.setting
@staticmethod @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() config = await UserTextToImageModelConfig.objects.filter(user=user).prefetch_related("setting").afirst()
if not config: 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 return config.setting
@staticmethod @staticmethod