diff --git a/server/utils/AiProviders/gemini/index.js b/server/utils/AiProviders/gemini/index.js
index b78c8db6ef27e805b2eb6321a27a85eca17aa330..f29d73e356efada1239d7327aec8b506aadebe6a 100644
--- a/server/utils/AiProviders/gemini/index.js
+++ b/server/utils/AiProviders/gemini/index.js
@@ -17,12 +17,14 @@ class GeminiLLM {
     this.gemini = genAI.getGenerativeModel(
       { model: this.model },
       {
-        // Gemini-1.5-pro and Gemini-1.5-flash are only available on the v1beta API.
-        apiVersion:
-          this.model === "gemini-1.5-pro-latest" ||
-          this.model === "gemini-1.5-flash-latest"
-            ? "v1beta"
-            : "v1",
+        // Gemini-1.5-pro-* and Gemini-1.5-flash are only available on the v1beta API.
+        apiVersion: [
+          "gemini-1.5-pro-latest",
+          "gemini-1.5-flash-latest",
+          "gemini-1.5-pro-exp-0801",
+        ].includes(this.model)
+          ? "v1beta"
+          : "v1",
       }
     );
     this.limits = {