mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-10-26 13:16:24 +00:00
1098fb4c60
* fix(harness): scope deferred tool promotion to user turn Signed-off-by: CorgiBoyG <111257566+CorgiBoyG@users.noreply.github.com> * fix(harness): preserve legacy promotions during routing resume --------- Signed-off-by: CorgiBoyG <111257566+CorgiBoyG@users.noreply.github.com> Co-authored-by: Willem Jiang <willem.jiang@gmail.com>
76 lines
2.9 KiB
Python
76 lines
2.9 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"}
|