grokkit/config/config.go

package config

import (
	"os"
	"path/filepath"

	"github.com/spf13/viper"
)

func Load() {
	home, _ := os.UserHomeDir()
	configPath := filepath.Join(home, ".config", "grokkit")

	viper.SetConfigName("config")
	viper.SetConfigType("toml")
	viper.AddConfigPath(configPath)
	viper.AddConfigPath(".")
	viper.AutomaticEnv()

	viper.SetDefault("default_model", "grok-4")
	viper.SetDefault("temperature", 0.7)

	_ = viper.ReadInConfig()
}

func GetModel(flagModel string) string {
	if flagModel != "" {
		if alias := viper.GetString("aliases." + flagModel); alias != "" {
			return alias
		}
		return flagModel
	}
	return viper.GetString("default_model")
}