discourse-ai/spec/system/embeddings/ai_embedding_definition_spe...

85 lines
3.2 KiB
Ruby

# frozen_string_literal: true
RSpec.describe "Managing Embeddings configurations", type: :system, js: true do
fab!(:admin)
let(:page_header) { PageObjects::Components::DPageHeader.new }
let(:form) { PageObjects::Components::FormKit.new("form") }
before { sign_in(admin) }
it "correctly sets defaults" do
preset = "text-embedding-3-small"
api_key = "abcd"
visit "/admin/plugins/discourse-ai/ai-embeddings"
find(".ai-embeddings-list-editor__new-button").click()
find("[data-preset-id='text-embedding-3-small'] button").click()
form.field("api_key").fill_in(api_key)
form.submit
expect(page).to have_current_path("/admin/plugins/discourse-ai/ai-embeddings")
embedding_def = EmbeddingDefinition.order(:id).last
expect(embedding_def.api_key).to eq(api_key)
preset = EmbeddingDefinition.presets.find { |p| p[:preset_id] == preset }
expect(embedding_def.display_name).to eq(preset[:display_name])
expect(embedding_def.url).to eq(preset[:url])
expect(embedding_def.tokenizer_class).to eq(preset[:tokenizer_class])
expect(embedding_def.dimensions).to eq(preset[:dimensions])
expect(embedding_def.max_sequence_length).to eq(preset[:max_sequence_length])
expect(embedding_def.pg_function).to eq(preset[:pg_function])
expect(embedding_def.provider).to eq(preset[:provider])
expect(embedding_def.provider_params.symbolize_keys).to eq(preset[:provider_params])
end
it "supports manual config" do
api_key = "abcd"
visit "/admin/plugins/discourse-ai/ai-embeddings"
find(".ai-embeddings-list-editor__new-button").click()
find("[data-preset-id='manual'] button").click()
form.field("display_name").fill_in("text-embedding-3-small")
form.field("provider").select(EmbeddingDefinition::OPEN_AI)
form.field("url").fill_in("https://api.openai.com/v1/embeddings")
form.field("api_key").fill_in(api_key)
form.field("tokenizer_class").select("DiscourseAi::Tokenizer::OpenAiTokenizer")
embed_prefix = "On creation:"
search_prefix = "On search:"
form.field("embed_prompt").fill_in(embed_prefix)
form.field("search_prompt").fill_in(search_prefix)
form.field("dimensions").fill_in(1536)
form.field("max_sequence_length").fill_in(8191)
form.field("pg_function").select("<=>")
form.field("provider_params.model_name").fill_in("text-embedding-3-small")
form.submit
expect(page).to have_current_path("/admin/plugins/discourse-ai/ai-embeddings")
embedding_def = EmbeddingDefinition.order(:id).last
expect(embedding_def.api_key).to eq(api_key)
preset = EmbeddingDefinition.presets.find { |p| p[:preset_id] == "text-embedding-3-small" }
expect(embedding_def.display_name).to eq(preset[:display_name])
expect(embedding_def.url).to eq(preset[:url])
expect(embedding_def.tokenizer_class).to eq(preset[:tokenizer_class])
expect(embedding_def.dimensions).to eq(preset[:dimensions])
expect(embedding_def.max_sequence_length).to eq(preset[:max_sequence_length])
expect(embedding_def.pg_function).to eq(preset[:pg_function])
expect(embedding_def.provider).to eq(preset[:provider])
expect(embedding_def.embed_prompt).to eq(embed_prefix)
expect(embedding_def.search_prompt).to eq(search_prefix)
end
end