llms-py 3.0.19__py3-none-any.whl → 3.0.20__py3-none-any.whl

llms/ui/ctx.mjs

@@ -1,7 +1,7 @@
 
 import { reactive, markRaw } from 'vue'
 import { EventBus, humanize, combinePaths } from "@servicestack/client"
-import { storageObject, isHtml } from './utils.mjs'
+import { storageObject, isHtml, sanitizeHtml } from './utils.mjs'
 
 export class ExtensionScope {
     constructor(ctx, id) {
@@ -126,7 +126,7 @@ export class ExtensionScope {
 }
 
 export class AppContext {
-    constructor({ app, routes, ai, fmt, utils, marked }) {
+    constructor({ app, routes, ai, fmt, utils, marked, markedFallback }) {
         this.app = app
         this.routes = routes
         this.ai = ai
@@ -134,6 +134,7 @@ export class AppContext {
         this.utils = utils
         this._components = {}
         this.marked = marked
+        this.markedFallback = markedFallback
 
         this.state = reactive({})
         this.events = new EventBus()
@@ -397,7 +398,19 @@ export class AppContext {
             const header = content.substring(3, headerEnd).trim()
             content = '<div class="frontmatter">' + header + '</div>\n' + content.substring(headerEnd + 3)
         }
-        return this.marked.parse(content || '')
+        let html = content || ''
+        try {
+            html = this.marked.parse(content || '')
+        } catch (e) {
+            console.log('Failed to parse markdown, using fallback', e)
+            try {
+                html = this.markedFallback.parse(content || '')
+            } catch (e2) {
+                console.log('Failed to parse markdown, using raw content', e2)
+                html = content || ''
+            }
+        }
+        return sanitizeHtml(html)
     }
 
     renderContent(content) {

llms/ui/index.mjs

@@ -9,7 +9,7 @@ import ChatModule from './modules/chat/index.mjs'
 import ModelSelectorModule from './modules/model-selector.mjs'
 import IconsModule from './modules/icons.mjs'
 import { utilsFunctions, utilsFormatters } from './utils.mjs'
-import { marked } from './markdown.mjs'
+import { marked, markedFallback } from './markdown.mjs'
 import { AppContext } from './ctx.mjs'
 
 const Components = {
@@ -35,7 +35,7 @@ export async function createContext() {
     const utils = Object.assign({}, utilsFunctions())
     const routes = []
 
-    const ctx = new AppContext({ app, routes, ai, fmt, utils, marked })
+    const ctx = new AppContext({ app, routes, ai, fmt, utils, marked, markedFallback })
     app.provide('ctx', ctx)
     await ctx.init()