Skip to content
This repository was archived by the owner on Jul 22, 2025. It is now read-only.

Commit 47ecf86

Browse files
authored
FIX: embedding validation (#1043)
1 parent 792df58 commit 47ecf86

File tree

2 files changed

+18
-2
lines changed

2 files changed

+18
-2
lines changed

lib/configuration/embeddings_module_validator.rb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,9 @@ def error_message
4242
def can_generate_embeddings?(val)
4343
DiscourseAi::Embeddings::VectorRepresentations::Base
4444
.find_representation(val)
45-
.new(DiscourseAi::Embeddings::Strategies::Truncation.new)
46-
.vector_from("this is a test")
45+
.new
46+
.inference_client
47+
.perform!("this is a test")
4748
.present?
4849
end
4950
end
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
# frozen_string_literal: true
2+
3+
RSpec.describe DiscourseAi::Configuration::EmbeddingsModuleValidator do
4+
let(:validator) { described_class.new }
5+
6+
describe "#can_generate_embeddings?" do
7+
it "returns true if embeddings can be generated" do
8+
stub_request(
9+
:post,
10+
"https://generativelanguage.googleapis.com/v1beta/models/embedding-001:embedContent?key=",
11+
).to_return(status: 200, body: { embedding: { values: [1, 2, 3] } }.to_json)
12+
expect(validator.can_generate_embeddings?("gemini")).to eq(true)
13+
end
14+
end
15+
end

0 commit comments

Comments
 (0)