cr

1_🏠_Home.py

@@ -28,9 +28,9 @@
     st.info("**NOTE**: The ability to add web search is enabled.")
 
 
+current_state = get_current_state()
 add_sidebar()
 
-current_state = get_current_state()
 
 st.info(f"Currently building/editing agent: {current_state.cache.agent_id}", icon="ℹ️")
 

pages/2_⚙️_RAG_Config.py

@@ -69,9 +69,10 @@ def delete_agent() -> None:
     menu_items=None,
 )
 st.title("RAG Pipeline Config")
-add_sidebar()
 
 current_state = get_current_state()
+add_sidebar()
+
 
 
 if current_state.agent_builder is not None:

pages/3_🤖_Generated_RAG_Agent.py

@@ -17,6 +17,7 @@
 )
 st.title("Generated RAG Agent")
 
+current_state = get_current_state()
 add_sidebar()
 
 if (
@@ -45,8 +46,6 @@ def display_messages() -> None:
                 raise ValueError(f"Unknown message type: {msg_type}")
 
 
-current_state = get_current_state()
-
 # if agent is created, then we can chat with it
 if current_state.cache is not None and current_state.cache.agent is not None:
     st.info(f"Viewing config for agent: {current_state.cache.agent_id}", icon="ℹ️")

st_utils.py

@@ -94,6 +94,13 @@ def get_current_state() -> CurrentSessionState:
     if "cur_agent_ids" not in st.session_state.keys():
         st.session_state.cur_agent_ids = agent_registry.get_agent_ids()
 
+    if "selected_id" not in st.session_state.keys():
+        st.session_state.selected_id = None
+
+    if "selected_cache" not in st.session_state.keys():
+        st.session_state.selected_cache = None
+
+    # set builder agent / agent builder
     if (
         "selected_cache" in st.session_state.keys()
         and st.session_state.selected_cache is not None