i18n-ai 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 91c87df3579aa11f253f670664cbcf4721dc4145cb12740ab05f9f81a8fef913
-  data.tar.gz: 590b0626a6e5dee174d34ddd429154485a7345539bc398815851ca5c4bd50ea3
+  metadata.gz: f98f84d08a1cbe5960d00ca2b76c1d1ce7adc930431247971694b4a93e497367
+  data.tar.gz: b42bc703a7bc1880a3503a1470b9f51923ce925ace01458f590e836313f80478
 SHA512:
-  metadata.gz: '08ecd1e198abe8f8363ca6bc227a36a00335f6cc52024b7c5f6ef2013e6a0205a860bdcbc102c34b7953b0708d7f02bc7379946deb7f0b2e702314ffd3f6a143'
-  data.tar.gz: 4e3b1a31c6112457ed53f274d32cf98c21bae53bffcc69d8b91e2ec79403790a4090fbb69a37bc4e7cdd2f30ad99a232777160e29c0944750906c4b1d477483b
+  metadata.gz: 2500f28ee2e4185e89b433a1265b863647a7fbbdd35c1288ecd33e939261cec5ff328875b96ddb531506462914c77bd7cd8d3db6145ebc324fbc80efa321f1a9
+  data.tar.gz: e5f6d3da9c17b2fd58d578aa3bd2037e213d9cf8a175df15f9d791bd24ae96e0be5eeb7432e1401d08fa94863dd6be2c94077e133e04375fd33a92fddb6b9ac6

data/.rubocop.yml

@@ -18,3 +18,14 @@ Layout/LineLength:
 Naming/FileName:
   Exclude:
     - 'lib/i18n-ai.rb'
+
+Metrics/BlockLength:
+  Exclude:
+    - 'i18n-ai.gemspec'
+
+Style/Documentation:
+  Enabled: false
+
+Style/BlockDelimiters:
+  Exclude:
+    - 'spec/**/*_spec.rb'

data/lib/i18n_ai/clients/anthropic_client.rb

@@ -10,33 +10,29 @@ module I18nAi
       def initialize
         super
         @client = Anthropic::Client.new(
-          access_token: @config[:access_token]
+          access_token: config[:access_token]
         )
       end
 
       def chat(locale, text)
         response = @client.messages(
           parameters: {
-            model: @config[:model],
-            messages: [{ role: "user", content: content(locale, text) }]
+            model: config[:model],
+            messages: [{ role: "user", content: chat_prompt(locale, text) }],
+            max_tokens: 4096
           }
         )
-
         parse_response(response)
-      rescue StandardError => e
-        handle_error(e)
       end
 
       private
 
       def parse_response(response)
         response.dig("content", 0, "text")
-      rescue TypeError, NoMethodError => e
-        handle_error(e)
       end
 
-      def handle_error(error)
-        puts "Error: #{error.message}"
+      def extract_translated_content(chat_content)
+        chat_content
       end
     end
   end

data/lib/i18n_ai/clients/base_client.rb

@@ -4,19 +4,50 @@ module I18nAi
   module Clients
     # The BaseClient class serves as a base class for all AI client implementations
     class BaseClient
+      attr_reader :config
+
       def initialize
         @config = I18nAi.configuration.ai_settings
       end
 
-      def content(locale, text_to_translate)
-        # rubocop:disable Layout/LineLength
-        "Translate the following YAML content to #{locale.to_s.upcase} and make sure to retain the keys in english except the first key which is the 2 letter language code:\n\n#{text_to_translate}"
-        # rubocop:enable Layout/LineLength
+      def translate_content(locale, content)
+        chat_content = chat(locale, content)
+        extract_translated_content(chat_content)
+      end
+
+      private
+
+      def chat_prompt(locale, text_to_translate)
+        <<~PROMPT
+          Translate the following YAML content to the language of the country using the ISO 639 language code
+          of #{locale}. Make sure to retain the keys in english except the first key which is the 2 letter language code:
+
+          """"
+          #{text_to_translate}"
+          """"
+
+          Return only the YAML content without explanation.
+
+          Example Return YAML:
+
+          """
+          {{ISO_639 language code}}:
+            key_1: "value1"
+          """
+        PROMPT
+      end
+
+      def chat(locale, text)
+        raise NotImplementedError, "Subclasses must implement this method"
       end
 
       def parse_response(response)
         raise NotImplementedError, "Subclasses must implement this method"
       end
+
+      def extract_translated_content(chat_content)
+        raise NotImplementedError, "Subclasses must implement this method"
+      end
     end
   end
 end

data/lib/i18n_ai/clients/open_ai_client.rb

@@ -5,12 +5,11 @@ require_relative "base_client"
 
 module I18nAi
   module Clients
-    # The AnthropicClient class is responsible for interacting with the OpenAI API
     class OpenAiClient < BaseClient
       def initialize
         super
         @client = OpenAI::Client.new(
-          access_token: @config[:access_token],
+          access_token: config[:access_token],
           log_errors: true
         )
       end
@@ -19,14 +18,11 @@ module I18nAi
         response = @client.chat(
           parameters: {
             model: @config[:model],
-            messages: [{ role: "user", content: content(locale, text) }],
-            max_tokens: 5000
+            messages: [{ role: "user", content: chat_prompt(locale, text) }],
+            max_tokens: 4096
           }
         )
-
         parse_response(response)
-      rescue StandardError => e
-        handle_error(e)
       end
 
       private
@@ -40,6 +36,11 @@ module I18nAi
       def handle_error(error)
         puts "Error: #{error.message}"
       end
+
+      def extract_translated_content(chat_content)
+        match_data = chat_content.match(/```yaml(.*?)```/m)
+        match_data ? match_data[1].strip : nil
+      end
     end
   end
 end

data/lib/i18n_ai/railtie.rb

@@ -11,6 +11,8 @@ module I18nAi
   class Railtie < Rails::Railtie
     # Initialize the I18nAi middleware
     class I18nAiMiddleware
+      attr_reader :client
+
       def initialize(app)
         @app = app
         @client = configure_client
@@ -83,10 +85,7 @@ module I18nAi
         generate_locales = I18nAi.configuration.generate_locales
 
         generate_locales.each do |locale|
-          response = translate_locales(locale, text_to_translate)
-          next unless response
-
-          translated_content = extract_translated_content(response)
+          translated_content = client.translate_content(locale, text_to_translate)
           save_translated_locales(locale, translated_content) if translated_content
         end
       end
@@ -95,15 +94,6 @@ module I18nAi
         YAML.load_file(locales_file)
       end
 
-      def translate_locales(locale, text_to_translate)
-        @client.chat(locale, text_to_translate)
-      end
-
-      def extract_translated_content(response)
-        match_data = response.match(/```yaml(.*?)```/m)
-        match_data ? match_data[1].strip : nil
-      end
-
       def save_translated_locales(locale, translated_content)
         locales_file = Rails.root.join("config", "locales", "#{locale}.yml")
         File.write(locales_file, translated_content)

data/lib/i18n_ai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module I18nAi
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: i18n-ai
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - William Estoque
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-06 00:00:00.000000000 Z
+date: 2024-08-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: railties
@@ -58,6 +58,34 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: vcr
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description:
 email:
 - william.estoque@gmail.com