25 lines
910 B
JavaScript
25 lines
910 B
JavaScript
const fs = require('fs');
|
|
const path = require('path');
|
|
const { chatWithAI } = require('../lib/ai-helper');
|
|
const config = require('../config.json');
|
|
|
|
module.exports = async function RestructureScript() {
|
|
console.log('[STATE] RestructureScript');
|
|
|
|
|
|
// Replace with actual context data later. This is place holder for getting a task script that isn't written.
|
|
const taskName = 'example-task';
|
|
const scriptPath = path.join(__dirname, '..', 'bot-tasks', `${taskName}.js`);
|
|
|
|
if (!fs.existsSync(scriptPath)) {
|
|
console.warn(`[AI] Cannot find script: ${taskName}.js`);
|
|
return;
|
|
}
|
|
|
|
const original = fs.readFileSync(scriptPath, 'utf8');
|
|
const prompt = `This script failed repeatedly:\n${original}\nRewrite it to be more reliable. Only return code.`;
|
|
const response = await chatWithAI(prompt, config.ai);
|
|
|
|
fs.writeFileSync(scriptPath, response);
|
|
console.log(`[AI] Rewrote ${taskName}`);
|
|
};
|