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 { ...@@ -17,12 +17,14 @@ class GeminiLLM {
this.gemini = genAI.getGenerativeModel( this.gemini = genAI.getGenerativeModel(
{ model: this.model }, { model: this.model },
{ {
// Gemini-1.5-pro and Gemini-1.5-flash are only available on the v1beta API. // Gemini-1.5-pro-* and Gemini-1.5-flash are only available on the v1beta API.
apiVersion: apiVersion: [
this.model === "gemini-1.5-pro-latest" || "gemini-1.5-pro-latest",
this.model === "gemini-1.5-flash-latest" "gemini-1.5-flash-latest",
? "v1beta" "gemini-1.5-pro-exp-0801",
: "v1", ].includes(this.model)
? "v1beta"
: "v1",
} }
); );
this.limits = { this.limits = {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment