fix: Disable predictions if model is absent (#7520)

2023-07-13 14:40:16 -07:00 · 2023-07-13 14:40:16 -07:00 · 7e415fbee3
commit 7e415fbee3
parent 50a927bac2
1 changed files with 2 additions and 0 deletions
--- a/enterprise/app/jobs/enterprise/sentiment_analysis_job.rb
+++ b/enterprise/app/jobs/enterprise/sentiment_analysis_job.rb
@ -17,6 +17,8 @@ class Enterprise::SentimentAnalysisJob < ApplicationJob
    # While gathering the maningfull node the Array/tensor index is going out of bound

    text = message.content&.truncate(2900)
+    return if model.blank?
+
    sentiment = model.predict(text)
    message.sentiment = sentiment.merge(value: label_val(sentiment))