i18n-ai 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 06e2c7c1e6b90add5766612403fefc33653d55b98e2a49b6ba0920a736545788
-  data.tar.gz: 19c739baa0b768bb12496cbb43dc2e58eb78a6afe629ab2db11e53dcfe5e94b0
+  metadata.gz: a4b8bb0bd9755345297a83d3c236a95fa9d68c1b432743f08cb48b9bc38fa03b
+  data.tar.gz: 62f2a9fc7281367224041c3222191e345c15848757ee6e6cfc4905de2dc4156f
 SHA512:
-  metadata.gz: a3abb0c81d6716e37df291fa0ef3b2b3853d97696de0247e0677b453cb781d154751b4dd8c4577596ef2e02ef35630a42956189547df3a046e556c85fc062764
-  data.tar.gz: 971b820565e02c742653a423e5eef1f8f6b4111cb3d9d2e535b0752fceb4c1996fc5e09338d866d7a19c2a9e9bd95299c110e40f8f6d0d17a2ffd26ef1c3154f
+  metadata.gz: 2a72a92c4f61419e2a76e9ad7df7086168128d24c5d6eafe9be395be4db948e6e6d06937d2db8d11356a85f295c53cf46952fb09eb1c7171eacfe4aa4e009d89
+  data.tar.gz: 45997aa4a669cc795ee91841a6de94974ba782387ad1e55384700d3a407749b097079abe77aec6c2f505a42bf31adfdb01a145e8cc24dab52b3a7089b860044c

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/clients/anthropic_client.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require "anthropic"
+require_relative "base_client"
+
+module I18nAi
+  module Clients
+    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,19 @@
+# frozen_string_literal: true
+
+module I18nAi
+  module Clients
+    class BaseClient
+      def initialize
+        @config = I18nAi.configuration.ai_settings
+      end
+
+      def content(locale, text_to_translate)
+        "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}"
+      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,44 @@
+# frozen_string_literal: true
+
+require "openai"
+require_relative "base_client"
+
+module I18nAi
+  module Clients
+    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

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

data/lib/i18n-ai/railtie.rb

@@ -1,16 +1,17 @@
+# frozen_string_literal: true
+
 require "rails/railtie"
-require "openai"
 require "digest"
 
+require_relative "clients/open_ai_client"
+require_relative "clients/anthropic_client"
+
 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
-        )
+        @client = configure_client
         @last_checksum = nil
       end
 
@@ -37,6 +38,18 @@ module I18nAi
 
       private
 
+      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 = YAML.load_file(locales_file)
         text_to_translate = locales.to_yaml
@@ -45,16 +58,12 @@ module I18nAi
         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
-            }
+            locale, text_to_translate
           )
 
-          translated_text = response["choices"][0]["message"]["content"]
-          match_data = translated_text.match(/```yaml(.*?)```/m)
+          next unless response
+
+          match_data = response.match(/```yaml(.*?)```/m)
           str = match_data ? match_data[1].strip : nil
 
           # Save the response to <locale>.yml

data/lib/i18n-ai/version.rb

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

data/lib/i18n-ai.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require_relative "i18n-ai/version"
 require_relative "i18n-ai/railtie"
 require_relative "i18n-ai/configuration"

metadata

@@ -1,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: i18n-ai
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - William Estoque
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-01 00:00:00.000000000 Z
+date: 2024-08-05 00:00:00.000000000 Z
 dependencies:
+- !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,6 +52,9 @@ files:
 - README.md
 - Rakefile
 - 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