feat: Rewrite APIs for ai editor message improvement (#13059)

Co-authored-by: Vishnu Narayanan <iamwishnu@gmail.com>

lib/integrations/openai/processor_service.rb

@@ -1,5 +1,4 @@
 class Integrations::Openai::ProcessorService < Integrations::LlmBaseService
-  AGENT_INSTRUCTION = 'You are a helpful support agent.'.freeze
   LANGUAGE_INSTRUCTION = 'Ensure that the reply should be in user language.'.freeze
   def reply_suggestion_message
     make_api_call(reply_suggestion_body)
@@ -9,39 +8,39 @@ class Integrations::Openai::ProcessorService < Integrations::LlmBaseService
     make_api_call(summarize_body)
   end
 
-  def rephrase_message
-    make_api_call(build_api_call_body("#{AGENT_INSTRUCTION} Please rephrase the following response. " \
-                                      "#{LANGUAGE_INSTRUCTION}"))
+  def confident_message
+    tone_instruction = determine_tone_instruction('confident')
+    make_api_call(build_api_call_body(tone_rewrite_prompt(tone_instruction)))
   end
 
   def fix_spelling_grammar_message
-    make_api_call(build_api_call_body("#{AGENT_INSTRUCTION} Please fix the spelling and grammar of the following response. " \
+    make_api_call(build_api_call_body('Please fix the spelling and grammar of the following response. ' \
                                       "#{LANGUAGE_INSTRUCTION}"))
   end
 
-  def shorten_message
-    make_api_call(build_api_call_body("#{AGENT_INSTRUCTION} Please shorten the following response. " \
-                                      "#{LANGUAGE_INSTRUCTION}"))
+  def straightforward_message
+    tone_instruction = determine_tone_instruction('straightforward')
+    make_api_call(build_api_call_body(tone_rewrite_prompt(tone_instruction)))
   end
 
-  def expand_message
-    make_api_call(build_api_call_body("#{AGENT_INSTRUCTION} Please expand the following response. " \
-                                      "#{LANGUAGE_INSTRUCTION}"))
+  def casual_message
+    tone_instruction = determine_tone_instruction('casual')
+    make_api_call(build_api_call_body(tone_rewrite_prompt(tone_instruction)))
   end
 
   def make_friendly_message
-    make_api_call(build_api_call_body("#{AGENT_INSTRUCTION} Please make the following response more friendly. " \
-                                      "#{LANGUAGE_INSTRUCTION}"))
+    tone_instruction = determine_tone_instruction('friendly')
+    make_api_call(build_api_call_body(tone_rewrite_prompt(tone_instruction)))
   end
 
   def make_formal_message
-    make_api_call(build_api_call_body("#{AGENT_INSTRUCTION} Please make the following response more formal. " \
-                                      "#{LANGUAGE_INSTRUCTION}"))
+    tone_instruction = determine_tone_instruction('formal')
+    make_api_call(build_api_call_body(tone_rewrite_prompt(tone_instruction)))
   end
 
-  def simplify_message
-    make_api_call(build_api_call_body("#{AGENT_INSTRUCTION} Please simplify the following response. " \
-                                      "#{LANGUAGE_INSTRUCTION}"))
+  def professional_message
+    tone_instruction = determine_tone_instruction('professional')
+    make_api_call(build_api_call_body(tone_rewrite_prompt(tone_instruction)))
   end
 
   private
@@ -51,6 +50,10 @@ class Integrations::Openai::ProcessorService < Integrations::LlmBaseService
     Rails.root.join(path, "#{file_name}.txt").read
   end
 
+  def tone_rewrite_prompt(tone_instruction)
+    format(prompt_from_file('tone_rewrite'), tone_instruction)
+  end
+
   def build_api_call_body(system_content, user_content = event['data']['content'])
     {
       model: GPT_MODEL,
@@ -133,6 +136,26 @@ class Integrations::Openai::ProcessorService < Integrations::LlmBaseService
       ].concat(conversation_messages(in_array_format: true))
     }.to_json
   end
+
+  def determine_tone_instruction(tone)
+    case tone
+    when 'friendly'
+      'Warm, approachable, and personable. Use conversational language, positive words, and show empathy. ' \
+      'May include phrases like \"Happy to help!\" or \"I\'d be glad to...\"'
+    when 'confident'
+      'Assertive and assured. Use definitive language, avoid hedging words like \"maybe\" or \"I think\". ' \
+      'Be direct and authoritative while remaining helpful.'
+    when 'straightforward'
+      'Clear, direct, and to-the-point. Remove unnecessary words, get straight to the information or solution. No fluff or extra pleasantries.'
+    when 'casual'
+      'Relaxed and informal. Use contractions, simpler words, and a conversational style. Friendly but less formal than professional tone.'
+    when 'professional'
+      'Formal, polished, and business-appropriate. Use complete sentences, proper grammar, ' \
+      'and maintain respectful distance. Avoid slang or overly casual language.'
+    else
+      determine_tone_instruction('friendly')
+    end
+  end
 end
 
 Integrations::Openai::ProcessorService.prepend_mod_with('Integrations::OpenaiProcessorService')