147 lines
4.9 KiB
Ruby
147 lines
4.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module AiBot
|
|
class EntryPoint
|
|
REQUIRE_TITLE_UPDATE = "discourse-ai-title-update"
|
|
|
|
GPT4_ID = -110
|
|
GPT3_5_TURBO_ID = -111
|
|
CLAUDE_V2_ID = -112
|
|
BOTS = [
|
|
[GPT4_ID, "gpt4_bot", "gpt-4"],
|
|
[GPT3_5_TURBO_ID, "gpt3.5_bot", "gpt-3.5-turbo"],
|
|
[CLAUDE_V2_ID, "claude_bot", "claude-2"],
|
|
]
|
|
|
|
def self.map_bot_model_to_user_id(model_name)
|
|
case model_name
|
|
in "gpt-3.5-turbo"
|
|
GPT3_5_TURBO_ID
|
|
in "gpt-4"
|
|
GPT4_ID
|
|
in "claude-2"
|
|
CLAUDE_V2_ID
|
|
else
|
|
nil
|
|
end
|
|
end
|
|
|
|
def load_files
|
|
require_relative "jobs/regular/create_ai_reply"
|
|
require_relative "jobs/regular/update_ai_bot_pm_title"
|
|
require_relative "bot"
|
|
require_relative "anthropic_bot"
|
|
require_relative "open_ai_bot"
|
|
require_relative "commands/command"
|
|
require_relative "commands/search_command"
|
|
require_relative "commands/categories_command"
|
|
require_relative "commands/tags_command"
|
|
require_relative "commands/time_command"
|
|
require_relative "commands/summarize_command"
|
|
require_relative "commands/image_command"
|
|
require_relative "commands/google_command"
|
|
require_relative "commands/read_command"
|
|
require_relative "commands/setting_context_command"
|
|
require_relative "commands/search_settings_command"
|
|
require_relative "commands/db_schema_command"
|
|
require_relative "personas/persona"
|
|
require_relative "personas/artist"
|
|
require_relative "personas/general"
|
|
require_relative "personas/sql_helper"
|
|
require_relative "personas/settings_explorer"
|
|
require_relative "personas/researcher"
|
|
require_relative "personas/creative"
|
|
require_relative "site_settings_extension"
|
|
end
|
|
|
|
def inject_into(plugin)
|
|
plugin.on(:site_setting_changed) do |name, _old_value, _new_value|
|
|
if name == :ai_bot_enabled_chat_bots || name == :ai_bot_enabled
|
|
DiscourseAi::AiBot::SiteSettingsExtension.enable_or_disable_ai_bots
|
|
end
|
|
end
|
|
|
|
plugin.register_seedfu_fixtures(
|
|
Rails.root.join("plugins", "discourse-ai", "db", "fixtures", "ai_bot"),
|
|
)
|
|
|
|
plugin.add_to_serializer(
|
|
:current_user,
|
|
:ai_enabled_personas,
|
|
include_condition: -> do
|
|
SiteSetting.ai_bot_enabled && scope.authenticated? &&
|
|
scope.user.in_any_groups?(SiteSetting.ai_bot_allowed_groups_map)
|
|
end,
|
|
) do
|
|
Personas.all.map { |persona| { name: persona.name, description: persona.description } }
|
|
end
|
|
|
|
plugin.add_to_serializer(
|
|
:current_user,
|
|
:ai_enabled_chat_bots,
|
|
include_condition: -> do
|
|
SiteSetting.ai_bot_enabled && scope.authenticated? &&
|
|
scope.user.in_any_groups?(SiteSetting.ai_bot_allowed_groups_map)
|
|
end,
|
|
) do
|
|
model_map = {}
|
|
SiteSetting
|
|
.ai_bot_enabled_chat_bots
|
|
.split("|")
|
|
.each do |bot_name|
|
|
model_map[
|
|
::DiscourseAi::AiBot::EntryPoint.map_bot_model_to_user_id(bot_name)
|
|
] = bot_name
|
|
end
|
|
|
|
# not 100% ideal, cause it is one extra query, but we need it
|
|
bots = DB.query_hash(<<~SQL, user_ids: model_map.keys)
|
|
SELECT username, id FROM users WHERE id IN (:user_ids)
|
|
SQL
|
|
|
|
bots.each { |hash| hash["model_name"] = model_map[hash["id"]] }
|
|
bots
|
|
end
|
|
|
|
plugin.register_svg_icon("robot")
|
|
|
|
plugin.add_to_serializer(
|
|
:topic_view,
|
|
:ai_persona_name,
|
|
include_condition: -> { SiteSetting.ai_bot_enabled && object.topic.private_message? },
|
|
) { topic.custom_fields["ai_persona"] }
|
|
|
|
plugin.on(:post_created) do |post|
|
|
bot_ids = BOTS.map(&:first)
|
|
|
|
if post.post_type == Post.types[:regular] && post.topic.private_message? &&
|
|
!bot_ids.include?(post.user_id)
|
|
if (SiteSetting.ai_bot_allowed_groups_map & post.user.group_ids).present?
|
|
bot_id = post.topic.topic_allowed_users.where(user_id: bot_ids).first&.user_id
|
|
|
|
if bot_id
|
|
if post.post_number == 1
|
|
post.topic.custom_fields[REQUIRE_TITLE_UPDATE] = true
|
|
post.topic.save_custom_fields
|
|
end
|
|
Jobs.enqueue(:create_ai_reply, post_id: post.id, bot_user_id: bot_id)
|
|
Jobs.enqueue_in(
|
|
5.minutes,
|
|
:update_ai_bot_pm_title,
|
|
post_id: post.id,
|
|
bot_user_id: bot_id,
|
|
)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
if plugin.respond_to?(:register_editable_topic_custom_field)
|
|
plugin.register_editable_topic_custom_field(:ai_persona)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|