@@ -1155,7 +1155,9 @@ def generate_client_response_streaming(
11551155) -> Generator [YieldMessage , Any , tuple [Message , Any ]]:
11561156 msg_stream : Generator [Message , Any , Any ]
11571157 assert isinstance (block .model , str ) # block is a "concrete block"
1158- assert isinstance (block .parameters , dict ) # block is a "concrete block"
1158+ assert block .parameters is None or isinstance (
1159+ block .parameters , dict
1160+ ) # block is a "concrete block"
11591161 match block :
11601162 case BamModelBlock ():
11611163 model_input_str = messages_to_str (model_input )
@@ -1222,7 +1224,9 @@ def generate_client_response_single(
12221224 model_input : Messages ,
12231225) -> Generator [YieldMessage , Any , tuple [Message , Any ]]:
12241226 assert isinstance (block .model , str ) # block is a "concrete block"
1225- assert isinstance (block .parameters , dict ) # block is a "concrete block"
1227+ assert block .parameters is None or isinstance (
1228+ block .parameters , dict
1229+ ) # block is a "concrete block"
12261230 msg : Message
12271231 match block :
12281232 case BamModelBlock ():
@@ -1256,7 +1260,9 @@ def generate_client_response_batching( # pylint: disable=too-many-arguments
12561260 model_input : Messages ,
12571261) -> Generator [YieldMessage , Any , Message ]:
12581262 assert isinstance (block .model , str ) # block is a "concrete block"
1259- assert isinstance (block .parameters , dict ) # block is a "concrete block"
1263+ assert block .parameters is None or isinstance (
1264+ block .parameters , dict
1265+ ) # block is a "concrete block"
12601266 match block :
12611267 case BamModelBlock ():
12621268 model_input_str = messages_to_str (model_input )
0 commit comments