Update

tests/unit/services/devii/session.py

@@ -41,8 +41,34 @@ def test_main_channel_keeps_full_catalog_and_behavior(local_db):
     main = _session("main", "sess-main")
     names = _tool_names(main)
     assert "search_docs" in names
+    assert "ui_prompt" in names
+    assert "ui_cancel" in names
     assert len(names) > 1
-    assert "TRUTH RULES AND BEHAVIOR" in main._compose_system_prompt()
+    prompt = main._compose_system_prompt()
+    assert "TRUTH RULES AND BEHAVIOR" in prompt
+    assert "CHANNEL: site-chat" in prompt
+    assert "Interactive asks (CA-IWP)" in prompt
+
+
+def test_docs_channel_excludes_ui_prompt(local_db):
+    docs = _session("docs", "sess-docs-ui")
+    assert "ui_prompt" not in _tool_names(docs)
+
+
+def test_telegram_channel_includes_ui_prompt(local_db):
+    tg = _session("telegram", "sess-tg-ui")
+    names = _tool_names(tg)
+    assert "ui_prompt" in names
+    prompt = tg._compose_system_prompt()
+    assert "CHANNEL: telegram" in prompt
+
+
+def test_guest_main_has_ui_prompt_without_pref_tools(local_db):
+    main = _session("main", "sess-guest-ui")
+    names = _tool_names(main)
+    assert "ui_prompt" in names
+    assert "interactions_get" not in names
+    assert "interactions_set" not in names
 
 
 def test_docs_channel_does_not_start_scheduler(local_db):