Initial commit: hardened DeerFlow factory

Vendored deer-flow upstream (bytedance/deer-flow) plus prompt-injection
hardening:

- New deerflow.security package: content_delimiter, html_cleaner,
  sanitizer (8 layers — invisible chars, control chars, symbols, NFC,
  PUA, tag chars, horizontal whitespace collapse with newline/tab
  preservation, length cap)
- New deerflow.community.searx package: web_search, web_fetch,
  image_search backed by a private SearX instance, every external
  string sanitized and wrapped in <<<EXTERNAL_UNTRUSTED_CONTENT>>>
  delimiters
- All native community web providers (ddg_search, tavily, exa,
  firecrawl, jina_ai, infoquest, image_search) replaced with hard-fail
  stubs that raise NativeWebToolDisabledError at import time, so a
  misconfigured tool.use path fails loud rather than silently falling
  back to unsanitized output
- Native client back-doors (jina_client.py, infoquest_client.py)
  stubbed too
- Native-tool tests quarantined under tests/_disabled_native/
  (collect_ignore_glob via local conftest.py)
- Sanitizer Layer 7 fix: only collapse horizontal whitespace, preserve
  newlines and tabs so list/table structure survives
- Hardened runtime config.yaml references only the searx-backed tools
- Factory overlay (backend/) kept in sync with deer-flow tree as a
  reference / source

See HARDENING.md for the full audit trail and verification steps.
This commit is contained in:
2026-04-12 14:23:57 +02:00
commit 6de0bf9f5b
889 changed files with 173052 additions and 0 deletions

View File

@@ -0,0 +1,175 @@
"""Tests for memory prompt injection formatting."""
import math
from deerflow.agents.memory.prompt import _coerce_confidence, format_memory_for_injection
def test_format_memory_includes_facts_section() -> None:
memory_data = {
"user": {},
"history": {},
"facts": [
{"content": "User uses PostgreSQL", "category": "knowledge", "confidence": 0.9},
{"content": "User prefers SQLAlchemy", "category": "preference", "confidence": 0.8},
],
}
result = format_memory_for_injection(memory_data, max_tokens=2000)
assert "Facts:" in result
assert "User uses PostgreSQL" in result
assert "User prefers SQLAlchemy" in result
def test_format_memory_sorts_facts_by_confidence_desc() -> None:
memory_data = {
"user": {},
"history": {},
"facts": [
{"content": "Low confidence fact", "category": "context", "confidence": 0.4},
{"content": "High confidence fact", "category": "knowledge", "confidence": 0.95},
],
}
result = format_memory_for_injection(memory_data, max_tokens=2000)
assert result.index("High confidence fact") < result.index("Low confidence fact")
def test_format_memory_respects_budget_when_adding_facts(monkeypatch) -> None:
# Make token counting deterministic for this test by counting characters.
monkeypatch.setattr("deerflow.agents.memory.prompt._count_tokens", lambda text, encoding_name="cl100k_base": len(text))
memory_data = {
"user": {},
"history": {},
"facts": [
{"content": "First fact should fit", "category": "knowledge", "confidence": 0.95},
{"content": "Second fact should not fit in tiny budget", "category": "knowledge", "confidence": 0.90},
],
}
first_fact_only_memory_data = {
"user": {},
"history": {},
"facts": [
{"content": "First fact should fit", "category": "knowledge", "confidence": 0.95},
],
}
one_fact_result = format_memory_for_injection(first_fact_only_memory_data, max_tokens=2000)
two_facts_result = format_memory_for_injection(memory_data, max_tokens=2000)
# Choose a budget that can include exactly one fact section line.
max_tokens = (len(one_fact_result) + len(two_facts_result)) // 2
first_only_result = format_memory_for_injection(memory_data, max_tokens=max_tokens)
assert "First fact should fit" in first_only_result
assert "Second fact should not fit in tiny budget" not in first_only_result
def test_coerce_confidence_nan_falls_back_to_default() -> None:
"""NaN should not be treated as a valid confidence value."""
result = _coerce_confidence(math.nan, default=0.5)
assert result == 0.5
def test_coerce_confidence_inf_falls_back_to_default() -> None:
"""Infinite values should fall back to default rather than clamping to 1.0."""
assert _coerce_confidence(math.inf, default=0.3) == 0.3
assert _coerce_confidence(-math.inf, default=0.3) == 0.3
def test_coerce_confidence_valid_values_are_clamped() -> None:
"""Valid floats outside [0, 1] are clamped; values inside are preserved."""
assert _coerce_confidence(1.5) == 1.0
assert _coerce_confidence(-0.5) == 0.0
assert abs(_coerce_confidence(0.75) - 0.75) < 1e-9
def test_format_memory_skips_none_content_facts() -> None:
"""Facts with content=None must not produce a 'None' line in the output."""
memory_data = {
"facts": [
{"content": None, "category": "knowledge", "confidence": 0.9},
{"content": "Real fact", "category": "knowledge", "confidence": 0.8},
],
}
result = format_memory_for_injection(memory_data, max_tokens=2000)
assert "None" not in result
assert "Real fact" in result
def test_format_memory_skips_non_string_content_facts() -> None:
"""Facts with non-string content (e.g. int/list) must be ignored."""
memory_data = {
"facts": [
{"content": 42, "category": "knowledge", "confidence": 0.9},
{"content": ["list"], "category": "knowledge", "confidence": 0.85},
{"content": "Valid fact", "category": "knowledge", "confidence": 0.7},
],
}
result = format_memory_for_injection(memory_data, max_tokens=2000)
# The formatted line for an integer content would be "- [knowledge | 0.90] 42".
assert "| 0.90] 42" not in result
# The formatted line for a list content would be "- [knowledge | 0.85] ['list']".
assert "| 0.85]" not in result
assert "Valid fact" in result
def test_format_memory_renders_correction_source_error() -> None:
memory_data = {
"facts": [
{
"content": "Use make dev for local development.",
"category": "correction",
"confidence": 0.95,
"sourceError": "The agent previously suggested npm start.",
}
]
}
result = format_memory_for_injection(memory_data, max_tokens=2000)
assert "Use make dev for local development." in result
assert "avoid: The agent previously suggested npm start." in result
def test_format_memory_renders_correction_without_source_error_normally() -> None:
memory_data = {
"facts": [
{
"content": "Use make dev for local development.",
"category": "correction",
"confidence": 0.95,
}
]
}
result = format_memory_for_injection(memory_data, max_tokens=2000)
assert "Use make dev for local development." in result
assert "avoid:" not in result
def test_format_memory_includes_long_term_background() -> None:
"""longTermBackground in history must be injected into the prompt."""
memory_data = {
"user": {},
"history": {
"recentMonths": {"summary": "Recent activity summary"},
"earlierContext": {"summary": "Earlier context summary"},
"longTermBackground": {"summary": "Core expertise in distributed systems"},
},
"facts": [],
}
result = format_memory_for_injection(memory_data, max_tokens=2000)
assert "Background: Core expertise in distributed systems" in result
assert "Recent: Recent activity summary" in result
assert "Earlier: Earlier context summary" in result