andre/gpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

入库prompt字符串从strip替换为正则表达式

Browse Source
This commit is contained in:
w_xiaolizu
2023-05-29 11:27:53 +08:00
parent 90325d6056
commit c5178b7038
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
func_box.py
View File

@ -481,8 +481,9 @@ def thread_write_chat(chatbot):
"""
private_key = toolbox.get_conf('private_key')[0]
chat_title = chatbot[0][0].split()
i_say = chatbot[-1][0].strip('<div class="markdown-body">/div<p>/p')
gpt_result = chatbot[-1][1].strip('<div class="markdown-body">/div<p>/p')
i_say = re.sub(r'^<div class="markdown-body"><p>|<\/p><\/div>$', '', chatbot[-1][0])
gpt_result = re.sub(r'^<div class="markdown-body"><p>|<\/p><\/div>$', '', chatbot[-1][1])
if private_key in chat_title:
SqliteHandle(f'ai_private_{chat_title[-2]}').inset_prompt({i_say: gpt_result})
else:
@ -543,4 +544,4 @@ class JsonHandle:
if __name__ == '__main__':
print(JsonHandle('/Users/kilig/Job/Python-project/academic_gpt/test.json').load())
pass