فهرست منبع

Merge pull request #9619 from XingjianXie/fix_after_tag

Fix tag_content_handler issue
Timothy Jaeryang Baek 2 ماه پیش
والد
کامیت
4271b652a5
1فایلهای تغییر یافته به همراه2 افزوده شده و 2 حذف شده
  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() :
                                 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