slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-06-27 17:49:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(impression): correct impression update behavior

Browse Source
This commit is contained in:
slhafzjw
2026-05-23 23:22:15 +08:00
parent 047d1b56fe
commit 26ef5d875d
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
Partner-Core/src/main/java/work/slhaf/partner/core/cognition/impression/Entity.kt
View File

@@ -33,24 +33,28 @@ class Entity @JvmOverloads constructor(
): IndexableData = impressionLock.withLock { ): IndexableData = impressionLock.withLock {
if (newImpression == null) { if (newImpression == null) {
impressions.computeIfAbsent(impression) { IndexableData(confidence) } impressions.computeIfAbsent(impression) { IndexableData(confidence) }
.also { it.confidence = confidence } .also {
it.confidence = confidence
if (it.confidence >= 0.9) {
featureLock.withLock { features[impression] = it }
}
}
} else { } else {
impressions.remove(impression) impressions.remove(impression)
IndexableData(confidence).also { IndexableData(confidence).also {
impressions[newImpression] = it impressions[newImpression] = it
} if (it.confidence >= 0.9) {
}.also { featureLock.withLock { features[newImpression] = it }
if (it.confidence >= 0.9) { }
featureLock.withLock { features[impression] = it }
} }
} }
} }
fun updateFeature(feature: String, newFeature: String? = null, confidence: Double = 1.0) = featureLock.withLock { fun updateFeature(feature: String, newFeature: String? = null, confidence: Double = 1.0) = featureLock.withLock {
if (newFeature == null) { if (newFeature == null) {
features.computeIfAbsent(feature){ IndexableData(confidence) } features.computeIfAbsent(feature) { IndexableData(confidence) }
.also { it.confidence = confidence } .also { it.confidence = confidence }
}else{ } else {
features.remove(feature) features.remove(feature)
IndexableData(confidence).also { IndexableData(confidence).also {
features[newFeature] = it features[newFeature] = it