wanwu/deer-flow
1
0
Fork 0
mirror of https://gitee.com/wanwujie/deer-flow synced 2026-04-22 05:34:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into fix-968

Browse Source
This commit is contained in:
Willem Jiang
2026-03-06 09:45:04 +08:00
committed by GitHub
parent 24a8ea76ee 3a5e0b935d
commit a8963f3ac7
3 changed files with 133 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
backend/tests/test_custom_agent.py
View File

@@ -305,23 +305,35 @@ class TestMemoryFilePath:
def test_global_memory_path(self, tmp_path):
"""None agent_name should return global memory file."""
import src.agents.memory.updater as updater_mod
from src.config.memory_config import MemoryConfig
with patch("src.agents.memory.updater.get_paths", return_value=_make_paths(tmp_path)):
with (
patch("src.agents.memory.updater.get_paths", return_value=_make_paths(tmp_path)),
patch("src.agents.memory.updater.get_memory_config", return_value=MemoryConfig(storage_path="")),
):
path = updater_mod._get_memory_file_path(None)
assert path == tmp_path / "memory.json"
def test_agent_memory_path(self, tmp_path):
"""Providing agent_name should return per-agent memory file."""
import src.agents.memory.updater as updater_mod
from src.config.memory_config import MemoryConfig
with patch("src.agents.memory.updater.get_paths", return_value=_make_paths(tmp_path)):
with (
patch("src.agents.memory.updater.get_paths", return_value=_make_paths(tmp_path)),
patch("src.agents.memory.updater.get_memory_config", return_value=MemoryConfig(storage_path="")),
):
path = updater_mod._get_memory_file_path("code-reviewer")
assert path == tmp_path / "agents" / "code-reviewer" / "memory.json"
def test_different_paths_for_different_agents(self, tmp_path):
import src.agents.memory.updater as updater_mod
from src.config.memory_config import MemoryConfig
with patch("src.agents.memory.updater.get_paths", return_value=_make_paths(tmp_path)):
with (
patch("src.agents.memory.updater.get_paths", return_value=_make_paths(tmp_path)),
patch("src.agents.memory.updater.get_memory_config", return_value=MemoryConfig(storage_path="")),
):
path_global = updater_mod._get_memory_file_path(None)
path_a = updater_mod._get_memory_file_path("agent-a")
path_b = updater_mod._get_memory_file_path("agent-b")