translation_api 1.1.1 → 1.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 18aab159f877af05a07c70b365d022ac0531868db5a00dbcbd7e861d299137a9
-  data.tar.gz: d8fe14a44a14753e1b3c8ea22239dce5c845d1b97914c16305501caaf1ba1af0
+  metadata.gz: fef1087b2a2ddc29d3a942318b98fee14c24d941250665e69e7a875eba931b56
+  data.tar.gz: 4104acc87346c85fc82f0dd6da55475ad308535f63b8ec0f09db6afa73e341bb
 SHA512:
-  metadata.gz: 2d8f18647569f3b6d7f11be1f554efe08a23e834d7a626e2420c15119f29bdb526f0b73d6f888525b2f03c3e1f4f757c1ae099014807f2dd7ed11f63de28fdb2
-  data.tar.gz: ba509a11536d06ea663b43dee3c81615d1a4a8bf8ebcf496aa7f709b6523aaa710b22d4baef92f63347ffe6d4a04e43e9461a2f0933ae7a0bd7c064cedae6f38
+  metadata.gz: 8bf6cac81b1c2e6153194bcd7ec173f3333a19972c07f66a425121c5c2276fa982c10f0ce6e3d81ff7c31457b63e657a0e6046f2e566a08705e38108bc739278
+  data.tar.gz: '06609efa3d55a13d4cd8eb54a6ea6580cfa94382bb6cc81c272fad94ee78ce029fb18bb7d9317324b575e7b611d8ea2b1d4f3c1845bcf74d027e2c95101fcb42'

data/CHANGELOG.md

@@ -48,3 +48,11 @@
 ## [1.1.1] - 2025-11-23
 
 - configの設定順序を逆に変更
+
+## [1.2.0] - 2025-11-23
+
+- Gemini対応
+
+## [1.3.0] - 2026-04-11
+
+- OpenAIのモデルを更新

data/README.md

@@ -6,22 +6,28 @@ Requires api key.
 ## For OpenAI
 
 1. `touch .env`
-2. Add `OPENAI_API_KEY=YOUR_API_KEY`  
-3. Optional: `ENV["OPENAI_MODEL"]`
+2. Add `OPENAI_API_KEY=YOUR_API_KEY` or `GEMINI_API_KEY=YOUR_API_KEY` or `DEEPL_API_KEY=YOUR_API_KEY` to `.env`
+3. Optional: `ENV["OPENAI_MODEL"]` or `ENV["GEMINI_MODEL"]`
 4. `TranslationAPI.translate("text")`
 
 ### Configuration Options
 
-* output_logs (default: true)  
 * language (default: "japanese")  
 * provider (default: :openai)  
+* output_logs (default: true)  
 * except_words (default: [])  
+* custom_prompt (default: nil)
+  * Only for OpenAI and Gemini
+* deepl_pro (default: false)
+  * Only for DeepL
 
-### Output
+### Output(Only for OpenAI and Gemini)
 
 * Translated_text
 * Used Tokens
-* Cost Spent(https://openai.com/api/pricing/)
+* Cost Spent
+  * https://openai.com/api/pricing/
+  * https://ai.google.dev/gemini-api/docs/pricing/
 
 ## Example
 
@@ -29,10 +35,11 @@ Exec `ruby example.rb "text"`
 
 ```ruby
 TranslationAPI.configure do |config|
-  config.language     = "english"
-  config.provider     = :deepl
-  config.output_logs  = false
-  config.except_words = %w[hoge fuga]
+  config.language      = "english"
+  config.provider      = :gemini
+  config.output_logs   = false
+  config.except_words  = %w[hoge fuga]
+  config.custom_prompt = "Please Samurai style."
 end
 
 TranslationAPI.translate("text")

data/Rakefile

@@ -1,39 +1,3 @@
 # frozen_string_literal: true
 
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
-require "rubocop/rake_task"
-require "yard"
-require_relative "rake_helper"
-
-desc "analysis"
-task :analysis do
-  sh "bundle install"
-
-  RakeHelper.init_rake_tasks
-
-  puts "--- rspec ---"
-  Rake::Task[:spec].invoke
-
-  puts "--- rubocop ---"
-  Rake::Task[:rubocop].invoke
-
-  puts "--- yard ---"
-  Rake::Task[:yard].invoke
-end
-
-desc "push to github packages and rubygems"
-task :push do
-  sh "bundle install"
-
-  puts "--- build ---"
-  RakeHelper.build_gem
-
-  puts "--- push to github packages ---"
-  RakeHelper.push_to_github_packages
-
-  puts "--- push to rubygems ---"
-  RakeHelper.push_to_rubygems
-end
-
-task default: :analysis
+require "push_gem"

data/example.rb

@@ -16,5 +16,10 @@ TranslationAPI.configure do |config|
   config.custom_prompt = "Please Samurai style."
 end
 
-translated_text = TranslationAPI.translate(text)
-p translated_text
+begin
+  translated_text = TranslationAPI.translate(text)
+  p translated_text
+rescue StandardError => e
+  puts e
+  puts e.backtrace
+end

data/how_to_publish.txt

@@ -2,9 +2,9 @@ some code change
 version.rb change
 changelog.md change
 
-$rake analysis
+$rake gem:check
 
 $git commit
 $git push
 
-$rake push
+$rake gem:push

data/lib/translation_api/llm/log.rb

@@ -0,0 +1,97 @@
+# frozen_string_literal: true
+
+require "fileutils"
+
+class TranslationAPI
+  module Llm
+    class Log
+      def initialize(provider:, response:, cost:)
+        @provider = provider
+        @response = response
+        @cost     = cost
+      end
+
+      def write
+        write_translated_text
+        write_used_tokens
+        write_total_cost
+      end
+
+      private
+
+      def write_translated_text
+        log_file_path = text_path("translated_text.txt")
+
+        File.open(log_file_path, "a") do |file|
+          file.puts(@response.translated_text)
+        end
+      end
+
+      def write_used_tokens
+        log_file_path = text_path("tokens.txt")
+
+        existing_input_tokens, existing_output_tokens = read_existing_tokens
+        tokens => { input_tokens:, output_tokens: }
+
+        total_input_tokens  = existing_input_tokens + input_tokens
+        total_output_tokens = existing_output_tokens + output_tokens
+
+        File.open(log_file_path, "w") do |file|
+          file.puts("input: #{total_input_tokens}")
+          file.puts("output: #{total_output_tokens}")
+        end
+      end
+
+      def read_existing_tokens
+        log_file_path = text_path("tokens.txt")
+        input_tokens, output_tokens = 0, 0
+
+        if File.exist?(log_file_path)
+          File.readlines(log_file_path).each do |line|
+            tokens = line.split(":").last.strip.to_i
+            input_tokens  = tokens if line.start_with?("input:")
+            output_tokens = tokens if line.start_with?("output:")
+          end
+        end
+
+        [input_tokens, output_tokens]
+      end
+
+      def write_total_cost
+        log_file_path = text_path("cost.txt")
+        tokens => { input_tokens:, output_tokens: }
+
+        this_cost = @cost.input_cost(input_tokens) + @cost.output_cost(output_tokens)
+        total_cost = this_cost + existing_cost
+
+        File.open(log_file_path, "w") do |file|
+          file.puts(format_cost(total_cost))
+        end
+      end
+
+      def format_cost(cost)
+        "$#{format("%.8f", cost)}"
+      end
+
+      def existing_cost
+        log_file_path = text_path("cost.txt")
+
+        File.exist?(log_file_path) ? File.read(log_file_path).gsub("$", "").to_f : 0.0
+      end
+
+      def tokens
+        {
+          input_tokens: @response.dig_used_tokens(type: :input),
+          output_tokens: @response.dig_used_tokens(type: :output)
+        }
+      end
+
+      def text_path(under_logs_path)
+        output_dir = "translator_logs/#{@provider.name}"
+        FileUtils.mkdir_p(output_dir) unless File.directory?(output_dir)
+
+        File.join(output_dir, under_logs_path)
+      end
+    end
+  end
+end

data/lib/translation_api/llm/model.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+class TranslationAPI
+  module Llm
+    class Model
+      MODEL_ERROR_MESSAGE =
+        "Specified model is not supported. Please check the model name."
+
+      attr_reader :name
+
+      def initialize(name)
+        @name = name
+        validate_model!
+      end
+
+      private
+
+      def validate_model!
+        supported_models =
+          self.class.const_defined?(:SUPPORTED_MODELS) ? self.class::SUPPORTED_MODELS : []
+
+        raise ArgumentError, MODEL_ERROR_MESSAGE unless supported_models.include?(@name)
+      end
+    end
+  end
+end

data/lib/translation_api/llm/prompt.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+class TranslationAPI
+  module Llm
+    class Prompt
+      SYSTEM_PROMPT_BASE = <<~TEXT
+        Translate only.
+        Return result only, no extra info
+        Keep symbols
+      TEXT
+
+      def initialize(except_words:, language:, custom_prompt:)
+        @except_words  = except_words
+        @language      = language
+        @custom_prompt = custom_prompt
+      end
+
+      def system_prompt
+        SYSTEM_PROMPT_BASE + except_option_text
+      end
+
+      def user_prompt
+        <<~TEXT
+          #{@custom_prompt || ""}
+          Please translate this text to #{@language}:
+        TEXT
+      end
+
+      private
+
+      def except_option_text
+        return "" if @except_words.empty?
+
+        <<~TEXT
+          Words listed next are not translated: [#{@except_words.join(", ")}]
+        TEXT
+      end
+    end
+  end
+end

data/lib/translation_api/provider/deepl.rb

@@ -43,11 +43,11 @@ class TranslationAPI
       end
 
       def validate_api_key!
-        raise API_KEY_ERROR_MESSAGE unless ENV["DEEPL_API_KEY"] || ENV["DEEPL_AUTH_KEY"]
+        raise ArgumentError, API_KEY_ERROR_MESSAGE unless ENV["DEEPL_API_KEY"] || ENV["DEEPL_AUTH_KEY"]
       end
 
       def validate_supported_language!
-        raise LANGUAGE_UNSUPPORTED_MESSAGE unless supported_language?
+        raise ArgumentError, LANGUAGE_UNSUPPORTED_MESSAGE unless supported_language?
       end
 
       def supported_languages

data/lib/translation_api/provider/gemini/chat.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require "faraday"
+require "json"
+require_relative "response"
+
+class TranslationAPI
+  module Provider
+    class Gemini
+      class Chat
+        API_KEY_ERROR_MESSAGE = "API key is not found."
+
+        def initialize(model:, prompt:)
+          validate_api_key!
+
+          @model  = model
+          @prompt = prompt
+        end
+
+        def call(text)
+          Response.new(request(text))
+        end
+
+        private
+
+        def request(text)
+          connection.post("/v1beta/models/#{@model.name}:generateContent") do |request|
+            request.body = body(text)
+          end
+        end
+
+        def connection
+          Faraday.new(
+            url: "https://generativelanguage.googleapis.com",
+            headers: {
+              "Content-Type" => "application/json",
+              "x-goog-api-key" => ENV["GEMINI_API_KEY"]
+            }
+          )
+        end
+
+        def body(text)
+          {
+            contents: [
+              {
+                parts: [
+                  {
+                    text: @prompt.system_prompt + @prompt.user_prompt + text
+                  }
+                ]
+              }
+            ]
+          }.to_json
+        end
+
+        def validate_api_key!
+          raise ArgumentError, API_KEY_ERROR_MESSAGE unless ENV["GEMINI_API_KEY"]
+        end
+      end
+    end
+  end
+end

data/lib/translation_api/provider/gemini/cost.rb

@@ -0,0 +1,73 @@
+# frozen_string_literal: true
+
+class TranslationAPI
+  module Provider
+    class Gemini
+      class Cost
+        ONE_MILLION = 1_000_000
+
+        def initialize(model)
+          @model = model
+        end
+
+        def input_cost(used_tokens)
+          calculate_cost(used_tokens, :input)
+        end
+
+        def output_cost(used_tokens)
+          calculate_cost(used_tokens, :output)
+        end
+
+        private
+
+        def calculate_cost(used_tokens, type)
+          used_tokens * token_rates[@model.name][type]
+        end
+
+        def token_rates
+          three_pro.merge(two_five_pro).merge(two_five_flash).merge(two_five_flash_lite)
+        end
+
+        def three_pro
+          {
+            @model.class.three_pro => {
+              input: 2.0 / ONE_MILLION,
+              output: (2.0 * pro_cost_diff_ratio) / ONE_MILLION
+            }
+          }
+        end
+
+        def two_five_pro
+          {
+            @model.class.two_five_pro => {
+              input: 1.25 / ONE_MILLION,
+              output: (1.25 * pro_cost_diff_ratio) / ONE_MILLION
+            }
+          }
+        end
+
+        def two_five_flash
+          {
+            @model.class.two_five_flash => {
+              input: 0.3 / ONE_MILLION,
+              output: 2.5 / ONE_MILLION
+            }
+          }
+        end
+
+        def two_five_flash_lite
+          {
+            @model.class.two_five_flash_lite => {
+              input: 0.1 / ONE_MILLION,
+              output: 0.4 / ONE_MILLION
+            }
+          }
+        end
+
+        def pro_cost_diff_ratio
+          2.0
+        end
+      end
+    end
+  end
+end

data/lib/translation_api/provider/gemini/model.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+require_relative "../../llm/model"
+
+class TranslationAPI
+  module Provider
+    class Gemini
+      class Model < Llm::Model
+        SUPPORTED_MODELS = [
+          "gemini-3-pro-preview",
+          "gemini-2.5-pro",
+          "gemini-2.5-flash",
+          "gemini-2.5-flash-lite"
+        ].freeze
+
+        def self.three_pro
+          SUPPORTED_MODELS[0]
+        end
+
+        def self.two_five_pro
+          SUPPORTED_MODELS[1]
+        end
+
+        def self.two_five_flash
+          SUPPORTED_MODELS[2]
+        end
+
+        def self.two_five_flash_lite
+          SUPPORTED_MODELS[3]
+        end
+      end
+    end
+  end
+end

data/lib/translation_api/provider/gemini/response.rb

@@ -0,0 +1,43 @@
+# frozen_string_literal: true
+
+require "json"
+
+class TranslationAPI
+  module Provider
+    class Gemini
+      class Response
+        REQUEST_FAILED_MESSAGE = "Request failed with status"
+
+        attr_reader :response
+
+        def initialize(response)
+          @response = response
+        end
+
+        def translated_text
+          failed_message = "#{REQUEST_FAILED_MESSAGE} #{@response.status}"
+          raise ArgumentError, failed_message unless @response.status == 200
+
+          body_json.dig("candidates", 0, "content", "parts", 0, "text")
+        end
+
+        def dig_used_tokens(type:)
+          case type
+          when :input
+            body_json.dig("usageMetadata", "promptTokenCount")
+          when :output
+            body_json.dig("usageMetadata", "candidatesTokenCount")
+          else
+            raise ArgumentError, "Invalid token type: #{type}"
+          end
+        end
+
+        private
+
+        def body_json
+          @body_json ||= JSON.parse(@response.body)
+        end
+      end
+    end
+  end
+end

data/lib/translation_api/provider/gemini.rb

@@ -0,0 +1,39 @@
+# frozen_string_literal: true
+
+require_relative "../llm/prompt"
+require_relative "../llm/log"
+require_relative "gemini/model"
+require_relative "gemini/chat"
+require_relative "gemini/cost"
+
+class TranslationAPI
+  module Provider
+    class Gemini
+      def initialize(output_logs:, except_words:, language:, custom_prompt: nil)
+        @model  = Model.new(ENV["GEMINI_MODEL"] || Model.two_five_flash)
+        @prompt = Llm::Prompt.new(except_words:, language:, custom_prompt:)
+        @chat   = Chat.new(model: @model, prompt: @prompt)
+        @output_logs = output_logs
+      end
+
+      def translate(text)
+        return text if text.strip.empty?
+
+        @response = @chat.call(text)
+        log.write if @output_logs
+
+        @response.translated_text
+      end
+
+      def name
+        "gemini"
+      end
+
+      private
+
+      def log
+        Llm::Log.new(provider: self, response: @response, cost: Cost.new(@model))
+      end
+    end
+  end
+end

data/lib/translation_api/provider/openai/chat.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require "openai"
+require_relative "response"
+
+class TranslationAPI
+  module Provider
+    class OpenAI
+      class Chat
+        API_KEY_ERROR_MESSAGE = "API key is not found."
+
+        def initialize(model:, prompt:)
+          validate_api_key!
+
+          @model  = model
+          @prompt = prompt
+          @client = init_client
+        end
+
+        def call(text)
+          Response.new(request(text))
+        end
+
+        private
+
+        def request(text)
+          @client.chat.completions.create(
+            model: @model.name,
+            messages: [
+              { role: "system", content: @prompt.system_prompt },
+              { role: "user", content: @prompt.user_prompt + text }
+            ]
+          )
+        end
+
+        def init_client
+          ::OpenAI::Client.new(api_key: ENV["OPENAI_API_KEY"])
+        end
+
+        def validate_api_key!
+          raise ArgumentError, API_KEY_ERROR_MESSAGE unless ENV["OPENAI_API_KEY"]
+        end
+      end
+    end
+  end
+end

data/lib/translation_api/provider/openai/cost.rb

@@ -4,12 +4,10 @@ class TranslationAPI
   module Provider
     class OpenAI
       class Cost
-        BASE_MODEL_NAME = "gpt-5"
         ONE_MILLION = 1_000_000
-        BASE_MODEL_COST = 1.25 / ONE_MILLION
 
-        def initialize(provider)
-          @provider = provider
+        def initialize(model)
+          @model = model
         end
 
         def input_cost(used_tokens)
@@ -23,62 +21,39 @@ class TranslationAPI
         private
 
         def calculate_cost(used_tokens, type)
-          used_tokens * token_rates[@provider.using_model][type]
+          used_tokens * token_rates[@model.name][type]
         end
 
         def token_rates
-          normal_models = base_model.merge(mini_model).merge(nano_model)
-          normal_models.merge(other_models)
+          base.merge(mini).merge(nano)
         end
 
-        def base_model
+        def base
           {
-            BASE_MODEL_NAME => {
-              input: BASE_MODEL_COST,
-              output: BASE_MODEL_COST * normal_io_ratio[:output]
+            @model.class.base => {
+              input: 2.50 / ONE_MILLION,
+              output: 15.00 / ONE_MILLION
             }
           }
         end
 
-        def mini_model
+        def mini
           {
-            "#{BASE_MODEL_NAME}-mini" => {
-              input: BASE_MODEL_COST / normal_cost_diff_ratio,
-              output: (BASE_MODEL_COST * normal_io_ratio[:output]) / normal_cost_diff_ratio
+            @model.class.mini => {
+              input: 0.75 / ONE_MILLION,
+              output: 4.50 / ONE_MILLION
             }
           }
         end
 
-        def nano_model
-          mini_model_cost = mini_model.values[0][:input]
-
-          {
-            "#{BASE_MODEL_NAME}-nano" => {
-              input: mini_model_cost / normal_cost_diff_ratio,
-              output: (mini_model_cost * normal_io_ratio[:output]) / normal_cost_diff_ratio
-            }
-          }
-        end
-
-        def other_models
+        def nano
           {
-            "#{BASE_MODEL_NAME}-chat-latest" => {
-              input: 1.25 / ONE_MILLION,
-              output: (1.25 * normal_io_ratio[:output]) / ONE_MILLION
+            @model.class.nano => {
+              input: 0.20 / ONE_MILLION,
+              output: 1.25 / ONE_MILLION
             }
           }
         end
-
-        def normal_io_ratio
-          {
-            input: 1.0,
-            output: 8.0
-          }
-        end
-
-        def normal_cost_diff_ratio
-          5.0
-        end
       end
     end
   end

data/lib/translation_api/provider/openai/model.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+require_relative "../../llm/model"
+
+class TranslationAPI
+  module Provider
+    class OpenAI
+      class Model < Llm::Model
+        SUPPORTED_MODELS = [
+          "gpt-5.4",
+          "gpt-5.4-mini",
+          "gpt-5.4-nano"
+        ].freeze
+
+        def self.base
+          SUPPORTED_MODELS[0]
+        end
+
+        def self.mini
+          SUPPORTED_MODELS[1]
+        end
+
+        def self.nano
+          SUPPORTED_MODELS[2]
+        end
+      end
+    end
+  end
+end

data/lib/translation_api/provider/openai/response.rb

@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+class TranslationAPI
+  module Provider
+    class OpenAI
+      class Response
+        def initialize(response)
+          @response = response
+        end
+
+        def translated_text
+          @response.choices[0].message.content
+        end
+
+        def dig_used_tokens(type:)
+          case type
+          when :input
+            @response.usage.prompt_tokens
+          when :output
+            @response.usage.completion_tokens
+          else
+            raise ArgumentError, "Invalid token type: #{type}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/translation_api/provider/openai.rb

@@ -1,97 +1,38 @@
 # frozen_string_literal: true
 
-require "openai"
-require_relative "openai/log"
+require_relative "../llm/prompt"
+require_relative "../llm/log"
+require_relative "openai/model"
+require_relative "openai/chat"
+require_relative "openai/cost"
 
 class TranslationAPI
   module Provider
     class OpenAI
-      SYSTEM_PROMPT_BASE = <<~TEXT
-        Translate only.
-        Return result only, no extra info
-        Keep symbols
-      TEXT
-
-      API_KEY_ERROR_MESSAGE = "API key is not found."
-
-      MODEL_ERROR_MESSAGE =
-        "Specified model is not supported. Please check the model name."
-
       def initialize(output_logs:, except_words:, language:, custom_prompt: nil)
-        validate_api_key!
-
-        @client = init_client
-        @output_logs   = output_logs
-        @system_prompt = SYSTEM_PROMPT_BASE + except_option_text(except_words)
-        @user_prompt   = user_prompt_text(language, custom_prompt)
+        @model  = Model.new(ENV["OPENAI_MODEL"] || Model.nano)
+        @prompt = Llm::Prompt.new(except_words:, language:, custom_prompt:)
+        @chat   = Chat.new(model: @model, prompt: @prompt)
+        @output_logs = output_logs
       end
 
       def translate(text)
         return text if text.strip.empty?
 
-        @response = chat_to_api(text)
-        Log.new(self).write if @output_logs
+        @response = @chat.call(text)
+        log.write if @output_logs
 
-        translated_text
+        @response.translated_text
       end
 
-      def translated_text
-        @response["choices"][0]["message"]["content"]
-      end
-
-      def using_model
-        ENV["OPENAI_MODEL"] || "gpt-5-mini"
-      end
-
-      def dig_used_tokens(type:)
-        case type
-        when :input
-          @response["usage"]["prompt_tokens"]
-        when :output
-          @response["usage"]["completion_tokens"]
-        else
-          raise ArgumentError, "Invalid token type: #{type}"
-        end
+      def name
+        "openai"
       end
 
       private
 
-      def validate_api_key!
-        raise API_KEY_ERROR_MESSAGE unless ENV["OPENAI_API_KEY"]
-      end
-
-      def init_client
-        ::OpenAI::Client.new(
-          access_token: ENV["OPENAI_API_KEY"],
-          log_errors: true
-        )
-      end
-
-      def chat_to_api(text)
-        @client.chat(
-          parameters: {
-            model: using_model,
-            messages: [
-              { role: "system", content: @system_prompt },
-              { role: "user", content: @user_prompt + text }
-            ]
-          }
-        )
-      end
-
-      def except_option_text(except_words)
-        return "" if except_words.empty?
-
-        <<~TEXT
-          Words listed next are not translated: [#{except_words.join(", ")}]
-        TEXT
-      end
-
-      def user_prompt_text(language, custom_prompt)
-        <<~TEXT
-          #{custom_prompt || ""}
-          Please translate this text to #{language}:
-        TEXT
+      def log
+        Llm::Log.new(provider: self, response: @response, cost: Cost.new(@model))
       end
     end
   end

data/lib/translation_api/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class TranslationAPI
-  VERSION = "1.1.1"
+  VERSION = "1.3.0"
 end

data/lib/translation_api.rb

@@ -4,6 +4,7 @@ require "dotenv"
 require_relative "translation_api/version"
 require_relative "translation_api/config"
 require_relative "translation_api/provider/openai"
+require_relative "translation_api/provider/gemini"
 require_relative "translation_api/provider/deepl"
 
 class TranslationAPI
@@ -62,6 +63,13 @@ class TranslationAPI
         language: @language,
         custom_prompt: @custom_prompt
       )
+    when :gemini
+      Provider::Gemini.new(
+        output_logs: @output_logs,
+        except_words: @except_words,
+        language: @language,
+        custom_prompt: @custom_prompt
+      )
     when :deepl
       Provider::DeepL.new(
         pro: config.deepl_pro,
@@ -69,7 +77,7 @@ class TranslationAPI
         language: @language
       )
     else
-      raise UNSUPPORTED_PROVIDER_MESSAGE
+      raise ArgumentError, UNSUPPORTED_PROVIDER_MESSAGE
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: translation_api
 version: !ruby/object:Gem::Version
-  version: 1.1.1
+  version: 1.3.0
 platform: ruby
 authors:
 - milkeclair
@@ -38,7 +38,21 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: ruby-openai
+  name: faraday
+  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: openai
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -68,12 +82,21 @@ files:
 - how_to_publish.txt
 - lib/translation_api.rb
 - lib/translation_api/config.rb
+- lib/translation_api/llm/log.rb
+- lib/translation_api/llm/model.rb
+- lib/translation_api/llm/prompt.rb
 - lib/translation_api/provider/deepl.rb
+- lib/translation_api/provider/gemini.rb
+- lib/translation_api/provider/gemini/chat.rb
+- lib/translation_api/provider/gemini/cost.rb
+- lib/translation_api/provider/gemini/model.rb
+- lib/translation_api/provider/gemini/response.rb
 - lib/translation_api/provider/openai.rb
+- lib/translation_api/provider/openai/chat.rb
 - lib/translation_api/provider/openai/cost.rb
-- lib/translation_api/provider/openai/log.rb
+- lib/translation_api/provider/openai/model.rb
+- lib/translation_api/provider/openai/response.rb
 - lib/translation_api/version.rb
-- rake_helper.rb
 homepage: https://github.com/milkeclair/translation_api
 licenses:
 - MIT

data/lib/translation_api/provider/openai/log.rb

@@ -1,99 +0,0 @@
-# frozen_string_literal: true
-
-require "fileutils"
-require_relative "cost"
-
-class TranslationAPI
-  module Provider
-    class OpenAI
-      class Log
-        def initialize(provider)
-          @provider = provider
-          @cost = Cost.new(@provider)
-        end
-
-        def write
-          write_translated_text
-          write_used_tokens
-          write_total_cost
-        end
-
-        private
-
-        def write_translated_text
-          log_file_path = text_path("translated_text.txt")
-
-          File.open(log_file_path, "a") do |file|
-            file.puts(@provider.translated_text)
-          end
-        end
-
-        def write_used_tokens
-          log_file_path = text_path("tokens.txt")
-
-          existing_input_tokens, existing_output_tokens = read_existing_tokens
-          tokens => { input_tokens:, output_tokens: }
-
-          total_input_tokens  = existing_input_tokens + input_tokens
-          total_output_tokens = existing_output_tokens + output_tokens
-
-          File.open(log_file_path, "w") do |file|
-            file.puts("input: #{total_input_tokens}")
-            file.puts("output: #{total_output_tokens}")
-          end
-        end
-
-        def read_existing_tokens
-          log_file_path = text_path("tokens.txt")
-          input_tokens, output_tokens = 0, 0
-
-          if File.exist?(log_file_path)
-            File.readlines(log_file_path).each do |line|
-              tokens = line.split(":").last.strip.to_i
-              input_tokens  = tokens if line.start_with?("input:")
-              output_tokens = tokens if line.start_with?("output:")
-            end
-          end
-
-          [input_tokens, output_tokens]
-        end
-
-        def write_total_cost
-          log_file_path = text_path("cost.txt")
-          tokens => { input_tokens:, output_tokens: }
-
-          this_cost = @cost.input_cost(input_tokens) + @cost.output_cost(output_tokens)
-          total_cost = this_cost + existing_cost
-
-          File.open(log_file_path, "w") do |file|
-            file.puts(format_cost(total_cost))
-          end
-        end
-
-        def format_cost(cost)
-          "$#{format("%.8f", cost)}"
-        end
-
-        def existing_cost
-          log_file_path = text_path("cost.txt")
-
-          File.exist?(log_file_path) ? File.read(log_file_path).gsub("$", "").to_f : 0.0
-        end
-
-        def tokens
-          {
-            input_tokens: @provider.dig_used_tokens(type: :input),
-            output_tokens: @provider.dig_used_tokens(type: :output)
-          }
-        end
-
-        def text_path(under_logs_path)
-          output_dir = "translator_logs/openai"
-          FileUtils.mkdir_p(output_dir) unless File.directory?(output_dir)
-
-          File.join(output_dir, under_logs_path)
-        end
-      end
-    end
-  end
-end

data/rake_helper.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-class RakeHelper
-  GITHUB_PACKAGES_PUSH_COMMAND =
-    "gem push --key github --host https://rubygems.pkg.github.com/milkeclair " \
-    "pkg/translation_api-#{TranslationAPI::VERSION}.gem".freeze
-
-  RUBYGEMS_PUSH_COMMAND =
-    "gem push --host https://rubygems.org " \
-    "pkg/translation_api-#{TranslationAPI::VERSION}.gem".freeze
-
-  def self.init_rake_tasks
-    RSpec::Core::RakeTask.new(:spec) { |task| task.verbose = false }
-    RuboCop::RakeTask.new
-    YARD::Rake::YardocTask.new
-  end
-
-  def self.build_gem
-    abort("gemのビルドに失敗しました") unless system("rake build")
-  end
-
-  def self.push_to_github_packages
-    abort("githubへのgemのpushに失敗しました") unless system(GITHUB_PACKAGES_PUSH_COMMAND)
-  end
-
-  def self.push_to_rubygems
-    abort("rubygemsへのgemのpushに失敗しました") unless system(RUBYGEMS_PUSH_COMMAND)
-  end
-end