From 231220d2dae2b518489d32ec888f0f74af43492f Mon Sep 17 00:00:00 2001 From: Hermann Kao Date: Tue, 25 Jun 2024 17:53:27 +0200 Subject: [PATCH] finished alpha version, --- .../ollamacommitsummarizer/actions/GenerateCommitAction.kt | 7 ++++++- src/main/resources/META-INF/plugin.xml | 2 +- src/main/resources/META-INF/pluginIcon.svg | 6 ++++++ 3 files changed, 13 insertions(+), 2 deletions(-) create mode 100644 src/main/resources/META-INF/pluginIcon.svg diff --git a/src/main/kotlin/me/itishermann/ollamacommitsummarizer/actions/GenerateCommitAction.kt b/src/main/kotlin/me/itishermann/ollamacommitsummarizer/actions/GenerateCommitAction.kt index f4c9974..2f81920 100644 --- a/src/main/kotlin/me/itishermann/ollamacommitsummarizer/actions/GenerateCommitAction.kt +++ b/src/main/kotlin/me/itishermann/ollamacommitsummarizer/actions/GenerateCommitAction.kt @@ -38,7 +38,7 @@ class GenerateCommitAction: AnAction(), DumbAware { if (commitPanel == null || processing) { return } - ProgressManager.getInstance().run(object : Task.Backgroundable(event.project, "Generating commit", false) { + ProgressManager.getInstance().run(object : Task.Backgroundable(event.project, "Ollama commit summarizer", false) { override fun run(@NotNull indicator: ProgressIndicator) { indicator.text = "Generating commit message" processing = true @@ -48,6 +48,7 @@ class GenerateCommitAction: AnAction(), DumbAware { } private fun handleEvent(e: AnActionEvent, commitPanel: CommitMessageI, indicator: ProgressIndicator) { + indicator.text = "Building prompt" val project = checkNotNull(e.project) // get included changes val abstractCommitWorkflowHandler = @@ -70,16 +71,20 @@ class GenerateCommitAction: AnAction(), DumbAware { } private fun generateCommitMessage(prompt: String, commitPanel: CommitMessageI, indicator: ProgressIndicator) { + indicator.text = "Preparing inference" val client = service().getOllamaClient() val modelName = OllamaSettingsState.instance.state.modelName val streamHandler = OllamaStreamHandler { s: String? -> ApplicationManager.getApplication().invokeLater { + indicator.text = "Streaming response" indicator.fraction += 0.05 commitPanel.setCommitMessage(s) } } + indicator.text = "Setting temperature and top-k" val options = OptionsBuilder().setTemperature(1.5f).setTopP(0.9f).setTopK(40).build() try { + indicator.text = "Inferring" client.generate(modelName, prompt, options, streamHandler) processing = false Notifications.Bus.notify( diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml index 6f47d68..6f71507 100644 --- a/src/main/resources/META-INF/plugin.xml +++ b/src/main/resources/META-INF/plugin.xml @@ -26,7 +26,7 @@ + icon="com.intellij.icons.AllIcons.Actions.EditScheme"> diff --git a/src/main/resources/META-INF/pluginIcon.svg b/src/main/resources/META-INF/pluginIcon.svg new file mode 100644 index 0000000..471ebd5 --- /dev/null +++ b/src/main/resources/META-INF/pluginIcon.svg @@ -0,0 +1,6 @@ + + + + Ollama Commit Summarizer Logo + +