llms-py 2.0.34__py3-none-any.whl → 3.0.0__py3-none-any.whl

llms/ui/markdown.mjs

@@ -1,11 +1,11 @@
 import { Marked } from "marked"
 import hljs from "highlight.js"
 
-export const marked = (() => {
+export function createMarked() {
     const aliases = {
         vue: 'html',
     }
-    
+
     const ret = new Marked(
         markedHighlight({
             langPrefix: 'hljs language-',
@@ -18,18 +18,23 @@ export const marked = (() => {
             }
         })
     )
-    ret.use({ extensions:[thinkTag()] })
+    ret.use({ extensions: [thinkTag()] })
     //ret.use({ extensions: [divExtension()] })
     return ret
-})();
+}
+
+export const marked = createMarked();
 
 export function renderMarkdown(content) {
+    if (Array.isArray(content)) {
+        content = content.filter(c => c.type === 'text').map(c => c.text).join('\n')
+    }
     if (content) {
         content = content
-            .replaceAll(`\\[ \\boxed{`,'\n<span class="inline-block text-xl text-blue-500 bg-blue-50 dark:text-blue-400 dark:bg-blue-950 px-3 py-1 rounded">')
-            .replaceAll('} \\]','</span>\n')
+            .replaceAll(`\\[ \\boxed{`, '\n<span class="inline-block text-xl text-blue-500 bg-blue-50 dark:text-blue-400 dark:bg-blue-950 px-3 py-1 rounded">')
+            .replaceAll('} \\]', '</span>\n')
     }
-    return marked.parse(content)
+    return marked.parse(content || '')
 }
 
 // export async function renderMarkdown(body) {
@@ -195,3 +200,9 @@ function toggleThink(id) {
         icon.classList.remove('rotate-180')
     }
 }
+
+export function markdownFormatters() {
+    return {
+        markdown: renderMarkdown
+    }
+}