Backport PR jupyterlab#1075: Fix magic commands when using non-chat p…

…roviders w/ history (jupyterlab#1080) Co-authored-by: Alan Meeson <[email protected]>
meeseeksmachine · Nov 1, 2024 · d42edd5 · d42edd5
1 parent fb4a833
commit d42edd5
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
@@ -603,17 +603,19 @@ def run_ai_cell(self, args: CellArgs, prompt: str):
         else:
             # generate output from model via provider
             if context:
-                inputs = [
-                    (
-                        f"AI: {message.content}"
-                        if message.type == "ai"
-                        else f"{message.type.title()}: {message.content}"
-                    )
-                    for message in context + [HumanMessage(content=prompt)]
-                ]
+                inputs = "\n\n".join(
+                    [
+                        (
+                            f"AI: {message.content}"
+                            if message.type == "ai"
+                            else f"{message.type.title()}: {message.content}"
+                        )
+                        for message in context + [HumanMessage(content=prompt)]
+                    ]
+                )
             else:
-                inputs = [prompt]
-            result = provider.generate(inputs)
+                inputs = prompt
+            result = provider.generate([inputs])
 
         output = result.generations[0][0].text