21 lines
674 B
Python
21 lines
674 B
Python
from deerflow.agents.memory.thread_updater import ThreadMemoryUpdater
|
|
|
|
|
|
def test_scrub_sensitive_tolerates_non_numeric_confidence():
|
|
updater = ThreadMemoryUpdater()
|
|
cleaned = updater._scrub_sensitive(
|
|
{
|
|
"user": {},
|
|
"history": {},
|
|
"facts": [
|
|
{"content": "Uses React", "category": "knowledge", "confidence": "high"},
|
|
{"content": "Uses TypeScript", "category": "knowledge", "confidence": None},
|
|
],
|
|
},
|
|
"thread-test",
|
|
)
|
|
|
|
assert len(cleaned["facts"]) == 2
|
|
assert cleaned["facts"][0]["confidence"] == 0.5
|
|
assert cleaned["facts"][1]["confidence"] == 0.5
|