Merge pull request #62 from jan-janssen/fix_anthropic_output

Fix bug with anthropic output
jan-janssen · Aug 30, 2024 · d96f3fb · d96f3fb
2 parents 8b90e5c + 11ba2b4
commit d96f3fb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/langsim/magics.py b/langsim/magics.py
@@ -85,6 +85,8 @@ def chat(self, line, cell=None):
         self.messages.append(("ai", output))
         if args.raw:
             return Markdown(f"```\n{output}\n```\n")
+        elif isinstance(output, list) and isinstance(output[-1], dict):
+            return Markdown(output[-1]['text'])
         else:
             return Markdown(output)