i18n-ai 0.1.0 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 06e2c7c1e6b90add5766612403fefc33653d55b98e2a49b6ba0920a736545788
-  data.tar.gz: 19c739baa0b768bb12496cbb43dc2e58eb78a6afe629ab2db11e53dcfe5e94b0
+  metadata.gz: 91c87df3579aa11f253f670664cbcf4721dc4145cb12740ab05f9f81a8fef913
+  data.tar.gz: 590b0626a6e5dee174d34ddd429154485a7345539bc398815851ca5c4bd50ea3
 SHA512:
-  metadata.gz: a3abb0c81d6716e37df291fa0ef3b2b3853d97696de0247e0677b453cb781d154751b4dd8c4577596ef2e02ef35630a42956189547df3a046e556c85fc062764
-  data.tar.gz: 971b820565e02c742653a423e5eef1f8f6b4111cb3d9d2e535b0752fceb4c1996fc5e09338d866d7a19c2a9e9bd95299c110e40f8f6d0d17a2ffd26ef1c3154f
+  metadata.gz: '08ecd1e198abe8f8363ca6bc227a36a00335f6cc52024b7c5f6ef2013e6a0205a860bdcbc102c34b7953b0708d7f02bc7379946deb7f0b2e702314ffd3f6a143'
+  data.tar.gz: 4e3b1a31c6112457ed53f274d32cf98c21bae53bffcc69d8b91e2ec79403790a4090fbb69a37bc4e7cdd2f30ad99a232777160e29c0944750906c4b1d477483b

data/.rubocop.yml

@@ -6,3 +6,15 @@ Style/StringLiterals:
 
 Style/StringLiteralsInInterpolation:
   EnforcedStyle: double_quotes
+
+Metrics/BlockLength:
+  Exclude:
+    - 'spec/**/*_spec.rb'
+
+Layout/LineLength:
+  Exclude:
+    - 'spec/**/*_spec.rb'
+
+Naming/FileName:
+  Exclude:
+    - 'lib/i18n-ai.rb'

data/README.md

@@ -14,19 +14,47 @@ gem "i18n-ai"
 
 And do `bundle install`.
 
-## Usage
+## Configuration
 
-First you need to set the environment variable `ENV["OPENAI_ACCESS_TOKEN"]`.
+To use I18nAi, you need to set the appropriate environment variables based on the AI service you wish to use.
 
-To configure and enable other locales, create a file `config/initializers/i18n_ai.rb` and add the following:
+### OpenAi
+
+If you are using OpenAI, set `ENV["OPENAI_ACCESS_TOKEN"]`
+
+To configure and enable other locales, create a file `config/initializers/i18n_aii.rb` and add the following:
 
 ```
 # config/initializers/i18n_ai.rb
 I18nAi.configure do |config|
+  config.ai_settings = {
+    provider: "openai",
+    model: "gpt-4o-mini",
+    access_token: ENV["OPENAI_ACCESS_TOKEN"]
+  }
   config.generate_locales = [:es, :it] # add your other supported locales to this array
 end
 ```
 
+### Anthropic's Claude
+
+If you prefer to use Anthropic's Claude, set `ENV[ANTHROPIC_API_KEY]`
+
+To configure and enable other locales, create a file `config/initializers/i18n_ai.rb` and add the following:
+
+```
+I18nAi.configure do |config|
+  config.ai_settings = {
+    provider: "anthropic",
+    model: "claude-3-haiku-20240307",
+    access_token: ENV["ANTHROPIC_ACCESS_TOKEN"]
+  }
+  config.generate_locales = [:es]
+end
+```
+
+## Usage
+
 Every page reload, the gem will check if the `en.yml` file changed and if it did, it will automatically generate the configured locale files.
 
 The gem has been setup to generate a `es.yml` file by default.

data/lib/i18n-ai.rb

@@ -1,19 +1,3 @@
-require_relative "i18n-ai/version"
-require_relative "i18n-ai/railtie"
-require_relative "i18n-ai/configuration"
+# frozen_string_literal: true
 
-module I18nAi
-  class Error < StandardError; end
-
-  class << self
-    attr_writer :configuration
-
-    def configuration
-      @configuration ||= Configuration.new
-    end
-
-    def configure
-      yield(configuration)
-    end
-  end
-end
+require_relative "i18n_ai"

data/lib/i18n_ai/clients/anthropic_client.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require "anthropic"
+require_relative "base_client"
+
+module I18nAi
+  module Clients
+    # The AnthropicClient class is responsible for interacting with the Anthropic API
+    class AnthropicClient < BaseClient
+      def initialize
+        super
+        @client = Anthropic::Client.new(
+          access_token: @config[:access_token]
+        )
+      end
+
+      def chat(locale, text)
+        response = @client.messages(
+          parameters: {
+            model: @config[:model],
+            messages: [{ role: "user", content: content(locale, text) }]
+          }
+        )
+
+        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}"
+      end
+    end
+  end
+end

data/lib/i18n_ai/clients/base_client.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+module I18nAi
+  module Clients
+    # The BaseClient class serves as a base class for all AI client implementations
+    class BaseClient
+      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
+      end
+
+      def parse_response(response)
+        raise NotImplementedError, "Subclasses must implement this method"
+      end
+    end
+  end
+end

data/lib/i18n_ai/clients/open_ai_client.rb

@@ -0,0 +1,45 @@
+# frozen_string_literal: true
+
+require "openai"
+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],
+          log_errors: true
+        )
+      end
+
+      def chat(locale, text)
+        response = @client.chat(
+          parameters: {
+            model: @config[:model],
+            messages: [{ role: "user", content: content(locale, text) }],
+            max_tokens: 5000
+          }
+        )
+
+        parse_response(response)
+      rescue StandardError => e
+        handle_error(e)
+      end
+
+      private
+
+      def parse_response(response)
+        response.dig("choices", 0, "message", "content")
+      rescue TypeError, NoMethodError => e
+        handle_error(e)
+      end
+
+      def handle_error(error)
+        puts "Error: #{error.message}"
+      end
+    end
+  end
+end

data/lib/i18n_ai/configuration.rb

@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+module I18nAi
+  # Configuration class for I18nAi
+  class Configuration
+    attr_accessor :ai_settings, :generate_locales
+
+    def initialize
+      @ai_settings = {}
+      @generate_locales = [:es]
+    end
+  end
+end

data/lib/i18n_ai/railtie.rb

@@ -0,0 +1,121 @@
+# frozen_string_literal: true
+
+require "rails/railtie"
+require "digest"
+
+require_relative "clients/open_ai_client"
+require_relative "clients/anthropic_client"
+
+module I18nAi
+  # The Railtie class provides a way to integrate I18nAi into an application
+  class Railtie < Rails::Railtie
+    # Initialize the I18nAi middleware
+    class I18nAiMiddleware
+      def initialize(app)
+        @app = app
+        @client = configure_client
+        @last_checksum = nil
+      end
+
+      def call(env)
+        locales_file = locate_locales_file
+
+        if file_exists?(locales_file)
+          process_locales_file(locales_file)
+        else
+          log_file_not_found
+        end
+
+        @app.call(env)
+      end
+
+      private
+
+      def locate_locales_file
+        Rails.root.join("config", "locales", "en.yml")
+      end
+
+      def file_exists?(file)
+        File.exist?(file)
+      end
+
+      def process_locales_file(file)
+        current_checksum = calculate_checksum(file)
+        log_checksum(current_checksum)
+
+        first_load = @last_checksum.nil?
+        file_changed = current_checksum != @last_checksum
+        log_generate_status(first_load, file_changed)
+
+        return unless first_load || file_changed
+
+        @last_checksum = current_checksum
+        generate_translations(file)
+      end
+
+      def log_checksum(checksum)
+        puts "==> en.yml checksum: #{checksum}"
+      end
+
+      def log_generate_status(first_load, file_changed)
+        puts "==> en.yml generate: #{first_load || file_changed}"
+      end
+
+      def log_file_not_found
+        puts "en.yml file not found"
+      end
+
+      def configure_client
+        config = I18nAi.configuration.ai_settings
+        case config[:provider]
+        when "anthropic"
+          I18nAi::Clients::AnthropicClient.new
+        when "openai"
+          I18nAi::Clients::OpenAiClient.new
+        else
+          raise "Unknown AI provider: #{config[:provider]}"
+        end
+      end
+
+      def generate_translations(locales_file)
+        locales = load_locales(locales_file)
+        text_to_translate = locales.to_yaml
+        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)
+          save_translated_locales(locale, translated_content) if translated_content
+        end
+      end
+
+      def load_locales(locales_file)
+        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)
+      end
+
+      def calculate_checksum(file_path)
+        Digest::SHA256.file(file_path).hexdigest
+      end
+    end
+
+    initializer "i18n_ai.configure_middleware", before: :build_middleware_stack do |app|
+      app.middleware.use I18nAiMiddleware
+    end
+  end
+end

data/lib/{i18n-ai → i18n_ai}/version.rb

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

data/lib/i18n_ai.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require_relative "i18n_ai/version"
+require_relative "i18n_ai/railtie"
+require_relative "i18n_ai/configuration"
+
+# The I18nAi module provides functionality for integrating AI-based translation services
+module I18nAi
+  class Error < StandardError; end
+
+  class << self
+    attr_writer :configuration
+
+    def configuration
+      @configuration ||= Configuration.new
+    end
+
+    def configure
+      yield(configuration)
+    end
+  end
+end

metadata

@@ -1,15 +1,49 @@
 --- !ruby/object:Gem::Specification
 name: i18n-ai
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.3
 platform: ruby
 authors:
 - William Estoque
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-01 00:00:00.000000000 Z
+date: 2024-08-06 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: railties
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 6.0.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 6.0.0
+    - - "<"
+      - !ruby/object:Gem::Version
+        version: '8'
+- !ruby/object:Gem::Dependency
+  name: anthropic
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: ruby-openai
   requirement: !ruby/object:Gem::Requirement
@@ -38,10 +72,13 @@ files:
 - README.md
 - Rakefile
 - lib/i18n-ai.rb
-- lib/i18n-ai/configuration.rb
-- lib/i18n-ai/railtie.rb
-- lib/i18n-ai/version.rb
-- sig/i18n/ai.rbs
+- lib/i18n_ai.rb
+- lib/i18n_ai/clients/anthropic_client.rb
+- lib/i18n_ai/clients/base_client.rb
+- lib/i18n_ai/clients/open_ai_client.rb
+- lib/i18n_ai/configuration.rb
+- lib/i18n_ai/railtie.rb
+- lib/i18n_ai/version.rb
 homepage: https://github.com/narralabs/i18n-ai
 licenses:
 - MIT

data/lib/i18n-ai/configuration.rb

@@ -1,9 +0,0 @@
-module I18nAi
-  class Configuration
-    attr_accessor :generate_locales
-
-    def initialize
-      @generate_locales = [:es]
-    end
-  end
-end

data/lib/i18n-ai/railtie.rb

@@ -1,75 +0,0 @@
-require "rails/railtie"
-require "openai"
-require "digest"
-
-module I18nAi
-  class Railtie < Rails::Railtie
-    class I18nAiMiddleware
-      def initialize(app)
-        @app = app
-        @client = OpenAI::Client.new(
-          access_token: ENV.fetch("OPENAI_ACCESS_TOKEN"),
-          log_errors: true
-        )
-        @last_checksum = nil
-      end
-
-      def call(env)
-        locales_file = Rails.root.join("config", "locales", "en.yml")
-
-        if File.exist?(locales_file)
-          current_checksum = calculate_checksum(locales_file)
-          puts "==> en.yml checksum: #{current_checksum}"
-          first_load = @last_checksum.nil?
-          file_changed = current_checksum != @last_checksum
-          puts "==> en.yml generate: #{first_load || file_changed}"
-
-          if first_load || file_changed
-            @last_checksum = current_checksum
-            generate_translations(locales_file)
-          end
-        else
-          puts "en.yml file not found"
-        end
-
-        @app.call(env)
-      end
-
-      private
-
-      def generate_translations(locales_file)
-        locales = YAML.load_file(locales_file)
-        text_to_translate = locales.to_yaml
-        generate_locales = I18nAi.configuration.generate_locales
-
-        generate_locales.each do |locale|
-          # Make a request to OpenAI to translate the locales to the specified locale
-          response = @client.chat(
-            parameters: {
-              model: "gpt-4o-mini",
-              messages: [{ role: "user",
-                           content: "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}" }],
-              max_tokens: 5000
-            }
-          )
-
-          translated_text = response["choices"][0]["message"]["content"]
-          match_data = translated_text.match(/```yaml(.*?)```/m)
-          str = match_data ? match_data[1].strip : nil
-
-          # Save the response to <locale>.yml
-          locales_file = Rails.root.join("config", "locales", "#{locale}.yml")
-          File.write(locales_file, str)
-        end
-      end
-
-      def calculate_checksum(file_path)
-        Digest::SHA256.file(file_path).hexdigest
-      end
-    end
-
-    initializer "i18n_ai.configure_middleware", before: :build_middleware_stack do |app|
-      app.middleware.use I18nAiMiddleware
-    end
-  end
-end

data/sig/i18n/ai.rbs

@@ -1,6 +0,0 @@
-module I18n
-  module Ai
-    VERSION: String
-    # See the writing guide of rbs: https://github.com/ruby/rbs#guides
-  end
-end