Skip to content

Commit cb585a8

Browse files
committed
Improve message formatting in MessageBuilder and add spacing in ClaudeResponseModel API call
1 parent d766684 commit cb585a8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/agentlab/llm/response_api.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,12 +89,12 @@ def to_markdown(self) -> str:
8989
parts = []
9090
for item in self.content:
9191
if "text" in item:
92-
parts.append(item["text"])
92+
parts.append(f"\n```\n{item["text"]}\n```\n")
9393
elif "image" in item:
9494
parts.append(f"![Image]({item['image']})")
9595

9696
markdown = f"### {self.role.capitalize()}\n"
97-
markdown += "\n\n---\n\n".join(parts)
97+
markdown += "\n".join(parts)
9898

9999
# if self.role == "tool":
100100
# assert self.tool_call_id is not None, "Tool call ID is required for tool messages"
@@ -509,6 +509,7 @@ def __init__(
509509

510510
def _call_api(self, messages: list[dict | MessageBuilder]) -> dict:
511511
input = []
512+
512513
for msg in messages:
513514
input.extend(msg.prepare_message() if isinstance(msg, MessageBuilder) else [msg])
514515

0 commit comments

Comments
 (0)