mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-02-18 01:14:53 +00:00
8849caf136
We no longer support the "provider:model" format in the "ai_helper_model" and "ai_embeddings_semantic_search_hyde_model" settings. We'll migrate existing values and work with our new data-driven LLM configs from now on.
89 lines
2.9 KiB
Ruby
89 lines
2.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe Jobs::StreamPostHelper do
|
|
subject(:job) { described_class.new }
|
|
|
|
before { assign_fake_provider_to(:ai_helper_model) }
|
|
|
|
describe "#execute" do
|
|
fab!(:topic)
|
|
fab!(:post) do
|
|
Fabricate(
|
|
:post,
|
|
topic: topic,
|
|
raw:
|
|
"I like to eat pie. It is a very good dessert. Some people are wasteful by throwing pie at others but I do not do that. I always eat the pie.",
|
|
)
|
|
end
|
|
fab!(:user) { Fabricate(:leader) }
|
|
|
|
before do
|
|
Group.find(Group::AUTO_GROUPS[:trust_level_3]).add(user)
|
|
SiteSetting.composer_ai_helper_enabled = true
|
|
end
|
|
|
|
describe "validates params" do
|
|
it "does nothing if there is no post" do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/ai-helper/explain/#{post.id}") do
|
|
job.execute(post_id: nil, user_id: user.id, term_to_explain: "pie")
|
|
end
|
|
|
|
expect(messages).to be_empty
|
|
end
|
|
|
|
it "does nothing if there is no user" do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/ai-helper/explain/#{post.id}") do
|
|
job.execute(post_id: post.id, user_id: nil, term_to_explain: "pie")
|
|
end
|
|
|
|
expect(messages).to be_empty
|
|
end
|
|
|
|
it "does nothing if there is no term to explain" do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/ai-helper/explain/#{post.id}") do
|
|
job.execute(post_id: post.id, user_id: user.id, term_to_explain: nil)
|
|
end
|
|
|
|
expect(messages).to be_empty
|
|
end
|
|
end
|
|
|
|
it "publishes updates with a partial result" do
|
|
explanation =
|
|
"In this context, \"pie\" refers to a baked dessert typically consisting of a pastry crust and filling."
|
|
|
|
partial_explanation = "I"
|
|
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([explanation]) do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/ai-helper/explain/#{post.id}") do
|
|
job.execute(post_id: post.id, user_id: user.id, term_to_explain: "pie")
|
|
end
|
|
|
|
partial_result_update = messages.first.data
|
|
expect(partial_result_update[:done]).to eq(false)
|
|
expect(partial_result_update[:result]).to eq(partial_explanation)
|
|
end
|
|
end
|
|
|
|
it "publishes a final update to signal we're done" do
|
|
explanation =
|
|
"In this context, \"pie\" refers to a baked dessert typically consisting of a pastry crust and filling."
|
|
|
|
DiscourseAi::Completions::Llm.with_prepared_responses([explanation]) do
|
|
messages =
|
|
MessageBus.track_publish("/discourse-ai/ai-helper/explain/#{post.id}") do
|
|
job.execute(post_id: post.id, user_id: user.id, term_to_explain: "pie")
|
|
end
|
|
|
|
final_update = messages.last.data
|
|
expect(final_update[:result]).to eq(explanation)
|
|
expect(final_update[:done]).to eq(true)
|
|
end
|
|
end
|
|
end
|
|
end
|