create `.ollama` directory if it doesnt exist

cmd/cmd.go

@@ -9,12 +9,12 @@ import (
 	"errors"
 	"fmt"
 	"io"
-	"io/ioutil"
 	"log"
 	"net"
 	"net/http"
 	"os"
 	"os/exec"
+	"path"
 	"path/filepath"
 	"runtime"
 	"strings"
@@ -585,7 +585,12 @@ func initializeKeypair() error {
 			return err
 		}
 
-		err = ioutil.WriteFile(privKeyPath, pem.EncodeToMemory(privKeyBytes), 0600)
+		err = os.MkdirAll(path.Dir(privKeyPath), 0o700)
+		if err != nil {
+			return fmt.Errorf("could not create directory %w", err)
+		}
+
+		err = os.WriteFile(privKeyPath, pem.EncodeToMemory(privKeyBytes), 0600)
 		if err != nil {
 			return err
 		}
@@ -597,7 +602,7 @@ func initializeKeypair() error {
 
 		pubKeyData := ssh.MarshalAuthorizedKey(sshPrivateKey.PublicKey())
 
-		err = ioutil.WriteFile(pubKeyPath, pubKeyData, 0644)
+		err = os.WriteFile(pubKeyPath, pubKeyData, 0644)
 		if err != nil {
 			return err
 		}

server/auth.go

@@ -63,11 +63,6 @@ func getAuthToken(redirData AuthRedirect, regOpts *RegistryOptions) (string, err
 
 	keyPath := path.Join(home, ".ollama", "id_ed25519")
 
-	err = os.MkdirAll(path.Dir(keyPath), 0700)
-	if err != nil {
-		return "", fmt.Errorf("could not create .ollama directory %w", err)
-	}
-
 	rawKey, err := os.ReadFile(keyPath)
 	if err != nil {
 		log.Printf("Failed to load private key: %v", err)