Populated the RequestId field of the LLMRequest

shmuelk · shmuelk · commit 05954ac65233 · 2025-05-08T12:50:33.000+03:00
Signed-off-by: Shmuel Kallner &lt;kallner@il.ibm.com&gt;
diff --git a/pkg/epp/requestcontrol/director.go b/pkg/epp/requestcontrol/director.go
@@ -31,6 +31,7 @@ import (
 	schedulingtypes "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/scheduling/types"
 	errutil "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/error"
 	logutil "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/logging"
+	requtil "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/request"
 )
 
 type Scheduler interface {
@@ -82,6 +83,7 @@ func (d *Director) HandleRequest(ctx context.Context, reqCtx *handlers.RequestCo
 	}
 
 	llmReq := &schedulingtypes.LLMRequest{
+		RequestId:           reqCtx.Request.Headers[requtil.RequestIdHeaderKey],
 		Model:               reqCtx.Model,
 		ResolvedTargetModel: reqCtx.ResolvedTargetModel,
 		Critical:            modelObj.Spec.Criticality != nil && *modelObj.Spec.Criticality == v1alpha2.Critical,

Original file line number	Diff line number	Diff line change
`@@ -31,6 +31,7 @@ import (`
`31`	`31`	`schedulingtypes "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/scheduling/types"`
`32`	`32`	`errutil "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/error"`
`33`	`33`	`logutil "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/logging"`
	`34`	`+ requtil "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/request"`
`34`	`35`	`)`
`35`	`36`
`36`	`37`	`type Scheduler interface {`
`@@ -82,6 +83,7 @@ func (d Director) HandleRequest(ctx context.Context, reqCtx handlers.RequestCo`
`82`	`83`	`}`
`83`	`84`
`84`	`85`	`llmReq := &schedulingtypes.LLMRequest{`
	`86`	`+ RequestId: reqCtx.Request.Headers[requtil.RequestIdHeaderKey],`
`85`	`87`	`Model: reqCtx.Model,`
`86`	`88`	`ResolvedTargetModel: reqCtx.ResolvedTargetModel,`
`87`	`89`	`Critical: modelObj.Spec.Criticality != nil && *modelObj.Spec.Criticality == v1alpha2.Critical,`