From 870e3716d949a2140d7eefb36f1dc105310fc91e Mon Sep 17 00:00:00 2001
From: Shivam Mishra <scm.mymail@gmail.com>
Date: Fri, 30 Jun 2023 14:00:39 +0530
Subject: [PATCH] refactor: rename suggest_label to label_suggestion

---
 app/javascript/dashboard/api/integrations/openapi.js   |  2 +-
 .../components/widgets/AIAssistanceButton.vue          | 10 ++++++++++
 lib/integrations/openai/processor_service.rb           |  8 ++++----
 lib/integrations/openai_processor_service.rb           |  4 ++--
 4 files changed, 17 insertions(+), 7 deletions(-)

diff --git a/app/javascript/dashboard/api/integrations/openapi.js b/app/javascript/dashboard/api/integrations/openapi.js
index 4926f075e..ad203a14c 100644
--- a/app/javascript/dashboard/api/integrations/openapi.js
+++ b/app/javascript/dashboard/api/integrations/openapi.js
@@ -28,7 +28,7 @@ class OpenAIAPI extends ApiClient {
     this.conversation_events = [
       'summarize',
       'reply_suggestion',
-      'suggest_label',
+      'label_suggestion',
     ];
 
     /**
diff --git a/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue b/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue
index 1da4109da..ce5bc7f04 100644
--- a/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue
+++ b/app/javascript/dashboard/components/widgets/AIAssistanceButton.vue
@@ -1,6 +1,16 @@
 <template>
   <div v-if="isAIIntegrationEnabled" class="position-relative">
     <div v-if="!message">
+      <woot-button
+        v-if="isPrivateNote"
+        v-tooltip.top-end="$t('INTEGRATION_SETTINGS.OPEN_AI.SUMMARY_TITLE')"
+        icon="book-pulse"
+        color-scheme="secondary"
+        variant="smooth"
+        size="small"
+        :is-loading="uiFlags.summarize"
+        @click="processEvent('summarize')"
+      />
       <woot-button
         v-if="isPrivateNote"
         v-tooltip.top-end="$t('INTEGRATION_SETTINGS.OPEN_AI.SUMMARY_TITLE')"
diff --git a/lib/integrations/openai/processor_service.rb b/lib/integrations/openai/processor_service.rb
index c11745d2a..ff9fa4ce0 100644
--- a/lib/integrations/openai/processor_service.rb
+++ b/lib/integrations/openai/processor_service.rb
@@ -11,11 +11,11 @@ class Integrations::Openai::ProcessorService < Integrations::OpenaiProcessorServ
     make_api_call(rephrase_body)
   end
 
-  def suggest_label_message
-    payload = suggest_label_body
+  def label_suggestion_message
+    payload = label_suggestion_body
     return nil if payload.blank?
 
-    make_api_call(suggest_label_body)
+    make_api_call(label_suggestion_body)
   end
 
   private
@@ -119,7 +119,7 @@ class Integrations::Openai::ProcessorService < Integrations::OpenaiProcessorServ
     }.to_json
   end
 
-  def suggest_label_body
+  def label_suggestion_body
     content = labels_with_messages
     return nil if content.blank?
 
diff --git a/lib/integrations/openai_processor_service.rb b/lib/integrations/openai_processor_service.rb
index cedc2bbf9..7552a09e5 100644
--- a/lib/integrations/openai_processor_service.rb
+++ b/lib/integrations/openai_processor_service.rb
@@ -6,8 +6,8 @@ class Integrations::OpenaiProcessorService
   API_URL = 'https://api.openai.com/v1/chat/completions'.freeze
   GPT_MODEL = 'gpt-3.5-turbo'.freeze
 
-  ALLOWED_EVENT_NAMES = %w[rephrase summarize reply_suggestion suggest_label].freeze
-  CACHEABLE_EVENTS = %w[suggest_label].freeze
+  ALLOWED_EVENT_NAMES = %w[rephrase summarize reply_suggestion label_suggestion].freeze
+  CACHEABLE_EVENTS = %w[label_suggestion].freeze
 
   pattr_initialize [:hook!, :event!]