furyhawk/deer-flow
1
0
Fork 0
mirror of https://github.com/bytedance/deer-flow.git synced 2026-05-24 17:06:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix the unit tests error of agent provider

Browse Source
This commit is contained in:
Willem Jiang
2026-04-26 22:09:12 +08:00
parent ac18b9c424
commit 60754f0c50
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/tests/test_checkpointer.py
+4 -4
View File
@@ -174,9 +174,9 @@ class TestGetCheckpointer:
with ( with (
patch.dict(sys.modules, {"langgraph.checkpoint.sqlite": mock_module}), patch.dict(sys.modules, {"langgraph.checkpoint.sqlite": mock_module}),
patch("deerflow.agents.checkpointer.provider.ensure_sqlite_parent_dir") as mock_ensure, patch("deerflow.runtime.checkpointer.provider.ensure_sqlite_parent_dir") as mock_ensure,
patch( patch(
"deerflow.agents.checkpointer.provider.resolve_sqlite_conn_str", "deerflow.runtime.checkpointer.provider.resolve_sqlite_conn_str",
return_value="/tmp/resolved/relative/test.db", return_value="/tmp/resolved/relative/test.db",
), ),
): ):
@@ -210,11 +210,11 @@ class TestGetCheckpointer:
with ( with (
patch.dict(sys.modules, {"langgraph.checkpoint.sqlite": mock_module}), patch.dict(sys.modules, {"langgraph.checkpoint.sqlite": mock_module}),
patch( patch(
"deerflow.agents.checkpointer.provider.ensure_sqlite_parent_dir", "deerflow.runtime.checkpointer.provider.ensure_sqlite_parent_dir",
side_effect=record_ensure, side_effect=record_ensure,
), ),
patch( patch(
"deerflow.agents.checkpointer.provider.resolve_sqlite_conn_str", "deerflow.runtime.checkpointer.provider.resolve_sqlite_conn_str",
return_value="/tmp/resolved/relative/test.db", return_value="/tmp/resolved/relative/test.db",
), ),
): ):
backend/tests/test_lead_agent_prompt.py
+4 -4
View File
@@ -84,7 +84,7 @@ def test_refresh_skills_system_prompt_cache_async_reloads_immediately(monkeypatc
state = {"skills": [make_skill("first-skill")]} state = {"skills": [make_skill("first-skill")]}
monkeypatch.setattr(prompt_module, "load_skills", lambda enabled_only=True: list(state["skills"])) monkeypatch.setattr(prompt_module, "load_skills", lambda enabled_only=True: list(state["skills"]))
prompt_module._reset_skills_system_prompt_cache_state() prompt_module.clear_skills_system_prompt_cache()
try: try:
prompt_module.warm_enabled_skills_cache() prompt_module.warm_enabled_skills_cache()
@@ -95,7 +95,7 @@ def test_refresh_skills_system_prompt_cache_async_reloads_immediately(monkeypatc
assert [skill.name for skill in prompt_module._get_enabled_skills()] == ["second-skill"] assert [skill.name for skill in prompt_module._get_enabled_skills()] == ["second-skill"]
finally: finally:
prompt_module._reset_skills_system_prompt_cache_state() prompt_module.clear_skills_system_prompt_cache()
def test_clear_cache_does_not_spawn_parallel_refresh_workers(monkeypatch, tmp_path): def test_clear_cache_does_not_spawn_parallel_refresh_workers(monkeypatch, tmp_path):
@@ -137,7 +137,7 @@ def test_clear_cache_does_not_spawn_parallel_refresh_workers(monkeypatch, tmp_pa
return [make_skill(f"skill-{current_call}")] return [make_skill(f"skill-{current_call}")]
monkeypatch.setattr(prompt_module, "load_skills", fake_load_skills) monkeypatch.setattr(prompt_module, "load_skills", fake_load_skills)
prompt_module._reset_skills_system_prompt_cache_state() prompt_module.clear_skills_system_prompt_cache()
try: try:
prompt_module.clear_skills_system_prompt_cache() prompt_module.clear_skills_system_prompt_cache()
@@ -151,7 +151,7 @@ def test_clear_cache_does_not_spawn_parallel_refresh_workers(monkeypatch, tmp_pa
assert [skill.name for skill in prompt_module._get_enabled_skills()] == ["skill-2"] assert [skill.name for skill in prompt_module._get_enabled_skills()] == ["skill-2"]
finally: finally:
release.set() release.set()
prompt_module._reset_skills_system_prompt_cache_state() prompt_module.clear_skills_system_prompt_cache()
def test_warm_enabled_skills_cache_logs_on_timeout(monkeypatch, caplog): def test_warm_enabled_skills_cache_logs_on_timeout(monkeypatch, caplog):