diff --git a/app/models/bot/alegre.rb b/app/models/bot/alegre.rb index b0e56c501d..d8997e3a3f 100644 --- a/app/models/bot/alegre.rb +++ b/app/models/bot/alegre.rb @@ -12,6 +12,7 @@ class Error < ::StandardError INDIAN_MODEL = 'indian-sbert' FILIPINO_MODEL = 'paraphrase-filipino-mpnet-base-v2' OPENAI_ADA_MODEL = 'openai-text-embedding-ada-002' + PARAPHRASE_MULTILINGUAL_MODEL = 'paraphrase-multilingual-mpnet-base-v2' ELASTICSEARCH_MODEL = 'elasticsearch' DEFAULT_ES_SCORE = 10 @@ -724,7 +725,7 @@ def self.send_post_create_message(source, target, relationship) end def self.relationship_model_not_allowed(relationship_model) - allowed_models = [MEAN_TOKENS_MODEL, INDIAN_MODEL, FILIPINO_MODEL, OPENAI_ADA_MODEL, ELASTICSEARCH_MODEL, 'audio', 'image', 'video'] + allowed_models = [MEAN_TOKENS_MODEL, INDIAN_MODEL, FILIPINO_MODEL, OPENAI_ADA_MODEL, PARAPHRASE_MULTILINGUAL_MODEL, ELASTICSEARCH_MODEL, 'audio', 'image', 'video'] models = relationship_model.split("|").collect{ |m| m.split('/').first } models.length != (allowed_models&models).length end