Initial standalone memabra release
This commit is contained in:
25
tests/test_router_smoke.py
Normal file
25
tests/test_router_smoke.py
Normal file
@@ -0,0 +1,25 @@
|
||||
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"]
|
||||
Reference in New Issue
Block a user