Strips prefix and suffix (jupyterlab#60)

Marchlak · Oct 28, 2024 · ee90e89 · ee90e89
1 parent 48ed36f
commit ee90e89
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
@@ -1,5 +1,5 @@
-import os
 import json
+import os
 import warnings
 from typing import Optional
 
@@ -189,6 +189,8 @@ def ai(self, line, cell=None):
 
         # if the user wants code, add another cell with the output.
         if args.format == 'code':
+            # Strip leading and trailing triple-backticks
+            output = output.removeprefix('```\n').removesuffix('\n```')
             new_cell_payload = dict(
                 source='set_next_input',
                 text=output,