diff --git a/outlines/models/__init__.py b/outlines/models/__init__.py
index d28fcb2d7..fe6f861ac 100644
--- a/outlines/models/__init__.py
+++ b/outlines/models/__init__.py
@@ -16,4 +16,4 @@
 from .transformers_vision import TransformersVision, transformers_vision
 from .vllm import VLLM, vllm
 
-LogitsGenerator = Union[Transformers, LlamaCpp, ExLlamaV2Model, MLXLM, VLLM]
+LogitsGenerator = Union[Transformers, LlamaCpp, OpenAI, ExLlamaV2Model, MLXLM, VLLM]