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

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

View File

@ -383,7 +383,8 @@ class ChatBot(ChatBotFrame):
# Start
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():

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