From 251628bfa168a5fc62c925efc2697ecac41e5232 Mon Sep 17 00:00:00 2001 From: Roman Rizzi Date: Tue, 26 Nov 2024 18:12:03 -0300 Subject: [PATCH] FIX: Shutdown embeddings thread pool after processing (#961) --- lib/embeddings/vector_representations/base.rb | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lib/embeddings/vector_representations/base.rb b/lib/embeddings/vector_representations/base.rb index 7055ea23..5f113002 100644 --- a/lib/embeddings/vector_representations/base.rb +++ b/lib/embeddings/vector_representations/base.rb @@ -81,6 +81,9 @@ module DiscourseAi .zip(*promised_embeddings) .value! .each { |e| save_to_db(e[:target], e[:embedding], e[:digest]) } + + pool.shutdown + pool.wait_for_termination end def generate_representation_from(target, persist: true)