Spaces:
Sleeping
Sleeping
Tuchuanhuhuhu
commited on
Commit
•
65dd753
1
Parent(s):
06630d4
bugfix: 修复重新生成时有时内容包含HTML标签的问题
Browse files
modules/models/base_model.py
CHANGED
@@ -523,10 +523,13 @@ class BaseLLMModel:
|
|
523 |
if len(self.history) > 0:
|
524 |
inputs = self.history[-2]["content"]
|
525 |
del self.history[-2:]
|
526 |
-
|
527 |
-
|
528 |
elif len(chatbot) > 0:
|
529 |
inputs = chatbot[-1][0]
|
|
|
|
|
|
|
530 |
else:
|
531 |
yield chatbot, f"{STANDARD_ERROR_MSG}上下文是空的"
|
532 |
return
|
|
|
523 |
if len(self.history) > 0:
|
524 |
inputs = self.history[-2]["content"]
|
525 |
del self.history[-2:]
|
526 |
+
if len(self.all_token_counts) > 0:
|
527 |
+
self.all_token_counts.pop()
|
528 |
elif len(chatbot) > 0:
|
529 |
inputs = chatbot[-1][0]
|
530 |
+
if '<div class="user-message">' in inputs:
|
531 |
+
inputs = inputs.split('<div class="user-message">')[1]
|
532 |
+
inputs = inputs.split("</div>")[0]
|
533 |
else:
|
534 |
yield chatbot, f"{STANDARD_ERROR_MSG}上下文是空的"
|
535 |
return
|