YueMi-Development/RakunNakun-AI
3
0
Fork 0
mirror of https://github.com/NekoMonci12/RakunNakun-AI.git synced 2025-12-19 14:59:15 +00:00
Code Issues Packages Releases Activity

Database Seeder

Browse Source
This commit is contained in:
NekoMonci12
2025-06-03 21:23:28 +07:00
parent 9ba4724282
commit a897c67032
1 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
database.js
View File

@@ -77,16 +77,32 @@ class Database {
try {
await this.pool.query(createModelsQuery);
console.log('Models table is ready or already exists.');
console.log('Models table is ready');
await this.pool.query(createGuildsQuery);
console.log('Guilds table is ready or already exists.');
console.log('Guilds table is ready');
await this.pool.query(createChatLogsQuery);
console.log('ChatLogs table is ready or already exists.');
console.log('ChatLogs table is ready');
await this.pool.query(createApiKeysQuery);
console.log('ApiKeys table is ready or already exists.');
console.log('ApiKeys table is ready');
} catch (error) {
console.error('Error creating tables:', error);
}
await this.seedModels();
}
async seedModels() {
try {
const [rows] = await this.pool.query('SELECT COUNT(*) as count FROM Models');
if (rows[0].count === 0) {
await this.pool.query(`
INSERT INTO Models (MODEL_ID, TOKEN_INPUT, TOKEN_OUTPUT, TOKEN_CACHED_INPUT, TOKEN_CACHED_OUPUT, API_KEY, API_URL)
VALUES ('gpt-4.1', 1, 1, 1, 1, '', 'https://api.llm7.io/v1/chat/completions');
`);
console.log('Seeded Models table.');
}
} catch (error) {
console.error('Error seeding Models table:', error);
}
}
}