Roman Rizzi
8d5f901a67
* DRAFT: Create AI Bot users dynamically and support custom LlmModels * Get user associated to llm_model * Track enabled bots with attribute * Don't store bot username. Minor touches to migrate default values in settings * Handle scenario where vLLM uses a SRV record * Made 3.5-turbo-16k the default version so we can remove hack
47 lines
1.4 KiB
Ruby
47 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
def has_rg?
|
|
if defined?(@has_rg)
|
|
@has_rg
|
|
else
|
|
@has_rg |= system("which rg")
|
|
end
|
|
end
|
|
|
|
RSpec.describe DiscourseAi::AiBot::Tools::SettingContext, if: has_rg? do
|
|
let(:bot_user) { DiscourseAi::AiBot::EntryPoint.find_user_from_model("gpt-3.5-turbo") }
|
|
let(:llm) { DiscourseAi::Completions::Llm.proxy("open_ai:gpt-3.5-turbo") }
|
|
|
|
before { SiteSetting.ai_bot_enabled = true }
|
|
|
|
def setting_context(setting_name)
|
|
described_class.new({ setting_name: setting_name }, bot_user: bot_user, llm: llm)
|
|
end
|
|
|
|
describe "#execute" do
|
|
it "returns the context for core setting" do
|
|
result = setting_context("moderators_view_emails").invoke
|
|
|
|
expect(result[:setting_name]).to eq("moderators_view_emails")
|
|
|
|
expect(result[:context]).to include("site_settings.yml")
|
|
expect(result[:context]).to include("moderators_view_emails")
|
|
end
|
|
|
|
it "returns the context for plugin setting" do
|
|
result = setting_context("ai_bot_enabled").invoke
|
|
|
|
expect(result[:setting_name]).to eq("ai_bot_enabled")
|
|
expect(result[:context]).to include("ai_bot_enabled:")
|
|
end
|
|
|
|
context "when the setting does not exist" do
|
|
it "returns an error message" do
|
|
result = setting_context("this_setting_does_not_exist").invoke
|
|
|
|
expect(result[:context]).to eq("This setting does not exist")
|
|
end
|
|
end
|
|
end
|
|
end
|