MigrateRepos/glm-free-api
1
0
Fork 0
mirror of https://github.com/LLM-Red-Team/glm-free-api.git synced 2024-09-21 12:14:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

调整prompt

Browse Source
This commit is contained in:
Vinlic 2024-03-17 16:58:04 +08:00
parent 14e1f6a43a
commit 613f204f1a
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/api/controllers/chat.ts
View File

@ -304,7 +304,7 @@ function extractRefFileUrls(messages: any[]) {
* @param messages 参考gpt系列消息格式，多轮对话请完整提供上下文 * @param messages 参考gpt系列消息格式，多轮对话请完整提供上下文
*/ */
function messagesPrepare(messages: any[], refs: any[]) { function messagesPrepare(messages: any[], refs: any[]) {
const headPrompt = '设定：【重要】回答用户问题，禁止输出双方的名字\n'; const headPrompt = '使用”你“这个角色回复”我“这个角色，以第一人称对话\n';
const content = messages.reduce((content, message) => { const content = messages.reduce((content, message) => {
if (_.isArray(message.content)) { if (_.isArray(message.content)) {
return message.content.reduce((_content, v) => { return message.content.reduce((_content, v) => {
@ -313,9 +313,10 @@ function messagesPrepare(messages: any[], refs: any[]) {
return _content + (v['text'] || ''); return _content + (v['text'] || '');
}, content); }, content);
} }
const role = message.role.replace('user', '我').replace('assistant', 'ChatGLM') || '我'; const role = message.role.replace('user', '我').replace('assistant', '你') || '我';
return content += `${role}:${message.content}\n`; return content += `${role}:${message.content}\n你:`;
}, ''); }, '');
console.log(content)
return [ return [
{ {
role: 'user', role: 'user',