Skip to content

Commit 4a2c02e

Browse files
committed
Update directory structure
1 parent 80e07e0 commit 4a2c02e

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

pkg/epp/datalayer/plugins/data_types.go renamed to pkg/epp/datalayer/plugins/approximateprefix/data_types.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ See the License for the specific language governing permissions and
1414
limitations under the License.
1515
*/
1616

17-
package plugins
17+
package approximateprefix
1818

1919
import (
2020
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/datalayer"

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ import (
2828
k8stypes "k8s.io/apimachinery/pkg/types"
2929
"sigs.k8s.io/controller-runtime/pkg/log"
3030

31-
dplugins "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/datalayer/plugins"
31+
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/datalayer/plugins/approximateprefix"
3232
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/metrics"
3333
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/plugins"
3434
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/requestcontrol"
@@ -208,7 +208,7 @@ func (p *Plugin) WithName(name string) *Plugin {
208208
}
209209

210210
func (p *Plugin) Produces() map[string]any {
211-
return map[string]any{dplugins.PrefixCacheMatchInfoKey: dplugins.PrefixCacheMatchInfo{}}
211+
return map[string]any{approximateprefix.PrefixCacheMatchInfoKey: approximateprefix.PrefixCacheMatchInfo{}}
212212
}
213213

214214
func (p *Plugin) Consumes() map[string]any {
@@ -226,7 +226,7 @@ func (p *Plugin) PrepareRequestData(ctx context.Context, request *types.LLMReque
226226

227227
for _, pod := range pods {
228228
matchLen := state.PrefixCacheServers[ServerID(pod.GetPod().NamespacedName)]
229-
pod.Put(dplugins.PrefixCacheMatchInfoKey, dplugins.NewPrefixCacheMatchInfo(matchLen, total))
229+
pod.Put(approximateprefix.PrefixCacheMatchInfoKey, approximateprefix.NewPrefixCacheMatchInfo(matchLen, total))
230230
}
231231
return nil
232232
}

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ import (
3030
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/backend"
3131
backendmetrics "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/backend/metrics"
3232
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/datalayer"
33-
dplugins "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/datalayer/plugins"
33+
dplugins "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/datalayer/plugins/approximateprefix"
3434
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/plugins"
3535
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/requestcontrol"
3636
"sigs.k8s.io/gateway-api-inference-extension/pkg/epp/scheduling/types"

0 commit comments

Comments
 (0)