pch_xyz/ai
1
0
Fork 0
mirror of https://github.com/syuilo/ai.git synced 2025-03-25 21:12:56 +00:00
Code Issues Projects Releases Packages Wiki Activity

aichatに対話相手の名前を渡す(friend.nameが優先)

Browse source
This commit is contained in:
tetsuya-ki 2025-01-13 15:36:19 +09:00
parent c4cf8c27fe
commit d2c6628f16
1 changed files with 25 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
src/modules/aichat/index.ts
View file

@ -14,6 +14,7 @@ type AiChat = {
api: string; api: string;
key: string; key: string;
history?: { role: string; content: string }[]; history?: { role: string; content: string }[];
friendName?: string;
}; };
type base64File = { type base64File = {
type: string; type: string;
@ -117,6 +118,10 @@ export default class extends Module {
}); });
// 設定のプロンプトに加え、現在時刻を渡す // 設定のプロンプトに加え、現在時刻を渡す
let systemInstructionText = aiChat.prompt + "。また、現在日時は" + now + "である(他の日時は無効とすること)。"; let systemInstructionText = aiChat.prompt + "。また、現在日時は" + now + "である(他の日時は無効とすること)。";
// 名前を伝えておく
if (aiChat.friendName != undefined) {
systemInstructionText += "なお、会話相手の名前は" + aiChat.friendName + "とする。";
}
const systemInstruction: GeminiSystemInstruction = {role: 'system', parts: [{text: systemInstructionText}]}; const systemInstruction: GeminiSystemInstruction = {role: 'system', parts: [{text: systemInstructionText}]};
parts = [{text: aiChat.question}]; parts = [{text: aiChat.question}];
@ -458,6 +463,20 @@ export default class extends Module {
reKigoType = RegExp(KIGO + GEMINI_PRO, 'i'); reKigoType = RegExp(KIGO + GEMINI_PRO, 'i');
} }
const friend: Friend | null = this.ai.lookupFriend(msg.userId);
this.log("msg.userId:"+msg.userId);
let friendName: string | undefined;
if (friend != null && friend.name != null) {
friendName = friend.name;
this.log("friend.name:" + friend.name);
} else if (msg.user.name) {
friendName = msg.user.name;
this.log("msg.user.username:" + msg.user.username);
} else {
friendName = msg.user.username;
this.log("msg.user.username:" + msg.user.username);
}
const question = extractedText const question = extractedText
.replace(reName, '') .replace(reName, '')
.replace(reKigoType, '') .replace(reKigoType, '')
@ -475,7 +494,8 @@ export default class extends Module {
prompt: prompt, prompt: prompt,
api: GEMINI_20_FLASH_API, api: GEMINI_20_FLASH_API,
key: config.geminiProApiKey, key: config.geminiProApiKey,
history: exist.history history: exist.history,
friendName: friendName
}; };
if (exist.api) { if (exist.api) {
aiChat.api = exist.api aiChat.api = exist.api
@ -494,7 +514,8 @@ export default class extends Module {
prompt: prompt, prompt: prompt,
api: PLAMO_API, api: PLAMO_API,
key: config.pLaMoApiKey, key: config.pLaMoApiKey,
history: exist.history history: exist.history,
friendName: friendName
}; };
text = await this.genTextByPLaMo(aiChat); text = await this.genTextByPLaMo(aiChat);
break; break;
@ -527,7 +548,8 @@ export default class extends Module {
createdAt: Date.now(), createdAt: Date.now(),
type: exist.type, type: exist.type,
api: aiChat.api, api: aiChat.api,
history: exist.history history: exist.history,
friendName: friendName
}); });
this.log('Subscribe&Set Timer...'); this.log('Subscribe&Set Timer...');