ask_chatgpt 0.3.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 81b5ff785019505f049a92996ae9a8f2afbb6bda16375df14a9b101d110a4e3f
-  data.tar.gz: 0c4c037173a02db028500cf6dbafe67331b3f3b90b2bcd80327f3e0336e678e6
+  metadata.gz: 86f36d546bd015947fcac7a2bb9366530ba0ff8a9012f21f2151ae0484ecf647
+  data.tar.gz: 136c817bc54317c77325ca0145857262d48680b8b25072e3364dcd0fe526e501
 SHA512:
-  metadata.gz: e7d950e4d0cd28e23a37a1987ae6600d8627eea1974360ae138dadf556a9976ced78423afa69b6da94d5f7a6d7d6fac9c41d057d5f9c78b40419cc779a003e93
-  data.tar.gz: 3062aee654d3d3505af33762a2c781d02a02b6b50edd1539a6ab248914ffaac31169899c1eda184782683c14fbeacad361859ae4b2930f739d90055d71597ee6
+  metadata.gz: 7b3112557fcb6e3fa11856693e26dc37df80dedafde058f6f181d303a336f75cc4ab21f9c1daf89dd61d253eea150b05d5f664408ab5299c07210f37d8f9d87c
+  data.tar.gz: 4c5a22ca8157813e15fa5e2f639eefd89967ba9bbd300d6c0929bfe18cd3b2ff4e600197d59411520da05626c77ec9ca1786e2b6f3bfd7b51a2d5403f4528098

data/bin/ask_chatgpt

@@ -2,6 +2,7 @@
 
 require 'optparse'
 require 'active_support'
+require 'irb'
 
 require_relative "../lib/ask_chatgpt"
 
@@ -24,6 +25,8 @@ parser = OptionParser.new do |opts|
       ask_chatgpt -f app/models/user.rb -q "create RSpec spec for this model"
       ask_chatgpt -f test/dummy/Gemfile -q "sort Ruby gems alphabetically"
 
+    Version: #{AskChatGPT::VERSION}
+
   USAGE
 
   opts.on("-q", "--question \"Your Prompt\"", String, "Specify your prompt with full context, language, etc.") do |prompt|

data/lib/ask_chatgpt/console.rb

@@ -1,7 +1,7 @@
 module AskChatgpt
   module Console
     def gpt
-      AskChatGPT::Core.call
+      AskChatGPT::Executor.call
     end
 
     alias :chatgpt :gpt

data/lib/ask_chatgpt/executor.rb

@@ -18,6 +18,11 @@ module AskChatgpt
 
     attr_reader :scope, :client, :spinner, :cursor
 
+    def self.call
+      client = OpenAI::Client.new(access_token: AskChatGPT.access_token)
+      AskChatgpt::Executor.new(client)
+    end
+
     def initialize(client)
       @scope   = AskChatGPT.included_prompts.dup
       @client  = client
@@ -26,6 +31,9 @@ module AskChatgpt
     end
 
     def inspect
+      return if @executed
+      @executed = true
+
       pp(executor_parameters) if AskChatGPT.debug
       call_with_validations do
         case AskChatGPT.mode
@@ -35,11 +43,14 @@ module AskChatgpt
           call_sync
         end
       end
-    rescue InputError => e
+      nil
+    rescue SystemExit, SignalException, InputError, NoMethodError => e
       puts e.message
     rescue StandardError => e
       puts e.message
       puts e.backtrace.take(5).join("\n")
+    rescue Exception => e
+      puts e.message
     ensure
       nil
     end
@@ -68,18 +79,22 @@ module AskChatgpt
           print content_part
         end
       }))
+      spinner&.stop if spinner&.spinning?
       if AskChatGPT.markdown
+        result = content.compact.join
+        shift = result.split("\n").size == 1 ? 1 : result.split("\n").size + 1
         # re-draw the screen
         # go back to the top by the number of new lines previously printed
-        print cursor.clear_lines(content.compact.join.split("\n").size + 1, :up)
+        print cursor.clear_lines(shift, :up)
         # print cursor.restore
         # print cursor.down
         # print cursor.clear_screen_down
         # $content = content.compact.join
-        puts(TTY::Markdown.parse(content.compact.join))
+        puts(TTY::Markdown.parse(result))
       else
         # nothing, content is already printed in the stream
       end
+      nil
     end
 
     # wait for the whole response and print it at once
@@ -98,6 +113,7 @@ module AskChatgpt
           puts(content)
         end
       end
+      nil
     end
 
     def executor_parameters

data/lib/ask_chatgpt/helpers.rb

@@ -7,6 +7,7 @@ module AskChatgpt
           method_or_class_or_str.source
         when Module, Class, String
           str = capture_io do
+            init_irb unless IRB.CurrentContext
             IRB.CurrentContext.main.irb_show_source(method_or_class_or_str.to_s)
           end.join("\n")
           # check if source was extracted
@@ -22,6 +23,48 @@ module AskChatgpt
 
       private
 
+      class DummyInputMethod < ::IRB::InputMethod
+        attr_reader :list, :line_no
+
+        def initialize(list = [])
+          super("test")
+          @line_no = 0
+          @list = list
+        end
+
+        def gets
+          @list[@line_no]&.tap {@line_no += 1}
+        end
+
+        def eof?
+          @line_no >= @list.size
+        end
+
+        def encoding
+          Encoding.default_external
+        end
+
+        def reset
+          @line_no = 0
+        end
+      end
+
+      # https://github.com/ruby/irb/blob/95782f2cf93b4b32e51104a21a860aa4491a04bc/test/irb/test_cmd.rb#L36
+      def init_irb
+        conf = {}
+        main = :self
+        irb_path = nil
+        IRB.init_config(nil)
+        IRB.conf[:VERBOSE] = false
+        IRB.conf[:PROMPT_MODE] = :SIMPLE
+        IRB.conf.merge!(conf)
+        input = DummyInputMethod.new("TestInputMethod#gets")
+        irb = IRB::Irb.new(IRB::WorkSpace.new(main), input)
+        irb.context.return_format = "=> %s\n"
+        irb.context.irb_path = irb_path if irb_path
+        IRB.conf[:MAIN_CONTEXT] = irb.context
+      end
+
       # from  https://github.com/minitest/minitest/blob/master/lib/minitest/assertions.rb#L542
       def capture_io
         _synchronize do

data/lib/ask_chatgpt/version.rb

@@ -1,3 +1,3 @@
 module AskChatgpt
-  VERSION = "0.3.0"
+  VERSION = "0.3.1"
 end

data/lib/ask_chatgpt.rb

@@ -9,11 +9,11 @@ require "openai"
 require "tty-markdown"
 require "tty-spinner"
 require "tty-cursor"
+require "irb"
 
 require_relative "ask_chatgpt/console"
 require_relative "ask_chatgpt/executor"
 require_relative "ask_chatgpt/helpers"
-require_relative "ask_chatgpt/core"
 
 module AskChatgpt
   ::AskChatGPT = AskChatgpt

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ask_chatgpt
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.3.1
 platform: ruby
 authors:
 - Igor Kasyanchuk
@@ -9,7 +9,7 @@ authors:
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-29 00:00:00.000000000 Z
+date: 2023-05-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -138,7 +138,6 @@ files:
 - bin/ask_chatgpt
 - lib/ask_chatgpt.rb
 - lib/ask_chatgpt/console.rb
-- lib/ask_chatgpt/core.rb
 - lib/ask_chatgpt/default_behavior.rb
 - lib/ask_chatgpt/executor.rb
 - lib/ask_chatgpt/helpers.rb
@@ -161,11 +160,11 @@ files:
 - lib/generators/ask_chatgpt/ask_chatgpt_generator.rb
 - lib/generators/ask_chatgpt/templates/template.rb
 - lib/tasks/ask_chatgpt_tasks.rake
-homepage: https://github.com/railsjazz.com/ask_chatgpt
+homepage: https://github.com/railsjazz/ask_chatgpt
 licenses:
 - MIT
 metadata:
-  homepage_uri: https://github.com/railsjazz.com/ask_chatgpt
+  homepage_uri: https://github.com/railsjazz/ask_chatgpt
 post_install_message:
 rdoc_options: []
 require_paths:

data/lib/ask_chatgpt/core.rb

@@ -1,8 +0,0 @@
-module AskChatgpt
-  class Core
-    def self.call
-      client = OpenAI::Client.new(access_token: AskChatGPT.access_token)
-      AskChatgpt::Executor.new(client)
-    end
-  end
-end