simulate: audio subcommand with impairment flags

cmd/lk/simulate.go

@@ -70,7 +70,10 @@
 		simulateProjectConfig = pc
 		return nil, nil
 	},
-	Action: runSimulate,
+	Action: func(ctx context.Context, cmd *cli.Command) error {
+		return runSimulate(ctx, cmd, livekit.SimulationMode_SIMULATION_MODE_TEXT)
+	},
+	Commands: []*cli.Command{simulateAudioCommand},
 	Flags: []cli.Flag{
 		&cli.IntFlag{
 			Name:    "num-simulations",
@@ -85,10 +88,6 @@
 			Name:  "scenarios",
 			Usage: "Path to a scenarios `FILE` (yaml). If omitted, scenarios are generated from the agent's source",
 		},
-		&cli.BoolFlag{
-			Name:  "audio",
-			Usage: "Simulate speech-to-speech interactions using the agent's full audio pipeline. By default, simulations run in text-only mode.",
-		},
 		&cli.BoolFlag{
 			Name:    "yes",
 			Aliases: []string{"y"},
@@ -109,6 +108,34 @@
 	},
 }
 
+// simulateAudioCommand inherits every flag on `simulate`: flags there are
+// persistent (cli.FlagBase.Local defaults to false), so they parse on either
+// side of the subcommand name.
+var simulateAudioCommand = &cli.Command{
+	Name:            "audio",
+	Usage:           "Simulate speech-to-speech interactions using the agent's full audio pipeline",
+	Description:     "Options on lk agent simulate apply here too, e.g. --scenarios and --agent-name.",
+	ArgsUsage:       "[entrypoint]",
+	HideHelpCommand: true,
+	Action: func(ctx context.Context, cmd *cli.Command) error {
+		return runSimulate(ctx, cmd, livekit.SimulationMode_SIMULATION_MODE_AUDIO)
+	},
+	Flags: []cli.Flag{
+		&cli.BoolFlag{
+			Name:  "background-noise",
+			Usage: "Mix ambient noise into the simulated user's audio",
+		},
+		&cli.BoolFlag{
+			Name:  "low-quality-microphone",
+			Usage: "Publish the simulated user's audio as a low-quality microphone would capture it",
+		},
+		&cli.BoolFlag{
+			Name:  "packet-loss",
+			Usage: "Drop packets from the simulated user's audio track",
+		},
+	},
+}
+
 // writeGeneratedScenariosTemp writes a generated run's scenarios to a temp
 // scenarios.yaml; "" when the run carries none.
 func writeGeneratedScenariosTemp(run *livekit.SimulationRun) (string, error) {
@@ -189,6 +216,12 @@
 	viewModeRunID  string   // non-empty when --view opens a pre-existing run
 	liveAgent      bool     // --agent-name: run against an already-running agent, don't spawn one
 	warnings       []string // config-level warnings surfaced at setup (e.g. ignored flags)
+
+	// impairments on the simulated user's audio, set only in SIMULATION_MODE_AUDIO
+	backgroundNoise      bool
+	lowQualityMicrophone bool
+	packetLoss           bool
+
 	// TODO (steveyoon): add agent deployment support
 	// agentDeployment string
 }
@@ -273,7 +306,7 @@
 	return ok, nil
 }
 
-func runSimulate(ctx context.Context, cmd *cli.Command) error {
+func runSimulate(ctx context.Context, cmd *cli.Command, simulationMode livekit.SimulationMode) error {
 	pc := simulateProjectConfig
 
 	// --export is a one-shot read of a finished run, so it short-circuits
@@ -368,11 +401,6 @@
 
 	simClient := lksdk.NewAgentSimulationClient(serverURL, pc.APIKey, pc.APISecret)
 
-	simulationMode := livekit.SimulationMode_SIMULATION_MODE_TEXT
-	if cmd.Bool("audio") {
-		simulationMode = livekit.SimulationMode_SIMULATION_MODE_AUDIO
-	}
-
 	simCfg := &simulateConfig{
 		ctx:            ctx,
 		client:         simClient,
@@ -392,6 +420,12 @@
 		warnings:       simulateConfigWarnings(mode, numSimulations),
 	}
 
+	if simulationMode == livekit.SimulationMode_SIMULATION_MODE_AUDIO {
+		simCfg.backgroundNoise = cmd.Bool("background-noise")
+		simCfg.lowQualityMicrophone = cmd.Bool("low-quality-microphone")
+		simCfg.packetLoss = cmd.Bool("packet-loss")
+	}
+
 	if !isInteractive() {
 		return runSimulateCI(ctx, simCfg)
 	}
@@ -522,9 +556,12 @@
 
 func createSimulationRun(ctx context.Context, c *simulateConfig) (string, *livekit.PresignedPostRequest, error) {
 	req := &livekit.SimulationRun_Create_Request{
-		AgentName:      c.agentName,
-		NumSimulations: c.numSimulations,
-		Mode:           c.simulationMode,
+		AgentName:            c.agentName,
+		NumSimulations:       c.numSimulations,
+		Mode:                 c.simulationMode,
+		BackgroundNoise:      c.backgroundNoise,
+		LowQualityMicrophone: c.lowQualityMicrophone,
+		PacketLoss:           c.packetLoss,
 	}
 	if c.concurrency > 0 {
 		req.Concurrency = &c.concurrency