26 lines
796 B
Python
26 lines
796 B
Python
from memabra.candidate_types import CandidateObject
|
|
from memabra.router import RuleBasedRouter, TaskContext
|
|
|
|
|
|
def test_router_prefers_memory_for_preference_queries():
|
|
router = RuleBasedRouter()
|
|
decision = router.choose(
|
|
TaskContext(user_input="Remember my preferred deployment region"),
|
|
memory_candidates=[
|
|
CandidateObject(
|
|
id="mem-1",
|
|
type="memory",
|
|
title="Preferred region",
|
|
summary="User prefers us-west-2",
|
|
confidence=0.9,
|
|
freshness=0.8,
|
|
success_rate=0.9,
|
|
)
|
|
],
|
|
skill_candidates=[],
|
|
tool_candidates=[],
|
|
)
|
|
|
|
assert decision.decision_type == "inject_memory"
|
|
assert decision.selected_ids == ["mem-1"]
|