Add check_for_missing_imports

autogen/handle_imports.py

@@ -0,0 +1,18 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from contextlib import contextmanager
+from typing import Generator
+
+
+@contextmanager
+def check_for_missing_imports() -> Generator[None, None, None]:
+    """
+    A context manager to temporarily suppress ImportErrors.
+    Use this to attempt imports without failing immediately on missing modules.
+    """
+    try:
+        yield
+    except ImportError:
+        pass  # Ignore ImportErrors during this context

test/test_handle_imports.py

@@ -0,0 +1,11 @@
+# Copyright (c) 2023 - 2024, Owners of https://github.com/ag2ai
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from autogen.handle_imports import check_for_missing_imports
+
+
+def test_check_for_missing_imports():
+    with check_for_missing_imports():
+        pass  # Safe to attempt, even if it fails
+    assert True