Export column analyses (#638)

holoviz · Aug 5, 2024 · 0f0583e · 0f0583e
1 parent 122e904
commit 0f0583e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lumen/ai/export.py b/lumen/ai/export.py
@@ -5,6 +5,9 @@
 
 import nbformat
 
+from panel import Column
+from panel.chat import ChatStep
+
 from lumen.ai.views import LumenOutput
 from lumen.pipeline import Pipeline
 from lumen.views import View
@@ -59,6 +62,11 @@ def export_notebook(assistant, preamble: str = ""):
             cells += format_markdown(msg)
         elif isinstance(msg.object, LumenOutput):
             cells += format_output(msg)
+        elif isinstance(msg.object, Column):
+            for obj in msg.object:
+                if isinstance(obj, ChatStep):
+                    continue
+                cells += format_output(obj)
 
     nb = nbformat.v4.new_notebook(cells=cells)
     return nbformat.v4.writes(nb)
diff --git a/lumen/ai/views.py b/lumen/ai/views.py
@@ -138,7 +138,7 @@ def __init__(self, **params):
         if config_options:
             options = pn.Param(self.analysis.param, parameters=config_options)
             b = pn.widgets.Button(icon='rocket', name='Run...', on_click=self._rerun, button_type='success', margin=20)
-            self._tabs.insert(0, ('Config', pn.Column(options, b)))
+            self._tabs.insert(1, ('Config', pn.Column(options, b)))
             with discard_events(self):
                 self._tabs.active = 2