# 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",
  )


  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]                                                                
                                                                                               
╭─ Arguments ─────────────────────────────────────────────────────────────────────────────────╮
│   prompt      [PROMPT]  The prompt to generate completions for.                             │
╰─────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ───────────────────────────────────────────────────────────────────────────────────╮
│ --model                             TEXT                       Large language model to use. │
│                                                                [default: gpt-3.5-turbo]     │
│ --temperature                       FLOAT RANGE [0.0<=x<=2.0]  Randomness of generated      │
│                                                                output.                      │
│                                                                [default: 0.1]               │
│ --top-probability                   FLOAT RANGE [0.1<=x<=1.0]  Limits highest probable      │
│                                                                tokens (words).              │
│                                                                [default: 1.0]               │
│ --editor             --no-editor                               Open $EDITOR to provide a    │
│                                                                prompt.                      │
│                                                                [default: no-editor]         │
│ --cache              --no-cache                                Cache completion results.    │
│                                                                [default: cache]             │
│ --help                                                         Show this message and exit.  │
╰─────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Assistance Options ────────────────────────────────────────────────────────────────────────╮
│ --shell           -s                 Generate and execute shell commands.                   │
│ --describe-shell  -d                 Describe a shell command.                              │
│ --code                --no-code      Generate only code. [default: no-code]                 │
╰─────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Chat Options ──────────────────────────────────────────────────────────────────────────────╮
│ --chat                             TEXT  Follow conversation with id, use "temp" for quick  │
│                                          session.                                           │
│                                          [default: None]                                    │
│ --repl                             TEXT  Start a REPL (Read–eval–print loop) session.       │
│                                          [default: None]                                    │
│ --show-chat                        TEXT  Show all messages from provided chat id.           │
│                                          [default: None]                                    │
│ --list-chats    --no-list-chats          List all existing chat ids.                        │
│                                          [default: no-list-chats]                           │
╰─────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Role Options ──────────────────────────────────────────────────────────────────────────────╮
│ --role                              TEXT  System role for GPT model. [default: None]        │
│ --create-role                       TEXT  Create role. [default: None]                      │
│ --show-role                         TEXT  Show role. [default: None]                        │
│ --list-roles     --no-list-roles          List roles. [default: no-list-roles]              │
╰─────────────────────────────────────────────────────────────────────────────────────────────╯