jekyll-chatgpt-translate 0.0.41 → 0.0.42

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9285d05f0fd8813ffa969d52df39481324035a038f91a4958ec9fe5e875cb3d9
-  data.tar.gz: 8e17f117460148801d2f8a94a7885a01853ca44e62cb38a898aed1031424c968
+  metadata.gz: 22b2e725c02a5de3447605daaad8d41fce5885908f59c8bc86cc676ac8d3a661
+  data.tar.gz: 5cb8610b0994bab60d9bf23d3aff7888f01f7ff0520e9dddc4e487e99ca94601
 SHA512:
-  metadata.gz: 3c375ccbe265fa4d5c8e33230f4f238de544fe5a7de6fdba18eb5891b96e3b0107af5dc8bbe659c8cf8477d39df32059c031566e7b892344d6bfbef3d5373a5f
-  data.tar.gz: a5f126c7abe7e84856963f17d24dadd6815fa3c4aa3beef989a96c6381ec5b663af0ee8ec0f93d43aad8ff7c436640bd10a26fb07473a2e99c3ed68fd6912615
+  metadata.gz: f1bf2b84b63f9b82ce5bc28771ab52b3571a5b4d5629fe6e10038c968bad2127407a33e9d7f1d08a7b11c247f4748021695fc569794e3f9966cca3ec30f567cf
+  data.tar.gz: b6385f234df97f3e6f5ea37e555667a59b68db7c4f7ec82ee26bd6e27a4e7cc3df759b7243f50fdcee424a7ecff785a989898ce0c050596a2c616a36f9d3ef55

data/.rultor.yml

@@ -1,10 +1,10 @@
 docker:
-  image: yegor256/rultor-image:1.21.0
+  image: yegor256/rultor-image:1.22.0
 assets:
   rubygems.yml: yegor256/home#assets/rubygems.yml
 install: |
   pdd -f /dev/null
-  sudo bundle install --no-color "--gemfile=$(pwd)/Gemfile"
+  bundle install --no-color
 release:
   script: |-
     bundle exec rake

data/jekyll-chatgpt-translate.gemspec

@@ -28,7 +28,7 @@ Gem::Specification.new do |s|
   s.required_rubygems_version = Gem::Requirement.new('>= 0') if s.respond_to? :required_rubygems_version=
   s.required_ruby_version = '>= 3.0'
   s.name = 'jekyll-chatgpt-translate'
-  s.version = '0.0.41'
+  s.version = '0.0.42'
   s.license = 'MIT'
   s.summary = 'Translate Jekyll Pages Through ChatGPT'
   s.description = [

data/lib/jekyll-chatgpt-translate/chatgpt.rb

@@ -45,12 +45,13 @@ class GptTranslate::ChatGPT
   # +key+ OpenAI API Key, which can't be nil, but can be empty string, which means dry mode (no calls to OpenAI)
   # +source+ The language to translate from
   # +target+ The language to translate into
-  def initialize(key, model, source, target)
+  def initialize(key, model, source, target, client: OpenAI::Client.new(access_token: key))
     raise 'OpenAI key cannot be nil' if key.nil?
     @key = key
     @model = model
     @source = source
     @target = target
+    @client = client
   end
 
   def translate(markdown, min: 32, window_length: 2000)
@@ -115,9 +116,8 @@ class GptTranslate::ChatGPT
   end
 
   def translate_par(par)
-    client = OpenAI::Client.new(access_token: @key)
     if @@models_printed
-      Jekyll.logger.info("Available ChatGPT models: #{client.models.list['data'].map { |m| m['id'] }.join(', ')}")
+      Jekyll.logger.info("Available ChatGPT models: #{@client.models.list['data'].map { |m| m['id'] }.join(', ')}")
       @@models_printed = true
     end
     prompt = GptTranslate::Prompt.new(par, @source, @target).to_s
@@ -125,14 +125,15 @@ class GptTranslate::ChatGPT
     answer = nil
     attempt = 0
     begin
-      response = client.chat(
+      response = @client.chat(
         parameters: {
           model: @model,
           messages: [{ role: 'user', content: prompt }],
           temperature: 0.7
         }
       )
-      answer = JSON.parse(response).dig('choices', 0, 'message', 'content')
+      json = response.is_a?(Hash) ? response : JSON.parse(response)
+      answer = json.dig('choices', 0, 'message', 'content')
       if answer.nil?
         Jekyll.logger.error("No content returned by ChatGPT: #{response}")
         raise 'No content returned by ChatGPT'

data/lib/jekyll-chatgpt-translate/version.rb

@@ -23,5 +23,5 @@
 # SOFTWARE.
 
 module GptTranslate
-  VERSION = '0.0.41'
+  VERSION = '0.0.42'
 end

data/test/test_chatgpt.rb

@@ -109,10 +109,26 @@ class GptTranslate::ChatGPTTest < Minitest::Test
     )
   end
 
+  def test_with_json
+    client = Object.new
+    def client.chat(*)
+      { 'choices' => [{ 'message' => { 'content' => 'done!' } }] }
+    end
+    chat = GptTranslate::ChatGPT.new('fake-key', 'gpt-3.5-turbo', 'en', 'ru', client: client)
+    assert_equal(
+      "done!\n\ndone!",
+      chat.translate(
+        "This is the first paragraph\n\nThis is second\n\nThis is third",
+        min: 1, window_length: 4
+      )
+    )
+  end
+
   private
 
   def stub_it!
-    stub_request(:any, 'https://api.openai.com/v1/chat/completions')
-      .to_return(body: '{"choices":[{"message":{"content": "done!"}}]}')
+    stub_request(:any, 'https://api.openai.com/v1/chat/completions').to_return(
+      body: '{"choices":[{"message":{"content": "done!"}}]}'
+    )
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: jekyll-chatgpt-translate
 version: !ruby/object:Gem::Version
-  version: 0.0.41
+  version: 0.0.42
 platform: ruby
 authors:
 - Yegor Bugayenko
@@ -186,7 +186,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.15
+rubygems_version: 3.4.10
 signing_key: 
 specification_version: 4
 summary: Translate Jekyll Pages Through ChatGPT