lzc256
/
AquaDX
mirror of https://github.com/hykilpikonna/AquaDX
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

[F] Fix safety :(

matching
Azalea 2024-12-26 07:55:54 -05:00
parent b7c8fba464
commit 812d910212
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/icu/samnyan/aqua/net/Safety.kt
View File

@ -54,6 +54,7 @@ class AquaNetSafetyService(
*/
suspend fun isSafeBatch(rawContents: List<String>): List<Boolean> {
val contents = rawContents.map { Normalizer.normalize(it, Normalizer.Form.NFKC) }
val origMap = safety.findAll().associateBy { it.content }.toMutableMap()
val map = safety.findAll().associateBy { it.content.lowercase().trim() }.toMutableMap()
// Process unseen content with OpenAI
@ -72,8 +73,9 @@ class AquaNetSafetyService(
}
}
if (news.isNotEmpty()) safety.saveAll(news)
news.associateByTo(origMap) { it.content }
news.associateByTo(map) { it.content.lowercase().trim() }
return contents.map { map[it.lowercase().trim()]!!.safe }
return contents.map { map[it.lowercase().trim()]?.safe ?: origMap[it]?.safe ?: true }
}
}