Spaces:
Running
Running
查找语法错误之前先清除换行符
Browse files- functional.py +4 -2
- predict.py +2 -1
- toolbox.py +6 -0
functional.py
CHANGED
|
@@ -2,6 +2,7 @@
|
|
| 2 |
# 'secondary' 颜色对应 theme.py 中的 neutral_hue
|
| 3 |
# 'stop' 颜色对应 theme.py 中的 color_er
|
| 4 |
# 默认按钮颜色是 secondary
|
|
|
|
| 5 |
|
| 6 |
def get_functionals():
|
| 7 |
return {
|
|
@@ -22,11 +23,12 @@ def get_functionals():
|
|
| 22 |
"查找语法错误": {
|
| 23 |
"Prefix": r"Below is a paragraph from an academic paper. " +
|
| 24 |
r"Can you help me ensure that the grammar and the spelling is correct? " +
|
| 25 |
-
r"
|
| 26 |
-
r"If you find grammar mistakes,please list mistakes you find in a two-column markdown table, " +
|
| 27 |
r"put the original text the first column, " +
|
| 28 |
r"put the corrected text in the second column and highlight the key words you fixed." + "\n\n",
|
| 29 |
"Suffix": r"",
|
|
|
|
| 30 |
},
|
| 31 |
"中译英": {
|
| 32 |
"Prefix": r"Please translate following sentence to English:" + "\n\n",
|
|
|
|
| 2 |
# 'secondary' 颜色对应 theme.py 中的 neutral_hue
|
| 3 |
# 'stop' 颜色对应 theme.py 中的 color_er
|
| 4 |
# 默认按钮颜色是 secondary
|
| 5 |
+
from toolbox import clear_line_break
|
| 6 |
|
| 7 |
def get_functionals():
|
| 8 |
return {
|
|
|
|
| 23 |
"查找语法错误": {
|
| 24 |
"Prefix": r"Below is a paragraph from an academic paper. " +
|
| 25 |
r"Can you help me ensure that the grammar and the spelling is correct? " +
|
| 26 |
+
r"Do not try to polish the text, if no mistake is found, tell me that this paragraph is good." +
|
| 27 |
+
r"If you find grammar or spelling mistakes, please list mistakes you find in a two-column markdown table, " +
|
| 28 |
r"put the original text the first column, " +
|
| 29 |
r"put the corrected text in the second column and highlight the key words you fixed." + "\n\n",
|
| 30 |
"Suffix": r"",
|
| 31 |
+
"PreProcess": clear_line_break, # 预处理:清除换行符
|
| 32 |
},
|
| 33 |
"中译英": {
|
| 34 |
"Prefix": r"Please translate following sentence to English:" + "\n\n",
|
predict.py
CHANGED
|
@@ -119,8 +119,9 @@ def predict(inputs, top_p, temperature, chatbot=[], history=[], system_prompt=''
|
|
| 119 |
"""
|
| 120 |
if additional_fn is not None:
|
| 121 |
import functional
|
| 122 |
-
importlib.reload(functional)
|
| 123 |
functional = functional.get_functionals()
|
|
|
|
| 124 |
inputs = functional[additional_fn]["Prefix"] + inputs + functional[additional_fn]["Suffix"]
|
| 125 |
|
| 126 |
if stream:
|
|
|
|
| 119 |
"""
|
| 120 |
if additional_fn is not None:
|
| 121 |
import functional
|
| 122 |
+
importlib.reload(functional) # 热更新prompt
|
| 123 |
functional = functional.get_functionals()
|
| 124 |
+
if "PreProcess" in functional[additional_fn]: inputs = functional[additional_fn]["PreProcess"](inputs) # 获取预处理函数(如果有的话)
|
| 125 |
inputs = functional[additional_fn]["Prefix"] + inputs + functional[additional_fn]["Suffix"]
|
| 126 |
|
| 127 |
if stream:
|
toolbox.py
CHANGED
|
@@ -230,3 +230,9 @@ def get_conf(*args):
|
|
| 230 |
assert False, "正确的API_KEY密钥是51位,请在config文件中修改API密钥, 添加海外代理之后再运行。" + \
|
| 231 |
"(如果您刚更新过代码,请确保旧版config_private文件中没有遗留任何新增键值)"
|
| 232 |
return res
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 230 |
assert False, "正确的API_KEY密钥是51位,请在config文件中修改API密钥, 添加海外代理之后再运行。" + \
|
| 231 |
"(如果您刚更新过代码,请确保旧版config_private文件中没有遗留任何新增键值)"
|
| 232 |
return res
|
| 233 |
+
|
| 234 |
+
def clear_line_break(txt):
|
| 235 |
+
txt = txt.replace('\n', ' ')
|
| 236 |
+
txt = txt.replace(' ', ' ')
|
| 237 |
+
txt = txt.replace(' ', ' ')
|
| 238 |
+
return txt
|