Skip to content

Commit 5641863

Browse files
committed
fix
Signed-off-by: yxia216 <[email protected]>
1 parent 68e1937 commit 5641863

File tree

1 file changed

+4
-20
lines changed

1 file changed

+4
-20
lines changed

internal/translator/openai_gcpvertexai_test.go

Lines changed: 4 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -781,27 +781,11 @@ func TestOpenAIToGCPVertexAITranslatorV1ChatCompletion_RequestBody(t *testing.T)
781781
},
782782
onRetry: false,
783783
wantError: false,
784-
wantHeaderMut: &extprocv3.HeaderMutation{
785-
SetHeaders: []*corev3.HeaderValueOption{
786-
{
787-
Header: &corev3.HeaderValue{
788-
Key: ":path",
789-
RawValue: []byte("publishers/google/models/gemini-1.5-pro:generateContent"),
790-
},
791-
},
792-
{
793-
Header: &corev3.HeaderValue{
794-
Key: "Content-Length",
795-
RawValue: []byte("190"),
796-
},
797-
},
798-
},
799-
},
800-
wantBody: &extprocv3.BodyMutation{
801-
Mutation: &extprocv3.BodyMutation_Body{
802-
Body: wantBdyWithEnterpriseWebSearch,
803-
},
784+
wantHeaderMut: []internalapi.Header{
785+
{":path", "publishers/google/models/gemini-1.5-pro:generateContent"},
786+
{"content-length", "190"},
804787
},
788+
wantBody: wantBdyWithEnterpriseWebSearch,
805789
},
806790
}
807791

0 commit comments

Comments
 (0)