diff --git a/lib/integrations/openai/processor_service.rb b/lib/integrations/openai/processor_service.rb index 17a34e23f..312886169 100644 --- a/lib/integrations/openai/processor_service.rb +++ b/lib/integrations/openai/processor_service.rb @@ -1,28 +1,22 @@ -class Integrations::Openai::ProcessorService - # 3.5 support 4,096 tokens - # 1 token is approx 4 characters - # 4,096 * 4 = 16,384 characters, sticking to 15,000 to be safe - TOKEN_LIMIT = 15_000 - API_URL = 'https://api.openai.com/v1/chat/completions'.freeze - GPT_MODEL = 'gpt-3.5-turbo'.freeze +class Integrations::Openai::ProcessorService < Integrations::OpenaiProcessorService + def reply_suggestion_message + make_api_call(reply_suggestion_body) + end - ALLOWED_EVENT_NAMES = %w[rephrase summarize reply_suggestion].freeze + def summarize_message + make_api_call(summarize_body) + end - pattr_initialize [:hook!, :event!] + def rephrase_message + make_api_call(rephrase_body) + end - def perform - event_name = event['name'] - return nil unless valid_event_name?(event_name) - - send("#{event_name}_message") + def suggest_label_message + make_api_call(suggest_label_body) end private - def valid_event_name?(event_name) - ALLOWED_EVENT_NAMES.include?(event_name) - end - def rephrase_body { model: GPT_MODEL, @@ -42,12 +36,19 @@ class Integrations::Openai::ProcessorService add_messages_until_token_limit(conversation, messages, in_array_format) end - def find_conversation - hook.account.conversations.find_by(display_id: event['data']['conversation_display_id']) + def labels_with_messages + labels = hook.account.labels.pluck(:title).join(', ') + + character_count = labels.length + conversation = find_conversation + messages = init_messages_body(false) + add_messages_until_token_limit(conversation, messages, false, character_count) + + "Messages:\n#{messages}\nLabels:\n#{labels}" end - def add_messages_until_token_limit(conversation, messages, in_array_format) - character_count = 0 + def add_messages_until_token_limit(conversation, messages, in_array_format, start_from = 0) + character_count = start_from conversation.messages.chat.reorder('id desc').each do |message| character_count, message_added = add_message_if_within_limit(character_count, message, messages, in_array_format) break unless message_added @@ -113,25 +114,24 @@ class Integrations::Openai::ProcessorService }.to_json end - def reply_suggestion_message - make_api_call(reply_suggestion_body) - end - - def summarize_message - make_api_call(summarize_body) - end - - def rephrase_message - make_api_call(rephrase_body) - end - - def make_api_call(body) - headers = { - 'Content-Type' => 'application/json', - 'Authorization' => "Bearer #{hook.settings['api_key']}" - } - - response = HTTParty.post(API_URL, headers: headers, body: body) - JSON.parse(response.body)['choices'].first['message']['content'] + def suggest_label_body + { + model: GPT_MODEL, + messages: [ + { + role: 'system', + content: 'Your role is as an assistant to a customer support agent. You will be provided with a transcript of a conversation between a ' \ + 'customer and the support agent, along with a list of potential labels. ' \ + 'Your task is to analyze the conversation and select the two labels from the given list that most accurately ' \ + 'represent the themes or issues discussed. Ensure you preserve the exact casing of the labels as they are provided in the list. ' \ + 'Do not create new labels; only choose from those provided. Once you have made your selections, please provide your response ' \ + 'as a comma-separated list of the provided labels. Remember, your response should only contain the labels you\'ve selected, ' \ + 'in their original casing, and nothing else. ' + }, + { + role: 'user', content: labels_with_messages + } + ] + }.to_json end end diff --git a/lib/integrations/openai_processor_service.rb b/lib/integrations/openai_processor_service.rb new file mode 100644 index 000000000..4f5170f0f --- /dev/null +++ b/lib/integrations/openai_processor_service.rb @@ -0,0 +1,36 @@ +class Integrations::OpenaiProcessorService + TOKEN_LIMIT = 15_000 + 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 + + pattr_initialize [:hook!, :event!] + + def perform + event_name = event['name'] + return nil unless valid_event_name?(event_name) + + send("#{event_name}_message") + end + + private + + def find_conversation + hook.account.conversations.find_by(display_id: event['data']['conversation_display_id']) + end + + def valid_event_name?(event_name) + ALLOWED_EVENT_NAMES.include?(event_name) + end + + def make_api_call(body) + headers = { + 'Content-Type' => 'application/json', + 'Authorization' => "Bearer #{hook.settings['api_key']}" + } + + response = HTTParty.post(API_URL, headers: headers, body: body) + JSON.parse(response.body)['choices'].first['message']['content'] + end +end