immosquare-translate 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2cf94a139c881e315445fb0df7f24699166c1833b2b5f0746b86c91f5025ba5c
-  data.tar.gz: b518012e6deafacb308f178c1c21ab0191c2e491d47f0bda97e976605155a37f
+  metadata.gz: 326589c34a947bb34891fdef41683bf0a0d2bc5dd649c0598d63d7591eac1341
+  data.tar.gz: e28ead0b001c5c3ffea1a322e35ba3293e7c916dbe71a8c677d9484efb86a5c8
 SHA512:
-  metadata.gz: 1f96fdafd54a457e86ee7d8b0475807e4f8308ff5e8080a08ca6df369a1c4d515dc45ec90a240d4a4143a531bdaaaf51881532313252054e133b5fd0d33281ae
-  data.tar.gz: 715d03787f074bff87e69039ddeb466f9cbe449e1e57021ea9a3587fc272cc71d8f431e6f74aa75c33cb9accab2976e559d4ad885989b6d6c2ceab21637208e3
+  metadata.gz: 8d74b478dbaa133200e77b2638e38bf986204ecfc5946b2653cb7687b40a8bfbfb28e6e163c772aac679f1c599169bc44c66c14cdb224be8fba830353ad21f69
+  data.tar.gz: bbf947a36c077cfc1b419936cac7a88e4cee75df414c11133f85172aa431cff10e95b5336e202171a18b34036a9964339be5515096ad554fa14a5d271b20e586

data/lib/immosquare-translate/translator.rb

@@ -20,7 +20,7 @@ module ImmosquareTranslate
           model         = OPEN_AI_MODELS.find {|m| m[:name] == model_name }
           model         = OPEN_AI_MODELS.find {|m| m[:name] == "gpt-4-0125-preview" } if model.nil?
           from_iso      = ISO_639.find_by_code(from).english_name.split(";").first
-          to_iso        = to.map {|l| ISO_639.find_by_code(l).english_name.split(";").first }
+          to_iso        = to.map {|iso| [iso, ISO_639.find_by_code(iso).english_name.split(";").first] }
           headers       = {
             "Content-Type"  => "application/json",
             "Authorization" => "Bearer #{ImmosquareTranslate.configuration.openai_api_key}"
@@ -28,19 +28,21 @@ module ImmosquareTranslate
 
           prompt_system = "As a sophisticated translation AI, your role is to translate sentences from a specified source language to multiple target languages. " \
                           "It is imperative that you return the translations in a single, pure JSON string format. Use ISO 639-1 language codes for specifying languages. " \
-                          "if string is html, you should return the translated html." \
+                          "If string is html, you should return the translated html." \
                           "Ensure that the output does not include markdown (```json) or any other formatting characters. Adhere to the JSON structure meticulously."
 
 
-          prompt = "Translate the following sentences from '#{from_iso}' into the languages #{to_iso.join(", ")}, and format the output as a single, pure JSON string. " \
-                   "Follow the structure: {\"datas\":[{\"en\":\"English Translation\",\"es\":\"Spanish Translation\",\"it\":\"Italian Translation\"}]}, using the correct ISO 639-1 language codes for each translation. " \
-                   "Your response should strictly conform to this JSON structure without any additional characters or formatting. Sentences to translate are:"
+          prompt = "Translate the #{text.size} following  #{text.size == 1 ? "sentence" : "sentences"} from #{from_iso} (with ISO 639-1 code : #{from}) into the languages #{to_iso.map {|iso, l| "#{l} (with ISO 639-1 code : #{iso})" }.join(", ")}.\n" \
+                   "Format the output as a single pure JSON string.\n" \
+                   "Follow this array structure: {\"datas\":[{\"locale_iso\":\"Translated Text\"}]}, using the correct ISO 639-1 language codes for each translation. " \
+                   "Your response should strictly conform to this JSON structure without any additional characters or formatting.\nSentences to translate are:"
 
           text.each_with_index do |sentence, index|
             prompt += "\n#{index + 1}: #{sentence}"
           end
 
 
+
           body = {
             :model       => model[:name],
             :messages    => [
@@ -74,8 +76,14 @@ module ImmosquareTranslate
           puts("Estimate price => #{input_price.round(3)} + #{output_price.round(3)} = #{price.round(3)} USD")
 
 
+          ##============================================================##
+          ## On s'assure de ne renvoyer que les locales demandées
+          ## car l'API peut renvoyer des locales non demandées...
+          ##============================================================##
           p = JSON.parse(choice["message"]["content"])
-          p["datas"]
+          p["datas"].map! do |my_hash|
+            my_hash.select! {|iso, _value| to.include?(iso.to_s) }
+          end
         rescue StandardError => e
           puts(e.message)
           puts(e.backtrace)

data/lib/immosquare-translate/version.rb

@@ -1,3 +1,3 @@
 module ImmosquareTranslate
-  VERSION = "0.1.3".freeze
+  VERSION = "0.1.4".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: immosquare-translate
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - IMMO SQUARE
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-31 00:00:00.000000000 Z
+date: 2024-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty