Skip to content

Commit 411d969

Browse files
committed
Use pragma lax no cover for defensive vendor_part_id checks
1 parent 8ecbca0 commit 411d969

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

pydantic_ai_slim/pydantic_ai/_parts_manager.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -251,12 +251,12 @@ def _handle_in_thinking_mode(
251251
yield self.handle_thinking_delta(vendor_part_id=vendor_part_id, content=before_end)
252252

253253
# Close the thinking part
254-
if vendor_part_id is not None:
254+
if vendor_part_id is not None: # pragma: lax no cover
255255
self._vendor_id_to_part_index.pop(vendor_part_id, None)
256256
self._tag_buffer.pop(vendor_part_id, None)
257257

258258
# Process any remaining content after the end tag
259-
if after_end:
259+
if after_end: # pragma: lax no cover
260260
yield from self._handle_text_delta_with_thinking_tags(
261261
vendor_part_id=vendor_part_id,
262262
content=after_end,
@@ -266,11 +266,11 @@ def _handle_in_thinking_mode(
266266
)
267267
elif self._could_be_tag_start(combined_content, end_tag):
268268
# Might be start of end tag, buffer it
269-
if vendor_part_id is not None:
269+
if vendor_part_id is not None: # pragma: lax no cover
270270
self._tag_buffer[vendor_part_id] = combined_content
271271
else:
272272
# Not an end tag, add to thinking content
273-
if vendor_part_id is not None:
273+
if vendor_part_id is not None: # pragma: lax no cover
274274
self._tag_buffer.pop(vendor_part_id, None)
275275
yield self.handle_thinking_delta(vendor_part_id=vendor_part_id, content=combined_content)
276276

@@ -290,7 +290,7 @@ def _handle_start_of_response(
290290
after_start = combined_content[len(start_tag) :]
291291

292292
# Clear any state for this vendor_part_id and start thinking part
293-
if vendor_part_id is not None:
293+
if vendor_part_id is not None: # pragma: lax no cover
294294
self._vendor_id_to_part_index.pop(vendor_part_id, None)
295295
self._tag_buffer.pop(vendor_part_id, None)
296296
yield self.handle_thinking_delta(vendor_part_id=vendor_part_id, content='')
@@ -306,11 +306,11 @@ def _handle_start_of_response(
306306
)
307307
elif self._could_be_tag_start(combined_content, start_tag):
308308
# Might be start of start tag at position 0, buffer it
309-
if vendor_part_id is not None:
309+
if vendor_part_id is not None: # pragma: lax no cover
310310
self._tag_buffer[vendor_part_id] = combined_content
311311
else:
312312
# Not a start tag, process as normal text
313-
if vendor_part_id is not None:
313+
if vendor_part_id is not None: # pragma: lax no cover
314314
self._tag_buffer.pop(vendor_part_id, None)
315315
yield from self._handle_text_delta_simple(
316316
vendor_part_id=vendor_part_id,

0 commit comments

Comments
 (0)