discourse-ai/spec/jobs/regular/manage_embedding_def_search_index_spec.rb

# frozen_string_literal: true

RSpec.describe Jobs::ManageEmbeddingDefSearchIndex do
  fab!(:embedding_definition)

  describe "#execute" do
    context "when there is no embedding def" do
      it "does nothing" do
        invalid_id = 999_999_999

        subject.execute(id: invalid_id)

        expect(
          DiscourseAi::Embeddings::Schema.correctly_indexed?(
            EmbeddingDefinition.new(id: invalid_id),
          ),
        ).to eq(false)
      end
    end

    context "when the embedding def is fresh" do
      it "creates the indexes" do
        subject.execute(id: embedding_definition.id)

        expect(DiscourseAi::Embeddings::Schema.correctly_indexed?(embedding_definition)).to eq(true)
      end

      it "creates them only once" do
        subject.execute(id: embedding_definition.id)
        subject.execute(id: embedding_definition.id)

        expect(DiscourseAi::Embeddings::Schema.correctly_indexed?(embedding_definition)).to eq(true)
      end

      context "when one of the idxs is missing" do
        it "automatically recovers by creating it" do
          DB.exec <<~SQL
            CREATE INDEX IF NOT EXISTS ai_topics_embeddings_#{embedding_definition.id}_1_search_bit ON ai_topics_embeddings
            USING hnsw ((binary_quantize(embeddings)::bit(#{embedding_definition.dimensions})) bit_hamming_ops)
            WHERE model_id = #{embedding_definition.id} AND strategy_id = 1;
          SQL

          expect(DiscourseAi::Embeddings::Schema.correctly_indexed?(embedding_definition)).to eq(
            false,
          )

          subject.execute(id: embedding_definition.id)

          expect(DiscourseAi::Embeddings::Schema.correctly_indexed?(embedding_definition)).to eq(
            true,
          )
        end
      end
    end
  end
end