From 2932d305b0f370244a63ea10333272a40317e88e Mon Sep 17 00:00:00 2001 From: sabaimran Date: Tue, 10 Sep 2024 21:09:43 -0700 Subject: [PATCH] Simplify redundant logic for constructing structured messages with image url --- src/khoj/processor/conversation/utils.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/khoj/processor/conversation/utils.py b/src/khoj/processor/conversation/utils.py index 54bf6334..999473e2 100644 --- a/src/khoj/processor/conversation/utils.py +++ b/src/khoj/processor/conversation/utils.py @@ -175,10 +175,9 @@ def generate_chatml_messages_with_context( message_content = chat["message"] + message_notes - if chat.get("uploadedImageData") and vision_enabled: - message_content = construct_structured_message( - message_content, chat.get("uploadedImageData"), model_type, vision_enabled - ) + message_content = construct_structured_message( + message_content, chat.get("uploadedImageData"), model_type, vision_enabled + ) reconstructed_message = ChatMessage(content=message_content, role=role)