Fix tag_content_handler issue: after_tag should be remove from the current content_blocks

This commit is contained in:
Xingjian Xie 2025-02-07 20:15:54 +00:00
parent 14398ab628
commit c56bedc5ff

View File

@ -1260,10 +1260,10 @@ async def process_chat_response(
match.end() : match.end() :
] # Content after opening tag ] # Content after opening tag
# Remove the start tag from the currently handling text block # Remove the start tag and after from the currently handling text block
content_blocks[-1]["content"] = content_blocks[-1][ content_blocks[-1]["content"] = content_blocks[-1][
"content" "content"
].replace(match.group(0), "") ].replace(match.group(0) + after_tag, "")
if before_tag: if before_tag:
content_blocks[-1]["content"] = before_tag content_blocks[-1]["content"] = before_tag