sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-20 06:55:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

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
Show stats 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