baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix relative path on create (#1222)

Browse source
This commit is contained in:
Bruce MacDonald 2023-11-21 15:43:17 -05:00 committed by GitHub
parent 2eaa95b417
commit 37d95157df
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
server/images.go
View file

@ -228,10 +228,10 @@ func GetModel(name string) (*Model, error) {
return model, nil
}
func realpath(p string) string {
abspath, err := filepath.Abs(p)
func realpath(mfDir, from string) string {
abspath, err := filepath.Abs(from)
if err != nil {
return p
return from
}
home, err := os.UserHomeDir()
@ -239,16 +239,21 @@ func realpath(p string) string {
return abspath
}
if p == "~" {
if from == "~" {
return home
} else if strings.HasPrefix(p, "~/") {
return filepath.Join(home, p[2:])
} else if strings.HasPrefix(from, "~/") {
return filepath.Join(home, from[2:])
}
if _, err := os.Stat(filepath.Join(mfDir, from)); err == nil {
// this is a file relative to the Modelfile
return filepath.Join(mfDir, from)
}
return abspath
}
func CreateModel(ctx context.Context, name string, commands []parser.Command, fn func(resp api.ProgressResponse)) error {
func CreateModel(ctx context.Context, name, modelFileDir string, commands []parser.Command, fn func(resp api.ProgressResponse)) error {
config := ConfigV2{
OS: "linux",
Architecture: "amd64",
@ -276,7 +281,7 @@ func CreateModel(ctx context.Context, name string, commands []parser.Command, fn
c.Args = blobPath
}
bin, err := os.Open(realpath(c.Args))
bin, err := os.Open(realpath(modelFileDir, c.Args))
if err != nil {
// not a file on disk so must be a model reference
modelpath := ParseModelPath(c.Args)
@ -372,7 +377,7 @@ func CreateModel(ctx context.Context, name string, commands []parser.Command, fn
layers = append(layers, layer)
case "adapter":
fn(api.ProgressResponse{Status: "creating adapter layer"})
bin, err := os.Open(realpath(c.Args))
bin, err := os.Open(realpath(modelFileDir, c.Args))
if err != nil {
return err
}

8
server/routes.go
View file

@ -423,14 +423,14 @@ func CreateModelHandler(c *gin.Context) {
var modelfile io.Reader = strings.NewReader(req.Modelfile)
if req.Path != "" && req.Modelfile == "" {
bin, err := os.Open(req.Path)
mf, err := os.Open(req.Path)
if err != nil {
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": fmt.Sprintf("error reading modelfile: %s", err)})
return
}
defer bin.Close()
defer mf.Close()
modelfile = bin
modelfile = mf
}
commands, err := parser.Parse(modelfile)
@ -449,7 +449,7 @@ func CreateModelHandler(c *gin.Context) {
ctx, cancel := context.WithCancel(c.Request.Context())
defer cancel()
if err := CreateModel(ctx, req.Name, commands, fn); err != nil {
if err := CreateModel(ctx, req.Name, filepath.Dir(req.Path), commands, fn); err != nil {
ch <- gin.H{"error": err.Error()}
}
}()