add customer config
This commit is contained in:
parent
fd26fa7459
commit
b0674e0f23
3 changed files with 43 additions and 7 deletions
|
|
@ -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)
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
|
|
|
||||||
|
|
@ -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)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue