mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-10-28 04:58:39 +00:00
98c89953d3
People tend to keep to 1 persona when working with the bot, this adds local browser memory for the last persona you interacted with so you do not need to select it over and over again. This is per browser, not per user memory. Also... clean up tests so they do not need to require stubs which were breaking the build --------- Co-authored-by: Martin Brennan <martin@discourse.org>
38 lines
1.1 KiB
Ruby
38 lines
1.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe Jobs::GenerateEmbeddings do
|
|
subject(:job) { described_class.new }
|
|
|
|
describe "#execute" do
|
|
before do
|
|
SiteSetting.ai_embeddings_discourse_service_api_endpoint = "http://test.com"
|
|
SiteSetting.ai_embeddings_enabled = true
|
|
SiteSetting.ai_embeddings_model = "all-mpnet-base-v2"
|
|
end
|
|
|
|
fab!(:topic) { Fabricate(:topic) }
|
|
fab!(:post) { Fabricate(:post, post_number: 1, topic: topic) }
|
|
|
|
let(:truncation) { DiscourseAi::Embeddings::Strategies::Truncation.new }
|
|
let(:vector_rep) do
|
|
DiscourseAi::Embeddings::VectorRepresentations::Base.current_representation(truncation)
|
|
end
|
|
|
|
it "works" do
|
|
expected_embedding = [0.0038493] * vector_rep.dimensions
|
|
|
|
text =
|
|
truncation.prepare_text_from(
|
|
topic,
|
|
vector_rep.tokenizer,
|
|
vector_rep.max_sequence_length - 2,
|
|
)
|
|
EmbeddingsGenerationStubs.discourse_service(vector_rep.name, text, expected_embedding)
|
|
|
|
job.execute(topic_id: topic.id)
|
|
|
|
expect(vector_rep.topic_id_from_representation(expected_embedding)).to eq(topic.id)
|
|
end
|
|
end
|
|
end
|