Fix spice chat error handling (spiceai#5433)

* Fix spice chat error handling

* fix: lint

Author: Sevenannn

bin/spice/cmd/chat.go

@@ -174,30 +174,40 @@ spice chat --model <model> --cloud
 			rtcontext.RequireModelsFlavor(cmd)
 		}
 
+		httpEndpoint, err := cmd.Flags().GetString("http-endpoint")
+		if err != nil {
+			slog.Error("could not get http-endpoint flag", "error", err)
+			os.Exit(1)
+		}
+		if httpEndpoint != "" {
+			rtcontext.SetHttpEndpoint(httpEndpoint)
+		}
+
 		model, err := cmd.Flags().GetString(modelKeyFlag)
 		if err != nil {
 			slog.Error("could not get model flag", "error", err)
 			os.Exit(1)
 		}
-		if model == "" {
-			models, err := api.GetDataSingle[api.ModelResponse](rtcontext, "/v1/models?status=true")
-			if err != nil {
-				slog.Error("could not list models", "error", err)
-				os.Exit(1)
-			}
 
-			if len(models.Data) == 0 {
-				slog.Error("No models found")
-				os.Exit(1)
-			}
+		models, err := api.GetDataSingle[api.ModelResponse](rtcontext, "/v1/models?status=true")
+		if err != nil {
+			slog.Error("could not list models", "error", err)
+			os.Exit(1)
+		}
 
-			availableModels := []string{}
-			for _, model := range models.Data {
-				if model.Status == "Ready" {
-					availableModels = append(availableModels, model.Id)
-				}
+		if len(models.Data) == 0 {
+			slog.Error("No models found")
+			os.Exit(1)
+		}
+
+		availableModels := []string{}
+		for _, model := range models.Data {
+			if model.Status == "Ready" {
+				availableModels = append(availableModels, model.Id)
 			}
+		}
 
+		if model == "" {
 			if len(availableModels) == 0 {
 				slog.Error("No models are ready")
 				os.Exit(1)
@@ -221,15 +231,20 @@ spice chat --model <model> --cloud
 
 			cmd.Printf("Using model: %s\n", selectedModel)
 			model = selectedModel
-		}
+		} else {
+			modelIsAvailable := false
+			for _, availableModel := range availableModels {
+				if availableModel == model {
+					modelIsAvailable = true
+					break
+				}
+			}
 
-		httpEndpoint, err := cmd.Flags().GetString("http-endpoint")
-		if err != nil {
-			slog.Error("could not get http-endpoint flag", "error", err)
-			os.Exit(1)
-		}
-		if httpEndpoint != "" {
-			rtcontext.SetHttpEndpoint(httpEndpoint)
+			if !modelIsAvailable {
+				slog.Error("Requested model is not available", "model", model)
+				slog.Info("Available models", "models", strings.Join(availableModels, ", "))
+				os.Exit(1)
+			}
 		}
 
 		var messages = []Message{}

bin/spice/pkg/api/util.go

@@ -84,7 +84,7 @@ func doRuntimeApiRequest[T interface{}](rtcontext *context.RuntimeContext, metho
 	}()
 
 	if resp.StatusCode == http.StatusUnauthorized {
-		return *new(T), fmt.Errorf("Unauthorized")
+		return *new(T), fmt.Errorf("unauthorized: invalid or missing Spice API key")
 	}
 
 	if resp.StatusCode == http.StatusNotFound {