From b2aae15c79ab02fc8a433f51e5d2a58e3f9cb1a8 Mon Sep 17 00:00:00 2001 From: "Erik C. Thauvin" Date: Sun, 24 Mar 2024 17:06:53 -0700 Subject: [PATCH] Added support for VertexAI (Gemini) version 0.7.0 --- src/bld/java/net/thauvin/erik/MobibotBuild.java | 2 +- src/main/kotlin/net/thauvin/erik/mobibot/modules/Gemini.kt | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/bld/java/net/thauvin/erik/MobibotBuild.java b/src/bld/java/net/thauvin/erik/MobibotBuild.java index 115512c..a3369d2 100644 --- a/src/bld/java/net/thauvin/erik/MobibotBuild.java +++ b/src/bld/java/net/thauvin/erik/MobibotBuild.java @@ -87,7 +87,7 @@ public class MobibotBuild extends Project { // Google .include(dependency("com.google.code.gson", "gson", "2.10.1")) .include(dependency("com.google.guava", "guava", "33.1.0-jre")) - .include(dependency("com.google.cloud", "google-cloud-vertexai", version(0, 6, 0))) + .include(dependency("com.google.cloud", "google-cloud-vertexai", version(0, 7, 0))) // Kotlin .include(dependency("org.jetbrains.kotlinx", "kotlinx-coroutines-core", "1.8.0")) .include(dependency("org.jetbrains.kotlinx", "kotlinx-cli-jvm", "0.3.6")) diff --git a/src/main/kotlin/net/thauvin/erik/mobibot/modules/Gemini.kt b/src/main/kotlin/net/thauvin/erik/mobibot/modules/Gemini.kt index b1a2c83..80f3d13 100644 --- a/src/main/kotlin/net/thauvin/erik/mobibot/modules/Gemini.kt +++ b/src/main/kotlin/net/thauvin/erik/mobibot/modules/Gemini.kt @@ -109,7 +109,9 @@ class Gemini : AbstractModule() { try { VertexAI(projectId, location).use { vertexAI -> val generationConfig = GenerationConfig.newBuilder().setMaxOutputTokens(maxToken).build() - val model = GenerativeModel("gemini-pro-vision", generationConfig, vertexAI) + val model = GenerativeModel.Builder().setModelName("gemini-pro-vision") + .setGenerationConfig(generationConfig) + .setVertexAi(vertexAI).build() val session = ChatSession(model) val response = session.sendMessage(query)