FIX: Embeddings validator test needs to use the new Vector class. (#1041)
This commit is contained in:
parent
bdb8f1d5e0
commit
ceac6e5efb
|
@ -35,11 +35,8 @@ module DiscourseAi
|
|||
end
|
||||
|
||||
def can_generate_embeddings?(val)
|
||||
DiscourseAi::Embeddings::VectorRepresentations::Base
|
||||
.find_representation(val)
|
||||
.new(DiscourseAi::Embeddings::Strategies::Truncation.new)
|
||||
.vector_from("this is a test")
|
||||
.present?
|
||||
vdef = DiscourseAi::Embeddings::VectorRepresentations::Base.find_representation(val).new
|
||||
DiscourseAi::Embeddings::Vector.new(vdef).vector_from("this is a test").present?
|
||||
end
|
||||
end
|
||||
end
|
||||
|
|
|
@ -0,0 +1,17 @@
|
|||
# frozen_string_literal: true
|
||||
|
||||
require_relative "../support/embeddings_generation_stubs"
|
||||
|
||||
RSpec.describe DiscourseAi::Configuration::EmbeddingsModelValidator do
|
||||
before { SiteSetting.ai_embeddings_discourse_service_api_endpoint = "http://test.com" }
|
||||
|
||||
describe "#can_generate_embeddings?" do
|
||||
it "works" do
|
||||
discourse_model = "all-mpnet-base-v2"
|
||||
|
||||
EmbeddingsGenerationStubs.discourse_service(discourse_model, "this is a test", [1] * 1024)
|
||||
|
||||
expect(subject.can_generate_embeddings?(discourse_model)).to eq(true)
|
||||
end
|
||||
end
|
||||
end
|
Loading…
Reference in New Issue