From a0b279f1054ede7d565d3f83cb5bf11c14bed39d Mon Sep 17 00:00:00 2001 From: Daniel Vainsencher Date: Wed, 29 May 2024 17:07:41 -0400 Subject: [PATCH] Refactored format_messages to use format_content for string content handling. --- aider/utils.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/aider/utils.py b/aider/utils.py index 1e76b39bf..3a09f0236 100644 --- a/aider/utils.py +++ b/aider/utils.py @@ -84,7 +84,12 @@ def safe_abs_path(res): return str(res) -def format_messages(messages, title=None): +def format_content(role, content): + formatted_lines = [] + for line in content.splitlines(): + formatted_lines.append(f"{role} {line}") + return "\n".join(formatted_lines) + output = [] if title: output.append(f"{title.upper()} {'*' * 50}") @@ -98,8 +103,7 @@ def format_messages(messages, title=None): if isinstance(item, dict) and "image_url" in item: output.append(f"{role} Image URL: {item['image_url']['url']}") elif isinstance(content, str): # Handle string content - for line in content.splitlines(): - output.append(f"{role} {line}") + output.append(format_content(role, content)) content = msg.get("function_call") if content: output.append(f"{role} {content}")