diff --git a/src/khoj/database/adapters/__init__.py b/src/khoj/database/adapters/__init__.py index 572e1f83..2280ea60 100644 --- a/src/khoj/database/adapters/__init__.py +++ b/src/khoj/database/adapters/__init__.py @@ -1182,7 +1182,7 @@ class ConversationAdapters: @staticmethod async def aget_default_chat_model(user: KhojUser = None, fallback_chat_model: Optional[ChatModel] = None): - """Get default conversation config. Prefer chat model by server admin > user > first created chat model""" + """Get default conversation config. Prefer chat model by server admin > agent > user > first created chat model""" # Get the server chat settings server_chat_settings: ServerChatSettings = ( await ServerChatSettings.objects.filter() diff --git a/src/khoj/routers/research.py b/src/khoj/routers/research.py index 330d669e..d534e812 100644 --- a/src/khoj/routers/research.py +++ b/src/khoj/routers/research.py @@ -113,6 +113,7 @@ async def apick_next_tool( today = datetime.today() location_data = f"{location}" if location else "Unknown" + agent_chat_model = agent.chat_model if agent else None personality_context = ( prompts.personality_context.format(personality=agent.personality) if agent and agent.personality else "" ) @@ -140,6 +141,7 @@ async def apick_next_tool( user=user, query_images=query_images, query_files=query_files, + agent_chat_model=agent_chat_model, tracer=tracer, ) except Exception as e: