2023-08-06 22:24:49 +00:00
|
|
|
import { db } from "./db.mjs"
|
|
|
|
|
2023-08-09 10:26:08 +00:00
|
|
|
export default async function getNewPrompt({ minCount = 200000, maxCount = 30000000, rarityBias = 0.5 }) {
|
2023-08-06 22:24:49 +00:00
|
|
|
const badWords = await db('bad_words')
|
|
|
|
.select('word')
|
|
|
|
|
|
|
|
const prompts = await db('dictionary')
|
|
|
|
.select('*')
|
|
|
|
.where({
|
|
|
|
derivative: 0,
|
|
|
|
scientific: 0,
|
|
|
|
})
|
|
|
|
.andWhere('count', '<', maxCount)
|
|
|
|
.andWhere('count', '>', minCount)
|
|
|
|
.andWhere('word', 'not in', badWords)
|
|
|
|
.whereNotNull('pronunciation')
|
2023-08-06 22:59:24 +00:00
|
|
|
.orderByRaw('count desc')
|
2023-08-08 14:52:17 +00:00
|
|
|
|
|
|
|
const getBiasedRng = (min, max, bias, influence) => {
|
|
|
|
const random = Math.random() * (max - min) + min
|
|
|
|
const mix = Math.random() * influence
|
|
|
|
return random * (1 - mix) + bias * mix
|
|
|
|
}
|
2023-08-06 22:24:49 +00:00
|
|
|
|
|
|
|
const randomEntry = (array) => {
|
2023-08-09 10:26:08 +00:00
|
|
|
const random = getBiasedRng(0, 1, rarityBias, 1)
|
2023-08-06 22:59:24 +00:00
|
|
|
const mix = Math.random()
|
2023-08-06 22:24:49 +00:00
|
|
|
return array[
|
|
|
|
parseInt(
|
|
|
|
array.length * random
|
|
|
|
)
|
|
|
|
]
|
|
|
|
}
|
2023-08-06 22:59:24 +00:00
|
|
|
|
2023-08-09 10:26:08 +00:00
|
|
|
await db.destroy()
|
|
|
|
return await randomEntry(prompts)
|
2023-08-06 22:24:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
2023-08-09 10:26:08 +00:00
|
|
|
//console.log(await getNewPrompt({}))
|
2023-08-06 22:59:24 +00:00
|
|
|
|
2023-08-06 22:24:49 +00:00
|
|
|
|
|
|
|
|