File tree Expand file tree Collapse file tree 2 files changed +19
-12
lines changed
crates/q_cli/src/cli/chat Expand file tree Collapse file tree 2 files changed +19
-12
lines changed Original file line number Diff line number Diff line change @@ -1807,13 +1807,25 @@ where
18071807 ) ?;
18081808 } ,
18091809 Some ( ToolsSubcommand :: ResetSingle { tool_name } ) => {
1810- self . tool_permissions . reset_tool ( & tool_name) ;
1811- queue ! (
1812- self . output,
1813- style:: SetForegroundColor ( Color :: Green ) ,
1814- style:: Print ( format!( "\n Reset tool '{}' to the default permission level." , tool_name) ) ,
1815- style:: SetForegroundColor ( Color :: Reset ) ,
1816- ) ?;
1810+ if self . tool_permissions . has ( & tool_name) {
1811+ self . tool_permissions . reset_tool ( & tool_name) ;
1812+ queue ! (
1813+ self . output,
1814+ style:: SetForegroundColor ( Color :: Green ) ,
1815+ style:: Print ( format!( "\n Reset tool '{}' to the default permission level." , tool_name) ) ,
1816+ style:: SetForegroundColor ( Color :: Reset ) ,
1817+ ) ?;
1818+ } else {
1819+ queue ! (
1820+ self . output,
1821+ style:: SetForegroundColor ( Color :: Red ) ,
1822+ style:: Print ( format!(
1823+ "\n Tool '{}' does not exist or is already in default settings." ,
1824+ tool_name
1825+ ) ) ,
1826+ style:: SetForegroundColor ( Color :: Reset ) ,
1827+ ) ?;
1828+ }
18171829 } ,
18181830 Some ( ToolsSubcommand :: Help ) => {
18191831 queue ! (
Original file line number Diff line number Diff line change @@ -27,7 +27,6 @@ use fs_read::FsRead;
2727use fs_write:: FsWrite ;
2828use gh_issue:: GhIssue ;
2929use serde:: Deserialize ;
30- use tracing:: warn;
3130use use_aws:: UseAws ;
3231
3332use super :: parser:: ToolUse ;
@@ -196,10 +195,6 @@ impl ToolPermissions {
196195 }
197196
198197 pub fn reset_tool ( & mut self , tool_name : & str ) {
199- if !self . permissions . contains_key ( tool_name) {
200- warn ! ( "No custom permissions set for tool '{tool_name}' to reset" ) ;
201- return ;
202- }
203198 self . permissions . remove ( tool_name) ;
204199 }
205200
You can’t perform that action at this time.
0 commit comments