pwn 0.4.618 → 0.4.620

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 46b245ad4f639c970cf5840e137c9d55deb5d6d79d46b423a9cec6c84048237d
-  data.tar.gz: '082bff94ad45ee0891caaab8ce17b0e17e9f95180eacfaedb7ea8f62800e901b'
+  metadata.gz: fafd153cff40ac71671f4350a1b00459b355b9648797f013c206272d42fa1d7e
+  data.tar.gz: 1915f04c0ee44c2e71a3b90208e3bffed04a2445a2bba3151900fb36fd46ad6c
 SHA512:
-  metadata.gz: 6aadac951190c38f228965ccff85b3af1c7e87053173ba0965ed39df4a96143ec398a45eeaad9cfc5ba1a064dab349a0f6965c6e1a2d4cdc0593df95662aa374
-  data.tar.gz: e0b8483c95d6c1717dcce9127f5575aabd88aeb5bfad5760a920f559f0b50cc7135c358da2e827db7b6e8ff16cbfec702e09cd21488e597db16101a0185598dd
+  metadata.gz: ff8533d123cd6ba697cd452d8526d11c82bed093f28cb2d5146396ebbf4df8b56ffffe343e32770f36cd9a12627b7a174d8b6bac8cc4ca4c569039a60e7c7dba
+  data.tar.gz: c4d482bd29938edf14ca98510ebc3885563f55000127c8a01d252a744dc6b671d4419bf89876dd3bafcfc8779c92b925785faeb605139d22b5c5ddb8501afbf3

data/Gemfile

@@ -11,7 +11,7 @@ gemspec
 # In some circumstances custom flags are passed to gems in order
 # to build appropriately.  Defer to ./reinstall_pwn_gemset.sh
 # to review these custom flags (e.g. pg, serialport, etc).
-gem 'activesupport', '7.0.4.2'
+gem 'activesupport', '7.0.4.3'
 gem 'anemone', '0.7.2'
 gem 'authy', '3.0.1'
 gem 'aws-sdk', '3.1.0'

data/README.md

@@ -37,7 +37,7 @@ $ rvm use ruby-ruby-3.2.1@pwn
 $ rvm list gemsets
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.4.618]:001 >>> PWN.help
+pwn[v0.4.620]:001 >>> PWN.help
 ```
 
 [![Installing the pwn Security Automation Framework](https://raw.githubusercontent.com/0dayInc/pwn/master/documentation/pwn_install.png)](https://youtu.be/G7iLUY4FzsI)
@@ -52,7 +52,7 @@ $ rvm use ruby-ruby-3.2.1@pwn
 $ gem uninstall --all --executables pwn
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.4.618]:001 >>> PWN.help
+pwn[v0.4.620]:001 >>> PWN.help
 ```
 
 

data/lib/pwn/plugins/open_ai.rb

@@ -91,33 +91,55 @@ module PWN
       end
 
       # Supported Method Parameters::
-      # response = PWN::Plugins::OpenAI.default_chat(
+      # response = PWN::Plugins::OpenAI.chat(
       #   token: 'required - Bearer token',
       #   request: 'required - message to ChatGPT'
-      #   model: 'optional - model to use for text generation (defaults to text-davinci-003)',
+      #   model: 'optional - model to use for text generation (defaults to gpt-3.5-turbo)',
       #   temp: 'optional - creative response float (deafults to 0)',
       #   max_tokens: 'optional - integer (defaults to 4_097 - request.length || 300)'
       # )
 
-      public_class_method def self.default_chat(opts = {})
+      public_class_method def self.chat(opts = {})
         token = opts[:token]
         request = opts[:request]
         model = opts[:model]
-        model ||= 'text-davinci-003'
+        model ||= 'gpt-3.5-turbo'
+        # model ||= 'text-davinci-003'
         temp = opts[:temp].to_f
         temp = 0 unless temp.positive?
         max_tokens = opts[:max_tokens].to_i
         max_tokens = 4_097 - request.to_s.length
         max_tokens = 300 unless max_tokens.positive?
 
-        rest_call = 'completions'
-
-        http_body = {
-          model: model,
-          prompt: request,
-          temperature: temp,
-          max_tokens: max_tokens
-        }
+        case model
+        when 'text-davinci-002',
+             'text-davinci-003'
+
+          rest_call = 'completions'
+
+          http_body = {
+            model: model,
+            prompt: request,
+            temperature: temp,
+            max_tokens: max_tokens
+          }
+        when 'gpt-3.5-turbo',
+             'gpt-4'
+
+          rest_call = 'chat/completions'
+
+          http_body = {
+            model: model,
+            messages: [
+              role: 'system',
+              content: request
+            ],
+            temperature: temp,
+            max_tokens: max_tokens
+          }
+        else
+          raise "ERROR: #{model} not supported."
+        end
 
         response = open_ai_rest_call(
           http_method: :post,
@@ -179,11 +201,11 @@ module PWN
 
       public_class_method def self.help
         puts "USAGE:
-          response = #{self}.default_chat(
+          response = #{self}.chat(
             token: 'required - Bearer token',
             request: 'required - message to ChatGPT',
-            model: 'optional - model to use for text generation (defaults to text-davinci-003)',
-            temp: 'optional - creative response float (deafults to 0)',
+            model: 'optional - model to use for text generation (defaults to gpt-3.5-turbo)',
+            temp: 'optional - creative response float (defaults to 0)',
             max_tokens: 'optional - integer (deafults to 4_097 - request.length || 300)'
           )
 

data/lib/pwn/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.4.618
+  version: 0.4.620
 platform: ruby
 authors:
 - 0day Inc.
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-13 00:00:00.000000000 Z
+date: 2023-03-14 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 7.0.4.2
+        version: 7.0.4.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - '='
       - !ruby/object:Gem::Version
-        version: 7.0.4.2
+        version: 7.0.4.3
 - !ruby/object:Gem::Dependency
   name: anemone
   requirement: !ruby/object:Gem::Requirement