diff --git a/cmd/picoclaw/internal/gateway/command.go b/cmd/picoclaw/internal/gateway/command.go index b304c2516..35976ff10 100644 --- a/cmd/picoclaw/internal/gateway/command.go +++ b/cmd/picoclaw/internal/gateway/command.go @@ -8,6 +8,7 @@ func NewGatewayCommand() *cobra.Command { var ( debug bool orchestration bool + enableStats bool ) cmd := &cobra.Command{ @@ -16,12 +17,13 @@ func NewGatewayCommand() *cobra.Command { Short: "Start picoclaw gateway", Args: cobra.NoArgs, RunE: func(_ *cobra.Command, _ []string) error { - return gatewayCmd(debug, orchestration) + return gatewayCmd(debug, orchestration, enableStats) }, } cmd.Flags().BoolVarP(&debug, "debug", "d", false, "Enable debug logging") cmd.Flags().BoolVar(&orchestration, "orchestration", false, "Enable subagent orchestration") + cmd.Flags().BoolVar(&enableStats, "stats", false, "Enable stats tracking") return cmd } diff --git a/cmd/picoclaw/internal/gateway/helpers.go b/cmd/picoclaw/internal/gateway/helpers.go index f332cdbf4..8001c1a23 100644 --- a/cmd/picoclaw/internal/gateway/helpers.go +++ b/cmd/picoclaw/internal/gateway/helpers.go @@ -47,7 +47,7 @@ import ( "github.com/sipeed/picoclaw/pkg/tools" ) -func gatewayCmd(debug bool, orchestration bool) error { +func gatewayCmd(debug bool, orchestration bool, enableStats bool) error { if debug { logger.SetLevel(logger.DEBUG) fmt.Println("šŸ” Debug mode enabled") @@ -72,7 +72,7 @@ func gatewayCmd(debug bool, orchestration bool) error { } msgBus := bus.NewMessageBus() - agentLoop := agent.NewAgentLoop(cfg, msgBus, provider) + agentLoop := agent.NewAgentLoop(cfg, msgBus, provider, enableStats) // Print agent startup info fmt.Println("\nšŸ“¦ Agent Status:")