From f572a7cc2c365cac1c1edf5fee2cf7ffca21d52c Mon Sep 17 00:00:00 2001 From: Rafael dos Santos Silva Date: Wed, 22 Feb 2023 20:48:51 -0300 Subject: [PATCH] fix lint --- lib/modules/sentiment/post_classifier.rb | 2 +- lib/modules/toxicity/classifier.rb | 2 +- lib/shared/inference_manager.rb | 17 +++-------------- 3 files changed, 5 insertions(+), 16 deletions(-) diff --git a/lib/modules/sentiment/post_classifier.rb b/lib/modules/sentiment/post_classifier.rb index 0b2638f0..a773da91 100644 --- a/lib/modules/sentiment/post_classifier.rb +++ b/lib/modules/sentiment/post_classifier.rb @@ -25,7 +25,7 @@ module ::DiscourseAI "#{SiteSetting.ai_sentiment_inference_service_api_endpoint}/api/v1/classify", model, content, - SiteSetting.ai_sentiment_inference_service_api_key + SiteSetting.ai_sentiment_inference_service_api_key, ) store_classification(model, classification) diff --git a/lib/modules/toxicity/classifier.rb b/lib/modules/toxicity/classifier.rb index ee7a951e..ba8ca7e0 100644 --- a/lib/modules/toxicity/classifier.rb +++ b/lib/modules/toxicity/classifier.rb @@ -26,7 +26,7 @@ module ::DiscourseAI "#{SiteSetting.ai_toxicity_inference_service_api_endpoint}/api/v1/classify", SiteSetting.ai_toxicity_inference_service_api_model, content, - SiteSetting.ai_toxicity_inference_service_api_key + SiteSetting.ai_toxicity_inference_service_api_key, ) store_classification consider_flagging diff --git a/lib/shared/inference_manager.rb b/lib/shared/inference_manager.rb index 03f060f8..554cd941 100644 --- a/lib/shared/inference_manager.rb +++ b/lib/shared/inference_manager.rb @@ -3,22 +3,11 @@ module ::DiscourseAI class InferenceManager def self.perform!(endpoint, model, content, api_key) + headers = { "Referer" => Discourse.base_url, "Content-Type" => "application/json" } - headers = { - "Referer" => Discourse.base_url, - "Content-Type" => "application/json", - } + headers["X-API-KEY"] = api_key if api_key.present? - if api_key.present? - headers["X-API-KEY"] = api_key - end - - response = - Faraday.post( - endpoint, - { model: model, content: content }.to_json, - headers, - ) + response = Faraday.post(endpoint, { model: model, content: content }.to_json, headers) raise Net::HTTPBadResponse unless response.status == 200