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
|
end
|
||||||
|
|
||||||
def can_generate_embeddings?(val)
|
def can_generate_embeddings?(val)
|
||||||
DiscourseAi::Embeddings::VectorRepresentations::Base
|
vdef = DiscourseAi::Embeddings::VectorRepresentations::Base.find_representation(val).new
|
||||||
.find_representation(val)
|
DiscourseAi::Embeddings::Vector.new(vdef).vector_from("this is a test").present?
|
||||||
.new(DiscourseAi::Embeddings::Strategies::Truncation.new)
|
|
||||||
.vector_from("this is a test")
|
|
||||||
.present?
|
|
||||||
end
|
end
|
||||||
end
|
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