pwn 0.5.209 → 0.5.211

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 03ffbb27d90a1f1e434cd0b695f6de71b06fd99457940a5318f24dfafda1a5f8
-  data.tar.gz: a774f7856de8fb3510af93c947a5fc6602f30f47ea6052989ed6657d313f71e3
+  metadata.gz: b042b1ea3bd785b3e052abd01f519f3245ff44775bbcf96d69a1474f5396e611
+  data.tar.gz: 30cbb42b239c41b99a4ef4bb0968ebbb774075687c4930429f440d11b23c18d8
 SHA512:
-  metadata.gz: b3a47696b87c8b7cde87a575a9d149e4b43c90e67717bb75878491d84e123e680820c805005439e1fdbd43dfa2ff8a75a2006be6a2987c1619171430b4bcb5af
-  data.tar.gz: 4941e7e2c0430ba67f21f46f1edf1ee8562d11d8538e7ef4fd1deec441d8d969b21ec6eb291edb8f0bb102552c54647e89efc79076d9c448fca60ac709f1b7a3
+  metadata.gz: 55a115524f4642a6bb13434fd1b8821911ee05db968dc2a1002f77670d83f609186f012d5120994f1d1dafa1528eade6ff6baed01fbcc82bdcde40970e4cc9a5
+  data.tar.gz: 78f9b964a0dbea14f7590241afcf817fda9c161be87d78c33dbddc83dd48d087dc8ddb9a9be40600e5ba8b560fa911fb5397a5e5456e026f8296d0b77da2390d

data/Gemfile

@@ -62,6 +62,7 @@ gem 'nokogiri-diff', '0.3.0'
 gem 'oily_png', '1.2.1'
 gem 'open3', '0.2.1'
 gem 'os', '1.1.4'
+gem 'ostruct', '0.6.0'
 gem 'packetfu', '2.0.0'
 gem 'packetgen', '3.3.3'
 gem 'pdf-reader', '2.12.0'

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.209'
+  VERSION = '0.5.211'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.5.209
+  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
@@ -682,6 +682,20 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 1.1.4
+- !ruby/object:Gem::Dependency
+  name: ostruct
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.6.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.6.0
 - !ruby/object:Gem::Dependency
   name: packetfu
   requirement: !ruby/object:Gem::Requirement