chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

处理一个core_functional.py中出现的mermaid渲染特例

Browse Source
This commit is contained in:
binary-husky 2024-01-24 01:38:06 +08:00
parent ef311c4859
commit 91926d24b7
1 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
shared_utils/advanced_markdown_format.py
View File

@ -292,13 +292,25 @@ def close_up_code_segment_during_stream(gpt_reply):
return 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): def compat_non_markdown_input(text):
""" """
改善非markdown输入的显示效果，例如将空格转换为&nbsp;，将换行符转换为</br>等。 改善非markdown输入的显示效果，例如将空格转换为&nbsp;，将换行符转换为</br>等。
""" """
if "```" in text: if "```" in text:
# careful input：markdown输入 # careful input：markdown输入
text = special_render_issues_for_mermaid(text) # 处理特殊的渲染问题
return text return text
elif "</div>" in text: elif "</div>" in text:
# careful input：html输入 # careful input：html输入
@ -313,19 +325,18 @@ def compat_non_markdown_input(text):
@lru_cache(maxsize=128) # 使用lru缓存 @lru_cache(maxsize=128) # 使用lru缓存
def simple_markdown_convertion(txt): def simple_markdown_convertion(text):
pre = '<div class="markdown-body">' pre = '<div class="markdown-body">'
suf = "</div>" suf = "</div>"
if txt.startswith(pre) and txt.endswith(suf): if text.startswith(pre) and text.endswith(suf):
return txt # 已经被转化过，不需要再次转化 return text # 已经被转化过，不需要再次转化
text = compat_non_markdown_input(text) # 兼容非markdown输入
txt = compat_non_markdown_input(txt) # 兼容非markdown输入 text = markdown.markdown(
txt = markdown.markdown( text,
txt,
extensions=["pymdownx.superfences", "tables", "pymdownx.highlight"], extensions=["pymdownx.superfences", "tables", "pymdownx.highlight"],
extension_configs=code_highlight_configs, extension_configs=code_highlight_configs,
) )
return pre + txt + suf return pre + text + suf
def format_io(self, y): def format_io(self, y):