Merge pull request #1070 from narengogi/fix/support-provider-specific-fields-in-snake-case

Fix/support provider specific fields for bedrock in snake case

Author: VisargD

src/providers/bedrock/chatComplete.ts

@@ -32,12 +32,18 @@ import {
 
 export interface BedrockChatCompletionsParams extends Params {
   additionalModelRequestFields?: Record<string, any>;
+  additional_model_request_fields?: Record<string, any>;
   additionalModelResponseFieldPaths?: string[];
   guardrailConfig?: {
     guardrailIdentifier: string;
     guardrailVersion: string;
     trace?: string;
   };
+  guardrail_config?: {
+    guardrailIdentifier: string;
+    guardrailVersion: string;
+    trace?: string;
+  };
   anthropic_version?: string;
   countPenalty?: number;
 }
@@ -312,10 +318,18 @@ export const BedrockConverseChatCompleteConfig: ProviderConfig = {
     param: 'guardrailConfig',
     required: false,
   },
+  guardrail_config: {
+    param: 'guardrailConfig',
+    required: false,
+  },
   additionalModelResponseFieldPaths: {
     param: 'additionalModelResponseFieldPaths',
     required: false,
   },
+  additional_model_response_field_paths: {
+    param: 'additionalModelResponseFieldPaths',
+    required: false,
+  },
   max_tokens: {
     param: 'inferenceConfig',
     transform: (params: BedrockChatCompletionsParams) =>
@@ -346,6 +360,11 @@ export const BedrockConverseChatCompleteConfig: ProviderConfig = {
     transform: (params: BedrockChatCompletionsParams) =>
       transformAdditionalModelRequestFields(params),
   },
+  additional_model_request_fields: {
+    param: 'additionalModelRequestFields',
+    transform: (params: BedrockChatCompletionsParams) =>
+      transformAdditionalModelRequestFields(params),
+  },
   top_k: {
     param: 'additionalModelRequestFields',
     transform: (params: BedrockChatCompletionsParams) =>
@@ -701,6 +720,11 @@ export const BedrockConverseAnthropicChatCompleteConfig: ProviderConfig = {
     transform: (params: BedrockConverseAnthropicChatCompletionsParams) =>
       transformAnthropicAdditionalModelRequestFields(params),
   },
+  additional_model_request_fields: {
+    param: 'additionalModelRequestFields',
+    transform: (params: BedrockConverseAnthropicChatCompletionsParams) =>
+      transformAnthropicAdditionalModelRequestFields(params),
+  },
   top_k: {
     param: 'additionalModelRequestFields',
     transform: (params: BedrockConverseAnthropicChatCompletionsParams) =>
@@ -722,7 +746,9 @@ export const BedrockConverseAnthropicChatCompleteConfig: ProviderConfig = {
       transformAnthropicAdditionalModelRequestFields(params),
   },
   anthropic_beta: {
-    param: 'anthropic_beta',
+    param: 'additionalModelRequestFields',
+    transform: (params: BedrockConverseAnthropicChatCompletionsParams) =>
+      transformAnthropicAdditionalModelRequestFields(params),
   },
 };
 
@@ -733,6 +759,11 @@ export const BedrockConverseCohereChatCompleteConfig: ProviderConfig = {
     transform: (params: BedrockConverseCohereChatCompletionsParams) =>
       transformCohereAdditionalModelRequestFields(params),
   },
+  additional_model_request_fields: {
+    param: 'additionalModelRequestFields',
+    transform: (params: BedrockConverseCohereChatCompletionsParams) =>
+      transformCohereAdditionalModelRequestFields(params),
+  },
   top_k: {
     param: 'additionalModelRequestFields',
     transform: (params: BedrockConverseCohereChatCompletionsParams) =>
@@ -767,6 +798,11 @@ export const BedrockConverseAI21ChatCompleteConfig: ProviderConfig = {
     transform: (params: BedrockConverseAI21ChatCompletionsParams) =>
       transformAI21AdditionalModelRequestFields(params),
   },
+  additional_model_request_fields: {
+    param: 'additionalModelRequestFields',
+    transform: (params: BedrockConverseAI21ChatCompletionsParams) =>
+      transformAI21AdditionalModelRequestFields(params),
+  },
   top_k: {
     param: 'additionalModelRequestFields',
     transform: (params: BedrockConverseAI21ChatCompletionsParams) =>

src/providers/bedrock/utils.ts

@@ -93,7 +93,9 @@ export const transformAdditionalModelRequestFields = (
   params: BedrockChatCompletionsParams
 ) => {
   const additionalModelRequestFields: Record<string, any> =
-    params.additionalModelRequestFields || {};
+    params.additionalModelRequestFields ||
+    params.additional_model_request_fields ||
+    {};
   if (params['top_k']) {
     additionalModelRequestFields['top_k'] = params['top_k'];
   }
@@ -107,7 +109,9 @@ export const transformAnthropicAdditionalModelRequestFields = (
   params: BedrockConverseAnthropicChatCompletionsParams
 ) => {
   const additionalModelRequestFields: Record<string, any> =
-    params.additionalModelRequestFields || {};
+    params.additionalModelRequestFields ||
+    params.additional_model_request_fields ||
+    {};
   if (params['top_k']) {
     additionalModelRequestFields['top_k'] = params['top_k'];
   }
@@ -130,7 +134,9 @@ export const transformCohereAdditionalModelRequestFields = (
   params: BedrockConverseCohereChatCompletionsParams
 ) => {
   const additionalModelRequestFields: Record<string, any> =
-    params.additionalModelRequestFields || {};
+    params.additionalModelRequestFields ||
+    params.additional_model_request_fields ||
+    {};
   if (params['top_k']) {
     additionalModelRequestFields['top_k'] = params['top_k'];
   }
@@ -155,7 +161,9 @@ export const transformAI21AdditionalModelRequestFields = (
   params: BedrockConverseAI21ChatCompletionsParams
 ) => {
   const additionalModelRequestFields: Record<string, any> =
-    params.additionalModelRequestFields || {};
+    params.additionalModelRequestFields ||
+    params.additional_model_request_fields ||
+    {};
   if (params['top_k']) {
     additionalModelRequestFields['top_k'] = params['top_k'];
   }