refactor(沙箱工具): 将沙箱镜像配置改为从settings读取
默认镜像配置改为从settings.SANDBOX_IMAGE读取,提高配置灵活性
This commit is contained in:
parent
99c0613cfa
commit
dbf03c5ac3
|
|
@ -14,6 +14,7 @@ from pydantic import BaseModel, Field
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
|
|
||||||
from .base import AgentTool, ToolResult
|
from .base import AgentTool, ToolResult
|
||||||
|
from app.core.config import settings
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
@ -21,7 +22,7 @@ logger = logging.getLogger(__name__)
|
||||||
@dataclass
|
@dataclass
|
||||||
class SandboxConfig:
|
class SandboxConfig:
|
||||||
"""沙箱配置"""
|
"""沙箱配置"""
|
||||||
image: str = "deepaudit/sandbox:latest"
|
image: str = None # 默认从 settings.SANDBOX_IMAGE 读取
|
||||||
memory_limit: str = "512m"
|
memory_limit: str = "512m"
|
||||||
cpu_limit: float = 1.0
|
cpu_limit: float = 1.0
|
||||||
timeout: int = 60
|
timeout: int = 60
|
||||||
|
|
@ -29,6 +30,10 @@ class SandboxConfig:
|
||||||
read_only: bool = True
|
read_only: bool = True
|
||||||
user: str = "1000:1000"
|
user: str = "1000:1000"
|
||||||
|
|
||||||
|
def __post_init__(self):
|
||||||
|
if self.image is None:
|
||||||
|
self.image = settings.SANDBOX_IMAGE
|
||||||
|
|
||||||
|
|
||||||
class SandboxManager:
|
class SandboxManager:
|
||||||
"""
|
"""
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue