add customer config

This commit is contained in:
Wikig 2026-02-25 03:26:16 +08:00
parent fd26fa7459
commit b0674e0f23
3 changed files with 43 additions and 7 deletions

View file

@ -24,6 +24,7 @@ func agentCmd() {
message := "" message := ""
sessionKey := "cli:default" sessionKey := "cli:default"
modelOverride := "" modelOverride := ""
configPath := ""
args := os.Args[2:] args := os.Args[2:]
for i := 0; i < len(args); i++ { for i := 0; i < len(args); i++ {
@ -46,10 +47,24 @@ func agentCmd() {
modelOverride = args[i+1] modelOverride = args[i+1]
i++ i++
} }
case "-c", "--config":
if i+1 < len(args) {
configPath = args[i+1]
i++
} else {
fmt.Println("Error: -c/--config requires a file path")
os.Exit(1)
}
default:
if strings.HasPrefix(args[i], "--config=") {
configPath = strings.TrimPrefix(args[i], "--config=")
} else if strings.HasPrefix(args[i], "-c=") {
configPath = strings.TrimPrefix(args[i], "-c=")
}
} }
} }
cfg, err := loadConfig() cfg, err := loadConfigFromPath(configPath)
if err != nil { if err != nil {
fmt.Printf("Error loading config: %v\n", err) fmt.Printf("Error loading config: %v\n", err)
os.Exit(1) os.Exit(1)

View file

@ -29,17 +29,31 @@ import (
) )
func gatewayCmd() { func gatewayCmd() {
// Check for --debug flag configPath := ""
args := os.Args[2:] args := os.Args[2:]
for _, arg := range args { for i := 0; i < len(args); i++ {
if arg == "--debug" || arg == "-d" { if args[i] == "--debug" || args[i] == "-d" {
logger.SetLevel(logger.DEBUG) logger.SetLevel(logger.DEBUG)
fmt.Println("🔍 Debug mode enabled") fmt.Println("🔍 Debug mode enabled")
break continue
}
if args[i] == "-c" || args[i] == "--config" {
if i+1 < len(args) {
configPath = args[i+1]
i++
continue
}
fmt.Println("Error: -c/--config requires a file path")
os.Exit(1)
}
if strings.HasPrefix(args[i], "--config=") {
configPath = strings.TrimPrefix(args[i], "--config=")
} else if strings.HasPrefix(args[i], "-c=") {
configPath = strings.TrimPrefix(args[i], "-c=")
} }
} }
cfg, err := loadConfig() cfg, err := loadConfigFromPath(configPath)
if err != nil { if err != nil {
fmt.Printf("Error loading config: %v\n", err) fmt.Printf("Error loading config: %v\n", err)
os.Exit(1) os.Exit(1)

View file

@ -195,5 +195,12 @@ func getConfigPath() string {
} }
func loadConfig() (*config.Config, error) { func loadConfig() (*config.Config, error) {
return config.LoadConfig(getConfigPath()) return loadConfigFromPath("")
}
func loadConfigFromPath(path string) (*config.Config, error) {
if path == "" {
path = getConfigPath()
}
return config.LoadConfig(path)
} }