Fix lazy openai.types to stay module-backed

Author: fede-kamel

src/openai/__init__.py

@@ -98,17 +98,6 @@
 if _t.TYPE_CHECKING:
     from . import types as types
     from .lib.azure import AzureADTokenProvider, AzureOpenAI, AsyncAzureOpenAI
-else:
-    from ._utils._proxy import LazyProxy as _LazyProxy
-
-    class _TypesProxy(_LazyProxy[_t.Any]):
-        @override
-        def __load__(self) -> _t.Any:
-            import importlib
-
-            return importlib.import_module("openai.types")
-
-    types = _TypesProxy().__as_proxied__()
 
 from .version import VERSION as VERSION
 from .lib._old_api import *
@@ -143,6 +132,12 @@ def _lazy_azure_openai() -> object:
     return AzureOpenAI
 
 
+def _lazy_types_module() -> object:
+    import importlib
+
+    return importlib.import_module("openai.types")
+
+
 def _lazy_async_azure_openai() -> object:
     from .lib.azure import AsyncAzureOpenAI
 
@@ -168,6 +163,7 @@ def _lazy_async_assistant_event_handler() -> object:
 
 
 _LAZY_EXPORTS: dict[str, _t.Callable[[], object]] = {
+    "types": _lazy_types_module,
     "AzureOpenAI": _lazy_azure_openai,
     "AsyncAzureOpenAI": _lazy_async_azure_openai,
     "pydantic_function_tool": _lazy_pydantic_function_tool,

tests/test_lazy_types_import.py

@@ -0,0 +1,39 @@
+from __future__ import annotations
+
+import importlib
+import sys
+from types import ModuleType
+
+
+def _openai_modules() -> dict[str, object]:
+    return {name: mod for name, mod in sys.modules.items() if name == "openai" or name.startswith("openai.")}
+
+
+def test_openai_types_are_lazy_imported() -> None:
+    original_modules = _openai_modules()
+
+    for name in original_modules:
+        sys.modules.pop(name, None)
+
+    try:
+        openai = importlib.import_module("openai")
+
+        assert "openai.types" not in sys.modules
+
+        # Attribute access should trigger the lazy import.
+        assert openai.types.ChatModel is not None
+        assert "openai.types" in sys.modules
+
+        types_module = sys.modules["openai.types"]
+        assert isinstance(openai.types, ModuleType)
+        assert openai.types is types_module
+
+        # Module reload should keep the binding pointed at the module object.
+        reloaded = importlib.reload(openai.types)
+        assert reloaded is openai.types
+        assert reloaded is sys.modules["openai.types"]
+    finally:
+        for name in list(sys.modules):
+            if name == "openai" or name.startswith("openai."):
+                sys.modules.pop(name, None)
+        sys.modules.update(original_modules)