Discourse AI - 埋め込み

最近のバージョンで rake ai:embeddings:backfill が失敗する小さなバグを見つけました。

root@nbg-webxj:/var/www/discourse# rake ai:embeddings:backfill
rake aborted!
NameError: uninitialized constant Parallel (NameError)

  Parallel.each(topics.all, in_processes: args[:concurrency].to_i, progress: "Topics") do |t|
  ^^^^^^
/var/www/discourse/plugins/discourse-ai/lib/tasks/modules/embeddings/database.rake:27:in `block in <main>'
/usr/local/bin/bundle:25:in `load'
/usr/local/bin/bundle:25:in `<main>'
Tasks: TOP => ai:embeddings:backfill
(See full trace by running task with --trace)

原因は、parallel gem がこのプラグインにも、Discourse コアにもインストールされていないことだと推測しています(if ENV["IMPORT"] == "1" ブロックで gem "parallel", require: false を見つけただけです)。

rake ai:embeddings:backfill を実行するために ruby-progressbar gem も必要であることがわかりました。

GitHub に簡単な PR を作成しました。

「いいね!」 2