translation_api 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c9c1c9798882d3591828278748d1aa763227e85a2c37be0d7583859c78fc0013
-  data.tar.gz: ac5378596044a829e4f653e258ade5074e023b2ca0d3ccb1a55b603c8e707c5d
+  metadata.gz: 9d36a40c0d987797c40e7ab34cf32d270a5b26b0b2d9372d62bbbd0ede73ca5e
+  data.tar.gz: d05a8271c9fa12105651eacbbfbfd04253da239c89932cdec9a2e64d7b5d7895
 SHA512:
-  metadata.gz: 94379f1ffbd3741c72fc87289b7887b434eb65e3bb8fb55438a98fdede95e4220f473aae4f56aeed5a126099d95fdc7abde0d2b9fd05f53d8ec7286bb03af54a
-  data.tar.gz: 17d351d60dc25125577f5a9c8a43490e86a91a8d77420d795131c17de6c7b1a33e3f9050c5d836272fd87ec825846a25379fd44d42f1725580bde08b4fe5c06a
+  metadata.gz: bcc16101336a524fd635c23f4028202a2f4fed6215936df5787da62ea07a902298b4cc63eb41f3351555f41ac72c293ea3498bcbc6f699fbc0605311f4288a69
+  data.tar.gz: 5d13abeeb461a92952f1c34715ae0e0b84dd43d65136584e409520b6021d392a849a5853cced542ab613aca7e6c8d93c8b110ee58d9b82006bbf22f826cc6c3e

data/CHANGELOG.md

@@ -37,5 +37,10 @@
 - OpenAIのモデル追加
 
 ## [1.0.0] - 2025-11-10
+
 - 全体的な設計を刷新
 - よりシンプルなインターフェースに変更
+
+## [1.1.0] - 2025-11-23
+
+- カスタムプロンプトを設定できる機能を追加

data/example.rb

@@ -9,9 +9,11 @@ end
 
 text = ARGV.join(" ")
 TranslationAPI.configure do |config|
-  config.language     = "english"
-  config.provider     = :deepl
-  config.except_words = %w[hoge fuga]
+  config.language      = "english"
+  config.provider      = :openai
+  config.output_logs   = false
+  config.except_words  = %w[hoge fuga]
+  config.custom_prompt = "Please Samurai style."
 end
 
 translated_text = TranslationAPI.translate(text)

data/lib/translation_api/config.rb

@@ -6,18 +6,19 @@ class TranslationAPI
   class Config
     include Singleton
 
-    attr_accessor :language, :provider, :output_logs, :except_words, :deepl_pro
+    attr_accessor :language, :provider, :output_logs, :except_words, :custom_prompt, :deepl_pro
 
     def self.configure(&block)
       block.call(instance)
     end
 
     def initialize
-      @language     = "japanese"
-      @provider     = :openai
-      @output_logs  = true
-      @except_words = []
-      @deepl_pro    = false
+      @language      = "japanese"
+      @provider      = :openai
+      @output_logs   = true
+      @except_words  = []
+      @custom_prompt = nil
+      @deepl_pro     = false
     end
   end
 end

data/lib/translation_api/provider/openai.rb

@@ -17,13 +17,13 @@ class TranslationAPI
       MODEL_ERROR_MESSAGE =
         "Specified model is not supported. Please check the model name."
 
-      def initialize(output_logs:, except_words:, language:)
+      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)
+        @user_prompt   = user_prompt_text(language, custom_prompt)
       end
 
       def translate(text)
@@ -87,8 +87,9 @@ class TranslationAPI
         TEXT
       end
 
-      def user_prompt_text(language)
+      def user_prompt_text(language, custom_prompt)
         <<~TEXT
+          #{custom_prompt || ""}
           Please translate this text to #{language}:
         TEXT
       end

data/lib/translation_api/version.rb

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

data/lib/translation_api.rb

@@ -23,16 +23,11 @@ class TranslationAPI
     new(**).translate(text)
   end
 
-  def initialize(
-    language:     config.language,
-    provider:     config.provider,
-    output_logs:  config.output_logs,
-    except_words: config.except_words
-  )
-    @language     = language
-    @output_logs  = output_logs
-    @except_words = except_words
-    @provider     = init_provider(provider)
+  def initialize(**options)
+    use_provided_options(options)
+    use_default_options
+
+    @provider = init_provider(@provider)
   end
 
   def config
@@ -45,13 +40,27 @@ class TranslationAPI
 
   private
 
+  def use_provided_options(options)
+    options.each do |key, value|
+      instance_variable_set("@#{key}", value)
+    end
+  end
+
+  def use_default_options
+    self.class.config.instance_variables.each do |var|
+      key = var.to_s.delete_prefix("@").to_sym
+      instance_variable_set("@#{key}", self.class.config.send(key))
+    end
+  end
+
   def init_provider(provider)
     case provider
     when :openai
       Provider::OpenAI.new(
         output_logs: @output_logs,
         except_words: @except_words,
-        language: @language
+        language: @language,
+        custom_prompt: @custom_prompt
       )
     when :deepl
       Provider::DeepL.new(

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: translation_api
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - milkeclair