lib/aia/tools/sgpt.rb in aia-0.3.20 vs lib/aia/tools/sgpt.rb in aia-0.4.1

@@ -1,19 +1,35 @@
 # lib/aia/tools/sgpt.rb
 
+require_relative 'backend_common'
+
 class AIA::Sgpt < AIA::Tools
+  include AIA::BackendCommon
 
   meta(
     name:     'sgpt',
     role:     :backend,
     desc:     "shell-gpt",
     url:      "https://github.com/TheR1D/shell_gpt",
     install:  "pip install shell-gpt",
   )
 
-  def initialize
-    # TODO: something
-  end
+
+  DEFAULT_PARAMETERS = [
+    # "--verbose",          # enable verbose logging (if applicable)
+    # Add default parameters here
+  ].join(' ').freeze
+
+  DIRECTIVES = %w[
+    model
+    temperature
+    max_tokens
+    top_p
+    frequency_penalty
+    presence_penalty
+    stop_sequence
+    api_key
+  ]
 end
 
 __END__
                                                                                                
  Usage: sgpt [OPTIONS] [PROMPT]