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 d804fc11f8
commit 03aa2ee115
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
__main__.py
View File

@ -383,7 +383,8 @@ class ChatBot(ChatBotFrame):
# Start # Start
self.auto_opentab_delay() self.auto_opentab_delay()
demo.queue(concurrency_count=CONCURRENT_COUNT).launch(server_name="0.0.0.0", server_port=PORT, auth=AUTHENTICATION) demo.queue(concurrency_count=CONCURRENT_COUNT).launch(server_name="0.0.0.0", server_port=PORT, auth=AUTHENTICATION,
blocked_paths=["config.py", "config_private.py", "docker-compose.yml", "Dockerfile"])
def check_proxy_free(): def check_proxy_free():

7
func_box.py
View File

@ -481,8 +481,9 @@ def thread_write_chat(chatbot):
""" """
private_key = toolbox.get_conf('private_key')[0] private_key = toolbox.get_conf('private_key')[0]
chat_title = chatbot[0][0].split() 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: if private_key in chat_title:
SqliteHandle(f'ai_private_{chat_title[-2]}').inset_prompt({i_say: gpt_result}) SqliteHandle(f'ai_private_{chat_title[-2]}').inset_prompt({i_say: gpt_result})
else: else:
@ -543,4 +544,4 @@ class JsonHandle:
if __name__ == '__main__': if __name__ == '__main__':
print(JsonHandle('/Users/kilig/Job/Python-project/academic_gpt/test.json').load()) pass