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

fix #4

Browse source
This commit is contained in:
tetsuya-ki 2024-08-22 09:44:26 +09:00
parent 260621950a
commit 634215765f
2 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -16,8 +16,8 @@ type Base64Image = {
type: string;
base64: string;
};
const GEMINI_API = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent';
const GEMINI_VISION_API = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-pro-vision:generateContent';
const GEMINI_15_FLASH_API = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent';
const GEMINI_15_PRO_API = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro:generateContent';
export default class extends Module {
public readonly name = 'aichat';
@ -131,6 +131,7 @@ export default class extends Module {
type = 'chatgpt3.5';
}
const question = msg.extractedText
.toLowerCase()
.replace(this.name, '')
.replace(kigo + type, '')
.trim();
@ -142,6 +143,7 @@ export default class extends Module {
}
switch(type) {
case 'gemini':
// geminiの場合、APIキーが必須
if (!config.geminiProApiKey) {
msg.reply(serifs.aichat.nothing(type));
return false;
@ -150,11 +152,11 @@ export default class extends Module {
aiChat = {
question: question,
prompt: prompt,
api: GEMINI_API,
api: GEMINI_15_PRO_API,
key: config.geminiProApiKey
};
if (base64Image !== null) {
aiChat.api = GEMINI_VISION_API;
if (msg.includes([kigo + 'gemini-flash'])) {
aiChat.api = GEMINI_15_FLASH_API;
}
text = await this.genTextByGemini(aiChat, base64Image);
break;
@ -165,7 +167,7 @@ export default class extends Module {
if (text == null) {
this.log('The result is invalid. It seems that tokens and other items need to be reviewed.')
msg.reply(serifs.aichat.nothing(type));
msg.reply(serifs.aichat.error(type));
return false;
}

1
src/serifs.ts
View file

@ -390,6 +390,7 @@ export default {
aichat: {
nothing: type => `あぅ... ${type}のAPIキーが登録されてないみたいです`,
error: type => `うぇ...${type}でエラーが発生しちゃったみたいです。gemini-flashだと動くかも？`,
post: (text, type) => `${text} (${type}) #aichat`,
},