Upload files to ''
This commit is contained in:
parent
85f778c05f
commit
cea6bb514b
2 changed files with 105 additions and 0 deletions
84
index.js
Normal file
84
index.js
Normal file
|
@ -0,0 +1,84 @@
|
||||||
|
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}`);
|
||||||
|
});
|
21
package.json
Normal file
21
package.json
Normal file
|
@ -0,0 +1,21 @@
|
||||||
|
{
|
||||||
|
"dependencies": {
|
||||||
|
"node-fetch": "^3.3.2",
|
||||||
|
"ollama-js-client": "^1.0.1",
|
||||||
|
"prompt": "^1.3.0"
|
||||||
|
},
|
||||||
|
"name": "kode",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "Feedback loop to help AI code",
|
||||||
|
"main": "index.js",
|
||||||
|
"type": "module",
|
||||||
|
"scripts": {
|
||||||
|
"test": "echo \"Error: no test specified\" && exit 1"
|
||||||
|
},
|
||||||
|
"repository": {
|
||||||
|
"type": "git",
|
||||||
|
"url": "https://nodemixaholic.com:3002/nodemixaholic/kode"
|
||||||
|
},
|
||||||
|
"author": "Samuel Lord",
|
||||||
|
"license": "SEE LICENSE IN LICENSE"
|
||||||
|
}
|
Loading…
Reference in a new issue