fix(google-genai): Update multimodal model check to include gemini-2 …

…support (#7389) Co-authored-by: jacoblee93 <[email protected]>
langchain-ai · Dec 19, 2024 · 9d6b434 · 9d6b434
1 parent 82cecae
commit 9d6b434
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/libs/langchain-google-genai/src/chat_models.ts b/libs/langchain-google-genai/src/chat_models.ts
@@ -580,7 +580,11 @@ export class ChatGoogleGenerativeAI
   private client: GenerativeModel;
 
   get _isMultimodalModel() {
-    return this.model.includes("vision") || this.model.startsWith("gemini-1.5");
+    return (
+      this.model.includes("vision") ||
+      this.model.startsWith("gemini-1.5") ||
+      this.model.startsWith("gemini-2")
+    );
   }
 
   constructor(fields?: GoogleGenerativeAIChatInput) {