diff --git a/src/agent/context.rs b/src/agent/context.rs index 05ad935..47e6c7a 100644 --- a/src/agent/context.rs +++ b/src/agent/context.rs @@ -146,9 +146,9 @@ impl NodeBody { out.push_str("\n\n"); } Self::ToolResult(text) => { - out.push_str("<|im_start|>tool\n"); + out.push_str("<|im_start|>user\n\n"); out.push_str(text); - out.push_str("<|im_end|>\n"); + out.push_str("\n<|im_end|>\n"); } Self::Memory { text, .. } => { out.push_str("<|im_start|>memory\n"); @@ -1035,7 +1035,7 @@ mod tests { #[test] fn test_render_tool_result() { let node = AstNode::tool_result("output here"); - assert_eq!(node.render(), "<|im_start|>tool\noutput here<|im_end|>\n"); + assert_eq!(node.render(), "<|im_start|>user\n\noutput here\n<|im_end|>\n"); } #[test]