From b490ae4c026a10f5b29b770190103c7d8591b0c7 Mon Sep 17 00:00:00 2001
From: dj-oyu <68707227+dj-oyu@users.noreply.github.com>
Date: Sun, 1 Mar 2026 04:13:34 +0900
Subject: [PATCH] fix(cli): restore --stats flag lost in Cobra migration

The --stats flag was not carried over when the CLI was migrated from
manual os.Args parsing to Cobra commands. Add it back to the gateway
command and forward enableStats to NewAgentLoop.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
---
 cmd/picoclaw/internal/gateway/command.go | 4 +++-
 cmd/picoclaw/internal/gateway/helpers.go | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

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:")