pwn 0.5.210 → 0.5.211

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 361b33380a6f528dc2cc78a4cd403b5a9114d319237e6703584f7ea325fe22a1
-  data.tar.gz: 03e63ef8cba3b0a0b77425b8e25d45292c8f660bc4ff6568ce28566cbf7d7a7e
+  metadata.gz: b042b1ea3bd785b3e052abd01f519f3245ff44775bbcf96d69a1474f5396e611
+  data.tar.gz: 30cbb42b239c41b99a4ef4bb0968ebbb774075687c4930429f440d11b23c18d8
 SHA512:
-  metadata.gz: 7143125cec3c6e0a3b594070aff1d33e6047727cd5fb2e3e8f79ef49ed7136cba384a1150bb32084e4bdd8bd88fed222b21196f27c90db35c99d7695a04b13fe
-  data.tar.gz: 59739d8a5df55eb280222fb49004043fb299d569ef9cb77769f7402ce3a9b6f56de81faead1f0a8b305138042b48bd6fdb4efcb0923882ebd9cdb3d15fa06b2d
+  metadata.gz: 55a115524f4642a6bb13434fd1b8821911ee05db968dc2a1002f77670d83f609186f012d5120994f1d1dafa1528eade6ff6baed01fbcc82bdcde40970e4cc9a5
+  data.tar.gz: 78f9b964a0dbea14f7590241afcf817fda9c161be87d78c33dbddc83dd48d087dc8ddb9a9be40600e5ba8b560fa911fb5397a5e5456e026f8296d0b77da2390d

data/lib/pwn/plugins/repl.rb

@@ -636,24 +636,28 @@ module PWN
             else
               raise "ERROR: Unsupported AI Engine: #{ai_engine}"
             end
+            # puts response.inspect
 
-            # puts response[:choices].inspect
             last_response = ''
-            if response[:choices].last.keys.include?(:text)
-              last_response = response[:choices].last[:text]
+            if response.nil?
+              last_response = "Model: #{model} not currently supported with API key."
             else
-              last_response = response[:choices].last[:content]
+              if response[:choices].last.keys.include?(:text)
+                last_response = response[:choices].last[:text]
+              else
+                last_response = response[:choices].last[:content]
+              end
+
+              response_history = {
+                id: response[:id],
+                object: response[:object],
+                model: response[:model],
+                usage: response[:usage]
+              }
+              response_history[:choices] ||= response[:choices]
             end
             puts "\n\001\e[32m\002#{last_response}\001\e[0m\002\n\n"
 
-            response_history = {
-              id: response[:id],
-              object: response[:object],
-              model: response[:model],
-              usage: response[:usage]
-            }
-            response_history[:choices] ||= response[:choices]
-
             if debug
               puts 'DEBUG: response_history => '
               pp response_history

data/lib/pwn/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module PWN
-  VERSION = '0.5.210'
+  VERSION = '0.5.211'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.5.210
+  version: 0.5.211
 platform: ruby
 authors:
 - 0day Inc.
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-26 00:00:00.000000000 Z
+date: 2024-09-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport