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

refactor(impression): add feature attribute as index, which will work with high-confidence impressions

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

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

@@ -8,11 +8,13 @@ import kotlin.concurrent.withLock
class Entity @JvmOverloads constructor(
val uuid: String = UUID.randomUUID().toString(),
private val relations: MutableMap<String, MutableMap<String, Double>> = mutableMapOf(),
private val impressions: MutableMap<String, ImpressionData> = mutableMapOf()
private val impressions: MutableMap<String, IndexableData> = mutableMapOf(),
private val features: MutableMap<String, IndexableData> = mutableMapOf()
) {
private val impressionLock = ReentrantLock()
private val relationLock = ReentrantLock()
private val featureLock = ReentrantLock()
@JvmOverloads
fun updateRelation(
@@ -27,19 +29,39 @@ class Entity @JvmOverloads constructor(
fun updateImpression(
impression: String,
newImpression: String? = null,
strength: Double = 1.0
): ImpressionData = impressionLock.withLock {
confidence: Double = 1.0
): IndexableData = impressionLock.withLock {
if (newImpression == null) {
impressions.computeIfAbsent(impression) { ImpressionData(strength) }
.also { it.confidence = strength }
impressions.computeIfAbsent(impression) { IndexableData(confidence) }
.also { it.confidence = confidence }
} else {
impressions.remove(impression)
ImpressionData(strength).also {
IndexableData(confidence).also {
impressions[newImpression] = it
}
}.also {
if (it.confidence >= 0.9) {
featureLock.withLock { features[impression] = it }
}
}
}
fun updateFeature(feature: String, newFeature: String? = null, confidence: Double = 1.0) = featureLock.withLock {
if (newFeature == null) {
features.computeIfAbsent(feature){ IndexableData(confidence) }
.also { it.confidence = confidence }
}else{
features.remove(feature)
IndexableData(confidence).also {
features[newFeature] = it
}
}
}
fun removeFeature(feature: String) = featureLock.withLock {
features.remove(feature)
}
fun removeImpression(impression: String) = impressionLock.withLock {
impressions.remove(impression)
}
@@ -78,7 +100,7 @@ class Entity @JvmOverloads constructor(
}.toSet()
}
data class ImpressionData(
data class IndexableData(
var confidence: Double
) {
private val vectors: ConcurrentHashMap<String, DoubleArray> = ConcurrentHashMap()