docugen/docugen-gpt3.py

45 lines
1.0 KiB
Python
Raw Permalink Normal View History

2022-12-05 20:25:34 -08:00
import openai
import yaml
from yaml import CLoader, CDumper
import os
file_path = os.path.dirname(os.path.realpath(__file__))
with open(f"{file_path}/config.yaml", "r") as file:
config = yaml.load(file, Loader=CLoader)
openai.api_key = config['OPENAI_API_KEY']
prompt = """Write example code that uses this function:
const mineBlock = async (data) => {
if (current_transactions.length != BLOCK_SIZE) return;
console.log("Mining block.")
let x = 5;
let y = 0;
while (true) {
let valid = false;
let val = await digestTxt(`${data}${x*y}`);
for (let c = 0; c < POW_DIFFICULTY; c++) {
res = parseInt(val[val.length - (c+1)]) ?? 100;
if (!(res == 0)) { // check if res character is acceptable
valid = false;
break;
} else {
valid = true;
}
}
if (!valid) {
y += 1;
} else {
console.log(res)
console.log(`Done.\nNonce: ${y}\nHash: ${val}`)
return {val, y};
}
}
}
"""
res = openai.Completion.create(engine="davinci", prompt=prompt)
print(res)