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)