From b3bec3cfb0ab75a5dc30dbd55b91e687a3588410 Mon Sep 17 00:00:00 2001 From: sigoden Date: Sun, 23 Jun 2024 23:09:47 +0800 Subject: refactor: cli options and shell completions (#640) --- scripts/completions/aichat.ps1 | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) (limited to 'scripts/completions/aichat.ps1') diff --git a/scripts/completions/aichat.ps1 b/scripts/completions/aichat.ps1 index ac582d7..dd0dc9c 100644 --- a/scripts/completions/aichat.ps1 +++ b/scripts/completions/aichat.ps1 @@ -28,8 +28,9 @@ Register-ArgumentCompleter -Native -CommandName 'aichat' -ScriptBlock { [CompletionResult]::new('-s', '-s', [CompletionResultType]::ParameterName, 'Start or join a session') [CompletionResult]::new('--session', '--session', [CompletionResultType]::ParameterName, 'Start or join a session') [CompletionResult]::new('--save-session', '--save-session', [CompletionResultType]::ParameterName, 'Forces the session to be saved') - [CompletionResult]::new('-b', '-b', [CompletionResultType]::ParameterName, 'Start a agent') + [CompletionResult]::new('-a', '-a', [CompletionResultType]::ParameterName, 'Start a agent') [CompletionResult]::new('--agent', '--agent', [CompletionResultType]::ParameterName, 'Start a agent') + [CompletionResult]::new('-R', '-R', [CompletionResultType]::ParameterName, 'Start a RAG') [CompletionResult]::new('--rag', '--rag', [CompletionResultType]::ParameterName, 'Start a RAG') [CompletionResult]::new('-f', '-f', [CompletionResultType]::ParameterName, 'Include files with the message') [CompletionResult]::new('--file', '--file', [CompletionResultType]::ParameterName, 'Include files with the message') @@ -70,17 +71,18 @@ Register-ArgumentCompleter -Native -CommandName 'aichat' -ScriptBlock { $offset=1 } $flag = $commandElements[$commandElements.Count-$offset].ToString() - if ($flag -eq "-m" -or $flag -eq "--model") { + dump-args $flag ($flag -eq "-R") > /tmp/file1 + if ($flag -ceq "-m" -or $flag -eq "--model") { $completions = Get-AichatValues "--list-models" - } elseif ($flag -eq "-r" -or $flag -eq "--role") { + } elseif ($flag -ceq "-r" -or $flag -eq "--role") { $completions = Get-AichatValues "--list-roles" - } elseif ($flag -eq "-s" -or $flag -eq "--session") { + } elseif ($flag -ceq "-s" -or $flag -eq "--session") { $completions = Get-AichatValues "--list-sessions" - } elseif ($flag -eq "-b" -or $flag -eq "--agent") { + } elseif ($flag -ceq "-a" -or $flag -eq "--agent") { $completions = Get-AichatValues "--list-agents" - } elseif ($flag -eq "--rag") { + } elseif ($flag -ceq "-R" -or $flag -eq "--rag") { $completions = Get-AichatValues "--list-rags" - } elseif ($flag -eq "-f" -or $flag -eq "--file") { + } elseif ($flag -ceq "-f" -or $flag -eq "--file") { $completions = @() } } -- cgit v1.2.3