Method handleStatusEvent

scaletest/chat/run.go:221–316

(ctx context.Context, chatID uuid.UUID, logger slog.Logger, status codersdk.ChatStatus)

}

func (r *Runner) handleStatusEvent(ctx context.Context, chatID uuid.UUID, logger slog.Logger, status codersdk.ChatStatus) (bool, error) {
	if status == r.lastStatus {
		return false, nil
	}
	if status == codersdk.ChatStatusWaiting &&
		!r.sawTurnFirstOutput &&
		(r.sawTurnRunning || r.result.turnsCompleted > 0) {
		return false, nil
	}
	r.lastStatus = status

	switch status {
	case codersdk.ChatStatusRunning:
		r.sawTurnRunning = true
		r.cfg.Metrics.ChatTimeToRunningSeconds.WithLabelValues(r.currentPhase).Observe(time.Since(r.turnStartTime).Seconds())
		logger.Info(ctx, "chat reached running status",
			slog.F("phase", r.currentPhase),
		)
		return false, nil
	case codersdk.ChatStatusWaiting:
		r.result.turnsCompleted++
		turnDuration := time.Since(r.turnStartTime)
		r.cfg.Metrics.ChatTimeToTerminalStatusSeconds.WithLabelValues(r.currentPhase).Observe(turnDuration.Seconds())
		r.cfg.Metrics.ChatTerminalStatusTotal.WithLabelValues(string(codersdk.ChatStatusWaiting)).Inc()
		r.cfg.Metrics.ChatTurnsCompletedTotal.Inc()
		logger.Info(ctx, "chat completed turn",
			slog.F("turn", r.result.turnsCompleted),
			slog.F("turns", r.cfg.Turns),
			slog.F("duration", turnDuration),
		)
		if r.result.turnsCompleted >= r.cfg.Turns {
			r.result.finalStatus = string(codersdk.ChatStatusWaiting)
			conversationDuration := time.Since(r.conversationStart)
			logger.Info(ctx, "chat reached terminal status",
				slog.F("status", codersdk.ChatStatusWaiting),
				slog.F("duration", conversationDuration),
				slog.F("turns_completed", r.result.turnsCompleted),
			)
			return true, nil
		}

		// After the very first turn completes, mark this runner ready
		// for the CLI-coordinated turn-start gate. The inter-phase
		// delay measures the gap between every chat actually finishing its
		// initial turn and the start of the follow-up turns, not the gap
		// between CreateChat returning and the next turn.
		if r.result.turnsCompleted == 1 {
			r.markTurnStartReady()
			if r.cfg.StartTurnsChan != nil {
				logger.Info(ctx, "chat waiting for turn start release",
					slog.F("turn_start_delay", r.cfg.TurnStartDelay),
				)
				select {
				case <-ctx.Done():
					return false, ctx.Err()
				case <-r.cfg.StartTurnsChan:
				}
			}