Skip to content

Commit 4a30052

Browse files
committed
chore: support fields for bedrock in snake_case as well because portkey-node-sdk converts all parameters to snake_case
1 parent cd0cc8c commit 4a30052

File tree

2 files changed

+26
-4
lines changed

2 files changed

+26
-4
lines changed

src/providers/bedrock/chatComplete.ts

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,12 +32,18 @@ import {
3232

3333
export interface BedrockChatCompletionsParams extends Params {
3434
additionalModelRequestFields?: Record<string, any>;
35+
additional_model_request_fields?: Record<string, any>;
3536
additionalModelResponseFieldPaths?: string[];
3637
guardrailConfig?: {
3738
guardrailIdentifier: string;
3839
guardrailVersion: string;
3940
trace?: string;
4041
};
42+
guardrail_config?: {
43+
guardrailIdentifier: string;
44+
guardrailVersion: string;
45+
trace?: string;
46+
};
4147
anthropic_version?: string;
4248
countPenalty?: number;
4349
}
@@ -312,10 +318,18 @@ export const BedrockConverseChatCompleteConfig: ProviderConfig = {
312318
param: 'guardrailConfig',
313319
required: false,
314320
},
321+
guardrail_config: {
322+
param: 'guardrailConfig',
323+
required: false,
324+
},
315325
additionalModelResponseFieldPaths: {
316326
param: 'additionalModelResponseFieldPaths',
317327
required: false,
318328
},
329+
additional_model_response_field_paths: {
330+
param: 'additionalModelResponseFieldPaths',
331+
required: false,
332+
},
319333
max_tokens: {
320334
param: 'inferenceConfig',
321335
transform: (params: BedrockChatCompletionsParams) =>

src/providers/bedrock/utils.ts

Lines changed: 12 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,9 @@ export const transformAdditionalModelRequestFields = (
9393
params: BedrockChatCompletionsParams
9494
) => {
9595
const additionalModelRequestFields: Record<string, any> =
96-
params.additionalModelRequestFields || {};
96+
params.additionalModelRequestFields ||
97+
params.additional_model_request_fields ||
98+
{};
9799
if (params['top_k']) {
98100
additionalModelRequestFields['top_k'] = params['top_k'];
99101
}
@@ -104,7 +106,9 @@ export const transformAnthropicAdditionalModelRequestFields = (
104106
params: BedrockConverseAnthropicChatCompletionsParams
105107
) => {
106108
const additionalModelRequestFields: Record<string, any> =
107-
params.additionalModelRequestFields || {};
109+
params.additionalModelRequestFields ||
110+
params.additional_model_request_fields ||
111+
{};
108112
if (params['top_k']) {
109113
additionalModelRequestFields['top_k'] = params['top_k'];
110114
}
@@ -127,7 +131,9 @@ export const transformCohereAdditionalModelRequestFields = (
127131
params: BedrockConverseCohereChatCompletionsParams
128132
) => {
129133
const additionalModelRequestFields: Record<string, any> =
130-
params.additionalModelRequestFields || {};
134+
params.additionalModelRequestFields ||
135+
params.additional_model_request_fields ||
136+
{};
131137
if (params['top_k']) {
132138
additionalModelRequestFields['top_k'] = params['top_k'];
133139
}
@@ -152,7 +158,9 @@ export const transformAI21AdditionalModelRequestFields = (
152158
params: BedrockConverseAI21ChatCompletionsParams
153159
) => {
154160
const additionalModelRequestFields: Record<string, any> =
155-
params.additionalModelRequestFields || {};
161+
params.additionalModelRequestFields ||
162+
params.additional_model_request_fields ||
163+
{};
156164
if (params['top_k']) {
157165
additionalModelRequestFields['top_k'] = params['top_k'];
158166
}

0 commit comments

Comments
 (0)