Discource-C
/
discourse-ai
mirror of https://github.com/discourse/discourse-ai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

FIX: Set ivfflat.probes using topic count, not post count (#421) 

Fixes a regression from 140359c which caused we to set this globally based on post count, rendering the cost of an index scan on the topics table too high and making the planner, correctly, not use the index anymore.

Hopefully https://github.com/pgvector/pgvector/issues/235 lands soon.
This commit is contained in:
Rafael dos Santos Silva 2024-01-12 11:20:23 -03:00 committed by GitHub
parent 3be76ebd7a
commit 705ef986b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/embeddings/vector_representations/base.rb
View File

@ -112,7 +112,13 @@ module DiscourseAi
DB.exec("RESET maintenance_work_mem;") DB.exec("RESET maintenance_work_mem;")
database = DB.query_single("SELECT current_database();").first database = DB.query_single("SELECT current_database();").first
DB.exec("ALTER DATABASE #{database} SET ivfflat.probes = #{probes};")
# This is a global setting, if we set it based on post count
# we will be unable to use the index for topics
# Hopefully https://github.com/pgvector/pgvector/issues/235 will make this better
if table_name == topic_table_name
DB.exec("ALTER DATABASE #{database} SET ivfflat.probes = #{probes};")
end
end end
def vector_from(text) def vector_from(text)