# frozen_string_literal: true RSpec.describe DiscourseAi::AiBot::Playground do subject(:playground) { described_class.new(bot) } fab!(:bot_user) do SiteSetting.ai_bot_enabled_chat_bots = "claude-2" SiteSetting.ai_bot_enabled = true User.find(DiscourseAi::AiBot::EntryPoint::CLAUDE_V2_ID) end fab!(:bot) do persona = AiPersona .find( DiscourseAi::AiBot::Personas::Persona.system_personas[ DiscourseAi::AiBot::Personas::General ], ) .class_instance .new DiscourseAi::AiBot::Bot.as(bot_user, persona: persona) end fab!(:admin) { Fabricate(:admin, refresh_auto_groups: true) } fab!(:user) { Fabricate(:user, refresh_auto_groups: true) } fab!(:pm) do Fabricate( :private_message_topic, title: "This is my special PM", user: user, topic_allowed_users: [ Fabricate.build(:topic_allowed_user, user: user), Fabricate.build(:topic_allowed_user, user: bot_user), ], ) end fab!(:first_post) do Fabricate(:post, topic: pm, user: user, post_number: 1, raw: "This is a reply by the user") end fab!(:second_post) do Fabricate(:post, topic: pm, user: bot_user, post_number: 2, raw: "This is a bot reply") end fab!(:third_post) do Fabricate( :post, topic: pm, user: user, post_number: 3, raw: "This is a second reply by the user", ) end describe "is_bot_user_id?" do it "properly detects ALL bots as bot users" do persona = Fabricate(:ai_persona, enabled: false) persona.create_user! expect(DiscourseAi::AiBot::Playground.is_bot_user_id?(persona.user_id)).to eq(true) end end describe "persona with user support" do before do Jobs.run_immediately! SiteSetting.ai_bot_allowed_groups = "#{Group::AUTO_GROUPS[:trust_level_0]}" end fab!(:persona) do persona = AiPersona.create!( name: "Test Persona", description: "A test persona", allowed_group_ids: [Group::AUTO_GROUPS[:trust_level_0]], enabled: true, system_prompt: "You are a helpful bot", ) persona.create_user! persona.update!(default_llm: "anthropic:claude-2", mentionable: true) persona end it "replies to whispers with a whisper" do post = nil DiscourseAi::Completions::Llm.with_prepared_responses(["Yes I can"]) do post = create_post( title: "My public topic", raw: "Hey @#{persona.user.username}, can you help me?", post_type: Post.types[:whisper], ) end post.topic.reload last_post = post.topic.posts.order(:post_number).last expect(last_post.raw).to eq("Yes I can") expect(last_post.user_id).to eq(persona.user_id) expect(last_post.post_type).to eq(Post.types[:whisper]) end it "allows mentioning a persona" do # we still should be able to mention with no bots SiteSetting.ai_bot_enabled_chat_bots = "" post = nil DiscourseAi::Completions::Llm.with_prepared_responses(["Yes I can"]) do post = create_post( title: "My public topic", raw: "Hey @#{persona.user.username}, can you help me?", ) end post.topic.reload last_post = post.topic.posts.order(:post_number).last expect(last_post.raw).to eq("Yes I can") expect(last_post.user_id).to eq(persona.user_id) end it "allows PMing a persona even when no particular bots are enabled" do SiteSetting.ai_bot_enabled = true SiteSetting.ai_bot_enabled_chat_bots = "" post = nil DiscourseAi::Completions::Llm.with_prepared_responses( ["Magic title", "Yes I can"], llm: "anthropic:claude-2", ) do post = create_post( title: "I just made a PM", raw: "Hey there #{persona.user.username}, can you help me?", target_usernames: "#{user.username},#{persona.user.username}", archetype: Archetype.private_message, user: admin, ) end last_post = post.topic.posts.order(:post_number).last expect(last_post.raw).to eq("Yes I can") expect(last_post.user_id).to eq(persona.user_id) last_post.topic.reload expect(last_post.topic.allowed_users.pluck(:user_id)).to include(persona.user_id) expect(last_post.topic.participant_count).to eq(2) end it "picks the correct llm for persona in PMs" do # If you start a PM with GPT 3.5 bot, replies should come from it, not from Claude SiteSetting.ai_bot_enabled = true SiteSetting.ai_bot_enabled_chat_bots = "gpt-3.5-turbo|claude-2" post = nil gpt3_5_bot_user = User.find(DiscourseAi::AiBot::EntryPoint::GPT3_5_TURBO_ID) # title is queued first, ensures it uses the llm targeted via target_usernames not claude DiscourseAi::Completions::Llm.with_prepared_responses( ["Magic title", "Yes I can"], llm: "open_ai:gpt-3.5-turbo-16k", ) do post = create_post( title: "I just made a PM", raw: "Hey @#{persona.user.username}, can you help me?", target_usernames: "#{user.username},#{gpt3_5_bot_user.username}", archetype: Archetype.private_message, user: admin, ) end last_post = post.topic.posts.order(:post_number).last expect(last_post.raw).to eq("Yes I can") expect(last_post.user_id).to eq(persona.user_id) last_post.topic.reload expect(last_post.topic.allowed_users.pluck(:user_id)).to include(persona.user_id) end end describe "#title_playground" do let(:expected_response) { "This is a suggested title" } before { SiteSetting.min_personal_message_post_length = 5 } it "updates the title using bot suggestions" do DiscourseAi::Completions::Llm.with_prepared_responses([expected_response]) do playground.title_playground(third_post) expect(pm.reload.title).to eq(expected_response) end end end describe "#reply_to" do it "streams the bot reply through MB and create a new post in the PM with a cooked responses" do expected_bot_response = "Hello this is a bot and what you just said is an interesting question" DiscourseAi::Completions::Llm.with_prepared_responses([expected_bot_response]) do messages = MessageBus.track_publish("discourse-ai/ai-bot/topic/#{pm.id}") do playground.reply_to(third_post) end reply = pm.reload.posts.last noop_signal = messages.pop expect(noop_signal.data[:noop]).to eq(true) done_signal = messages.pop expect(done_signal.data[:done]).to eq(true) expect(done_signal.data[:cooked]).to eq(reply.cooked) expect(messages.first.data[:raw]).to eq("") messages[1..-1].each_with_index do |m, idx| expect(m.data[:raw]).to eq(expected_bot_response[0..idx]) end expect(reply.cooked).to eq(PrettyText.cook(expected_bot_response)) end end it "supports multiple function calls" do response1 = (<<~TXT).strip search search testing various things search search another search TXT response2 = "I found stuff" DiscourseAi::Completions::Llm.with_prepared_responses([response1, response2]) do playground.reply_to(third_post) end last_post = third_post.topic.reload.posts.order(:post_number).last expect(last_post.raw).to include("testing various things") expect(last_post.raw).to include("another search") expect(last_post.raw).to include("I found stuff") end it "does not include placeholders in conversation context but includes all completions" do response1 = (<<~TXT).strip search search testing various things TXT response2 = "I found some really amazing stuff!" DiscourseAi::Completions::Llm.with_prepared_responses([response1, response2]) do playground.reply_to(third_post) end last_post = third_post.topic.reload.posts.order(:post_number).last custom_prompt = PostCustomPrompt.where(post_id: last_post.id).first.custom_prompt expect(custom_prompt.length).to eq(3) expect(custom_prompt.to_s).not_to include("
") expect(custom_prompt.last.first).to eq(response2) expect(custom_prompt.last.last).to eq(bot_user.username) end context "with Dall E bot" do let(:bot) do persona = AiPersona .find( DiscourseAi::AiBot::Personas::Persona.system_personas[ DiscourseAi::AiBot::Personas::DallE3 ], ) .class_instance .new DiscourseAi::AiBot::Bot.as(bot_user, persona: persona) end it "does not include placeholders in conversation context (simulate DALL-E)" do SiteSetting.ai_openai_api_key = "123" response = (<<~TXT).strip dall_e dall_e ["a pink cow"] TXT image = "iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg==" data = [{ b64_json: image, revised_prompt: "a pink cow 1" }] WebMock.stub_request(:post, SiteSetting.ai_openai_dall_e_3_url).to_return( status: 200, body: { data: data }.to_json, ) DiscourseAi::Completions::Llm.with_prepared_responses([response]) do playground.reply_to(third_post) end last_post = third_post.topic.reload.posts.order(:post_number).last custom_prompt = PostCustomPrompt.where(post_id: last_post.id).first.custom_prompt # DALL E has custom_raw, we do not want to inject this into the prompt stream expect(custom_prompt.length).to eq(2) expect(custom_prompt.to_s).not_to include("
") end end end describe "#conversation_context" do context "with limited context" do before do @old_persona = playground.bot.persona persona = Fabricate(:ai_persona, max_context_posts: 1) playground.bot.persona = persona.class_instance.new end after { playground.bot.persona = @old_persona } it "respects max_context_post" do context = playground.conversation_context(third_post) expect(context).to contain_exactly( *[{ type: :user, id: user.username, content: third_post.raw }], ) end end it "includes previous posts ordered by post_number" do context = playground.conversation_context(third_post) expect(context).to contain_exactly( *[ { type: :user, id: user.username, content: third_post.raw }, { type: :model, content: second_post.raw }, { type: :user, id: user.username, content: first_post.raw }, ], ) end it "only include regular posts" do first_post.update!(post_type: Post.types[:whisper]) context = playground.conversation_context(third_post) expect(context).to contain_exactly( *[ { type: :user, id: user.username, content: third_post.raw }, { type: :model, content: second_post.raw }, ], ) end context "with custom prompts" do it "When post custom prompt is present, we use that instead of the post content" do custom_prompt = [ [ { args: { timezone: "Buenos Aires" }, time: "2023-12-14 17:24:00 -0300" }.to_json, "time", "tool", ], [ { name: "time", arguments: { name: "time", timezone: "Buenos Aires" } }.to_json, "time", "tool_call", ], ["I replied this thanks to the time command", bot_user.username], ] PostCustomPrompt.create!(post: second_post, custom_prompt: custom_prompt) context = playground.conversation_context(third_post) expect(context).to contain_exactly( *[ { type: :user, id: user.username, content: third_post.raw }, { type: :model, content: custom_prompt.third.first }, { type: :tool_call, content: custom_prompt.second.first, id: "time" }, { type: :tool, id: "time", content: custom_prompt.first.first }, { type: :user, id: user.username, content: first_post.raw }, ], ) end it "include replies generated from tools" do custom_prompt = [ [ { args: { timezone: "Buenos Aires" }, time: "2023-12-14 17:24:00 -0300" }.to_json, "time", "tool", ], [ { name: "time", arguments: { name: "time", timezone: "Buenos Aires" } }.to_json, "time", "tool_call", ], ["I replied", bot_user.username], ] PostCustomPrompt.create!(post: second_post, custom_prompt: custom_prompt) PostCustomPrompt.create!(post: first_post, custom_prompt: custom_prompt) context = playground.conversation_context(third_post) expect(context).to contain_exactly( *[ { type: :user, id: user.username, content: third_post.raw }, { type: :model, content: custom_prompt.third.first }, { type: :tool_call, content: custom_prompt.second.first, id: "time" }, { type: :tool, id: "time", content: custom_prompt.first.first }, { type: :tool_call, content: custom_prompt.second.first, id: "time" }, { type: :tool, id: "time", content: custom_prompt.first.first }, { type: :model, content: "I replied" }, ], ) end end end end