feat(clients): support gemini

Author: Blarc

build.gradle.kts

@@ -127,6 +127,7 @@ dependencies {
     implementation("dev.langchain4j:langchain4j-open-ai:0.32.0")
     implementation("dev.langchain4j:langchain4j-ollama:0.32.0")
     implementation("dev.langchain4j:langchain4j-qianfan:0.32.0") // The Baidu Qianfan Large Model Platform, including the ERNIE series, can be accessed at https://docs.langchain4j.dev/integrations/language-models/qianfan/.
+    implementation("dev.langchain4j:langchain4j-vertex-ai-gemini:0.32.0")
 //    implementation("dev.langchain4j:langchain4j-hugging-face:0.28.0")
 //    implementation("dev.langchain4j:langchain4j-milvus:0.28.0")
 //    implementation("dev.langchain4j:langchain4j-local-ai:0.28.0")

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/Icons.kt

@@ -7,4 +7,5 @@ object Icons {
     val OPEN_AI = IconLoader.getIcon("/icons/openai.svg", javaClass)
     val OLLAMA = IconLoader.getIcon("/icons/ollama15.svg", javaClass)
     val QIANFAN = IconLoader.getIcon("/icons/qianfan.png", javaClass)
+    val GEMINI = IconLoader.getIcon("/icons/gemini.png", javaClass)
 }

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/gemini/GeminiClientConfiguration.kt

@@ -0,0 +1,58 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients.gemini
+
+import com.github.blarc.ai.commits.intellij.plugin.Icons
+import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClientConfiguration
+import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClientSharedState
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.vcs.ui.CommitMessage
+import com.intellij.util.xmlb.annotations.Attribute
+import javax.swing.Icon
+
+class GeminiClientConfiguration : LLMClientConfiguration(
+    "Gemini",
+    "gemini-pro",
+    "0.7"
+) {
+    @Attribute
+    var projectId: String = ""
+    @Attribute
+    var location: String = "us-central1"
+
+    companion object {
+        const val CLIENT_NAME = "Gemini"
+    }
+
+    override fun getClientName(): String {
+        return CLIENT_NAME
+    }
+
+    override fun getClientIcon(): Icon {
+        return Icons.GEMINI
+    }
+
+    override fun getSharedState(): LLMClientSharedState {
+        return GeminiClientSharedState.getInstance()
+    }
+
+    override fun generateCommitMessage(prompt: String, project: Project, commitMessage: CommitMessage) {
+        return GeminiClientService.getInstance().generateCommitMessage(this, prompt, project, commitMessage)
+    }
+
+    // Model names are hard-coded and do not need to be refreshed.
+    override fun getRefreshModelsFunction() = null
+
+    override fun clone(): LLMClientConfiguration {
+        val copy = GeminiClientConfiguration()
+        copy.id = id
+        copy.name = name
+        copy.modelId = modelId
+        copy.temperature = temperature
+        copy.projectId = projectId
+        copy.location = location
+        return copy
+    }
+
+    override fun panel() = GeminiClientPanel(this)
+
+
+}

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/gemini/GeminiClientPanel.kt

@@ -0,0 +1,64 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients.gemini
+
+import com.github.blarc.ai.commits.intellij.plugin.AICommitsBundle.message
+import com.github.blarc.ai.commits.intellij.plugin.notBlank
+import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClientPanel
+import com.intellij.ui.components.JBTextField
+import com.intellij.ui.dsl.builder.Panel
+import com.intellij.ui.dsl.builder.bindText
+import com.intellij.ui.dsl.builder.panel
+
+class GeminiClientPanel private constructor(
+    private val clientConfiguration: GeminiClientConfiguration,
+    val service: GeminiClientService
+) : LLMClientPanel(clientConfiguration) {
+
+    private val projectIdTextField = JBTextField()
+    private val locationTextField = JBTextField()
+
+    constructor(configuration: GeminiClientConfiguration): this(configuration, GeminiClientService.getInstance())
+
+    override fun create() = panel {
+        nameRow()
+        projectIdRow()
+        locationRow()
+        modelIdRow()
+        temperatureRow()
+        verifyRow()
+    }
+
+    private fun Panel.projectIdRow() {
+        row {
+            label(message("settings.gemini.project-id"))
+                .widthGroup("label")
+
+            cell(projectIdTextField)
+                .bindText(clientConfiguration::projectId)
+                .resizableColumn()
+                .widthGroup("input")
+                .validationOnInput { notBlank(it.text) }
+                .comment(message("settings.gemini.project-id.comment"))
+        }
+
+    }
+
+    private fun Panel.locationRow() {
+        row {
+            label(message("settings.gemini.location"))
+                .widthGroup("label")
+            cell(locationTextField)
+                .bindText(clientConfiguration::location)
+                .resizableColumn()
+                .validationOnInput { notBlank(it.text) }
+                .widthGroup("input")
+        }
+    }
+
+    override fun verifyConfiguration() {
+        // Configuration passed to panel is already a copy of the original or a new configuration
+        clientConfiguration.modelId = modelComboBox.item
+        clientConfiguration.temperature = temperatureTextField.text
+
+        service.verifyConfiguration(clientConfiguration, verifyLabel)
+    }
+}

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/gemini/GeminiClientService.kt

@@ -0,0 +1,27 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients.gemini
+
+import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClientService
+import com.intellij.openapi.components.Service
+import com.intellij.openapi.components.service
+import dev.langchain4j.model.chat.ChatLanguageModel
+import dev.langchain4j.model.vertexai.VertexAiGeminiChatModel
+import kotlinx.coroutines.CoroutineScope
+
+@Service(Service.Level.APP)
+class GeminiClientService(private val cs: CoroutineScope): LLMClientService<GeminiClientConfiguration>(cs) {
+
+    companion object {
+        @JvmStatic
+        fun getInstance(): GeminiClientService = service()
+    }
+
+    override suspend fun buildChatModel(client: GeminiClientConfiguration): ChatLanguageModel {
+        return VertexAiGeminiChatModel.builder()
+            .project(client.projectId)
+            .location(client.location)
+            .modelName(client.modelId)
+            .temperature(client.temperature.toFloat())
+            .build()
+    }
+
+}

src/main/kotlin/com/github/blarc/ai/commits/intellij/plugin/settings/clients/gemini/GeminiClientSharedState.kt

@@ -0,0 +1,31 @@
+package com.github.blarc.ai.commits.intellij.plugin.settings.clients.gemini
+
+import com.github.blarc.ai.commits.intellij.plugin.settings.clients.LLMClientSharedState
+import com.intellij.openapi.components.*
+import com.intellij.util.xmlb.XmlSerializerUtil
+import com.intellij.util.xmlb.annotations.XCollection
+
+@Service(Service.Level.APP)
+@State(name = "GeminiClientSharedState", storages = [Storage("AICommitsGemini.xml")])
+class GeminiClientSharedState : PersistentStateComponent<GeminiClientSharedState>, LLMClientSharedState {
+
+    companion object {
+        @JvmStatic
+        fun getInstance(): GeminiClientSharedState = service()
+    }
+
+    @XCollection(style = XCollection.Style.v2)
+    override val hosts = mutableSetOf("http://localhost:11434/")
+
+    @XCollection(style = XCollection.Style.v2)
+    override val modelIds: MutableSet<String> = mutableSetOf(
+        "gemini-pro",
+        "gemini-ultra"
+    )
+
+    override fun getState(): GeminiClientSharedState = this
+
+    override fun loadState(state: GeminiClientSharedState) {
+        XmlSerializerUtil.copyBean(state, this)
+    }
+}

src/main/resources/icons/gemini.png

@@ -79,3 +79,6 @@ settings.openAi.organizationId=Organization ID
 settings.qianfan.apiKey=API key
 settings.qianfan.secretKey=Secret key
 notifications.client-not-set=LLM client is not set.
+settings.gemini.project-id=Project ID
+settings.gemini.project-id.comment=Google Cloud project's ID.
+settings.gemini.location=Location