Public/Enable-AIShortCutKey.ps1
|
function Enable-AIShortCutKey { <# .SYNOPSIS Enable a shortcut key for getting completions .DESCRIPTION Enables a shortcut key for getting completions. The -ShortcutKey can be customized. If not in Visual Studio Code, the ShortcutKey will default to 'CTRL+G' In Visual Studio Code, the ShortcutKey will default to 'ALT+G' .EXAMPLE # Enables the shortcut key. Outside of VSCode, CTRL+G. Inside of VSCode, ALT+G. Enable-AIShortCutKey .EXAMPLE Enable-AIShortCutKey -ShortcutKey "CTRL+ALT+P" #> param( [string] $ShortcutKey = $( # In Visual Studio Code, CTRL+G is "goto", if ((Get-Process -id $pid).Parent.ProcessName -eq 'code') { "ALT+G" # so we'll use ALT+G by default. } else { "CTRL+G" # If we're not running in code, use CTRL+G by default } ) ) Set-PSReadLineKeyHandler -Key $ShortcutKey ` -BriefDescription OpenAICli ` -LongDescription "Calls Open AI on the current buffer" ` -ScriptBlock { param($key, $arg) $line = $null $cursor = $null [Microsoft.PowerShell.PSConsoleReadLine]::GetBufferState([ref]$line, [ref]$cursor) $prompt = "Using PowerShell, just code: $($line)" $output = Get-GPT3Completion $prompt -max_tokens 256 $output = $output -replace "`r", "" # check if output is not null if ($null -ne $output) { foreach ($str in $output) { if ($null -ne $str -and $str -ne "") { [Microsoft.PowerShell.PSConsoleReadLine]::AddLine() [Microsoft.PowerShell.PSConsoleReadLine]::Insert($str) } } } } } |