plugin.rb

# frozen_string_literal: true
# name: discourse-ai-topic-summary
# about: Uses a remote (OpenAI) AI language model to prepare and post a summary of a Topic
# version: 0.5.0
# authors: Robert Barrow
# contact_emails: merefield@gmail.com
# url: https://github.com/merefield/discourse-ai-topic-summary

gem 'multipart-post', '2.4.0', { require: false }
gem 'faraday-multipart', '1.0.4', { require: false }
gem 'event_stream_parser', '1.0.0', { require: false }
gem "ruby-openai", '7.0.0', { require: false }

register_asset 'stylesheets/common/ai_topic_summary.scss'

enabled_site_setting :ai_topic_summary_enabled

module ::AiTopicSummary

  def progress_debug_message(message)
    puts "AI Topic Summary: #{message}" if SiteSetting.ai_topic_summary_verbose_console_logging
    Rails.logger.info("AI Topic Summary: #{message}") if SiteSetting.ai_topic_summary_verbose_rails_logging
  end

  module_function :progress_debug_message
end

after_initialize do
  %w[
  ../lib/ai_topic_summary/engine.rb
  ../lib/ai_topic_summary/call_bot.rb
  ../app/models/ai_topic_summary/tag_embedding.rb
  ../lib/ai_topic_summary/embeddings/embedding_process.rb
  ../lib/ai_topic_summary/embeddings/tag_embedding_process.rb
  ../lib/ai_topic_summary/embeddings/embedding_completionist_process.rb
  ../lib/ai_topic_summary/summarise.rb
  ../app/controllers/ai_topic_summary/vote.rb
  ../app/controllers/ai_topic_summary/ai_summary.rb
  ../app/jobs/regular/ai_topic_summary_summarise_topic.rb
  ../app/jobs/regular/ai_topic_summary_tag_embedding.rb
  ../app/jobs/scheduled/ai_topic_summary_embeddings_set_completer.rb
  ../app/serializers/topic_list_item_edits.rb
  ../config/routes.rb
  ].each do |path|
    load File.expand_path(path, __FILE__)
  end

  Topic.register_custom_field_type('ai_summary', :json)

  add_to_class(:topic, :ai_summary) { self.custom_fields['ai_summary'] }

  add_to_serializer(:topic_view, :ai_summary, respect_plugin_enabled: true ) { object.topic.ai_summary }

  add_preloaded_topic_list_custom_field("ai_summary")

  DiscourseEvent.on(:post_created) do |*params|
    post, opts, user = params

    if SiteSetting.ai_topic_summary_enabled
      skip = false

      posts_count = post.topic.posts_count

      skip = true if posts_count <= SiteSetting.ai_topic_summary_enabled_min_posts 

      is_private_msg = post.topic.private_message?

      skip = true if !SiteSetting.ai_topic_summary_permitted_in_private_messages && is_private_msg

      permitted_categories = SiteSetting.ai_topic_summary_permitted_categories.split('|')

      skip = true if !SiteSetting.ai_topic_summary_permitted_all_categories && !permitted_categories.include?(post.topic.category_id.to_s)

      if !skip &&
        (post.topic.ai_summary.nil? ||
        (!post.topic.ai_summary.nil? &&
         !post.topic.ai_summary["post_count"].nil? &&
          posts_count >= post.topic.ai_summary["post_count"] + SiteSetting.ai_topic_summary_enabled_post_interval_rerun &&
          posts_count <= SiteSetting.ai_topic_summary_post_limit))
          Jobs.enqueue(:ai_topic_summary_summarise_topic, topic_id: post.topic.id)
      end
    end
  end
end