fixed file issues

Author: GauravJangra9988

cmd/cli/llmSetup.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	
 	"fmt"
 
 	"github.com/dfanso/commit-msg/cmd/cli/store"
@@ -51,7 +52,12 @@ func UpdateLLM() error {
 
 	SavedModels, err := store.ListSavedModels()
 	if err != nil {
-		fmt.Println(err)
+		return err
+	}
+
+	if len(SavedModels.LLMProviders) == 0{
+		 fmt.Println("No model exists, Please add atleast one model")
+		 return nil
 	}
 
 	models := []string{}
@@ -68,7 +74,7 @@ func UpdateLLM() error {
 
 	_,model,err := prompt.Run()
 	if err != nil {
-		fmt.Println(err)
+		return err
 	}
 
 
@@ -98,15 +104,17 @@ func UpdateLLM() error {
 			if err !=  nil {
 				return err
 			}
-			store.UpdateAPIKey(model, apiKey)
+			err = store.UpdateAPIKey(model, apiKey)
+			if err != nil {
+				return err
+			}
 			fmt.Printf("%s API Key Updated", model)
 		case 2:
 			err := store.DeleteModel(model)
-			if err == nil {
-				fmt.Printf("%s model deleted", model)
+			if err != nil {
+				return err
 			}
-			
-				
+			fmt.Printf("%s model deleted", model)			
 	}
 
 	return nil

cmd/cli/store/store.go

@@ -4,7 +4,6 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
-	"log"
 	"os"
 	"path/filepath"
 	"runtime"
@@ -37,22 +36,24 @@ func Save(LLMConfig LLMProvider) error {
 	if !isConfigExists {
 		err := createConfigFile(configPath)
 		if err != nil {
-			log.Fatal(err)
+			return err
 		}
 	}
 
 
 	data, err := os.ReadFile(configPath)
-	if err != nil {
-		fmt.Println("Error", err)
-		log.Fatal(err)
+	if errors.Is(err, os.ErrNotExist){
+		data = []byte("{}")
+	} else if
+	 err != nil {
+		return err
 	}
 
+
 	if len(data) > 0 {
 		err = json.Unmarshal(data, &cfg)
 		if err != nil {
-			fmt.Println("unmarshal error", err)
-			log.Fatal(err)
+			return err
 		}
 	}
 
@@ -162,15 +163,13 @@ func DefaultLLMKey() (*LLMProvider, error) {
 
 	data, err := os.ReadFile(configPath)
 	if err != nil {
-		fmt.Println("Error", err)
 		return nil, err
 	}
 
 
 	if len(data) > 0 {
 		err = json.Unmarshal(data, &cfg)
 		if err != nil {
-			fmt.Println("unmarshal error", err)
 			return nil, err
 		}
 	} else {
@@ -195,25 +194,23 @@ func ListSavedModels() (*Config, error){
 
 	configPath, err := getConfigPath()
 	if err != nil {
-		log.Fatal(err)
+		return nil, err
 	}
 
 	isConfigExists := checkConfig(configPath)
 	if !isConfigExists {
-		log.Fatal("config file not exists")
+		return nil, errors.New("config file not exists")
 	}
 
 	data, err := os.ReadFile(configPath)
 	if err != nil {
-		fmt.Println("Error", err)
-		log.Fatal(err)
+		return nil, err
 	}
 
 	if len(data) > 0 {
 		err = json.Unmarshal(data, &cfg)
 		if err != nil {
-			fmt.Println("unmarshal error", err)
-			log.Fatal(err)
+			return nil, err
 		}
 	} else {
 		fmt.Println("Config file is empty, Please add atlead one LLM Key")