discourse/discourse-ai
1
0
Fork 0
mirror of https://github.com/discourse/discourse-ai.git synced 2025-06-28 10:32:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

FIX: Embeddings backfill job compat when transitioning models (#1122)

Browse Source 
When you already have embeddings for a model stored and change models,
our backfill script was failing to backfill the newly configured model.

Regression introduced most likely in 1686a8a
This commit is contained in:
Rafael dos Santos Silva 2025-02-12 10:37:45 -03:00 committed by GitHub
parent 708a3bd2b8
commit 77c6543c5d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
app/jobs/scheduled/embeddings_backfill.rb
View File

@ -26,7 +26,9 @@ module Jobs
topics = topics =
Topic Topic
.joins("LEFT JOIN #{table_name} ON #{table_name}.topic_id = topics.id") .joins(
"LEFT JOIN #{table_name} ON #{table_name}.topic_id = topics.id AND #{table_name}.model_id = #{vector_def.id}",
)
.where(archetype: Archetype.default) .where(archetype: Archetype.default)
.where(deleted_at: nil) .where(deleted_at: nil)
.order("topics.bumped_at DESC") .order("topics.bumped_at DESC")
@ -43,7 +45,7 @@ module Jobs
#{table_name}.strategy_version < #{vector_def.strategy_version} #{table_name}.strategy_version < #{vector_def.strategy_version}
SQL SQL
rebaked += populate_topic_embeddings(vector, relation) rebaked += populate_topic_embeddings(vector, relation, force: true)
return if rebaked >= limit return if rebaked >= limit
@ -67,7 +69,9 @@ module Jobs
posts = posts =
Post Post
.joins("LEFT JOIN #{table_name} ON #{table_name}.post_id = posts.id") .joins(
"LEFT JOIN #{table_name} ON #{table_name}.post_id = posts.id AND #{table_name}.model_id = #{vector_def.id}",
)
.where(deleted_at: nil) .where(deleted_at: nil)
.where(post_type: Post.types[:regular]) .where(post_type: Post.types[:regular])