Skip to content

Commit

Permalink
opti 问答方式
Browse files Browse the repository at this point in the history
  • Loading branch information
Bistutu committed May 25, 2024
1 parent 960cc93 commit 309232a
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 23 deletions.
4 changes: 2 additions & 2 deletions entrypoints/utils/option.ts
Original file line number Diff line number Diff line change
Expand Up @@ -343,8 +343,8 @@ export const defaultOption = {
hotkey: 'Control',
service: services.microsoft,
custom: 'http://localhost:11434/v1/chat/completions',
system_role: 'You are a professional, authentic translation engine, only returns translations.',
user_role: `Please translate them into {{to}}, please do not explain my original text.:
system_role: 'You are a professional, authentic machine translation engine. You only return the translated text, without any explanations.',
user_role: `Translate the following text into {{to}}, output translation text directly without any extra information:
{{origin}}`,
count: 0,
Expand Down
31 changes: 10 additions & 21 deletions entrypoints/utils/template.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,30 +9,27 @@ export function commonMsgTemplate(config: Config, origin: string) {

let system = config.system_role[config.service] || defaultOption.system_role;
let user = (config.user_role[config.service] || defaultOption.user_role)
.replace('{{to}}', config.to).replace('{{origin}}', 'hello');
.replace('{{to}}', config.to).replace('{{origin}}', origin);

return JSON.stringify({
'model': model,
"temperature": 0.3,
"temperature": 0.7,
'messages': [
{'role': 'system', 'content': system},
{'role': 'user', 'content': user},
{'role': "assistant", 'content': '你好'},
{'role': 'user', 'content': origin}
]
})
}

// gemini
export function geminiMsgTemplate(config: Config, origin: string) {
let user = (config.user_role[config.service] || defaultOption.user_role)
.replace('{{to}}', config.to).replace('{{origin}}', 'hello');
.replace('{{to}}', config.to).replace('{{origin}}', origin);

return JSON.stringify({
"contents": [
{"role": "user", "parts": [{"text": user}]},
{"role": "model", "parts": [{"text": "你好"}]},
{"role": "user", "parts": [{"text": origin}]}]
]
})
}

Expand All @@ -45,7 +42,7 @@ export function claudeMsgTemplate(config: Config, origin: string) {

let system = config.system_role[config.service] || defaultOption.system_role;
let user = (config.user_role[config.service] || defaultOption.user_role)
.replace('{{to}}', config.to).replace('{{origin}}', 'hello');
.replace('{{to}}', config.to).replace('{{origin}}', origin);

return JSON.stringify({
model: model,
Expand All @@ -54,8 +51,6 @@ export function claudeMsgTemplate(config: Config, origin: string) {
system: system,
messages: [
{role: "user", content: user},
{role: "assistant", content: "你好"},
{role: "user", content: origin}
]
})
}
Expand All @@ -66,16 +61,14 @@ export function tongyiMsgTemplate(config: Config, origin: string) {

let system = config.system_role[config.service] || defaultOption.system_role;
let user = (config.user_role[config.service] || defaultOption.user_role)
.replace('{{to}}', config.to).replace('{{origin}}', 'hello');
.replace('{{to}}', config.to).replace('{{origin}}', origin);

return JSON.stringify({
"model": model,
"input": {
"messages": [
{"role": "system", "content": system},
{"role": "user", "content": user},
{"role": "assistant", "content": "你好"},
{"role": "user", "content": origin}
]
},
"parameters": {}
Expand All @@ -85,15 +78,13 @@ export function tongyiMsgTemplate(config: Config, origin: string) {
// 文心一言
export function yiyanMsgTemplate(config: Config, origin: string) {
let user = (config.user_role[config.service] || defaultOption.user_role)
.replace('{{to}}', config.to).replace('{{origin}}', 'hello');
.replace('{{to}}', config.to).replace('{{origin}}', origin);

return JSON.stringify({
'temperature': 0.3, // 随机度
'temperature': 0.7,
'disable_search': true, // 禁用搜索
'messages': [
{"role": "user", "content": user},
{"role": "assistant", "content": "你好"},
{"role": "user", "content": origin}
],
})
}
Expand All @@ -102,17 +93,15 @@ export function minimaxTemplate(config: Config, origin: string) {

let system = config.system_role[config.service] || defaultOption.system_role;
let user = (config.user_role[config.service] || defaultOption.user_role)
.replace('{{to}}', config.to).replace('{{origin}}', 'hello');
.replace('{{to}}', config.to).replace('{{origin}}', origin);

return JSON.stringify({
model: "abab6.5-chat",
stream: false,
temperature: 0.3,
temperature: 0.7,
messages: [
{role: 'system', content: system},
{role: 'user', content: user},
{role: 'assistant', content: '你好'},
{role: 'user', content: origin}
]
})
}

0 comments on commit 309232a

Please sign in to comment.