Cookies_Github_mirror/AquaDX
4
0
Fork 1
mirror of https://github.com/MewoLab/AquaDX.git synced 2026-02-08 04:57:26 +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> { suspend fun isSafeBatch(rawContents: List<String>): List<Boolean> {
val contents = rawContents.map { Normalizer.normalize(it, Normalizer.Form.NFKC) } 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() val map = safety.findAll().associateBy { it.content.lowercase().trim() }.toMutableMap()
// Process unseen content with OpenAI // Process unseen content with OpenAI
@@ -72,8 +73,9 @@ class AquaNetSafetyService(
} }
} }
if (news.isNotEmpty()) safety.saveAll(news) if (news.isNotEmpty()) safety.saveAll(news)
news.associateByTo(origMap) { it.content }
news.associateByTo(map) { it.content.lowercase().trim() } 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 }
} }
} }