mirror of
https://github.com/langgenius/dify.git
synced 2026-05-13 08:57:28 +08:00
a0c388f283
- Add helpers.py with connection management utilities:
- with_connection: context manager for connection lifecycle
- submit_command: execute command and return CommandFuture
- execute: run command with auto connection, raise on failure
- try_execute: run command with auto connection, return result
- Add CommandExecutionError to exec.py for typed error handling
with access to exit_code, stderr, and full result
- Remove run_command method from VirtualEnvironment base class
(now available as submit_command helper)
- Update all call sites to use new helper functions:
- sandbox/session.py
- sandbox/storage/archive_storage.py
- sandbox/bash/bash_tool.py
- workflow/nodes/command/node.py
- Add comprehensive unit tests for helpers with connection reuse
31 lines
752 B
Python
31 lines
752 B
Python
from core.sandbox.bash.bash_tool import SandboxBashTool
|
|
from core.sandbox.bash.dify_cli import (
|
|
DifyCliBinary,
|
|
DifyCliConfig,
|
|
DifyCliEnvConfig,
|
|
DifyCliLocator,
|
|
DifyCliToolConfig,
|
|
)
|
|
from core.sandbox.constants import (
|
|
DIFY_CLI_CONFIG_PATH,
|
|
DIFY_CLI_PATH,
|
|
DIFY_CLI_PATH_PATTERN,
|
|
)
|
|
from core.sandbox.initializer.initializer import DifyCliInitializer, SandboxInitializer
|
|
from core.sandbox.session import SandboxSession
|
|
|
|
__all__ = [
|
|
"DIFY_CLI_CONFIG_PATH",
|
|
"DIFY_CLI_PATH",
|
|
"DIFY_CLI_PATH_PATTERN",
|
|
"DifyCliBinary",
|
|
"DifyCliConfig",
|
|
"DifyCliEnvConfig",
|
|
"DifyCliInitializer",
|
|
"DifyCliLocator",
|
|
"DifyCliToolConfig",
|
|
"SandboxBashTool",
|
|
"SandboxInitializer",
|
|
"SandboxSession",
|
|
]
|