diff --git a/api/tests/unit_tests/core/app/apps/agent_app/test_runtime_request_builder.py b/api/tests/unit_tests/core/app/apps/agent_app/test_runtime_request_builder.py index f8fe4dc942e..61bf3443cc8 100644 --- a/api/tests/unit_tests/core/app/apps/agent_app/test_runtime_request_builder.py +++ b/api/tests/unit_tests/core/app/apps/agent_app/test_runtime_request_builder.py @@ -348,9 +348,7 @@ class TestAgentAppDriveLayer: assert names.index(DIFY_SHELL_LAYER_ID) == names.index("execution_context") + 1 assert names.index("drive") == names.index(DIFY_SHELL_LAYER_ID) + 1 - def test_drive_layer_injected_with_empty_catalog_and_drive_depends_on_shell( - self, monkeypatch: pytest.MonkeyPatch - ): + def test_drive_layer_injected_with_empty_catalog_and_drive_depends_on_shell(self, monkeypatch: pytest.MonkeyPatch): monkeypatch.setattr( "core.app.apps.agent_app.runtime_request_builder.dify_config.AGENT_DRIVE_MANIFEST_ENABLED", True ) diff --git a/dify-agent/src/dify_agent/layers/drive/layer.py b/dify-agent/src/dify_agent/layers/drive/layer.py index d370501ae0e..e6941b50cbe 100644 --- a/dify-agent/src/dify_agent/layers/drive/layer.py +++ b/dify-agent/src/dify_agent/layers/drive/layer.py @@ -171,9 +171,7 @@ class DifyDriveLayer(PlainLayer[DifyDriveDeps, DifyDriveLayerConfig, EmptyRuntim lines = [ "set -eu", f"base={shlex.quote(base_path)}", - "dify-agent drive pull " - + " ".join(shlex.quote(target) for target in pull_targets) - + ' --to "$base"', + "dify-agent drive pull " + " ".join(shlex.quote(target) for target in pull_targets) + ' --to "$base"', ] for skill_key in self.config.mentioned_skill_keys: skill_path = self._shell_local_path(skill_key)