From af4cf55884ac54b9e637cd71dadfe9b7a5685877 Mon Sep 17 00:00:00 2001
From: Bruce MacDonald <brucewmacdonald@gmail.com>
Date: Fri, 6 Oct 2023 16:06:20 -0400
Subject: [PATCH] not found error before pulling model (#718)

---
 server/routes.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/server/routes.go b/server/routes.go
index c68011df..3e5fa5e8 100644
--- a/server/routes.go
+++ b/server/routes.go
@@ -3,8 +3,10 @@ package server
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"io"
+	"io/fs"
 	"log"
 	"net"
 	"net/http"
@@ -170,6 +172,11 @@ func GenerateHandler(c *gin.Context) {
 
 	model, err := GetModel(req.Model)
 	if err != nil {
+		var pErr *fs.PathError
+		if errors.As(err, &pErr) {
+			c.JSON(http.StatusNotFound, gin.H{"error": fmt.Sprintf("model '%s' not found, try pulling it first", req.Model)})
+			return
+		}
 		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
 		return
 	}