highperfocused/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-10-10 07:23:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

refactor create model

Browse Source
This commit is contained in:
Michael Yang
2023-11-14 12:30:34 -08:00
parent f61f340279
commit b0d14ed51c
2 changed files with 162 additions and 184 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
server/routes.go
View File

@@ -26,6 +26,7 @@ import (
"github.com/jmorganca/ollama/api"
"github.com/jmorganca/ollama/llm"
"github.com/jmorganca/ollama/parser"
"github.com/jmorganca/ollama/version"
)
@@ -414,6 +415,19 @@ func CreateModelHandler(c *gin.Context) {
return
}
modelfile, err := os.Open(req.Path)
if err != nil {
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
defer modelfile.Close()
commands, err := parser.Parse(modelfile)
if err != nil {
c.AbortWithStatusJSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
ch := make(chan any)
go func() {
defer close(ch)
@@ -424,7 +438,7 @@ func CreateModelHandler(c *gin.Context) {
ctx, cancel := context.WithCancel(c.Request.Context())
defer cancel()
if err := CreateModel(ctx, req.Name, req.Path, fn); err != nil {
if err := CreateModel(ctx, req.Name, commands, fn); err != nil {
ch <- gin.H{"error": err.Error()}
}
}()