Functions/GenXdev.AI/Invoke-AIPowershellCommand.ps1
|
function Set-AICommandSuggestion { param( [Parameter(Mandatory)] [string]$Command ) return @{ command = $Command.Trim() success = $true } | ConvertTo-Json } ################################################################################ <# .SYNOPSIS Generates and executes PowerShell commands using AI assistance. .DESCRIPTION Uses LM-Studio to generate PowerShell commands based on natural language queries and either sends them to the current PowerShell window or copies to clipboard. .PARAMETER Query The natural language query to generate a PowerShell command. .PARAMETER Model The LM-Studio model to use for generating the response. Defaults to "qwen". .PARAMETER Temperature Controls randomness in the response (0.0-1.0). Lower values are more focused. .PARAMETER Clipboard Switch to copy the generated command to clipboard instead of executing it. .EXAMPLE Invoke-AIPowershellCommand -Query "list all running processes" -Model "qwen" .EXAMPLE hint "list files modified today" #> function Invoke-AIPowershellCommand { [CmdletBinding()] [Alias("hint")] param ( ######################################################################## [Parameter( Position = 0, Mandatory = $true, ValueFromPipeline = $true, HelpMessage = "The query string for the LLM." )] [ValidateNotNullOrEmpty()] [string] $Query, ######################################################################## [Parameter( Position = 1, Mandatory = $false, HelpMessage = "The LM-Studio model to use for generating the response" )] [PSDefaultValue(Value = "qwen")] [string] $Model = "qwen", ######################################################################## [Parameter( Position = 2, Mandatory = $false, HelpMessage = "Temperature parameter for response randomness" )] [ValidateRange(0.0, 1.0)] [double] $Temperature = 0.01, ######################################################################## [Parameter( Mandatory = $false, HelpMessage = "Only set clipboard" )] [switch] $Clipboard ######################################################################## ) begin { Write-Verbose "Initializing AI command generation with model: $Model" # simplified instructions focusing on command generation $instructions = "You are a PowerShell expert. Analyze the user's request and suggest a PowerShell command that accomplishes their goal. Use the Get-GenXDevCmdlets and the Get-Command cmdlets to find the right commands to use. Use the Set-AICommandSuggestion function with the command you suggest. return only what Set-AICommandSuggestion returns in json format. " } process { Write-Verbose "Generating PowerShell command for query: $Query" # generate command using LM-Studio with exposed cmdlet $result = Invoke-LMStudioQuery ` -Query $Query ` -Model $Model ` -Temperature $Temperature ` -Instructions $instructions ` -ExposedCmdLets (Get-Command Set-AICommandSuggestion, Get-GenXDevCmdlets, Get-Help) ` -NoConfirmationFor @("Set-AICommandSuggestion", "Get-GenXDevCmdlets", "Get-Help") | ` ConvertFrom-Json # extract command from result $command = ($result | ConvertFrom-Json).Command if (-not $command.success) { Write-Warning "Failed to generate command: $command" return } if ($Clipboard) { # copy to clipboard $command | Set-Clipboard Write-Verbose "Command copied to clipboard" } else { # get main window and send command $window = Get-PowershellMainWindow if ($null -ne $window) { $window.SetForeground() } Send-Keys ("`r`n$command".Replace("`r`n", " ``+{ENTER}")) } } } ################################################################################ |