mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-11-15 07:41:20 +00:00
3e2489429f
* fix(gateway): reject an unusable promoted record at the input boundary * fix(gateway): refuse an Overwrite-marker promotion record * test(gateway): pin the server-owned metadata strip on promoted --------- Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
90 lines
3.5 KiB
Python
90 lines
3.5 KiB
Python
from deerflow.agents.thread_state import merge_promoted
|
|
|
|
|
|
def test_merge_promoted_preserves_existing_when_new_is_none():
|
|
existing = {"catalog_hash": "abc", "names": ["search"]}
|
|
|
|
assert merge_promoted(existing, None) is existing
|
|
|
|
|
|
def test_merge_promoted_preserves_existing_when_new_is_empty_dict():
|
|
existing = {"catalog_hash": "abc", "names": ["search"]}
|
|
|
|
assert merge_promoted(existing, {}) is existing
|
|
|
|
|
|
def test_merge_promoted_replaces_none_existing_with_deduplicated_new_names():
|
|
result = merge_promoted(None, {"catalog_hash": "abc", "names": ["search", "search", "fetch"]})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["search", "fetch"]}
|
|
|
|
|
|
def test_merge_promoted_replaces_when_catalog_hash_changes():
|
|
existing = {"catalog_hash": "abc", "names": ["old"]}
|
|
|
|
result = merge_promoted(existing, {"catalog_hash": "def", "names": ["new", "new", "old"]})
|
|
|
|
assert result == {"catalog_hash": "def", "names": ["new", "old"]}
|
|
|
|
|
|
def test_merge_promoted_unions_names_when_catalog_hash_matches():
|
|
existing = {"catalog_hash": "abc", "names": ["search", "fetch"]}
|
|
|
|
result = merge_promoted(existing, {"catalog_hash": "abc", "names": ["fetch", "scrape"]})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["search", "fetch", "scrape"]}
|
|
|
|
|
|
def test_merge_promoted_replaces_malformed_existing_without_crash():
|
|
# A forward-incompatible / externally-injected persisted state could lack
|
|
# catalog_hash; the reducer must treat it as a mismatch and replace, not crash.
|
|
result = merge_promoted({"names": ["stale"]}, {"catalog_hash": "abc", "names": ["search"]})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["search"]}
|
|
|
|
|
|
def test_merge_promoted_unions_names_within_same_turn():
|
|
existing = {"catalog_hash": "abc", "names": ["search"], "turn_id": "user-1"}
|
|
|
|
result = merge_promoted(existing, {"catalog_hash": "abc", "names": ["fetch"], "turn_id": "user-1"})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["search", "fetch"], "turn_id": "user-1"}
|
|
|
|
|
|
def test_merge_promoted_replaces_names_on_new_turn():
|
|
existing = {"catalog_hash": "abc", "names": ["search"], "turn_id": "user-1"}
|
|
|
|
result = merge_promoted(existing, {"catalog_hash": "abc", "names": ["fetch"], "turn_id": "user-2"})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["fetch"], "turn_id": "user-2"}
|
|
|
|
|
|
def test_merge_promoted_tool_search_update_inherits_current_turn():
|
|
existing = {"catalog_hash": "abc", "names": [], "turn_id": "user-1"}
|
|
|
|
result = merge_promoted(existing, {"catalog_hash": "abc", "names": ["search"]})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["search"], "turn_id": "user-1"}
|
|
|
|
|
|
def test_merge_promoted_recovers_from_malformed_existing_names():
|
|
existing = {"catalog_hash": "abc", "names": "not-a-list", "turn_id": "user-1"}
|
|
|
|
result = merge_promoted(existing, {"catalog_hash": "abc", "names": ["search"]})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["search"], "turn_id": "user-1"}
|
|
|
|
|
|
def test_merge_promoted_replaces_non_object_existing():
|
|
# A checkpoint poisoned before the gateway validated this channel stores a
|
|
# raw Overwrite value; genuine promotion writes must not keep failing on it.
|
|
result = merge_promoted("oops", {"catalog_hash": "abc", "names": ["search"]})
|
|
|
|
assert result == {"catalog_hash": "abc", "names": ["search"]}
|
|
|
|
|
|
def test_merge_promoted_ignores_non_object_update():
|
|
existing = {"catalog_hash": "abc", "names": ["search"], "turn_id": "user-1"}
|
|
|
|
assert merge_promoted(existing, "oops") is existing
|