pwn 0.5.217 → 0.5.219

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5fcd08cace1206ca512f8505dc2b697164a92f518143fc3510f555963c02f44a
-  data.tar.gz: f540dd48550734a5c7ee259bd6e6451f40a288afaffb3431bd23c6295f9182e0
+  metadata.gz: 58a55be13ad9968ee1a2ec11d4d6a4f83fa5d690f4742c1cdcdf35ca23513bfa
+  data.tar.gz: 418f44ce4b2bfec5b0cdacd073f19fb616c4fcc8e268fc1ed448b18a18384f21
 SHA512:
-  metadata.gz: 19056325157ea846b3717a610038b0232ab6679073a79d1804ce078dfbae7e2931f9233aa763c74d307ae3b19fd1cd862715ec91dbb9182ddcc93aa439e9fd23
-  data.tar.gz: e5ad3899be169546126534ca45b5cc3e52a244a1fcb0d52571b047fcad3242d6504f58a55c3bb1309623417918b285861c4b17d7de109be2d73715b85971b33f
+  metadata.gz: aa7ab4d1286a3bbd0c842bddc1ffe90ad78d36036565460b6b8aa861de325958ac5ff6f77ead67071b109c6a6ebee040e3b8a4469a71fb55365b6364b05aaa51
+  data.tar.gz: 6f5fe523e33592e80e2da2117890838cb0ca89abe67a2e0432552760e7328f337adb1cddb4e53c7098478fa5f80fb7f8b198dfc2af72723a6baecad3bd77b3d3

data/README.md

@@ -37,7 +37,7 @@ $ cd /opt/pwn
 $ ./install.sh
 $ ./install.sh ruby-gem
 $ pwn
-pwn[v0.5.217]:001 >>> PWN.help
+pwn[v0.5.219]: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.3.5@pwn
 $ gem uninstall --all --executables pwn
 $ gem install --verbose pwn
 $ pwn
-pwn[v0.5.217]:001 >>> PWN.help
+pwn[v0.5.219]:001 >>> PWN.help
 ```
 
 If you're using a multi-user install of RVM do:
@@ -62,7 +62,7 @@ $ rvm use ruby-3.3.5@pwn
 $ rvmsudo gem uninstall --all --executables pwn
 $ rvmsudo gem install --verbose pwn
 $ pwn
-pwn[v0.5.217]:001 >>> PWN.help
+pwn[v0.5.219]:001 >>> PWN.help
 ```
 
 PWN periodically upgrades to the latest version of Ruby which is reflected in `/opt/pwn/.ruby-version`.  The easiest way to upgrade to the latest version of Ruby from a previous PWN installation is to run the following script:

data/bin/pwn_rdoc_to_jsonl

@@ -61,10 +61,15 @@ begin
       # puts "Desc: #{method_description.inspect}"
       # gets
 
-      prompt = "#{module_name}.#{method_name} Usage"
-      completion = "`#{module_name}.#{method_name}`: #{method_description}"
-
-      jsonl_hash = { prompt: prompt, completion: completion }
+      user_content = "#{module_name}.#{method_name} Usage"
+      assistant_content = "`#{module_name}.#{method_name}`: #{method_description}"
+
+      jsonl_hash = {
+        messages: [
+          { role: 'user', content: user_content },
+          { role: 'assistant', content: assistant_content }
+        ]
+      }
       # puts "JSONL: #{jsonl}"
       # gets
 

data/lib/pwn/plugins/open_ai.rb

@@ -448,17 +448,14 @@ module PWN
         token = opts[:token]
         training_file = opts[:training_file]
         validation_file = opts[:validation_file]
-        model = opts[:model]
-        model ||= :davinci
+        model = opts[:model] ||= 'gpt-4o-mini'
 
-        n_epochs = opts[:n_epochs]
-        n_epochs ||= 4
+        n_epochs = opts[:n_epochs] ||= 4
 
         batch_size = opts[:batch_size]
         learning_rate_multipler = opts[:learning_rate_multipler]
 
-        prompt_loss_weight = opts[:prompt_loss_weight]
-        prompt_loss_weight ||= 0.01
+        prompt_loss_weight = opts[:prompt_loss_weight] ||= 0.01
 
         computer_classification_metrics = true if opts[:computer_classification_metrics]
         classification_n_classes = opts[:classification_n_classes]
@@ -498,7 +495,7 @@ module PWN
         response = open_ai_rest_call(
           http_method: :post,
           token: token,
-          rest_call: 'fine-tunes',
+          rest_call: 'fine_tuning/jobs',
           http_body: http_body,
           timeout: timeout
         )
@@ -520,7 +517,7 @@ module PWN
 
         response = open_ai_rest_call(
           token: token,
-          rest_call: 'fine-tunes',
+          rest_call: 'fine_tuning/jobs',
           timeout: timeout
         )
 
@@ -541,7 +538,7 @@ module PWN
         fine_tune_id = opts[:fine_tune_id]
         timeout = opts[:timeout]
 
-        rest_call = "fine-tunes/#{fine_tune_id}"
+        rest_call = "fine_tuning/jobs/#{fine_tune_id}"
 
         response = open_ai_rest_call(
           token: token,
@@ -566,7 +563,7 @@ module PWN
         fine_tune_id = opts[:fine_tune_id]
         timeout = opts[:timeout]
 
-        rest_call = "fine-tunes/#{fine_tune_id}/cancel"
+        rest_call = "fine_tuning/jobs/#{fine_tune_id}/cancel"
 
         response = open_ai_rest_call(
           http_method: :post,
@@ -592,7 +589,7 @@ module PWN
         fine_tune_id = opts[:fine_tune_id]
         timeout = opts[:timeout]
 
-        rest_call = "fine-tunes/#{fine_tune_id}/events"
+        rest_call = "fine_tuning/jobs/#{fine_tune_id}/events"
 
         response = open_ai_rest_call(
           token: token,
@@ -665,8 +662,7 @@ module PWN
         file = opts[:file]
         raise "ERROR: #{file} not found." unless File.exist?(file)
 
-        purpose = opts[:purpose]
-        purpose ||= 'fine-tune'
+        purpose = opts[:purpose] ||= 'fine-tune'
 
         timeout = opts[:timeout]
 

data/lib/pwn/version.rb

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