Skip to content

Commit fb6753a

Browse files
committed
Move to internal package
1 parent f5f2638 commit fb6753a

File tree

3 files changed

+6
-7
lines changed

3 files changed

+6
-7
lines changed

pkg/controller/generic/interface.go

Lines changed: 0 additions & 5 deletions
This file was deleted.

pkg/controller/launcher-populator/interface.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,4 +26,8 @@ const (
2626
LauncherConfigNameLabelKey = "dual-pods.llm-d.ai/launcher-config-name"
2727

2828
NodeNameLabelKey = "dual-pods.llm-d.ai/node-name"
29+
30+
NominalHashAnnotationKey = "dual-pods.llm-d.ai/nominal"
31+
32+
RequesterAnnotationKey = "dual-pods.llm-d.ai/requester"
2933
)

pkg/controller/launcher-populator/populator.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -446,7 +446,7 @@ func (ctl *controller) deleteExcessLaunchers(ctx context.Context, launchers []co
446446
// isLauncherBoundToServerRequestingPod checks if the launcher pod is bound to any server-requesting pod
447447
func (ctl *controller) isLauncherBoundToServerRequestingPod(launcherPod corev1.Pod) bool {
448448
// Check if the launcher pod has annotations indicating assignment to a server-requesting pod
449-
requesterAnnotationValue, exists := launcherPod.Annotations[genctlr.RequesterAnnotationKey]
449+
requesterAnnotationValue, exists := launcherPod.Annotations[RequesterAnnotationKey]
450450
if !exists {
451451
return false
452452
}
@@ -494,7 +494,7 @@ func (ctl *controller) buildPodFromTemplate(template corev1.PodTemplateSpec, key
494494
if pod.Annotations == nil {
495495
pod.Annotations = make(map[string]string)
496496
}
497-
pod.Annotations = dualpods.MapSet(pod.Annotations, genctlr.NominalHashAnnotationKey, nominalHash)
497+
pod.Annotations = dualpods.MapSet(pod.Annotations, NominalHashAnnotationKey, nominalHash)
498498

499499
cIdx, serverPort, err := utils.GetInferenceServerPort(pod)
500500
if err != nil {

0 commit comments

Comments
 (0)