mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-07-08 15:22:47 +00:00
478f31de47
* FEATURE: add inferred concepts system This commit adds a new inferred concepts system that: - Creates a model for storing concept labels that can be applied to topics - Provides AI personas for finding new concepts and matching existing ones - Adds jobs for generating concepts from popular topics - Includes a scheduled job that automatically processes engaging topics * FEATURE: Extend inferred concepts to include posts * Adds support for concepts to be inferred from and applied to posts * Replaces daily task with one that handles both topics and posts * Adds database migration for posts_inferred_concepts join table * Updates PersonaContext to include inferred concepts Co-authored-by: Roman Rizzi <rizziromanalejandro@gmail.com> Co-authored-by: Keegan George <kgeorge13@gmail.com>
64 lines
1.9 KiB
Ruby
64 lines
1.9 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe DiscourseAi::Personas::ConceptFinder do
|
|
let(:persona) { described_class.new }
|
|
|
|
describe ".default_enabled" do
|
|
it "is disabled by default" do
|
|
expect(described_class.default_enabled).to eq(false)
|
|
end
|
|
end
|
|
|
|
describe "#system_prompt" do
|
|
before do
|
|
Fabricate(:inferred_concept, name: "programming")
|
|
Fabricate(:inferred_concept, name: "testing")
|
|
Fabricate(:inferred_concept, name: "ruby")
|
|
end
|
|
|
|
it "includes existing concepts when available" do
|
|
prompt = persona.system_prompt
|
|
|
|
InferredConcept.all.each { |concept| expect(prompt).to include(concept.name) }
|
|
end
|
|
|
|
it "handles empty existing concepts" do
|
|
InferredConcept.destroy_all
|
|
prompt = persona.system_prompt
|
|
|
|
expect(prompt).not_to include("following concepts already exist")
|
|
expect(prompt).to include("advanced concept tagging system")
|
|
end
|
|
|
|
it "limits existing concepts to 100" do
|
|
manager = instance_double(DiscourseAi::InferredConcepts::Manager)
|
|
allow(DiscourseAi::InferredConcepts::Manager).to receive(:new).and_return(manager)
|
|
allow(manager).to receive(:list_concepts).with(limit: 100).and_return(%w[concept1 concept2])
|
|
|
|
persona.system_prompt
|
|
end
|
|
|
|
it "includes format instructions" do
|
|
prompt = persona.system_prompt
|
|
|
|
expect(prompt).to include("<o>")
|
|
expect(prompt).to include('{"concepts": ["concept1", "concept2", "concept3"]}')
|
|
expect(prompt).to include("</o>")
|
|
end
|
|
|
|
it "includes language preservation instruction" do
|
|
prompt = persona.system_prompt
|
|
|
|
expect(prompt).to include("original language of the text")
|
|
end
|
|
end
|
|
|
|
describe "#response_format" do
|
|
it "defines correct response format" do
|
|
format = persona.response_format
|
|
|
|
expect(format).to eq([{ "array_type" => "string", "key" => "concepts", "type" => "array" }])
|
|
end
|
|
end
|
|
end
|