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 default chat model to use user model if no server chat model set

Browse Source 
- Advanced chat model should also fallback to user chat model if set
- Get conversation config should falback to user chat model if set

These assume no server chat model settings is configured
This commit is contained in:
Debanjum Singh Solanky
2024-10-13 03:02:29 -07:00
parent 8ff13e4cf6
commit 9356e66b94
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -937,21 +937,21 @@ class ConversationAdapters:
def get_conversation_config(user: KhojUser): def get_conversation_config(user: KhojUser):
subscribed = is_user_subscribed(user) subscribed = is_user_subscribed(user)
if not subscribed: if not subscribed:
return ConversationAdapters.get_default_conversation_config() return ConversationAdapters.get_default_conversation_config(user)
config = UserConversationConfig.objects.filter(user=user).first() config = UserConversationConfig.objects.filter(user=user).first()
if config: if config:
return config.setting return config.setting
return ConversationAdapters.get_advanced_conversation_config() return ConversationAdapters.get_advanced_conversation_config(user)
@staticmethod @staticmethod
async def aget_conversation_config(user: KhojUser): async def aget_conversation_config(user: KhojUser):
subscribed = await ais_user_subscribed(user) subscribed = await ais_user_subscribed(user)
if not subscribed: if not subscribed:
return await ConversationAdapters.aget_default_conversation_config() return await ConversationAdapters.aget_default_conversation_config(user)
config = await UserConversationConfig.objects.filter(user=user).prefetch_related("setting").afirst() config = await UserConversationConfig.objects.filter(user=user).prefetch_related("setting").afirst()
if config: if config:
return config.setting return config.setting
return ConversationAdapters.aget_advanced_conversation_config() return ConversationAdapters.aget_advanced_conversation_config(user)
@staticmethod @staticmethod
async def aget_voice_model_config(user: KhojUser) -> Optional[VoiceModelOption]: async def aget_voice_model_config(user: KhojUser) -> Optional[VoiceModelOption]:
@@ -1012,22 +1012,22 @@ class ConversationAdapters:
return await ChatModelOptions.objects.filter().prefetch_related("openai_config").afirst() return await ChatModelOptions.objects.filter().prefetch_related("openai_config").afirst()
@staticmethod @staticmethod
def get_advanced_conversation_config(): def get_advanced_conversation_config(user: KhojUser):
server_chat_settings = ServerChatSettings.objects.first() server_chat_settings = ServerChatSettings.objects.first()
if server_chat_settings is not None and server_chat_settings.chat_advanced is not None: if server_chat_settings is not None and server_chat_settings.chat_advanced is not None:
return server_chat_settings.chat_advanced return server_chat_settings.chat_advanced
return ConversationAdapters.get_default_conversation_config() return ConversationAdapters.get_default_conversation_config(user)
@staticmethod @staticmethod
async def aget_advanced_conversation_config(): async def aget_advanced_conversation_config(user: KhojUser = None):
server_chat_settings: ServerChatSettings = ( server_chat_settings: ServerChatSettings = (
await ServerChatSettings.objects.filter() await ServerChatSettings.objects.filter()
.prefetch_related("chat_advanced", "chat_advanced__openai_config") .prefetch_related("chat_advanced", "chat_advanced__openai_config")
.afirst() .afirst()
) )
if server_chat_settings is not None or server_chat_settings.chat_advanced is not None: if server_chat_settings is not None and server_chat_settings.chat_advanced is not None:
return server_chat_settings.chat_advanced return server_chat_settings.chat_advanced
return await ConversationAdapters.aget_default_conversation_config() return await ConversationAdapters.aget_default_conversation_config(user)
@staticmethod @staticmethod
def create_conversation_from_public_conversation( def create_conversation_from_public_conversation(