Skip to content
Snippets Groups Projects
Unverified Commit f06ef618 authored by Timothy Carambat's avatar Timothy Carambat Committed by GitHub
Browse files

add exp model to v1Beta (#2082)

parent 01e85b0f
No related branches found
No related tags found
No related merge requests found
......@@ -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 = {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment