@@ -200,8 +200,8 @@ def _validate_request_metadata(self, metadata: dict) -> None:
200
200
llm_provider = "bedrock" ,
201
201
)
202
202
203
- key_pattern = re .compile (r' ^[a-zA-Z0-9\s:_@$#=/+,.-]{1,256}$' )
204
- value_pattern = re .compile (r' ^[a-zA-Z0-9\s:_@$#=/+,.-]{0,256}$' )
203
+ key_pattern = re .compile (r" ^[a-zA-Z0-9\s:_@$#=/+,.-]{1,256}$" )
204
+ value_pattern = re .compile (r" ^[a-zA-Z0-9\s:_@$#=/+,.-]{0,256}$" )
205
205
206
206
for key , value in metadata .items ():
207
207
if not isinstance (key , str ):
@@ -762,7 +762,9 @@ def _handle_top_k_value(self, model: str, inference_params: dict) -> dict:
762
762
763
763
return {}
764
764
765
- def _prepare_request_params (self , optional_params : dict , model : str ) -> tuple [dict , dict , dict ]:
765
+ def _prepare_request_params (
766
+ self , optional_params : dict , model : str
767
+ ) -> Tuple [dict , dict , dict ]:
766
768
"""Prepare and separate request parameters."""
767
769
inference_params = copy .deepcopy (optional_params )
768
770
supported_converse_params = list (
@@ -797,7 +799,13 @@ def _prepare_request_params(self, optional_params: dict, model: str) -> tuple[di
797
799
798
800
return inference_params , additional_request_params , request_metadata
799
801
800
- def _process_tools_and_beta (self , original_tools : list , model : str , headers : Optional [dict ], additional_request_params : dict ) -> tuple [List [ToolBlock ], list ]:
802
+ def _process_tools_and_beta (
803
+ self ,
804
+ original_tools : list ,
805
+ model : str ,
806
+ headers : Optional [dict ],
807
+ additional_request_params : dict ,
808
+ ) -> tuple [List [ToolBlock ], list ]:
801
809
"""Process tools and collect anthropic_beta values."""
802
810
bedrock_tools : List [ToolBlock ] = []
803
811
@@ -871,12 +879,16 @@ def _transform_request_helper(
871
879
)
872
880
873
881
# Prepare and separate parameters
874
- inference_params , additional_request_params , request_metadata = self ._prepare_request_params (optional_params , model )
882
+ inference_params , additional_request_params , request_metadata = (
883
+ self ._prepare_request_params (optional_params , model )
884
+ )
875
885
876
886
original_tools = inference_params .pop ("tools" , [])
877
887
878
888
# Process tools and collect beta values
879
- bedrock_tools , anthropic_beta_list = self ._process_tools_and_beta (original_tools , model , headers , additional_request_params )
889
+ bedrock_tools , anthropic_beta_list = self ._process_tools_and_beta (
890
+ original_tools , model , headers , additional_request_params
891
+ )
880
892
881
893
bedrock_tool_config : Optional [ToolConfigBlock ] = None
882
894
if len (bedrock_tools ) > 0 :
@@ -1157,9 +1169,7 @@ def apply_tool_call_transformation_if_needed(
1157
1169
1158
1170
return message , returned_finish_reason
1159
1171
1160
- def _translate_message_content (
1161
- self , content_blocks : List [ContentBlock ]
1162
- ) -> Tuple [
1172
+ def _translate_message_content (self , content_blocks : List [ContentBlock ]) -> Tuple [
1163
1173
str ,
1164
1174
List [ChatCompletionToolCallChunk ],
1165
1175
Optional [List [BedrockConverseReasoningContentBlock ]],
@@ -1174,9 +1184,9 @@ def _translate_message_content(
1174
1184
"""
1175
1185
content_str = ""
1176
1186
tools : List [ChatCompletionToolCallChunk ] = []
1177
- reasoningContentBlocks : Optional [
1178
- List [ BedrockConverseReasoningContentBlock ]
1179
- ] = None
1187
+ reasoningContentBlocks : Optional [List [ BedrockConverseReasoningContentBlock ]] = (
1188
+ None
1189
+ )
1180
1190
for idx , content in enumerate (content_blocks ):
1181
1191
"""
1182
1192
- Content is either a tool response or text
@@ -1297,9 +1307,9 @@ def _transform_response(
1297
1307
chat_completion_message : ChatCompletionResponseMessage = {"role" : "assistant" }
1298
1308
content_str = ""
1299
1309
tools : List [ChatCompletionToolCallChunk ] = []
1300
- reasoningContentBlocks : Optional [
1301
- List [ BedrockConverseReasoningContentBlock ]
1302
- ] = None
1310
+ reasoningContentBlocks : Optional [List [ BedrockConverseReasoningContentBlock ]] = (
1311
+ None
1312
+ )
1303
1313
1304
1314
if message is not None :
1305
1315
(
@@ -1312,12 +1322,12 @@ def _transform_response(
1312
1322
chat_completion_message ["provider_specific_fields" ] = {
1313
1323
"reasoningContentBlocks" : reasoningContentBlocks ,
1314
1324
}
1315
- chat_completion_message [
1316
- "reasoning_content"
1317
- ] = self . _transform_reasoning_content ( reasoningContentBlocks )
1318
- chat_completion_message [
1319
- "thinking_blocks"
1320
- ] = self . _transform_thinking_blocks ( reasoningContentBlocks )
1325
+ chat_completion_message ["reasoning_content" ] = (
1326
+ self . _transform_reasoning_content ( reasoningContentBlocks )
1327
+ )
1328
+ chat_completion_message ["thinking_blocks" ] = (
1329
+ self . _transform_thinking_blocks ( reasoningContentBlocks )
1330
+ )
1321
1331
chat_completion_message ["content" ] = content_str
1322
1332
if (
1323
1333
json_mode is True
0 commit comments