Merge pull request #3718 from ollama/mxyng/modelname-3

update delete handler to use model.Name
This commit is contained in:
Michael Yang
2024-05-29 12:02:07 -07:00
committed by GitHub
8 changed files with 587 additions and 122 deletions

View File

@@ -88,3 +88,26 @@ func (l *Layer) Open() (io.ReadSeekCloser, error) {
return os.Open(blob)
}
func (l *Layer) Remove() error {
ms, err := Manifests()
if err != nil {
return err
}
for _, m := range ms {
for _, layer := range append(m.Layers, m.Config) {
if layer.Digest == l.Digest {
// something is using this layer
return nil
}
}
}
blob, err := GetBlobsPath(l.Digest)
if err != nil {
return err
}
return os.Remove(blob)
}