Use <role>: prefix instead of pseudo-XML for non-chat models

jupyterlab · Sep 30, 2024 · bead0a7 · bead0a7
1 parent 7ed670d
commit bead0a7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
@@ -610,9 +610,9 @@ def run_ai_cell(self, args: CellArgs, prompt: str):
             if self.transcript:
                 transcript = [
                     (
-                        f"<HUMAN>{message.content}</HUMAN>"
-                        if message.type == "human"
-                        else message.content
+                        f"AI: {message.content}"
+                        if message.type == "ai"
+                        else f"{message.type.title()}: {message.content}"
                     )
                     for message in self.transcript[-2 * self.max_history :]
                     + [HumanMessage(content=prompt)]