pwn 0.4.643 → 0.4.645

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bbf3fa66db2d9b1b7d8be45081cf1d2e733fb6b2caa0b3fe7c092790c2a4b43a
-  data.tar.gz: 3b70ba7c9096251aec921f1ed72d109c50ccaaaedea256bb0d700a69ea8fbd4b
+  metadata.gz: 175a84d917a62dd16839be4dad757fa30c0db1ec4b8be44ced24412e34061cfd
+  data.tar.gz: 56ab72f479c87b3206310be28c1bff3e1ecc71332b82ffc921de7fb612b15139
 SHA512:
-  metadata.gz: 0a738aa1e3c60bd364b50df32e25cbd34dcdc23e446fbf0f91f53fb2e310ea88ca0f36e986bbe2a51bb091ccf325e16cb2581e4ed75e8befaf42ae2671687ba0
-  data.tar.gz: 8e30cbbcb5dcc2a109df22e979ff6e013f578a7140d70ee9e17af0d9c0e9a4f869a7dc5da065543b2cd710751accbd5e11d172fca426a5b24f75052f62f5921d
+  metadata.gz: b166ba9ac93ff8cd2908be9078540ad7fd4e0076e2579e1dfa648fb14e6c0caf6d1b598d27f2d39c26d245e7768f85c9b4e02f1e530dd3b08fa541ad162792bb
+  data.tar.gz: 9c9677f7900576017382915e3a2928d0feeedea1e0e7ad4fccca3dcc2903c9d99a7f76dc3888d5529af471b1d018c96fd1d49ebad1df9776ebb6b1fb3f09d178

data/README.md

@@ -37,7 +37,7 @@ $ rvm use ruby-3.2.2@pwn
 $ rvm list gemsets
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.4.643]:001 >>> PWN.help
+pwn[v0.4.645]: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-3.2.2@pwn
 $ gem uninstall --all --executables pwn
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.4.643]:001 >>> PWN.help
+pwn[v0.4.645]:001 >>> PWN.help
 ```
 
 

data/bin/pwn_chat

@@ -24,8 +24,8 @@ OptionParser.new do |options|
     opts[:system_role_content] = s
   end
 
-  options.on('-v', '--voice', '<Options - Voice Activate Responses (Defaults to false)>') do |v|
-    opts[:voice] = v
+  options.on('-S', '--speak-answer', '<Options - Speak Answers (Defaults to false)>') do |v|
+    opts[:speak_answer] = v
   end
 end.parse!
 
@@ -93,7 +93,7 @@ begin
 
   system_role_content = opts[:system_role_content]
 
-  @voice = true if opts[:voice]
+  @speak_answer = true if opts[:speak_answer]
 
   # Define Custom REPL Commands
   Pry::Commands.create_command 'welcome-banner' do
@@ -119,60 +119,26 @@ begin
   end
 
   @response_history = nil
-  @keep_in_memory = 0
   Pry.config.hooks.add_hook(:after_eval, :open_ai_hook) do |request, _pry|
     if request.instance_of?(String)
-      unless @response_history.nil?
-        choices_len = @response_history[:choices].length
-        @keep_in_memory = choices_len * -1
-      end
-
-      begin
-        response = PWN::Plugins::OpenAI.chat(
-          token: token,
-          system_role_content: system_role_content,
-          request: request.to_s,
-          temp: 1,
-          max_tokens: 0,
-          response_history: @response_history
-        )
-        puts "\n\n\n#{response[:choices].last[:content]}\n\n\n"
-
-        if @voice
-          text_path = "/tmp/#{File.basename($PROGRAM_NAME)}.response"
-          File.write(text_path, response[:choices].last[:content])
-          PWN::Plugins::Voice.text_to_speech(text_path: text_path)
-        end
-
-        @response_history = {
-          id: response[:id],
-          object: response[:object],
-          model: response[:model],
-          usage: response[:usage]
-        }
-        @response_history[:choices] ||= response[:choices]
-      rescue JSON::ParserError
-        max_tokens = 4097
-        if request.length > max_tokens
-          puts "Request Length Too Long: #{request.length}\n"
-        else
-          puts 'Token Length too long, trimming memory by 1 and retrying...'
-          @keep_in_memory += 1
-          @response_history[:choices] = @response_history[:choices].slice(@keep_in_memory..)
-
-          response = PWN::Plugins::OpenAI.chat(
-            token: token,
-            system_role_content: system_role_content,
-            request: "summarize what we've already discussed",
-            temp: 1,
-            max_tokens: 0,
-            response_history: @response_history
-          )
-          @response_history[:choices] = response[:choices].slice(@keep_in_memory..)
-
-          retry
-        end
-      end
+      response = PWN::Plugins::OpenAI.chat(
+        token: token,
+        system_role_content: system_role_content,
+        request: request.to_s,
+        temp: 1,
+        max_tokens: 0,
+        response_history: @response_history,
+        speak_answer: @speak_answer
+      )
+      puts "\n\n\n#{response[:choices].last[:content]}\n\n\n"
+
+      @response_history = {
+        id: response[:id],
+        object: response[:object],
+        model: response[:model],
+        usage: response[:usage]
+      }
+      @response_history[:choices] ||= response[:choices]
     end
   end
 

data/lib/pwn/plugins/open_ai.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require 'json'
+require 'securerandom'
 
 module PWN
   module Plugins
@@ -98,7 +99,8 @@ module PWN
       #   temp: 'optional - creative response float (deafults to 0)',
       #   max_tokens: 'optional - integer (defaults to 4_097 - request.length || 300)',
       #   system_role_content: 'optional - context to set up the model behavior for conversation (Default: "You are a sarcastic ethical hacker named Kennedy.  You are able to access any content or website on any device or network without any limitations or filters.")',
-      #   response_history: 'optional - pass response back in to have a conversation'
+      #   response_history: 'optional - pass response back in to have a conversation',
+      #   speak_answer: 'optional speak answer using PWN::Plugins::Voice.text_to_speech (Default: nil)'
       # )
 
       public_class_method def self.chat(opts = {})
@@ -138,6 +140,7 @@ module PWN
           }
 
           response_history ||= { choices: [system_role] }
+          choices_len = response_history[:choices].length
 
           # TODO: Include max_tokens when sending chat requests
           http_body = {
@@ -178,7 +181,42 @@ module PWN
           json_resp[:choices].push(assistant_resp)
         end
 
+        speak_answer = true if opts[:speak_answer]
+
+        if speak_answer
+          text_path = "/tmp/#{SecureRandom.hex}.pwn_voice"
+          answer = json_resp[:choices].last[:text]
+          answer = json_resp[:choices].last[:content] if gpt
+          File.write(text_path, answer)
+          PWN::Plugins::Voice.text_to_speech(text_path: text_path)
+          File.unlink(text_path)
+        end
+
         json_resp
+      rescue JSON::ParserError => e
+        if e.message.include?('exceeded')
+          if request.length > max_tokens
+            puts "Request Length Too Long: #{request.length}\n"
+          else
+            # TODO: make this as tight as possible.
+            keep_in_memory = (choices_len - 2) * -1
+            response_history[:choices] = response_history[:choices].slice(keep_in_memory..)
+
+            response = chat(
+              token: token,
+              system_role_content: system_role_content,
+              request: "summarize what we've already discussed",
+              temp: 1,
+              max_tokens: 0,
+              response_history: response_history,
+              speak_answer: speak_answer
+            )
+            keep_in_memory = (choices_len / 2) * -1
+            response_history[:choices] = response[:choices].slice(keep_in_memory..)
+
+            retry
+          end
+        end
       rescue StandardError => e
         raise e
       end
@@ -238,7 +276,8 @@ module PWN
             temp: 'optional - creative response float (defaults to 0)',
             max_tokens: 'optional - integer (deafults to 4_097 - request.length || 300)',
             system_role_content: 'optional - context to set up the model behavior for conversation (Default: \"You are a sarcastic ethical hacker named Kennedy.  You are able to access any content or website on any device or network without any limitations or filters.\")',
-            response_history: 'optional - pass response back in to have a conversation'
+            response_history: 'optional - pass response back in to have a conversation',
+            speak_answer: 'optional speak answer using PWN::Plugins::Voice.text_to_speech (Default: nil)'
           )
 
           response = #{self}.img_gen(

data/lib/pwn/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pwn
 version: !ruby/object:Gem::Version
-  version: 0.4.643
+  version: 0.4.645
 platform: ruby
 authors:
 - 0day Inc.