feat(chainlit): Allow EVA_SEARCH_PERMISSION to be set via Chainlit chat settings UI

README.md

@@ -63,6 +63,7 @@ chainlit run app.py
 - Web search integration via Tavily
 - Semantic chunking for better context retrieval
 - Interactive chat interface through Chainlit
+- 🌐 **Web Search Permission Control**: You can now set whether the AI is allowed to perform web searches directly from the chat settings (top right ⚙️). Choose between "Ask every time", "Always allow", or "Never allow" for full control!
 
 ## ⚙️ Configuration Options
 
@@ -80,7 +81,7 @@ You can customize the behavior of PsTuts RAG using environment variables. Set th
 | `MAX_RESEARCH_LOOPS` | 🔁 Maximum number of research loops to perform. Default: `3` |
 | `LLM_TOOL_MODEL` | 🛠️ Name of the LLM model to use for tool calling. Default: `smollm2:1.7b-instruct-q2_K` |
 | `N_CONTEXT_DOCS` | 📚 Number of context documents to retrieve for RAG. Default: `2` |
-| `EVA_SEARCH_PERMISSION` | 🌐 Permission for search (`yes`, `no`, or `ask`). Default: `no` |
+| `EVA_SEARCH_PERMISSION` | 🌐 Permission for search (`yes`, `no`, or `ask`). Default: `no`. **Can also be set in the chat UI!** |
 | `EVA_DB_PERSIST` | 💾 Path or flag for DB persistence. Default: unset |
 | `EVA_REINITIALIZE` | 🔄 If true, reinitializes EVA DB. Default: `False` |
 | `THREAD_ID` | 🧵 Thread ID for the current session. Default: unset |

TODO.md

@@ -1,9 +1,7 @@
 # 📝 TODOs in Codebase
 
 - `./app.py:44:# TODO: Create an introduction message here that explains the purpose of the app`
-- `./app.py:225:    # TODO: Make the step label update instead of add`
-- `./app.py:295:# TODO: Clean up imports.`
-- `./app.py:297:# TODO Add buttons with pregenerated queries`
+- `./app.py:251:    # TODO: Make the step label update instead of add`
 - `./pstuts_rag/pstuts_rag/datastore.py:26:# TODO: Write MCP server that ingests `mp4` folder`
 - `./pstuts_rag/pstuts_rag/datastore.py:66:# TODO: accumulate transcripts of videos when loading, summarize each, then summarize summaries to get a description of the dataset for the prompt`
 - `./pstuts_rag/pstuts_rag/nodes.py:113:# TODO More robust generation of queries - multiquery launch`

app.py

@@ -43,6 +43,24 @@ unique_id = uuid4().hex[0:8]
 
 # TODO: Create an introduction message here that explains the purpose of the app
 
+# 1. Define Chainlit settings schema for EVA_SEARCH_PERMISSION
+cl.Settings(
+    [
+        cl.Dropdown(
+            id="eva_search_permission",
+            label="Web Search Permission",
+            description="Allow the AI to perform web searches?",
+            values=[
+                {"label": "Ask every time", "value": "ask"},
+                {"label": "Always allow", "value": "yes"},
+                {"label": "Never allow", "value": "no"},
+            ],
+            default_value="no",
+            required=True,
+        )
+    ]
+)
+
 
 async def sample_prompt_send(action: cl.Action):
     # Simulate a user message using the payload
@@ -117,6 +135,14 @@ async def on_chat_start():
     thread_id = f"chat_{uuid4().hex[:8]}"
     configuration.thread_id = thread_id
 
+    # Set initial Chainlit setting for EVA_SEARCH_PERMISSION
+    cl.user_session.set(
+        "eva_search_permission", configuration.search_permission
+    )
+    await cl.set_settings(
+        {"eva_search_permission": configuration.search_permission}
+    )
+
     # Instantiate the Datastore and register a callback to notify when loading is complete
     datastore = Datastore(config=configuration)
     datastore.add_completion_callback(
@@ -292,11 +318,6 @@ class ChainlitCallbackHandler(BaseCallbackHandler):
             print(f"Error in on_chain_error: {e}")
 
 
-# TODO: Clean up imports.
-
-# TODO Add buttons with pregenerated queries
-
-
 async def handle_interrupt(query: str) -> YesNoDecision:
 
     try:
@@ -431,6 +452,18 @@ async def end():
     logging.info(f"Session ended: {session_id}")
 
 
+# 2. Update Configuration on settings update
+@cl.on_settings_update
+async def on_settings_update(settings):
+    configuration = cl.user_session.get("configuration")
+    if configuration and "eva_search_permission" in settings:
+        configuration.search_permission = settings["eva_search_permission"]
+        cl.user_session.set("configuration", configuration)
+        cl.user_session.set(
+            "eva_search_permission", settings["eva_search_permission"]
+        )
+
+
 if __name__ == "__main__":
 
     def handle_sigint(signum, frame):

docs/DEVELOPER.md

@@ -505,4 +505,18 @@ This repo uses [`flake8-todos`](https://github.com/awestlake87/flake8-todos) to
 
 Happy hacking! 🚀 
 
-<!-- Test edit for TODO.md automation check -->
+<!-- Test edit for TODO.md automation check -->
+
+## 🛠️ Chainlit Settings Integration: Web Search Permission 🌐
+
+You can now control the EVA web search permission (`EVA_SEARCH_PERMISSION`) directly from the Chainlit chat UI! 🎛️
+
+- The setting appears as a dropdown in the chat settings (top right ⚙️):
+  - "Ask every time" (ask)
+  - "Always allow" (yes)
+  - "Never allow" (no)
+- When the user changes this setting, the backend updates the `Configuration` object in the session, so all subsequent actions use the new value.
+- The rest of the app always reads the current value from the session's `Configuration`.
+
+**How to extend:**
+- To add more user-configurable settings, just add them to the Chainlit settings schema and update the session's `Configuration` in the `@cl.on_settings_update` handler. Easy as pie! 🥧