Browse Source

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

Xingjian Xie 2 tháng trước cách đây
mục cha
commit
c56bedc5ff
1 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 2 2
      backend/open_webui/utils/middleware.py

+ 2 - 2
backend/open_webui/utils/middleware.py

@@ -1260,10 +1260,10 @@ async def process_chat_response(
                                 match.end() :
                             ]  # 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"
-                            ].replace(match.group(0), "")
+                            ].replace(match.group(0) + after_tag, "")
 
                             if before_tag:
                                 content_blocks[-1]["content"] = before_tag