diff --git a/shared_utils/advanced_markdown_format.py b/shared_utils/advanced_markdown_format.py index 653cf07..a015fd6 100644 --- a/shared_utils/advanced_markdown_format.py +++ b/shared_utils/advanced_markdown_format.py @@ -292,13 +292,25 @@ def close_up_code_segment_during_stream(gpt_reply): return gpt_reply +def special_render_issues_for_mermaid(text): + # 用不太优雅的方式处理一个core_functional.py中出现的mermaid渲染特例: + # 我不希望"总结绘制脑图"prompt中的mermaid渲染出来 + @lru_cache(maxsize=1) + def get_special_case(): + from core_functional import get_core_functions + special_case = get_core_functions()["总结绘制脑图"]["Suffix"] + return special_case + if text.endswith(get_special_case()): text = text.replace("```mermaid", "```") + return text + + def compat_non_markdown_input(text): """ 改善非markdown输入的显示效果,例如将空格转换为 ,将换行符转换为等。 """ - if "```" in text: # careful input:markdown输入 + text = special_render_issues_for_mermaid(text) # 处理特殊的渲染问题 return text elif "" in text: # careful input:html输入 @@ -313,19 +325,18 @@ def compat_non_markdown_input(text): @lru_cache(maxsize=128) # 使用lru缓存 -def simple_markdown_convertion(txt): +def simple_markdown_convertion(text): pre = '