tcsenpai/ollama
1
0
Fork 0
mirror of https://github.com/tcsenpai/ollama.git synced 2025-06-08 20:25:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(windows): modelpath and list

Browse Source
This commit is contained in:
Michael Yang 2023-12-15 15:50:51 -08:00
parent 2bb2bdd5d4
commit 2b9892a808
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
server/modelpath.go
View File

@ -46,7 +46,7 @@ func ParseModelPath(name string) ModelPath {
name = after name = after
} }
parts := strings.Split(name, string(os.PathSeparator)) parts := strings.Split(name, "/")
switch len(parts) { switch len(parts) {
case 3: case 3:
mp.Registry = parts[0] mp.Registry = parts[0]

15
server/routes.go
View File

@ -711,7 +711,7 @@ func GetModelInfo(req api.ShowRequest) (*api.ShowResponse, error) {
func ListModelsHandler(c *gin.Context) { func ListModelsHandler(c *gin.Context) {
models := make([]api.ModelResponse, 0) models := make([]api.ModelResponse, 0)
fp, err := GetManifestPath() manifestsPath, err := GetManifestPath()
if err != nil { if err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return return
@ -741,13 +741,14 @@ func ListModelsHandler(c *gin.Context) {
walkFunc := func(path string, info os.FileInfo, _ error) error { walkFunc := func(path string, info os.FileInfo, _ error) error {
if !info.IsDir() { if !info.IsDir() {
dir, file := filepath.Split(path) path, tag := filepath.Split(path)
dir = strings.Trim(strings.TrimPrefix(dir, fp), string(os.PathSeparator)) model := strings.Trim(strings.TrimPrefix(path, manifestsPath), string(os.PathSeparator))
tag := strings.Join([]string{dir, file}, ":") modelPath := strings.Join([]string{model, tag}, ":")
canonicalModelPath := strings.ReplaceAll(modelPath, string(os.PathSeparator), "/")
resp, err := modelResponse(tag) resp, err := modelResponse(canonicalModelPath)
if err != nil { if err != nil {
log.Printf("skipping file: %s", fp) log.Printf("skipping file: %s", canonicalModelPath)
// nolint: nilerr // nolint: nilerr
return nil return nil
} }
@ -759,7 +760,7 @@ func ListModelsHandler(c *gin.Context) {
return nil return nil
} }
if err := filepath.Walk(fp, walkFunc); err != nil { if err := filepath.Walk(manifestsPath, walkFunc); err != nil {
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
return return
} }