rbs_goose 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 10bed3f0456e4a9262ed23aa377238af9fa6297fabc1792aac15146c6ef20325
-  data.tar.gz: 5b7ac52a808a4a62b71e9bb603f14c92d5dadb897b897fde004e9dcc1cbb9fff
+  metadata.gz: e0b8c7bfd0c2fc9e55cce1f96ec6049b84bebddf51f0ff12ed4b37ff92f4707d
+  data.tar.gz: aea4e28ded2e9c0c949a05b0dea9906b23a790937f9957da7e1675827d9236e0
 SHA512:
-  metadata.gz: 9f9e69a4443e9f08cb98f4282df0b4787f13a07563e74debea467da6216bb0e7034f9661c68896f1083e3edad2c9d61c2aa01c11b5864ca7368a1dd9abcd5d48
-  data.tar.gz: c92d6b1a3c4ac8b6988a8b33192b44a41404cc0e122d13323e945d2a44ba08a39bc261aa33b3c5c99bd9bc45d83c1d24f0702a1412880bafa542ebf6346747e8
+  metadata.gz: 590bc72265e0ec354e41767107fa56468a17dd3c458dae461909c4c5460a3213f1620ae642c3db9c0bbbe61a4cf6c7866f056492c6aefbef329f558bf6cb315b
+  data.tar.gz: 22e5dbccd22117e3cb7efbf0fac45569c0c0d1e336859c34b91b4509b46672a47dbeea2115067eeb5e7e7adacb94faf8084bbf1a3e5ed5de6e9d12dd99ea53e5

data/Gemfile.lock

@@ -121,7 +121,7 @@ GEM
       diff-lcs (>= 1.2.0, < 2.0)
       rspec-support (~> 3.13.0)
     rspec-support (3.13.0)
-    rubocop (1.63.4)
+    rubocop (1.63.5)
       json (~> 2.3)
       language_server-protocol (>= 3.17.0)
       parallel (~> 1.10)

data/lib/rbs_goose/type_inferrer.rb

@@ -48,7 +48,7 @@ module RbsGoose
       end.chat_completion
     end
 
-    def llm_debug(prompt)
+    def llm_debug(prompt) # rubocop:disable Metrics/MethodLength
       return yield if ENV['DEBUG'].nil?
 
       puts "!!!!!!!! Prompt !!!!!!!!\n\n#{prompt}\n\n"
@@ -56,8 +56,20 @@ module RbsGoose
       sec = Benchmark.realtime do
         result = yield
       end
-      puts "!!!!!!!! Stats !!!!!!!!\n\n  spend: #{sec}[s]\n  prompt_tokens: #{result.prompt_tokens}\n  completion_tokens: #{result.completion_tokens}\n" # rubocop:disable Layout/LineLength
-      puts "!!!!!!!! Result !!!!!!!!\n\n#{result.chat_completion}\n\n"
+      puts "!!!!!!!! Stats !!!!!!!!\n\n  spend: #{sec}[s]"
+      # prompt_token はOllamaなどでサポートされないため、rescueして無視する
+      begin
+        puts "  prompt_tokens: #{result.prompt_tokens}\n  completion_tokens: #{result.completion_tokens}\n"
+      rescue NoMethodError
+        nil
+      end
+
+      # Ollama ではchat_completionではなくcompletionなのでrescueでフォールバックする
+      begin
+        puts "!!!!!!!! Result !!!!!!!!\n\n#{result.chat_completion}\n\n"
+      rescue NoMethodError
+        puts "!!!!!!!! Result !!!!!!!!\n\n#{result.completion}\n\n"
+      end
       result
     end
 

data/lib/rbs_goose/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RbsGoose
-  VERSION = '0.2.0'
+  VERSION = '0.2.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbs_goose
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - kokuyouwind
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-05-09 00:00:00.000000000 Z
+date: 2024-05-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday