@@ -93,7 +93,9 @@ export const transformAdditionalModelRequestFields = (
93
93
params : BedrockChatCompletionsParams
94
94
) => {
95
95
const additionalModelRequestFields : Record < string , any > =
96
- params . additionalModelRequestFields || { } ;
96
+ params . additionalModelRequestFields ||
97
+ params . additional_model_request_fields ||
98
+ { } ;
97
99
if ( params [ 'top_k' ] ) {
98
100
additionalModelRequestFields [ 'top_k' ] = params [ 'top_k' ] ;
99
101
}
@@ -104,7 +106,9 @@ export const transformAnthropicAdditionalModelRequestFields = (
104
106
params : BedrockConverseAnthropicChatCompletionsParams
105
107
) => {
106
108
const additionalModelRequestFields : Record < string , any > =
107
- params . additionalModelRequestFields || { } ;
109
+ params . additionalModelRequestFields ||
110
+ params . additional_model_request_fields ||
111
+ { } ;
108
112
if ( params [ 'top_k' ] ) {
109
113
additionalModelRequestFields [ 'top_k' ] = params [ 'top_k' ] ;
110
114
}
@@ -127,7 +131,9 @@ export const transformCohereAdditionalModelRequestFields = (
127
131
params : BedrockConverseCohereChatCompletionsParams
128
132
) => {
129
133
const additionalModelRequestFields : Record < string , any > =
130
- params . additionalModelRequestFields || { } ;
134
+ params . additionalModelRequestFields ||
135
+ params . additional_model_request_fields ||
136
+ { } ;
131
137
if ( params [ 'top_k' ] ) {
132
138
additionalModelRequestFields [ 'top_k' ] = params [ 'top_k' ] ;
133
139
}
@@ -152,7 +158,9 @@ export const transformAI21AdditionalModelRequestFields = (
152
158
params : BedrockConverseAI21ChatCompletionsParams
153
159
) => {
154
160
const additionalModelRequestFields : Record < string , any > =
155
- params . additionalModelRequestFields || { } ;
161
+ params . additionalModelRequestFields ||
162
+ params . additional_model_request_fields ||
163
+ { } ;
156
164
if ( params [ 'top_k' ] ) {
157
165
additionalModelRequestFields [ 'top_k' ] = params [ 'top_k' ] ;
158
166
}
0 commit comments