check if docker is installed if user_docker is True (#1145)

microsoft · Jan 5, 2024 · 55b03bf · 55b03bf
1 parent 871e9e2
commit 55b03bf
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/autogen/code_utils.py b/autogen/code_utils.py
@@ -272,6 +272,11 @@ def execute_code(
         logger.error(error_msg)
         raise AssertionError(error_msg)
 
+    if use_docker and docker is None:
+        error_msg = "Cannot use docker because the python docker package is not available."
+        logger.error(error_msg)
+        raise AssertionError(error_msg)
+
     # Warn if use_docker was unspecified (or None), and cannot be provided (the default).
     # In this case the current behavior is to fall back to run natively, but this behavior
     # is subject to change.