Cookies_Github_mirror/AquaDX
4
0
Fork 1
mirror of https://github.com/MewoLab/AquaDX.git synced 2026-02-05 01:37:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

[F] Fix safety :(

Browse Source
This commit is contained in:
Azalea
2024-12-26 07:55:54 -05:00
parent b7c8fba464
commit 812d910212
1 changed files with 3 additions and 1 deletions
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 }
}
}