llms-py 3.0.0b3__py3-none-any.whl → 3.0.0b5__py3-none-any.whl

llms/ui/ctx.mjs

@@ -11,12 +11,13 @@ export class ExtensionScope {
         this.baseUrl = `${ctx.ai.base}/ext/${this.id}`
         this.storageKey = `llms.${this.id}`
         this.state = reactive({})
+        this.prefs = reactive(storageObject(this.storageKey))
     }
     getPrefs() {
-        return storageObject(this.storageKey)
+        return this.prefs
     }
     setPrefs(o) {
-        storageObject(this.storageKey, Object.assign(this.getPrefs(), o))
+        storageObject(this.storageKey, Object.assign(this.prefs, o))
     }
     get(url, options) {
         return this.ctx.ai.get(combinePaths(this.baseUrl, url), options)
@@ -45,7 +46,6 @@ export class AppContext {
         this.events = new EventBus()
         this.modalComponents = {}
         this.extensions = []
-        this.layout = reactive(storageObject(`llms.layout`))
         this.chatRequestFilters = []
         this.chatResponseFilters = []
         this.chatErrorFilters = []
@@ -53,12 +53,15 @@ export class AppContext {
         this.updateThreadFilters = []
         this.top = {}
         this.left = {}
+        this.layout = reactive(storageObject(`llms.layout`))
+        this.prefs = reactive(storageObject(ai.prefsKey))
 
         if (!Array.isArray(this.layout.hide)) {
             this.layout.hide = []
         }
         Object.assign(app.config.globalProperties, {
             $ctx: this,
+            $prefs: this.prefs,
             $state: this.state,
             $layout: this.layout,
             $ai: ai,
@@ -80,6 +83,12 @@ export class AppContext {
             this[name] = global
         })
     }
+    getPrefs() {
+        return this.prefs
+    }
+    setPrefs(o) {
+        storageObject(this.ai.prefsKey, Object.assign(this.prefs, o))
+    }
     _validateIcons(icons) {
         Object.entries(icons).forEach(([id, icon]) => {
             if (!icon.component) {
@@ -173,12 +182,6 @@ export class AppContext {
     layoutVisible(key) {
         return !this.layout.hide.includes(key)
     }
-    getPrefs() {
-        return storageObject(this.ai.prefsKey)
-    }
-    setPrefs(o) {
-        storageObject(this.ai.prefsKey, Object.assign(this.getPrefs(), o))
-    }
     toggleTop(name) {
         console.log('toggleTop', name)
         this.layout.top = this.layout.top == name ? undefined : name
@@ -193,4 +196,10 @@ export class AppContext {
             this.router.push({ path })
         }
     }
+    getJson(url, options) {
+        return this.ai.getJson(url, options)
+    }
+    postJson(url, options) {
+        return this.ai.postJson(url, options)
+    }
 }

llms/ui/index.mjs

@@ -9,6 +9,7 @@ import ChatModule from './modules/chat/index.mjs'
 import ThreadsModule from './modules/threads/index.mjs'
 import ModelSelectorModule from './modules/model-selector.mjs'
 import AnalyticsModule from './modules/analytics.mjs'
+import ToolsModule from './modules/tools.mjs'
 import { utilsFunctions, utilsFormatters } from './utils.mjs'
 import { markdownFormatters } from './markdown.mjs'
 import { AppContext } from './ctx.mjs'
@@ -22,6 +23,7 @@ const BuiltInModules = {
     ThreadsModule,
     ModelSelectorModule,
     AnalyticsModule,
+    ToolsModule,
 }
 
 
@@ -53,10 +55,13 @@ export async function createContext() {
         }
     }))
 
+    const installedModules = []
+
     // Install built-in modules sequentially
     Object.entries(BuiltInModules).forEach(([name, module]) => {
         try {
             module.install(ctx)
+            installedModules.push({ extension: { id: name }, module: { default: module } })
             console.log(`Installed built-in: ${name}`)
         } catch (e) {
             console.error(`Failed to install built-in ${name}:`, e)
@@ -68,6 +73,7 @@ export async function createContext() {
         if (result && result.module.default && result.module.default.install) {
             try {
                 result.module.default.install(ctx)
+                installedModules.push(result)
                 console.log(`Installed extension: ${result.extension.id}`)
             } catch (e) {
                 console.error(`Failed to install extension ${result.extension.id}:`, e)
@@ -103,8 +109,11 @@ export async function createContext() {
         ctx.router.push({ path: ctx.layout.path })
     }
 
+    const loadModules = installedModules.filter(x => x.module.default && x.module.default.load)
+    console.log('Loading modules: ', loadModules.map(x => x.extension.id))
+
     // Load all extensions in parallel
-    await Promise.all(ctx.modules.filter(x => x.module.default && x.module.default.load).map(async result => {
+    await Promise.all(loadModules.map(async result => {
         try {
             await result.module.default.load(ctx)
             console.log(`Loaded extension: ${result.extension.id}`)