mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-07-08 15:22:47 +00:00
37 lines
997 B
Ruby
37 lines
997 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
RSpec.describe DiscourseAi::Personas::ConceptMatcher 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
|
||
|
|
let(:prompt) { persona.system_prompt }
|
||
|
|
|
||
|
|
it "includes placeholder for concept list" do
|
||
|
|
expect(prompt).to include("{inferred_concepts}")
|
||
|
|
end
|
||
|
|
|
||
|
|
it "specifies output format" do
|
||
|
|
expect(prompt).to include("matching_concepts")
|
||
|
|
expect(prompt).to include("<o>")
|
||
|
|
expect(prompt).to include('{"matching_concepts": ["concept1", "concept3", "concept5"]}')
|
||
|
|
expect(prompt).to include("</o>")
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
describe "#response_format" do
|
||
|
|
it "defines correct response format" do
|
||
|
|
format = persona.response_format
|
||
|
|
|
||
|
|
expect(format).to eq(
|
||
|
|
[{ "array_type" => "string", "key" => "matching_concepts", "type" => "array" }],
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|