Tighten context onboarding and actuator suggestions
This commit is contained in:
@@ -57,7 +57,7 @@ _STOPWORDS = frozenset(
|
||||
"value",
|
||||
}
|
||||
)
|
||||
_GENERIC_AREA_NAMES = frozenset({"monitoring", "system", "technik"})
|
||||
_GENERIC_AREA_NAMES = frozenset({"energie", "monitoring", "power", "strom", "system", "technik"})
|
||||
_NUMERIC_AUTO_ACCEPT_SCORE = 0.82
|
||||
_NUMERIC_AUTO_ACCEPT_MIN_SCORE = 0.5
|
||||
_NUMERIC_MIN_MARGIN = 0.18
|
||||
@@ -88,6 +88,7 @@ _DIAGNOSTIC_TOKENS = frozenset({
|
||||
"diagnostic",
|
||||
"firmware",
|
||||
"gesehen",
|
||||
"heat",
|
||||
"last",
|
||||
"linkquality",
|
||||
"knoten",
|
||||
@@ -100,10 +101,28 @@ _DIAGNOSTIC_TOKENS = frozenset({
|
||||
"signal",
|
||||
"ssid",
|
||||
"status",
|
||||
"overheat",
|
||||
"overheating",
|
||||
"overload",
|
||||
"uptime",
|
||||
"uberhitzung",
|
||||
"ueberhitzung",
|
||||
"ueberlast",
|
||||
"überhitzung",
|
||||
"überlast",
|
||||
"wifi",
|
||||
"zuletzt",
|
||||
})
|
||||
_AUTO_CONTEXT_CLASSES = frozenset({
|
||||
"door",
|
||||
"garage_door",
|
||||
"illuminance",
|
||||
"motion",
|
||||
"occupancy",
|
||||
"opening",
|
||||
"presence",
|
||||
"window",
|
||||
})
|
||||
|
||||
|
||||
class ActuatorReconciliationService:
|
||||
@@ -628,8 +647,12 @@ class ActuatorReconciliationService:
|
||||
if context
|
||||
else _NUMERIC_AUTO_ACCEPT_MIN_SCORE
|
||||
)
|
||||
can_auto_accept_context = (
|
||||
not context or _eligible_for_auto_context(actuator, candidate)
|
||||
)
|
||||
auto_accepted = (
|
||||
candidate.score >= minimum_score
|
||||
can_auto_accept_context
|
||||
and candidate.score >= minimum_score
|
||||
and confidence >= auto_score
|
||||
and (context or margin >= _NUMERIC_MIN_MARGIN)
|
||||
)
|
||||
@@ -743,6 +766,23 @@ def _has_context_relationship(actuator: HaEntitySummary, entity: HaEntitySummary
|
||||
)
|
||||
|
||||
|
||||
def _eligible_for_auto_context(
|
||||
actuator: HaEntitySummary,
|
||||
candidate: AssignmentCandidate,
|
||||
) -> bool:
|
||||
device_class = candidate.device_class or ""
|
||||
if device_class in _AUTO_CONTEXT_CLASSES:
|
||||
return True
|
||||
if (
|
||||
actuator.device_name
|
||||
and candidate.device_name
|
||||
and actuator.device_name == candidate.device_name
|
||||
and candidate.domain in {"light", "switch"}
|
||||
):
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def _score_candidate(
|
||||
actuator: HaEntitySummary,
|
||||
entity: HaEntitySummary,
|
||||
|
||||
Reference in New Issue
Block a user