Remove the 'User Message' tag in the quick answer process.

This commit is contained in:
黄伟韬 2024-08-19 19:15:33 +08:00
parent 408174ed8f
commit 6b4c00f286

View file

@ -295,6 +295,9 @@ class RoleZero(Role):
if "QUICK" in intent_result or "AMBIGUOUS" in intent_result: # llm call with the original context
async with ThoughtReporter(enable_llm_stream=True) as reporter:
await reporter.async_report({"type": "quick"})
# remove "[Message] from A to B :" tag.
pattern = r"\[Message\] from .* to .*:\s*"
memory[-1].content = re.sub(pattern, "", memory[-1].content, count=1)
answer = await self.llm.aask(
self.llm.format_msg(memory),
system_msgs=[QUICK_RESPONSE_SYSTEM_PROMPT.format(role_info=self._get_prefix())],