Added support for VertexAI (Gemini) version 0.7.0

ethauvin · Mar 25, 2024 · b2aae15 · b2aae15
1 parent ee5b5ad
commit b2aae15
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/bld/java/net/thauvin/erik/MobibotBuild.java b/src/bld/java/net/thauvin/erik/MobibotBuild.java
@@ -87,7 +87,7 @@ public MobibotBuild() {
                 // 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
@@ -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)