mirror of
https://github.com/mudler/LocalAI.git
synced 2024-12-18 20:27:57 +00:00
0d8bf91699
* WIP: add models to webui Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Register routes Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * fix: don't cache models Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * small fixups Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * fix: fixup multiple installs (strings.Clone) Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
40 lines
1.2 KiB
Go
40 lines
1.2 KiB
Go
package downloader
|
|
|
|
import "hash"
|
|
|
|
type progressWriter struct {
|
|
fileName string
|
|
total int64
|
|
fileNo int
|
|
totalFiles int
|
|
written int64
|
|
downloadStatus func(string, string, string, float64)
|
|
hash hash.Hash
|
|
}
|
|
|
|
func (pw *progressWriter) Write(p []byte) (n int, err error) {
|
|
n, err = pw.hash.Write(p)
|
|
pw.written += int64(n)
|
|
|
|
if pw.total > 0 {
|
|
percentage := float64(pw.written) / float64(pw.total) * 100
|
|
if pw.totalFiles > 1 {
|
|
// This is a multi-file download
|
|
// so we need to adjust the percentage
|
|
// to reflect the progress of the whole download
|
|
// This is the file pw.fileNo of pw.totalFiles files. We assume that
|
|
// the files before successfully downloaded.
|
|
percentage = percentage / float64(pw.totalFiles)
|
|
if pw.fileNo > 1 {
|
|
percentage += float64(pw.fileNo-1) * 100 / float64(pw.totalFiles)
|
|
}
|
|
}
|
|
//log.Debug().Msgf("Downloading %s: %s/%s (%.2f%%)", pw.fileName, formatBytes(pw.written), formatBytes(pw.total), percentage)
|
|
pw.downloadStatus(pw.fileName, formatBytes(pw.written), formatBytes(pw.total), percentage)
|
|
} else {
|
|
pw.downloadStatus(pw.fileName, formatBytes(pw.written), "", 0)
|
|
}
|
|
|
|
return
|
|
}
|