Generate custom field server-side (capture all cases)

Author: markvanlan

assets/javascripts/discourse/services/ai-bot-conversations-hidden-submit.js

@@ -53,7 +53,6 @@ export default class AiBotConversationsHiddenSubmit extends Service {
           archetype: "private_message",
           target_recipients: this.targetUsername,
           meta_data: { ai_persona_id: this.personaId },
-          topic_custom_fields: { is_ai_bot_pm: true },
         },
       });
 

lib/ai_bot/entry_point.rb

@@ -65,7 +65,33 @@ def self.ai_share_error(topic, guardian)
       end
 
       def inject_into(plugin)
-        plugin.register_editable_topic_custom_field(TOPIC_AI_BOT_PM_FIELD)
+        plugin.register_topic_custom_field_type(TOPIC_AI_BOT_PM_FIELD, :string)
+
+        plugin.on(:topic_created) do |topic|
+          next if !topic.private_message?
+          creator = topic.user
+
+          # Only process if creator is not a bot or system user
+          next if DiscourseAi::AiBot::Playground.is_bot_user_id?(creator.id)
+          # Get all bot user IDs defined by the discourse-ai plugin
+
+          bot_ids = DiscourseAi::AiBot::EntryPoint.all_bot_ids
+
+          # Check if the only recipients are bots
+          recipients = topic.topic_allowed_users.pluck(:user_id)
+
+          # Remove creator from recipients for checking
+          recipients -= [creator.id]
+
+          # If all remaining recipients are AI bots and there's exactly one recipient
+          if recipients.length == 1 && (recipients - bot_ids).empty?
+            # The only recipient is an AI bot - add the custom field to the topic
+            topic.custom_fields[TOPIC_AI_BOT_PM_FIELD] = true
+
+            # Save the custom fields
+            topic.save_custom_fields
+          end
+        end
 
         plugin.register_modifier(:chat_allowed_bot_user_ids) do |user_ids, guardian|
           if guardian.user

spec/lib/modules/ai_bot/entry_point_spec.rb

@@ -35,6 +35,20 @@
         expect(serializer[:current_user][:can_debug_ai_bot_conversations]).to eq(true)
       end
 
+      describe "adding TOPIC_AI_BOT_PM_FIELD to topic custom fields" do
+        it "is added when user PMs a single bot" do
+          topic = PostCreator.create!(admin, post_args).topic
+          expect(topic.reload.custom_fields[DiscourseAi::AiBot::TOPIC_AI_BOT_PM_FIELD]).to eq("t")
+        end
+
+        it "is not added when user PMs a bot and another user" do
+          user = Fabricate(:user)
+          post_args[:target_usernames] = [gpt_bot.username, user.username].join(",")
+          topic = PostCreator.create!(admin, post_args).topic
+          expect(topic.reload.custom_fields[DiscourseAi::AiBot::TOPIC_AI_BOT_PM_FIELD]).to be_nil
+        end
+      end
+
       it "adds information about forcing default llm to current_user_serializer" do
         Group.refresh_automatic_groups!