mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-09 02:40:50 +00:00
d59ed1091b
This change-set connects GPT based chat with the forum it runs on. Allowing it to perform search, lookup tags and categories and summarize topics. The integration is currently restricted to public portions of the forum. Changes made: - Do not run ai reply job for small actions - Improved composable system prompt - Trivial summarizer for topics - Image generator - Google command for searching via Google - Corrected trimming of posts raw (was replacing with numbers) - Bypass of problem specs The feature works best with GPT-4 --------- Co-authored-by: Roman Rizzi <rizziromanalejandro@gmail.com>
75 lines
2.2 KiB
Ruby
75 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module AiBot
|
|
class EntryPoint
|
|
GPT4_ID = -110
|
|
GPT3_5_TURBO_ID = -111
|
|
CLAUDE_V1_ID = -112
|
|
BOTS = [
|
|
[GPT4_ID, "gpt4_bot"],
|
|
[GPT3_5_TURBO_ID, "gpt3.5_bot"],
|
|
[CLAUDE_V1_ID, "claude_v1_bot"],
|
|
]
|
|
|
|
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-v1"
|
|
CLAUDE_V1_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"
|
|
end
|
|
|
|
def inject_into(plugin)
|
|
plugin.register_seedfu_fixtures(
|
|
Rails.root.join("plugins", "discourse-ai", "db", "fixtures", "ai_bot"),
|
|
)
|
|
|
|
plugin.register_svg_icon("robot")
|
|
|
|
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
|
|
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
|
|
end
|
|
end
|
|
end
|
|
end
|