84 lines
2.5 KiB
JavaScript
84 lines
2.5 KiB
JavaScript
import readline from 'readline';
|
|
import Ollama from 'ollama-js-client';
|
|
import spawn from 'child_process';
|
|
|
|
var answerParsed = ""
|
|
|
|
let generation = 1;
|
|
|
|
function prompt(q) {
|
|
const rl = readline.createInterface({
|
|
input: process.stdin,
|
|
output: process.stdout,
|
|
});
|
|
|
|
return new Promise((resolve) => {
|
|
rl.question(`${q}`, function (a) {
|
|
rl.close();
|
|
console.log("coding!");
|
|
resolve(a);
|
|
});
|
|
});
|
|
}
|
|
|
|
let problem = await prompt("What's the project? (no external libs or reqs): ");
|
|
|
|
let lang = await prompt("What's the lang? (js, python, ppython [panda3d python]): ");
|
|
|
|
function langExec(langCode) {
|
|
if (lang == "js") {
|
|
return eval(langCode);
|
|
} else if (lang == "python") {
|
|
const pythonProcess = spawn('python', ['-c', langCode]);
|
|
// Handle stderr data from the Python process
|
|
return pythonProcess.stderr.on('data', (data) => {
|
|
return Error(`${data}`);
|
|
});
|
|
} else if (lang == "ppython") {
|
|
const ppythonProcess = spawn('ppython', ['-c', langCode]);
|
|
// Handle stderr data from the Python process
|
|
return ppythonProcess.stderr.on('data', (data) => {
|
|
return Error(`${data}`);
|
|
});
|
|
} else {
|
|
console.error("Language command not found!")
|
|
}
|
|
}
|
|
|
|
const instance = new Ollama({
|
|
model: "codellama",
|
|
url: "http://127.0.0.1:11434/api/",
|
|
});
|
|
|
|
function getLangID() {
|
|
if (lang == "ppython") {
|
|
return "panda3d python"
|
|
} else {
|
|
return lang;
|
|
}
|
|
}
|
|
|
|
let answer = await instance.prompt(`${problem} - This must be coded in pure ${getLangID()}, no external libraries or requirements. Please provide the code, the full code, and nothing but the code. No chit-chat, no markdown, just code.`);
|
|
|
|
async function main() {
|
|
|
|
let problemSolved = false;
|
|
while (problemSolved == false) {
|
|
try {
|
|
console.log(`Generation ${generation}`)
|
|
console.log(answer.response)
|
|
answerParsed = answer.response.replaceAll("```javascript","").replaceAll("```","");
|
|
langExec(answerParsed);
|
|
problemSolved = true;
|
|
generation = generation + 1;
|
|
} catch (error) {
|
|
answer = await instance.prompt(`There was an error: ${error.message}. Please only provide the code, the full code, and nothing but the code. No chit-chat, no markdown, just code. Also, make sure it's written in ${getLangID()} without any libraries besides included.`)
|
|
}
|
|
}
|
|
}
|
|
|
|
main().then(() => {
|
|
console.log(`!!!ANSWER COMPUTED!!!
|
|
|
|
${answerParsed}`);
|
|
});
|