diff options
Diffstat (limited to 'src/repl')
| -rw-r--r-- | src/repl/mod.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/repl/mod.rs b/src/repl/mod.rs index f40bdd0..4482b84 100644 --- a/src/repl/mod.rs +++ b/src/repl/mod.rs @@ -63,7 +63,7 @@ lazy_static::lazy_static! { ReplCommand::new( ".exit role", "Leave the role", - AssertState::True(StateFlags::ROLE), + AssertState::TrueFalse(StateFlags::ROLE, StateFlags::SESSION), ), ReplCommand::new( ".session", @@ -108,7 +108,7 @@ lazy_static::lazy_static! { ReplCommand::new( ".edit rag-docs", "Edit the RAG documents", - AssertState::True(StateFlags::RAG), + AssertState::TrueFalse(StateFlags::RAG, StateFlags::AGENT), ), ReplCommand::new( ".rebuild rag", @@ -139,7 +139,7 @@ lazy_static::lazy_static! { ReplCommand::new( ".variable", "Set agent variable", - AssertState::True(StateFlags::AGENT) + AssertState::TrueFalse(StateFlags::AGENT, StateFlags::SESSION) ), ReplCommand::new( ".info agent", |