Skip to content

Commit f83d798

Browse files
authored
prefill header is set in the form of ip:port (#233)
1 parent fb952f4 commit f83d798

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pkg/plugins/pre-request/pd_prerequest.go

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,6 @@ func (p *PrefillHeaderHandler) PreRequest(_ context.Context, request *types.LLMR
7474
return // prefill profile failed to run or we chose not to run it, no-op in this case
7575
}
7676

77-
// TODO: should the scheme be configurable (e.g., https://)?
78-
prefillURL := "http://" + net.JoinHostPort(prefillProfileRunResult.TargetPod.GetPod().Address, strconv.Itoa(targetPort))
79-
request.Headers[prefillPodHeader] = prefillURL
77+
prefillHostPort := net.JoinHostPort(prefillProfileRunResult.TargetPod.GetPod().Address, strconv.Itoa(targetPort))
78+
request.Headers[prefillPodHeader] = prefillHostPort // in the form of <ip:port>
8079
}

0 commit comments

Comments
 (0)