remove request control hooks

Author: RishabhSaini

pkg/plugins/scorer/pd_slo_aware_router_hooks.go

@@ -74,16 +74,11 @@ func PDSLOAwareScorerFactory(name string, rawConfig json.RawMessage, handle plug
 		return nil, fmt.Errorf("failed to start latency predictor: %w", err)
 	}
 
-	// Create the SLO aware router and inject P/D-aware request builder
-	// This builder will populate the PodType field based on llm-d.ai/role labels
-	baseRouter := predictedlatency.NewPredictedLatency(cfg, predictor).WithName(name)
-	baseRouter.SetRequestBuilder(NewPDPredictionRequestBuilder())
+	// Create the SLO aware router with P/D-aware request builder
+	// The builder populates the PodType field based on llm-d.ai/role labels
+	// to distinguish prefill vs decode pods in training data
+	router := predictedlatency.NewPredictedLatency(cfg, predictor).WithName(name)
+	router.SetRequestBuilder(NewPDPredictionRequestBuilder())
 
-	// Wrap with PDSLOAwareRouter to add P/D-specific hook logic
-	// The wrapper delegates to the base router while adding P/D-specific header extraction
-	pdRouter := &PDSLOAwareRouter{
-		PredictedLatency: baseRouter,
-	}
-
-	return pdRouter, nil
+	return router, nil
 }

pkg/plugins/scorer/pd_slo_aware_scorer.go

@@ -174,19 +174,12 @@ func (s *Server) runNIXLProtocolV2(w http.ResponseWriter, r *http.Request, prefi
 	dreq.Body = io.NopCloser(strings.NewReader(string(dbody)))
 	dreq.ContentLength = int64(len(dbody))
 
-	// 2. Forward to local decoder with prefill timing headers
+	// 2. Forward to local decoder
 
 	s.logger.V(5).Info("sending request to decoder", "body", string(dbody))
 
-	// Wrap response writer to inject prefill timing headers for EPP training
-	timingWriter := &timingResponseWriter{
-		ResponseWriter:   w,
-		prefillLatencyMs: float64(prefillLatency.Milliseconds()),
-		prefillPodHost:   prefillPodHostPort,
-	}
-
-	if !s.forwardDataParallel || !s.dataParallelHandler(timingWriter, dreq) {
+	if !s.forwardDataParallel || !s.dataParallelHandler(w, dreq) {
 		s.logger.V(4).Info("sending request to decoder", "to", s.decoderURL.Host)
-		s.decoderProxy.ServeHTTP(timingWriter, dreq)
+		s.decoderProxy.ServeHTTP(w, dreq)
 	}
 }