feat: allow db any path with flag

cmd/jenn-ai/main.go

@@ -16,6 +16,7 @@ var (
 	topP        = flag.Float64("topP", 0.999, "topP setting")
 	topK        = flag.Int("topK", 250, "topK setting")
 	maxTokens   = flag.Int("max-tokens", 500, "max tokens to sample")
+	dbPath      = flag.String("db-path", "./chat.db", "The path to the sqlite database")
 )
 
 func main() {
@@ -31,5 +32,5 @@ func main() {
 		TopK:        *topK,
 		MaxTokens:   *maxTokens,
 	}
-	api.Serve(ctx, mc)
+	api.Serve(ctx, mc, *dbPath)
 }

internal/api/server.go

@@ -13,7 +13,7 @@ import (
 	"jenn-ai/internal/state"
 )
 
-func Serve(ctx context.Context, c config.ModelConfig) {
+func Serve(ctx context.Context, c config.ModelConfig, dbPath string) {
 	router := gin.Default()
 
 	// Use the embedded template set
@@ -22,7 +22,7 @@ func Serve(ctx context.Context, c config.ModelConfig) {
 
 	mc := handlers.NewParameters(c.Platform, c.ModelID, c.Region, c.Temperature, c.TopP, c.TopK, c.MaxTokens)
 
-	err := db.NewDB()
+	err := db.NewDB(dbPath)
 	if err != nil {
 		log.Fatal(err)
 	}

internal/db/database.go

@@ -10,10 +10,10 @@ var DB *sql.DB
 
 // NewDB creates a new database if it doesn't exist
 // and returns an error if it fails
-func NewDB() error {
+func NewDB(dbPath string) error {
 	var err error
 
-	DB, err = sql.Open("sqlite3", "./chat.db")
+	DB, err = sql.Open("sqlite3", dbPath)
 	if err != nil {
 		return err
 	}