diff options
| author | sigoden <sigoden@gmail.com> | 2024-11-05 17:16:01 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-11-05 17:16:01 +0800 |
| commit | 42deaa082fe026b955f1bc11b5168a2309aad99b (patch) | |
| tree | 46a3675c7f97848d5b51db984a0151cc53436c04 /src/repl/mod.rs | |
| parent | 0a324b6ddd4b89cc61a8dd15eeb06d60273308fa (diff) | |
| download | aichat-42deaa082fe026b955f1bc11b5168a2309aad99b.tar.gz | |
feat: support session-scoped agent variables (#969)
Diffstat (limited to 'src/repl/mod.rs')
| -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", |