gai18n 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 80ea1d6a4a1a9340d8149b99546887182e9e8c8111e5d97aabc3443e495e7518
-  data.tar.gz: 87973911b51f123a2a49b1729dc97385bffeee649663dd5fda836f2537f64b10
+  metadata.gz: 872e8312f105964dda9b785fbecb875fc66c64e87c51c527b3b44bb6369642f9
+  data.tar.gz: d7647cdf55655c08aafe5210fca2679bcc5125001ace9dfab573b57f00e5fb98
 SHA512:
-  metadata.gz: bcde0839b720b180d9e97cddccc3a8de21e05a2cd2094054f6015f87c87248bcd61a8e7fbb1c4502e7678b7cc1c53ea835cf01b9f24a31c3afc0b93a8e0daf03
-  data.tar.gz: 713b081ea1ac3d8b1b2d20be55a7e5f6f307fc81a51ac9411de7680def97eee5a829f0626a2d50632f21569583c3547f5d42a73bb3cde26c22460887a90000e4
+  metadata.gz: 71981288cb3de0ab007efa0de5026182e1f1e836aab007e38d1089764c22ef4eeeeb05a0e06e6bdeaef55b4a03d59d633646b0ab518f83cff472894e9fc885c4
+  data.tar.gz: 49fb2d2d9f3f49470086ffce89427833f28bb6a47395be05b13a616379d0cbbd4c109b3f8133b3c089f62b5bc999266a7e244eaa0e8a32377ffbe5da1a80075f

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,133 @@
+
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+We as members, contributors, and leaders pledge to make participation in our
+community a harassment-free experience for everyone, regardless of age, body
+size, visible or invisible disability, ethnicity, sex characteristics, gender
+identity and expression, level of experience, education, socio-economic status,
+nationality, personal appearance, race, caste, color, religion, or sexual
+identity and orientation.
+
+We pledge to act and interact in ways that contribute to an open, welcoming,
+diverse, inclusive, and healthy community.
+
+## Our Standards
+
+Examples of behavior that contributes to a positive environment for our
+community include:
+
+* Demonstrating empathy and kindness toward other people
+* Being respectful of differing opinions, viewpoints, and experiences
+* Giving and gracefully accepting constructive feedback
+* Accepting responsibility and apologizing to those affected by our mistakes,
+  and learning from the experience
+* Focusing on what is best not just for us as individuals, but for the overall
+  community
+
+Examples of unacceptable behavior include:
+
+* The use of sexualized language or imagery, and sexual attention or advances of
+  any kind
+* Trolling, insulting or derogatory comments, and personal or political attacks
+* Public or private harassment
+* Publishing others' private information, such as a physical or email address,
+  without their explicit permission
+* Other conduct which could reasonably be considered inappropriate in a
+  professional setting
+
+## Enforcement Responsibilities
+
+Community leaders are responsible for clarifying and enforcing our standards of
+acceptable behavior and will take appropriate and fair corrective action in
+response to any behavior that they deem inappropriate, threatening, offensive,
+or harmful.
+
+Community leaders have the right and responsibility to remove, edit, or reject
+comments, commits, code, wiki edits, issues, and other contributions that are
+not aligned to this Code of Conduct, and will communicate reasons for moderation
+decisions when appropriate.
+
+## Scope
+
+This Code of Conduct applies within all community spaces, and also applies when
+an individual is officially representing the community in public spaces.
+Examples of representing our community include using an official email address,
+posting via an official social media account, or acting as an appointed
+representative at an online or offline event.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported to the community leaders responsible for enforcement at
+[INSERT CONTACT METHOD].
+All complaints will be reviewed and investigated promptly and fairly.
+
+All community leaders are obligated to respect the privacy and security of the
+reporter of any incident.
+
+## Enforcement Guidelines
+
+Community leaders will follow these Community Impact Guidelines in determining
+the consequences for any action they deem in violation of this Code of Conduct:
+
+### 1. Correction
+
+**Community Impact**: Use of inappropriate language or other behavior deemed
+unprofessional or unwelcome in the community.
+
+**Consequence**: A private, written warning from community leaders, providing
+clarity around the nature of the violation and an explanation of why the
+behavior was inappropriate. A public apology may be requested.
+
+### 2. Warning
+
+**Community Impact**: A violation through a single incident or series of
+actions.
+
+**Consequence**: A warning with consequences for continued behavior. No
+interaction with the people involved, including unsolicited interaction with
+those enforcing the Code of Conduct, for a specified period of time. This
+includes avoiding interactions in community spaces as well as external channels
+like social media. Violating these terms may lead to a temporary or permanent
+ban.
+
+### 3. Temporary Ban
+
+**Community Impact**: A serious violation of community standards, including
+sustained inappropriate behavior.
+
+**Consequence**: A temporary ban from any sort of interaction or public
+communication with the community for a specified period of time. No public or
+private interaction with the people involved, including unsolicited interaction
+with those enforcing the Code of Conduct, is allowed during this period.
+Violating these terms may lead to a permanent ban.
+
+### 4. Permanent Ban
+
+**Community Impact**: Demonstrating a pattern of violation of community
+standards, including sustained inappropriate behavior, harassment of an
+individual, or aggression toward or disparagement of classes of individuals.
+
+**Consequence**: A permanent ban from any sort of public interaction within the
+community.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage],
+version 2.1, available at
+[https://www.contributor-covenant.org/version/2/1/code_of_conduct.html][v2.1].
+
+Community Impact Guidelines were inspired by
+[Mozilla's code of conduct enforcement ladder][Mozilla CoC].
+
+For answers to common questions about this code of conduct, see the FAQ at
+[https://www.contributor-covenant.org/faq][FAQ]. Translations are available at
+[https://www.contributor-covenant.org/translations][translations].
+
+[homepage]: https://www.contributor-covenant.org
+[v2.1]: https://www.contributor-covenant.org/version/2/1/code_of_conduct.html
+[Mozilla CoC]: https://github.com/mozilla/diversity
+[FAQ]: https://www.contributor-covenant.org/faq
+[translations]: https://www.contributor-covenant.org/translations

data/LICENSE.txt

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Philip Q Nguyen
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/lib/gai18n/cli.rb

@@ -0,0 +1,79 @@
+require 'optparse'
+
+module GAI18n
+  class CLI
+    class Parser
+      Options = Struct.new(:secret)
+
+      def self.parse(options)
+        args = Options.new
+
+        opt_parser = OptionParser.new do |opts|
+          opts.banner = "Usage: gai18n [setup|translate|assistant:create] [options]"
+
+          if options[0] == 'setup'
+            opts.on("-sSECRET", "--secret=SECRET", "OpenAI Secret Key") do |secret|
+              args.secret = secret
+            end
+          end
+
+          opts.on("-h", "--help", "Prints this help") do
+            puts opts
+            exit
+          end
+        end
+
+        opt_parser.parse!(options)
+        return args
+      end
+    end
+
+    def run(argv)
+      if argv[0] == 'setup'
+        setup openai_api_key: Parser.parse(argv).secret
+      elsif argv[0] == 'translate'
+        translate
+      elsif argv[0] == 'assistant:create'
+        assistant_create
+      else
+        Parser.parse(argv + ['-h'])
+      end
+    end
+
+    private
+
+    def setup(openai_api_key: nil)
+      GAI18n::Setup.new.run(openai_api_key: openai_api_key)
+    end
+
+    def translate
+      require_config_file
+      GAI18n::Translator.new.translate
+    end
+
+    def assistant_create
+      require_config_file
+      puts 'Creating OpenAI assistant...'
+      assistant = GAI18n::OpenAI::Assistant.create
+      puts 'Created OpenAI assistant.'
+      puts 'Please add the following to your gai18n.rb config file.'
+      puts 'GAI18n.configure do |config|'
+      puts "  config.openai_assistant_id = '#{assistant.id}'"
+      puts 'end'
+    end
+
+    def require_config_file
+      if File.exist?('./config/gai18n.rb' )
+        require './config/gai18n'
+      elsif File.exist?('./gai18n.rb' )
+        require './gai18n'
+      else
+        message = [
+          'GAI18n configuration file not found.',
+          'Ensure there is one at ./config/gai18n.rb or ./gai18n.rb'
+        ].join(' ')
+        raise GAI18n::LoadError, message
+      end
+    end
+  end
+end

data/lib/gai18n/config_template.rb

@@ -0,0 +1,59 @@
+GAI18n.configure do |config|
+  config.source_locale = {
+    english: {
+      # `files`: this specifies where to match one or more source yml files.
+      # Change this to match your source language file(s) location.
+      files: 'config/**/en.yml',
+
+      # `file_identifier`: this is used to specify the identifier for the
+      # source language file. It could be a part of the file name or the entire
+      # file path. This identifier is used by the gem to locate and process the
+      # source language file. For example, if your source language file is
+      # located at 'config/dashboards/en.yml' or 'config/en/dashboard.yml' you
+      # should set `file_identifier` to 'en'.
+      file_identifier: 'en',
+
+      # `root_key`: this is the root key of the yaml content in your source
+      #file(s). For example, if your root key in the source language file is
+      # `en`, you should set `root_key` to 'en'.
+      root_key: 'en'
+    }
+  }
+
+  config.target_locales = {
+    japanese: {
+      # `file_identifier`: this is used to specify the identifier for the
+      # target language file. It could be a part of the file name or the entire
+      # file path. This identifier is used by the gem to locate and process the
+      # target language file. For example, if your target language file is
+      # located at 'config/dashboards/jp.yml' or 'config/jp/dashboard.yml' you
+      # should set `file_identifier` to 'jp'.
+      file_identifier: 'jp',
+
+      # `root_key`: this is the root key of the yaml content in your source
+      # file(s). For example, if your root key in the target language file is
+      # `jp`, you should set `root_key` to 'jp'.
+      root_key: 'jp'
+    },
+    french: {
+      file_identifier: 'fr',
+      root_key: 'fr'
+    }
+  }
+
+  # `openai_secret_key`: this is the secret key from OpenAI. You can get this
+  # from your OpenAI account.
+  config.openai_secret_key = 'replace_with_the_openai_secret_key'
+
+  # `base_git_branch`: we need the base branch to compare changes between the
+  # current source file to the source file in the base branch. This is used to
+  # determine the changes that need to be translated.
+  config.base_git_branch = 'main'
+
+  # `openai_assistant_id`: this is the assistant id from OpenAI. Please create
+  # this assistant using the `bundle exec gai18n assistant:create` command.
+  # Do not use an existing assisntant. This is because the assistant created
+  # from the command is configured with specific instructions and tools that
+  # are required for translation.
+  config.openai_assistant_id = 'replace_with_the_assistant_id'
+end

data/lib/gai18n/configuration.rb

@@ -0,0 +1,45 @@
+module GAI18n
+  class Configuration
+    attr_accessor :source_locale, :openai_secret_key, :locales,
+                  :openai_assistant_id, :target_locales
+
+    attr_writer :base_git_branch, :model, :keys_per_paginated_requests
+
+    def openai_client
+      args = {
+        access_token: openai_secret_key
+      }
+      @openai_client ||= ::OpenAI::Client.new args
+    end
+
+    def project_root
+      return Rails.root if defined? Rails
+      return Bundler.root if defined? Bundler
+      Dir.pwd
+    end
+
+    def base_git_branch
+      @base_git_branch ||= 'main'
+    end
+
+    def model
+      @model ||= 'gpt-3.5-turbo-1106'
+    end
+
+    def keys_per_paginated_requests
+      @keys_per_paginated_requests ||= 20
+    end
+  end
+
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
+
+  def self.config
+    configuration
+  end
+
+  def self.configure
+    yield configuration if block_given?
+  end
+end

data/lib/gai18n/content.rb

@@ -0,0 +1,48 @@
+module GAI18n
+  class Content
+    attr_reader :current
+
+    def initialize(raw)
+      @current = raw
+    end
+
+    def deep_merge!(second)
+      merger = proc do |key, v1, v2|
+        if Hash === v1 && Hash === v2
+          v1.merge!(v2, &merger)
+        else
+          v2
+        end
+      end
+      current.merge!(second, &merger)
+      self
+    end
+
+    def to_h
+      current
+    end
+
+    def keys
+      flatten_keys current
+    end
+
+    def value_for(key)
+      keys = key.split('.')
+      keys.inject(current) do |hash, key|
+        hash[key]
+      end
+    end
+
+    private
+
+    def flatten_keys(hash, prefix = nil)
+      hash.flat_map do |key, value|
+        if value.is_a? Hash
+          flatten_keys(value, prefix ? "#{prefix}.#{key}" : key)
+        else
+          prefix ? "#{prefix}.#{key}" : key
+        end
+      end
+    end
+  end
+end

data/lib/gai18n/errors.rb

@@ -0,0 +1,5 @@
+module GAI18n
+  class GAI18nError < StandardError; end
+  class IncorrectResponseError < GAI18nError; end
+  class LoadError < GAI18nError; end
+end

data/lib/gai18n/git_comparison.rb

@@ -0,0 +1,38 @@
+module GAI18n
+  class GitComparison
+    attr_reader :base_git_branch, :project_root, :source_path, :source_root_key
+
+    def initialize(locale_file)
+      @base_git_branch = GAI18n.config.base_git_branch
+      @project_root = GAI18n.config.project_root
+      @source_path = locale_file.source_path
+      @source_root_key = locale_file.source_root_key
+    end
+
+    def changes
+      git = Git.open(project_root, :log => Logger.new(STDOUT))
+      diff = git.diff(base_git_branch).path(source_path)
+      base_content = diff.first&.blob(:src)&.contents
+      return [] if base_content.nil?
+      base_yaml = YAML.load(base_content)[source_root_key.to_s]
+      current_yaml = YAML.load_file(source_path)[source_root_key.to_s]
+      keys_with_changed_values(base_yaml, current_yaml)
+    end
+
+    private
+
+    def keys_with_changed_values(base_yaml, current_yaml, prefix = '')
+      base_yaml.keys.inject([]) do |keys, key|
+        full_key = prefix.empty? ? key : "#{prefix}.#{key}"
+        if base_yaml[key] != current_yaml[key]
+          if base_yaml[key].is_a?(Hash) && current_yaml[key].is_a?(Hash)
+            keys.concat(keys_with_changed_values(base_yaml[key], current_yaml[key], full_key))
+          else
+            keys << full_key
+          end
+        end
+        keys
+      end
+    end
+  end
+end

data/lib/gai18n/locale.rb

@@ -0,0 +1,60 @@
+module GAI18n
+  class Locale
+    class << self
+      def source_locale
+        GAI18n.config.source_locale
+      end
+
+      def source_language
+        source_locale.keys.first
+      end
+
+      def source_file_identifier
+        source_locale[source_language][:file_identifier]
+      end
+
+      def source_paths
+        Dir.glob(source_locale[source_language][:files])
+      end
+
+      def source_root_key
+        source_locale[source_language][:root_key] || source_file_identifier
+      end
+
+      def target_locale_files
+        target_locales = GAI18n.config.target_locales
+        target_locales.flat_map do |(lang, options)|
+          source_paths.map do |source_path|
+            file_identifier = options[:file_identifier]
+            root_key = options[:root_key] || file_identifier
+            path = source_path.gsub source_file_identifier, file_identifier
+            Locale.new lang: lang,
+                       root_key: root_key,
+                       path: path,
+                       source_lang: source_language,
+                       source_path: source_path,
+                       source_root_key: source_root_key
+          end
+        end
+      end
+    end
+
+    attr_reader :lang, :root_key, :path, :file_identifier, :source_lang,
+                :source_path, :source_root_key, :source_file_identifier,
+                :locale_file_class
+
+    def initialize(lang:, root_key:, path:, source_lang:, source_path:,
+                   source_root_key:)
+      @lang = lang
+      @root_key = root_key
+      @path = path
+      @source_lang = source_lang
+      @source_path = source_path
+      @source_root_key = source_root_key
+    end
+
+    def translate
+      LocaleFile.new(self).translate
+    end
+  end
+end

data/lib/gai18n/locale_file.rb

@@ -0,0 +1,118 @@
+module GAI18n
+  class LocaleFile
+    extend Forwardable
+
+    attr_reader :locale, :assistant_id, :skip_keys
+
+    def_delegators :@locale, :lang, :root_key, :path, :source_lang,
+                   :source_path, :source_root_key
+
+    def initialize(locale)
+      @locale = locale
+      @assistant_id = GAI18n.config.openai_assistant_id
+      @skip_keys = []
+    end
+
+    def translate(skip_keys = [])
+      @skip_keys = skip_keys
+      write_file and return if translatable_key_values.empty?
+      thread = GAI18n::OpenAI::Thread.create
+      OpenAI::Message.create thread_id: thread.id, content: message_content
+      run = OpenAI::Run.create assistant_id: assistant_id, thread_id: thread.id
+      run = run.wait_until_done
+      if run.requires_action?
+        run.accept_translations.each {|translation| insert translation}
+        write_file
+        locale_file = self.class.new locale
+        locale_file.translate(skip_keys + translatable_keys)
+      else
+        msg = ['Either the Run took too long or received incorrect',
+               "Run status from OpenAI: #{run.status}"].join(' ')
+        raise IncorrectResponseError, msg
+      end
+    end
+
+    private
+
+    def insert(translated_hash)
+      content.deep_merge!(translated_hash)
+    end
+
+    def write_file
+      ordered_content = source_content.deep_merge!(content.to_h)
+      filtered_content = undeleted_content ordered_content.to_h, content.to_h
+      content_yaml = {root_key.to_s => filtered_content}.to_yaml
+      file_path = if path.start_with? '/'
+        path
+      else
+        GAI18n.config.project_root.join(path)
+      end
+      File.open(file_path, 'w') { |f| f.write content_yaml }
+    end
+
+    def translatable_key_values
+      @translatable_key_values ||= translatable_keys.map do |key|
+        [key, source_content.value_for(key)]
+      end.to_h
+    end
+
+    def translatable_keys
+      @_translatable_keys ||= begin
+        keys = source_content.keys - content.keys + keys_with_changed_values - skip_keys
+        keys.first GAI18n.config.keys_per_paginated_requests
+      end
+    end
+
+    def keys_with_changed_values
+      GitComparison.new(self).changes
+    end
+
+    def deep_merge(first, second)
+      merger = proc do |key, v1, v2|
+        if Hash === v1 && Hash === v2
+          v1.merge(v2, &merger)
+        else
+          v2
+        end
+      end
+      first.merge(second, &merger)
+    end
+
+    def source_content
+      @source_content ||= begin
+        raw = YAML.load_file(source_path)[source_root_key.to_s]
+        Content.new raw
+      end
+    end
+
+    def content
+      @content ||= begin
+        raw_content = if File.exist?(path)
+          raw_content = YAML.load_file(path)
+          raw_content.respond_to?(:[]) ? raw_content[root_key.to_s] : {}
+        else
+          {}
+        end
+        Content.new undeleted_content(raw_content, source_content.to_h)
+      end
+    end
+
+    def undeleted_content(first_content, second_content)
+      first_content.each_with_object({}) do |(key, value), new_hash|
+        if second_content.key?(key)
+          new_hash[key] = value.is_a?(Hash) ? undeleted_content(value, second_content[key]) : value
+        end
+      end
+    end
+
+    def message_content
+      <<~HEREDOC
+        Below are keys and values. Please translate the values from #{source_lang} to #{lang}, then submit the translations.
+
+        ```
+        #{translatable_key_values.map {|key, val| "#{key}: #{val}"}.join("\n")}
+        ```
+      HEREDOC
+    end
+  end
+end

data/lib/gai18n/openai/assistant.rb

@@ -0,0 +1,72 @@
+module GAI18n
+  module OpenAI
+    class Assistant
+      class << self
+        def create
+          description = ['This assisistant was created by GAI18n',
+                        'to help with internationalization.'].join ' '
+          instructions = "You are a software localization engineer. You have been tasked with translating the English source to the target language. You will be given a key and a source string. You will be asked to translate the source string to the target language. Once translated, please call the translation.submit function with the key, translated string, and the target language. If you are asked to translate into multiple target languages, please call the translation.submit function for each target language."
+          openai_client = GAI18n.config.openai_client
+          model = GAI18n.config.model
+          parameters = {
+            model: model,
+            name: "GAI18n-#{Time.now.to_i}",
+            description: description,
+            instructions: instructions,
+            tools: [
+              {
+                type: "function",
+                function: {
+                  name: "translation.submit",
+                  description: "Submit a translation",
+                  parameters: {
+                    type: "object",
+                    properties: {
+                      key: {
+                        type: "string",
+                        description: "The given key that's associated to the string needing translation."
+                      },
+                      translation: {
+                        type: "string",
+                        description: "The translated string."
+                      },
+                      language: {
+                        type: "string",
+                        description: "The language of the translation."
+                      }
+                    },
+                    required: ["key", "translation", "language"]
+                  }
+                }
+              }
+            ]
+          }
+          response = openai_client.assistants.create parameters: parameters
+          new response
+        end
+
+        def find(id)
+          openai_client = GAI18n.config.openai_client
+          response = openai_client.assistants.retrieve id: id
+          new response
+        end
+      end
+
+      attr_reader :id, :object, :created_at, :name, :description,
+                  :model, :instructions, :tools, :file_ids, :metadata
+
+      def initialize(response)
+        @id = response['id']
+        @object = response['object']
+        @created_at = response['created_at']
+        @name = response['name']
+        @description = response['description']
+        @model = response['model']
+        @instructions = response['instructions']
+        @tools = response['tools']
+        @file_ids = response['file_ids']
+        @metadata = response['metadata']
+      end
+    end
+  end
+end

data/lib/gai18n/openai/message.rb

@@ -0,0 +1,40 @@
+module GAI18n
+  module OpenAI
+    class Message
+      class << self
+        def create(thread_id:, content:, uploads: [])
+          openai_client = GAI18n.config.openai_client
+          parameters = {
+            role: 'user',
+            content: content
+          }
+          response = openai_client.messages.create thread_id: thread_id,
+                                                   parameters: parameters
+          new response
+        end
+
+        def all(thread_id:)
+          openai_client = GAI18n.config.openai_client
+          response = openai_client.messages.list(thread_id: thread_id)
+          response['data'].map { |message| new message }
+        end
+      end
+
+      attr_reader :id, :object, :created_at, :role, :thread_id, :content,
+                  :file_ids, :assistant_id, :run_id, :metadata
+
+      def initialize(response)
+        @id = response['id']
+        @object = response['object']
+        @created_at = response['created_at']
+        @role = response['role']
+        @thread_id = response['thread_id']
+        @content = response['content']
+        @file_ids = response['file_ids']
+        @assisistant_id = response['assistant_id']
+        @run_id = response['run_id']
+        @metadata = response['metadata']
+      end
+    end
+  end
+end

data/lib/gai18n/openai/run.rb

@@ -0,0 +1,116 @@
+module GAI18n
+  module OpenAI
+    class Run
+      class << self
+        def create(assistant_id:, thread_id:)
+          openai_client = GAI18n.config.openai_client
+          parameters = {assistant_id: assistant_id}
+          response = openai_client.runs.create thread_id: thread_id,
+                                               parameters: parameters
+          new response
+        end
+
+        def find(thread_id:, id:)
+          openai_client = GAI18n.config.openai_client
+          response = openai_client.runs.retrieve thread_id: thread_id, id: id
+          new response
+        end
+      end
+
+      attr_accessor :id, :object, :created_at, :assistant_id, :thread_id,
+                   :status, :started_at, :expires_at, :cancelled_at, :failed_at,
+                   :completed_at, :last_error, :model, :instructions, :tools,
+                   :file_ids, :metadata, :tool_calls
+
+      def initialize(response)
+        set_attributes_from(response)
+      end
+
+      def completed?
+        status == 'completed'
+      end
+
+      def requires_action?
+        status == 'requires_action'
+      end
+
+      def failed?
+        status == 'failed'
+      end
+
+      def cancelled?
+        status == 'cancelled'
+      end
+
+      def expired?
+        status == 'expired'
+      end
+
+      def accept_translations
+        submit_tool_outputs
+        submit_translations
+      end
+
+      def reload
+        openai_client = GAI18n.config.openai_client
+        response = openai_client.runs.retrieve thread_id: thread_id, id: id
+        set_attributes_from response
+      end
+
+      def wait_until_done(count = 0)
+        return self if count > 10
+        reload
+        halt_statuses = %w[requires_action completed failed cancelled expired]
+        return self if halt_statuses.include? status
+        sleep 5
+        wait_until_done count + 1
+      end
+
+      private
+
+      def set_attributes_from(response)
+        @id = response['id']
+        @object = response['object']
+        @created_at = response['created_at']
+        @assistant_id = response['assistant_id']
+        @thread_id = response['thread_id']
+        @status = response['status']
+        @started_at = response['started_at']
+        @expires_at = response['expires_at']
+        @cancelled_at = response['cancelled_at']
+        @failed_at = response['failed_at']
+        @completed_at = response['completed_at']
+        @last_error = response['last_error']
+        @model = response['model']
+        @instructions = response['instructions']
+        @tools = response['tools']
+        @file_ids = response['file_ids']
+        @metadata = response['metadata']
+        required_action = response.fetch('required_action', {}) || {}
+        submit_tool_outputs = required_action.fetch('submit_tool_outputs', {})
+        @tool_calls = submit_tool_outputs.fetch('tool_calls', [])
+      end
+
+      def submit_tool_outputs
+        parameters = tool_calls.inject({tool_outputs: []}) do |hash, tool_call|
+          hash[:tool_outputs] << {
+            tool_call_id: tool_call['id'],
+            output: "Accepted"
+          }
+          hash
+        end
+        openai_client = GAI18n.config.openai_client
+        openai_client.runs.submit_tool_outputs run_id: id,
+                                               thread_id: thread_id,
+                                               parameters: parameters
+      end
+
+      def submit_translations
+        tool_calls.map do |tool_call|
+          args = JSON.parse(tool_call["function"]["arguments"]).transform_keys(&:to_sym)
+          Translation.new.submit **args
+        end
+      end
+    end
+  end
+end

data/lib/gai18n/openai/thread.rb

@@ -0,0 +1,23 @@
+module GAI18n
+  module OpenAI
+    class Thread
+      class << self
+        def create
+          openai_client = GAI18n.config.openai_client
+          response = openai_client.threads.create
+          new response
+        end
+
+      end
+
+      attr_reader :id, :object, :created_at, :metadata
+
+      def initialize(response)
+        @id = response['id']
+        @object = response['object']
+        @created_at = response['created_at']
+        @metadata = response['metadata']
+      end
+    end
+  end
+end

data/lib/gai18n/setup.rb

@@ -0,0 +1,38 @@
+module GAI18n
+  class Setup
+    def run(openai_api_key: nil)
+      copy_template
+      puts "GAI18n configuration file created at #{destination}"
+      if openai_api_key
+        add_api_key_to_template openai_api_key
+        puts 'Added OpenAI API key to configuration file.'
+        puts 'Please run `bundle exec gai18n assistant:create` to create an OpenAI assistant, and then add the assistant id to the configuration file.'
+      else
+        puts '1. Please add your OpenAI API key to the configuration file.'
+        puts '2. Then run `bundle exec gai18n assistant:create` to create an OpenAI assistant, and then add the assistant id to the configuration file.'
+      end
+    end
+
+    private
+
+    def destination
+      if File.directory?('./config')
+        "#{GAI18n.config.project_root}/config/gai18n.rb"
+      else
+        "#{GAI18n.config.project_root}/gai18n.rb"
+      end
+    end
+
+    def copy_template
+      template_file = File.expand_path '../config_template.rb', __FILE__
+      FileUtils.cp template_file, destination
+    end
+
+    def add_api_key_to_template(openai_api_key)
+      content = File.read(destination).gsub('replace_with_the_openai_secret_key', openai_api_key)
+      File.open(destination, 'w') do |file|
+        file.write content
+      end
+    end
+  end
+end

data/lib/gai18n/translation.rb

@@ -0,0 +1,15 @@
+module GAI18n
+  class Translation
+    def submit(key:, translation:, language:)
+      arr = key.split('.') + [translation]
+      result = arr.reverse.inject({}) do |hash, k|
+        if k == translation
+          k
+        else
+          { k => hash }
+        end
+      end
+      result
+    end
+  end
+end

data/lib/gai18n/translator.rb

@@ -0,0 +1,10 @@
+module GAI18n
+  class Translator
+    def translate
+      threads = Locale.target_locale_files.map do |target_locale_file|
+        Thread.new { target_locale_file.translate }
+      end
+      threads.each(&:join)
+    end
+  end
+end

data/lib/gai18n/version.rb

@@ -0,0 +1,3 @@
+module GAI18n
+  VERSION = '0.1.1'
+end

data/lib/gai18n.rb

@@ -0,0 +1,22 @@
+require 'openai'
+require 'json'
+require 'yaml'
+require 'git'
+require 'openai/overrides'
+require 'gai18n/errors'
+require 'gai18n/openai/assistant'
+require 'gai18n/openai/message'
+require 'gai18n/openai/thread'
+require 'gai18n/openai/run'
+require 'gai18n/version'
+require 'gai18n/configuration'
+require 'gai18n/git_comparison'
+require 'gai18n/setup'
+require 'gai18n/translator'
+require 'gai18n/translation'
+require 'gai18n/locale_file'
+require 'gai18n/locale'
+require 'gai18n/content'
+
+module GAI18n
+end

data/lib/openai/overrides.rb

@@ -0,0 +1,16 @@
+
+if ENV['ENABLE_OPENAI_LOGS'] == 'true'
+  module OpenAI
+    module HTTP
+      def conn(multipart: false)
+        Faraday.new do |f|
+          f.options[:timeout] = @request_timeout
+          f.request(:multipart) if multipart
+          f.response :raise_error
+          f.response :json
+          f.response :logger, ::Logger.new(STDOUT), bodies: true
+        end
+      end
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gai18n
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Philip Q Nguyen
@@ -116,9 +116,29 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- CODE_OF_CONDUCT.md
+- LICENSE.txt
 - README.md
 - Rakefile
 - bin/gai18n
+- lib/gai18n.rb
+- lib/gai18n/cli.rb
+- lib/gai18n/config_template.rb
+- lib/gai18n/configuration.rb
+- lib/gai18n/content.rb
+- lib/gai18n/errors.rb
+- lib/gai18n/git_comparison.rb
+- lib/gai18n/locale.rb
+- lib/gai18n/locale_file.rb
+- lib/gai18n/openai/assistant.rb
+- lib/gai18n/openai/message.rb
+- lib/gai18n/openai/run.rb
+- lib/gai18n/openai/thread.rb
+- lib/gai18n/setup.rb
+- lib/gai18n/translation.rb
+- lib/gai18n/translator.rb
+- lib/gai18n/version.rb
+- lib/openai/overrides.rb
 homepage: https://github.com/philipqnguyen/gai18n
 licenses: []
 metadata: