feat: prioritize local models in model list

Local models (source=local or format=gguf/safetensors) now appear
first in the model list, making them easier to find and select.

Generated with AI

Co-Authored-By: csglite <xzhgan@gmail.com>

Author: ganisback

internal/server/cloud_models.go

@@ -130,6 +130,12 @@ func (s *Server) refreshCloudChatModels(ctx context.Context) ([]api.ModelInfo, e
 
 func sortModelsByPriority(models []api.ModelInfo) {
 	sort.SliceStable(models, func(i, j int) bool {
+		iIsLocal := isLocalModelInfo(models[i])
+		jIsLocal := isLocalModelInfo(models[j])
+		if iIsLocal != jIsLocal {
+			return iIsLocal
+		}
+
 		iType := strings.TrimSpace(strings.ToLower(models[i].LLMType))
 		jType := strings.TrimSpace(strings.ToLower(models[j].LLMType))
 		iOwner := strings.TrimSpace(strings.ToLower(models[i].OwnedBy))
@@ -150,3 +156,9 @@ func sortModelsByPriority(models []api.ModelInfo) {
 		return models[i].Model < models[j].Model
 	})
 }
+
+func isLocalModelInfo(model api.ModelInfo) bool {
+	source := strings.TrimSpace(strings.ToLower(model.Source))
+	format := strings.TrimSpace(strings.ToLower(model.Format))
+	return source == "local" || format == "gguf" || format == "safetensors"
+}

internal/server/cloud_models_test.go

@@ -213,6 +213,33 @@ func TestHandleCloudAuthTokenSaveInvalidatesCloudModelCache(t *testing.T) {
 	}
 }
 
+func TestSortModelsByPriorityKeepsLocalModelsFirst(t *testing.T) {
+	models := []api.ModelInfo{
+		{Model: "z/cloud-external", Source: "cloud", Format: "cloud", LLMType: "external_llm"},
+		{Model: "z/local", Source: "local", Format: "gguf"},
+		{Model: "a/cloud-opencsg", Source: "cloud", Format: "cloud", OwnedBy: "opencsg"},
+		{Model: "a/local-safetensors", Format: "safetensors"},
+		{Model: "b/cloud-other", Source: "cloud", Format: "cloud"},
+	}
+
+	sortModelsByPriority(models)
+
+	got := make([]string, 0, len(models))
+	for _, model := range models {
+		got = append(got, model.Model)
+	}
+	want := []string{
+		"a/local-safetensors",
+		"z/local",
+		"z/cloud-external",
+		"a/cloud-opencsg",
+		"b/cloud-other",
+	}
+	if strings.Join(got, ",") != strings.Join(want, ",") {
+		t.Fatalf("sorted models = %#v, want %#v", got, want)
+	}
+}
+
 func containsModelID(modelIDs []string, want string) bool {
 	for _, item := range modelIDs {
 		if item == want {