Skip to content

Commit af76cee

Browse files
fix: Rename missing instances
Signed-off-by: Alberto Perdomo <aperdomo@redhat.com>
1 parent 87666fb commit af76cee

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

pkg/epp/framework/plugins/requestcontrol/dataproducer/tokenizer/tokenizer_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -792,8 +792,8 @@ func TestProduce_MessagesRequest(t *testing.T) {
792792
},
793793
}
794794
require.NoError(t, p.Produce(context.Background(), req, nil))
795-
require.NotNil(t, req.Body.TokenizedPrompt)
796-
assert.Equal(t, [][]uint32{wantTokens}, req.Body.TokenizedPrompt.PerPromptTokens)
795+
require.NotNil(t, req.Body.TokenizedRequest)
796+
assert.Equal(t, []fwkrh.PromptTokens{{TokenIDs: wantTokens}}, req.Body.TokenizedRequest.Prompts)
797797

798798
pm, ok := gotPayload.AsMap()
799799
require.True(t, ok, "RenderChat payload must be a map")

0 commit comments

Comments
 (0)