2023-04-10 00:41:34 +08:00
|
|
|
const cleanupPreset = ({ preset: _preset, endpointsConfig = {} }) => {
|
2023-04-04 01:12:14 +08:00
|
|
|
const { endpoint } = _preset;
|
|
|
|
|
|
|
|
|
|
let preset = {};
|
|
|
|
|
if (endpoint === 'azureOpenAI' || endpoint === 'openAI') {
|
|
|
|
|
preset = {
|
|
|
|
|
endpoint,
|
2023-04-05 17:25:35 +08:00
|
|
|
presetId: _preset?.presetId ?? null,
|
2023-04-10 00:41:34 +08:00
|
|
|
model: _preset?.model ?? endpointsConfig[endpoint]?.availableModels?.[0] ?? 'gpt-3.5-turbo',
|
2023-04-05 17:25:35 +08:00
|
|
|
chatGptLabel: _preset?.chatGptLabel ?? null,
|
|
|
|
|
promptPrefix: _preset?.promptPrefix ?? null,
|
|
|
|
|
temperature: _preset?.temperature ?? 1,
|
|
|
|
|
top_p: _preset?.top_p ?? 1,
|
|
|
|
|
presence_penalty: _preset?.presence_penalty ?? 0,
|
|
|
|
|
frequency_penalty: _preset?.frequency_penalty ?? 0,
|
|
|
|
|
title: _preset?.title ?? 'New Preset'
|
2023-04-04 01:12:14 +08:00
|
|
|
};
|
2023-05-13 16:29:06 -04:00
|
|
|
} else if (endpoint === 'google') {
|
|
|
|
|
preset = {
|
|
|
|
|
endpoint,
|
|
|
|
|
presetId: _preset?.presetId ?? null,
|
|
|
|
|
model: _preset?.model ?? endpointsConfig[endpoint]?.availableModels?.[0] ?? 'chat-bison',
|
|
|
|
|
modelLabel: _preset?.modelLabel ?? null,
|
|
|
|
|
examples: _preset?.examples ?? [{ input: { content: '' }, output: { content: '' } }],
|
|
|
|
|
promptPrefix: _preset?.promptPrefix ?? null,
|
|
|
|
|
temperature: _preset?.temperature ?? 0.2,
|
|
|
|
|
maxOutputTokens: _preset?.maxOutputTokens ?? 1024,
|
|
|
|
|
topP: _preset?.topP ?? 0.95,
|
|
|
|
|
topK: _preset?.topK ?? 40,
|
|
|
|
|
title: _preset?.title ?? 'New Preset'
|
|
|
|
|
};
|
2023-04-04 01:12:14 +08:00
|
|
|
} else if (endpoint === 'bingAI') {
|
|
|
|
|
preset = {
|
|
|
|
|
endpoint,
|
2023-04-05 17:25:35 +08:00
|
|
|
presetId: _preset?.presetId ?? null,
|
|
|
|
|
jailbreak: _preset?.jailbreak ?? false,
|
|
|
|
|
context: _preset?.context ?? null,
|
|
|
|
|
systemMessage: _preset?.systemMessage ?? null,
|
|
|
|
|
toneStyle: _preset?.toneStyle ?? 'fast',
|
|
|
|
|
title: _preset?.title ?? 'New Preset'
|
2023-04-04 01:12:14 +08:00
|
|
|
};
|
2023-04-05 21:21:31 +08:00
|
|
|
} else if (endpoint === 'chatGPTBrowser') {
|
2023-04-04 01:12:14 +08:00
|
|
|
preset = {
|
|
|
|
|
endpoint,
|
2023-04-05 17:25:35 +08:00
|
|
|
presetId: _preset?.presetId ?? null,
|
2023-04-05 21:21:02 +08:00
|
|
|
model:
|
2023-05-18 11:09:31 -07:00
|
|
|
_preset?.model ??
|
|
|
|
|
endpointsConfig[endpoint]?.availableModels?.[0] ??
|
|
|
|
|
'text-davinci-002-render-sha',
|
2023-04-05 17:25:35 +08:00
|
|
|
title: _preset?.title ?? 'New Preset'
|
2023-04-04 01:12:14 +08:00
|
|
|
};
|
|
|
|
|
} else if (endpoint === null) {
|
|
|
|
|
preset = {
|
|
|
|
|
endpoint,
|
|
|
|
|
presetId: _preset?.presetId || null,
|
2023-04-05 17:25:35 +08:00
|
|
|
title: _preset?.title ?? 'New Preset'
|
2023-04-04 01:12:14 +08:00
|
|
|
};
|
|
|
|
|
} else {
|
|
|
|
|
console.error(`Unknown endpoint ${endpoint}`);
|
|
|
|
|
preset = {
|
|
|
|
|
endpoint: null,
|
2023-04-05 17:25:35 +08:00
|
|
|
presetId: _preset?.presetId ?? null,
|
|
|
|
|
title: _preset?.title ?? 'New Preset'
|
2023-04-04 01:12:14 +08:00
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return preset;
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
export default cleanupPreset;
|