diff --git a/setup.py b/setup.py
index 2056a0a371..2782259d3a 100644
--- a/setup.py
+++ b/setup.py
@@ -215,7 +215,7 @@
 )
 
 ag2_extra_require = [
-    "ag2[gemini]",
+    "autogen[gemini] >= 0.14.1, < 1",
     "openinference-instrumentation-autogen >= 0.1.6, < 0.2",
 ]