feat(brain): add verify quant callback

README.md

@@ -557,11 +557,12 @@ _See code: [src/commands/brain/list.ts](https://github.com/offline-ai/cli/blob/v
 
 ```
 USAGE
-  $ ai brain refresh [--json] [-c <value>] [--banner]
+  $ ai brain refresh [--json] [-b <value>] [-u <value>] [-v]
 
 FLAGS
-  -c, --config=<value>  the config file
-      --[no-]banner     show banner
+  -b, --brainDir=<value>  the brains(LLM) directory
+  -u, --hubUrl=<value>    the hub mirror url
+  -v, --verifyQuant       whether verify quant when refresh
 
 GLOBAL FLAGS
   --json  Format output as json.

src/lib/brain.ts

@@ -125,8 +125,17 @@ export async function downloadBrain(brain: AIModelSettings, options: {
   logLevel?: string,
 }) {
   const quant = options.quant
+  const onProgress = options.onProgress
   const brains = ToolFunc.get(BRAINS_FUNC_NAME) as LlmModelsFunc
   const dryRun = options.dryRun
+
+  if (typeof onProgress === 'function') {
+    brains.on('model:'+DownloadStatusEventName, (_name: string, status: string, info: any) => {
+      if (info.old === info.quant) {delete info.old}
+      onProgress('status', status, info)
+    })
+  }
+
   let downTasks = await brains.$download({id: brain._id, quant, url: options.url, dryRun})
   if (downTasks) {
     if (!Array.isArray(downTasks)) {downTasks = [downTasks]}
@@ -142,8 +151,8 @@ export async function downloadBrain(brain: AIModelSettings, options: {
       let leftCount = downTasks.length
       const errs: string[] = []
       for (const downTask of downTasks) {
-        if (typeof options.onProgress === 'function') {
-          download.on(DownloadProgressEventName+':'+downTask!.id, options.onProgress)
+        if (typeof onProgress === 'function') {
+          download.on(DownloadProgressEventName+':'+downTask!.id, onProgress)
         }
         download.on(DownloadStatusEventName+':'+downTask!.id, function(_name: string, status: FileDownloadStatus, idInfo: {url: string, id: string, filepath: string}) {
           // console.log(_name, status, idInfo)

src/oclif/commands/brain/download.ts

@@ -119,12 +119,20 @@ export default class DownloadBrainCommand extends AICommand {
     const quant = AIModelQuantType[flags.quant]
     const progresses: any = {}
     const onProgress = function(_name: string, progress: {percent:number, totalBytes:number, transferredBytes:number}, idInfo: {url: string, id?: string, filepath?: string}) {
+      if (_name === 'status') {
+        const status = progress as unknown as string
+        const info = idInfo as unknown as {url: string, quant: number, old: number}
+        const quant = AIModelQuantType[info.quant]
+        const old = AIModelQuantType[info.old]
+        logUpdate(status + ' ' + info.url + ' quant' + (info.old ? `(from ${old} to ${quant})` : `(${quant})`) )
+        return
+      }
       // console.log('🚀 ~ DownloadBrainCommand ~ onProgress ~ progress:', arguments)
       progresses[idInfo.url] = `Downloading ${idInfo.url}... ${(progress.percent * 100).toFixed(2)}% ${progress.transferredBytes} bytes`
       const info = Object.keys(progresses).map(k => progresses[k]).join('\n')
       logUpdate(info)
     }
-    this.log('Downloading to ' + userConfig.brainDir)
+    this.log(brain._id, flags.quant, 'Downloading to ' + userConfig.brainDir)
     let result: any[]
     try {
       result = await downloadBrain(brain, {