diff --git a/api/db/services/dialog_service.py b/api/db/services/dialog_service.py index ab43b944c..71429198a 100644 --- a/api/db/services/dialog_service.py +++ b/api/db/services/dialog_service.py @@ -168,7 +168,7 @@ def chat(dialog, messages, stream=True, **kwargs): gen_conf = dialog.llm_setting msg = [{"role": "system", "content": prompt_config["system"].format(**kwargs)}] - msg.extend([{"role": m["role"], "content": m["content"]} + msg.extend([{"role": m["role"], "content": re.sub(r"##\d+\$\$", "", m["content"])} for m in messages if m["role"] != "system"]) used_token_count, msg = message_fit_in(msg, int(max_tokens * 0.97)) assert len(msg) >= 2, f"message_fit_in has bug: {msg}"