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

1
src/serifs.ts
View file

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