wip

Author: davorrunje

.gitignore

@@ -174,7 +174,7 @@ test/test_files/agenteval-in-out/out/
 
 # local cache or coding foler
 local_cache/
-coding/
+# coding/
 
 # Files created by tests
 *tmp_code_*

autogen/coding/jupyter/helpers.py

@@ -0,0 +1,20 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import subprocess
+
+__all__ = ["is_jupyter_kernel_gateway_installed"]
+
+
+def is_jupyter_kernel_gateway_installed() -> bool:
+    try:
+        subprocess.run(
+            ["jupyter-kernel-gateway", "--version"],
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            check=True,
+        )
+        return True
+    except (subprocess.CalledProcessError, FileNotFoundError):
+        return False

test/coding/__init__.py

@@ -17,9 +17,7 @@
 from autogen.coding.base import CodeBlock, CodeExecutor
 from autogen.coding.factory import CodeExecutorFactory
 
-# add ../.. to sys.path to import conftest
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
-from conftest import MOCK_OPEN_AI_API_KEY, skip_docker  # noqa: E402
+from ..conftest import MOCK_OPEN_AI_API_KEY, skip_docker  # noqa: E402
 
 try:
     from autogen.coding.jupyter import (