Skip to content

Commit 8e2ef13

Browse files
committed
Fix assertions
1 parent a3774a5 commit 8e2ef13

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

src/pdl/pdl_interpreter.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)