mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-06 17:30:20 +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>
114 lines
3.5 KiB
Ruby
114 lines
3.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe DiscourseAi::AiBot::EntryPoint do
|
|
describe "#inject_into" do
|
|
describe "subscribes to the post_created event" do
|
|
fab!(:admin) { Fabricate(:admin) }
|
|
let(:gpt_bot) { User.find(described_class::GPT4_ID) }
|
|
fab!(:bot_allowed_group) { Fabricate(:group) }
|
|
|
|
let(:post_args) do
|
|
{
|
|
title: "Dear AI, I want to ask a question",
|
|
raw: "Hello, Can you please tell me a story?",
|
|
archetype: Archetype.private_message,
|
|
target_usernames: [gpt_bot.username].join(","),
|
|
}
|
|
end
|
|
|
|
before do
|
|
SiteSetting.ai_bot_allowed_groups = bot_allowed_group.id
|
|
bot_allowed_group.add(admin)
|
|
end
|
|
|
|
it "queues a job to generate a reply by the AI" do
|
|
expect { PostCreator.create!(admin, post_args) }.to change(
|
|
Jobs::CreateAiReply.jobs,
|
|
:size,
|
|
).by(1)
|
|
end
|
|
|
|
it "does not queue a job for small actions" do
|
|
post = PostCreator.create!(admin, post_args)
|
|
|
|
expect {
|
|
post.topic.add_moderator_post(
|
|
admin,
|
|
"this is a small action",
|
|
post_type: Post.types[:small_action],
|
|
)
|
|
}.not_to change(Jobs::CreateAiReply.jobs, :size)
|
|
|
|
expect {
|
|
post.topic.add_moderator_post(
|
|
admin,
|
|
"this is a small action",
|
|
post_type: Post.types[:moderator_action],
|
|
)
|
|
}.not_to change(Jobs::CreateAiReply.jobs, :size)
|
|
|
|
expect {
|
|
post.topic.add_moderator_post(
|
|
admin,
|
|
"this is a small action",
|
|
post_type: Post.types[:whisper],
|
|
)
|
|
}.not_to change(Jobs::CreateAiReply.jobs, :size)
|
|
end
|
|
|
|
it "includes the bot's user_id" do
|
|
claude_bot = User.find(described_class::CLAUDE_V1_ID)
|
|
claude_post_attrs = post_args.merge(target_usernames: [claude_bot.username].join(","))
|
|
|
|
expect { PostCreator.create!(admin, claude_post_attrs) }.to change(
|
|
Jobs::CreateAiReply.jobs,
|
|
:size,
|
|
).by(1)
|
|
|
|
job_args = Jobs::CreateAiReply.jobs.last["args"].first
|
|
expect(job_args["bot_user_id"]).to eq(claude_bot.id)
|
|
end
|
|
|
|
context "when the post is not from a PM" do
|
|
it "does nothing" do
|
|
expect {
|
|
PostCreator.create!(admin, post_args.merge(archetype: Archetype.default))
|
|
}.not_to change(Jobs::CreateAiReply.jobs, :size)
|
|
end
|
|
end
|
|
|
|
context "when the bot doesn't have access to the PM" do
|
|
it "does nothing" do
|
|
user_2 = Fabricate(:user)
|
|
expect {
|
|
PostCreator.create!(admin, post_args.merge(target_usernames: [user_2.username]))
|
|
}.not_to change(Jobs::CreateAiReply.jobs, :size)
|
|
end
|
|
end
|
|
|
|
context "when the user is not allowed to interact with the bot" do
|
|
it "does nothing" do
|
|
bot_allowed_group.remove(admin)
|
|
expect { PostCreator.create!(admin, post_args) }.not_to change(
|
|
Jobs::CreateAiReply.jobs,
|
|
:size,
|
|
)
|
|
end
|
|
end
|
|
|
|
context "when the post was created by the bot" do
|
|
it "does nothing" do
|
|
gpt_topic_id = PostCreator.create!(admin, post_args).topic_id
|
|
reply_args =
|
|
post_args.except(:archetype, :target_usernames, :title).merge(topic_id: gpt_topic_id)
|
|
|
|
expect { PostCreator.create!(gpt_bot, reply_args) }.not_to change(
|
|
Jobs::CreateAiReply.jobs,
|
|
:size,
|
|
)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|