erik/mobibot
erik/mobibot
1
1
Fork 0
Code Issues Releases Activity

Added support for VertexAI (Gemini) version 0.7.0

Browse source
This commit is contained in:
Erik C. Thauvin 2024-03-24 17:06:53 -07:00
parent ee5b5ad11a
commit b2aae15c79
Signed by: erik
GPG key ID: 776702A6A2DA330E
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/bld/java/net/thauvin/erik/MobibotBuild.java
View file

@ -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"))

4
src/main/kotlin/net/thauvin/erik/mobibot/modules/Gemini.kt
View file

@ -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)