Add on-demand follow-up suggestion icon after successful queries

Author: ghostiee-11

lumen/ai/ui.py

@@ -1635,10 +1635,10 @@ async def _generate_follow_up():
         if len(self.interface):
             message = self.interface.objects[-1]
             existing = [
-                obj for obj in (message.footer_actions or [])
+                obj for obj in (message.footer_objects or [])
                 if getattr(obj, 'name', '') != "FollowUp"
             ]
-            message.footer_actions = existing + [follow_up_button]
+            message.footer_objects = existing + [follow_up_button]
 
     def __panel__(self):
         return self._main

lumen/tests/ai/test_ui.py

@@ -1615,12 +1615,9 @@ async def noop_callback(contents, user, instance):
 
 # --- Follow-up suggestions tests ---
 
-def _get_footer_actions(msg):
-    """Get footer actions from message, falling back to footer_objects for older pmui."""
-    try:
-        return msg.footer_actions or []
-    except AttributeError:
-        return msg.footer_objects or []
+def _get_footer_objects(msg):
+    """Get footer objects from message."""
+    return msg.footer_objects or []
 
 
 def test_follow_up_suggestion_model():
@@ -1663,7 +1660,7 @@ async def test_follow_up_icon_after_query(explorer_ui):
 
     last_msg = ui.interface.objects[-1]
     follow_up_icons = [
-        f for f in _get_footer_actions(last_msg)
+        f for f in _get_footer_objects(last_msg)
         if getattr(f, 'name', '') == "FollowUp"
     ]
     assert len(follow_up_icons) == 1
@@ -1677,7 +1674,7 @@ async def test_follow_up_icon_not_on_error(explorer_ui_with_error):
 
     last_msg = ui.interface.objects[-1]
     follow_up_icons = [
-        f for f in _get_footer_actions(last_msg)
+        f for f in _get_footer_objects(last_msg)
         if getattr(f, 'name', '') == "FollowUp"
     ]
     assert len(follow_up_icons) == 0
@@ -1699,7 +1696,7 @@ async def test_follow_up_icon_not_shown_without_pipeline(explorer_ui):
 
     last_msg = ui.interface.objects[-1]
     follow_up_icons = [
-        f for f in _get_footer_actions(last_msg)
+        f for f in _get_footer_objects(last_msg)
         if getattr(f, 'name', '') == "FollowUp"
     ]
     assert len(follow_up_icons) == 0
@@ -1733,7 +1730,7 @@ async def test_follow_up_icon_description(explorer_ui):
 
     last_msg = ui.interface.objects[-1]
     follow_up_icons = [
-        f for f in _get_footer_actions(last_msg)
+        f for f in _get_footer_objects(last_msg)
         if getattr(f, 'name', '') == "FollowUp"
     ]
     assert len(follow_up_icons) == 1

pixi.toml

@@ -20,7 +20,7 @@ lint = ["py312", "lint"]
 [dependencies]
 bokeh = "*"
 holoviews = ">=1.17.0"
-panel-material-ui = ">=0.9.0"
+panel-material-ui = ">=0.9.1"
 hvplot = "*"
 intake = "<2"
 jinja2 = ">3.0"

pyproject.toml

@@ -42,7 +42,7 @@ dependencies = [
     "pandas",
     "panel >=1.7.5",
     "panel-graphic-walker[kernel] >=0.6.4",
-    "panel-material-ui >=0.8.1",
+    "panel-material-ui >=0.9.1",
     "panel-splitjs >=0.3.2",
     "param >=2.2.1",
     "pyarrow",