From e9ae28f773a5d7464d1114d70d94a5b972111033 Mon Sep 17 00:00:00 2001 From: Rafael dos Santos Silva Date: Wed, 17 May 2023 12:10:10 -0300 Subject: [PATCH] FIX: Non instructor OSS embeddings was broken (#65) --- lib/modules/embeddings/model.rb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/modules/embeddings/model.rb b/lib/modules/embeddings/model.rb index e939e895..93b42126 100644 --- a/lib/modules/embeddings/model.rb +++ b/lib/modules/embeddings/model.rb @@ -70,7 +70,7 @@ module DiscourseAi truncated_input = DiscourseAi::Tokenizer::BertTokenizer.truncate(input, max_sequence_lenght) if name.start_with?("instructor") - instructed_input = [ + truncated_input = [ [SiteSetting.ai_embeddings_semantic_related_instruction, truncated_input], ] end @@ -78,7 +78,7 @@ module DiscourseAi DiscourseAi::Inference::DiscourseClassifier.perform!( "#{SiteSetting.ai_embeddings_discourse_service_api_endpoint}/api/v1/classify", name.to_s, - instructed_input, + truncated_input, SiteSetting.ai_embeddings_discourse_service_api_key, ) end