From 7380f0e7ce23d5ffb6387cec2443033e0784989f Mon Sep 17 00:00:00 2001
From: Sojan Jose <sojan@pepalo.com>
Date: Fri, 17 Nov 2023 19:54:15 -0800
Subject: [PATCH] chore: Making OpenAI label suggestions optional (#8374)

Co-authored-by: Pranav Raj S <pranav@chatwoot.com>
---
 .../dashboard/assets/scss/_formulate.scss     | 20 ++++++++++++++
 .../widgets/conversation/MessagesView.vue     |  2 +-
 app/javascript/dashboard/mixins/aiMixin.js    | 20 +++++++++-----
 .../settings/integrationapps/NewHook.vue      |  2 +-
 config/integration/apps.yml                   |  9 ++++++-
 .../integrations/openai_processor_service.rb  | 27 ++++++++++++++-----
 6 files changed, 65 insertions(+), 15 deletions(-)

diff --git a/app/javascript/dashboard/assets/scss/_formulate.scss b/app/javascript/dashboard/assets/scss/_formulate.scss
index 57c43a3d6..d683ce3c7 100644
--- a/app/javascript/dashboard/assets/scss/_formulate.scss
+++ b/app/javascript/dashboard/assets/scss/_formulate.scss
@@ -16,3 +16,23 @@
     width: 100%;
   }
 }
+
+.integration-hooks {
+  .formulate-input[data-type='checkbox'] {
+    .formulate-input-wrapper {
+      @apply flex;
+
+      .formulate-input-element {
+        @apply pr-2;
+
+        input {
+          @apply mb-0;
+        }
+      }
+    }
+
+    .formulate-input-element-decorator {
+      @apply hidden;
+    }
+  }
+}
diff --git a/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue b/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue
index c6d492061..10b1dce32 100644
--- a/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue
+++ b/app/javascript/dashboard/components/widgets/conversation/MessagesView.vue
@@ -340,7 +340,7 @@ export default {
       // method available in mixin, need to ensure that integrations are present
       await this.fetchIntegrationsIfRequired();
 
-      if (!this.isAIIntegrationEnabled) {
+      if (!this.isLabelSuggestionFeatureEnabled) {
         return;
       }
 
diff --git a/app/javascript/dashboard/mixins/aiMixin.js b/app/javascript/dashboard/mixins/aiMixin.js
index 3a9fc4a75..f3558a6c7 100644
--- a/app/javascript/dashboard/mixins/aiMixin.js
+++ b/app/javascript/dashboard/mixins/aiMixin.js
@@ -15,18 +15,26 @@ export default {
       currentChat: 'getSelectedChat',
       replyMode: 'draftMessages/getReplyEditorMode',
     }),
-    isAIIntegrationEnabled() {
-      return !!this.appIntegrations.find(
+    aiIntegration() {
+      return this.appIntegrations.find(
         integration => integration.id === 'openai' && !!integration.hooks.length
-      );
+      ).hooks[0];
+    },
+    isAIIntegrationEnabled() {
+      return !!this.aiIntegration;
+    },
+    isLabelSuggestionFeatureEnabled() {
+      if (this.aiIntegration) {
+        const { settings = {} } = this.aiIntegration || {};
+        return settings.label_suggestion;
+      }
+      return false;
     },
     isFetchingAppIntegrations() {
       return this.uiFlags.isFetching;
     },
     hookId() {
-      return this.appIntegrations.find(
-        integration => integration.id === 'openai' && !!integration.hooks.length
-      ).hooks[0].id;
+      return this.aiIntegration.id;
     },
     draftMessage() {
       return this.$store.getters['draftMessages/get'](this.draftKey);
diff --git a/app/javascript/dashboard/routes/dashboard/settings/integrationapps/NewHook.vue b/app/javascript/dashboard/routes/dashboard/settings/integrationapps/NewHook.vue
index 93eed8795..454f03dac 100644
--- a/app/javascript/dashboard/routes/dashboard/settings/integrationapps/NewHook.vue
+++ b/app/javascript/dashboard/routes/dashboard/settings/integrationapps/NewHook.vue
@@ -1,6 +1,6 @@
 <!-- eslint-disable vue/v-slot-style -->
 <template>
-  <div class="flex flex-col h-auto overflow-auto">
+  <div class="flex flex-col h-auto overflow-auto integration-hooks">
     <woot-modal-header
       :header-title="integration.name"
       :header-content="integration.description"
diff --git a/config/integration/apps.yml b/config/integration/apps.yml
index f55fd9f56..408c56e1c 100644
--- a/config/integration/apps.yml
+++ b/config/integration/apps.yml
@@ -139,6 +139,7 @@ openai:
     "type": "object",
     "properties": {
         "api_key": { "type": "string" },
+        "label_suggestion": { "type": "boolean" },
       },
     "required": ["api_key"],
     "additionalProperties": false,
@@ -150,5 +151,11 @@ openai:
       "name": "api_key",
       "validation": "required",
     },
+    {
+      "label": "Show label suggestions",
+      "type": "checkbox",
+      "name": "label_suggestion",
+      "validation": "",
+    },
   ]
-  visible_properties: ['api_key']
+  visible_properties: ['api_key', 'label_suggestion']
diff --git a/enterprise/lib/enterprise/integrations/openai_processor_service.rb b/enterprise/lib/enterprise/integrations/openai_processor_service.rb
index 8cc481f43..bf2f4ccf3 100644
--- a/enterprise/lib/enterprise/integrations/openai_processor_service.rb
+++ b/enterprise/lib/enterprise/integrations/openai_processor_service.rb
@@ -21,11 +21,7 @@ module Enterprise::Integrations::OpenaiProcessorService
   def labels_with_messages
     conversation = find_conversation
 
-    # return nil if conversation is not present
-    return nil if conversation.nil?
-
-    # return nil if conversation has less than 3 incoming messages
-    return nil if conversation.messages.incoming.count < 3
+    return nil unless valid_conversation?(conversation)
 
     labels = hook.account.labels.pluck(:title).join(', ')
     character_count = labels.length
@@ -38,6 +34,19 @@ module Enterprise::Integrations::OpenaiProcessorService
     "Messages:\n#{messages}\nLabels:\n#{labels}"
   end
 
+  def valid_conversation?(conversation)
+    return false if conversation.nil?
+    return false if conversation.messages.incoming.count < 3
+
+    # Think Mark think, at this point the conversation is beyond saving
+    return false if conversation.messages.count > 100
+
+    # if there are more than 20 messages, only trigger this if the last message is from the client
+    return false if conversation.messages.count > 20 && !conversation.messages.last.incoming?
+
+    true
+  end
+
   def summarize_body
     {
       model: self.class::GPT_MODEL,
@@ -50,8 +59,10 @@ module Enterprise::Integrations::OpenaiProcessorService
   end
 
   def label_suggestion_body
+    return unless label_suggestions_enabled?
+
     content = labels_with_messages
-    return nil if content.blank?
+    return value_from_cache if content.blank?
 
     {
       model: self.class::GPT_MODEL,
@@ -64,4 +75,8 @@ module Enterprise::Integrations::OpenaiProcessorService
       ]
     }.to_json
   end
+
+  def label_suggestions_enabled?
+    hook.settings['label_suggestion'].present?
+  end
 end