2023-02-17 09:33:47 -05:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2023-02-22 18:46:53 -05:00
|
|
|
# name: discourse-ai
|
2023-05-11 22:54:25 -04:00
|
|
|
# about: Enables AI modules features in Discourse
|
2023-02-17 09:33:47 -05:00
|
|
|
# version: 0.0.1
|
|
|
|
# authors: Discourse
|
2023-05-11 22:54:25 -04:00
|
|
|
# url: https://meta.discourse.org/t/discourse-ai/259214
|
2023-02-17 09:33:47 -05:00
|
|
|
# required_version: 2.7.0
|
|
|
|
|
2023-07-27 12:55:32 -04:00
|
|
|
gem "tokenizers", "0.3.3"
|
2023-07-13 11:41:36 -04:00
|
|
|
gem "tiktoken_ruby", "0.0.5"
|
2023-10-02 11:58:36 -04:00
|
|
|
gem "aws-eventstream", "1.2.0"
|
2023-04-19 10:55:59 -04:00
|
|
|
|
2023-02-23 09:08:34 -05:00
|
|
|
enabled_site_setting :discourse_ai_enabled
|
2023-02-17 09:33:47 -05:00
|
|
|
|
2023-03-15 16:02:20 -04:00
|
|
|
register_asset "stylesheets/modules/ai-helper/common/ai-helper.scss"
|
2023-05-18 11:34:49 -04:00
|
|
|
|
2023-05-05 14:28:31 -04:00
|
|
|
register_asset "stylesheets/modules/ai-bot/common/bot-replies.scss"
|
2023-03-15 16:02:20 -04:00
|
|
|
|
2023-07-13 15:34:02 -04:00
|
|
|
register_asset "stylesheets/modules/embeddings/common/semantic-related-topics.scss"
|
2023-09-05 10:08:23 -04:00
|
|
|
register_asset "stylesheets/modules/embeddings/common/semantic-search.scss"
|
2023-07-13 15:34:02 -04:00
|
|
|
|
2023-03-14 15:03:50 -04:00
|
|
|
module ::DiscourseAi
|
|
|
|
PLUGIN_NAME = "discourse-ai"
|
|
|
|
end
|
|
|
|
|
2023-03-07 13:39:28 -05:00
|
|
|
require_relative "lib/discourse_ai/engine"
|
|
|
|
|
2023-02-24 05:53:43 -05:00
|
|
|
after_initialize do
|
2023-03-07 14:14:39 -05:00
|
|
|
require_relative "lib/shared/inference/discourse_classifier"
|
|
|
|
require_relative "lib/shared/inference/discourse_reranker"
|
|
|
|
require_relative "lib/shared/inference/openai_completions"
|
|
|
|
require_relative "lib/shared/inference/openai_embeddings"
|
2023-04-10 10:04:42 -04:00
|
|
|
require_relative "lib/shared/inference/anthropic_completions"
|
2023-05-19 19:38:08 -04:00
|
|
|
require_relative "lib/shared/inference/stability_generator"
|
2023-07-27 12:55:32 -04:00
|
|
|
require_relative "lib/shared/inference/hugging_face_text_generation"
|
2023-10-02 11:58:36 -04:00
|
|
|
require_relative "lib/shared/inference/amazon_bedrock_inference"
|
2023-10-04 12:47:51 -04:00
|
|
|
require_relative "lib/shared/inference/cloudflare_workers_ai"
|
2023-08-22 17:49:36 -04:00
|
|
|
require_relative "lib/shared/inference/function"
|
|
|
|
require_relative "lib/shared/inference/function_list"
|
2023-03-07 14:14:39 -05:00
|
|
|
|
2023-02-28 09:17:03 -05:00
|
|
|
require_relative "lib/shared/classificator"
|
|
|
|
require_relative "lib/shared/post_classificator"
|
|
|
|
require_relative "lib/shared/chat_message_classificator"
|
2023-02-24 05:53:43 -05:00
|
|
|
|
2023-04-19 10:55:59 -04:00
|
|
|
require_relative "lib/shared/tokenizer/tokenizer"
|
|
|
|
|
2023-03-15 16:21:45 -04:00
|
|
|
require_relative "lib/shared/database/connection"
|
|
|
|
|
2023-02-24 05:53:43 -05:00
|
|
|
require_relative "lib/modules/nsfw/entry_point"
|
|
|
|
require_relative "lib/modules/toxicity/entry_point"
|
|
|
|
require_relative "lib/modules/sentiment/entry_point"
|
2023-03-15 16:02:20 -04:00
|
|
|
require_relative "lib/modules/ai_helper/entry_point"
|
2023-03-15 16:21:45 -04:00
|
|
|
require_relative "lib/modules/embeddings/entry_point"
|
2023-04-04 10:24:09 -04:00
|
|
|
require_relative "lib/modules/summarization/entry_point"
|
2023-05-05 14:28:31 -04:00
|
|
|
require_relative "lib/modules/ai_bot/entry_point"
|
2023-10-02 17:55:30 -04:00
|
|
|
require_relative "lib/discourse_automation/llm_triage"
|
2023-02-22 18:46:53 -05:00
|
|
|
|
2023-03-07 13:39:28 -05:00
|
|
|
[
|
2023-03-15 16:21:45 -04:00
|
|
|
DiscourseAi::Embeddings::EntryPoint.new,
|
2023-03-14 15:03:50 -04:00
|
|
|
DiscourseAi::NSFW::EntryPoint.new,
|
|
|
|
DiscourseAi::Toxicity::EntryPoint.new,
|
|
|
|
DiscourseAi::Sentiment::EntryPoint.new,
|
2023-03-15 16:02:20 -04:00
|
|
|
DiscourseAi::AiHelper::EntryPoint.new,
|
2023-04-04 10:24:09 -04:00
|
|
|
DiscourseAi::Summarization::EntryPoint.new,
|
2023-05-05 14:28:31 -04:00
|
|
|
DiscourseAi::AiBot::EntryPoint.new,
|
2023-03-07 13:39:28 -05:00
|
|
|
].each do |a_module|
|
2023-02-23 13:50:10 -05:00
|
|
|
a_module.load_files
|
|
|
|
a_module.inject_into(self)
|
|
|
|
end
|
2023-03-07 13:39:28 -05:00
|
|
|
|
2023-03-14 15:03:50 -04:00
|
|
|
register_reviewable_type ReviewableAiChatMessage
|
|
|
|
register_reviewable_type ReviewableAiPost
|
2023-03-07 13:39:28 -05:00
|
|
|
|
|
|
|
on(:reviewable_transitioned_to) do |new_status, reviewable|
|
|
|
|
ModelAccuracy.adjust_model_accuracy(new_status, reviewable)
|
|
|
|
end
|
2023-02-22 18:46:53 -05:00
|
|
|
end
|