feat(linking): use canonical model name when linking to lm-studio (#209)

Author: Vigilans

operations.go

@@ -16,6 +16,7 @@ import (
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
 	"github.com/ollama/ollama/api"
+	ollama_model "github.com/ollama/ollama/types/model"
 	"github.com/sammcj/gollama/config"
 	"github.com/sammcj/gollama/logging"
 	"github.com/sammcj/gollama/styles"
@@ -1052,22 +1053,24 @@ func linkModel(modelName, lmStudioModelsDir string, noCleanup bool, dryRun bool,
 	if err != nil {
 		return "", fmt.Errorf("error getting model files for %s: %v", modelName, err)
 	}
+	fullModelName := ollama_model.ParseName(modelName)
 
-	// Extract author from model name (e.g., "fleo/tiny-r1-32b-preview:latest" -> "fleo")
-	parts := strings.Split(modelName, ":")
-	author := "unknown"
-	if len(parts) > 1 && strings.Contains(parts[0], "/") {
-		authorParts := strings.Split(parts[0], "/")
-		author = authorParts[0]
-	} else if strings.Contains(modelName, "/") {
-		authorParts := strings.Split(modelName, "/")
-		author = authorParts[0]
+	// Extract author from model name, e.g.:
+	// * "fleo/tiny-r1-32b-preview:latest" -> "registry.ollama.ai/fleo/tiny-r1-32b-preview" -> "fleo"
+	// * "huggingface.co/fleo/tiny-r1-32b-preview" -> "fleo"
+	// * "tiny-r1-32b-preview:latest" -> "registry.ollama.ai/library/tiny-r1-32b-preview" -> "registry.ollama.ai" (Use host as author if the namespace is default `library`)
+	author := fullModelName.Namespace
+	if author == ollama_model.DefaultName().Namespace {
+		author = fullModelName.Host
 	}
 
 	// Create a clean model name for the file
-	lmStudioModelName := strings.ReplaceAll(strings.ReplaceAll(modelName, ":", "-"), "_", "-")
-	lmStudioModelName = strings.ReplaceAll(lmStudioModelName, "/", "-")
-	lmStudioModelDir := filepath.Join(lmStudioModelsDir, author, lmStudioModelName+"-GGUF")
+	// * model tag is appended to the model name if it's not the default `latest` tag
+	lmStudioModelName := fullModelName.Model
+	if fullModelName.Tag != ollama_model.DefaultName().Tag {
+		lmStudioModelName += "-" + fullModelName.Tag
+	}
+	lmStudioModelDir := filepath.Join(lmStudioModelsDir, author, lmStudioModelName)
 
 	// Check if the main model path is a valid file
 	fileInfo, err := os.Stat(modelFiles.MainModel)