diff --git a/app/services/llm_formatter/conversation_llm_formatter.rb b/app/services/llm_formatter/conversation_llm_formatter.rb index 1444d75c1..af56cfee3 100644 --- a/app/services/llm_formatter/conversation_llm_formatter.rb +++ b/app/services/llm_formatter/conversation_llm_formatter.rb @@ -11,6 +11,8 @@ class LlmFormatter::ConversationLlmFormatter < LlmFormatter::DefaultLlmFormatter end sections << "Contact Details: #{@record.contact.to_llm_text}" if config[:include_contact_details] + sections << 'Conversation Attributes:' + sections << build_attributes sections.join("\n") end @@ -30,4 +32,12 @@ class LlmFormatter::ConversationLlmFormatter < LlmFormatter::DefaultLlmFormatter sender = message.message_type == 'incoming' ? 'User' : 'Support agent' "#{sender}: #{message.content}\n" end + + def build_attributes + attributes = [] + @record.account.custom_attribute_definitions.with_attribute_model('conversation_attribute').each do |attribute| + attributes << "#{attribute.attribute_display_name}: #{@record.custom_attributes[attribute.attribute_key]}" + end + attributes.join("\n") + end end