Skip to content

Commit 4175bc7

Browse files
fix: update stale TokenizedPrompt reference missed during rebase
The auto-merge left one test using the old TokenizedPrompt type and PerPromptTokens field in producer_test.go. Updated to use TokenizedRequest with Prompts/PromptTokens. Signed-off-by: Alberto Perdomo <aperdomo@redhat.com>
1 parent af76cee commit 4175bc7

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

  • pkg/epp/framework/plugins/requestcontrol/dataproducer/preciseprefixcache

pkg/epp/framework/plugins/requestcontrol/dataproducer/preciseprefixcache/producer_test.go

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -414,8 +414,11 @@ func TestProduce_WritesCachedBlocksByTier(t *testing.T) {
414414
RequestID: "req-by-tier",
415415
TargetModel: "test-model",
416416
Body: &fwkrh.InferenceRequestBody{
417-
TokenizedPrompt: &fwkrh.TokenizedPrompt{
418-
PerPromptTokens: [][]uint32{promptA, promptB},
417+
TokenizedRequest: &fwkrh.TokenizedRequest{
418+
Prompts: []fwkrh.PromptTokens{
419+
{TokenIDs: promptA},
420+
{TokenIDs: promptB},
421+
},
419422
},
420423
},
421424
}

0 commit comments

Comments
 (0)