fix: run local python in session workspace (#8792)

tests/unit/test_python_tools.py

@@ -1,5 +1,10 @@
 import platform
+from types import SimpleNamespace
+from unittest.mock import AsyncMock
 
+import pytest
+
+from astrbot.core.agent.run_context import ContextWrapper
 from astrbot.core.tools.computer_tools.python import LocalPythonTool, PythonTool
 
 
@@ -18,3 +23,48 @@ def test_local_python_tool_description_contains_os():
     assert current_os in tool.description
     assert "Python environment" in tool.description
     assert "system-compatible" in tool.description
+
+
+@pytest.mark.asyncio
+async def test_local_python_tool_uses_session_workspace(tmp_path, monkeypatch):
+    """Local Python execution should use the same workspace as local shell."""
+    tool = LocalPythonTool()
+    python_exec = AsyncMock(
+        return_value={"data": {"output": {"text": "ok", "images": []}, "error": ""}}
+    )
+    monkeypatch.setattr(
+        "astrbot.core.tools.computer_tools.python.get_local_booter",
+        lambda: SimpleNamespace(python=SimpleNamespace(exec=python_exec)),
+    )
+    monkeypatch.setattr(
+        "astrbot.core.tools.computer_tools.python.workspace_root",
+        lambda umo: tmp_path / umo.replace(":", "_"),
+    )
+
+    event = SimpleNamespace(
+        unified_msg_origin="onebot:GroupMessage:12345",
+        role="admin",
+        get_platform_name=lambda: "onebot",
+    )
+    context = ContextWrapper(
+        context=SimpleNamespace(
+            event=event,
+            context=SimpleNamespace(
+                get_config=lambda **_kwargs: {
+                    "provider_settings": {"computer_use_require_admin": True}
+                }
+            ),
+        ),
+        tool_call_timeout=60,
+    )
+
+    await tool.call(context, code="print('ok')", timeout=30)
+
+    workspace = tmp_path / "onebot_GroupMessage_12345"
+    assert workspace.is_dir()
+    python_exec.assert_awaited_once_with(
+        "print('ok')",
+        timeout=30,
+        silent=False,
+        cwd=str(workspace.resolve(strict=False)),
+    )