feat: Refactor imports to use pod_executor and add container verification script
This commit is contained in:
parent
17577d3fba
commit
114381b71c
19 changed files with 79 additions and 79 deletions
|
|
@ -6,11 +6,11 @@ from pathlib import Path
|
|||
|
||||
from mcp_forge.execution.jupyter.backend import JupyterBackend
|
||||
from mcp_forge.execution.jupyter.sessions import SessionManager, Session, SessionState
|
||||
from mcp_forge.execution.simple.executor import ExecutionResult
|
||||
from pod_executor.simple.executor import ExecutionResult
|
||||
from mcp_forge.config.schema import ForgeConfig, ExecutionConfig, ImageConfig, SessionConfig
|
||||
from mcp_forge.podman.containers import SecureContainerManager
|
||||
from mcp_forge.security.audit import AuditLogger
|
||||
from mcp_forge.security.resource_limits import ResourceLimits
|
||||
from pod_executor.security.resource_limits import ResourceLimits
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
|
|||
|
|
@ -11,8 +11,8 @@ from mcp_forge.execution.jupyter.kernel import (
|
|||
KernelError
|
||||
)
|
||||
from mcp_forge.podman.containers import SecureContainerManager
|
||||
from mcp_forge.security.resource_limits import ResourceLimits
|
||||
from mcp_forge.execution.simple.executor import ExecutionResult
|
||||
from pod_executor.security.resource_limits import ResourceLimits
|
||||
from pod_executor.simple.executor import ExecutionResult
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
|
|||
|
|
@ -14,8 +14,8 @@ from mcp_forge.execution.jupyter.sessions import (
|
|||
from mcp_forge.execution.jupyter.kernel import JupyterKernelManager
|
||||
from mcp_forge.config.schema import SessionConfig
|
||||
from mcp_forge.security.audit import AuditLogger
|
||||
from mcp_forge.security.resource_limits import ResourceLimits
|
||||
from mcp_forge.execution.simple.executor import ExecutionResult
|
||||
from pod_executor.security.resource_limits import ResourceLimits
|
||||
from pod_executor.simple.executor import ExecutionResult
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
|
|||
|
|
@ -4,12 +4,12 @@ import pytest
|
|||
from unittest.mock import Mock, MagicMock, patch
|
||||
from pathlib import Path
|
||||
|
||||
from mcp_forge.execution.simple.backend import SimpleBackend
|
||||
from mcp_forge.execution.simple.executor import ExecutionResult
|
||||
from mcp_forge.adapters import SimpleBackend
|
||||
from pod_executor.simple.executor import ExecutionResult
|
||||
from mcp_forge.config.schema import ForgeConfig, ExecutionConfig, ImageConfig
|
||||
from mcp_forge.podman.containers import SecureContainerManager
|
||||
from mcp_forge.security.audit import AuditLogger
|
||||
from mcp_forge.security.resource_limits import ResourceLimits
|
||||
from pod_executor.security.resource_limits import ResourceLimits
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
|
|||
|
|
@ -4,9 +4,9 @@ import pytest
|
|||
from unittest.mock import Mock
|
||||
import json
|
||||
|
||||
from mcp_forge.execution.simple.executor import CodeExecutor, ExecutionResult
|
||||
from pod_executor.simple.executor import CodeExecutor, ExecutionResult
|
||||
from mcp_forge.podman.containers import SecureContainerManager
|
||||
from mcp_forge.security.resource_limits import ResourceLimits
|
||||
from pod_executor.security.resource_limits import ResourceLimits
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue