github/deer-flow
1
0
Fork 0
mirror of https://github.com/bytedance/deer-flow.git synced 2026-10-26 21:26:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(harness): anchor DeerFlowClient agent-name validation with fullmatch (#6104)

Browse Source
This commit is contained in:
Fish 2026-09-30 19:51:47 +08:00 committed by GitHub
parent 25671538af
commit 2000bffa8a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/packages/harness/deerflow/client.py
View File

@ -254,7 +254,7 @@ class DeerFlowClient:
self._checkpoint_channel_mode = freeze_checkpoint_channel_mode(self._app_config.database.checkpoint_channel_mode)
self._checkpoint_snapshot_frequency = freeze_checkpoint_snapshot_frequency(self._app_config.database.checkpoint_delta.snapshot_frequency)
if agent_name is not None and not AGENT_NAME_PATTERN.match(agent_name):
if agent_name is not None and not AGENT_NAME_PATTERN.fullmatch(agent_name):
raise ValueError(f"Invalid agent name '{agent_name}'. Must match pattern: {AGENT_NAME_PATTERN.pattern}")
self._checkpointer = checkpointer

7
backend/tests/test_client.py
View File

@ -121,6 +121,13 @@ class TestClientInit:
with pytest.raises(ValueError, match="Invalid agent name"):
DeerFlowClient(agent_name="../path/traversal")
def test_agent_name_with_trailing_newline_rejected(self, mock_app_config):
with patch("deerflow.client.get_app_config", return_value=mock_app_config):
# The client's own guard must reject this at construction; the
# memory store's later fullmatch check uses different phrasing.
with pytest.raises(ValueError, match="Must match pattern"):
DeerFlowClient(agent_name="reviewer\n")
def test_custom_config_path(self, mock_app_config):
with (
patch("deerflow.client.reload_app_config") as mock_reload,