use cobra RunE to simplfy the cmds

cmd/cache.go

@@ -4,7 +4,6 @@ Copyright © 2024 NAME HERE <EMAIL ADDRESS>
 package cmd
 
 import (
-	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 
 	"github.com/robbyt/llm_proxy/config"
@@ -21,12 +20,9 @@ and retrieve the responses. This mode is useful for development and for CI, beca
 number of requests to the upstream server. The cache server will respond with the same status code,
 headers, and body as the previous response. The cache server will not store responses with a status
 code of 500 or higher.`,
-	Run: func(cmd *cobra.Command, args []string) {
+	RunE: func(cmd *cobra.Command, args []string) error {
 		cfg.AppMode = config.CacheMode
-		err := proxy.Run(cfg)
-		if err != nil {
-			log.Fatal(err)
-		}
+		return proxy.Run(cfg)
 	},
 }
 

cmd/config.go

@@ -5,6 +5,7 @@ import "github.com/robbyt/llm_proxy/config"
 // cfg is a reasonable default configuration, used by all commands
 var cfg *config.Config = config.NewDefaultConfig()
 
+// suggestions are here instead of their respective files bc it's easier to see them all in one place
 var cache_suggestions = []string{
 	"cache-proxy", "caching-proxy", "cash-proxy", "cash",
 }

cmd/dir_logger.go

@@ -1,7 +1,6 @@
 package cmd
 
 import (
-	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 
 	"github.com/robbyt/llm_proxy/config"
@@ -16,12 +15,9 @@ var dirLoggerCmd = &cobra.Command{
 Each request/response pair will be written to a file, identified by a unique ID. The file
 will contain the request and response in JSON format.
 `,
-	Run: func(cmd *cobra.Command, args []string) {
+	RunE: func(cmd *cobra.Command, args []string) error {
 		cfg.AppMode = config.DirLoggerMode
-		err := proxy.Run(cfg)
-		if err != nil {
-			log.Fatal(err)
-		}
+		return proxy.Run(cfg)
 	},
 }
 

cmd/simple.go

@@ -1,7 +1,6 @@
 package cmd
 
 import (
-	log "github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
 
 	"github.com/robbyt/llm_proxy/config"
@@ -13,12 +12,9 @@ var simpleCmd = &cobra.Command{
 	Use:   "simple",
 	Short: "Run a 'simple' proxy server, traffic will not be stored or cached.",
 	Long:  "Useful as a simple proxy to test connectivity, or as a base for a more complex proxy.",
-	Run: func(cmd *cobra.Command, args []string) {
+	RunE: func(cmd *cobra.Command, args []string) error {
 		cfg.AppMode = config.SimpleMode
-		err := proxy.Run(cfg)
-		if err != nil {
-			log.Fatal(err)
-		}
+		return proxy.Run(cfg)
 	},
 }