|
|
@@ -17,7 +17,7 @@ OPENAI_MODEL = ""
|
|
|
# 后端可选“openai”或“ernie”
|
|
|
BACKEND = ""
|
|
|
|
|
|
-ernie_prompt = """\
|
|
|
+PROMPT = """
|
|
|
你是一个Linux命令行专家，请将我的请求转换成一条可执行的bash命令。
|
|
|
只给一条命令，不要做任何解释或说明。
|
|
|
|
|
|
@@ -27,19 +27,6 @@ ernie_prompt = """\
|
|
|
```bash
|
|
|
uname -a
|
|
|
```
|
|
|
-我的请求是：{0}\
|
|
|
-"""
|
|
|
-
|
|
|
-openai_prompt = """\
|
|
|
-你是一个Linux命令行专家，请将我的请求转换成一条可执行的bash命令。
|
|
|
-只给一条命令，不要做任何解释或说明。
|
|
|
-
|
|
|
-示例：
|
|
|
-请求：显示系统版本信息。
|
|
|
-输出：
|
|
|
-```bash
|
|
|
-uname -a
|
|
|
-```\
|
|
|
"""
|
|
|
|
|
|
# 设置API_KEY和SECRET_KEY
|
|
|
@@ -154,7 +141,7 @@ def askERNIE(quest):
|
|
|
"messages": [
|
|
|
{
|
|
|
"role": "user",
|
|
|
- "content": ernie_prompt.format(quest)
|
|
|
+ "content": PROMPT+"我的问题是:{0}".format(quest)
|
|
|
}
|
|
|
],
|
|
|
"temperature": 0.95,
|
|
|
@@ -193,7 +180,7 @@ def askOpenAI(quest):
|
|
|
"messages": [
|
|
|
{
|
|
|
"role": "system",
|
|
|
- "content": openai_prompt
|
|
|
+ "content": PROMPT
|
|
|
},
|
|
|
{
|
|
|
"role": "user",
|
