Merge pull request #1266 from Portkey-AI/chore/pass-missing-arguments-to-transformer-service

fix: add missing parameter for bedrock and vertex transformer calls

Author: VisargD

src/providers/bedrock/uploadFile.ts

@@ -142,7 +142,8 @@ class AwsMultipartUploadHandler {
               this,
               partNumber,
               purpose ?? 'batch',
-              modelType ?? 'chat'
+              modelType ?? 'chat',
+              this.providerOptions
             );
           this.contentLength += uploadLength;
           partNumber++;
@@ -251,7 +252,8 @@ const transformAndUploadFileContentParts = async (
   handler: AwsMultipartUploadHandler,
   partNumber: number,
   purpose: string,
-  modelType: string
+  modelType: string,
+  providerOptions: Options
 ): Promise<[string, number]> => {
   let transformedChunkToUpload = '';
   const jsonLines = chunk.split('\n');
@@ -279,7 +281,11 @@ const transformAndUploadFileContentParts = async (
       }
       const transformedLine = {
         recordId: json.custom_id,
-        modelInput: transformUsingProviderConfig(providerConfig, json.body),
+        modelInput: transformUsingProviderConfig(
+          providerConfig,
+          json.body,
+          providerOptions
+        ),
       };
       transformedChunkToUpload += JSON.stringify(transformedLine) + '\r\n';
       buffer = buffer.slice(line.length + 1);

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

@@ -122,7 +122,8 @@ export const GoogleFileUploadRequestHandler: RequestHandler<
           const toTranspose = purpose === 'batch' ? json.body : json;
           const transformedBody = transformUsingProviderConfig(
             providerConfig,
-            toTranspose
+            toTranspose,
+            providerOptions
           );
 
           delete transformedBody['model'];