enable the servers group

cmd/thv/app/commands.go

@@ -53,7 +53,7 @@ func NewRootCmd(enableUpdates bool) *cobra.Command {
 	rootCmd.AddCommand(newSecretCommand())
 	rootCmd.AddCommand(inspectorCommand())
 	rootCmd.AddCommand(newMCPCommand())
-	// rootCmd.AddCommand(groupCmd) //  TODO: add back in once we have a working group command, and update the docs
+	rootCmd.AddCommand(groupCmd)
 
 	// Silence printing the usage on error
 	rootCmd.SilenceUsage = true

cmd/thv/app/common.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/stacklok/toolhive/pkg/config"
 	"github.com/stacklok/toolhive/pkg/secrets"
+	"github.com/stacklok/toolhive/pkg/validation"
 	"github.com/stacklok/toolhive/pkg/workloads"
 )
 
@@ -147,7 +148,7 @@ func completeLogsArgs(cmd *cobra.Command, args []string, _ string) ([]string, co
 
 // ValidateGroupFlag returns a cobra PreRunE-compatible function
 // that validates the --group flag *if provided*.
-/*func validateGroupFlag() func(cmd *cobra.Command, args []string) error {
+func validateGroupFlag() func(cmd *cobra.Command, args []string) error {
 	return func(cmd *cobra.Command, _ []string) error {
 		groupName, err := cmd.Flags().GetString("group")
 		if err != nil {
@@ -166,4 +167,4 @@ func completeLogsArgs(cmd *cobra.Command, args []string, _ string) ([]string, co
 
 		return nil
 	}
-}*/
+}

cmd/thv/app/list.go

@@ -30,10 +30,9 @@ func init() {
 	listCmd.Flags().BoolVarP(&listAll, "all", "a", false, "Show all workloads (default shows just running)")
 	listCmd.Flags().StringVar(&listFormat, "format", FormatText, "Output format (json, text, or mcpservers)")
 	listCmd.Flags().StringArrayVarP(&listLabelFilter, "label", "l", []string{}, "Filter workloads by labels (format: key=value)")
-	// TODO: Re-enable when group functionality is complete
-	// listCmd.Flags().StringVar(&listGroupFilter, "group", "", "Filter workloads by group")
+	listCmd.Flags().StringVar(&listGroupFilter, "group", "", "Filter workloads by group")
 
-	// listCmd.PreRunE = validateGroupFlag()
+	listCmd.PreRunE = validateGroupFlag()
 }
 
 func listCmdFunc(cmd *cobra.Command, _ []string) error {

cmd/thv/app/restart.go

@@ -27,13 +27,12 @@ var restartCmd = &cobra.Command{
 
 func init() {
 	restartCmd.Flags().BoolVarP(&restartAll, "all", "a", false, "Restart all MCP servers")
-	// TODO: Uncomment when groups are fully supported
-	//restartCmd.Flags().StringVarP(&restartGroup, "group", "g", "", "Restart all MCP servers in a specific group")
-	//
-	//// Mark the flags as mutually exclusive
-	//restartCmd.MarkFlagsMutuallyExclusive("all", "group")
+	restartCmd.Flags().StringVarP(&restartGroup, "group", "g", "", "Restart all MCP servers in a specific group")
 
-	// restartCmd.PreRunE = validateGroupFlag()
+	// Mark the flags as mutually exclusive
+	restartCmd.MarkFlagsMutuallyExclusive("all", "group")
+
+	restartCmd.PreRunE = validateGroupFlag()
 }
 
 func restartCmdFunc(cmd *cobra.Command, args []string) error {

cmd/thv/app/rm.go

@@ -20,10 +20,9 @@ var rmCmd = &cobra.Command{
 }
 
 func init() {
-	// TODO: Re-enable when group functionality is complete
-	// rmCmd.Flags().String("group", "", "Delete all workloads in the specified group")
+	rmCmd.Flags().String("group", "", "Delete all workloads in the specified group")
 
-	// rmCmd.PreRunE = validateGroupFlag()
+	rmCmd.PreRunE = validateGroupFlag()
 }
 
 //nolint:gocyclo // This function is complex but manageable

cmd/thv/app/run_flags.go

@@ -84,9 +84,8 @@ func AddRunFlags(cmd *cobra.Command, config *RunFlags) {
 	cmd.Flags().StringVar(&config.Transport, "transport", "", "Transport mode (sse, streamable-http or stdio)")
 	cmd.Flags().StringVar(&config.ProxyMode, "proxy-mode", "sse", "Proxy mode for stdio transport (sse or streamable-http)")
 	cmd.Flags().StringVar(&config.Name, "name", "", "Name of the MCP server (auto-generated from image if not provided)")
-	// TODO: Re-enable when group functionality is complete
-	// cmd.Flags().StringVar(&config.Group, "group", "default",
-	//	"Name of the group this workload belongs to (defaults to 'default' if not specified)")
+	cmd.Flags().StringVar(&config.Group, "group", "default",
+		"Name of the group this workload belongs to (defaults to 'default' if not specified)")
 	cmd.Flags().StringVar(&config.Host, "host", transport.LocalhostIPv4, "Host for the HTTP proxy to listen on (IP or hostname)")
 	cmd.Flags().IntVar(&config.ProxyPort, "proxy-port", 0, "Port for the HTTP proxy to listen on (host port)")
 	cmd.Flags().IntVar(&config.TargetPort, "target-port", 0,

cmd/thv/main.go

@@ -7,6 +7,7 @@ import (
 	"github.com/stacklok/toolhive/cmd/thv/app"
 	"github.com/stacklok/toolhive/pkg/client"
 	"github.com/stacklok/toolhive/pkg/container/runtime"
+	"github.com/stacklok/toolhive/pkg/groups"
 	"github.com/stacklok/toolhive/pkg/logger"
 )
 
@@ -20,8 +21,7 @@ func main() {
 
 	// Check and perform default group migration if needed
 	// Migrates existing workloads to the default group, only executes once
-	// TODO: Re-enable when group functionality is complete
-	// groups.CheckAndPerformDefaultGroupMigration()
+	groups.CheckAndPerformDefaultGroupMigration()
 
 	// Skip update check for completion command or if we are running in kubernetes
 	if err := app.NewRootCmd(!app.IsCompletionCommand(os.Args) && !runtime.IsKubernetesRuntime()).Execute(); err != nil {

test/e2e/e2e_suite_test.go

@@ -1,4 +1,4 @@
-package e2e_test
+package e2e
 
 import (
 	"testing"