From 45e725ac9cc9b346b5848ba41cd7cc28699b607f Mon Sep 17 00:00:00 2001 From: sabaimran Date: Wed, 12 Jun 2024 17:41:12 +0530 Subject: [PATCH] Use the summarizer model for generating improved image prompts --- src/khoj/routers/helpers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/khoj/routers/helpers.py b/src/khoj/routers/helpers.py index f144044e..503d5b1f 100644 --- a/src/khoj/routers/helpers.py +++ b/src/khoj/routers/helpers.py @@ -449,8 +449,10 @@ async def generate_better_image_prompt( online_results=simplified_online_results, ) + summarizer_model: ChatModelOptions = await ConversationAdapters.aget_summarizer_conversation_config() + with timer("Chat actor: Generate contextual image prompt", logger): - response = await send_message_to_model_wrapper(image_prompt) + response = await send_message_to_model_wrapper(image_prompt, chat_model_option=summarizer_model) return response.strip()