refactor(sandbox): update imports and remove unused bash tool files, adjust DIFY_CLI_CONFIG_PATH

api/core/sandbox/bash_tool.py

@@ -0,0 +1,95 @@
+from collections.abc import Generator
+from typing import Any
+
+from core.tools.__base.tool import Tool
+from core.tools.__base.tool_runtime import ToolRuntime
+from core.tools.entities.common_entities import I18nObject
+from core.tools.entities.tool_entities import (
+    ToolDescription,
+    ToolEntity,
+    ToolIdentity,
+    ToolInvokeMessage,
+    ToolParameter,
+    ToolProviderType,
+)
+from core.virtual_environment.__base.virtual_environment import VirtualEnvironment
+
+SANDBOX_BASH_TOOL_NAME = "bash"
+SANDBOX_BASH_TOOL_PROVIDER = "sandbox"
+COMMAND_TIMEOUT_SECONDS = 60
+
+
+class SandboxBashTool(Tool):
+    def __init__(self, sandbox: VirtualEnvironment, tenant_id: str):
+        self._sandbox = sandbox
+
+        entity = ToolEntity(
+            identity=ToolIdentity(
+                author="Dify",
+                name=SANDBOX_BASH_TOOL_NAME,
+                label=I18nObject(en_US="Bash", zh_Hans="Bash"),
+                provider=SANDBOX_BASH_TOOL_PROVIDER,
+            ),
+            parameters=[
+                ToolParameter.get_simple_instance(
+                    name="command",
+                    llm_description="The bash command to execute in the sandbox environment",
+                    typ=ToolParameter.ToolParameterType.STRING,
+                    required=True,
+                ),
+            ],
+            description=ToolDescription(
+                human=I18nObject(
+                    en_US="Execute bash commands in the sandbox environment",
+                ),
+                llm="Execute bash commands in the sandbox environment. "
+                "Use this tool to run shell commands, scripts, or interact with the system. "
+                "The command will be executed in an isolated sandbox environment.",
+            ),
+        )
+
+        runtime = ToolRuntime(tenant_id=tenant_id)
+        super().__init__(entity=entity, runtime=runtime)
+
+    def tool_provider_type(self) -> ToolProviderType:
+        return ToolProviderType.BUILT_IN
+
+    def _invoke(
+        self,
+        user_id: str,
+        tool_parameters: dict[str, Any],
+        conversation_id: str | None = None,
+        app_id: str | None = None,
+        message_id: str | None = None,
+    ) -> Generator[ToolInvokeMessage, None, None]:
+        command = tool_parameters.get("command", "")
+        if not command:
+            yield self.create_text_message("Error: No command provided")
+            return
+
+        connection_handle = self._sandbox.establish_connection()
+        try:
+            cmd_list = ["sh", "-c", command]
+            future = self._sandbox.run_command(connection_handle, cmd_list)
+            timeout = COMMAND_TIMEOUT_SECONDS if COMMAND_TIMEOUT_SECONDS > 0 else None
+            result = future.result(timeout=timeout)
+
+            stdout = result.stdout.decode("utf-8", errors="replace") if result.stdout else ""
+            stderr = result.stderr.decode("utf-8", errors="replace") if result.stderr else ""
+            exit_code = result.exit_code
+
+            output_parts: list[str] = []
+            if stdout:
+                output_parts.append(f"stdout:\n{stdout}")
+            if stderr:
+                output_parts.append(f"stderr:\n{stderr}")
+            output_parts.append(f"exit_code: {exit_code}")
+
+            yield self.create_text_message("\n".join(output_parts))
+
+        except TimeoutError:
+            yield self.create_text_message(f"Error: Command timed out after {COMMAND_TIMEOUT_SECONDS}s")
+        except Exception as e:
+            yield self.create_text_message(f"Error: {e!s}")
+        finally:
+            self._sandbox.release_connection(connection_handle)