From f06ef6180d4e604a76a87c26f44c0650c687721f Mon Sep 17 00:00:00 2001
From: Timothy Carambat <rambat1010@gmail.com>
Date: Fri, 9 Aug 2024 14:19:49 -0700
Subject: [PATCH] add exp model to v1Beta (#2082)
---
server/utils/AiProviders/gemini/index.js | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/server/utils/AiProviders/gemini/index.js b/server/utils/AiProviders/gemini/index.js
index b78c8db6e..f29d73e35 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 = {
--
GitLab