Private/ArgumentCompletion.ps1
|
function ModelArgumentCompleter { param($commandName, $parameterName, $wordToComplete, $commandAst, $fakeBoundParameter) $models = $( "gpt-3.5-turbo-0125" "gpt-3.5-turbo-0613" "gpt-3.5-turbo-1106" "gpt-3.5-turbo-16k-0613" "gpt-3.5-turbo-16k" "gpt-3.5-turbo" "gpt-4-0125-preview" "gpt-4-0314" "gpt-4-0613" "gpt-4-1106-preview" "gpt-4-32k-0314" "gpt-4-32k-0613" "gpt-4-32k" "gpt-4-turbo-2024-04-09" "gpt-4-turbo-preview" "gpt-4-turbo" "gpt-4-vision-preview" "gpt-4" "gpt-4o-2024-05-13" "gpt-4o-mini-2024-07-18" "gpt-4o-mini" "gpt-4o" ) foreach ($model in $models) { New-Object -TypeName System.Management.Automation.CompletionResult -ArgumentList "'$model'", $model , 'ParameterValue' , $model } } if (Get-Command -ErrorAction SilentlyContinue -name Register-ArgumentCompleter) { $functionNames = $( 'Invoke-AIPrompt' 'Invoke-Chat' 'Invoke-OAIChat' 'Invoke-OAIChatCompletion' 'New-OAIAssistant' 'New-OAIRun' 'New-OpenAIChat' 'Update-OAIAssistant' ) foreach ($functionName in $functionNames) { Register-ArgumentCompleter -CommandName $functionName -ParameterName Model -ScriptBlock $Function:ModelArgumentCompleter } } |