From 42deaa082fe026b955f1bc11b5168a2309aad99b Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 5 Nov 2024 17:16:01 +0800
Subject: feat: support session-scoped agent variables (#969)

---
 src/repl/mod.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/repl')

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",
-- 
cgit v1.2.3