Wrap image with XML open/close tags

This commit is contained in:
Charles Cunningham
2026-01-11 22:20:45 -08:00
parent aa07b6b6fd
commit 95bc7bbf8a
2 changed files with 33 additions and 11 deletions
+9 -1
View File
@@ -37,7 +37,12 @@ fn parse_user_message(message: &[ContentItem]) -> Option<UserMessageItem> {
match content_item {
ContentItem::InputText { text } => {
if is_local_image_label_text(text)
&& matches!(message.get(idx + 1), Some(ContentItem::InputImage { .. }))
&& (matches!(message.get(idx + 1), Some(ContentItem::InputImage { .. }))
|| (idx > 0
&& matches!(
message.get(idx - 1),
Some(ContentItem::InputImage { .. })
)))
{
continue;
}
@@ -198,6 +203,9 @@ mod tests {
ContentItem::InputImage {
image_url: image_url.clone(),
},
ContentItem::InputText {
text: "</image>".to_string(),
},
ContentItem::InputText {
text: user_text.clone(),
},