mirror of
https://github.com/langgenius/dify.git
synced 2026-03-23 07:17:55 +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
19 lines
527 B
Python
19 lines
527 B
Python
"""Sandbox debug utilities. TODO: Remove this module when sandbox debugging is complete."""
|
|
|
|
from typing import Any
|
|
|
|
from core.callback_handler.agent_tool_callback_handler import print_text
|
|
|
|
SANDBOX_DEBUG_ENABLED = True
|
|
|
|
|
|
def sandbox_debug(tag: str, message: str, data: Any = None) -> None:
|
|
if not SANDBOX_DEBUG_ENABLED:
|
|
return
|
|
|
|
print_text(f"\n[{tag}]\n", color="blue")
|
|
if data is not None:
|
|
print_text(f"{message}: {data}\n", color="blue")
|
|
else:
|
|
print_text(f"{message}\n", color="blue")
|