mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-08 10:20:07 +00:00
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
40 lines
1.3 KiB
Ruby
40 lines
1.3 KiB
Ruby
#frozen_string_literal: true
|
|
|
|
RSpec.describe DiscourseAi::AiBot::Tools::SearchSettings do
|
|
fab!(:gpt_35_bot) { Fabricate(:llm_model, name: "gpt-3.5-turbo") }
|
|
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 do
|
|
SiteSetting.ai_bot_enabled = true
|
|
toggle_enabled_bots(bots: [gpt_35_bot])
|
|
end
|
|
|
|
def search_settings(query)
|
|
described_class.new({ query: query }, bot_user: bot_user, llm: llm)
|
|
end
|
|
|
|
describe "#process" do
|
|
it "can handle no results" do
|
|
results = search_settings("this will not exist frogs").invoke
|
|
expect(results[:args]).to eq({ query: "this will not exist frogs" })
|
|
expect(results[:rows]).to eq([])
|
|
end
|
|
|
|
it "can return more many settings with no descriptions if there are lots of hits" do
|
|
results = search_settings("a").invoke
|
|
|
|
expect(results[:rows].length).to be > 30
|
|
expect(results[:rows][0].length).to eq(1)
|
|
end
|
|
|
|
it "can return descriptions if there are few matches" do
|
|
results = search_settings("this will not be found!@,default_locale,ai_bot_enabled").invoke
|
|
|
|
expect(results[:rows].length).to eq(2)
|
|
|
|
expect(results[:rows][0][1]).not_to eq(nil)
|
|
end
|
|
end
|
|
end
|