chat_gpt_error_handler 0.1.2 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3ff5d23745282dc3a2d6b01cd67e998e89f73074ae16ceb3e1ee63988b5361e0
-  data.tar.gz: 32ede69316cc4964c2497baab47e244d8e5fb8de13db2ff68d808edc71c589b3
+  metadata.gz: 8eb4d741fe11dd031e02e4405a131ba4a440a0306575a97246b6486de5f7ae3e
+  data.tar.gz: 589d87cf005f91a365b9399e99a9d45711dce60d9847500790395c3173f9b097
 SHA512:
-  metadata.gz: 33d465f4944e318d13766a19d9328cb8bdd914c3eb98aca2e491b02a94df38cff4bfa345ad68f157f3e3c8df3e81bb5d6b887cc70adf3427302ebfff91a08a7f
-  data.tar.gz: e1913f58570b9ac5e3f88dbb145bfcc89ff91f58929b25235a88c6f9ba8770c29e190ba5f4877bb45436ddb9db405beab964d477e2ab48f0e3fb77737a776af8
+  metadata.gz: ef2028789e012e4856f1397fcf98c8e2ee3c593a4d22dec620d05a1b20c539b3103e6892e8463e59e1f19f2bee4fa892f49035f3ec80f3818ecf7d7a68663795
+  data.tar.gz: cbf4c7bb9ebe54e539e543c36108387c089f9a2fb891968ddf0e7bf0f23a57c5a6d9eb179c3ba63a07da2603201a74e822fe74f1203eeac8648d158e2bf20477

data/README.md

@@ -12,11 +12,12 @@ Add this line to your application's Gemfile:
 gem "chat_gpt_error_handler"
 ```
 
-Set your [OpenAI API Token](https://openai.com/product#made-for-developers) in an initializer:
+Set your [OpenAI API Token](https://openai.com/product#made-for-developers) and enable the gem in an initializer:
 
 ```ruby
 # config/initializers/chat_gpt_error_handler.rb
 ChatGptErrorHandler.openai_access_token = 'your_openai_access_token_here'
+ChatGptErrorHandler.enabled = true
 ```
 
 And you're off!

data/lib/chat_gpt_error_handler/error_handler.rb

@@ -5,8 +5,10 @@ module ChatGptErrorHandler
   class ErrorHandler
     def initialize(app)
       @app = app
-      OpenAI.configure do |config|
-        config.access_token = ChatGptErrorHandler.openai_access_token || ENV.fetch('OPENAI_ACCESS_TOKEN')
+      if ChatGptErrorHandler.enabled
+        OpenAI.configure do |config|
+          config.access_token = ChatGptErrorHandler.openai_access_token || ENV.fetch('OPENAI_ACCESS_TOKEN')
+        end
       end
     end
 
@@ -14,7 +16,7 @@ module ChatGptErrorHandler
       begin
         @app.call(env)
       rescue => error
-        send_to_gpt(error)
+        send_to_gpt(error) if ChatGptErrorHandler.enabled
         raise error
       end
     end
@@ -43,29 +45,49 @@ module ChatGptErrorHandler
                 stop: nil
           }
         )
-        response_text = response["choices"].map { |c| c["text"].to_s }
 
-        response_text = response_text.join("")
-        shortened_response_text = []
-        begin
-          # try to format the response text to be 75 characters per line
-          response_text.split(" ").reduce("") do |line, word|
-            if line.length + word.length > 75
-              shortened_response_text << line
-              line = word
-            elsif word == response_text.split(" ").last
-              shortened_response_text << [line + " " + word]
-            else
-              line += (" " + word)
+
+        # Try a second time if the response is empty. Raise the second time
+        if response["choices"].nil?
+          response = client.completions(
+            parameters: {
+              model: "text-davinci-002",
+                  prompt: prompt,
+                  n: 1,
+                  temperature: 0.85,
+                  max_tokens: 250,
+                  stop: nil
+            }
+          )
+        end
+
+        if true #response["choices"].nil?
+          print_error_and_solution(error, "ChatGPT returned an empty response to your error message. You may need to try again.")
+        else
+          response_text = response["choices"].map { |c| c["text"].to_s }
+
+          response_text = response_text.join("")
+          shortened_response_text = []
+          begin
+            # try to format the response text to be 75 characters per line
+            response_text.split(" ").reduce("") do |line, word|
+              if line.length + word.length > 75
+                shortened_response_text << line
+                line = word
+              elsif word == response_text.split(" ").last
+                shortened_response_text << [line + " " + word]
+              else
+                line += (" " + word)
+              end
             end
+            shortened_response_text = shortened_response_text.join("\n")
+          rescue => e
+            # The formatting does not always work, so if it fails, just use the original response text
+            shortened_response_text = response_text
           end
-          shortened_response_text = shortened_response_text.join("\n")
-        rescue => e
-          # The formatting does not always work, so if it fails, just use the original response text
-          shortened_response_text = response_text
-        end
 
-        print_error_and_solution(error, shortened_response_text.strip)
+          print_error_and_solution(error, shortened_response_text.strip)
+        end
       rescue => gpt_error
         puts "\e[31mAn error occurred while communicating with GPT:\e[0m"
         puts gpt_error.message

data/lib/chat_gpt_error_handler/version.rb

@@ -1,3 +1,3 @@
 module ChatGptErrorHandler
-  VERSION = "0.1.2"
+  VERSION = "0.2.1"
 end

data/lib/chat_gpt_error_handler.rb

@@ -2,8 +2,11 @@ require "chat_gpt_error_handler/version"
 require "chat_gpt_error_handler/engine"
 
 module ChatGptErrorHandler
-    mattr_accessor :openai_access_token
+    mattr_accessor :openai_access_token, :enabled
 
   # Default value for the openai_access_token, used when not explicitly set by the user
   @@openai_access_token = nil
+
+  # Default value to allow opting-in
+  @@enabled = false
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: chat_gpt_error_handler
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.1
 platform: ruby
 authors:
 - Nick Schwaderer
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-05-17 00:00:00.000000000 Z
+date: 2023-06-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails