refactor: Allow export prefix SchedulingContextState for use across plugins

Signed-off-by: Kfir Toledo <kfir.toledo@ibm.com>

Author: kfirtoledo

pkg/epp/scheduling/framework/plugins/multi/prefix/plugin.go

@@ -90,25 +90,25 @@ func (s ServerID) String() string {
 }
 
 // compile-time type validation
-var _ types.StateData = &schedulingContextState{}
+var _ types.StateData = &SchedulingContextState{}
 
-// This is the state of this plugin to be used during a scheduling cycle.
-type schedulingContextState struct {
+// SchedulingContextState is the state of this plugin to be used during a scheduling cycle.
+type SchedulingContextState struct {
 	// PrefixHashes is a list of prefix hashes of the request prompt broken into blocks.
 	PrefixHashes []BlockHash
 	// A map of server to its longest prefix cache match length.
 	PrefixCacheServers map[ServerID]int
 }
 
-func (s *schedulingContextState) Clone() types.StateData {
+func (s *SchedulingContextState) Clone() types.StateData {
 	prefixHashes := make([]BlockHash, len(s.PrefixHashes))
 	copy(prefixHashes, s.PrefixHashes)
 	prefixCacheServers := make(map[ServerID]int, len(s.PrefixCacheServers))
 	for key, value := range s.PrefixCacheServers {
 		prefixCacheServers[key] = value
 	}
 
-	return &schedulingContextState{
+	return &SchedulingContextState{
 		PrefixHashes:       prefixHashes,
 		PrefixCacheServers: prefixCacheServers,
 	}
@@ -171,7 +171,7 @@ func (m *Plugin) Score(ctx context.Context, cycleState *types.CycleState, reques
 	loggerTrace := log.FromContext(ctx).V(logutil.TRACE)
 	// pre score step, hashing prompt and find longest prefix match.
 	hashes := hashPrompt(ctx, request, m.HashBlockSize, m.MaxPrefixBlocksToMatch)
-	state := &schedulingContextState{
+	state := &SchedulingContextState{
 		PrefixHashes:       hashes,
 		PrefixCacheServers: m.matchLongestPrefix(ctx, hashes),
 	}
@@ -235,15 +235,15 @@ func (m *Plugin) matchLongestPrefix(ctx context.Context, hashes []BlockHash) map
 	return res
 }
 
-// getPrefixState returns the cycle state as a schedulingContextState.
-func (m *Plugin) getPrefixState(cycleState *types.CycleState) (*schedulingContextState, error) {
+// getPrefixState returns the cycle state as a SchedulingContextState.
+func (m *Plugin) getPrefixState(cycleState *types.CycleState) (*SchedulingContextState, error) {
 	prefixStateKey := types.StateKey(m.Type())
 	state, err := cycleState.Read(prefixStateKey)
 	if err != nil {
 		return nil, fmt.Errorf("failed reading %q from CycleState: %w", prefixStateKey, err)
 	}
 
-	prefixSchedulingState, ok := state.(*schedulingContextState)
+	prefixSchedulingState, ok := state.(*SchedulingContextState)
 	if !ok {
 		return nil, fmt.Errorf("invalid Prefix state, got type %T", state)
 	}