Skip to content

Commit 1e5f9de

Browse files
authored
Merge branch 'main' into feat/release-1.11.1
2 parents d779a4d + 0c6d2ff commit 1e5f9de

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

src/providers/bedrock/uploadFile.ts

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,8 @@ class AwsMultipartUploadHandler {
142142
this,
143143
partNumber,
144144
purpose ?? 'batch',
145-
modelType ?? 'chat'
145+
modelType ?? 'chat',
146+
this.providerOptions
146147
);
147148
this.contentLength += uploadLength;
148149
partNumber++;
@@ -251,7 +252,8 @@ const transformAndUploadFileContentParts = async (
251252
handler: AwsMultipartUploadHandler,
252253
partNumber: number,
253254
purpose: string,
254-
modelType: string
255+
modelType: string,
256+
providerOptions: Options
255257
): Promise<[string, number]> => {
256258
let transformedChunkToUpload = '';
257259
const jsonLines = chunk.split('\n');
@@ -279,7 +281,11 @@ const transformAndUploadFileContentParts = async (
279281
}
280282
const transformedLine = {
281283
recordId: json.custom_id,
282-
modelInput: transformUsingProviderConfig(providerConfig, json.body),
284+
modelInput: transformUsingProviderConfig(
285+
providerConfig,
286+
json.body,
287+
providerOptions
288+
),
283289
};
284290
transformedChunkToUpload += JSON.stringify(transformedLine) + '\r\n';
285291
buffer = buffer.slice(line.length + 1);

src/providers/google-vertex-ai/uploadFile.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,8 @@ export const GoogleFileUploadRequestHandler: RequestHandler<
122122
const toTranspose = purpose === 'batch' ? json.body : json;
123123
const transformedBody = transformUsingProviderConfig(
124124
providerConfig,
125-
toTranspose
125+
toTranspose,
126+
providerOptions
126127
);
127128

128129
delete transformedBody['model'];

0 commit comments

Comments
 (0)