FIX: Embeddings validator test needs to use the new Vector class. (#1041)

This commit is contained in:
Roman Rizzi 2024-12-23 14:19:22 -03:00 committed by GitHub
parent bdb8f1d5e0
commit ceac6e5efb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 19 additions and 5 deletions

View File

@ -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

View File

@ -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