github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Implement docker based jupyter executor (#1794)

Browse Source 
* Implement docker based jupyter client

* formatting

* skip docker tests when asked

* feedback

* add log

* update build

* formatting

* structural changes

* update setup.py

* update tests

---------

Co-authored-by: Chi Wang <wang.chi@microsoft.com>
This commit is contained in:
Jack Gerrits
2024-02-29 09:54:11 -05:00
committed by GitHub
parent b8ceb866e6
commit 76ef0789c0
10 changed files with 268 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
test/coding/test_embedded_ipython_code_executor.py
View File

@@ -9,11 +9,25 @@ from autogen.agentchat.conversable_agent import ConversableAgent
from autogen.coding.base import CodeBlock, CodeExecutor
from autogen.coding.factory import CodeExecutorFactory
from autogen.oai.openai_utils import config_list_from_json
from conftest import MOCK_OPEN_AI_API_KEY, skip_openai # noqa: E402
from conftest import MOCK_OPEN_AI_API_KEY, skip_openai, skip_docker # noqa: E402
try:
from autogen.coding.embedded_ipython_code_executor import EmbeddedIPythonCodeExecutor
from autogen.coding.jupyter_code_executor import LocalJupyterCodeExecutor
from autogen.coding.jupyter import (
DockerJupyterServer,
EmbeddedIPythonCodeExecutor,
JupyterCodeExecutor,
LocalJupyterServer,
)
class DockerJupyterExecutor(JupyterCodeExecutor):
def __init__(self, **kwargs):
jupyter_server = DockerJupyterServer()
super().__init__(jupyter_server=jupyter_server, **kwargs)
class LocalJupyterCodeExecutor(JupyterCodeExecutor):
def __init__(self, **kwargs):
jupyter_server = LocalJupyterServer()
super().__init__(jupyter_server=jupyter_server, **kwargs)
# Skip on windows due to kernelgateway bug https://github.com/jupyter-server/kernel_gateway/issues/398
if sys.platform == "win32":
@@ -21,21 +35,27 @@ try:
else:
classes_to_test = [EmbeddedIPythonCodeExecutor, LocalJupyterCodeExecutor]
if not skip_docker:
classes_to_test.append(DockerJupyterExecutor)
skip = False
skip_reason = ""
except ImportError:
except ImportError as e:
skip = True
skip_reason = "Dependencies for EmbeddedIPythonCodeExecutor or LocalJupyterCodeExecutor not installed."
skip_reason = "Dependencies for EmbeddedIPythonCodeExecutor or LocalJupyterCodeExecutor not installed. " + e.msg
classes_to_test = []
@pytest.mark.skipif(skip, reason=skip_reason)
def test_create_dict() -> None:
config: Dict[str, Union[str, CodeExecutor]] = {"executor": "ipython-embedded"}
executor = CodeExecutorFactory.create(config)
assert isinstance(executor, EmbeddedIPythonCodeExecutor)
@pytest.mark.skipif(skip, reason=skip_reason)
@pytest.mark.parametrize("cls", classes_to_test)
def test_create(cls) -> None:
config: Dict[str, Union[str, CodeExecutor]] = {"executor": "ipython-embedded"}
executor = CodeExecutorFactory.create(config)
assert isinstance(executor, EmbeddedIPythonCodeExecutor)
config = {"executor": cls()}
executor = CodeExecutorFactory.create(config)
assert executor is config["executor"]