feat(brainCommand): add refresh flag option

offline-ai · Jun 11, 2024 · 129dd3f · 129dd3f
1 parent e3c1e51
commit 129dd3f
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 2 deletions.
diff --git a/src/lib/brain.ts b/src/lib/brain.ts
@@ -12,6 +12,11 @@ import EventEmitter from 'events'
 EventEmitter.defaultMaxListeners = 1000
 
 // note: need initTools() first
+export function upgradeBrains() {
+  const brains = ToolFunc.get(BRAINS_FUNC_NAME) as LlmModelsFunc
+  brains.updateDBFromDir()
+}
+
 export function listBrains(userConfig: any, flags: any) {
   const brainDir = userConfig.brainDir
   const result: any[] = searchBrains(brainDir, flags)

diff --git a/src/oclif/commands/brain/index.ts b/src/oclif/commands/brain/index.ts
@@ -3,7 +3,7 @@ import { initTools } from '../../../lib/init-tools.js'
 import { AICommand } from '../../lib/ai-command.js'
 import { showBanner } from '../../lib/help.js'
 import { parseJsJson } from '@isdk/ai-tool'
-import { listBrains, printBrains } from '../../../lib/brain.js'
+import { listBrains, printBrains, upgradeBrains } from '../../../lib/brain.js'
 
 export default class Brain extends AICommand {
   static args = {
@@ -25,6 +25,10 @@ export default class Brain extends AICommand {
       description: 'the max number of brains to list, 0 means all.',
       default: 100,
     }),
+    refresh: Flags.boolean({
+      char: 'r',
+      description: 'refresh the online brains list',
+    }),
   }
 
   static summary = '🧠 The AI Agent Brains(LLM) Manager.'
@@ -50,6 +54,9 @@ export default class Brain extends AICommand {
     const {args, flags} = opts
     const userConfig = this.loadConfig(flags.config, opts)
     initTools(userConfig)
+    if (flags.refresh) {
+      upgradeBrains()
+    }
 
     if (userConfig.banner && !isJson) {showBanner('Brain')}
     flags.name = args.name

diff --git a/src/oclif/commands/brain/list.ts b/src/oclif/commands/brain/list.ts
@@ -1,6 +1,6 @@
 // import util from 'util'
 import { parseJsJson } from '@isdk/ai-tool'
-import { listBrains, printBrains } from '../../../lib/brain.js'
+import { listBrains, printBrains, upgradeBrains } from '../../../lib/brain.js'
 import { initTools } from '../../../lib/init-tools.js'
 import { AICommand } from '../../lib/ai-command.js'
 import { showBanner } from '../../lib/help.js'
@@ -46,6 +46,10 @@ export default class AIBrainListCommand extends AICommand {
       description: 'the max number of brains to list, 0 means all.',
       default: 100,
     }),
+    refresh: Flags.boolean({
+      char: 'r',
+      description: 'refresh the online brains list',
+    }),
   }
 
   async run(): Promise<any> {
@@ -55,6 +59,10 @@ export default class AIBrainListCommand extends AICommand {
     const userConfig = this.loadConfig(flags.config, opts)
     initTools(userConfig)
 
+    if (flags.refresh) {
+      upgradeBrains()
+    }
+
     if (userConfig.banner && !isJson) {showBanner('Brain')}
     flags.name = args.name
     const result = listBrains(userConfig, flags)