fixes #15

Author: jph00

dialoghelper/_modidx.py

@@ -5,7 +5,8 @@
                 'doc_host': 'https://AnswerDotAI.github.io',
                 'git_url': 'https://github.com/AnswerDotAI/dialoghelper',
                 'lib_path': 'dialoghelper'},
-  'syms': { 'dialoghelper.core': { 'dialoghelper.core._msg': ('core.html#_msg', 'dialoghelper/core.py'),
+  'syms': { 'dialoghelper.core': { 'dialoghelper.core._add_msg_unsafe': ('core.html#_add_msg_unsafe', 'dialoghelper/core.py'),
+                                   'dialoghelper.core._msg': ('core.html#_msg', 'dialoghelper/core.py'),
                                    'dialoghelper.core._umsg': ('core.html#_umsg', 'dialoghelper/core.py'),
                                    'dialoghelper.core.add_html': ('core.html#add_html', 'dialoghelper/core.py'),
                                    'dialoghelper.core.add_msg': ('core.html#add_msg', 'dialoghelper/core.py'),

dialoghelper/core.py

@@ -123,6 +123,17 @@ def add_msg(
     assert msg_type not in ('note') or not output, "'note' messages cannot have an output."
     run_cmd('add_msg', content=content, msg_type=msg_type, output=output, placement=placement, sid=sid, **kwargs)
 
+# %% ../nbs/00_core.ipynb
+@delegates(add_msg)
+def _add_msg_unsafe(
+    content:str, # Content of the message (i.e the message prompt, code, or note text)
+    run:bool=False, # For prompts, send it to the AI; for code, execute it (*DANGEROUS -- be careful of what you run!)
+    **kwargs
+):
+    """Add/update a message to the queue to show after code execution completes, and optionally run it. Be sure to pass a `sid` (stable id) not a `mid` (which is used only for sorting, and can change).
+    *WARNING*--This can execute arbitrary code, so check carefully what you run!--*WARNING"""
+    add_msg(content=content, run=run, **kwargs)
+
 # %% ../nbs/00_core.ipynb
 def _umsg(
     content:str|None = None, # Content of the message (i.e the message prompt, code, or note text)

nbs/00_core.ipynb

@@ -349,6 +349,24 @@
     "    run_cmd('add_msg', content=content, msg_type=msg_type, output=output, placement=placement, sid=sid, **kwargs)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#| export\n",
+    "@delegates(add_msg)\n",
+    "def _add_msg_unsafe(\n",
+    "    content:str, # Content of the message (i.e the message prompt, code, or note text)\n",
+    "    run:bool=False, # For prompts, send it to the AI; for code, execute it (*DANGEROUS -- be careful of what you run!)\n",
+    "    **kwargs\n",
+    "):\n",
+    "    \"\"\"Add/update a message to the queue to show after code execution completes, and optionally run it. Be sure to pass a `sid` (stable id) not a `mid` (which is used only for sorting, and can change).\n",
+    "    *WARNING*--This can execute arbitrary code, so check carefully what you run!--*WARNING\"\"\"\n",
+    "    add_msg(content=content, run=run, **kwargs)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,