mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-04 00:10:07 +00:00
f5cf1019fb
* Use AR model for embeddings features * endpoints * Embeddings CRUD UI * Add presets. Hide a couple more settings * system specs * Seed embedding definition from old settings * Generate search bit index on the fly. cleanup orphaned data * support for seeded models * Fix run test for new embedding * fix selected model not set correctly
46 lines
1.4 KiB
Ruby
46 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
RSpec.describe "AI Composer helper", type: :system, js: true do
|
|
let(:search_page) { PageObjects::Pages::Search.new }
|
|
let(:query) { "apple_pie" }
|
|
let(:hypothetical_post) { "This is an hypothetical post generated from the keyword apple_pie" }
|
|
|
|
fab!(:user) { Fabricate(:admin) }
|
|
fab!(:topic)
|
|
fab!(:post) { Fabricate(:post, topic: topic, raw: "Apple pie is a delicious dessert to eat") }
|
|
|
|
before do
|
|
prompt = DiscourseAi::Embeddings::HydeGenerators::OpenAi.new.prompt(query)
|
|
OpenAiCompletionsInferenceStubs.stub_response(
|
|
prompt,
|
|
hypothetical_post,
|
|
req_opts: {
|
|
max_tokens: 400,
|
|
},
|
|
)
|
|
|
|
hyde_embedding = [0.049382, 0.9999]
|
|
EmbeddingsGenerationStubs.hugging_face_service(hypothetical_post, hyde_embedding)
|
|
|
|
SearchIndexer.enable
|
|
SearchIndexer.index(topic, force: true)
|
|
SiteSetting.ai_embeddings_semantic_search_enabled = true
|
|
sign_in(user)
|
|
end
|
|
|
|
after do
|
|
described_class.clear_cache_for(query)
|
|
SearchIndexer.disable
|
|
end
|
|
|
|
describe "when performing a search in the full page search page" do
|
|
skip "TODO: Implement test after doing LLM abstraction" do
|
|
it "performs AI search in the background and hides results by default" do
|
|
visit("/search?expanded=true")
|
|
search_page.type_in_search("apple pie")
|
|
search_page.click_search_button
|
|
end
|
|
end
|
|
end
|
|
end
|